java agent使用

本文主要是介绍java agent使用，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

什么是java agent

本质是一个jar包中的类，有两种实现，第一种是通过permain()函数实现。这种javaagent会在宿主程序的main函数的启动前启动自己premain函数，这时候会得到一个Instrumentation对象，我们可以通过Instrumentation对象对还未加载的class进行拦截与修改。

还有一种实现方式是利用agentmain()函数。VirtualMachine类的attach(pid)方法可以将当前进程attach到一个运行中的java进程上，接着利用loadAgent(agentJarPath)来将含符合格式且含有agentmain函数的jar包注入到对应的进程，调用loadAgent函数后，对应的进程中会多出一个Instrumentation对象，这个对象会被当作agentmain的一个参数。对应进程接着会调用agentmain函数，进而操作Instrumentation对象，Instrumentation对象可以在class加载前拦截字节码进行修改，也可以对已经加载的class重新让它加载，并拦截且修改其中的内容，跟进程注入差不多，具体做什么操作，取决于我们的jar文件中的agentmain函数怎么写。

Java agent的使用方式有两种：

实现premain方法，在JVM启动前加载。

实现agentmain方法，在JVM启动后加载。

agent基础使用

环境搭建

agent项目源码

agent：

package com.naihe;import java.io.IOException;
import java.lang.instrument.*;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.security.ProtectionDomain;public class agent {//java agent 入口public static void premain(String agentOps, Instrumentation inst) {System.out.println("=========premain方法执行========");simpleDemo(agentOps, inst);}public static void agentmain(String agentOps, Instrumentation inst) {System.out.println("=========agentmain方法执行========");simpleDemo(agentOps, inst);//transform是会对尚未加载的类进行增加代理层，这里是已经运行中的jvm，所以类以及被加载了//必须主动调用retransformClasses让jvm再对运行中的类进行加上代理层for (Class allLoadedClass : inst.getAllLoadedClasses()) {//这里的Test路径，修改成你自己机器agent-demo-web工程的Test类的路径if(allLoadedClass.getName().contains("com.naihe.Demo")){try {inst.retransformClasses(allLoadedClass);} catch (UnmodifiableClassException e) {e.printStackTrace();}}}}public static void simpleDemo(String agentOps, Instrumentation inst) {inst.addTransformer(new ClassFileTransformer() {@Overridepublic byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {//判断是指定的classif ("com/naihe/Demo".equals(className)) {try {//获取更改后的类class 字节数组String path="C:/Users/12107/Desktop/Demo2.class";classfileBuffer = Files.readAllBytes(Paths.get(path));} catch (IOException e) {e.printStackTrace();}}return classfileBuffer;}},true);}}

app项目源码

application：

Demo：

attach：

package com.naihe;import com.sun.tools.attach.*;import java.io.IOException;
import java.util.List;
import java.util.Scanner;
import java.util.stream.Collectors;public class attach {public static void main(String[] args) {//查找所有jvm进程，排除attach测试工程List<VirtualMachineDescriptor> attach = VirtualMachine.list().stream().filter(jvm -> {return !jvm.displayName().contains("attach");}).collect(Collectors.toList());for (int i = 0; i < attach.size(); i++) {System.out.println("[" + i + "] " + attach.get(i).displayName()+":"+attach.get(i).id());}System.out.println("请输入需要attach的pid编号");Scanner scanner = new Scanner(System.in);String s = scanner.nextLine();VirtualMachineDescriptor virtualMachineDescriptor = attach.get(new Integer(s));try {VirtualMachine virtualMachine = VirtualMachine.attach(virtualMachineDescriptor.id());virtualMachine.loadAgent("C:\\Users\\12107\\Desktop\\agent.jar", "param");virtualMachine.detach();} catch (AttachNotSupportedException e) {System.out.println("AttachNotSupportedException：" + e.getMessage());} catch (IOException e) {System.out.println("IOException：" + e.getMessage());} catch (AgentLoadException e) {System.out.println("AgentLoadException：" + e.getMessage());} catch (AgentInitializationException e) {System.out.println("AgentInitializationException：" + e.getMessage());}}
}

retransformClasses：对于已经加载的类重新进行转换处理，即会触发重新加载类定义，需要注意的是，新加载的类不能修改旧有的类声明，譬如不能增加属性、不能修改方法声明