novice/plugin-privacy
1
0
Fork 0
Code Pull Requests Packages Projects Releases 1 Wiki Activity

feat: support rainbow v2025.3.1

Browse Source
This commit is contained in:
yelochick 2025-07-10 21:02:51 +08:00
parent 6da7112443
commit f69a459850
13 changed files with 178 additions and 88 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/main/java/com/novitechie/ClassLoaderTransformer.java
View File

@ -25,12 +25,12 @@ public class ClassLoaderTransformer implements MyTransformer {
for (MethodNode m : node.methods) {
if ("loadClass".equals(m.name)) {
InsnList list = new InsnList();
LabelNode L0 = new LabelNode();
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 JumpInsnNode(IFEQ, L0));
list.add(new VarInsnNode(ALOAD, 1));
list.add(new MethodInsnNode(INVOKESTATIC, "com/novitechie/rules/LoadClassRule", "check", "(Ljava/lang/String;)V", false));
list.add(labelNode);
list.add(L0);
m.instructions.insert(list);
}
}

25
src/main/java/com/novitechie/ClassTransformer.java
View File

@ -25,36 +25,23 @@ public class ClassTransformer implements MyTransformer {
for (MethodNode m : node.methods) {
if ("getResource".equals(m.name) && m.desc.equals("(Ljava/lang/String;)Ljava/net/URL;")) {
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));
LabelNode L0 = new LabelNode();
list.add(new VarInsnNode(ALOAD, 1));
list.add(new MethodInsnNode(INVOKESTATIC, "com/novitechie/rules/ResourceRule", "check", "(Ljava/lang/String;)Z", false));
list.add(new JumpInsnNode(IFEQ, labelNode));
list.add(new JumpInsnNode(IFEQ, L0));
list.add(new InsnNode(ACONST_NULL));
list.add(new InsnNode(ARETURN));
list.add(labelNode);
list.add(L0);
m.instructions.insert(list);
} else if ("getResourceAsStream".equals(m.name) && m.desc.equals("(Ljava/lang/String;)Ljava/io/InputStream;")) {
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));
LabelNode L0 = new LabelNode();
list.add(new VarInsnNode(ALOAD, 1));
list.add(new MethodInsnNode(INVOKESTATIC, "com/novitechie/rules/ResourceRule", "check", "(Ljava/lang/String;)Z", false));
list.add(new JumpInsnNode(IFEQ, labelNode));
list.add(new JumpInsnNode(IFEQ, L0));
list.add(new InsnNode(ACONST_NULL));
list.add(new InsnNode(ARETURN));
list.add(labelNode);
m.instructions.insert(list);
} else if ("getDeclaredMethod".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 VarInsnNode(ALOAD, 1));
list.add(new MethodInsnNode(INVOKESTATIC, "com/novitechie/rules/ClassRule", "checkMethod", "(Ljava/lang/String;)V", false));
list.add(labelNode);
list.add(L0);
m.instructions.insert(list);
}
}

17
src/main/java/com/novitechie/LogUtil.java Normal file
View File

@ -0,0 +1,17 @@
package com.novitechie;
import com.janetfilter.core.commons.DebugInfo;
/**
* @author YeloChick
*/
public class LogUtil {
public static void printStackTrace() {
StackTraceElement[] stackTrace = Thread.currentThread().getStackTrace();
DebugInfo.output("===========================stackTrace: ");
for (StackTraceElement stackTraceElement : stackTrace) {
DebugInfo.output(stackTraceElement.getClassName() + ":" + stackTraceElement.getMethodName());
}
}
}

60
src/main/java/com/novitechie/MethodTransformer.java Normal file
View File

@ -0,0 +1,60 @@
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 MethodTransformer implements MyTransformer {
@Override
public String getHookClassName() {
return "java/lang/reflect/Method";
}
@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 ("invoke".equals(m.name) && m.desc.equals("(Ljava/lang/Object;[Ljava/lang/Object;)Ljava/lang/Object;")) {
InsnList list = new InsnList();
LabelNode L0 = new LabelNode();
list.add(new LdcInsnNode("java.lang.ClassLoader"));
list.add(new VarInsnNode(ALOAD, 0));
list.add(new FieldInsnNode(GETFIELD, "java/lang/reflect/Method", "clazz", "Ljava/lang/Class;"));
list.add(new MethodInsnNode(INVOKEVIRTUAL, "java/lang/Class", "getName", "()Ljava/lang/String;", false));
list.add(new MethodInsnNode(INVOKEVIRTUAL, "java/lang/String", "equals", "(Ljava/lang/Object;)Z", false));
list.add(new JumpInsnNode(IFEQ, L0));
list.add(new LdcInsnNode("findBootstrapClass"));
list.add(new VarInsnNode(ALOAD, 0));
list.add(new FieldInsnNode(GETFIELD, "java/lang/reflect/Method", "name", "Ljava/lang/String;"));
list.add(new MethodInsnNode(INVOKEVIRTUAL, "java/lang/String", "equals", "(Ljava/lang/Object;)Z", false));
list.add(new JumpInsnNode(IFEQ, L0));
list.add(new VarInsnNode(ALOAD, 2));
list.add(new InsnNode(ICONST_0));
list.add(new InsnNode(AALOAD));
list.add(new MethodInsnNode(INVOKESTATIC, "java/lang/String", "valueOf", "(Ljava/lang/Object;)Ljava/lang/String;", false));
list.add(new VarInsnNode(ASTORE, 3));
list.add(new VarInsnNode(ALOAD, 3));
list.add(new MethodInsnNode(INVOKESTATIC, "com/novitechie/rules/BootstrapClassRule", "check", "(Ljava/lang/String;)Z", false));
list.add(new JumpInsnNode(IFEQ, L0));
list.add(new MethodInsnNode(INVOKESTATIC, "com/novitechie/rules/StackTraceRule", "check", "()Z", false));
list.add(new JumpInsnNode(IFEQ, L0));
list.add(new InsnNode(ACONST_NULL));
list.add(new InsnNode(ARETURN));
list.add(L0);
m.instructions.insert(list);
}
}
ClassWriter writer = new ClassWriter(ClassWriter.COMPUTE_FRAMES | ClassWriter.COMPUTE_MAXS);
node.accept(writer);
return writer.toByteArray();
}
}

36
src/main/java/com/novitechie/PluginManagerCoreTransformer.java
View File

@ -25,12 +25,12 @@ public class PluginManagerCoreTransformer implements MyTransformer {
for (MethodNode m : node.methods) {
if ("getPlugins".equals(m.name)) {
InsnList list = new InsnList();
list.add(new MethodInsnNode(INVOKESTATIC, "com/novitechie/rules/StackTraceRule", "check", "()Z", false));
LabelNode L0 = new LabelNode();
LabelNode L1 = new LabelNode();
LabelNode L2 = new LabelNode();
LabelNode L3 = new LabelNode();
LabelNode L4 = new LabelNode();
list.add(new JumpInsnNode(IFEQ, L1));
list.add(new MethodInsnNode(INVOKESTATIC, "com/novitechie/rules/StackTraceRule", "check", "()Z", false));
list.add(new JumpInsnNode(IFEQ, L0));
list.add(new FieldInsnNode(GETSTATIC, "com/intellij/ide/plugins/PluginManagerCore", "INSTANCE", "Lcom/intellij/ide/plugins/PluginManagerCore;"));
list.add(new MethodInsnNode(INVOKEVIRTUAL, "com/intellij/ide/plugins/PluginManagerCore", "getPluginSet", "()Lcom/intellij/ide/plugins/PluginSet;", false));
list.add(new FieldInsnNode(GETFIELD, "com/intellij/ide/plugins/PluginSet", "allPlugins", "Ljava/util/Set;"));
@ -42,10 +42,10 @@ public class PluginManagerCoreTransformer implements MyTransformer {
list.add(new VarInsnNode(ALOAD, 0));
list.add(new MethodInsnNode(INVOKEINTERFACE, "java/util/Set", "iterator", "()Ljava/util/Iterator;", true));
list.add(new VarInsnNode(ASTORE, 2));
list.add(L3);
list.add(L2);
list.add(new VarInsnNode(ALOAD, 2));
list.add(new MethodInsnNode(INVOKEINTERFACE, "java/util/Iterator", "hasNext", "()Z", true));
list.add(new JumpInsnNode(IFEQ, L2));
list.add(new JumpInsnNode(IFEQ, L1));
list.add(new VarInsnNode(ALOAD, 2));
list.add(new MethodInsnNode(INVOKEINTERFACE, "java/util/Iterator", "next", "()Ljava/lang/Object;", true));
list.add(new TypeInsnNode(CHECKCAST, "com/intellij/ide/plugins/IdeaPluginDescriptor"));
@ -54,47 +54,47 @@ public class PluginManagerCoreTransformer implements MyTransformer {
list.add(new MethodInsnNode(INVOKEINTERFACE, "com/intellij/ide/plugins/IdeaPluginDescriptor", "getPluginId", "()Lcom/intellij/openapi/extensions/PluginId;", true));
list.add(new MethodInsnNode(INVOKEVIRTUAL, "com/intellij/openapi/extensions/PluginId", "getIdString", "()Ljava/lang/String;", false));
list.add(new MethodInsnNode(INVOKESTATIC, "com/novitechie/rules/IdeaPluginRule", "check", "(Ljava/lang/String;)Z", false));
list.add(new JumpInsnNode(IFNE, L4));
list.add(new JumpInsnNode(IFNE, L3));
list.add(new VarInsnNode(ALOAD, 1));
list.add(new VarInsnNode(ALOAD, 3));
list.add(new MethodInsnNode(INVOKEINTERFACE, "java/util/Set", "add", "(Ljava/lang/Object;)Z", true));
list.add(new InsnNode(POP));
list.add(L4);
list.add(new JumpInsnNode(GOTO, L3));
list.add(L2);
list.add(L3);
list.add(new JumpInsnNode(GOTO, L2));
list.add(L1);
list.add(new VarInsnNode(ALOAD, 1));
list.add(new InsnNode(ICONST_0));
list.add(new TypeInsnNode(ANEWARRAY, "com/intellij/ide/plugins/IdeaPluginDescriptor"));
list.add(new MethodInsnNode(INVOKEINTERFACE, "java/util/Set", "toArray", "([Ljava/lang/Object;)[Ljava/lang/Object;", true));
list.add(new TypeInsnNode(CHECKCAST, "[Lcom/intellij/ide/plugins/IdeaPluginDescriptor;"));
list.add(new InsnNode(ARETURN));
list.add(L1);
list.add(L0);
m.instructions.insert(list);
} else if ("isPluginInstalled".equals(m.name)) {
InsnList list = new InsnList();
LabelNode L0 = new LabelNode();
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 JumpInsnNode(IFEQ, L0));
list.add(new VarInsnNode(ALOAD, 0));
list.add(new MethodInsnNode(INVOKEVIRTUAL, "com/intellij/openapi/extensions/PluginId", "getIdString", "()Ljava/lang/String;", false));
list.add(new MethodInsnNode(INVOKESTATIC, "com/novitechie/rules/IdeaPluginRule", "check", "(Ljava/lang/String;)Z", false));
list.add(new JumpInsnNode(IFEQ, labelNode));
list.add(new JumpInsnNode(IFEQ, L0));
list.add(new InsnNode(ICONST_0));
list.add(new InsnNode(IRETURN));
list.add(labelNode);
list.add(L0);
m.instructions.insert(list);
} else if ("isDisabled".equals(m.name)) {
InsnList list = new InsnList();
LabelNode L0 = new LabelNode();
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 JumpInsnNode(IFEQ, L0));
list.add(new VarInsnNode(ALOAD, 0));
list.add(new MethodInsnNode(INVOKEVIRTUAL, "com/intellij/openapi/extensions/PluginId", "getIdString", "()Ljava/lang/String;", false));
list.add(new MethodInsnNode(INVOKESTATIC, "com/novitechie/rules/IdeaPluginRule", "check", "(Ljava/lang/String;)Z", false));
list.add(new JumpInsnNode(IFEQ, labelNode));
list.add(new JumpInsnNode(IFEQ, L0));
list.add(new InsnNode(ICONST_1));
list.add(new InsnNode(IRETURN));
list.add(labelNode);
list.add(L0);
m.instructions.insert(list);
}
}

3
src/main/java/com/novitechie/PrivacyPlugin.java
View File

@ -27,6 +27,7 @@ public class PrivacyPlugin implements PluginEntry {
new RuntimeMXBeanTransformer(),
new SystemTransformer(),
new ClassTransformer(),
new ClassLoaderTransformer());
new ClassLoaderTransformer(),
new MethodTransformer());
}
}

16
src/main/java/com/novitechie/SystemTransformer.java
View File

@ -25,27 +25,23 @@ public class SystemTransformer implements MyTransformer {
for (MethodNode m : node.methods) {
if ("getenv".equals(m.name) && m.desc.equals("(Ljava/lang/String;)Ljava/lang/String;")) {
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));
LabelNode L0 = new LabelNode();
list.add(new VarInsnNode(ALOAD, 0));
list.add(new MethodInsnNode(INVOKESTATIC, "com/novitechie/rules/SystemRule", "checkEnv", "(Ljava/lang/String;)Z", false));
list.add(new JumpInsnNode(IFEQ, labelNode));
list.add(new JumpInsnNode(IFEQ, L0));
list.add(new InsnNode(ACONST_NULL));
list.add(new InsnNode(ARETURN));
list.add(labelNode);
list.add(L0);
m.instructions.insert(list);
} else if ("getProperty".equals(m.name) && (m.desc.equals("(Ljava/lang/String;)Ljava/lang/String;") || m.desc.equals("(Ljava/lang/String;Ljava/lang/String;)Ljava/lang/String;"))) {
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));
LabelNode L0 = new LabelNode();
list.add(new VarInsnNode(ALOAD, 0));
list.add(new MethodInsnNode(INVOKESTATIC, "com/novitechie/rules/SystemRule", "checkProperty", "(Ljava/lang/String;)Z", false));
list.add(new JumpInsnNode(IFEQ, labelNode));
list.add(new JumpInsnNode(IFEQ, L0));
list.add(new InsnNode(ACONST_NULL));
list.add(new InsnNode(ARETURN));
list.add(labelNode);
list.add(L0);
m.instructions.insert(list);
}
}

29
src/main/java/com/novitechie/rules/BootstrapClassRule.java Normal file
View File

@ -0,0 +1,29 @@
package com.novitechie.rules;
import com.janetfilter.core.commons.DebugInfo;
import com.novitechie.LogUtil;
import java.util.ArrayList;
import java.util.List;
/**
* @author YeloChick
*/
public class BootstrapClassRule {
private static final List<String> PREVENT_LOAD_PACKAGES = new ArrayList<String>() {
{
add("com.janetfilter");
}
};
public static boolean check(String name) throws Exception {
boolean f = PREVENT_LOAD_PACKAGES.stream().anyMatch(name::startsWith);
if (f) {
DebugInfo.output("======================LoadBootstrapClass: " + name);
LogUtil.printStackTrace();
return true;
}
return false;
}
}

7
src/main/java/com/novitechie/rules/ClassRule.java
View File

@ -1,6 +1,7 @@
package com.novitechie.rules;
import com.janetfilter.core.commons.DebugInfo;
import com.novitechie.LogUtil;
import java.util.ArrayList;
import java.util.List;
@ -20,11 +21,7 @@ public class ClassRule {
public static void checkMethod(String name) throws Exception {
if (PREVENT_LOAD_METHOD.stream().anyMatch(name::equals)) {
DebugInfo.output("======================LoadMethod: " + name);
StackTraceElement[] stackTrace = Thread.currentThread().getStackTrace();
DebugInfo.output("===========================stackTrace: ");
for (StackTraceElement stackTraceElement : stackTrace) {
DebugInfo.output(stackTraceElement.getClassName() + ":" + stackTraceElement.getMethodName());
}
LogUtil.printStackTrace();
throw new NoSuchMethodException();
}
}

7
src/main/java/com/novitechie/rules/IdeaPluginRule.java
View File

@ -1,6 +1,7 @@
package com.novitechie.rules;
import com.janetfilter.core.commons.DebugInfo;
import com.novitechie.LogUtil;
import java.util.ArrayList;
import java.util.List;
@ -23,11 +24,7 @@ public class IdeaPluginRule {
boolean f = PREVENT_LOAD_PLUGINS.stream().anyMatch(name::equals);
if (f) {
DebugInfo.output("======================LoadPlugin: " + name);
StackTraceElement[] stackTrace = Thread.currentThread().getStackTrace();
DebugInfo.output("===========================stackTrace: ");
for (StackTraceElement stackTraceElement : stackTrace) {
DebugInfo.output(stackTraceElement.getClassName() + ":" + stackTraceElement.getMethodName());
}
LogUtil.printStackTrace();
}
return f;
}

10
src/main/java/com/novitechie/rules/LoadClassRule.java
View File

@ -1,6 +1,7 @@
package com.novitechie.rules;
import com.janetfilter.core.commons.DebugInfo;
import com.novitechie.LogUtil;
import java.util.ArrayList;
import java.util.List;
@ -16,13 +17,10 @@ public class LoadClassRule {
};
public static void check(String name) throws Exception {
if (PREVENT_LOAD_PACKAGES.stream().anyMatch(name::startsWith)) {
boolean f = PREVENT_LOAD_PACKAGES.stream().anyMatch(name::startsWith);
if (f) {
DebugInfo.output("======================LoadClass: " + name);
StackTraceElement[] stackTrace = Thread.currentThread().getStackTrace();
DebugInfo.output("===========================stackTrace: ");
for (StackTraceElement stackTraceElement : stackTrace) {
DebugInfo.output(stackTraceElement.getClassName() + ":" + stackTraceElement.getMethodName());
}
LogUtil.printStackTrace();
throw new ClassNotFoundException(name);
}
}

11
src/main/java/com/novitechie/rules/ResourceRule.java
View File

@ -1,6 +1,7 @@
package com.novitechie.rules;
import com.janetfilter.core.commons.DebugInfo;
import com.novitechie.LogUtil;
import java.util.ArrayList;
import java.util.List;
@ -12,7 +13,7 @@ public class ResourceRule {
private static final List<String> PREVENT_LOAD_RESOURCE = new ArrayList<String>() {
{
add("/6c81ec87e55d331c267262e892427a3d93d76683.txt");
// add("/6c81ec87e55d331c267262e892427a3d93d76683.txt");
add("/com/janetfilter");
}
};
@ -20,14 +21,12 @@ public class ResourceRule {
public static boolean check(String name) {
boolean f = PREVENT_LOAD_RESOURCE.stream().anyMatch(name::startsWith);
if (f) {
if (StackTraceRule.check()) {
DebugInfo.output("======================LoadResource: " + name);
StackTraceElement[] stackTrace = Thread.currentThread().getStackTrace();
DebugInfo.output("===========================stackTrace: ");
for (StackTraceElement stackTraceElement : stackTrace) {
DebugInfo.output(stackTraceElement.getClassName() + ":" + stackTraceElement.getMethodName());
}
LogUtil.printStackTrace();
return true;
}
}
return false;
}
}

33
src/main/java/com/novitechie/rules/SystemRule.java
View File

@ -1,6 +1,7 @@
package com.novitechie.rules;
import com.janetfilter.core.commons.DebugInfo;
import com.novitechie.LogUtil;
import java.util.ArrayList;
import java.util.List;
@ -12,7 +13,19 @@ public class SystemRule {
private static final List<String> PREVENT_LOAD_ENV = new ArrayList<String>() {
{
add("_VM_OPTIONS");
add("IDEA_VM_OPTIONS");
add("CLION_VM_OPTIONS");
add("PHPSTORM_VM_OPTIONS");
add("GOLAND_VM_OPTIONS");
add("PYCHARM_VM_OPTIONS");
add("WEBIDE_VM_OPTIONS");
add("RIDER_VM_OPTIONS");
add("DATAGRIP_VM_OPTIONS");
add("APPCODE_VM_OPTIONS");
add("DATASPELL_VM_OPTIONS");
add("GATEWAY_VM_OPTIONS");
add("JETBRAINS_CLIENT_VM_OPTIONS");
add("JETBRAINSCLIENT_VM_OPTIONS");
add("JANF_DEBUG");
add("JANF_OUTPUT");
}
@ -26,30 +39,26 @@ public class SystemRule {
};
public static boolean checkEnv(String name) {
boolean f = PREVENT_LOAD_ENV.stream().anyMatch(name::startsWith) || PREVENT_LOAD_ENV.stream().anyMatch(name::endsWith);
boolean f = PREVENT_LOAD_ENV.stream().anyMatch(name::equals);
if (f) {
if (StackTraceRule.check()) {
DebugInfo.output("======================LoadEnv: " + name);
StackTraceElement[] stackTrace = Thread.currentThread().getStackTrace();
DebugInfo.output("===========================stackTrace: ");
for (StackTraceElement stackTraceElement : stackTrace) {
DebugInfo.output(stackTraceElement.getClassName() + ":" + stackTraceElement.getMethodName());
}
LogUtil.printStackTrace();
return true;
}
}
return false;
}
public static boolean checkProperty(String name) {
boolean f = PREVENT_LOAD_PROPERTY.stream().anyMatch(name::equals);
if (f) {
if (StackTraceRule.check()) {
DebugInfo.output("======================LoadProperty: " + name);
StackTraceElement[] stackTrace = Thread.currentThread().getStackTrace();
DebugInfo.output("===========================stackTrace: ");
for (StackTraceElement stackTraceElement : stackTrace) {
DebugInfo.output(stackTraceElement.getClassName() + ":" + stackTraceElement.getMethodName());
}
LogUtil.printStackTrace();
return true;
}
}
return false;
}
}