优化

modules/check/axfr.py

@@ -22,7 +22,7 @@ class CheckAXFR(Module):
 
     def __init__(self, domain: str):
         Module.__init__(self)
-        self.domain = self.register(domain)
+        self.domain = self.get_maindomain(domain)
         self.module = 'Check'
         self.source = 'AXFRCheck'
         self.results = []
@@ -45,7 +45,7 @@ class CheckAXFR(Module):
         names = zone.nodes.keys()
         for name in names:
             full_domain = str(name) + '.' + self.domain
-            subdomain = self.match_subdomains(self.domain, full_domain)
+            subdomain = self.match_subdomains(full_domain)
             self.subdomains = self.subdomains.union(subdomain)
             record = zone[name].to_text(name)
             self.results.append(record)

modules/check/cdx.py

@@ -12,7 +12,7 @@ class CheckCDX(Module):
     """
     def __init__(self, domain: str):
         Module.__init__(self)
-        self.domain = self.register(domain)
+        self.domain = self.get_maindomain(domain)
         self.module = 'Check'
         self.source = "CrossDomainXml"
 
@@ -31,7 +31,7 @@ class CheckCDX(Module):
             if not resp:
                 return
             if resp and len(resp.content):
-                self.subdomains = self.match_subdomains(self.domain, resp.text)
+                self.subdomains = self.match_subdomains(resp.text)
 
     def run(self):
         """

modules/check/cert.py

@@ -14,7 +14,7 @@ from config.log import logger
 class CheckCert(Module):
     def __init__(self, domain):
         Module.__init__(self)
-        self.domain = self.register(domain)
+        self.domain = self.get_maindomain(domain)
         self.port = 443  # ssl port
         self.module = 'Check'
         self.source = 'CertInfo'
@@ -32,7 +32,7 @@ class CheckCert(Module):
         except Exception as e:
             logger.log('DEBUG', e.args)
             return
-        subdomains = self.match_subdomains(self.domain, str(cert_dict))
+        subdomains = self.match_subdomains(str(cert_dict))
         self.subdomains = self.subdomains.union(subdomains)
 
     def run(self):

modules/check/csp.py

@@ -14,7 +14,7 @@ class CheckCSP(Module):
     """
     def __init__(self, domain, header):
         Module.__init__(self)
-        self.domain = self.register(domain)
+        self.domain = self.get_maindomain(domain)
         self.module = 'Check'
         self.source = 'ContentSecurityPolicy'
         self.csp_header = header
@@ -52,7 +52,7 @@ class CheckCSP(Module):
         if not csp:
             logger.log('DEBUG', f'There is no Content-Security-Policy in the header of {self.domain}')
             return
-        self.subdomains = self.match_subdomains(self.domain, csp)
+        self.subdomains = self.match_subdomains(csp)
 
     def run(self):
         """

modules/check/nsec.py

@@ -8,7 +8,7 @@ from common import utils
 class CheckNSEC(Module):
     def __init__(self, domain):
         Module.__init__(self)
-        self.domain = self.register(domain)
+        self.domain = self.get_maindomain(domain)
         self.module = 'check'
         self.source = "CheckNSEC"
 
@@ -21,7 +21,7 @@ class CheckNSEC(Module):
             subdomain = str()
             for item in answer:
                 record = item.to_text()
-                subdomains = self.match_subdomains(self.domain, record)
+                subdomains = self.match_subdomains(record)
                 subdomain = ''.join(subdomains)  # 其实这里的subdomains的长度为1 也就是说只会有一个子域
                 self.subdomains = self.subdomains.union(subdomains)
                 self.gen_record(subdomains, record)

modules/check/robots.py

@@ -11,7 +11,7 @@ class CheckRobots(Module):
     """
     def __init__(self, domain):
         Module.__init__(self)
-        self.domain = self.register(domain)
+        self.domain = self.get_maindomain(domain)
         self.module = 'Check'
         self.source = 'Robots'
 
@@ -30,7 +30,7 @@ class CheckRobots(Module):
             if not resp:
                 return
             if resp and len(resp.content):
-                self.subdomains = self.match_subdomains(self.domain, resp.text)
+                self.subdomains = self.match_subdomains(resp.text)
 
     def run(self):
         """

modules/check/sitemap.py

@@ -11,7 +11,7 @@ class CheckRobots(Module):
     """
     def __init__(self, domain):
         Module.__init__(self)
-        self.domain = self.register(domain)
+        self.domain = self.get_maindomain(domain)
         self.module = 'Check'
         self.source = 'Sitemap'
 
@@ -35,7 +35,7 @@ class CheckRobots(Module):
             if not resp:
                 return
             if resp and len(resp.content):
-                self.subdomains = self.match_subdomains(self.domain, resp.text)
+                self.subdomains = self.match_subdomains(resp.text)
 
     def run(self):
         """