Question

我想将类的字段转换为有效的常量。我使用的是ASM 5.0.3。

这是我的测试课程：

public class BytecodeUtilsTest {

  @BeforeClass
  public static void beforeClass(){
    String replaceFieldClassName = "com.mypackage.ClassWithFieldToReplaceWithConstant";
    String replaceFieldClassNameAsPath = replaceFieldClassName.replace('.', '/') + ".class";
    // standard code to redefine class (inspired by ASM FAQ http://asm.ow2.org/doc/faq.html, sec. 5)
    ClassLoader contextClassLoader = Thread.currentThread().getContextClassLoader();
    InputStream stream = contextClassLoader.getResourceAsStream(replaceFieldClassNameAsPath);
    byte[] classBytes;
    try {
      classBytes = IOUtils.toByteArray(stream);

      // here is the interesting part
      byte[] patchedClassBytes = BytecodeUtils.patch(classBytes, "_fieldToReplace", true);

      Reflection.invoke(contextClassLoader, "defineClass", Class.class,
              new Class[]{String.class, byte[].class, int.class, int.class},
              new Object[]{replaceFieldClassName, patchedClassBytes, 0, patchedClassBytes.length});
    } catch (IOException e) {
      throw new RuntimeException(e);
    }    
  }

  @Test
  public void testFieldReplace(){
    Assert.assertTrue(new ClassWithFieldToReplaceWithConstant().getFieldToReplace());
    Assert.assertTrue(new ClassWithFieldToReplaceWithConstant()._fieldToReplace);
  }
}

以下是要更新的测试类：

public class ClassWithFieldToReplaceWithConstant {
  boolean _fieldToReplace;

  public boolean getFieldToReplace() {
    return _fieldToReplace;
  }
}

以下是修补程序：

public class BytecodeUtils {
  public static byte[] patch(byte[] bytecode, final String fieldToReplace, final boolean value) {
    ClassReader classReader = new ClassReader(bytecode);
    final ClassWriter classWriter = new ClassWriter(classReader, 0);
    ClassVisitor classVisitor = new ClassVisitor(Opcodes.ASM4, classWriter) {
      @Override
      public MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) {
        return new MethodVisitor(Opcodes.ASM4, super.visitMethod(access, name, desc, signature, exceptions)) {
          @Override
          public void visitFieldInsn(int opcode, String owner, String name, String desc) {
            if (opcode == Opcodes.GETFIELD && name.equals(fieldToReplace)) {
              mv.visitInsn(Opcodes.POP);
              mv.visitInsn(value ? Opcodes.ICONST_1 : Opcodes.ICONST_0);
            } else {
              super.visitFieldInsn(opcode, owner, name, desc);
            }
          }
        };
      }
    };
    classReader.accept(classVisitor, 0);
    return classWriter.toByteArray();
  }    
}

问题是第二个断言测试失败。因此，如果我使用getter，它将按预期返回true，但如果我直接读取字段，则返回false。考虑到getter产生INVOKEVIRTUAL指令和字段读取产生GETFIELD这一事实，这是非常意外的，而visitInsn通过调用true方法进行更新。

我做错了什么以及如何让直接字段访问返回{{1}}？

Answer 1

要使第二个断言工作，你需要修补BytecodeUtilsTest，而不是ClassWithFieldToReplaceWithConstant，因为第二个案例的读取字段的字节码指令实际上是在BytecodeUtilsTest＃testFieldReplace方法中。

Getter案例工作正常，因为读取字段的指令在getter体内，即在ClassWithFieldToReplaceWithConstant类中。

如果在真实场景字段中是私有的，则此代码应该没问题（因为将无法从类字段外部对字段进行访问）。否则，您将必须修补每个读取或写入此字段的类。

如何使用ASM用常量值替换布尔字段

1 个答案: