利用instrument+Attach API+javassist动态改变方法逻辑

1 instrument

instrument是jdk 1.5之后提供的一个功能，它通过代理的方式运行在 JVM 上的程序的服务。作为代理的类必须首先打成jar包。在jdk1.6中支持两种方式来启动代理:

(1) 在程序启动的时候添加-javaagent:jarpath=options参数指定代理的jar来启动代理，这种情况下

代理入口类通过在META-INF/MENIFEST.MF清单文件中的Premain-Class属性指定，代理入口类必须实现以下两个函数之一（两个方法同时存在时优先调用第一个）：

public static void premain(String agentArgs, Instrumentation inst);

或

public static void premain(String agentArgs);

(2)当目标程序已经在运行了，这种情况就只能采用第二种方式了。

首先必须在META-INF/MENIFEST.MF清单文件中通过Agent-Class属性来指定代理入口类。同样，代理入口类必须是实现以下两个函数之一（两个方法同时存在时优先调用第一个）：

public static void agentmain(String agentArgs, Instrumentation inst);

或

public static void agentmain(String agentArgs);

具体内容可以参考http://docs.oracle.com/javase/6/docs/api/java/lang/instrument/package-summary.html中的描述

这里需要介绍一下Instrumentation类，此类提供检测 Java 编程语言代码所需的服务。检测是向方法中添加字节码，以搜集各种工具所使用的数据。通过它，我们才能操作JVM，并修改Class中的一些内容。

这个类中有两个addTransformer方法，

void addTransformer(ClassFileTransformer transformer, boolean canRetransform)

该方法向Instrumentation注册一个转换类文件转换器。
ClassFileTransformer是一个接口，提供了一个

byte[] transform(ClassLoader loader,                 String className,                 Class<?> classBeingRedefined,                 ProtectionDomain protectionDomain,                 byte[] classfileBuffer)                 throws IllegalClassFormatException

该方法实现可以转换提供的类文件，并返回一个新的替换类文件。若不想替换类文件，可以返回null。

第二种方式显然已经不能再通过java命令的启动参数来指定了，这里就需要用到Attach API

2 Attach API

The Attach API 这篇文章中介绍得很详细了。

3.javassist

javassist是一个修改字节码创建Java字节码的类库。比asm工具容易上手。

4.动态改变类的行为

在运行的程序中出现问题时，有时候我们不希望停止程序来排查问题，这个时候可以通过动态的改变运行中的某些类的行为，或者打印某个值来排查。这个时候就需要我们能在程序运行过程中动态的改变类的行为。通过上面三种工具的结合，可以解决这个问题。
首先我们给出运行的目标程序：

public class HelloServiceImpl{    @Override    public void sayHello() {        System.out.println("hello");    }}

public class Client {    /**     * @param args     */    public static void main(String[] args) {        HelloServiceImpl service = new HelloServiceImpl();        while (true) {            service.sayHello();            try {                synchronized (service) {                    service.wait(3000);                }            } catch (InterruptedException e) {                e.printStackTrace();            }        }    }}

这里Client每隔3秒钟调用一次sayHello方法。输出结果为hello.
现在想在hello输出前再输出方法的名字。
agent：

public class MyAgent {    public static String className="com.alibaba.study.thread.HelloServiceImpl";    public static void agentmain(String args, Instrumentation inst) throws UnmodifiableClassException {        Class[] allClass = inst.getAllLoadedClasses();        for (Class c : allClass) {            if(c.getName().equals(className)){                System.out.println("agent loaded");                inst.addTransformer(new DynamicClassTransformer(), true);                inst.retransformClasses(c);            }        }    }}public class DynamicClassTransformer  implements ClassFileTransformer {    @Override    public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined,                            ProtectionDomain protectionDomain, byte[] classfileBuffer)                                                                                      throws IllegalClassFormatException {        try {            CtClass ctClass = ClassPool.getDefault().get("com.alibaba.study.thread.HelloServiceImpl");            String methodName = "sayHello";            CtMethod ctMethod=ctClass.getDeclaredMethod(methodName);            System.out.println(ctMethod.getName());            ctMethod.insertBefore("System.out.println(\" sayHello\");");            ctClass.writeFile();            return ctClass.toBytecode();        } catch (Exception e) {            System.out.println(e.getMessage());        }        return null;    }}

然后在配置清单中指定Agent-Class属性为MyAgent，并打成jar包。
将agent attach到指定的java虚拟机进程上。

public class AttachMain {    public static void main(String args[]) throws AttachNotSupportedException{        VirtualMachine vm;        List<VirtualMachineDescriptor> vmList= VirtualMachine.list();        if(vmList!=null){        for(int i=0;i<vmList.size();i++){            System.out.println("["+i+"]  "+vmList.get(i).displayName()+" ,id:"+vmList.get(i).id()+" ,provider:"+vmList.get(i).provider());        }        try{        int num=System.in.read()-48;        if(num!=-1&&num<vmList.size()){            vm= VirtualMachine.attach(vmList.get(num));            vm.loadAgent("/home/tanfeng/myagent.jar");            System.in.read();        }       }catch(Exception e){           e.printStackTrace();       }        }    }}

这样，当agent被加载之后，就可以看到在输出hello之前，会输出sayHello这个字符串。
这种功能相当强大，我们可以用它来做些其它用途。
当然我们没有必要自己去写这些程序了，因为已经有人帮我们做了，Btrace就是这样一款工具。