javascript - 如何为给定的javascript生成调用图？

Question

我看过“ https://stackoverflow.com/questions/1385335/how-to-generate-function-call-graphs-for-javascript ”，并尝试过。如果你想得到一个抽象的语法树，它工作得很好。

不幸的是，闭包编译器似乎只提供--print_tree,--print_ast和--print_pass_graph. 它们都对我没有用。

我想看看哪个函数调用了其他函数的图表。

Answer 1

code2flow正是这样做的。完全披露，我开始了这个项目

跑步

$ code2flow source1.js source2.js -o out.gv

然后，用 graphviz 打开 out.gv

编辑：目前，该项目未维护。我建议在使用 code2flow 之前尝试不同的解决方案。

Answer 2

如果你过滤你的输出，closure --print_tree你会得到你想要的。

例如采取以下文件：

var fib = function(n) {
    if (n < 2) {
        return n;
    } else {
        return fib(n - 1) + fib(n - 2);
    }
};

console.log(fib(fib(5)));

过滤输出closure --print_tree

            NAME fib 1 
                FUNCTION  1 
                                    CALL 5 
                                        NAME fib 5 
                                        SUB 5 
                                            NAME a 5 
                                            NUMBER 1.0 5 
                                    CALL 5 
                                        NAME fib 5 
                                        SUB 5 
                                            NAME a 5 
                                            NUMBER 2.0 5 
        EXPR_RESULT 9 
            CALL 9 
                GETPROP 9 
                    NAME console 9 
                    STRING log 9 
                CALL 9 
                CALL 9 
                    NAME fib 9 
                    CALL 9 
                    CALL 9 
                        NAME fib 9 
                        NUMBER 5.0 9 

你可以看到所有的调用语句。

我编写了以下脚本来执行此操作。

./call_tree

#! /usr/bin/env sh
function make_tree() {
    closure --print_tree $1 | grep $1
}

function parse_tree() {
    gawk -f parse_tree.awk
}

if [[ "$1" = "--tree" ]]; then
    make_tree $2
else
    make_tree $1 | parse_tree
fi

parse_tree.awk

BEGIN {
    lines_c = 0
    indent_width = 4
    indent_offset = 0
    string_offset = ""
    calling = 0
    call_indent = 0
}

{
    sub(/\[source_file.*$/, "")
    sub(/\[free_call.*$/, "")
}

/SCRIPT/ {
    indent_offset = calculate_indent($0)
    root_indent = indent_offset - 1
}

/FUNCTION/ {
    pl  = get_previous_line()
    if (calculate_indent(pl) < calculate_indent($0))
        print pl
    print
}

{
    lines_v[lines_c] = $0
    lines_c += 1
}

{
    indent = calculate_indent($0)
    if (indent <= call_indent) {
        calling = 0
    }
    if (calling) {
        print
    }
}

/CALL/ {
    calling = 1
    call_indent = calculate_indent($0)
    print
}

/EXPR/{
    line_indent = calculate_indent($0)
    if (line_indent == root_indent) {
        if ($0 !~ /(FUNCTION)/) {
            print
        }
    }
}

function calculate_indent(line) {
    match(line, /^ */)
    return int(RLENGTH / indent_width) - indent_offset
}

function get_previous_line() {
    return lines_v[lines_c - 1]
}

Answer 3

我最终使用UglifyJS2和Dot/GraphViz管理了这个，结合了上述答案和链接问题的答案。

对我来说，缺少的部分是如何过滤解析的 AST。事实证明，UglifyJS 有 TreeWalker 对象，它基本上对 AST 的每个节点都应用了一个函数。这是我到目前为止的代码：

//to be run using nodejs
var UglifyJS = require('uglify-js')
var fs = require('fs');
var util = require('util');

var file = 'path/to/file...';
//read in the code
var code = fs.readFileSync(file, "utf8");
//parse it to AST
var toplevel = UglifyJS.parse(code);
//open the output DOT file
var out = fs.openSync('path/to/output/file...', 'w');
//output the start of a directed graph in DOT notation
fs.writeSync(out, 'digraph test{\n');

//use a tree walker to examine each node
var walker = new UglifyJS.TreeWalker(function(node){
    //check for function calls
    if (node instanceof UglifyJS.AST_Call) {
        if(node.expression.name !== undefined)
        {
        //find where the calling function is defined
        var p = walker.find_parent(UglifyJS.AST_Defun);

        if(p !== undefined)
        {
            //filter out unneccessary stuff, eg calls to external libraries or constructors
            if(node.expression.name == "$" || node.expression.name == "Number" || node.expression.name =="Date")
            {
                //NOTE: $ is from jquery, and causes problems if it's in the DOT file.
                //It's also very frequent, so even replacing it with a safe string
                //results in a very cluttered graph
            }
            else
            {

                fs.writeSync(out, p.name.name);
                fs.writeSync(out, " -> ");
                fs.writeSync(out, node.expression.name);
                fs.writeSync(out, "\n");
            }
        }
        else
        {
            //it's a top level function
            fs.writeSync(out, node.expression.name);
            fs.writeSync(out, "\n");
        }

    }
}
if(node instanceof UglifyJS.AST_Defun)
{
    //defined but not called
    fs.writeSync(out, node.name.name);
    fs.writeSync(out, "\n");
}
});
//analyse the AST
toplevel.walk(walker);

//finally, write out the closing bracket
fs.writeSync(out, '}');

我用node运行它，然后将输出通过

dot -Tpng -o graph_name.png dot_file_name.dot

笔记：

它提供了一个非常基本的图表——只有黑白，没有格式。

正如其他人提到eval的那样，它根本没有捕获 ajax，并且可能不是类似的东西。with

此外，就目前而言，它在图中包括：由其他函数调用的函数（以及因此调用其他函数的函数），独立调用的函数，以及已定义但未调用的函数。

由于这一切，它可能会遗漏相关的内容，或包含不相关的内容。不过，这是一个开始，似乎完成了我所追求的目标，也是最初导致我提出这个问题的原因。

Answer 4

https://github.com/mishoo/UglifyJS 允许访问 javascript 中的 ast。

ast.coffee

util = require 'util'
jsp = require('uglify-js').parser

orig_code = """

var a = function (x) {
  return x * x;
};

function b (x) {
  return a(x)
}

console.log(a(5));
console.log(b(5));

"""

ast = jsp.parse(orig_code)

console.log util.inspect ast, true, null, true