利用 python-nmap模块进行端口扫描,简单的实现方式

实现了三种输入方式

1)单个端口扫描,例如 python scanport.py -H host -P port

2)多个端口扫描,例如 python scanport.py -H host -P port1,port2,.....

3)连续端口扫描,例如 python scanport.py -H host -P start_port-end_port

#coding:utf-8

import nmap

import optparse

import sys

class INFO:

HEADER = '\033[95m'

OKBLUE = '\033[94m'

OKGREEN = '\033[92m'

WARNING = '\033[93m'

ARNING = '\033[93m'

FAIL = '\033[91m'

ENDC = '\033[0m'

BOLD = '\033[1m'

UNDERLINE = '\033[4m'

def scanport(host, port):

nm =nmap.PortScanner()

try:

result = nm.scan(host,port)

state = result['scan'][host]['tcp'][int(port)]['state']# 多层字典嵌套取值

if state =='open':

print INFO.OKBLUE+'[*] '+host+ ' tcp/'+port+" "+state+INFO.ENDC

else:

print INFO.WARNING+'[*] '+host+ ' tcp/'+port+" "+state+INFO.ENDC

except Exception,e:

raise e

if __name__ == '__main__':

parser = optparse.OptionParser("usage %prog -H -p or -p -")

parser.add_option('-H', dest='Host', type='string', help='specify target host')

parser.add_option('-P', dest='Port', type='string', help='specify target port[s] separated by "-"'or ",")

(options, args) = parser.parse_args()

if options.Host ==None or options.Port==None:

print parser.usage

sys.exit()

host = options.Host

ports = options.Port

print host,ports

#如果是连续的端口

if '-' in ports:

ports_list = ports.split('-')

for p in range(int(ports_list[0]),int(ports_list[1])+1):

scanport(host, str(p))

#其他的情况

else:

ports_list = ports.split(',')

for p in ports_list:

scanport(host, p)

上面的代码有点复杂,改良之后的代码如下:

优化结构

import nmap,argparse, sys

class INFO:

HEADER = '\033[95m'

OKBLUE = '\033[94m'

OKGREEN = '\033[92m'

WARNING = '\033[93m'

ARNING = '\033[93m'

FAIL = '\033[91m'

ENDC = '\033[0m'

BOLD = '\033[1m'

UNDERLINE = '\033[4m'

def scan_port(host, port):

nm = nmap.PortScanner()

try:

result= nm.scan(host,port)

print result

state = result['scan'][host]['tcp'][int(port)]['state']

if state == 'open':

print INFO.OKBLUE+'[*] '+host+ ' tcp/'+port+" "+state+INFO.ENDC

else:

print INFO.WARNING + '[*] ' + host + ' tcp/' + port + " " + state + INFO.ENDC

except Exception,e:

raise e

def get_args():

parger = argparse.ArgumentParser(description='port scan script')

parger.add_argument('-H', '--host', help='specify target host')

parger.add_argument('-P', '--port', help='specify target port')

args = parger.parse_args()

print args

if args.host == None:

parger.error('host is required')

sys.exit()

if args.port == None:

parger.error('host is required')

sys.exit()

else:

return args

def main():

args = get_args()

if '-' in args.port:

ports_list = args.port.split('-')

for p in range(int(ports_list[0]),int(ports_list[1])+1):

scan_port(args.host, str(p))

else:

ports_list = args.port.split(',')

for p in ports_list:

scan_port(args.host, p)

if __name__ == '__main__':

main()