river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-10-25 00:56:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

模块优化

Browse Source
This commit is contained in:
Jing Ling
2020-07-08 15:16:26 +08:00
parent 6de0776c0b
commit dda116b9e4
16 changed files with 34 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/datasets/binaryedge_api.py
+1 -1
View File
@@ -22,7 +22,7 @@ class BinaryEdgeAPI(Query):
resp = self.get(url)
if not resp:
return
subdomains = self.match_subdomains(self.domain, str(resp.json()))
subdomains = self.match_subdomains(self.domain, resp.text)
self.subdomains = self.subdomains.union(subdomains)
def run(self):
modules/datasets/bufferover.py
+1 -1
View File
@@ -26,7 +26,7 @@ class BufferOver(Query):
return
if resp.status_code != 200:
return
subdomains = self.match_subdomains(self.domain, str(resp.json()))
subdomains = self.match_subdomains(self.domain, resp.text)
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains)
modules/datasets/cebaidu.py
+1 -1
View File
@@ -19,7 +19,7 @@ class CeBaidu(Query):
resp = self.get(self.addr, params)
if not resp:
return
subdomains = self.match_subdomains(self.domain, str(resp.json()))
subdomains = self.match_subdomains(self.domain, resp.text)
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains)
modules/datasets/chinaz_api.py
+1 -1
View File
@@ -21,7 +21,7 @@ class ChinazAPI(Query):
resp = self.get(self.addr, params)
if not resp:
return
subdomains = self.match_subdomains(self.domain, str(resp.json()))
subdomains = self.match_subdomains(self.domain, resp.text)
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains)
modules/datasets/circl_api.py
+1 -1
View File
@@ -21,7 +21,7 @@ class CirclAPI(Query):
resp = self.get(self.addr + self.domain, auth=(self.user, self.pwd))
if not resp:
return
subdomains = self.match_subdomains(self.domain, str(resp.json()))
subdomains = self.match_subdomains(self.domain, resp.text)
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains)
modules/datasets/passivedns_api.py
+1 -1
View File
@@ -22,7 +22,7 @@ class PassiveDnsAPI(Query):
resp = self.get(url)
if not resp:
return
subdomains = self.match_subdomains(self.domain, str(resp.json()))
subdomains = self.match_subdomains(self.domain, resp.text)
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains)
modules/datasets/phonebook.py
+9 -5
View File
@@ -15,11 +15,13 @@ class PhoneBook(Query):
"""
self.header = self.get_header()
self.proxy = self.get_proxy(self.source)
self.header.update({'Referer': 'https://phonebook.cz/', 'Origin': 'https://phonebook.cz'})
self.header.update({'Referer': 'https://phonebook.cz/',
'Origin': 'https://phonebook.cz'})
addr = 'https://public.intelx.io/phonebook/search'
key = 'ac572eea-3902-4e9a-972d-f5996d76174c'
key = 'd7d1ed06-f0c5-49d4-a9ca-a167e6d2ffab'
url = f'{addr}?k={key}'
data = {"term": self.domain, "maxresults": 1000000, "media": 0, "target": 1,
data = {"term": self.domain, "maxresults": 10000,
"media": 0, "target": 1,
"terminate": [], "timeout": 20}
resp = self.post(url, json=data)
if not resp:
@@ -29,8 +31,10 @@ class PhoneBook(Query):
if not ids:
logger.log('ALERT', f'Get PhoneBook id fail')
return
url = f'{addr}/result?k={key}&id={ids}&limit=1000000'
url = f'{addr}/result?k={key}&id={ids}&limit=10000'
resp = self.get(url)
if not resp:
return
subdomains = self.match_subdomains(self.domain, resp.text)
self.subdomains = self.subdomains.union(subdomains)
@@ -57,4 +61,4 @@ def do(domain): # 统一入口名字 方便多线程调用
if __name__ == '__main__':
do('example.com')
do('freebuf.com')
modules/datasets/threatcrowd.py
+1 -1
View File
@@ -25,7 +25,7 @@ class ThreatCrowd(Query):
return
if resp.status_code != 200:
return
subdomains = self.match_subdomains(self.domain, str(resp.json()))
subdomains = self.match_subdomains(self.domain, resp.text)
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains)