fix: kafka exception.

Author: ‘niuerzhuang’

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

@@ -29,14 +29,23 @@ public class PluginRegister {
 
     public PluginRegister() {
         this.plugins = new ArrayList<DispatchPlugin>();
-        this.plugins.add(new DispatchSpringApplication());
+        if(!"false".equals(System.getProperty("dongtai.plugin.api"))){
+            this.plugins.add(new DispatchSpringApplication());
+        }
         this.plugins.add(new DispatchJ2ee());
-        this.plugins.add(new DispatchKafka());
+        if(!"false".equals(System.getProperty("dongtai.plugin.kafka"))) {
+            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());
-
+        if(!"false".equals(System.getProperty("dongtai.plugin.shiro"))) {
+            this.plugins.add(new DispatchShiro());
+        }
+        if(!"false".equals(System.getProperty("dongtai.plugin.feign"))) {
+            this.plugins.add(new DispatchFeign());
+        }
+        if(!"false".equals(System.getProperty("dongtai.plugin.dubbo"))) {
+            this.plugins.add(new DispatchDubbo());
+        }
         this.plugins.add(new DispatchClassPlugin());
     }
 

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