Soot - 从 CFG 获取 JimpleBody

Soot - Get JimpleBody from a CFG

我想从 Java Class 得到 UnitGraph。 我通过 ClassFile 加载它并获得 main()method_info。 然后我创建一个 CFG 并尝试将其转换为 UnitGraph。 我的方法是获取CFGJimpleBody,然后创建一个UnitGraph。 但是,我无法通过调用 cfg.jimplify(...) 获得 JimpleBody,因为它会抛出以下错误:

Exception in thread "main" java.lang.RuntimeException: no method associated w/ body
    at soot.Body.getMethod(Body.java:137)
    at soot.coffi.CFG.jimplify(CFG.java:814)
    at com.LiveVariableAnalysis.main(LiveVariableAnalysis.java:151)

我的代码如下:

        String mainClassName = "Calculate";
        String mainClassPath = String.format("./target/test-classes/%s.class", mainClassName);
        ClassFile mainClassFile = new ClassFile(mainClassName);
        FileInputStream is = new FileInputStream(mainClassPath);
        mainClassFile.loadClassFile(is);
        logger.info(String.format("Loading Class: %s ...", mainClassFile));

        method_info methodInfo = null;
        for (method_info method: mainClassFile.methods) {
            if (Objects.equals(method.toName(mainClassFile.constant_pool), "main")) {
                methodInfo = method;
            }
        }
        logger.info(String.format("Loading method_info: %s ...", methodInfo.toName(mainClassFile.constant_pool)));

        mainClassFile.parseMethod(methodInfo);
        CFG cfg = new CFG(methodInfo);

        JimpleBody jimpleBody = new JimpleBody();
        // Error occurs here
        cfg.jimplify(mainClassFile.constant_pool, mainClassFile.this_class, mainClassFile.bootstrap_methods_attribute, jimpleBody);

        UnitGraph unitGraph = new ClassicCompleteUnitGraph(jimpleBody);
        logger.info(String.format("Creating unitGraph with %d units ...", unitGraph.size()));

我知道还有其他方法可以创建 UnitGraph,例如:

        String mainClassName = "Calculate";
        SootClass mainClass = Scene.v().loadClassAndSupport(className);
        Scene.v().setMainClass(mainClass);
        soot.Main.main(args);
        SootClass mainClass = Scene.v().getMainClass();
        String methodSignature = "void main(java.lang.String[])";
        SootMethod mainMethod = mainClass.getMethod(methodSignature);
        Body jimpleBody = mainMethod.retrieveActiveBody();
        UnitGraph unitGraph = new ClassicCompleteUnitGraph(jimpleBody);

但是,通过这种方式,我需要为 jce.jarrt.jar 设置 Scene.v().setSootClassPath(path),我不想在我的代码中出现这种情况。所以如果有另一种方法我可以在不设置这样的路径的情况下获得UnitGraph,请帮助我。

虽然我还是不能把method_info变成一个SootMathod然后得到一个UnitGraph,我可以用Options.v().set_prepend_classpath(true)来避免set jce.jarrt.jar 直接。这也达到了我的目的。