init

2024-04-21 10:21:39 +00:00 · 2018-08-31 11:49:48 +08:00
parent ebfb0c6f96
commit 3fbfab4288
235 changed files with 30452 additions and 0 deletions
@@ -0,0 +1,96 @@
+package java.arthas;
+
+import java.lang.reflect.Method;
+
+/**
+ * 间谍类<br/>
+ * 藏匿在各个ClassLoader中
+ * Created by vlinux on 15/8/23.
+ */
+public class Spy {
+
+
+    // -- 各种Advice的钩子引用 --
+    public static volatile Method ON_BEFORE_METHOD;
+    public static volatile Method ON_RETURN_METHOD;
+    public static volatile Method ON_THROWS_METHOD;
+    public static volatile Method BEFORE_INVOKING_METHOD;
+    public static volatile Method AFTER_INVOKING_METHOD;
+    public static volatile Method THROW_INVOKING_METHOD;
+
+    /**
+     * arthas's classloader 引用
+     */
+    public static volatile ClassLoader CLASSLOADER;
+
+    /**
+     * 代理重设方法
+     */
+    public static volatile Method AGENT_RESET_METHOD;
+
+    /*
+     * 用于普通的间谍初始化
+     */
+    public static void init(
+            ClassLoader classLoader,
+            Method onBeforeMethod,
+            Method onReturnMethod,
+            Method onThrowsMethod,
+            Method beforeInvokingMethod,
+            Method afterInvokingMethod,
+            Method throwInvokingMethod) {
+        CLASSLOADER = classLoader;
+        ON_BEFORE_METHOD = onBeforeMethod;
+        ON_RETURN_METHOD = onReturnMethod;
+        ON_THROWS_METHOD = onThrowsMethod;
+        BEFORE_INVOKING_METHOD = beforeInvokingMethod;
+        AFTER_INVOKING_METHOD = afterInvokingMethod;
+        THROW_INVOKING_METHOD = throwInvokingMethod;
+    }
+
+    /*
+     * 用于启动线程初始化
+     */
+    public static void initForAgentLauncher(
+            ClassLoader classLoader,
+            Method onBeforeMethod,
+            Method onReturnMethod,
+            Method onThrowsMethod,
+            Method beforeInvokingMethod,
+            Method afterInvokingMethod,
+            Method throwInvokingMethod,
+            Method agentResetMethod) {
+        CLASSLOADER = classLoader;
+        ON_BEFORE_METHOD = onBeforeMethod;
+        ON_RETURN_METHOD = onReturnMethod;
+        ON_THROWS_METHOD = onThrowsMethod;
+        BEFORE_INVOKING_METHOD = beforeInvokingMethod;
+        AFTER_INVOKING_METHOD = afterInvokingMethod;
+        THROW_INVOKING_METHOD = throwInvokingMethod;
+        AGENT_RESET_METHOD = agentResetMethod;
+    }
+
+    /**
+     * Clean up the reference to com.taobao.arthas.agent.AgentLauncher$1
+     * to avoid classloader leak.
+     */
+    public static void destroy() {
+        CLASSLOADER = null;
+        ON_BEFORE_METHOD = null;
+        ON_RETURN_METHOD = null;
+        ON_THROWS_METHOD = null;
+        BEFORE_INVOKING_METHOD = null;
+        AFTER_INVOKING_METHOD = null;
+        THROW_INVOKING_METHOD = null;
+        // clear the reference to ArthasClassLoader in AgentLauncher
+        if (AGENT_RESET_METHOD != null) {
+            try {
+                AGENT_RESET_METHOD.invoke(null);
+            } catch (Exception e) {
+                e.printStackTrace();
+            }
+        }
+        AGENT_RESET_METHOD = null;
+    }
+
+}