novice/plugin-privacy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 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
Download Patch File Download Diff File Expand all files Collapse all files

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();
}
}