从 AST 获取方法调用信息

Question

如何使用 AST（抽象语法树）解析器获取程序的每个方法声明中调用的方法的名称？到目前为止，我已经成功获取了方法声明的所有名称以及正在调用的方法的所有名称，但我想知道哪个方法调用了哪个方法。例如，我想看到方法 m1 调用方法 mA 和 mB，而方法 m2 调用方法 < code>mC 和 mD 等。

[编辑 11/9/2011 IDB，将新手的扩展评论转录回原始问题的正文中。我希望我已经正确地转录了它。我希望作者回来并根据需要进行修改]：

我的问题似乎是（Eclipse 的）MethodDeclaration api 没有可以调用的 GetInvokedMethodName 函数。这是我的代码：

 public class MethodVisitor extends ASTVisitor { 

         List<MethodDeclaration> methods = new ArrayList<MethodDeclaration>();

         @Override public boolean visit(MethodDeclaration node) { 
             methods.add(node); 
             return super.visit(node); } 

         public List<MethodDeclaration> getMethods() 
             { return methods; }

         List<MethodInvocation> methods1 = new ArrayList<MethodInvocation>();

         @Override public boolean visit(MethodInvocation node)
             { methods1.add(node); 
               return super.visit(node); } 

        public List<MethodInvocation> getMethods1() 
             { return methods1; } 
        } 

    ...

    for (MethodDeclaration method : visitor .getMethods()) 
        { System.out.println("Method name: " + method.getName() 
                            + " Return type: " + method.getReturnType2() 
                            + " Is constructor: " + method.isConstructor()
                            + " Method invoked: " + ASTNode.METHOD_INVOCATION );
           ); }

    for (MethodInvocation method1 : visitor .getMethods1())
           { System.out.println("Method name invoked: " + method1.getName() ); }

Answer 1

我也有同样的问题。这是我的解决方案：

final HashMap<MethodDeclaration, ArrayList<MethodInvocation>> invocationsForMethods =
    new HashMap<MethodDeclaration, ArrayList<MethodInvocation>>();

        CompilationUnit cu = (CompilationUnit) ap.createAST(null);
        cu.accept(new ASTVisitor() {

            private MethodDeclaration activeMethod;

            @Override
            public boolean visit(MethodDeclaration node) {
                activeMethod = node;
                return super.visit(node);
            }

            @Override
            public boolean visit(MethodInvocation node) {
                if (invocationsForMethods.get(activeMethod) == null) {
                    invocationsForMethods.put(activeMethod, new ArrayList<MethodInvocation>());
                }
                invocationsForMethods.get(activeMethod).add(node);
                return super.visit(node);
            }

        });

现在，我们可以要求 invokingsForMethods.keySet() 获取所使用的 AST 和 invokingsForMethods.get( key) 返回作为键给出的声明的所有方法调用。

Answer 2

如果您想知道 m1 调用了哪个特定方法 mB（在您庞大的类数组中所有名为“mB”的方法中），您需要的不仅仅是 AST。您需要一个完整的符号表，将每个符号使用绑定到与其匹配的可能定义。

计算这样一个符号表的过程对于许多语言来说都很困难，对于 Java 来说也非常困难（但不像 C++ 那么糟糕）。必须有人对如何在（本地）作用域、继承、重载、隐式转换等情况下查找标识符的规则进行编码，并且 Java 参考手册用了很大一部分内容试图解释这一点。您不想自己做这件事。

您真正需要的是一个完整的 Java 前端，对于您想要检查的每个方法都有 AST 和相应的符号表。我想，你可以从（Sun？）Java 编译器的接口（我个人不知道如何做到这一点）、Jikes 编译器、Eclipse Java AST（？）模块以及诸如此类的工具中获得这一点。作为我们的 Java 前端。另一种方法是处理类文件，其中包含 JVM 形式的方法调用，其优点是 JVM 指令都是利用符号表构建的。

如果您想计算 m1 调用 mA 调用 mQ 调用 .... mZ，您需要一个愿意立即读取整个源代码库的工具。编译器不会为您执行此操作，但您可以使用 Eclipse 或我们的前端来执行此操作。