Bug Hunter - Program Python Clean IPv6 extraction

[@sharingilmubarengibnu]

from rich.console import Console
from rich.panel import Panel
from rich.text import Text
from rich.table import Table
from rich import print as rprint
import re
import sys
import os
console = Console()
def clear_screen():
os.system('cls' if os.name == 'nt' else 'clear')
def show_banner():
banner = Text()
banner.append("╔══════════════════════════════════════════╗
", style="cyan")
banner.append("║ IPv6 Extractor Tool v1.0 ║
", style="cyan bold")
banner.append("║ DNS Enumeration & Analysis Tool ║
", style="cyan")
banner.append("╚══════════════════════════════════════════╝
", style="cyan")
console.print(Panel(banner, border_style="cyan"))
def show_author():
author = Table(show_header=False, border_style="cyan")
author.add_row("[cyan]Developer[/cyan]", "[white]:[/white]", "[yellow]Ibnu Rusdianto[/yellow]")
author.add_row("[cyan]Version[/cyan]", "[white]:[/white]", "[yellow]1.0[/yellow]")
author.add_row("[cyan]Github[/cyan]", "[white]:[/white]", "[yellow]github.com/ibnurusdianto[/yellow]")
console.print(Panel(author, title="[cyan]Author Information[/cyan]", border_style="cyan"))
def show_description():
desc = """
This tool is designed to extract IPv6 addresses from DNS enumeration results.
It processes MassDNS output format and cleanly extracts only the IPv6 addresses.
Features:
• Clean IPv6 extraction
• Duplicate removal
• Formatted output
• File handling
"""
console.print(Panel(desc, title="[cyan]Tool Description[/cyan]", border_style="cyan"))
def show_usage():
usage = """
[cyan]Usage:[/cyan]
1. Place your input file (output_ipv6.txt) in the same directory
2. Select option 1 to process the file
3. Results will be saved to 'extracted_ipv6.txt'
[cyan]Input Format:[/cyan]
domain.example.com. AAAA 2606:4700:20::681a:804
[cyan]Output Format:[/cyan]
2606:4700:20::681a:804
"""
console.print(Panel(usage, title="[cyan]Usage Guide[/cyan]", border_style="cyan"))
def process_file(filename):
try:
with open(filename, 'r') as file:
content = file.read()
# Extract IPv6 addresses using regex
ipv6_pattern = r'AAAA\s+([0-9a-fA-F:]+)'
ipv6_addresses = re.findall(ipv6_pattern, content)
# Remove duplicates while preserving order
unique_ipv6 = list(dict.fromkeys(ipv6_addresses))
# Save to output file
with open('extracted_ipv6.txt', 'w') as outfile:
for ip in unique_ipv6:
outfile.write(f"{ip}
")
# Show results
result_table = Table(title="Processing Results", border_style="cyan")
result_table.add_column("Description", style="cyan")
result_table.add_column("Count", style="green")
result_table.add_row("Total IPv6 addresses found", str(len(ipv6_addresses)))
result_table.add_row("Unique IPv6 addresses", str(len(unique_ipv6)))
result_table.add_row("Output file", "extracted_ipv6.txt")
console.print(result_table)
return True
except FileNotFoundError:
console.print("[red]Error: Input file not found![/red]")
return False
except Exception as e:
console.print(f"[red]Error: {str(e)}[/red]")
return False
def main_menu():
while True:
clear_screen()
show_banner()
menu = Table(show_header=False, border_style="cyan")
menu.add_row("[cyan]1.[/cyan]", "[white]Process IPv6 File[/white]")
menu.add_row("[cyan]2.[/cyan]", "[white]Show Tool Description[/white]")
menu.add_row("[cyan]3.[/cyan]", "[white]Show Usage Guide[/white]")
menu.add_row("[cyan]4.[/cyan]", "[white]Show Author Information[/white]")
menu.add_row("[cyan]5.[/cyan]", "[white]Exit[/white]")
console.print(Panel(menu, title="[cyan]Main Menu[/cyan]", border_style="cyan"))
choice = console.input("[cyan]Enter your choice (1-5): [/cyan]")
if choice == "1":
clear_screen()
show_banner()
console.print("[cyan]Processing file...[/cyan]")
if process_file('output_ipv6.txt'):
console.input("
[green]Press Enter to continue...[/green]")
elif choice == "2":
clear_screen()
show_banner()
show_description()
console.input("
[cyan]Press Enter to continue...[/cyan]")
elif choice == "3":
clear_screen()
show_banner()
show_usage()
console.input("
[cyan]Press Enter to continue...[/cyan]")
elif choice == "4":
clear_screen()
show_banner()
show_author()
console.input("
[cyan]Press Enter to continue...[/cyan]")
elif choice == "5":
clear_screen()
console.print("[cyan]Thank you for using IPv6 Extractor Tool![/cyan]")
sys.exit(0)
else:
console.print("[red]Invalid choice! Please try again.[/red]")
console.input("
Press Enter to continue...")
if __name__ == "__main__":
main_menu()