动态更改JVM里的class

近来需要完成一个feature：修改已load到JVM中的某个class，对其加一些代码，以此来动态修改运行中的程序。

对着这个feature我找到的方案是agent+Instrumentation+ASM

一路做下来有以下几点比较有意思：

1）动态attach agent到某个JVM进程

一般使用agent都是静态的，直接在运行某程序时加agent参数，这样agent会先于程序启动，这个不符合我的需求，我找到一个动态attach agent的方法，具体细节见以下代码：

public static void attach(String pid) throws Exception { try { String agentPath = "/cutemock-agent.jar"; String tmp = Main.class.getClassLoader().getResource("com/taobao/lp/cutemock/agent/Main.class").toString(); tmp = tmp.substring(0, tmp.indexOf("!")); tmp = tmp.substring("jar:".length(), tmp.lastIndexOf("/")); agentPath = tmp + agentPath; agentPath = new File(new URI(agentPath)).getAbsolutePath(); VirtualMachine vm = null; if (debug) { debugPrint("attaching to " + pid); } vm = VirtualMachine.attach(pid); if (debug) { debugPrint("attached to " + pid); } if (debug) { debugPrint("loading " + agentPath); } String agentArgs = "port=" + port; if (debug) { agentArgs += ",debug=true"; } if (debug) { debugPrint("agent args: " + agentArgs); } vm.loadAgent(agentPath, agentArgs); if (debug) { debugPrint("loaded " + agentPath); } } catch (RuntimeException re) { throw re; } catch (IOException ioexp) { throw ioexp; } catch (Exception exp) { throw exp; } }

这段代码的关键是要找到agent的jar包，然后通过VirtualMachine.attach和VirtualMachine.loadAgent把agent attach到pid上

2）通过Instrumentation修改已load了的class

见如下代码：

Class[] classes = inst.getAllLoadedClasses(); for(Class clazz : classes){ if(clazz.getName().equals(CLASS_NAME)){ System.out.println("add transformer to TBRemotingRPCProtocolComponent.class"); inst.addTransformer(new MyClassFileTransformer(),true); inst.retransformClasses(clazz); } }

关键在于inst.addTransformer(new MyClassFileTransformer(),true);这个true参数，inst.retransformClasses(clazz);只会重新修改addTransformer中canRetransform==true的

3）通过asm eclipse plugin方便修改class

大家都知道可以通过asm来修改class，但其api及其难用，比如我仅仅只想加一行：

targetURL = MockUtil.getTargetUrl(metadata.getUniqueName(), request.getMethodName(), targetURL);

翻译为asm：

mv.visitVarInsn(Opcodes.ALOAD, 2); mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, "com/taobao/hsf/model/metadata/ServiceMetadata", "getUniqueName", "()Ljava/lang/String;"); mv.visitVarInsn(Opcodes.ALOAD, 1); mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, "com/taobao/hsf/domain/HSFRequest", "getMethodName", "()Ljava/lang/String;"); mv.visitVarInsn(Opcodes.ALOAD, 3); mv.visitMethodInsn(Opcodes.INVOKESTATIC, "com/taobao/lp/cutemock/agent/MockUtil", "getTargetUrl", "(Ljava/lang/String;Ljava/lang/String;Ljava/lang/String;)Ljava/lang/String;"); mv.visitVarInsn(Opcodes.ASTORE, 3); Label l4 = new Label(); mv.visitLabel(l4);

但asm提供了一个eclipse plugin，更新地址为：http://andrei.gmxhome.de/eclipse/

它可以对比出修改前后的class的差异，并自动翻译为asm代码

以上是我这两天玩动态修改class的一些心得，有点乱，但确实是不断尝试后的心得