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

This commit is contained in:
shmilylty
2019-10-30 23:59:28 +08:00
parent 9f23e580e4
commit 9119e051b7
+10 -3
View File
@@ -70,6 +70,7 @@ class OneForAll(object):
:param str port: 请求验证子域的端口范围(默认只探测80端口)
:param int valid: 导出子域的有效性(默认1)
:param str format: 导出格式(默认csv)
:param bool takeover: 检查子域接管(默认True)
:param bool show: 终端显示导出数据(默认False)
"""
def __init__(self, target, brute=None, dns=None, req=None, port='default',
@@ -79,15 +80,21 @@ class OneForAll(object):
self.domains = set()
self.domain = str()
self.data = list()
self.brute = brute or config.enable_brute_module
self.dns = dns or config.enable_dns_resolve
self.req = req or config.enable_http_request
self.brute = brute
self.dns = dns
self.req = req
self.takeover = takeover
self.valid = valid
self.format = format
self.show = show
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'
new_table = self.domain + '_now'
collect = Collect(self.domain, export=False)