修改Javassist中的行号

所以我最近一直在使用Javassist，我遇到了一个我无法找到答案的问题。 CtMethod的insertAt方法允许您在特定的行号处插入代码，但它是否会覆盖该行或保留它，如何使其与默认情况相反？ 我有一个应用程序，它在运行时之前使用Javassist修改源代码，基于XML文件中的“钩子”。 我想这样做，以便可以覆盖一条线，或者可以在线上方放置一条线而不是覆盖它。 显然有一些黑客的方法可以做到这一点，但我宁愿使用正确的方法。

Answer 1

简单的部分

CtMethod对象中存在的insertAt（int lineNumber，String src）方法允许在给定行中的代码之前注入在src中编写的代码。

例如，采用以下（简单）示例程序：

public class TestSubject { public static void main(String[] args) { TestSubject testSubject = new TestSubject(); testSubject.print(); } private void print() { System.out.println("One"); // line 9 System.out.println("Two"); // line 10 System.out.println("Three"); // line 11 } } 

通过简单编码（请记住，方法变量必须是print方法的CtMethod表示）：

  // notice that I said line 10, which is where the sysout of "two" is method.insertAt(10, true, "System.out.println(\"one and an half\");"); 

将在类中注入一个新的sysout指令。 新类的输出将是：

  one one and an half two three 

困难的部分

Javassist没有提供一种简单的方法来删除一行代码，所以如果你真的想要替换它，你就别无选择，只能破解你的方法。

怎么做？ 好吧，让我向你介绍你的新朋友（如果你还不知道）， CodeAttribute对象。

CodeAttribute对象负责保存表示方法流的字节码，此外该代码属性还有另一个名为LineNumberAttribute的属性，可帮助您将行号映射到字节码数组中。 所以总结这个对象有你需要的一切！

以下示例中的想法非常简单。 将字节码数组中的字节与应删除的行相关联，并用无操作代码替换字节。

方法再次是方法打印的CtMethod表示

  // let's erase the sysout "Two" int lineNumberToReplace = 10; // Access the code attribute CodeAttribute codeAttribute = method.getMethodInfo().getCodeAttribute(); // Access the LineNumberAttribute LineNumberAttribute lineNumberAttribute = (LineNumberAttribute) codeAttribute.getAttribute(LineNumberAttribute.tag); // Index in bytecode array where the instruction starts int startPc = lineNumberAttribute.toStartPc(lineNumberToReplace); // Index in the bytecode array where the following instruction starts int endPc = lineNumberAttribute.toStartPc(lineNumberToReplace+1); System.out.println("Modifying from " + startPc + " to " + endPc); // Let's now get the bytecode array byte[] code = codeAttribute.getCode(); for (int i = startPc; i < endPc; i++) { // change byte to a no operation code code[i] = CodeAttribute.NOP; } 

在原始 TestSubject类中运行此修改将导致注入类具有以下输出：

  one three 

加起来

当您需要添加一行并仍保留现有行时，您只需要使用easy部分中给出的示例，如果要替换该行，则必须先使用下面给出的示例删除现有行。 硬件然后使用第一个例子注入新行。

还要记住，在示例中，我假设您已经熟悉javassist的基础知识，只显示多汁的部分，而不是所有交易。 这就是为什么，例如，在示例中没有ctClass.writeFile ...你仍然需要这样做，我只是把它留下来因为我确实你应该知道你必须这样做。

如果您在代码示例中需要任何额外帮助，请询问。 我很乐意帮忙。