finish

src/main/java/com/novitechie/LoadClassRule.java

@@ -3,6 +3,7 @@ package com.novitechie;
 public class LoadClassRule {
     public static void check(String name) throws Exception {
         if (name.startsWith("com.janetfilter")) {
+            System.out.println("----------------------"+name);
             throw new ClassNotFoundException(name);
         }
     }

src/main/java/com/novitechie/PluginClassLoaderTransformer.java

@@ -1,10 +1,11 @@
 package com.novitechie;
 
 import com.janetfilter.core.plugin.MyTransformer;
-import jdk.internal.org.objectweb.asm.*;
+import jdk.internal.org.objectweb.asm.ClassReader;
-import jdk.internal.org.objectweb.asm.commons.AdviceAdapter;
+import jdk.internal.org.objectweb.asm.ClassWriter;
+import jdk.internal.org.objectweb.asm.tree.*;
 
-import static jdk.internal.org.objectweb.asm.Opcodes.ASM5;
+import static jdk.internal.org.objectweb.asm.Opcodes.*;
 
 public class PluginClassLoaderTransformer implements MyTransformer {
     @Override
@@ -14,24 +15,19 @@ public class PluginClassLoaderTransformer implements MyTransformer {
 
     @Override
     public byte[] transform(String className, byte[] classBytes, int order) throws Exception {
-        ClassReader cr = new ClassReader(classBytes);
+        ClassReader reader = new ClassReader(classBytes);
-        ClassWriter cw = new ClassWriter(cr, ClassWriter.COMPUTE_FRAMES);
+        ClassNode node = new ClassNode(ASM5);
-        cr.accept(new ClassVisitor(ASM5, cw) {
+        reader.accept(node, 0);
-            @Override
+        for (MethodNode m : node.methods) {
-            public MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) {
+            if ("loadClass".equals(m.name)) {
-                if (name.equals("loadClass")) {
+                InsnList list = new InsnList();
-                    MethodVisitor mv = super.visitMethod(access, name, desc, signature, exceptions);
+                list.add(new VarInsnNode(ALOAD, 1));
-                    return new AdviceAdapter(api, mv, access, name, desc) {
+                list.add(new MethodInsnNode(INVOKESTATIC, "com/novitechie/LoadClassRule", "check", "(Ljava/lang/String;)V", false));
-                        @Override
+                m.instructions.insert(list);
-                        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();
+        ClassWriter writer = new SafeClassWriter(null,null,ClassWriter.COMPUTE_FRAMES | ClassWriter.COMPUTE_MAXS);
+        node.accept(writer);
+        return writer.toByteArray();
     }
 }

src/main/java/com/novitechie/PrivacyPlugin.java

@@ -7,6 +7,10 @@ import java.util.Arrays;
 import java.util.List;
 
 public class PrivacyPlugin  implements PluginEntry {
+
+    static {
+        System.out.println("-------------PrivacyPlugin------------------");
+    }
     @Override
     public String getName() {
         return "PRIVACY";

src/main/java/com/novitechie/SafeClassWriter.java

@@ -0,0 +1,186 @@
+/***
+ * Copyright (c) 2000-2011 INRIA, France Telecom
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ * 3. Neither the name of the copyright holders nor the names of its
+ *    contributors may be used to endorse or promote products derived from
+ *    this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
+ * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
+ * THE POSSIBILITY OF SUCH DAMAGE.
+ */
+package com.novitechie;
+
+
+import java.io.IOException;
+import java.io.InputStream;
+
+import jdk.internal.org.objectweb.asm.ClassReader;
+import jdk.internal.org.objectweb.asm.ClassWriter;
+import jdk.internal.org.objectweb.asm.Opcodes;
+
+
+
+/**
+ * A ClassWriter that computes the common super class of two classes without
+ * actually loading them with a ClassLoader.
+ * 
+ * @author Eric Bruneton
+ */
+public class SafeClassWriter extends ClassWriter {
+
+    private final ClassLoader loader;
+
+    
+    public SafeClassWriter(ClassReader cr, ClassLoader loader, final int flags) {
+        super(cr, flags);
+        this.loader = loader != null ? loader : ClassLoader.getSystemClassLoader();
+    }
+
+    @Override
+    protected String getCommonSuperClass(final String type1, final String type2) {
+        try {
+            ClassReader info1 = typeInfo(type1);
+            ClassReader info2 = typeInfo(type2);
+            if ((info1.getAccess() & Opcodes.ACC_INTERFACE) != 0) {
+                if (typeImplements(type2, info2, type1)) {
+                    return type1;
+                } else {
+                    return "java/lang/Object";
+                }
+            }
+            if ((info2.getAccess() & Opcodes.ACC_INTERFACE) != 0) {
+                if (typeImplements(type1, info1, type2)) {
+                    return type2;
+                } else {
+                    return "java/lang/Object";
+                }
+            }
+            StringBuilder b1 = typeAncestors(type1, info1);
+            StringBuilder b2 = typeAncestors(type2, info2);
+            String result = "java/lang/Object";
+            int end1 = b1.length();
+            int end2 = b2.length();
+            while (true) {
+                int start1 = b1.lastIndexOf(";", end1 - 1);
+                int start2 = b2.lastIndexOf(";", end2 - 1);
+                if (start1 != -1 && start2 != -1
+                        && end1 - start1 == end2 - start2) {
+                    String p1 = b1.substring(start1 + 1, end1);
+                    String p2 = b2.substring(start2 + 1, end2);
+                    if (p1.equals(p2)) {
+                        result = p1;
+                        end1 = start1;
+                        end2 = start2;
+                    } else {
+                        return result;
+                    }
+                } else {
+                    return result;
+                }
+            }
+        } catch (IOException e) {
+            throw new RuntimeException(e.toString());
+        }
+    }
+
+    /**
+     * Returns the internal names of the ancestor classes of the given type.
+     * 
+     * @param type
+     *            the internal name of a class or interface.
+     * @param info
+     *            the ClassReader corresponding to 'type'.
+     * @return a StringBuilder containing the ancestor classes of 'type',
+     *         separated by ';'. The returned string has the following format:
+     *         ";type1;type2 ... ;typeN", where type1 is 'type', and typeN is a
+     *         direct subclass of Object. If 'type' is Object, the returned
+     *         string is empty.
+     * @throws IOException
+     *             if the bytecode of 'type' or of some of its ancestor class
+     *             cannot be loaded.
+     */
+    private StringBuilder typeAncestors(String type, ClassReader info)
+            throws IOException {
+        StringBuilder b = new StringBuilder();
+        while (!"java/lang/Object".equals(type)) {
+            b.append(';').append(type);
+            type = info.getSuperName();
+            info = typeInfo(type);
+        }
+        return b;
+    }
+
+    /**
+     * Returns true if the given type implements the given interface.
+     * 
+     * @param type
+     *            the internal name of a class or interface.
+     * @param info
+     *            the ClassReader corresponding to 'type'.
+     * @param itf
+     *            the internal name of a interface.
+     * @return true if 'type' implements directly or indirectly 'itf'
+     * @throws IOException
+     *             if the bytecode of 'type' or of some of its ancestor class
+     *             cannot be loaded.
+     */
+    private boolean typeImplements(String type, ClassReader info, String itf)
+            throws IOException {
+        while (!"java/lang/Object".equals(type)) {
+            String[] itfs = info.getInterfaces();
+            for (int i = 0; i < itfs.length; ++i) {
+                if (itfs[i].equals(itf)) {
+                    return true;
+                }
+            }
+            for (int i = 0; i < itfs.length; ++i) {
+                if (typeImplements(itfs[i], typeInfo(itfs[i]), itf)) {
+                    return true;
+                }
+            }
+            type = info.getSuperName();
+            info = typeInfo(type);
+        }
+        return false;
+    }
+
+    /**
+     * Returns a ClassReader corresponding to the given class or interface.
+     * 
+     * @param type
+     *            the internal name of a class or interface.
+     * @return the ClassReader corresponding to 'type'.
+     * @throws IOException
+     *             if the bytecode of 'type' cannot be loaded.
+     */
+    private ClassReader typeInfo(final String type) throws IOException {
+        String resource = type + ".class";
+        InputStream is = loader.getResourceAsStream(resource);
+        if (is == null) {
+            throw new IOException("Cannot create ClassReader for type " + type);
+        }
+        try {
+            return new ClassReader(is);
+        } finally {
+            is.close();
+        }
+    }
+}

src/main/java/com/novitechie/StackTraceRule.java

@@ -2,6 +2,7 @@ package com.novitechie;
 
 public class StackTraceRule {
     public static boolean check() {
+        System.out.println("------------------StackTraceRule");
         RuntimeException e = new RuntimeException();
         for (StackTraceElement stackTraceElement : e.getStackTrace()) {
             if (stackTraceElement.getFileName() == null) {

src/main/java/com/novitechie/VMOptionsTransformer.java

@@ -3,8 +3,9 @@ package com.novitechie;
 import com.janetfilter.core.plugin.MyTransformer;
 import jdk.internal.org.objectweb.asm.*;
 import jdk.internal.org.objectweb.asm.commons.AdviceAdapter;
+import jdk.internal.org.objectweb.asm.tree.*;
 
-import static jdk.internal.org.objectweb.asm.Opcodes.ASM5;
+import static jdk.internal.org.objectweb.asm.Opcodes.*;
 
 public class VMOptionsTransformer implements MyTransformer {
     @Override
@@ -14,28 +15,23 @@ public class VMOptionsTransformer implements MyTransformer {
 
     @Override
     public byte[] transform(String className, byte[] classBytes, int order) throws Exception {
-        ClassReader cr = new ClassReader(classBytes);
+        ClassReader reader = new ClassReader(classBytes);
-        ClassWriter cw = new ClassWriter(cr, ClassWriter.COMPUTE_FRAMES);
+        ClassNode node = new ClassNode(ASM5);
-        cr.accept(new ClassVisitor(ASM5, cw) {
+        reader.accept(node, 0);
-            @Override
+        for (MethodNode m : node.methods) {
-            public MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) {
+            if ("getUserOptionsFile".equals(m.name)) {
-                if (name.equals("getUserOptionsFile")) {
+                InsnList list = new InsnList();
-                    MethodVisitor mv = super.visitMethod(access, name, desc, signature, exceptions);
+                list.add(new MethodInsnNode(INVOKESTATIC, "com/novitechie/StackTraceRule", "check", "()Z", false));
-                    return new AdviceAdapter(api, mv, access, name, desc) {
+                LabelNode labelNode = new LabelNode();
-                        @Override
+                list.add(new JumpInsnNode(IFEQ,labelNode));
-                        protected void onMethodEnter() {
+                list.add(new InsnNode(ACONST_NULL));
-                            mv.visitMethodInsn(INVOKESTATIC, "com/novitechie/StackTraceRule", "check", "()Z", false);
+                list.add(new InsnNode(ARETURN));
-                            Label l0 = new Label();
+                list.add(labelNode);
-                            mv.visitJumpInsn(IFEQ, l0);
+                m.instructions.insert(list);
-                            mv.visitInsn(ACONST_NULL);
-                            mv.visitInsn(ARETURN);
-                            mv.visitLabel(l0);
-                        }
-                    };
-                }
-                return super.visitMethod(access, name, desc, signature, exceptions);
             }
-        }, 6);
+        }
-        return cw.toByteArray();
+        ClassWriter writer = new ClassWriter(ClassWriter.COMPUTE_FRAMES | ClassWriter.COMPUTE_MAXS);
+        node.accept(writer);
+        return writer.toByteArray();
     }
 }