【基于javaagent premain快速运行版】

期望：当被代理的类传输电话号码的时候，对其字节码进行自定义处理

环境准备
IntelliJ IDEA Community Edition
jdk1.8.0_45 64位

置信息（路径相关修改为自己的）
配置位置：Run/Debug Configurations -> VM options
配置内容：-javaagent:/home/sq/IdeaProjects/agent-demo/target/agent-demo-1.0-SNAPSHOT.jar=ddddd

1. 创建agent-demo

创建PreMainAgent方法，使用bytebutty操作字节码

package com.itheima.agent;
 
 
import com.itheima.server.TimeInterceptor;
import net.bytebuddy.agent.builder.AgentBuilder;
import net.bytebuddy.description.method.MethodDescription;
import net.bytebuddy.description.type.TypeDescription;
import net.bytebuddy.dynamic.DynamicType;
import net.bytebuddy.implementation.MethodDelegation;
import net.bytebuddy.matcher.ElementMatchers;
import net.bytebuddy.utility.JavaModule;
 
import java.lang.instrument.Instrumentation;
 
public class PreMainAgent {
 
    /***
     * 执行方法拦截
     * @param agentArgs：-javaagent 命令携带的参数。在前面介绍 SkyWalking Agent 接入时提到
     *                 agent.service_name 这个配置项的默认值有三种覆盖方式，
     *                 其中，使用探针配置进行覆盖，探针配置的值就是通过该参数传入的。
     * @param inst：java.lang.instrumen.Instrumentation 是 Instrumention 包中定义的一个接口，它提供了操作类定义的相关方法。
     */
    public static void premain(String agentArgs, Instrumentation inst){
        if(agentArgs != null){
            System.out.println("agentArgs: "+agentArgs);
        }
        //动态构建操作，根据transformer规则执行拦截操作
        AgentBuilder.Transformer transformer = new AgentBuilder.Transformer() {
            @Override
            public DynamicType.Builder<?> transform(DynamicType.Builder<?> builder,
                                                    // 匹配上的具体的类型描述
                                                    TypeDescription typeDescription,
                                                    ClassLoader classLoader,
                                                    JavaModule javaModule) {
//                System.out.println("class:" + typeDescription.getName());
                //构建拦截规则
                return builder
                        //method()指定哪些方法需要被拦截，ElementMatchers.any()表示拦截所有方法
                        .method(ElementMatchers.<MethodDescription>any())
                        //intercept()指定拦截上述方法的拦截器
                        .intercept(MethodDelegation.to(TimeInterceptor.class));
            }
        };
 
        //采用Byte Buddy的AgentBuilder结合Java Agent处理程序
        new AgentBuilder
                //采用ByteBuddy作为默认的Agent实例
                .Default()
                //拦截匹配方式：类以com.itheima.driver开始（其实就是com.itheima.driver包下的所有类）
                .type(ElementMatchers.nameStartsWith("com.itheima.driver"))
                //拦截到的类由transformer处理
                .transform(transformer)
                //安装到 Instrumentation
                .installOn(inst);
 
    }
}

创建拦截器phoneNumberInterceptor对拦截到的类进行处理

package com.itheima.server;
 
import net.bytebuddy.implementation.bind.annotation.Origin;
import net.bytebuddy.implementation.bind.annotation.RuntimeType;
import net.bytebuddy.implementation.bind.annotation.SuperCall;
 
import java.lang.reflect.Method;
import java.util.concurrent.Callable;
 
public class phoneNumberInterceptor {
 
    /***
     * 拦截方法
     * @param method：拦截的方法
     * @param callable：调用对象的代理对象
     * @return
     * @throws Exception
     */
    @RuntimeType // 声明为static
    public static Object intercept(@Origin Method method,
                                   @SuperCall Callable<?> callable) throws Exception {
        //时间统计开始
        long start = System.currentTimeMillis();
        String phoneNumber = (String)callable.call();   // 执行原函数
        if (phoneNumber != null && phoneNumber.length() == 11) {
            StringBuilder encryptedPhoneNumber = new StringBuilder(phoneNumber);
            encryptedPhoneNumber.replace(3, 7, "****");
            phoneNumber = encryptedPhoneNumber.toString();
        }
 
        //执行时间统计
        System.out.println(method.getName() + ":执行耗时" + (System.currentTimeMillis() - start) + "ms");
        return phoneNumber;
    }
}

在pom.xml中配置自动生成META-INF/MANIFEST.MF

<?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>com.itheima</groupId>
    <artifactId>agent-demo</artifactId>
    <version>1.0-SNAPSHOT</version>

    <properties>
        <maven.compiler.source>8</maven.compiler.source>
        <maven.compiler.target>8</maven.compiler.target>
        <bytebuddy.version>1.11.17</bytebuddy.version>

        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
    </properties>

    <dependencies>
        <!-- ByteBuddy -->
        <dependency>
            <groupId>net.bytebuddy</groupId>
            <artifactId>byte-buddy</artifactId>
            <version>${bytebuddy.version}</version>
        </dependency>

        <!-- ByteBuddy Java Agent -->
        <dependency>
            <groupId>net.bytebuddy</groupId>
            <artifactId>byte-buddy-agent</artifactId>
            <version>${bytebuddy.version}</version>
        </dependency>
    </dependencies>


    <build>
        <plugins>
            <plugin>
                <artifactId>maven-assembly-plugin</artifactId>
                <configuration>
                    <appendAssemblyId>false</appendAssemblyId>
                    <descriptorRefs>
                        <descriptorRef>jar-with-dependencies</descriptorRef>
                    </descriptorRefs>
                    <archive> <!--自动添加META-INF/MANIFEST.MF -->
                        <manifest>
                            <!-- 添加 mplementation-*和Specification-*配置项-->
                            <addDefaultImplementationEntries>true</addDefaultImplementationEntries>
                            <addDefaultSpecificationEntries>true</addDefaultSpecificationEntries>
                        </manifest>
                        <manifestEntries>
                            <!--指定premain方法所在的类-->
                            <Premain-Class>com.itheima.agent.PreMainAgent</Premain-Class>
                            <Can-Redefine-Classes>true</Can-Redefine-Classes>
                            <Can-Retransform-Classes>true</Can-Retransform-Classes>
                        </manifestEntries>
                    </archive>
                </configuration>
                <executions>
                    <execution>
                        <id>make-assembly</id>
                        <phase>package</phase>
                        <goals>
                            <goal>single</goal>
                        </goals>
                    </execution>
                </executions>
            </plugin>
        </plugins>
    </build>
</project>

使用mvn clean install 打包

2. 创建agent-test类对javaagent进行测试

public class Application {

    public String generatePhoneNumber() {
        // 假设这个方法返回一个11位的电话号码
        return "12345678901";
    }

    public static void main(String[] args) {
        Application app = new Application();
        String phoneNumber = app.generatePhoneNumber();
        System.out.println("Original phone number: " + phoneNumber);
    }


}

使用mvn clean install 打包

进行测试

java -javaagent:/home/sq/IdeaProjects/agent-demo/target/agent-demo-1.0-SNAPSHOT.jar=xxx111 -jar target/agent-test.jar

参考文档：

https://my.oschina.net/jiagoushi/blog/5541311

https://www.cnblogs.com/crazymakercircle/p/16635330.html#autoid-h2-13-0-0

https://github.com/ShehanPerera/javaagent-bytebuddy