river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-08-26 12:57:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

简化命名

Browse Source
This commit is contained in:
shmilylty
2019-10-19 21:26:48 +08:00
parent 39d9adb6bc
commit ecc1cb53dc
2 changed files with 107 additions and 108 deletions
Download Patch File Download Diff File Expand all files Collapse all files
oneforall/data/srv_names.json → oneforall/data/srv_prefixes.json
View File
oneforall/modules/dnsquery/srv.py
+5 -6
View File
@@ -42,10 +42,10 @@ class BruteSRV(Module):
""" """
枚举域名的SRV记录 枚举域名的SRV记录
""" """
names_path = data_storage_path.joinpath('srv_names.json') path = data_storage_path.joinpath('srv_prefixes.json')
with open(names_path) as fp: with open(path) as file:
names_dict = json.load(fp) prefixes = json.load(file)
names = map(lambda prefix: prefix + self.domain, names_dict) names = map(lambda prefix: prefix + self.domain, prefixes)
tasks = [] tasks = []
for name in names: for name in names:
@@ -53,7 +53,6 @@ class BruteSRV(Module):
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
group = asyncio.gather(*tasks) group = asyncio.gather(*tasks)
results = loop.run_until_complete(group) results = loop.run_until_complete(group)
loop.close()
for result in results: for result in results:
if result is None: if result is None:
continue continue
@@ -87,4 +86,4 @@ def do(domain): # 统一入口名字 方便多线程调用
if __name__ == '__main__': if __name__ == '__main__':
do('zonetransfer.me') do('zonetransfer.me')
do('example.com') # do('example.com')