#!/usr/bin/env python3
# -*- coding: UTF-8 -*-
# 編程思路
# 1、定義portscan函數(shù)，創(chuàng)建socket對(duì)象進(jìn)行TCP端口掃描
# 2、啟動(dòng)多線(xiàn)程運(yùn)行PortScan函數(shù)
# 3、記錄并輸出掃描結(jié)果與時(shí)間

import socket  # 創(chuàng)建TCP連接
from threading import Thread  # 多線(xiàn)程模塊，進(jìn)行多線(xiàn)程掃描
import time  # 時(shí)間模塊，記錄掃描所需時(shí)間


def main():
    target = input("IP:")
    start_time = time.time()
    s_time = time.ctime()
    print("[*] Start port scan at %s" % s_time)
    for port in range(1, 65536):  # 定義掃描的端口范圍
        # 2、啟動(dòng)多線(xiàn)程運(yùn)行PortScan函數(shù)
        t = Thread(target=portscan, args=(target, port))  # 創(chuàng)建線(xiàn)程對(duì)象
        t.start()  # 開(kāi)始線(xiàn)程

    end_time = time.time()
    print("[*] All done in %.2f s" % (end_time - start_time))


def portscan(target, port):
    # 1、定義portscan函數(shù)，進(jìn)行TCP端口掃描
    try:
        client = socket.socket(
            socket.AF_INET, socket.SOCK_STREAM)  # 創(chuàng)建socket對(duì)象
        client.connect((target, port))  # 建立TCP連接
        print("[*] %s:%d端口開(kāi)放" % (target, port))
        client.close()
    except:
        pass  # 捕獲異常


if __name__ == "__main__":
    main()