fix: kafka exception.

dongtai-agent/src/main/java/io/dongtai/iast/agent/Agent.java

@@ -38,6 +38,7 @@ private static String[] parseAgentArgs(String[] args) throws ParseException {
         attachOptions.addOption(build("log_level", "log_level", "optional: DongTai agent log print level."));
         attachOptions.addOption(build("log_path", "log_path", "optional: DongTai agent log print path."));
         attachOptions.addOption(build("log_disable_collector", "log_disable_collector", "optional: DongTai agent disable log collector."));
+        attachOptions.addOption(build("disabled_plugins", "disabled_plugins", "optional: DongTai agent disable plugins."));
 
         CommandLineParser parser = new DefaultParser();
         HelpFormatter formatter = new HelpFormatter();

dongtai-agent/src/main/java/io/dongtai/iast/agent/AgentLauncher.java

@@ -2,7 +2,6 @@
 
 import io.dongtai.iast.agent.manager.EngineManager;
 import io.dongtai.iast.agent.monitor.MonitorDaemonThread;
-import io.dongtai.iast.agent.monitor.impl.AgentStateMonitor;
 import io.dongtai.iast.agent.report.AgentRegisterReport;
 import io.dongtai.iast.common.constants.AgentConstant;
 import io.dongtai.iast.common.scope.ScopeManager;

dongtai-agent/src/main/java/io/dongtai/iast/agent/IastProperties.java

@@ -33,6 +33,7 @@ public class IastProperties {
         put("log_path", PropertyConstant.PROPERTY_LOG_PATH);
         put("log_disable_collector", PropertyConstant.PROPERTY_LOG_DISABLE_COLLECTOR);
         put("uuid_path", PropertyConstant.PROPERTY_UUID_PATH);
+        put("disabled_plugins", PropertyConstant.PROPERTY_DISABLED_PLUGINS);
     }};
 
     private static IastProperties instance;

dongtai-common/src/main/java/io/dongtai/iast/common/constants/PropertyConstant.java

@@ -32,4 +32,5 @@ public class PropertyConstant {
     public static final String PROPERTY_RESPONSE_LENGTH = "dongtai.response.length";
     public static final String PROPERTY_POLICY_PATH = "dongtai.policy.path";
     public static final String PROPERTY_UUID_PATH = "dongtai.uuid.path";
+    public static final String PROPERTY_DISABLED_PLUGINS = "dongtai.disabled.plugins";
 }

dongtai-core/src/main/java/io/dongtai/iast/core/bytecode/enhance/plugin/DispatchPlugin.java

@@ -16,4 +16,6 @@ public interface DispatchPlugin {
      * @return ClassVisitor 命中的类访问起
      */
     ClassVisitor dispatch(ClassVisitor classVisitor, ClassContext context, Policy policy);
+
+    String getName();
 }

dongtai-core/src/main/java/io/dongtai/iast/core/bytecode/enhance/plugin/PluginRegister.java

@@ -9,13 +9,12 @@
 import io.dongtai.iast.core.bytecode.enhance.plugin.hardcoded.DispatchHardcodedPlugin;
 import io.dongtai.iast.core.bytecode.enhance.plugin.service.jdbc.DispatchJdbc;
 import io.dongtai.iast.core.bytecode.enhance.plugin.service.kafka.DispatchKafka;
-import io.dongtai.iast.core.bytecode.enhance.plugin.spring.DispatchSpringApplication;
+import io.dongtai.iast.core.bytecode.enhance.plugin.spring.DispatchApiCollector;
 import io.dongtai.iast.core.handler.hookpoint.models.policy.Policy;
 import io.dongtai.iast.core.handler.hookpoint.models.policy.PolicyManager;
 import org.objectweb.asm.ClassVisitor;
 
-import java.util.ArrayList;
-import java.util.List;
+import java.util.*;
 
 /**
  * @author dongzhiyong@huoxian.cn
@@ -28,18 +27,28 @@ public class PluginRegister {
     private final List<DispatchPlugin> plugins;
 
     public PluginRegister() {
-        this.plugins = new ArrayList<DispatchPlugin>();
-        this.plugins.add(new DispatchSpringApplication());
-        this.plugins.add(new DispatchJ2ee());
-        this.plugins.add(new DispatchKafka());
-        this.plugins.add(new DispatchJdbc());
-        this.plugins.add(new DispatchShiro());
-        this.plugins.add(new DispatchFeign());
-        this.plugins.add(new DispatchDubbo());
-
+        this.plugins = new ArrayList<>();
+        List<String> disabledPlugins = getdisabledPlugins();
+        List<DispatchPlugin> allPlugins = new ArrayList<>(Arrays.asList(
+                new DispatchApiCollector(),
+                new DispatchJ2ee(),
+                new DispatchKafka(),
+                new DispatchJdbc(),
+                new DispatchShiro(),
+                new DispatchFeign(),
+                new DispatchDubbo()
+        ));
+        allPlugins.removeIf(plugin -> disabledPlugins != null && disabledPlugins.contains(plugin.getName()));
+        this.plugins.addAll(allPlugins);
         this.plugins.add(new DispatchClassPlugin());
     }
 
+    private List<String> getdisabledPlugins() {
+        return Optional.ofNullable(System.getProperty("dongtai.disabled.plugins"))
+                .map(s -> Arrays.asList(s.split(",")))
+                .orElse(null);
+    }
+
     public ClassVisitor initial(ClassVisitor classVisitor, ClassContext context, PolicyManager policyManager) {
         Policy policy = policyManager.getPolicy();
         if (policy == null) {

dongtai-core/src/main/java/io/dongtai/iast/core/bytecode/enhance/plugin/authentication/shiro/DispatchShiro.java

@@ -17,4 +17,9 @@ public ClassVisitor dispatch(ClassVisitor classVisitor, ClassContext context, Po
         }
         return classVisitor;
     }
+
+    @Override
+    public String getName() {
+        return "shiro";
+    }
 }

dongtai-core/src/main/java/io/dongtai/iast/core/bytecode/enhance/plugin/core/DispatchClassPlugin.java

@@ -39,6 +39,11 @@ public ClassVisitor dispatch(ClassVisitor classVisitor, ClassContext classContex
         return new ClassVisit(classVisitor, classContext, policy);
     }
 
+    @Override
+    public String getName() {
+        return "class";
+    }
+
     public class ClassVisit extends AbstractClassVisitor {
         private int classVersion;
         private final MethodAdapter[] methodAdapters;

dongtai-core/src/main/java/io/dongtai/iast/core/bytecode/enhance/plugin/framework/dubbo/DispatchDubbo.java

@@ -36,4 +36,9 @@ public ClassVisitor dispatch(ClassVisitor classVisitor, ClassContext context, Po
 
         return classVisitor;
     }
+
+    @Override
+    public String getName() {
+        return "dubbo";
+    }
 }

dongtai-core/src/main/java/io/dongtai/iast/core/bytecode/enhance/plugin/framework/feign/DispatchFeign.java

@@ -18,4 +18,9 @@ public ClassVisitor dispatch(ClassVisitor classVisitor, ClassContext context, Po
 
         return classVisitor;
     }
+
+    @Override
+    public String getName() {
+        return "feign";
+    }
 }

dongtai-core/src/main/java/io/dongtai/iast/core/bytecode/enhance/plugin/framework/j2ee/dispatch/DispatchJ2ee.java

@@ -48,6 +48,11 @@ public ClassVisitor dispatch(ClassVisitor classVisitor, ClassContext context, Po
         return classVisitor;
     }
 
+    @Override
+    public String getName() {
+        return "j2ee";
+    }
+
     private boolean isJavaxServlet(String className, Set<String> diagram) {
         return JAVAX_FACES_SERVLET.equals(className) || JAVAX_HTTP_SERVLET.equals(className)
                 || diagram.contains(JAVAX_HTTP_SERVLET) || diagram.contains(JAVAX_FILTER);

dongtai-core/src/main/java/io/dongtai/iast/core/bytecode/enhance/plugin/hardcoded/DispatchHardcodedPlugin.java

@@ -33,6 +33,11 @@ public ClassVisitor dispatch(ClassVisitor classVisitor, ClassContext context, Po
         return classVisitor;
     }
 
+    @Override
+    public String getName() {
+        return "hardcode";
+    }
+
     private class ExtractClassContent extends AbstractClassVisitor {
 
         private String source;

dongtai-core/src/main/java/io/dongtai/iast/core/bytecode/enhance/plugin/service/jdbc/DispatchJdbc.java

@@ -27,4 +27,9 @@ public ClassVisitor dispatch(ClassVisitor classVisitor, ClassContext context, Po
 
         return classVisitor;
     }
+
+    @Override
+    public String getName() {
+        return "jdbc";
+    }
 }

dongtai-core/src/main/java/io/dongtai/iast/core/bytecode/enhance/plugin/service/kafka/DispatchKafka.java

@@ -18,4 +18,9 @@ public ClassVisitor dispatch(ClassVisitor classVisitor, ClassContext context, Po
 
         return classVisitor;
     }
+
+    @Override
+    public String getName() {
+        return "kafka";
+    }
 }

dongtai-core/src/main/java/io/dongtai/iast/core/bytecode/enhance/plugin/service/kafka/KafkaAbstractConfigInitAdviceAdapter.java

@@ -4,6 +4,7 @@
 import io.dongtai.iast.core.bytecode.enhance.asm.AsmTypes;
 import io.dongtai.iast.core.handler.hookpoint.service.ServiceType;
 import io.dongtai.iast.core.utils.AsmUtils;
+import org.objectweb.asm.Label;
 import org.objectweb.asm.MethodVisitor;
 import org.objectweb.asm.Type;
 import org.objectweb.asm.commons.AdviceAdapter;
@@ -20,6 +21,12 @@ protected KafkaAbstractConfigInitAdviceAdapter(MethodVisitor mv, int access, Str
     @Override
     protected void onMethodExit(int opcode) {
         if (opcode != ATHROW) {
+            Label tryL = new Label();
+            Label catchL = new Label();
+            Label exHandlerL = new Label();
+            visitTryCatchBlock(tryL, catchL, exHandlerL, ASM_TYPE_THROWABLE.getInternalName());
+            visitLabel(tryL);
+
             localServers = newLocal(Type.getType(List.class));
             loadThis();
             push("bootstrap.servers");
@@ -39,6 +46,13 @@ protected void onMethodExit(int opcode) {
             push("");
             push("KafkaUrlHandler");
             invokeInterface(ASM_TYPE_SPY_DISPATCHER, SPY$reportService);
+
+            visitLabel(catchL);
+            Label endL = new Label();
+            visitJumpInsn(GOTO, endL);
+            visitLabel(exHandlerL);
+            visitVarInsn(ASTORE, this.nextLocal);
+            visitLabel(endL);
         }
     }
 }

dongtai-core/src/main/java/io/dongtai/iast/core/bytecode/enhance/plugin/spring/DispatchSpringApplication.java → dongtai-core/src/main/java/io/dongtai/iast/core/bytecode/enhance/plugin/spring/DispatchApiCollector.java

@@ -5,7 +5,7 @@
 import io.dongtai.iast.core.handler.hookpoint.models.policy.Policy;
 import org.objectweb.asm.ClassVisitor;
 
-public class DispatchSpringApplication implements DispatchPlugin {
+public class DispatchApiCollector implements DispatchPlugin {
 
     private static final String FRAMEWORK_SERVLET = " org.springframework.web.servlet.FrameworkServlet".substring(1);
 
@@ -17,4 +17,9 @@ public ClassVisitor dispatch(ClassVisitor classVisitor, ClassContext context, Po
         }
         return classVisitor;
     }
+
+    @Override
+    public String getName() {
+        return "api";
+    }
 }