java 探针修改字节码

通过java agent 探针和javassist 实现从字节码层面的aop功能
首先我们要创建一个探针jar包
项目结构如下：

image.png

• 1、创建入口类

package com.wan;

import java.lang.instrument.Instrumentation;

public class MyAgent {
    public static void premain(String agentOpts, Instrumentation instrumentation) {
        System.out.println("====premain 方法执行===");
        instrumentation.addTransformer(new InsertTransformer());
    }
    public static void premain(String agentOps){

        System.out.println("====premain方法执行2====");
        System.out.println(agentOps);
    }
    public static void main(String[] args) {
        // TODO Auto-generated method stub


    }
} 

• 2、创建MANIFEST.MF文件，指定Premain-Class类

Manifest-Version: 1.0
Premain-Class: com.wan.MyAgent
Can-Redefine-Classes: true

• 3、创建pom文件，让maven可以将MANIFEST.MF文件打入到jar包

    4.0.0

    org.example
    my-agent
    1.0-SNAPSHOT
    
        
            org.javassist
            javassist
            3.27.0-GA
        
    
    
        
            
                org.apache.maven.plugins
                maven-shade-plugin
                2.4.3
                
                    
                        package
                        
                            shade
                        
                        
                            
                                
                                    
                                        
                                        com.wan.MyAgent
                                    
                                
                            
                        
                    
                
            
            
                org.apache.maven.plugins
                maven-compiler-plugin
                
                    1.8
                    1.8
                
            
        
    

• 4、创建转换类

主要实现的作用就是在TongShuaiStockController的shareStockList ()方法一开始打印出我是before,关于javassit 可以查看相关资料 javassist使用全解析

这里有个关键是我们通过线程获得上下文的类加载器，不然会找不到 TongShuaiStockController类，因为项目是web项目，使用的是自定义的类加载器

package com.wan;

import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtMethod;
import javassist.LoaderClassPath;

import java.lang.instrument.ClassFileTransformer;
import java.lang.instrument.IllegalClassFormatException;
import java.security.ProtectionDomain;

public class InsertTransformer implements ClassFileTransformer {
    public final String injectedClassName = "com.haier.cbs.web.controller.stock.TongShuaiStockController";
    public final String methodName = "shareStockList";
//    public final String injectedClassName = "com.wan.spi.Solution";
//    public final String methodName = "oddEvenList";

    public byte[] transform(ClassLoader loader, String className, Class classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {
        className = className.replace("/", ".");
//        System.out.println(className);
        if (className.equals(injectedClassName)) {

            CtClass ctclass = null;
            try {
                ClassPool classPool = ClassPool.getDefault();
                //使用相同的类加载器，不然找不controller
                classPool.appendClassPath(new LoaderClassPath(this.getClass().getClassLoader()));
                ctclass = classPool.get(className);// 使用全称,用于取得字节码类<使用javassist>
                CtMethod ctmethod = ctclass.getDeclaredMethod(methodName);// 得到这方法实例
                ctmethod.insertBefore("System.out.println("我是before");");
                return ctclass.toBytecode();
            } catch (Exception e) {
                System.out.println(e.getMessage());
                e.printStackTrace();
            }
        }
        return null;
    }
}

• 5、使用maven进行打包得到jar包

image.png

• 6、在需要使用探针的项目中加入虚拟机启动参数

-javaagent:D:/workspace/my-agent/target/my-agent-1.0-SNAPSHOT.jar="1212"

这个方法就是我们要处理的方法：

image.png

在我们调用接口的时候就会先打印“我是before”。