This commit is contained in:
Jing Ling
2020-09-16 00:56:57 +08:00
parent 685d9ff5d2
commit 30423e7bf5
3 changed files with 3 additions and 9 deletions
+2 -2
View File
@@ -689,11 +689,11 @@ def match_subdomains(domain, html, distinct=True, fuzzy=True):
def check_random_subdomain(subdomains): def check_random_subdomain(subdomains):
if not subdomains: if not subdomains:
logger.log('ALERT', f'The generated dictionary is empty') logger.log('ALERT', f'The generated dictionary is empty')
return False return
for subdomain in subdomains: for subdomain in subdomains:
if subdomain: if subdomain:
logger.log('ALERT', f'Please check whether {subdomain} is correct or not') logger.log('ALERT', f'Please check whether {subdomain} is correct or not')
return True return
def get_url_resp(url): def get_url_resp(url):
+1 -2
View File
@@ -43,7 +43,7 @@ class MultiIdentify(Module):
pool = Pool(processes_num) pool = Pool(processes_num)
rules = self.load_rules() rules = self.load_rules()
_identify = Identify(rules) _identify = Identify(rules)
for i in range(processes_num): for _ in range(processes_num):
pool.apply_async(func=_identify.run, args=(task_queue, done_queue)) pool.apply_async(func=_identify.run, args=(task_queue, done_queue))
pool.close() pool.close()
pool.join() pool.join()
@@ -116,7 +116,6 @@ class Identify(object):
self.parse(item) self.parse(item)
banners = [] banners = []
for name, rule in self.RULES.items(): for name, rule in self.RULES.items():
# print(name)
r = self._check_rule(rule) r = self._check_rule(rule)
if r: if r:
if 'implies' in rule: if 'implies' in rule:
-5
View File
@@ -17,11 +17,6 @@ cdn_cname_keyword = utils.load_json(data_dir.joinpath('cdn_cname_keywords.json')
cdn_header_key = utils.load_json(data_dir.joinpath('cdn_header_keys.json')) cdn_header_key = utils.load_json(data_dir.joinpath('cdn_header_keys.json'))
def check_cname_count(cnames):
if len(cnames) > 1:
return True
def check_cname_keyword(cname): def check_cname_keyword(cname):
names = cname.lower().split(',') names = cname.lower().split(',')
for name in names: for name in names: