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包

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>

    <groupId>org.example</groupId>
    <artifactId>my-agent</artifactId>
    <version>1.0-SNAPSHOT</version>
    <dependencies>
        <dependency>
            <groupId>org.javassist</groupId>
            <artifactId>javassist</artifactId>
            <version>3.27.0-GA</version>
        </dependency>
    </dependencies>
    <build>
        <plugins>
            <plugin>
                <groupId>org.apache.maven.plugins</groupId>
                <artifactId>maven-shade-plugin</artifactId>
                <version>2.4.3</version>
                <executions>
                    <execution>
                        <phase>package</phase>
                        <goals>
                            <goal>shade</goal>
                        </goals>
                        <configuration>
                            <transformers>
                                <transformer
                                        implementation="org.apache.maven.plugins.shade.resource.ManifestResourceTransformer">
                                    <manifestEntries>
                                        <!--指明包含 premain 方法的类名，否则打包出来的文件会找不到 MANIFEST.MF -->
                                        <Premain-Class>com.wan.MyAgent</Premain-Class>
                                    </manifestEntries>
                                </transformer>
                            </transformers>
                        </configuration>
                    </execution>
                </executions>
            </plugin>
            <plugin>
                <groupId>org.apache.maven.plugins</groupId>
                <artifactId>maven-compiler-plugin</artifactId>
                <configuration>
                    <source>1.8</source>
                    <target>1.8</target>
                </configuration>
            </plugin>
        </plugins>
    </build>
</project>

• 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”。