commit 439a1277995236e0e71e69615ac7c9931eb6328e Author: novice.li Date: Sun Feb 25 12:18:32 2024 +0800 init diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..bb2df64 --- /dev/null +++ b/.gitignore @@ -0,0 +1,38 @@ +target/ +!.mvn/wrapper/maven-wrapper.jar +!**/src/main/**/target/ +!**/src/test/**/target/ + +### IntelliJ IDEA ### +.idea/modules.xml +.idea/jarRepositories.xml +.idea/compiler.xml +.idea/libraries/ +*.iws +*.iml +*.ipr + +### Eclipse ### +.apt_generated +.classpath +.factorypath +.project +.settings +.springBeans +.sts4-cache +.idea/ +### NetBeans ### +/nbproject/private/ +/nbbuild/ +/dist/ +/nbdist/ +/.nb-gradle/ +build/ +!**/src/main/**/build/ +!**/src/test/**/build/ + +### VS Code ### +.vscode/ + +### Mac OS ### +.DS_Store \ No newline at end of file diff --git a/pom.xml b/pom.xml new file mode 100644 index 0000000..7c70e2a --- /dev/null +++ b/pom.xml @@ -0,0 +1,71 @@ + + + 4.0.0 + + com.novitechie + plugin-privacy + 1.0.0 + + + 8 + 8 + UTF-8 + + + + + org.apache.maven.plugins + maven-compiler-plugin + 3.7.0 + + 8 + 8 + UTF-8 + -XDignore.symbol.file + true + + + + org.apache.maven.plugins + maven-assembly-plugin + 3.0.0 + + false + + jar-with-dependencies + + + + true + + + novice.li + com.novitechie.PrivacyPlugin + + + privacy + + + + make-assembly + package + + single + + + + + + + + + com.ja-netfilter + ja-netfilter + 2.0.1 + provided + + + + \ No newline at end of file diff --git a/src/main/java/com/novitechie/LoadClassRule.java b/src/main/java/com/novitechie/LoadClassRule.java new file mode 100644 index 0000000..0979e22 --- /dev/null +++ b/src/main/java/com/novitechie/LoadClassRule.java @@ -0,0 +1,9 @@ +package com.novitechie; + +public class LoadClassRule { + public static void check(String name) throws Exception { + if (name.startsWith("com.janetfilter")) { + throw new ClassNotFoundException(name); + } + } +} diff --git a/src/main/java/com/novitechie/PluginClassLoaderTransformer.java b/src/main/java/com/novitechie/PluginClassLoaderTransformer.java new file mode 100644 index 0000000..f372cb6 --- /dev/null +++ b/src/main/java/com/novitechie/PluginClassLoaderTransformer.java @@ -0,0 +1,37 @@ +package com.novitechie; + +import com.janetfilter.core.plugin.MyTransformer; +import jdk.internal.org.objectweb.asm.*; +import jdk.internal.org.objectweb.asm.commons.AdviceAdapter; + +import static jdk.internal.org.objectweb.asm.Opcodes.ASM5; + +public class PluginClassLoaderTransformer implements MyTransformer { + @Override + public String getHookClassName() { + return "com/intellij/ide/plugins/cl/PluginClassLoader"; + } + + @Override + public byte[] transform(String className, byte[] classBytes, int order) throws Exception { + ClassReader cr = new ClassReader(classBytes); + ClassWriter cw = new ClassWriter(cr, ClassWriter.COMPUTE_FRAMES); + cr.accept(new ClassVisitor(ASM5, cw) { + @Override + public MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) { + if (name.equals("loadClass")) { + MethodVisitor mv = super.visitMethod(access, name, desc, signature, exceptions); + return new AdviceAdapter(api, mv, access, name, desc) { + @Override + protected void onMethodEnter() { + mv.visitVarInsn(ALOAD, 1); + mv.visitMethodInsn(INVOKESTATIC, "com/novitechie/LoadClassRule", "check", "(Ljava/lang/String;)V", false); + } + }; + } + return super.visitMethod(access, name, desc, signature, exceptions); + } + }, 6); + return cw.toByteArray(); + } +} diff --git a/src/main/java/com/novitechie/PrivacyPlugin.java b/src/main/java/com/novitechie/PrivacyPlugin.java new file mode 100644 index 0000000..3e518b4 --- /dev/null +++ b/src/main/java/com/novitechie/PrivacyPlugin.java @@ -0,0 +1,24 @@ +package com.novitechie; + +import com.janetfilter.core.plugin.MyTransformer; +import com.janetfilter.core.plugin.PluginEntry; + +import java.util.Arrays; +import java.util.List; + +public class PrivacyPlugin implements PluginEntry { + @Override + public String getName() { + return "PRIVACY"; + } + + @Override + public String getAuthor() { + return "novice.li"; + } + + @Override + public List getTransformers() { + return Arrays.asList(new VMOptionsTransformer(),new PluginClassLoaderTransformer()); + } +} diff --git a/src/main/java/com/novitechie/StackTraceRule.java b/src/main/java/com/novitechie/StackTraceRule.java new file mode 100644 index 0000000..5abf63f --- /dev/null +++ b/src/main/java/com/novitechie/StackTraceRule.java @@ -0,0 +1,13 @@ +package com.novitechie; + +public class StackTraceRule { + public static boolean check() { + RuntimeException e = new RuntimeException(); + for (StackTraceElement stackTraceElement : e.getStackTrace()) { + if (stackTraceElement.getFileName() == null) { + return true; + } + } + return false; + } +} diff --git a/src/main/java/com/novitechie/VMOptionsTransformer.java b/src/main/java/com/novitechie/VMOptionsTransformer.java new file mode 100644 index 0000000..b1afc2b --- /dev/null +++ b/src/main/java/com/novitechie/VMOptionsTransformer.java @@ -0,0 +1,41 @@ +package com.novitechie; + +import com.janetfilter.core.plugin.MyTransformer; +import jdk.internal.org.objectweb.asm.*; +import jdk.internal.org.objectweb.asm.commons.AdviceAdapter; + +import static jdk.internal.org.objectweb.asm.Opcodes.ASM5; + +public class VMOptionsTransformer implements MyTransformer { + @Override + public String getHookClassName() { + return "com/intellij/diagnostic/VMOptions"; + } + + @Override + public byte[] transform(String className, byte[] classBytes, int order) throws Exception { + ClassReader cr = new ClassReader(classBytes); + ClassWriter cw = new ClassWriter(cr, ClassWriter.COMPUTE_FRAMES); + cr.accept(new ClassVisitor(ASM5, cw) { + @Override + public MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) { + if (name.equals("getUserOptionsFile")) { + MethodVisitor mv = super.visitMethod(access, name, desc, signature, exceptions); + return new AdviceAdapter(api, mv, access, name, desc) { + @Override + protected void onMethodEnter() { + mv.visitMethodInsn(INVOKESTATIC, "com/novitechie/StackTraceRule", "check", "()Z", false); + Label l0 = new Label(); + mv.visitJumpInsn(IFEQ, l0); + mv.visitInsn(ACONST_NULL); + mv.visitInsn(ARETURN); + mv.visitLabel(l0); + } + }; + } + return super.visitMethod(access, name, desc, signature, exceptions); + } + }, 6); + return cw.toByteArray(); + } +}