java - 如何在ASM java中将ClassVisitor转换为ClassNode

Question

我想同时使用 ASM Core 和 Tree API。目前，系统正在使用一定数量的ClassVisitor链接在一起来分析一个类文件。现在，在此链的末尾，我想添加另一个在 ClassNode 上工作的 Transformer。是否可以将ClassVisitor转换为ClassNode。

...
_cv = new FirstCV(_cv);
_cv = new SecondCV(_cv);
...
_cv = new ClassNodeTransformer((ClassNode) _cv); // Not sure how to achieve this step.

Answer 1

ClassVisitor 表示处理步骤，而 ClassNode 表示特定的类状态。要集成在 ClassNode 上运行的操作，您需要一个访问者，它在访问类时构建树，并在最后开始访问处理后的结果。

public abstract class IntegrateNodeTransformer extends ClassVisitor {
    ClassVisitor target;

    public IntegrateNodeTransformer(int api, ClassVisitor target) {
        super(api, null);
        this.target = target;
    }

    /**
     * Initiate the building of a tree when the visit of a class starts.
     */
    @Override
    public void visit(int version, int access, String name, String signature,
                      String superName, String[] interfaces) {
        super.cv = new ClassNode();
        super.visit(version, access, name, signature, superName, interfaces);
    }

    /**
     * On completion of visiting the source class, process the tree and initiate
     * visiting of the result by the target visitor, if there is one.
     */
    @Override
    public void visitEnd() {
        super.visitEnd();
        ClassNode source = (ClassNode)super.cv;
        ClassNode result = process(source);
        if(target != null) result.accept(target);
    }

    /**
     * Do the actual work with the Tree API.
     * May manipulate {@code source} directly and return it or return an entirely new tree.
     */
    protected abstract ClassNode process(ClassNode source);
}

如果您只有一个使用 Tree API 的处理步骤，您可以考虑直接执行它，而不将其改造为 ClassVisitor:

ClassReader cr = new ClassReader(…);
// steps before using the tree API

// mind that composed visitor process in the opposite order of their construction
ClassNode cn = new ClassNode();
ClassVisitor cv = new SomeClassVisitor(cn); // 3rd step
cv = new AnotherClassVisitor(cv);           // 2nd step
cv = new YetAnotherClassVisitor(cv);        // 1st step

cr.accept(cv, 0); // perform steps 1 to 3 and build result tree

cn = process(cn); // the Tree API based operation (4th step)

// steps after using the tree API
ClassWriter cw = new ClassWriter(cr, 0);    // last generate class
cv = new YourLastClassVisitor(cw);          // 7th step
cv = new InCaseYouHaveMoreClassVisitor(cv); // 6th step
cv = new YouKnowWhatIMean(cv);              // 5th step

cn.accept(cv); // perform steps 5 to 7 on tree input and generate class
final byte[] code = cw.toByteArray();