3

我已经从这篇文章中了解了如何使用 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
4

1 回答 1

8

正如此评论中所说，使用 ASM，您根本不需要处理常量池；您只处理常量的实际用途。

关于您的示例，您必须意识到声明 like
… final String a = "Hello World";是编译时常量。因此，在描述值的字段中会有一个ConstantValue属性，但是对常量的每次读取访问都将在编译时被替换。

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

示例代码中的用法很容易更改；你只需要寻找ldc说明。替换字段本身的常量值声明也相当容易。更多涉及的是在注释中替换常量，但真正很难将常量替换为语句的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();
}
于 2018-03-23T16:16:56.813 回答