acidvegas/ptrstream
1
Fork 0
Code Issues Pull Requests Activity

colors fixed, script improved

Browse Source
This commit is contained in:
Dionysus 2023-11-23 04:46:49 -05:00
parent 581efb8f6d
commit 6e769ac39e
Signed by: acidvegas
GPG Key ID: EF4B922DB85DC9DE
1 changed files with 45 additions and 59 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

104
ptrstream.py
View File

@ -13,7 +13,6 @@ try:
except ImportError: except ImportError:
raise ImportError('missing required \'aiodns\' library (pip install aiodns)') raise ImportError('missing required \'aiodns\' library (pip install aiodns)')
# Colors # Colors
class colors: class colors:
ip = '\033[35m' ip = '\033[35m'
@ -24,6 +23,7 @@ class colors:
reset = '\033[0m' reset = '\033[0m'
separator = '\033[90m' separator = '\033[90m'
dns_resolvers = []
def get_dns_servers() -> list: def get_dns_servers() -> list:
'''Get a list of DNS servers to use for lookups.''' '''Get a list of DNS servers to use for lookups.'''
@ -31,18 +31,15 @@ def get_dns_servers() -> list:
results = source.read().decode().split('\n') results = source.read().decode().split('\n')
return [server for server in results if ':' not in server] return [server for server in results if ':' not in server]
async def rdns(semaphore: asyncio.Semaphore, ip_address: str, resolver: aiodns.DNSResolver):
async def rdns(semaphore: asyncio.Semaphore, ip_address: str, custom_dns_server: str):
''' '''
Perform a reverse DNS lookup on an IP address. Perform a reverse DNS lookup on an IP address.
:param ip_address: The IP address to lookup. :param ip_address: The IP address to lookup.
:param custom_dns_server: The DNS server to use for lookups. :param resolver: The DNS resolver to use for lookups.
:param semaphore: A semaphore to limit the number of concurrent lookups. :param semaphore: A semaphore to limit the number of concurrent lookups.
:param timeout: The timeout for the lookup.
''' '''
async with semaphore: async with semaphore:
resolver = aiodns.DNSResolver(nameservers=[custom_dns_server], rotate=False, timeout=args.timeout)
reverse_name = ipaddress.ip_address(ip_address).reverse_pointer reverse_name = ipaddress.ip_address(ip_address).reverse_pointer
try: try:
answer = await resolver.query(reverse_name, 'PTR') answer = await resolver.query(reverse_name, 'PTR')
@ -50,7 +47,6 @@ async def rdns(semaphore: asyncio.Semaphore, ip_address: str, custom_dns_server:
except: except:
return ip_address, None return ip_address, None
def rig(seed: int) -> str: def rig(seed: int) -> str:
''' '''
Random IP generator. Random IP generator.
@ -64,67 +60,57 @@ def rig(seed: int) -> str:
ip = ipaddress.ip_address(shuffled_index) ip = ipaddress.ip_address(shuffled_index)
yield str(ip) yield str(ip)
async def main(): async def main():
'''Generate random IPs and perform reverse DNS lookups.''' '''Generate random IPs and perform reverse DNS lookups.'''
semaphore = asyncio.Semaphore(args.concurrency) semaphore = asyncio.Semaphore(args.concurrency)
tasks = [] tasks = []
results_cache = [] results_cache = []
if args.resolvers: global dns_resolvers
with open(args.resolvers) as file: if not dns_resolvers:
dns_servers = [server.strip() for server in file.readlines()] dns_resolvers = [aiodns.DNSResolver(nameservers=[server], rotate=False, timeout=args.timeout) for server in get_dns_servers()[:args.concurrency]]
while True: seed = random.randint(10**9, 10**10 - 1)
if not args.resolvers: ip_generator = rig(seed)
dns_servers = []
while not dns_servers:
try:
dns_servers = get_dns_servers()
except:
time.sleep(300)
seed = random.randint(10**9, 10**10 - 1) for ip in ip_generator:
ip_generator = rig(seed) if len(tasks) < args.concurrency:
resolver = random.choice(dns_resolvers)
task = asyncio.create_task(rdns(semaphore, ip, resolver))
tasks.append(task)
else:
done, pending = await asyncio.wait(tasks, return_when=asyncio.FIRST_COMPLETED)
tasks = list(pending)
for task in done:
ip, result = task.result()
if result:
if result in ('127.0.0.1', 'localhost'):
print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}-> {result}{colors.reset}')
elif ip in result:
result = result.replace(ip, f'{colors.ip_match}{ip}{colors.ptr}')
elif (daship := ip.replace('.', '-')) in result:
result = result.replace(daship, f'{colors.ip_match}{daship}{colors.ptr}')
print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}->{colors.reset} {colors.ptr}{result}{colors.reset}')
elif (revip := '.'.join(ip.split('.')[::-1])) in result:
result = result.replace(revip, f'{colors.ip_match}{revip}{colors.ptr}')
print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}->{colors.reset} {colors.ptr}{result}{colors.reset}')
elif result.endswith('.gov') or result.endswith('.mil'):
result = result.replace('.gov', f'{colors.spooky}.gov{colors.reset}')
result = result.replace('.mil', f'{colors.spooky}.gov{colors.reset}')
print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}->{colors.reset} {colors.ptr}{result}{colors.reset}')
elif '.gov.' in result or '.mil.' in result:
result = result.replace('.gov.', f'{colors.spooky}.gov.{colors.ptr}')
result = result.replace('.mil.', f'{colors.spooky}.mil.{colors.ptr}')
print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}->{colors.reset} {colors.ptr}{result}{colors.reset}')
else:
print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}->{colors.reset} {colors.ptr}{result}{colors.reset}')
for ip in ip_generator: results_cache.append(f'{ip}:{result}')
if len(tasks) < args.concurrency: if len(results_cache) >= 1000:
dns = random.choice(dns_servers) stamp = time.strftime('%Y%m%d')
task = asyncio.create_task(rdns(semaphore, ip, dns)) with open(f'ptr_{stamp}_{seed}.txt', 'a') as file:
tasks.append(task) file.writelines(f"{record}\n" for record in results_cache)
else: results_cache = []
done, pending = await asyncio.wait(tasks, return_when=asyncio.FIRST_COMPLETED)
tasks = list(pending)
for task in done:
ip, result = task.result()
if result:
if result in ('127.0.0.1', 'localhost'):
print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}-> {result}{colors.reset}')
elif ip in result:
result = result.replace(ip, f'{colors.ip_match}{ip}{colors.ptr}')
elif (daship := ip.replace('.', '-')) in result:
result = result.replace(daship, f'{colors.ip_match}{daship}{colors.ptr}')
print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}->{colors.reset} {colors.ptr}{result}{colors.reset}')
elif (revip := '.'.join(ip.split('.')[::-1])) in result:
result = result.replace(revip, f'{colors.ip_match}{revip}{colors.ptr}')
print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}->{colors.reset} {colors.ptr}{result}{colors.reset}')
elif result.endswith('.gov') or result.endswith('.mil'):
result = result.replace('.gov', f'{colors.spooky}.gov{colors.reset}')
result = result.replace('.mil', f'{colors.spooky}.gov{colors.reset}')
print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}->{colors.reset} {colors.ptr}{result}{colors.reset}')
elif '.gov.' in result or '.mil.' in result:
result = result.replace('.gov.', f'{colors.spooky}.gov.{colors.reset}')
result = result.replace('.mil.', f'{colors.spooky}.mil.{colors.reset}')
print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}->{colors.reset} {colors.ptr}{result}{colors.reset}')
else:
print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}->{colors.reset} {colors.ptr}{result}{colors.reset}')
results_cache.append(f'{ip}:{result}')
if len(results_cache) >= 1000:
stamp = time.strftime('%Y%m%d')
with open(f'ptr_{stamp}_{seed}.txt', 'a') as file:
file.writelines(f"{record}\n" for record in results_cache)
results_cache = []