利用Python實(shí)現(xiàn)端口掃描器的全過程

import socket
from datetime import datetime
from multiprocessing.dummy import Pool as ThreadPool


class ScanPort:
    def __init__(self):
        self.ip = None

    def scan_port(self, port):
        try:
            s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            res = s.connect_ex((self.ip, port))
            if res == 0:  # 端口開啟
                print('Ip:{} Port:{} IS OPEN'.format(self.ip, port))
            else:
                print('Ip:{} Port:{}: IS NOT OPEN'.format(self.ip, port))
        except Exception as e:
            print(e)
        finally:
            s.close()

    def start(self):
        remote_server = input("輸入要掃描的遠(yuǎn)程主機(jī):")
        self.ip = socket.gethostbyname(remote_server)
        ports = [i for i in range(1, 1025)]
        socket.setdefaulttimeout(0.5)
        # 開始時(shí)間
        t1 = datetime.now()
        # 設(shè)置多進(jìn)程
        threads = []
        pool = ThreadPool(processes=8)
        pool.map(self.scan_port, ports)
        pool.close()
        pool.join()

        print('端口掃描已完成，耗時(shí)：', datetime.now() - t1)


ScanPort().start()