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

修复dns和req参数传入无效的问题

Browse Source
This commit is contained in:
shmilylty
2019-10-30 23:59:28 +08:00
parent 9f23e580e4
commit 9119e051b7
1 changed files with 18 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
oneforall/oneforall.py
+18 -11
View File
@@ -63,14 +63,15 @@ class OneForAll(object):
参数format可选格式有'txt', 'rst', 'csv', 'tsv', 'json', 'yaml', 'html', 参数format可选格式有'txt', 'rst', 'csv', 'tsv', 'json', 'yaml', 'html',
'jira', 'xls', 'xlsx', 'dbf', 'latex', 'ods' 'jira', 'xls', 'xlsx', 'dbf', 'latex', 'ods'
:param str target: 单个域名或者每行一个域名的文件路径(必需参数) :param str target: 单个域名或者每行一个域名的文件路径(必需参数)
:param bool brute: 使用爆破模块(默认False) :param bool brute: 使用爆破模块(默认False)
:param bool dns: DNS解析子域(默认True) :param bool dns: DNS解析子域(默认True)
:param bool req: HTTP请求子域(默认True) :param bool req: HTTP请求子域(默认True)
:param str port: 请求验证子域的端口范围(默认只探测80端口) :param str port: 请求验证子域的端口范围(默认只探测80端口)
:param int valid: 导出子域的有效性(默认1) :param int valid: 导出子域的有效性(默认1)
:param str format: 导出格式(默认csv) :param str format: 导出格式(默认csv)
:param bool show: 终端显示导出数据(默认False) :param bool takeover: 检查子域接管(默认True)
:param bool show: 终端显示导出数据(默认False)
""" """
def __init__(self, target, brute=None, dns=None, req=None, port='default', def __init__(self, target, brute=None, dns=None, req=None, port='default',
valid=1, format='csv', takeover=True, show=False): valid=1, format='csv', takeover=True, show=False):
@@ -79,15 +80,21 @@ class OneForAll(object):
self.domains = set() self.domains = set()
self.domain = str() self.domain = str()
self.data = list() self.data = list()
self.brute = brute or config.enable_brute_module self.brute = brute
self.dns = dns or config.enable_dns_resolve self.dns = dns
self.req = req or config.enable_http_request self.req = req
self.takeover = takeover self.takeover = takeover
self.valid = valid self.valid = valid
self.format = format self.format = format
self.show = show self.show = show
def main(self): def main(self):
if self.brute is None:
self.brute = config.enable_brute_module
if self.dns is None:
self.dns = config.enable_dns_resolve
if self.req is None:
self.req = config.enable_http_request
old_table = self.domain + '_last' old_table = self.domain + '_last'
new_table = self.domain + '_now' new_table = self.domain + '_now'
collect = Collect(self.domain, export=False) collect = Collect(self.domain, export=False)