修改程序入口为fire

This commit is contained in:
shmilylty
2019-08-06 00:42:49 +08:00
parent 60d8a2a88a
commit 365162a474
+4 -4
View File
@@ -161,7 +161,7 @@ class AIOBrute(Module):
self.target = target self.target = target
self.processes = processes or config.brute_processes_num or os.cpu_count() self.processes = processes or config.brute_processes_num or os.cpu_count()
self.coroutine = coroutine or config.brute_coroutine_num self.coroutine = coroutine or config.brute_coroutine_num
self.wordlist = wordlist or config.brute_wordlist_path or get_wordlist('subdomains.txt') self.wordlist = wordlist or config.brute_wordlist_path or get_wordlist('test.txt')
self.recursive_brute = recursive or config.enable_recursive_brute self.recursive_brute = recursive or config.enable_recursive_brute
self.recursive_depth = depth or config.brute_recursive_depth self.recursive_depth = depth or config.brute_recursive_depth
self.recursive_namelist = namelist or config.recursive_namelist_path or get_wordlist('next_subdomains.txt') self.recursive_namelist = namelist or config.recursive_namelist_path or get_wordlist('next_subdomains.txt')
@@ -279,6 +279,6 @@ def do(domain, result): # 统一入口名字 方便多线程调用
if __name__ == '__main__': if __name__ == '__main__':
# fire.Fire(AIOBrute) fire.Fire(AIOBrute)
result_queue = queue.Queue() # result_queue = queue.Queue()
do('example.com', result_queue) # do('example.com', result_queue)