新手学Python之端口扫描系统
无聊写的一个小小的程序，主要功能如下： 1，从自有API接口获取所有的外网IP段； 2，用Nmap 遍历扫描所有的IP段，-oX 生成XML的扫描报告； 3，用xml.etree.ElementTree模块方法读取XML文件，将ip，开放端口，对应服务等写入Mysql数据库。 功能很简单，没有满足老大高大上的需求，所以这个小项目这么英勇的挂掉了！~~~完全都还没考虑程序异常终止，扫描服务器异常歇菜的情况。 贴上代码： #coding:utf-8 import sys，os，time，subprocess import MySQLdb import re，urllib2 import ConfigParser from IPy import IP import xml.etree.ElementTree as ET nowtime = time.strftime('%Y-%m-%d'，time.localtime(time.time())) configpath=r'c:portscanconfig.ini' #传入api接口主路径，遍历获取所有的ip列表，用IPy模块格式成127.0.0.1/24的格式 def getiplist(ipinf): serverarea=['tj101'，'tj103'，'dh'，'dx'] iplist=[] for area in serverarea: ipapi=urllib2.urlopen(ipinf+area).read() for ip in ipapi.split(' '): #判断如果ip列表不为空，转换成ip/网关格式，再格式化成ip/24的格式 if ip: ip=ip.replace('_'，'/') ip=(IP(ip)) iplist.append(str(ip)) ipscan(iplist，nmapathx) #传递ip地址文件和nmap路径 def ipscan(iplist，nmapath): #古老的去重，对ip文件中的ip地址进行去重 newiplist=[] scaniplist=[] for ip in iplist: if ip not in newiplist: newiplist.append(ip) #遍历所有ip段，批量扫描，生成xml格式报告 for ip in newiplist: filename=nowtime+ip.split('/')[0]+'.xml' filepath=r"c:portscanscanres\" nmapcmd=nmapath+' -PT '+ip.strip(' ')+' -oX '+filepath+filename os.system(nmapcmd) scaniplist.append(ip) writeinmysql(scaniplist) #入库模块是某大婶发写好的给我 我只是简单修改了哈，主要是xml.etree.ElementTree模块。 def writeinmysql(scaniplist): filepath=r"c:portscanscanres" for ip in scaniplist: xmlfile=filepath+'\'+ip+'.xml' #缩进哈 发文章的时候临时改的，懒得缩进了 root=ET.parse(xmlfile).getroot() allhost=root.findall('host') conn=MySQLdb.connect(host='10.1.11.11'，user='nxadmin'，passwd='nxadmin.com'，port=3306，db='scandatabase'，charset='utf8') cur= conn.cursor() for host in allhost: address = host.find('address') #首先判断端口是不是open的，如果是再进行入库 for port in host.find('ports').findall('port'): if port.find('state').attrib['state']=="open": ip=address.attrib['addr'] portval=port.attrib['portid'] state=port.find('state').attrib['state'] sql = "INSERT INTO portscan (ip，port，state) VALUES(%s，%s，%s)" params=[ip，portval，state] cur.execute(sql，params) conn.commit() cur.close() conn.close() if __name__=="__main__": #读取配置文件中要扫描的IP apiurl和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安装路径。
