python3实现端口扫描脚本

啊 好久没写代码了，想着多线程啥的都忘得差不多了
就下面这个啦，

#coding=utf-8
import socket,threading
from socket import *

def tcp_connect(ip,port):
    tcpsock=socket(AF_INET,SOCK_STREAM)
    tcpsock.settimeout(0.5)
    try:
        result=tcpsock.connect_ex((ip,port))
        if(result==0):
            print ("%s:%s 开启" % (ip, port))
        else:
            #print ("%s:%s 关闭" % (ip, port))
            pass
        tcpsock.close()
    except socket.error as e:
        print ("[!]错误",e)

def url_ip(url):
    if('http://' in url or 'https://' in url):
        url=url[url.find('://')+3:]
        targetip = socket.getaddrinfo(url, 'http')[0][4][0]
        print (targetip)
        port_scan(targetip)
    else:
        targetip = socket.getaddrinfo(url, 'http')[0][4][0]
        print (targetip)
        port_scan(targetip)

def port_scan(ip):
    for i in range(200):
        tcp_connect(ip,i)

def main():
    print "-------------"
    print "1:ip扫描端口"
    print "2:url扫描端口"
    print "-------------"
    q=raw_input("选择功能:")
    if(int(q)==1):
        w=raw_input("ip=")
        t=threading.Thread(target=port_scan,args=(w,))
        t.start()
    else:
        e=raw_input("url=")
        url_ip(e)
main()

#!/usr/bin/python3
# -*- coding: utf-8 -*-
from socket import *
import threading

lock = threading.Lock()
openNum = 0
threads = []

def portScanner(host,port):
    global openNum
    try:
        s = socket(AF_INET,SOCK_STREAM)
        s.connect((host,port))
        lock.acquire()
        openNum+=1
        print('[+] %d open' % port)
        lock.release()
        s.close()
    except:
        pass

def main():
    setdefaulttimeout(1)
    for p in range(1,6558):
        t = threading.Thread(target=portScanner,args=('14.215.177.38',p))
        threads.append(t)
        t.start()

    for t in threads:
        t.join()

    print('[*] The scan is complete!')
    print('[*] A total of %d open port ' % (openNum))

if __name__ == '__main__':
    main()

这是协程

# -*-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 = raw_input("[+]输入扫描目标IP:")
    print("[*]正在扫描")
    scan_port(ip)

def scan_web():
    """扫描目标网址"""
    web = raw_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,200):
        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(raw_input("[+]请输入选择:"))
    if 1 == uc:
        scan_ip()
    elif 2 == uc:
        scan_web()
    else:
        print("[!]输入有误")

if __name__ == "__main__":
    main()