Java Agent入门实战(三)-JVM Attach原理与使用

简介: Java Agent入门实战(三)-JVM Attach原理与使用

之前的permain方法只能在java程序启动之前执行,并不能程序启动之后再执行,但是在实际的很多的情况下,我们没有办法在虚拟机启动之时就为其设定代理,这样实际上限制了instrument的应用。而Java SE 6的新特性改变了这种情况,可以通过Java Tool API中的attach方式来达到这种程序启动之后设置代理的效果。

Attach API 不是 Java 的标准 API,而是 Sun 公司提供的一套扩展 API,用来向目标 JVM “附着”(Attach)代理工具程序的。有了它,开发者可以方便的监控一个 JVM,运行一个外加的代理程序。

Attach API 很简单,只有 2 个主要的类,都在 com.sun.tools.attach包里面: VirtualMachine 代表一个 Java 虚拟机,也就是程序需要监控的目标虚拟机,提供了 JVM 枚举,Attach 动作和 Detach 动作(从 JVM 上面解除一个代理)等等 ; VirtualMachineDescriptor 则是一个描述虚拟机的容器类,配合VirtualMachine类完成各种功能。

结合上一篇文章和Attach,看看如何使用

Agent类增加了2个agentmain()方法,它们的参数不用,2个参数的优先级大于1个参数的,所以这里只有

agentmain (String agentArgs, Instrumentation inst)

会被执行

public class JpAgent {
    public static void premain(String agentArgs){
        System.out.println("我是一个参数的 Java Agent premain");
    }
    public static void agentmain (String agentArgs, Instrumentation inst) throws UnmodifiableClassException {
        inst.addTransformer(new JpClassFileTransformerDemo(), true);
        // retransformClasses 是 Java SE 6 里面的新方法,它跟 redefineClasses 一样,可以批量转换类定义
        inst.retransformClasses(Dog.class);
        System.out.println("我是两个参数的 Java Agent agentmain");
    }
    public static void agentmain (String agentArgs){
        System.out.println("我是一个参数的 Java Agent agentmain");
    }
}

新增一个Dog类,内容如下,关键点在package 路径要和目标程序一样,其他无所谓

package cn.jpsite.learning;
public class Dog {
}

在之前的example01工程中新建一个带有main函数的类WhileMain.java

public class WhileMain {
    public static void main(String[] args) throws InterruptedException {
        System.out.println(new Dog().say());
        int count = 0;
        while (true) {
            // 等待0.5秒
            Thread.sleep(500);
            count++;
            String say = new Dog().say();
            // 输出内容和次数
            System.out.println(say + count);
            // 内容不对或者次数达到1000次以上输出
            if (!"dog".equals(say) || count >= 1000) {
                System.out.println("有人偷了我的狗!");
                //break;
            }
        }
    }
}

准备一个修改过的Dog.class,内容如下,单独保存到目录D:\learning\Dog.class

// 这是一个修改后编译的.class文件,单独存放
public class Dog {
    public String say() {
        return "cat";
    }
}

resource/META-INF/MANIFEST.MF新增内容

Agent-Class: cn.jpsite.learning.javaagent01.JpAgent
Can-Retransform-Classes: true

到了这里,准备工作基本已经完成,执行打包构建,此时执行以下无论哪一条,都不会有结果agentmain输出

java -javaagent:jpAgent.jar -cp example01-1.0-SNAPSHOT.jar cn.jpsite.learning.Main
java -javaagent:jpAgent.jar -cp example01-1.0-SNAPSHOT.jar cn.jpsite.learning.WhileMain
java -cp example01-1.0-SNAPSHOT.jar cn.jpsite.learning.WhileMain

结果如下,并没用调用 agentmain 方法,这该怎么办呢?

learning>jaua-javaagentjpAgent.jar-cpexample01-1.0-SNAPSHOT.jar cn.jpsite.learning.Mair我是一个参数的 Jaua Agent premain夜太黑
-dog
D:\learning>jaua-jauaagent:jpAgent.jar-cpexample01-1.0-SNAPSHOT.jar cnjpsite.learning.WhileM我是一个参数的 Java Agent premain dog dog1 dog2 dog3 dog4
D:\learning>iava-cpexample01-1.0-SNAPSHOT.iar cn.ipsite.learning.WhileMain dog dog1
dog2  

这时候就要用到com.sun.tools.attach来帮助我们达到虚拟机启动之后的代理设置,代码如下:

import com.sun.tools.attach.AttachNotSupportedException;
import com.sun.tools.attach.VirtualMachine;
import com.sun.tools.attach.VirtualMachineDescriptor;
import java.io.IOException;
import java.util.List;
import java.util.Objects;
/**
 *
 * @author jiangpeng
 * @date 2019/12/1 0001
 */
public class AttachThread extends Thread {
    /**
     * 记录程序启动时的 VM 集合
     */
    private final List<VirtualMachineDescriptor> listBefore;
    /**
     要加载的agent.jar
     */
    private final String jar;
    private AttachThread(String attachJar, List<VirtualMachineDescriptor> vms) {
        listBefore = vms;
        jar = attachJar;
    }
    @Override
    public void run() {
        VirtualMachine vm;
        List<VirtualMachineDescriptor> listAfter;
        int count = 0;
        try {
            while (true) {
                listAfter = VirtualMachine.list();
                vm = hasNewVm(listAfter);
                if(vm == null){
                    System.out.println("没有新jvm程序,请手动指定java pid");
                    try{
                        vm = VirtualMachine.attach("7716");
                    }catch (AttachNotSupportedException e){
                        //System.out.println("拒绝访问 Disconnected from the target VM");
                    }
                }
                Thread.sleep(1000);
                System.out.println(count++);
                if (null != vm || count >= 100) {
                    break;
                }
            }
            Objects.requireNonNull(vm).loadAgent(jar);
            vm.detach();
        } catch (Exception e) {
            System.out.println("异常:" + e);
        }
    }
    /**
     * 判断是否有新的 JVM 程序运行
     */
    private VirtualMachine hasNewVm(List<VirtualMachineDescriptor> listAfter) throws IOException, AttachNotSupportedException {
        for (VirtualMachineDescriptor vmd : listAfter) {
            if (!listBefore.contains(vmd)) {
                // 如果 VM 有增加,,我们开始监控这个 VM
                System.out.println("有新的 vm 程序:"+ vmd.displayName());
                return VirtualMachine.attach(vmd);
            }
        }
        return null;
    }
    public static void main(String[] args)  {
        new AttachThread("D:/learning/jpAgent.jar", VirtualMachine.list()).start();
    }
}

其中while循环部分每隔1秒获取一次java进程集合,如果没有的话就会提示手动指定一个java程序进行attach, 当循环了100次或者 获取到了VirtualMachine ,则退出while(true) 去加载指定的agent.jar

2种测试方法

先执行

java -cp example01-1.0-SNAPSHOT.jar cn.jpsite.learning.WhileMain

启动example01程序,看到如下结果,还记得我们之前准备了一份Dog.class,里面的say()内容是cat的嘛,稍后你就会看到神奇的一幕

D : learning >
 D : Vlearning >
 D : learning >
 D : learning >
 D : learning 
 D : learning >
 D : vlearning >
 D : learning >
 D : Vlearning >
 D : Nlearning >
 D : Vlearning >
 D :\ learning > D : vlearning >
 D :eaHn1N9
 D :1earning> java - cp exanple01-1.D-SNAPSH0T. jar cn . jpsite . learning . WhileMain 

接下运行刚刚的AttachThread.java,看到如下内容

D:\Java\jdk1.8.0_111\bin\java. eXe
Connected to the target VM, address:'127.0.0.1:5407', transport: 'socket'
没有新jvm程序,请手动指定java pid
0
没有新jvm程序,请手动指定java pid
1
没有新jvm程序,请手动指定java pid
2
没有新jvm程序,请手动指定java pid  

使用jps -l 命令查看所有有运行中的java程序端口号

dog146
dog147
dog148
dog149
dog150
dog151
dog152
dog153
dog154
dog155
dogi56
dog157
dog158
dog159
dog160
dog161
dog162
dog163
dog164
dog165
dog166
dog167

修改AttachThread.java中的VirtualMachine.attach("7716");代码为

VirtualMachine.attach("16304");

16304为上图中WhileMain java 进程id,重新启动AttachThread.java,结果如下

loader className: cn/jpsite/learning/Dog
我是两个参数的 Java Agent agentmain

可以看到agentmain已经执行了,而且原Dog.say()方法里的内容dog也被改变了,成了cat

没有新jvm程序,请手动指定java pid
40
没有新jvm程序,请手动指定java pid
41
没有新jvm程序,请手动指定java pid
42
有新的vm 程序:cn.jpsite.learning.WhileMain
43
Disconnected from the target VM, address:'127.0.0.1:5656', transport:'socket

第二种

首先启动AttachThread.java,然后执行

java -cp example01-1.0-SNAPSHOT.jar cn.jpsite.learning.WhileMain

,可以很快就看到结果被改变了

D:\learning>
D:\learning>
D:\learning>
D:\learning>
D:\learning>
D:\learning>
D:\learning>
D:\learning>
D:\learning>java-cpexample01-1.0-SNAPSHOT.jar cn.jpsite.learning.WhileMain

MANIFEST.MF的其他属性

Can-Set-Native-Method-Prefix
System-Class-Path
Boot-Class-Path

Java SE 6 新特性:BootClassPath / SystemClassPath 的动态增补

注意几点。首先,我们加入到 classpath 的 jar 文件中不应当带有任何和系统的 instrumentation 有关的系统同名类,不然,一切都陷入不可预料之中

我们要注意到虚拟机的 ClassLoader 的工作方式,它会记载解析结果。比如,我们曾经要求读入某个类 someclass,但是失败了,ClassLoader 会记得这一点。即使我们在后面动态地加入了某一个 jar,含有这个类,ClassLoader 依然会认为我们无法解析这个类,与上次出错的相同的错误会被报告。

再次我们知道在 Java 语言中有一个系统参数“java.class.path”,这个 property 里面记录了我们当前的 classpath,但是,我们使用这两个函数,虽然真正地改变了实际的 classpath,却不会对这个 property 本身产生任何影响。

## 资料

Java Attach API

JavaAgent源码分析

JavaAgent



相关文章
|
23天前
|
消息中间件 Java Linux
2024年最全BATJ真题突击:Java基础+JVM+分布式高并发+网络编程+Linux(1),2024年最新意外的惊喜
2024年最全BATJ真题突击:Java基础+JVM+分布式高并发+网络编程+Linux(1),2024年最新意外的惊喜
|
29天前
|
自然语言处理 前端开发 Java
深入浅出JVM(六)之前端编译过程与语法糖原理
深入浅出JVM(六)之前端编译过程与语法糖原理
|
12天前
|
存储 安全 Java
深入探究Java虚拟机(JVM)的技术细节
深入探究Java虚拟机(JVM)的技术细节
|
14天前
|
存储 算法 Java
深入理解Java虚拟机(JVM)的垃圾回收机制
【5月更文挑战第30天】 在Java开发领域,垃圾回收(Garbage Collection, GC)是确保应用程序性能和内存效率的关键因素。本文将深入探讨Java虚拟机(JVM)的垃圾回收机制,解析其工作原理、不同算法的特点以及如何通过调优来提高应用性能。我们将透过JVM的内存结构,探索垃圾回收过程中涉及的关键技术点,并讨论现代Java应用中常见的垃圾回收器实现。
|
14天前
|
监控 Java 测试技术
滚雪球学Java(45):探秘Java Runtime类:深入了解JVM运行时环境
【5月更文挑战第20天】🏆本文收录于「滚雪球学Java」专栏,专业攻坚指数级提升,希望能够助你一臂之力,帮你早日登顶实现财富自由🚀;同时,欢迎大家关注&&收藏&&订阅!持续更新中,up!up!up!!
18 1
滚雪球学Java(45):探秘Java Runtime类:深入了解JVM运行时环境
|
16天前
|
存储 Java 开发者
深入理解Java虚拟机:JVM内存模型解析
【5月更文挑战第27天】 在Java程序的运行过程中,JVM(Java Virtual Machine)扮演着至关重要的角色。作为Java语言的核心执行环境,JVM不仅负责代码的执行,还管理着程序运行时的内存分配与回收。本文将深入探讨JVM的内存模型,包括其结构、各部分的作用以及它们之间的相互关系。通过对JVM内存模型的剖析,我们能够更好地理解Java程序的性能特征,并针对性地进行调优,从而提升应用的执行效率和稳定性。
|
16天前
|
Java
<Java SE> 5道递归计算,创建数组,数组遍历,JVM内存分配...
<Java SE> 5道递归计算,创建数组,数组遍历,JVM内存分配
38 2
|
19天前
|
存储 算法 Java
深入解析Java虚拟机(JVM):技术原理与性能优化
深入解析Java虚拟机(JVM):技术原理与性能优化
30 1
|
27天前
|
Java 数据库连接 Spring
K8S+Docker理论与实践深度集成java面试jvm原理
K8S+Docker理论与实践深度集成java面试jvm原理
|
29天前
|
Java 索引
深入浅出JVM(五)之Java中方法调用
深入浅出JVM(五)之Java中方法调用