This commit is contained in:
Jing Ling
2020-08-17 19:10:42 +08:00
parent 482c7d6ba2
commit daba572a6a
5 changed files with 11 additions and 9 deletions
+1 -1
View File
@@ -638,7 +638,7 @@ class Brute(Module):
delete_file(dict_path, output_paths)
end = time.time()
self.elapse = round(end - start, 1)
logger.log('INFOR', f'{self.source} module takes {self.elapse} seconds, '
logger.log('ALERT', f'{self.source} module takes {self.elapse} seconds, '
f'found {len(self.subdomains)} subdomains of {domain}')
logger.log('DEBUG', f'{self.source} module found subdomains of {domain}:\n'
f'{self.subdomains}')
+1 -2
View File
@@ -152,8 +152,7 @@ def get_domains(target, targets=None):
domains = list(target_domains.union(targets_domains))
logger.log('INFOR', f'Get {len(domains)} domains')
if not domains:
logger.log('FATAL', f'Did not get a valid domain name')
exit(1)
logger.log('ERROR', f'Did not get a valid domain name')
logger.log('DEBUG', f'The obtained domains \n{domains}')
return domains
+1 -1
View File
@@ -51,7 +51,7 @@ class MultiIdentify(Module):
result_data = done_queue.get()
self.end = time.time()
self.elapse = round(self.end - self.start, 1)
logger.log('INFOR', f'The Identify module took {self.elapse} seconds')
logger.log('ALERT', f'The Identify module took {self.elapse} seconds')
return result_data
def listener(self, total, done_queue):
+3
View File
@@ -269,10 +269,13 @@ class OneForAll(object):
self.config_param()
self.check_param()
self.domains = utils.get_domains(self.target, self.targets)
if self.domains:
for domain in self.domains:
self.domain = utils.get_main_domain(domain)
self.main()
utils.export_all(self.alive, self.format, self.path, self.datas)
else:
logger.log('FATAL', 'Failed to obtain domain')
logger.log('INFOR', 'Finished OneForAll')
@staticmethod
+1 -1
View File
@@ -158,7 +158,7 @@ class Takeover(Module):
logger.log('FATAL', f'Failed to obtain domain')
end = time.time()
elapse = round(end - start, 1)
logger.log('INFOR', f'{self.source} module takes {elapse} seconds, '
logger.log('ALERT', f'{self.source} module takes {elapse} seconds, '
f'There are {len(self.results)} subdomains exists takeover')
logger.log('INFOR', f'Subdomain takeover results: {self.path}')
logger.log('INFOR', f'Finished {self.source} module')