有时,我们需要在不修改源代码的前提下往一个第三方的JAVA程序里注入自己的代码逻辑。一种情况是拿不到它的源代码,另一种情况是即使有源代码也不想修改,想让注入的代码与第三方程序代码保持相对独立。
有两种方法可以让我们达到这样的目标。一种方法是使用JDK 1.5引入的Java Instrumentation API.
Instrumentation允许
一个独立于应用程序的代理程序(Agent),用来监测和协助运行在 JVM 上的程序,甚至能够替换和修改某些类的定义。另一种方法是编写一个定制的Class Loader,在合适的点注入自己的代码。
下面用一个简单的例子来描述一下如何用这两种方法分别来达到注入代码的目的。
A.java:
public
class A {
public void run() {
System.out.println( "A is running.");
}
}
public void run() {
System.out.println( "A is running.");
}
}
App.java:
public
class App {
public static void main(String... args) {
A a = new A();
a.run();
}
}
public static void main(String... args) {
A a = new A();
a.run();
}
}
我们的目的是替换Class A中的run方法。首先创建A的一个子类B,覆盖run方法:
B.java:
public
class B
extends A {
public void run() {
System.out.println( "B is running.");
}
}
public void run() {
System.out.println( "B is running.");
}
}
基本思路是在JVM load App类的时候,把对A的引用修改为对B的引用。我们甚至不用修改App的byte code,只需将App.class中常量池(constant pool)中类A的名字的字符串改为类B的名字,效果就是将语句A a = new A()改为A a = new B()。为了修改类的class文件,我们用到了一个开源的JAVA字节码操作和分析框架ASM (
http://asm.ow2.org/)。为了运行这个例子,下载
asm-4.0.jar到当前目录。