藉助nmap寫一個簡單的python掃描 全連接

# coding=UTF-8
import optparse
import socket
import threading

screenLock = threading.Semaphore(value=1)
def connScan(tgtHost, tgtPort):
	try:
		connSkt = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
		connSkt.connect((tgtHost, tgtPort))
                connSkt.send('ViolentPython\r\n')
                results = connSkt.recv(100)
                screenLock.acquire()
		print('[+]%d/tcp open' % tgtPort)
                print('[+] ' + str(results))
	except:
                screenLock.acquire()
		print('[-]%d/tcp closed' % tgtPort)
        finally:
            screenLock.release()
            connSkt.close()

def portScan(tgtHost, tgtPorts):
	try:
		tgtIP = socket.gethostbyname(tgtHost)
	except:
		print("[-] Cannot resolve '%s': Unknown host" % tgtHost)
		return
	try:
		tgtName = socket.gethostbyaddr(tgtIP)
		print('\n[+] Scan Results for: ' + tgtName[0])
	except:
		print('\n[+] Scan Results for: ' + tgtIP)
	socket.setdefaulttimeout(1)
	for tgtPort in tgtPorts:
		print('Scanning port ' + str(tgtPort))
		t = threading.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')
    (option, args)=parser.parse_args()
    tgtHost = option.tgtHost
    tgtPort = option.tgtPort
    args.append(tgtPort)
    if((tgtHost==None)|(tgtPort==None)):
        print('[-] You must specify a target host and port[s]!')
        exit(0)
    portScan(tgtHost, args)
    
if __name__ == '__main__':
    main()