python3扫描_Python3 实现高效的多线程端口扫描器

这几天写了一个端口扫描器，可以快速的扫描发现目前电脑开启的端口，思路是通过socket来创建TCP连接来测试端口是否开放。

#！usr/bin/env python

# -*-coding:UTF-8 -*-

# @Time: 2018/9/7 11:07

# @Author:gumgui

# @File: 端口扫描.py

import time

import socket

from gevent import monkey

monkey.patch_all()

import gevent

import gevent.pool

def TCP_connect(ip,port):

"""模拟TCP连接"""

TCP_sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

TCP_sock.settimeout(0.5) #设置连接超时

try:

result = TCP_sock.connect_ex((ip,int(port)))

if result == 0:

print("[*]%s 端口 开启\t"%port)

else:

# print("[!]%s端口 关闭"%port)

pass

TCP_sock.close()

except socket.error as e:

print("[!]错误:",e)

def scan_ip():

"""扫描目标IP"""

ip = input("[+]输入扫描目标IP:")

print("[*]正在扫描")

scan_port(ip)

def scan_web():

"""扫描目标网址"""

web = input("[+]输入扫描网址:")

if "http://" in web or "https://" in web:

web = web[web.find('://')+3:]

print(web)

print("[*]正在分析网站服务器IP")

try:

server_ip = socket.gethostbyname(str(web))

print("[*]服务器IP为%s"%server_ip)

scan_port(server_ip)

except Exception as e:

print("[!]服务器IP获取失败")

pass

def scan_port(ip):

"""扫描端口"""

print("[*]开始扫描目标端口")

start = time.time()

g = gevent.pool.Pool(50) #设置线程数

run_list = []

for port in range(1,65535):

run_list.append(g.spawn(TCP_connect,ip,port))

gevent.joinall(run_list)

end = time.time()

print("[*]总耗时%s"%time.strftime("%H:%M:%S",time.gmtime(end-start)))

def main():

print(

"""

1.通过IP扫描端口

2.通过网址扫描端口

"""

)

uc = int(input("[+]请输入选择:"))

if 1 == uc:

scan_ip()

elif 2 == uc:

scan_web()

else:

print("[!]输入有误")

if __name__ == "__main__":

main()

运行结果如下：