如何使用ASM修改常量池?

时间:2018-03-23 04:00:43

标签: java java-bytecode-asm jvm-bytecode

我已经了解了如何使用this post中的ASM在运行时操作类。

但我还有关于如何修改常量池的问题。下面是我想修改的示例java程序

主要jar文件:

    public class test {    
    private static final String a = "Hello World";
    private static final String b = "ASM is awasome";

    public static void main(String[] args) { 
         int x = 10;
         int y = 25;
         int z = x * y;
         System.out.println(a);
         System.out.println(z);
         System.out.println(b);

    }

}

我想将变量a"Hello World"修改为"Multiply Of x*y is: "

我的座席类

import java.lang.instrument.*;
import java.security.ProtectionDomain;
import org.objectweb.asm.*;

public class ExampleAgent implements ClassFileTransformer {
    private static final String TRANSFORM_CLASS = "src/test";
    private static final String TRANSFORM_METHOD_NAME = "main";
    private static final String TRANSFORM_METHOD_DESC = "([Ljava/lang/String;)V";

    public static void premain(String arg, Instrumentation instrumentation) {


        instrumentation.addTransformer(new ExampleAgent());

    }

    public byte[] transform(ClassLoader loader, String className, Class<?> cl,
                            ProtectionDomain pd, byte[] classfileBuffer) {
        if(!TRANSFORM_CLASS.equals(className)) return null;

        ClassReader cr = new ClassReader(classfileBuffer);
        ClassWriter cw = new ClassWriter(cr, 0);
        cr.accept(new ClassVisitor(Opcodes.ASM5, cw) {
            @Override
            public MethodVisitor visitMethod(int access, String name, String desc,
                                             String signature, String[] exceptions) {
                MethodVisitor mv = super.visitMethod(
                                       access, name, desc, signature, exceptions);


                //Code to modify the value of a




                return mv;
            }
        }, 0);

        return cw.toByteArray();

    }
}

控制台窗口上的结果应为

Multiply Of x*y is: 
250
ASM is awasome

1 个答案:

答案 0 :(得分:3)

正如在this comment中所述,使用ASM,你根本不处理常量池;你只处理常数的实际用途。

关于你的例子,你必须要知道像这样的声明 … final String a = "Hello World";是编译时常量。因此,在字段中将有一个ConstantValue attribute来描述该值,但是对该常量的每次读访问都将在编译时被替换。

所以要有效地“改变a”,你必须替换常量值的每个实际用法,但是你必须意识到你无法识别值的出现是否真正源于一个字段访问a或只是使用相同的常量值。

在示例代码中使用的代码很容易改变;你只需要寻找ldc instructions。替换字段本身的常量值声明也很容易。更多参与将是替换注释中的常量,但实际上很难将常量的使用替换为case语句的switch标签,因为如果您只是替换此类中的常量,程序逻辑将会中断场景。您必须重写更多代码才能使字符串切换使用不同的常量。

只关注更简单的任务,转换代码变为:

public byte[] transform(ClassLoader loader, String className, Class<?> cl,
                        ProtectionDomain pd, byte[] classfileBuffer) {
    if(!TRANSFORM_CLASS.equals(className)) return null;

    ClassReader cr = new ClassReader(classfileBuffer);
    ClassWriter cw = new ClassWriter(cr, 0);
    cr.accept(new ClassVisitor(Opcodes.ASM5, cw) {
        @Override
        public FieldVisitor visitField(int access, String name, String desc,
                                       String signature, Object cst) {
            if("Hello World".equals(cst)) cst = "Multiply Of x*y is: ";
            return super.visitField(access, name, desc, signature, cst);
        }

        @Override
        public MethodVisitor visitMethod(int access, String name, String desc,
                                         String signature, String[] exceptions) {
            MethodVisitor mv = super.visitMethod(
                                   access, name, desc, signature, exceptions);
            if(name.equals(TRANSFORM_METHOD_NAME)
            && desc.equals(TRANSFORM_METHOD_DESC)) {
                return new MethodVisitor(Opcodes.ASM5, mv) {
                    @Override
                    public void visitLdcInsn(Object cst) {
                        if("Hello World".equals(cst)) cst = "Multiply Of x*y is: ";
                        super.visitLdcInsn(cst);
                    }
                };
            }
            return mv;
        }
    }, 0);
    return cw.toByteArray();
}