操纵Java字节代码

动态编译Java源文件在一般情况下，开发人员都是在程序运行之前就编写完成了全部的Java源代码并且成功编译。对有些应用来说，Java源代码的内容在运行时刻才能确定。这个时候就需要动态编译源代码来生成Java字节代码，再由JVM来加载执行。典型的场景是很多算法竞赛的在线评测系统（如PKUJudgeOnline），允许用户上传Java代码，由系统在后台编译、运行并进行判定。在动态编译Java源文件时，使用的做法是直接在程序中调用Java编译器。JSR199引入了Java编译器API。如果使用JDK6的话，可以通过此API来动态编译Java代码。比如下面的代码用来动态编译最简单的HelloWorld类。该Java类的代码是保存在一个字符串中的。publicclassCompilerTest{publicstaticvoidmain(String[]args)throwsException{Stringsource=publicclassMain{publicstaticvoidmain(String[]args){System.out.println(\HelloWorld!\);}};JavaCompilercompiler=ToolProvider.getSystemJavaCompiler();StandardJavaFileManagerfileManager=compiler.getStandardFileManager(null,null,null);StringSourceJavaObjectsourceObject=newCompilerTest.StringSourceJavaObject(Main,source);IterableextendsJavaFileObjectfileObjects=Arrays.asList(sourceObject);CompilationTasktask=compiler.getTask(null,fileManager,null,null,null,fileObjects);booleanresult=task.call();if(result){System.out.println(编译成功。);}}staticclassStringSourceJavaObjectextendsSimpleJavaFileObject{privateStringcontent=null;publicStringSourceJavaObject(Stringname,Stringcontent)??throwsURISyntaxException{super(URI.create(string:///+name.replace('.','/')+Kind.SOURCE.extension),Kind.SOURCE);this.content=content;}publicCharSequencegetCharContent(booleanignoreEncodingErrors)??throwsIOException{returncontent;}}}如果不能使用JDK6提供的Java编译器API的话，可以使用JDK中的工具类com.sun.tools.javac.Main，不过该工具类只能编译存放在磁盘上的文件，类似于直接使用javac命令。另外一个可用的工具是EclipseJDTCore提供的编译器。这是EclipseJava开发环境使用的增量式Java编译器，支持运行和调试有错误的代码。该编译器也可以单独使用。Play框架在内部使用了JDT的编译器来动态编译Java源代码。在开发模式下，Play框架会定期扫描项目中的Java源代码文件，一旦发现有修改，会自动编译Java源代码。因此在修改代码之后，刷新页面就可以看到变化。使用这些动态编译的方式的时候，需要确保JDK中的tools.jar在应用的CLASSPATH中。下面介绍一个例子，是关于如何在Java里面做四则运算，比如求出来(3+4)*7-10的值。一般的做法是分析输入的运算表达式，自己来模拟计算过程。考虑到括号的存在和运算符的优先级等问题，这样的计算过程会比较复杂，而且容易出错。另外一种做法是可以用JSR223引入的脚本语言支持，直接把输入的表达式当做JavaScript或是JavaFX脚本来执行，得到结果。下面的代码使用的做法是动态生成Java源代码并编译，接着加载Java类来执行并获取结果。这种做法完全使用Java来实现。privatestaticdoublecalculate(Stringexpr)throwsCalculationException{StringclassName=CalculatorMain;StringmethodName=calculate;Stringsource=publicclass+className+{publicstaticdouble+methodName+(){return+expr+;}};//省略动态编译Java源代码的相关代码，参见上一节booleanresult=task.call();if(result){ClassLoaderloader=Calculator.class.getClassLoader();try{Class?clazz=loader.loadClass(className);Methodmethod=clazz.getMethod(methodName,newClass?[]{});Objectvalue=method.invoke(null,newObject[]{});return(Double)value;}catch(Exceptione){thrownewCalculationException(内部错误。);}}else{thrownewCalculationException(错误的表达式。);}}上面的代码给出了使用动态生成的Java字节代码的基本模式，即通过类加载器来加载字节代码，创建Java类的对象的实例，再通过Java反射API来调用对象中的方法。Java字节代码增强Java字节代码增强指的是在Java字节代码生成之后，对其进行修改，增强其功能。这种做法相当于对应用程序的二进制文件进行修改。在很多Java框架中都可以见到这种实现方式。Java字节代码增强通常与Java源文件中的注解（annotation）一块使用。注解在Java源代码中声明了需要增强的行为及相关的元数据，由框架在运行时刻完成对字节代码的增强。Java字节代码增强应用的场景比较多，一般都集中在减少冗余代码和对开发人员屏蔽底层的实现细节上。用过JavaBeans的人可能对其中那些必须添加的getter/setter方法感到很繁琐，并且难以维护。而通过字节代码增强，开发人员只需要声明Bean中的属性即可，getter/setter方法可以通过修改字节代码来自动添加。用过JPA的人，在调试程序的时候，会发现实体类中被添加了一些额外的域和方法。这些域和方法是在运行时刻由JPA的实现动态添加的。字节代码增强在面向方面编程（AOP）的一些实现中也有使用。在讨论如何进行字节代码增强之前，首先介绍一下表示一个Java类或接口的字节代码的组织形式。类文件{0xCAFEBABE，小版本号，大版本号，常量池大小，常量池数组，访问控制标记，当前类信息，父类信息，实现的接口个数，实现的接口信息数组，域个数，域信息数组，方法个数，方法信息数组，属性个数，属性信息数组}如上所示，一个类或接口的字节代码使用的是一种松散的组织结构，其中所包含的内容依次排列。对于可能包含多个条目的内容，如所实现的接口、域、方法和属性等，是以数组来表示的。而在数组之前的是该数组中条目的个数。不同的内容类型，有其不同的内部结构。对于开发人员来说，直接操纵包含字节代码的字节数组的话，开发效率比较低，而且容易出错。已经有不少的开源库可以对字节代码进行修改或是从头开始创建新的Java类的字节代码内容。这些类库包括ASM、cglib、serp和BCEL等。使用这些类库可以在一定程度上降低增强字节代码的复杂度。比如考虑下面一个简单的需求，在一个Java类的所有方法执行之前输出相应的日志。熟悉AOP的人都知道，可以用一个前增强（beforeadvice）来解决这个问题。如果使用ASM的话，相关的代码如下：ClassReadercr=newClassReader(is);ClassNodecn=newClassNode();cr.accept(cn,0);for(Objectobject:cn.methods){MethodNodemn=(MethodNode)object;if(init.equals(mn.name)||clinit.equals(mn.name)){continue;}InsnListinsns=mn.instructions;InsnListil=newInsnList();il.add(newFieldInsnNode(GETSTATIC,java/lang/System,out,Ljava/io/PrintStream;));il.add(newLdcInsnNode(Entermethod-+mn.name));il.add(newMethodInsnNode(INVOKEVIRTUAL,java/io/PrintStream,println,(Ljava/lang/String;)V));insns.insert(il);mn.maxStack+=3;}ClassWritercw=newClassWriter(0);cn.accept(cw);byte[]b=cw.toByteArray();从ClassWriter就可以获取到包含增强之后的字节代码的字节数组，可以把字节代码写回磁盘或是由类加载器直接使用。上述示例中，增强部分的逻辑比较简单，只是遍历Java类中的所有方法并添加对System.out.println方法的调用。在字节代码中，Java方法体是由一系列的指令组成的。而要做的是生成调用System.out.println方法的指令，并把这些指令插入到指令集合的最前面。ASM对这些指令做了抽象，不过熟悉全部的指令比较困难。ASM提供了一个工具类ASMifierClassVisitor，可以打印出Java类的字节代码的结构信息。当需要增强某个类的时候，可以先在源代码上做出修改，再通过此工具类来比较修改前后的字节代码的差异，从而确定该如何编写增强的代码。对类文件进行增强的时机是需要在Java源代码编译之后，在JVM执行之前。比较常见的做法有：由IDE在完成编译操作之后执行。如GoogleAppEngine的Eclipse插件会在编译之后运行DataNucleus来对实体类进行增强。在构建过程中完成，比如通过Ant或Maven来执行相关的操作。实现自己的Java类加载器。当获取到Java类的字节代码之后，先进行增强处理，再从修改过的字节代码中定义出Java类。通过JDK5引入的java.lang.instrument包来完成。java.lang.instrument由于存在着大量对Java字节代码进行修改的需求，JDK5引入了java.lang.instrument包并在JDK6中得到了进一步的增强。基本的思路是在JVM启动的时候添加一些代理（agent）。每个代理是一个jar包，其清单（manifest）文件中会指定一个代理类。这个类会包含一个premain方法。JVM在启动的时候会首先执行代理类的premain方法，再执行Java程序本身的main方法。在premain方法中就可以对程序本身的字节代码进行修改。JDK6中还允许在JVM启动之后动态添加代理。java.lang.instrument包支持两种修改的场景，一种是重定义一个Java类，即完全替换一