如何使用 ASM 修改常量池?

Question

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

Answer 1

正如此评论中所述，使用 ASM，您根本不处理常量池; 你只处理常量的实际用法。

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

因此，为了有效地"更改a"，您必须替换常量值的每个实际使用，但您必须意识到，您无法识别该值的出现是否真正源于对a的字段访问，或者只是对同一常量值的另一种使用。

示例代码中的用法很容易更改;您只需查找ldc说明。替换字段本身的常量值声明也相当容易。更涉及的是替换注释中的常量，但真正困难的是将常量用作switch语句的标签case因为在这种情况下替换常量，程序逻辑会中断。您必须重写更多代码才能使字符串切换使用不同的常量。

只关注较简单的任务，转换代码变为：

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