Pythonでポートスキャンするプログラムを作成

portscan.py

from socket import socket, gethostbyname, AF_INET, SOCK_STREAM
from threading import Thread, Lock
from time import sleep, time


def _thread(address, ports, opened, lock):
    while True:
        with lock:
            try:
                port = ports.pop()
            except IndexError:
                break
        s = socket(AF_INET, SOCK_STREAM)
        s.settimeout(5)
        if s.connect_ex((address, port)) == 0:
            with lock:
                opened.append(port)
        if port % 2000 == 0:
            print("Now.. ", port)
        s.close()


def port_scan(address, start=1, finish=65535, t=8000):
    s = time()
    threads = list()
    ports = list(reversed(range(start+1, finish+1)))
    opened = list()
    lock = Lock()
    host = gethostbyname(address)
    print("Port scan", address, "(", host, ") form", start, "to", finish)
    for i in range(t):
        threads.append(Thread(
            target=_thread, args=(host, ports, opened, lock), daemon=True))
    for t in threads:
        t.start()
        sleep(0.005)
    print(len(threads), "thread working.")
    for t in threads:
        t.join()
    print("Finished", round(time()-s, 3), "Sec")
    return opened


if __name__ == '__main__':
    r = port_scan('google.com')
    print(r)