路径修复

brute.py

@@ -653,6 +653,13 @@ class Brute(Module):
                 logger.log('FATAL', f'Incorrect domain for fuzz')
                 exit(1)
 
+    def init_dict_path(self):
+        data_dir = settings.data_storage_dir
+        if self.wordlist is None:
+            self.wordlist = settings.brute_wordlist_path or data_dir.joinpath('subnames.txt')
+        if self.recursive_nextlist is None:
+            self.recursive_nextlist = settings.recursive_nextlist_path or data_dir.joinpath('subnames_next.txt')
+
     def main(self, domain):
         start = time.time()
         logger.log('INFOR', f'Blasting {domain} ')
@@ -721,6 +728,7 @@ class Brute(Module):
         for self.domain in self.domains:
             self.results = list()  # 置空
             all_subdomains = list()
+            self.init_dict_path()
             self.check_brute_params()
             if self.recursive_brute:
                 logger.log('INFOR', f'Start recursively brute the 1 layer subdomain'

config/default.py

@@ -56,14 +56,14 @@ brute_process_num = 1  # 默认1
 brute_concurrent_num = 2000  # 并发查询数量(默认2000，最大推荐10000)
 brute_socket_num = 1  # 爆破时每个进程下的socket数量
 brute_resolve_num = 15  # 解析失败时尝试换名称服务器重查次数
-# 爆破所使用的字典路径 自定义字典请使用绝对路径
-brute_wordlist_path = data_storage_dir.joinpath('subnames.txt')
+# 爆破所使用的字典路径(默认None则使用data/subdomains.txt，自定义字典请使用绝对路径)
+brute_wordlist_path = None
 # 域名的权威DNS名称服务器的保存路径 当域名开启了泛解析时会使用该名称服务器来进行A记录查询
 authoritative_dns_path = data_storage_dir.joinpath('authoritative_dns.txt')
 enable_recursive_brute = False  # 是否使用递归爆破(默认False)
 brute_recursive_depth = 2  # 递归爆破深度(默认2层)
-# 爆破下一层子域所使用的字典路径 自定义字典请使用绝对路径
-recursive_nextlist_path = data_storage_dir.joinpath('subnames_next.txt')
+# 爆破下一层子域所使用的字典路径(默认None则使用data/subnames_next.txt，自定义字典请使用绝对路径)
+recursive_nextlist_path = None
 enable_check_dict = False  # 是否开启字典配置检查提示(默认False)
 delete_generated_dict = True  # 是否删除爆破时临时生成的字典(默认True)
 delete_massdns_result = True  # 是否删除爆破时massdns输出的解析结果 (默认True)