scanner.py

#/usr/bin/python3.12

from argparse import ArgumentParser
import socket
from threading import Thread
from time import time 

open_ports = []

def prepare_args():
    """prepare arguments
    
        return:
           args(argparse.Namespace)
    """
    parser = ArgumentParser(description="Python based fast port scanner", usage="%(prog)s 192.168.1.1", epilog = "example - %(prog)s -s 20 -e 40000 -t 500 -V 192.8.1.1 ")
    parser.add_argument(metavar="IPV4",dest="ip",help="host to scan")
    parser.add_argument("-s","--start",dest="start",metavar="",type=int, help="starting port", default=1)
    parser.add_argument("-e","--end",dest="end",metavar="",type=int,help="ending port",default=65535)
    parser.add_argument("-t","--threads",dest="threads",metavar="",type=int,help="thread hopping",default=500)
    parser.add_argument("-V","--verbose",dest="verbose",action="store_true",help="verbose output")
    parser.add_argument("-v","--version",action="version",version="%(prog)s 3.0",help="display version")
    args = parser.parse_args()
    return args 

def prepare_ports(Start:int,end:int):
    """generator function for ports

       arguments:
         start(int) - starting port
         end(int) - ending port
     """
    for port in range(Start,end+1):
        yield port

def scan_port():
    """scan ports
    
    """    
    while True:
        try:
            s = socket.socket()
            s.settimeout(1)
            port = next(ports)
            s.connect((arguments.ip,port))
            open_ports.append(port)
            if arguments.verbose:
                print(f"\r{open_ports}", end="") 
        except (ConnectionRefusedError,socket.timeout):
           continue
        except StopIteration:
            break 


def prepare_threads(threads:int):
    """create,start,join threads

        arguments:
            threads(int) - number of threads to use
     """

    thread_list = []
    for _ in range(threads+1):
        thread_list.append(threads(target=scan_port))

    for thread in thread_list:
        thread.start()

    for thread in thread_list:
        thread.join()

        

if __name__ == "__main__":
   arguments = prepare_args()
   ports = prepare_ports(arguments.start,arguments.end)
   start_time = time()
   prepare_threads(arguments.threads)
   end_time = time()
   if arguments.verbose:
       print()   
   print (f"Open Ports Found - {open_ports}")
   print (f"Time Taken - {round(end_time-start_time),2}")