滲透測試python編程之端口掃描

一、環境

    windows 10 +python2.7

二、代碼

#!/bin/env python
# coding=utf-8

import optparse
from threading import *
from socket import *

screenLock = Semaphore(value=1)


def connScan(tgtHost, tgtPort):
    global connSkt
    try:
        connSkt = socket(AF_INET, SOCK_STREAM)
        connSkt.connect((tgtHost, tgtPort))
        connSkt.send('ViolentPython\r\n')
        total_data = []
        while True:
            data = connSkt.recv(20480)
            if not data:
                break
            total_data.append(data)
        screenLock.acquire()
        print '[+]%d/tcp open' %tgtPort
        print '[+] '.join(total_data)
    except:
        screenLock.acquire()
        print '[-]%d/tcp closed' % tgtPort
    finally:
        screenLock.release()
        connSkt.close()


def portScan(tgtHost, tgtPorts):
    try:
        tgtIP = gethostbyname(tgtHost)
    except:
        print "[-]Cannot resolve '%s': Unknown host" % tgtHost
        return
    try:
        tgtName = gethostbyaddr(tgtIP)
        print '\n[+] Scan Results for: ' + tgtName[0]
    except:
        print '\n[+] Scan Results for: ' + tgtIP
    setdefaulttimeout(2)
    for tgtPort in tgtPorts:
        print 'Scanning port ' + tgtPort
        t = Thread(target=connScan, args=(tgtHost, int(tgtPort)))
        t.start()


def main():
    parser = optparse.OptionParser('usage %prog -H' +\
                                   '<target host> -p <target port>')
    parser.add_option('-H', dest='tgtHost', type='string', \
                      help='specify target host')
    parser.add_option('-p', dest='tgtPort', type='int', \
                      help='specify target port[s] separated by comma')
    (options, args) = parser.parse_args()
    tgtHost = options.tgtHost
    tgtPorts = str(options.tgtPort).split(', ')

    if (tgtHost is None) | (tgtPorts[0] is None):
        print '[-] You must specify a target host and port[s].'
        exit(0)
    portScan(tgtHost, tgtPorts)

if __name__ == '__main__':
    main()

cmd下運行python PortScan.py -H IP -p port