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

我想同时使用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.

最佳答案

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();