threaded_port_scan.py

import socket # module for connections
from colorama import init, Fore # module for colored text
import argparse
from threading import Thread, Lock
from queue import Queue

# colorama initialization and Color Constants assignment
init()
GREEN = Fore.GREEN # if port open
GRAY = Fore.LIGHTBLACK_EX # if port closed
RESET = Fore.RESET # reset to default color

# threading and queue setup
NUM_OF_THREADS = 200
q = Queue()
print_lock = Lock()

def port_scan(port):
    # Scan a port on the global variable `host`
    # currently broken, need time to figure out
    try:
        s = socket.socket()
        s.connect((host, port))
    except:
        with print_lock:
            print(f"{GRAY}{host:15}:{port:5} is closed {RESET}", end="\r")
    else:
        with print_lock:
            print(f"{GREEN}{host:15}:{port:5} is open {RESET}")
    finally:
        s.close()

# gets port numbers from the queue to scan it and then add it to the done tasks
def scan_thread():
    global q
    while True:
        # get the port number from the queue
        worker = q.get()
        #scan that port number
        port_scan(worker)
        q.task_done()

# populates the queue with the port numbers and spwaning NUM_OF_THREADS threads to consume them
def main(host, ports):
    global q
    for t in range(NUM_OF_THREADS):
        # for each thread, start it
        t = Thread(target=scan_thread)
        # By setting the daemon to true, that thread will end when the main thread ends
        t.daemon = True
        # start the daemon thread
        t.start()
    for worker in ports:
        # for each port, put that port into the queue
        # to start scanning
        q.put(worker)
    # wait the for the threads to finish
    q.join()