优化代码结构 删除无用队列结果集参数

oneforall/modules/check/axfr.py

@@ -7,7 +7,6 @@ DNS域传送(DNS zone transfer)指的是一台备用域名服务器使用来自
 目的是为了做冗余备份，防止主域名服务器出现故障时 dns 解析不可用。
 当主服务器开启DNS域传送同时又对来请求的备用服务器未作访问控制和身份验证便可以利用此漏洞获取某个域的所有记录。
 """
-import queue
 import dns.resolver
 import dns.zone
 
@@ -63,7 +62,7 @@ class CheckAXFR(Module):
                 logger.log('DEBUG', '\n'.join(self.results))
                 self.results = []
 
-    def run(self, rx_queue):
+    def run(self):
         """
         类执行入口
         """
@@ -73,23 +72,19 @@ class CheckAXFR(Module):
         self.save_json()
         self.gen_result()
         self.save_db()
-        rx_queue.put(self.results)
         logger.log('DEBUG', f'结束执行{self.source}检查{self.domain}的域传送漏洞')
         self.finish()
 
 
-def do(domain, rx_queue):  # 统一入口名字 方便多线程调用
+def do(domain):  # 统一入口名字 方便多线程调用
     """
     类统一调用入口
 
     :param str domain: 域名
-    :param rx_queue: 结果集队列
     """
     check = CheckAXFR(domain)
-    check.run(rx_queue)
+    check.run()
 
 
 if __name__ == '__main__':
-    # do('ZoneTransfer.me')
-    result_queue = queue.Queue()
-    do('example.com', result_queue)
+    do('ZoneTransfer.me')

oneforall/modules/check/cdx.py

@@ -2,7 +2,6 @@
 """
 检查crossdomain.xml文件收集子域名
 """
-import queue
 
 from common.module import Module
 from common.utils import match_subdomain
@@ -25,7 +24,6 @@ class CheckCDX(Module):
         检查crossdomain.xml收集子域名
         :return:
         """
-        url = f'http://{self.domain}/crossdomain.xml'
         urls = [f'http://{self.domain}/crossdomain.xml', f'https://{self.domain}/crossdomain.xml',
                 f'http://www.{self.domain}/crossdomain.xml', f'https://www.{self.domain}/crossdomain.xml']
         response = None
@@ -39,7 +37,7 @@ class CheckCDX(Module):
             return
         self.subdomains = match_subdomain(self.domain, response.text)
 
-    def run(self, rx_queue):
+    def run(self):
         """
         类执行入口
         """
@@ -49,21 +47,19 @@ class CheckCDX(Module):
         self.save_json()
         self.gen_result()
         self.save_db()
-        rx_queue.put(self.results)
         logger.log('DEBUG', f'结束执行{self.source}检查{self.domain}域的crossdomain.xml')
         self.finish()
 
 
-def do(domain, rx_queue):  # 统一入口名字 方便多线程调用
+def do(domain):  # 统一入口名字 方便多线程调用
     """
     类统一调用入口
+
     :param domain: 域名
-    :param rx_queue: 结果集队列
     """
     check = CheckCDX(domain)
-    check.run(rx_queue)
+    check.run()
 
 
 if __name__ == '__main__':
-    result_queue = queue.Queue()
-    do('163.com', result_queue)
+    do('example.com')

oneforall/modules/check/cert.py

@@ -4,7 +4,6 @@
 """
 检查域名证书收集子域名
 """
-import queue
 import socket
 import ssl
 
@@ -36,7 +35,7 @@ class CheckCert(Module):
         subdomains_find = utils.match_subdomain(self.domain, str(cert_dict))
         self.subdomains = self.subdomains.union(subdomains_find)
 
-    def run(self, rx_queue):
+    def run(self):
         """
         类执行入口
         """
@@ -45,22 +44,19 @@ class CheckCert(Module):
         self.save_json()
         self.gen_result()
         self.save_db()
-        rx_queue.put(self.results)
         logger.log('DEBUG', f'结束执行{self.source}检查{self.domain}域的证书中的子域')
         self.finish()
 
 
-def do(domain, rx_queue):  # 统一入口名字 方便多线程调用
+def do(domain):  # 统一入口名字 方便多线程调用
     """
     类统一调用入口
 
     :param str domain: 域名
-    :param rx_queue: 结果集队列
     """
     check = CheckCert(domain)
-    check.run(rx_queue)
+    check.run()
 
 
 if __name__ == '__main__':
-    result_queue = queue.Queue()
-    do('example.com', result_queue)
+    do('example.com')

oneforall/modules/check/csp.py

@@ -2,8 +2,8 @@
 """
 检查内容安全策略收集子域名收集子域名
 """
-import queue
 import requests
+
 from common import utils
 from common.module import Module
 from config import logger
@@ -45,7 +45,7 @@ class CheckCSP(Module):
         logger.log('DEBUG', f'{self.domain}域的响应头存在内容安全策略字段')
         self.subdomains = utils.match_subdomain(self.domain, csp)
 
-    def run(self, rx_queue):
+    def run(self):
         """
         类执行入口
         """
@@ -54,25 +54,21 @@ class CheckCSP(Module):
         self.save_json()
         self.gen_result()
         self.save_db()
-        rx_queue.put(self.results)
         logger.log('DEBUG', f'结束执行{self.source}检查{self.domain}域响应头中的内容安全策略字段')
         self.finish()
 
 
-def do(domain, rx_queue, header=None):  # 统一入口名字 方便多线程调用
+def do(domain, header=None):  # 统一入口名字 方便多线程调用
     """
     类统一调用入口
 
     :param str domain: 域名
-    :param rx_queue: 结果集队列
     :param dict or None header: 响应头
     """
     check = CheckCSP(domain, header)
-    check.run(rx_queue)
+    check.run()
 
 
 if __name__ == '__main__':
-    # resp = requests.get('https://content-security-policy.com/')
-    result_queue = queue.Queue()
-    resp = requests.get('https://www.baidu.com/')
-    do('google-analytics.com', result_queue, resp.headers)
+    resp = requests.get('https://content-security-policy.com/')
+    do('google-analytics.com', resp.headers)