本文实例讲述了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(\’n\’):
#判断如果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(\’rn\’)+\’ -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.5.10.57\’,user=\’nxadmin\’,passwd=\’nxadmin.com\’,port=3306,db=\’scandb\’,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安装路径。
感兴趣的朋友可以进一步完善该实例的功能。希望本文所述对大家的Python程序设计有所帮助。