添加api检查

oneforall/modules/intelligence/riskiq_api.py

@@ -3,6 +3,7 @@ import queue
 
 import config
 from common.query import Query
+from config import logger
 
 
 class RiskIQ(Query):
@@ -34,6 +35,10 @@ class RiskIQ(Query):
         """
         类执行入口
         """
+        if not self.username or not self.key:
+            logger.log('ERROR', f'{self.source}模块API配置错误')
+            logger.log('ALERT', f'不执行{self.source}模块')
+            return
         self.begin()
         self.query()
         self.save_json()

oneforall/modules/search/fofa_api.py

@@ -41,10 +41,12 @@ class FoFa(Search):
         """
         类执行入口
         """
+        if not self.email or not self.key:
+            logger.log('ERROR', f'{self.source}模块API配置错误')
+            logger.log('ALERT', f'不执行{self.source}模块')
+            return
         logger.log('DEBUG', f'开始执行{self.source}模块搜索{self.domain}的子域')
-
         self.search()
-
         self.save_json()
         self.gen_result()
         self.save_db()