通过线程监控Java进程，完成attach agent

最新推荐文章于 2025-04-02 16:04:04 发布

转载最新推荐文章于 2025-04-02 16:04:04 发布 · 1.7k 阅读

Java 专栏收录该内容

169 篇文章

订阅专栏

本文介绍了一个使用Java Attach API的实战案例，通过一个线程子类监控并为新启动的虚拟机加载代理程序。该程序能够检测到新增的虚拟机实例，并自动为其加载指定的代理JAR文件。

import java.io.IOException;
import java.util.List;

import com.sun.tools.attach.AgentInitializationException;
import com.sun.tools.attach.AgentLoadException;
import com.sun.tools.attach.AttachNotSupportedException;
import com.sun.tools.attach.VirtualMachine;
import com.sun.tools.attach.VirtualMachineDescriptor;

public class AttachAgent {
  // 一个运行 Attach API 的线程子类
  static class AttachThread extends Thread {

    private final List<VirtualMachineDescriptor> listBefore;

    private final String jar;

    AttachThread(String attachJar, List<VirtualMachineDescriptor> vms) {
      listBefore = vms; // 记录程序启动时的 VM 集合
      jar = attachJar;
    }

    public void run() {
      VirtualMachine vm = null;
      List<VirtualMachineDescriptor> listAfter = null;
      try {
        int count = 0;
        while (true) {
          listAfter = VirtualMachine.list();
          for (VirtualMachineDescriptor vmd : listAfter) {
            
            System.out.println("displayName:" + vmd.displayName());
            System.out.println("toString:" + vmd.toString());
            
            
            
            if (!listBefore.contains(vmd)) {
              // 如果 VM 有增加，我们就认为是被监控的 VM 启动了
              // 这时，我们开始监控这个 VM
              vm = VirtualMachine.attach(vmd);
              break;
            }
          }
          Thread.sleep(500);
          count++;
          if (null != vm || count >= 100) {
            break;
          }
        }
        vm.loadAgent(jar);
        vm.detach();
      } catch (Exception e) {

      }
    }
  }

  public static void main(String[] args) throws InterruptedException {
    new AttachThread("D:\\indigo\\eclipse\\workspace\\AttachAgent\\agent.jar", VirtualMachine.list()).start();
    
    //new AttachThread("/home/admin/attach/agent.jar", VirtualMachine.list()).start();
  }
}

转载地址：http://memory.blog.51cto.com/6054201/1174700