feat: dubbo api gather plugin support disable

Author: CC11001100

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

@@ -15,7 +15,9 @@
 import io.dongtai.iast.core.utils.PropertyUtils;
 import org.objectweb.asm.ClassVisitor;
 
-import java.util.*;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
 
 /**
  * @author [email protected]
@@ -63,4 +65,15 @@ public ClassVisitor initial(ClassVisitor classVisitor, ClassContext context, Pol
         }
         return classVisitor;
     }
+
+    /**
+     * 查询给定的插件是否被禁用
+     *
+     * @return
+     */
+    public static boolean isPluginDisable(String pluginName) {
+        // 现在插件比较少，O(n)性能损耗能够接受
+        return PropertyUtils.getDisabledPlugins().contains(pluginName);
+    }
+
 }

dongtai-core/src/main/java/io/dongtai/iast/core/handler/hookpoint/api/DubboApiGatherThread.java

@@ -1,6 +1,7 @@
 package io.dongtai.iast.core.handler.hookpoint.api;
 
 import io.dongtai.iast.common.constants.AgentConstant;
+import io.dongtai.iast.core.bytecode.enhance.plugin.PluginRegister;
 import io.dongtai.iast.core.handler.hookpoint.controller.impl.DubboImpl;
 import io.dongtai.log.DongTaiLog;
 
@@ -12,6 +13,9 @@ public class DubboApiGatherThread extends AbstractApiGatherThread {
 
     public static final String FRAMEWORK_NAME = "dubbo";
 
+    // dubbo api采集插件的名字，可以在启动agent的时候通过指定属性禁用它
+    public static final String PLUGIN_NAME = "dubbo-api";
+
     // Dubbo的包名是alibaba还是apache的
     public static enum DubboPackage {
         ALIBABA,
@@ -27,6 +31,12 @@ public static void gather(Class handlerClass) {
         }
         isStarted = true;
 
+        // 判断插件是否开启，仅当开启的情况下才采集
+        if (PluginRegister.isPluginDisable(PLUGIN_NAME)) {
+            DongTaiLog.debug("dubbo api gather plugin disable");
+            return;
+        }
+
         DubboPackage dubboPackage = parseDubboPackage(handlerClass);
         if (dubboPackage == null) {
             return;

dongtai-core/src/main/java/io/dongtai/iast/core/handler/hookpoint/api/SpringGatherApiThread.java

@@ -22,6 +22,7 @@ public static void gather(Object applicationContext) {
             return;
         }
         isStarted = true;
+        
         new SpringGatherApiThread(applicationContext).start();
     }