在pre-main方法中使用javassist在运行时重命名字段(java instrumentation)

Question

我想在运行时重命名java类中的字段.此外,任何访问该字段的方法;读取或写入; 我需要将其修改为使用新名称而不是旧名称....

所有这些都将在pre-main方法中完成......

作为Exmaple,给出以下代码:

public class Class1
{
    String strCompany;

    public String Test()
    {
         strCompany = "TestCompany";
         return strCompany;
    }
}

在上面的类中,我需要将字段"strCompany"更改为"strCompany2",另外我需要使用方法Test来使用新名称而不是旧名称....

可以使用ctField类中的setName方法更改字段名称,但是如何修改方法体以使用新名称.

Answer 1

好吧，我的答案很晚，但我希望你仍然觉得它有用（或者至少对需要这种东西的其他人）。

尽管您可以使用像 Raphw 在评论中建议的低级别字节码 api，javassist 确实允许您使用更高级别的 API（我推荐）来执行此操作。

我将在下面介绍的解决方案将更改字段名称，并将所有引用从旧字段名称更改为新字段名称，这可能是您想要的，因为您正在重命名该字段。

代码

让我们使用您的 Class1 示例。

  ClassPool classpool = ClassPool.getDefault();
  CtClass ctClass = classpool.get(Class1.class.getName());
  CtField field = ctClass.getField("strCompany");
  CodeConverter codeConverter = new CodeConverter();
  codeConverter.redirectFieldAccess(field, ctClass, "strCompany2");
  ctClass.instrument(codeConverter);

  field.setName("strCompany2");
  ctClass.writeFile("./injectedClasses");

我假设访问 CtField 并设置其名称 - 由于您的问题 - 您已经知道如何做到这一点。“重新连接”所有字段引用的技巧是使用 CodeConverter 完成的，它将 替换对 ctField字段的所有引用，以替换对ctClass中名为 strCompany2 的字段的引用（恰好是同一个类）。请记住，这需要在将该字段重命名为 strCompany2之前完成。

在此运行结束时，您将在injectedClasses 文件夹中获得新的Class1，准备好使用strCompany2 而不是strCompany。:-)

边注

请记住，CodeConverter 真正所做的是在类常量池中创建一个新条目，并将所有引用从有关旧字段的条目重新路由到定义“新”（读取重命名）字段的条目。

因此，在 Class1 示例中，会发生以下情况：

注入前的常量池

Constant pool:
#1 = Class              #2             //  test/Class1
#2 = Utf8               test/Class1
#3 = Class              #4             //  java/lang/Object
#4 = Utf8               java/lang/Object
#5 = Utf8               strCompany
#6 = Utf8               Ljava/lang/String;
#7 = Utf8               <init>
#8 = Utf8               ()V
#9 = Utf8               Code
#10 = Methodref          #3.#11         //  java/lang/Object."<init>":()V
#11 = NameAndType        #7:#8          //  "<init>":()V
#12 = Utf8               LineNumberTable
#13 = Utf8               LocalVariableTable
#14 = Utf8               this
#15 = Utf8               Ltest/Class1;
#16 = Utf8               test
#17 = Utf8               ()Ljava/lang/String;
#18 = String             #19            //  TestCompany
#19 = Utf8               TestCompany
#20 = Fieldref           #1.#21         // test/Class1.strCompany:Ljava/lang/String;
#21 = NameAndType        #5:#6          //  strCompany:Ljava/lang/String;
#22 = Utf8               SourceFile
#23 = Utf8               Class1.java

注入后的常量池

Constant pool:
#1 = Class              #2             //  test/Class1
#2 = Utf8               test/Class1
#3 = Class              #4             //  java/lang/Object
#4 = Utf8               java/lang/Object
#5 = Utf8               strCompany
#6 = Utf8               Ljava/lang/String;
#7 = Utf8               <init>
#8 = Utf8               ()V
#9 = Utf8               Code
#10 = Methodref          #3.#11         //  java/lang/Object."<init>":()V
#11 = NameAndType        #7:#8          //  "<init>":()V
#12 = Utf8               LineNumberTable
#13 = Utf8               LocalVariableTable
#14 = Utf8               this
#15 = Utf8               Ltest/Class1;
#16 = Utf8               test
#17 = Utf8               ()Ljava/lang/String;
#18 = String             #19            //  TestCompany
#19 = Utf8               TestCompany
#20 = Fieldref           #1.#21         // test/Class1.strCompany:Ljava/lang/String; 
#21 = NameAndType        #5:#6          //  strCompany:Ljava/lang/String;
#22 = Utf8               SourceFile
#23 = Utf8               Class1.java
#24 = Utf8               strCompany2
#25 = NameAndType        #24:#6         //  strCompany2:Ljava/lang/String;
#26 = Fieldref           #1.#25         //test/Class1.strCompany2:Ljava/lang/String;

在这种情况下，通过单个字段重写，你的constantPool增长了3个帧，它们代表了新字段的定义。通常这不是问题，但我宁愿提前提到它。