feat: hook ClassLoader, Class, RuntimeMXBean, System, PluginManagerCore

2025-06-23 10:26:07 +08:00
parent 3388fad6fc
commit 6da7112443
17 changed files with 493 additions and 38 deletions
--- a/src/main/java/com/novitechie/RuntimeMXBeanTransformer.java
+++ b/src/main/java/com/novitechie/RuntimeMXBeanTransformer.java
@@ -0,0 +1,42 @@
+package com.novitechie;
+
+import com.janetfilter.core.plugin.MyTransformer;
+import jdk.internal.org.objectweb.asm.ClassReader;
+import jdk.internal.org.objectweb.asm.ClassWriter;
+import jdk.internal.org.objectweb.asm.tree.*;
+
+import static jdk.internal.org.objectweb.asm.Opcodes.*;
+
+/**
+ * @author YeloChick
+ */
+public class RuntimeMXBeanTransformer implements MyTransformer {
+
+    @Override
+    public String getHookClassName() {
+        return "sun/management/RuntimeImpl";
+    }
+
+    @Override
+    public byte[] transform(String className, byte[] classBytes, int order) throws Exception {
+        ClassReader reader = new ClassReader(classBytes);
+        ClassNode node = new ClassNode(ASM5);
+        reader.accept(node, 0);
+        for (MethodNode m : node.methods) {
+            if ("getInputArguments".equals(m.name)) {
+                InsnList list = new InsnList();
+                list.add(new MethodInsnNode(INVOKESTATIC, "com/novitechie/rules/StackTraceRule", "check", "()Z", false));
+                LabelNode labelNode = new LabelNode();
+                list.add(new JumpInsnNode(IFEQ, labelNode));
+                list.add(new LdcInsnNode(""));
+                list.add(new MethodInsnNode(INVOKESTATIC, "java/util/Collections", "singletonList", "(Ljava/lang/Object;)Ljava/util/List;", false));
+                list.add(new InsnNode(ARETURN));
+                list.add(labelNode);
+                m.instructions.insert(list);
+            }
+        }
+        ClassWriter writer = new SafeClassWriter(null, null, ClassWriter.COMPUTE_FRAMES | ClassWriter.COMPUTE_MAXS);
+        node.accept(writer);
+        return writer.toByteArray();
+    }
+}