快捷導(dǎo)航

python端口掃描系統(tǒng)實(shí)現(xiàn)方法

更新時間：2014年11月19日 11:59:55 投稿：shichen2014

這篇文章主要介紹了python端口掃描系統(tǒng)實(shí)現(xiàn)方法,可實(shí)現(xiàn)簡單的外網(wǎng)IP掃描及寫入MySQL數(shù)據(jù)庫等功能,需要的朋友可以參考下

本文實(shí)例講述了python端口掃描系統(tǒng)實(shí)現(xiàn)方法。分享給大家供大家參考。具體實(shí)現(xiàn)方法如下：

該程序的主要功能如下：

1. 從自有API接口獲取所有的外網(wǎng)IP段；
2. 用Nmap 遍歷掃描所有的IP段，-oX 生成XML的掃描報(bào)告；
3. 用xml.etree.ElementTree模塊方法讀取XML文件，將ip，開放端口，對應(yīng)服務(wù)等寫入Mysql數(shù)據(jù)庫。
功能很簡單，沒有滿足老大高大上的需求，所以這個小項(xiàng)目就這么英勇的掛掉了！~~~完全都還沒考慮程序異常終止，掃描服務(wù)器異常歇菜的情況。
貼上代碼：

復(fù)制代碼代碼如下:

#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列表不為空，轉(zhuǎn)換成ip/網(wǎng)關(guān)格式,再格式化成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地址進(jìn)行去重 

newiplist=[] 

scaniplist=[] 

for ip in iplist: 

if ip not in newiplist: 

newiplist.append(ip) 

#遍歷所有ip段，批量掃描，生成xml格式報(bào)告 

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) 

#入庫模塊是某大嬸發(fā)寫好的給我 我只是簡單修改了哈，主要是xml.etree.ElementTree模塊。 

def writeinmysql(scaniplist): 

filepath=r"c:portscanscanres"

for ip in scaniplist: 

xmlfile=filepath+'\'+ip+'.xml' 

#縮進(jìn)哈 發(fā)文章的時候臨時改的，懶得縮進(jìn)了 

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的，如果是再進(jìn)行入庫 

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)