laramies/theHarvester
1
1
Fork 0
mirror of https://github.com/laramies/theHarvester.git synced 2025-02-26 07:23:17 +08:00
Code Issues Projects Releases Packages Wiki Activity

Allow the user to specify the name server used for the lookups

Browse source
This commit is contained in:
David Mougeolle 2020-03-20 19:40:36 +01:00
parent c5a9a7fb6f
commit 86df2ca295
2 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
theHarvester/__main__.py
View file

@ -490,7 +490,8 @@ async def handler(lst):
callback=generate_postprocessing_callback( callback=generate_postprocessing_callback(
target=word, target=word,
local_results=dnsrev, local_results=dnsrev,
overall_results=full))) overall_results=full),
nameservers=[dnsserver] if dnsserver else None))
# run all the reversing tasks concurrently # run all the reversing tasks concurrently
await asyncio.gather(*__reverse_dns_tasks.values()) await asyncio.gather(*__reverse_dns_tasks.values())

9
theHarvester/discovery/dnssearch.py
View file

@ -14,7 +14,7 @@
from aiodns import DNSResolver from aiodns import DNSResolver
from ipaddress import IPv4Network from ipaddress import IPv4Network
from typing import Callable, List from typing import Callable, List, Optional
# TODO: need big focus on performance and results parsing, now does the basic. # TODO: need big focus on performance and results parsing, now does the basic.
@ -155,7 +155,8 @@ async def reverse_single_ip(
async def reverse_all_ips_in_range( async def reverse_all_ips_in_range(
iprange: str, iprange: str,
callback: Callable) -> None: callback: Callable,
nameservers: Optional[List[str]] = None) -> None:
""" """
Reverse all the IPs stored in a network range. Reverse all the IPs stored in a network range.
All the queries are made concurrently. All the queries are made concurrently.
@ -168,12 +169,14 @@ async def reverse_all_ips_in_range(
they will be ignored. they will be ignored.
callback: Callable. callback: Callable.
Arbitrary postprocessing function. Arbitrary postprocessing function.
nameservers: List[str].
Optional list of DNS servers.
Returns Returns
------- -------
out: None. out: None.
""" """
__resolver = DNSResolver(timeout=4) __resolver = DNSResolver(timeout=4, nameservers=nameservers)
for __ip in list_ips_in_network_range(iprange): for __ip in list_ips_in_network_range(iprange):
log_query(__ip) log_query(__ip)
__host = await reverse_single_ip(ip=__ip, resolver=__resolver) __host = await reverse_single_ip(ip=__ip, resolver=__resolver)