laramies/theHarvester
1
1
Fork 0
mirror of https://github.com/laramies/theHarvester.git synced 2025-02-24 14:32:57 +08:00
Code Issues Projects Releases Packages Wiki Activity

small logic fix.

Browse source
This commit is contained in:
NotoriousRebel 2020-06-04 18:46:06 -04:00
parent db6c908a8a
commit 88c8a12dff
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
theHarvester/discovery/dnssearch.py
View file

@ -28,8 +28,8 @@ def __init__(self, domain, dnsserver, verbose=False):
self.domain = domain
self.subdo = False
self.verbose = verbose
#self.dnsserver = [dnsserver] if isinstance(dnsserver, str) else dnsserver
self.dnsserver = list(map(str, dnsserver.split(',')))
# self.dnsserver = [dnsserver] if isinstance(dnsserver, str) else dnsserver
self.dnsserver = list(map(str, dnsserver.split(','))) if isinstance(dnsserver, str) else dnsserver
try:
with open('wordlists/dns-names.txt', 'r') as file:
self.list = file.readlines()
@ -41,10 +41,13 @@ def __init__(self, domain, dnsserver, verbose=False):
async def run(self):
print(f'Created checker with this many words {len(self.list)}')
checker = hostchecker.Checker(self.list) if self.dnsserver == [] else hostchecker.Checker(self.list,
nameserver=self.dnsserver)
checker = hostchecker.Checker(
self.list) if self.dnsserver == [] or self.dnsserver == "" or self.dnsserver is None \
else hostchecker.Checker(self.list, nameserver=self.dnsserver)
hosts, ips = await checker.check()
return hosts, ips
#####################################################################
# DNS REVERSE
#####################################################################
@ -160,6 +163,7 @@ async def reverse_all_ips_in_range(iprange: str, callback: Callable, nameservers
callback(__host)
log_result(__host)
#####################################################################
# IO
#####################################################################
@ -218,6 +222,7 @@ def generate_postprocessing_callback(target: str, **allhosts: List[str]) -> Call
A function that will update the collection of target subdomains
when the query result is satisfying.
"""
def append_matching_hosts(host: str) -> None:
if host and target in host:
for __name, __hosts in allhosts.items():