Skip to content

Refactor main.py entry point #2

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
Chocapikk wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Refactor main.py entry point #2

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
286 changes: 120 additions & 166 deletions src/main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,199 +1,153 @@
# THIS MODULE WILL RUN MODULER CODE FROM HERE
import argparse
import sys



# ETC IMPORTS
import argparse, sys


# UI IMPORTS
from rich.console import Console
from rich.panel import Panel
console = Console()


# NSM IMPORTS
from nsm_scanner import Mass_IP_Scanner
from nsm_database import Database

console = Console()




class Program_Vars():
"""This will be used to pass vars from argparse to a class that stores them"""


api_key_ipinfo = False




class Main():
"""This wil launch program wide logic"""


data = (
"\n [bold cyan]Mass IP Scanning Framework[/bold cyan]"
"\n\n [bold yellow]Find Vulnerable Devices[/bold yellow]"
"\n\n [bold magenta]Made by NSM-Barii[/bold magenta]\n"
IRAN_PRESETS = {
"1": {
"name": "iran_ips_critical_infrastructure.txt",
"asn": "5542,6736,21170,31303,34837,35285,39200,42143,42750,42867,42907,43358,44244,44436,47603,48159,49666,50722,50992,51119,51168,51554,51732,52196,57218,57577,58169,59654,59961,60605,62039,197207,199633,200376,201442,202788,203026,205490,205899,206929,208072,209079,214419,214737",
},
"2": {
"name": "iran_ips_defense_research.txt",
"asn": "12660,29068,29577,41620,47981,48898,49792,57563,59794,61239",
},
"3": {
"name": "iran_ips_state_media.txt",
"asn": "42586,44609,47188,204999",
},
"4": {
"name": "iran_ips_financial.txt",
"asn": "16018,31182,34871,35615,41061,42990,47817,49433,50855,51460,51618,51785,52070,57241,57574,59754,60407,60516,61250,62238,203162,208493,208651,209941,210470,213682,213872,213916,215700",
},
}

PORT_PRESETS = {
"1": lambda: Database.CRITICAL_INFRASTRUCTURE_PORTS,
}

DEVICE_PRESETS = {
"iot": lambda: (Database.IOT_PORTS, None),
"nas": lambda: (Database.paths_nas, Database.paths_nas),
"router": lambda: (Database.ROUTER_PORTS, Database.paths_router),
"remote": lambda: (Database.REMOTE_PORTS, None),
"camera": lambda: (Database.CAMERA_PORTS, Database.paths_camera),
"database": lambda: (Database.DATABASE_PORTS, None),
}

PATH_MAP = {
"nas": lambda: Database.paths_nas,
"router": lambda: Database.paths_router,
"camera": lambda: Database.paths_camera,
}

BANNER = Panel(
renderable=(
"\n [bold cyan]Mass IP Scanning Framework[/bold cyan]"
"\n\n [bold yellow]Find Vulnerable Devices[/bold yellow]"
"\n\n [bold magenta]Made by NSM-Barii[/bold magenta]\n"
),
expand=False,
style="bold red",
)

panel = Panel(renderable=data, expand=False, style="bold red")


def parse_args():
parser = argparse.ArgumentParser(
description="Mass IP Scanning framework meant to find vulnerable devices left uncheck open to the internet"
)

parser.add_argument("-p", help="Port to scan.")
parser.add_argument("-t", help="Maximum number of threads to spawn.")

parser.add_argument("-t", type=int, default=250, help="Maximum number of threads to spawn.")
parser.add_argument("--save", action="store_true", help="Save all active IPs to database/ips.txt.")
parser.add_argument("--x", help="Use this to set a custom file name")


parser.add_argument("--x", help="Use this to set a custom file name")
parser.add_argument("--country", help="Only create IP Blocks within x Country")
parser.add_argument("--asn", help="Pass asn value(s) Example: 215892, 214735, 214145, 213727, 212056")
parser.add_argument("--bloom-size", help="BloomFilter capacity for global scans (default: 100000000). Warning: Higher values use more memory.")

parser.add_argument("--bloom-size", type=int, default=100_000_000, help="BloomFilter capacity for global scans (default: 100000000).")
parser.add_argument("--geo", choices=["local", "ipinfo"], help="Enable IP geolocation lookup.")
parser.add_argument("--ipinfo", help="Optional ipinfo.io API key. Defaults to none.")

parser.add_argument("--iot", action="store_true", help="Scan for IoT devices (MQTT, CoAP, mDNS).")
parser.add_argument("--nas", action="store_true", help="Scan for NAS devices (SMB, Synology, web panels).")
parser.add_argument("--ipinfo", help="Optional ipinfo.io API key.")
parser.add_argument("--iot", action="store_true", help="Scan for IoT devices (MQTT, CoAP, mDNS).")
parser.add_argument("--nas", action="store_true", help="Scan for NAS devices (SMB, Synology, web panels).")
parser.add_argument("--camera", action="store_true", help="Scan for IP cameras (RTSP, ONVIF, web interfaces).")
parser.add_argument("--router", action="store_true", help="Scan for routers and network infrastructure (admin panels, SSH, Telnet, TR-069).")
parser.add_argument("--router", action="store_true", help="Scan for routers and network infrastructure.")
parser.add_argument("--remote", action="store_true", help="Scan for remote access services (RDP, VNC, SSH, FTP).")
parser.add_argument("--database", action="store_true", help="Scan for open databases (3306, 5432, 27017, 6379, 9200).")

parser.add_argument("--show-all", action="store_true", help="Show all active IPS")
parser.add_argument("--paths", help="Manually set path for directory bruteforcing (nas, router, camera).")
parser.add_argument("--paths", choices=["nas", "router", "camera"], help="Set path for directory bruteforcing.")

if len(sys.argv) == 1:
console.print(BANNER)
parser.print_help()
sys.exit()

return parser.parse_args()

args = parser.parse_args()

def resolve_asn(args):
if args.asn in IRAN_PRESETS:
preset = IRAN_PRESETS[args.asn]
return preset["asn"], True, preset["name"]
return args.asn, args.save, args.x

if len(sys.argv) == 1:
console.print(panel)
parser.print_help(); exit()


# REQUIRED VARS
port = args.p or False
max_threads = args.t or 250


# ADDITIONS
country = args.country or False
asn = args.asn or False
bloom_size = int(args.bloom_size) if args.bloom_size else 100000000
lookup = args.geo or False
api_key_ipinfo = args.ipinfo or False
save = args.save or False
save_name = args.x or False

# WARNING
if not country:
console.print("\n[bold red][!] WARNING:[/bold red] [bold yellow]Scanning without --country will use a 100M BloomFilter limit.")
console.print("[bold yellow] After 100M IPs, duplicates may be scanned. Use --country for memory-efficient scanning.")
console.print("[bold yellow] Or increase with --bloom-size (e.g., --bloom-size 500000000) but this uses more RAM.\n")

# PRESET OPTIONS
iot = args.iot or False
nas = args.nas or False
router = args.router or False
remote = args.remote or False
camera = args.camera or False
database = args.database or False


# FOR PRESETS
paths = args.paths or False
all = args.show_all or False


# IRANIAN CRITICAL INFRASTRUCTURE ASN SHORTCUTS (must come before port assignment)
# TIER 1: CRITICAL INFRASTRUCTURE (Energy, Power Grid, Major Telecom, Transport)
if asn=="1":
asn = "5542,6736,21170,31303,34837,35285,39200,42143,42750,42867,42907,43358,44244,44436,47603,48159,49666,50722,50992,51119,51168,51554,51732,52196,57218,57577,58169,59654,59961,60605,62039,197207,199633,200376,201442,202788,203026,205490,205899,206929,208072,209079,214419,214737"
save = True
save_name = "iran_ips_critical_infrastructure.txt"

# TIER 2: INTELLIGENCE & DEFENSE RESEARCH (Universities with Strategic Programs)
elif asn=="2":
asn = "12660,29068,29577,41620,47981,48898,49792,57563,59794,61239"
save = True
save_name = "iran_ips_defense_research.txt"

# TIER 3: INFORMATION & PROPAGANDA (State Media, Broadcasting)
elif asn=="3":
asn = "42586,44609,47188,204999"
save = True
save_name = "iran_ips_state_media.txt"

# TIER 4: FINANCIAL (Major Banks, Payment Systems)
elif asn=="4":
asn = "16018,31182,34871,35615,41061,42990,47817,49433,50855,51460,51618,51785,52070,57241,57574,59754,60407,60516,61250,62238,203162,208493,208651,209941,210470,213682,213872,213916,215700"
save = True
save_name = "iran_ips_financial.txt"


# SET CONSTANTS
Mass_IP_Scanner.country = country
Mass_IP_Scanner.asn = asn
Mass_IP_Scanner.all = all
Mass_IP_Scanner.save = save
Mass_IP_Scanner.save_name = save_name
Mass_IP_Scanner.bloom_size = bloom_size


# ASSIGN PRESETS
if iot: port = Database.IOT_PORTS
elif nas: port = Database.paths_nas ; Database.paths = Database.paths_nas
elif router: port = Database.ROUTER_PORTS ; Database.paths = Database.paths_router
elif remote: port = Database.REMOTE_PORTS
elif camera: port = Database.CAMERA_PORTS ; Database.paths = Database.paths_camera
elif database: port = Database.DATABASE_PORTS

if port == "1": port = Database.CRITICAL_INFRASTRUCTURE_PORTS


if paths:
if paths in ["nas"]: Database.paths = Database.paths_nas
elif paths in ["router"]: Database.paths = Database.paths_router
elif paths in ["camera"]: Database.paths = Database.paths_camera


Database.ports = port
Database.lookup = lookup
Database.api_key_ipinfo = api_key_ipinfo



c1 = "red"; c2 = "bold green"; c3 = "bold blue"; c4 = "bold yellow"

stats = (
f"[{c1}][+] Port(s):[{c4}] {port}"
f"\n[{c1}] [+] Max Workers:[{c4}] {max_threads}"
f"\n[{c1}] [+] File Saving:[{c4}] {save}"
f"\n[{c1}] [+] GEO Lookup:[{c4}] {lookup}"
f"\n[{c1}] [+] API Key:[{c4}] {api_key_ipinfo}"
)

panel = Panel(renderable= stats,
title="Constants",
border_style="purple",
style="bold red",
expand=False
)

def resolve_port(args):
if args.p in PORT_PRESETS:
return PORT_PRESETS[args.p]()

for flag, resolver in DEVICE_PRESETS.items():
if getattr(args, flag):
port, paths = resolver()
if paths:
Database.paths = paths
return port

return args.p


def main():
args = parse_args()

if not args.country:
console.print(
"\n[bold red][!] WARNING:[/bold red] [bold yellow]Scanning without --country will use a 100M BloomFilter limit."
"\n After 100M IPs, duplicates may be scanned. Use --country for memory-efficient scanning."
"\n Or increase with --bloom-size (e.g., --bloom-size 500000000) but this uses more RAM.\n"
)

asn, save, save_name = resolve_asn(args)
port = resolve_port(args)

if args.paths and args.paths in PATH_MAP:
Database.paths = PATH_MAP[args.paths]()

Mass_IP_Scanner.country = args.country
Mass_IP_Scanner.asn = asn
Mass_IP_Scanner.all = args.show_all
Mass_IP_Scanner.save = save
Mass_IP_Scanner.save_name = save_name
Mass_IP_Scanner.bloom_size = args.bloom_size

Database.ports = port
Database.lookup = args.geo
Database.api_key_ipinfo = args.ipinfo

console.print(
f"[{c1}]========= CONSTANTS =========\n",
stats,
f"\n[{c1}]=================================",
"[red]========= CONSTANTS =========\n",
f"[red][+] Port(s):[bold yellow] {port}"
f"\n[red] [+] Max Workers:[bold yellow] {args.t}"
f"\n[red] [+] File Saving:[bold yellow] {save}"
f"\n[red] [+] GEO Lookup:[bold yellow] {args.geo}"
f"\n[red] [+] API Key:[bold yellow] {args.ipinfo}",
"\n[red]=================================",
)


Mass_IP_Scanner._main(port=port, threads=args.t)


Mass_IP_Scanner._main(port=port, threads=max_threads)
main()