python端口扫描系统实现方法
本文实例讲述了python端口扫描系统实现方法。分享给大家供大家参考。具体实现方法如下：
该程序的主要功能如下：
1.从自有API接口获取所有的外网IP段；
2.用Nmap遍历扫描所有的IP段，-oX生成XML的扫描报告；
3.用xml.etree.ElementTree模块方法读取XML文件，将ip，开放端口，对应服务等写入Mysql数据库。
功能很简单，没有满足老大高大上的需求，所以这个小项目就这么英勇的挂掉了！~~~完全都还没考虑程序异常终止，扫描服务器异常歇菜的情况。
贴上代码：
#coding:utf-8
importsys,os,time,subprocess
importMySQLdb
importre,urllib2
importConfigParser
fromIPyimportIP
importxml.etree.ElementTreeasET
nowtime=time.strftime('%Y-%m-%d',time.localtime(time.time()))
configpath=r'c:portscanconfig.ini'
#传入api接口主路径,遍历获取所有的ip列表，用IPy模块格式成127.0.0.1/24的格式
defgetiplist(ipinf):
serverarea=['tj101','tj103','dh','dx']
iplist=[]
forareainserverarea:
ipapi=urllib2.urlopen(ipinf+area).read()
foripinipapi.split('n'):
#判断如果ip列表不为空，转换成ip/网关格式,再格式化成ip/24的格式
ifip:
ip=ip.replace('_','/')
ip=(IP(ip))
iplist.append(str(ip))
ipscan(iplist,nmapathx)
#传递ip地址文件和nmap路径
defipscan(iplist,nmapath):
#古老的去重，对ip文件中的ip地址进行去重
newiplist=[]
scaniplist=[]
foripiniplist:
ifipnotinnewiplist:
newiplist.append(ip)
#遍历所有ip段，批量扫描，生成xml格式报告
foripinnewiplist:
filename=nowtime+ip.split('/')[0]+'.xml'
filepath=r"c:portscanscanres\"
nmapcmd=nmapath+'-PT'+ip.strip('rn')+'-oX'+filepath+filename
os.system(nmapcmd)
scaniplist.append(ip)
writeinmysql(scaniplist)
#入库模块是某大婶发写好的给我我只是简单修改了哈，主要是xml.etree.ElementTree模块。
defwriteinmysql(scaniplist):
filepath=r"c:portscanscanres"
foripinscaniplist:
xmlfile=filepath+'\'+ip+'.xml'
#缩进哈发文章的时候临时改的，懒得缩进了
root=ET.parse(xmlfile).getroot()
allhost=root.findall('host')
conn=MySQLdb.connect(host='10.5.10.57',user='nxadmin',passwd='nxadmin.com',port=3306,db='scandb',charset='utf8')
cur=conn.cursor()
forhostinallhost:
address=host.find('address')
#首先判断端口是不是open的，如果是再进行入库
forportinhost.find('ports').findall('port'):
ifport.find('state').attrib['state']=="open":
ip=address.attrib['addr']
portval=port.attrib['portid']
state=port.find('state').attrib['state']
sql="INSERTINTOportscan(ip,port,state)VALUES(%s,%s,%s)"
params=[ip,portval,state]
cur.execute(sql,params)
conn.commit()
cur.close()
conn.close()
if__name__=="__main__":
#读取配置文件中要扫描的IPapiurl和nmap安装文件路径
config=ConfigParser.ConfigParser()
config.readfp(open(configpath,'rb'))
nmapathx=config.get('nmap','nmapath')
ipinf=config.get('ip','ipinf')
getiplist(ipinf)配置文件c:portscanconfig.ini中主要是api接口主url，nmap安装路径。
感兴趣的朋友可以进一步完善该实例的功能。希望本文所述对大家的Python程序设计有所帮助。