Java深度歷險(xiǎn)（一）——Java字節(jié)代碼的操縱作者 成富 發(fā)布于 2010年12月20日 【編者按】Java作為業(yè)界應(yīng)用最為廣泛的語(yǔ)言之一，深得眾多軟件廠(chǎng)商和開(kāi)發(fā)者的推崇，更是被包括Oracle在內(nèi)的眾多JCP成 員積極地推動(dòng)發(fā)展。但是對(duì)于Java語(yǔ)言的深度理解和運(yùn)用，畢竟是很少會(huì)有人涉及的話(huà)題。InfoQ中文站特地邀請(qǐng)IBM高級(jí)工程師成富為大家撰寫(xiě)這個(gè) 《Java深度歷險(xiǎn)》專(zhuān)欄，旨在就Java的一些深度和高級(jí)特性分享他的經(jīng)驗(yàn)。 在一般的Java應(yīng)用開(kāi)發(fā)過(guò)程中，開(kāi)發(fā)人員使用Java的方式比較簡(jiǎn)單。打開(kāi)慣用的IDE，編寫(xiě)Java源代碼，再利用IDE提供的功能直接運(yùn)行 Java 程序就可以了。這種開(kāi)發(fā)模式背后的過(guò)程是：開(kāi)發(fā)人員編寫(xiě)的是Java源代碼文件（.java），IDE會(huì)負(fù)責(zé)調(diào)用Java的編譯器把Java源代碼編譯成 平臺(tái)無(wú)關(guān)的字節(jié)代碼（byte code），以類(lèi)文件的形式保存在磁盤(pán)上（.class）。Java虛擬機(jī)（JVM）會(huì)負(fù)責(zé)把Java字節(jié)代碼加載并執(zhí)行。Java通過(guò)這種方式來(lái)實(shí)現(xiàn)其 “編寫(xiě)一次，到處運(yùn)行（Write once, run anywhere）” 的目標(biāo)。Java類(lèi)文件中包含的字節(jié)代碼可以被不同平臺(tái)上的JVM所使用。Java字節(jié)代碼不僅可以以文件形式存在于磁盤(pán)上，也可以通過(guò)網(wǎng)絡(luò)方式來(lái)下載， 還可以只存在于內(nèi)存中。JVM中的類(lèi)加載器會(huì)負(fù)責(zé)從包含字節(jié)代碼的字節(jié)數(shù)組（byte[]）中定義出Java類(lèi)。在某些情況下，可能會(huì)需要?jiǎng)討B(tài)的生成 Java字節(jié)代碼，或是對(duì)已有的Java字節(jié)代碼進(jìn)行修改。這個(gè)時(shí)候就需要用到本文中將要介紹的相關(guān)技術(shù)。首先介紹一下如何動(dòng)態(tài)編譯Java源文件。 動(dòng)態(tài)編譯Java源文件在一般情況下，開(kāi)發(fā)人員都是在程序運(yùn)行之前就編寫(xiě)完成了全部的Java源代碼并且成功編譯。對(duì)有些應(yīng)用來(lái)說(shuō)，Java源代碼的內(nèi)容在運(yùn)行時(shí)刻才能確 定。這個(gè)時(shí)候就需要?jiǎng)討B(tài)編譯源代碼來(lái)生成Java字節(jié)代碼，再由JVM來(lái)加載執(zhí)行。典型的場(chǎng)景是很多算法競(jìng)賽的在線(xiàn)評(píng)測(cè)系統(tǒng)（如PKU JudgeOnline），允許用戶(hù)上傳Java代碼，由系統(tǒng)在后臺(tái)編譯、運(yùn)行并進(jìn)行判定。在動(dòng)態(tài)編譯Java源文件時(shí)，使用的做法是直接在程序中調(diào)用Java編譯器。 JSR 199引入了Java編譯器API。如果使用JDK 6的話(huà)，可以通過(guò)此API來(lái)動(dòng)態(tài)編譯Java代碼。比如下面的代碼用來(lái)動(dòng)態(tài)編譯最簡(jiǎn)單的Hello World類(lèi)。該Java類(lèi)的代碼是保存在一個(gè)字符串中的。 public class CompilerTest {
public static void main(String[] args) throws Exception {
String source = "public class Main { public static void main(String[] args) {System.out.println(\"Hello World!\");} }";
JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
StandardJavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);
StringSourceJavaObject sourceObject = new CompilerTest.StringSourceJavaObject("Main", source);
Iterable< extends JavaFileObject> fileObjects = Arrays.asList(sourceObject);
CompilationTask task = compiler.getTask(null, fileManager, null, null, null, fileObjects);
boolean result = task.call();
if (result) {
System.out.println("編譯成功。");
}
}
static class StringSourceJavaObject extends SimpleJavaFileObject {
private String content = null;
public StringSourceJavaObject(String name, String content) ??throws URISyntaxException {
super(URI.create("string:///" + name.replace('.','/') + Kind.SOURCE.extension), Kind.SOURCE);
this.content = content;
}
public CharSequence getCharContent(boolean ignoreEncodingErrors) ??throws IOException {
return content;
}
}
}
如果不能使用JDK 6提供的Java編譯器API的話(huà)，可以使用JDK中的工具類(lèi)com.sun.tools.javac.Main，不過(guò)該工具類(lèi)只能編譯存放在磁盤(pán)上的文件，類(lèi)似于直接使用javac命令。 另外一個(gè)可用的工具是Eclipse JDT Core提供的編譯器。這是Eclipse Java開(kāi)發(fā)環(huán)境使用的增量式Java編譯器，支持運(yùn)行和調(diào)試有錯(cuò)誤的代碼。該編譯器也可以單獨(dú)使用。Play框架在 內(nèi)部使用了JDT的編譯器來(lái)動(dòng)態(tài)編譯Java源代碼。在開(kāi)發(fā)模式下，Play框架會(huì)定期掃描項(xiàng)目中的Java源代碼文件，一旦發(fā)現(xiàn)有修改，會(huì)自動(dòng)編譯 Java源代碼。因此在修改代碼之后，刷新頁(yè)面就可以看到變化。使用這些動(dòng)態(tài)編譯的方式的時(shí)候，需要確保JDK中的tools.jar在應(yīng)用的 CLASSPATH中。 下面介紹一個(gè)例子，是關(guān)于如何在Java里面做四則運(yùn)算，比如求出來(lái)(3+4)*7-10的值。一般的做法是分析輸入的運(yùn)算表達(dá)式，自己來(lái)模擬計(jì)算過(guò)程?？紤]到括號(hào)的存在和運(yùn)算符的優(yōu)先級(jí)等問(wèn)題，這樣的計(jì)算過(guò)程會(huì)比較復(fù)雜，而且容易出錯(cuò)。另外一種做法是可以用JSR 223引入的腳本語(yǔ)言支持，直接把輸入的表達(dá)式當(dāng)做JavaScript或是JavaFX腳本來(lái)執(zhí)行，得到結(jié)果。下面的代碼使用的做法是動(dòng)態(tài)生成Java源代碼并編譯，接著加載Java類(lèi)來(lái)執(zhí)行并獲取結(jié)果。這種做法完全使用Java來(lái)實(shí)現(xiàn)。 private static double calculate(String expr) throws CalculationException {
String className = "CalculatorMain";
String methodName = "calculate";
String source = "public class " + className
+ " { public static double " + methodName + "() { return " + expr + "; } }";
//省略動(dòng)態(tài)編譯Java源代碼的相關(guān)代碼，參見(jiàn)上一節(jié)
boolean result = task.call();
if (result) {
ClassLoader loader = Calculator.class.getClassLoader();
try {
Class<?> clazz = loader.loadClass(className);
Method method = clazz.getMethod(methodName, new Class<?>[] {});
Object value = method.invoke(null, new Object[] {});
return (Double) value;
} catch (Exception e) {
throw new CalculationException("內(nèi)部錯(cuò)誤。");
}
} else {
throw new CalculationException("錯(cuò)誤的表達(dá)式。");
}
}
上面的代碼給出了使用動(dòng)態(tài)生成的Java字節(jié)代碼的基本模式，即通過(guò)類(lèi)加載器來(lái)加載字節(jié)代碼，創(chuàng)建Java類(lèi)的對(duì)象的實(shí)例，再通過(guò)Java反射API來(lái)調(diào)用對(duì)象中的方法。 Java字節(jié)代碼增強(qiáng)Java 字節(jié)代碼增強(qiáng)指的是在Java字節(jié)代碼生成之后，對(duì)其進(jìn)行修改，增強(qiáng)其功能。這種做法相當(dāng)于對(duì)應(yīng)用程序的二進(jìn)制文件進(jìn)行修改。在很多Java框架中都可以 見(jiàn)到這種實(shí)現(xiàn)方式。Java字節(jié)代碼增強(qiáng)通常與Java源文件中的注解（annotation）一塊使用。注解在Java源代碼中聲明了需要增強(qiáng)的行為及 相關(guān)的元數(shù)據(jù)，由框架在運(yùn)行時(shí)刻完成對(duì)字節(jié)代碼的增強(qiáng)。Java字節(jié)代碼增強(qiáng)應(yīng)用的場(chǎng)景比較多，一般都集中在減少冗余代碼和對(duì)開(kāi)發(fā)人員屏蔽底層的實(shí)現(xiàn)細(xì)節(jié) 上。用過(guò)JavaBeans的人可能對(duì)其中那些必須添加的getter/setter方法感到很繁瑣，并且難以維護(hù)。而通過(guò)字節(jié)代碼增強(qiáng)，開(kāi)發(fā)人員只需要聲明Bean中的屬性即可，getter/setter方法可以通過(guò)修改字節(jié)代碼來(lái)自動(dòng)添加。用過(guò)JPA的人，在調(diào)試程序的時(shí)候，會(huì)發(fā)現(xiàn)實(shí)體類(lèi)中被添加了一些額外的 域和方法。這些域和方法是在運(yùn)行時(shí)刻由JPA的實(shí)現(xiàn)動(dòng)態(tài)添加的。字節(jié)代碼增強(qiáng)在面向方面編程（AOP）的一些實(shí)現(xiàn)中也有使用。 在討論如何進(jìn)行字節(jié)代碼增強(qiáng)之前，首先介紹一下表示一個(gè)Java類(lèi)或接口的字節(jié)代碼的組織形式。 類(lèi)文件 {
0xCAFEBABE，小版本號(hào)，大版本號(hào)，常量池大小，常量池?cái)?shù)組，
訪(fǎng)問(wèn)控制標(biāo)記，當(dāng)前類(lèi)信息，父類(lèi)信息，實(shí)現(xiàn)的接口個(gè)數(shù)，實(shí)現(xiàn)的接口信息數(shù)組，域個(gè)數(shù)，
域信息數(shù)組，方法個(gè)數(shù)，方法信息數(shù)組，屬性個(gè)數(shù)，屬性信息數(shù)組
}
如上所示，一個(gè)類(lèi)或接口的字節(jié)代碼使用的是一種松散的組織結(jié)構(gòu)，其中所包含的內(nèi)容依次排列。對(duì)于可能包含多個(gè)條目的內(nèi)容，如所實(shí)現(xiàn)的接口、域、方法 和屬性等，是以數(shù)組來(lái)表示的。而在數(shù)組之前的是該數(shù)組中條目的個(gè)數(shù)。不同的內(nèi)容類(lèi)型，有其不同的內(nèi)部結(jié)構(gòu)。對(duì)于開(kāi)發(fā)人員來(lái)說(shuō)，直接操縱包含字節(jié)代碼的字節(jié) 數(shù)組的話(huà)，開(kāi)發(fā)效率比較低，而且容易出錯(cuò)。已經(jīng)有不少的開(kāi)源庫(kù)可以對(duì)字節(jié)代碼進(jìn)行修改或是從頭開(kāi)始創(chuàng)建新的Java類(lèi)的字節(jié)代碼內(nèi)容。這些類(lèi)庫(kù)包括ASM、cglib、serp和BCEL等。使用這些類(lèi)庫(kù)可以在一定程度上降低增強(qiáng)字節(jié)代碼的復(fù)雜度。比如考慮下面一個(gè)簡(jiǎn)單的需求，在一個(gè)Java類(lèi)的所有方法執(zhí)行之前輸出相應(yīng)的日志。熟悉AOP的人都知道，可以用一個(gè)前增強(qiáng)（before advice）來(lái)解決這個(gè)問(wèn)題。如果使用ASM的話(huà)，相關(guān)的代碼如下： ClassReader cr = new ClassReader(is);
ClassNode cn = new ClassNode();
cr.accept(cn, 0);
for (Object object : cn.methods) {
MethodNode mn = (MethodNode) object;
if ("<init>".equals(mn.name) || "<clinit>".equals(mn.name)) {
continue;
}
InsnList insns = mn.instructions;
InsnList il = new InsnList();
il.add(new FieldInsnNode(GETSTATIC, "java/lang/System", "out", "Ljava/io/PrintStream;"));
il.add(new LdcInsnNode("Enter method -> " + mn.name));
il.add(new MethodInsnNode(INVOKEVIRTUAL, "java/io/PrintStream", "println", "(Ljava/lang/String;)V"));
insns.insert(il); mn.maxStack += 3;
}
ClassWriter cw = new ClassWriter(0);
cn.accept(cw);
byte[] b = cw.toByteArray();
從ClassWriter就 可以獲取到包含增強(qiáng)之后的字節(jié)代碼的字節(jié)數(shù)組，可以把字節(jié)代碼寫(xiě)回磁盤(pán)或是由類(lèi)加載器直接使用。上述示例中，增強(qiáng)部分的邏輯比較簡(jiǎn)單，只是遍歷Java類(lèi) 中的所有方法并添加對(duì)System.out.println方法的調(diào)用。在字節(jié)代碼中，Java方法體是由一系列的指令組成的。而要做的是生成調(diào)用 System.out.println方法的指令，并把這些指令插入到指令集合的最前面。ASM對(duì)這些指令做了抽象，不過(guò)熟悉全部的指令比較困難。ASM 提供了一個(gè)工具類(lèi)ASMifierClassVisitor，可以打印出Java類(lèi)的字節(jié)代碼的結(jié)構(gòu)信息。當(dāng)需要增強(qiáng)某個(gè)類(lèi)的時(shí)候，可以先在源代碼上做出修改，再通過(guò)此工具類(lèi)來(lái)比較修改前后的字節(jié)代碼的差異，從而確定該如何編寫(xiě)增強(qiáng)的代碼。 對(duì)類(lèi)文件進(jìn)行增強(qiáng)的時(shí)機(jī)是需要在Java源代碼編譯之后，在JVM執(zhí)行之前。比較常見(jiàn)的做法有：
java.lang.instrument由于存在著大量對(duì)Java字節(jié)代碼進(jìn)行修改的需求，JDK 5引入了java.lang.instrument包并在JDK 6中 得到了進(jìn)一步的增強(qiáng)?；镜乃悸肥窃贘VM啟動(dòng)的時(shí)候添加一些代理（agent）。每個(gè)代理是一個(gè)jar包，其清單（manifest）文件中會(huì)指定一個(gè) 代理類(lèi)。這個(gè)類(lèi)會(huì)包含一個(gè)premain方法。JVM在啟動(dòng)的時(shí)候會(huì)首先執(zhí)行代理類(lèi)的premain方法，再執(zhí)行Java程序本身的main方法。在 premain方法中就可以對(duì)程序本身的字節(jié)代碼進(jìn)行修改。JDK 6中還允許在JVM啟動(dòng)之后動(dòng)態(tài)添加代理。java.lang.instrument包支持兩種修改的場(chǎng)景，一種是重定義一個(gè)Java類(lèi)，即完全替換一個(gè) Java類(lèi)的字節(jié)代碼；另外一種是轉(zhuǎn)換已有的Java類(lèi)，相當(dāng)于前面提到的類(lèi)字節(jié)代碼增強(qiáng)。還是以前面提到的輸出方法執(zhí)行日志的場(chǎng)景為例，首先需要實(shí)現(xiàn)java.lang.instrument.ClassFileTransformer接口來(lái)完成對(duì)已有Java類(lèi)的轉(zhuǎn)換。 static class MethodEntryTransformer implements ClassFileTransformer {
public byte[] transform(ClassLoader loader, String className,
Class<?> classBeingRedefined, ?ProtectionDomain protectionDomain, byte[] classfileBuffer)
throws IllegalClassFormatException {
try {
ClassReader cr = new ClassReader(classfileBuffer);
ClassNode cn = new ClassNode();
//省略使用ASM進(jìn)行字節(jié)代碼轉(zhuǎn)換的代碼
ClassWriter cw = new ClassWriter(0);
cn.accept(cw);
return cw.toByteArray();
} catch (Exception e){
return null;
}
}
}
有了這個(gè)轉(zhuǎn)換類(lèi)之后，就可以在代理的premain方法中使用它。 public static void premain(String args, Instrumentation inst) {
inst.addTransformer(new MethodEntryTransformer());
}
把該代理類(lèi)打成一個(gè)jar包，并在jar包的清單文件中通過(guò)Premain-Class聲明代理類(lèi)的名稱(chēng)。運(yùn)行Java程序的時(shí)候，添加JVM啟動(dòng) 參數(shù)-javaagent:myagent.jar。這樣的話(huà)，JVM會(huì)在加載Java類(lèi)的字節(jié)代碼之前，完成相關(guān)的轉(zhuǎn)換操作。 總結(jié)操縱Java字節(jié)代碼是一件很有趣的事情。通過(guò)它，可以很容易的對(duì)二進(jìn)制分發(fā)的Java程序進(jìn)行修改，非常適合于性能分析、調(diào)試跟蹤和日志記錄等任 務(wù)。另外一個(gè)非常重要的作用是把開(kāi)發(fā)人員從繁瑣的Java語(yǔ)法中解放出來(lái)。開(kāi)發(fā)人員應(yīng)該只需要負(fù)責(zé)編寫(xiě)與業(yè)務(wù)邏輯相關(guān)的重要代碼。對(duì)于那些只是因?yàn)檎Z(yǔ)法要 求而添加的，或是模式固定的代碼，完全可以將其字節(jié)代碼動(dòng)態(tài)生成出來(lái)。字節(jié)代碼增強(qiáng)和源代碼生成是不同的概念。源代碼生成之后，就已經(jīng)成為了程序的一部 分，開(kāi)發(fā)人員需要去維護(hù)它：要么手工修改生成出來(lái)的源代碼，要么重新生成。而字節(jié)代碼的增強(qiáng)過(guò)程，對(duì)于開(kāi)發(fā)人員是完全透明的。妥善使用Java字節(jié)代碼的 操縱技術(shù)，可以更好的解決某一類(lèi)開(kāi)發(fā)問(wèn)題。 參考資料 |
|
|
