使用 ASM 监控 java.lang.ProcessBuilder答案

【问题标题】：Using ASM to monitor java.lang.ProcessBuilder使用 ASM 监控 java.lang.ProcessBuilder
【发布时间】：2015-01-16 17:55:42
【问题描述】：

我希望创建一个使用 ASM 检测的监视器来记录对 java.lang.ProcessBuilder 的调用。但是，我的尝试似乎失败了。一些示例代码如下。基本上，这段代码只是在 java.lang.ProcessBuilder 的构造函数中添加了一个 nop，然后将新的（和原始的）类文件写入修改后的类目录。这完全是为了测试，因为我想确保一切正常，并且我可以在运行代理之外检查更改的字节码。如果我将此示例代码指向不同的（非 java.lang）类，则一切正常。是否可以使用 javaagent 修改 java.lang.ProcessBuilder？

提前致谢，

杰里米

public class Instrumentor {

public static void agentmain(String agentArgs, Instrumentation instrumentation) {
    instrument(instrumentation);
}

public static void premain(String agentArgument, Instrumentation instrumentation) {
    instrument(instrumentation);
}

private static void instrument(Instrumentation instrumentation) {
    try {
        instrumentation.addTransformer(new SampleTransformer());
    } catch (TransformerException ex) {
        Logger.getLogger(Instrumentor.class.getName()).log(Level.SEVERE, null, ex);
    }
}
}

//--------------------------------
public class SampleTransformer implements ClassFileTransformer {

public SampleTransformer() throws TransformerException {
}

@Override
public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined,
        ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {
    if ("java/lang/ProcessBuilder".equals(className)) {
        System.out.println("attempting to process " + className);
        ClassReader cr = new ClassReader(classfileBuffer);
        ClassWriter cw = new ClassWriter(cr, ClassWriter.COMPUTE_MAXS);
        SampleClassAdapter adapter = new SampleClassAdapter(cw);
        cr.accept(adapter, 0);
        byte[] clz = cw.toByteArray();
        try {
            System.out.println("write " + className);
            File p = new File("target\\modified-classes\\");
            p.mkdirs();
            File f = new File(p, className.replace('/', '.') + ".class");
            File o = new File(p, className.replace('/', '.') + ".original.class");
            Files.write(Paths.get(f.getAbsolutePath()), clz);
            Files.write(Paths.get(o.getAbsolutePath()), classfileBuffer);
        } catch (IOException ex) {
            Logger.getLogger(SampleTransformer.class.getName()).log(Level.SEVERE, null, ex);
        }
        return clz;
    }
    return classfileBuffer;
}
}
//------------------------
public class SampleClassAdapter extends ClassVisitor {

public SampleClassAdapter(ClassVisitor cv) {
    super(Opcodes.ASM5, cv);
}

@Override
public void visit(int version, int access, String name, String signature, String superName, String[] interfaces) {
    super.visit(version, access, name, signature, superName, interfaces);
}

@Override
public MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) {
    MethodVisitor mv = super.visitMethod(access, name, desc, signature, exceptions);
    if (mv != null && "<init>".equals(name)) {
        mv = new SampleMethodAdapter(mv, access, name, desc);
    }
    return mv;
}
}
//----------------------------------------
public class SampleMethodAdapter extends AdviceAdapter {

public SampleMethodAdapter(final MethodVisitor methodVisitor,
        final int access, final String name, final String descriptor) {
    super(Opcodes.ASM5, methodVisitor, access, name, descriptor);
}

@Override
protected void onMethodEnter() {
    mv.visitInsn(Opcodes.NOP);
}

}

【问题讨论】：

标签： java bytecode instrumentation java-bytecode-asm

【解决方案1】：

您可以在运行时修改类，但首先我建议您尝试编译您自己的版本，您可以在引导类路径上覆盖该版本。一旦你完成了这项工作，你就可以使用 ASM 来使代码注入在运行时工作。

【讨论】：

【解决方案2】：

我发现了问题所在。接受的 ClassReader 调用需要包含 EXPAND_FRAMES 标志。

cr.accept(adapter, ClassReader.EXPAND_FRAMES);

更正此调用后，我能够成功修改 java.lang.ProcessBuilder 的字节码。

【讨论】：