# coding: utf-8
import os
import sys
try:
import xml.etree.cElementTree as ET
except ImportError:
import xml.etree.ElementTree as ET
reload(__import__('sys')).setdefaultencoding('utf-8')
'''
nmap -Pn --open -n -vv -F -oX d:\\xxx.xml 10.129.246.1/24
nmap -n -vv --open -oX d:\\vm.xml 192.168.106.1/24
--open這個(gè)參數(shù)必須加
nmap -Pn -n -vv -p80 -oX d:\\xxx.xml 10.129.246.1/24 檢測(cè)單一端口的就不用執(zhí)行端口提了
'''
def deal_nmap_xml(xml_name, save_name):
try:
root = ET.fromstring(xml_name)
ip_list = []
except:
print u"請(qǐng)檢查nmap掃描文件標(biāo)簽是否完整"
sys.exit()
res = open(save_name + '.txt', 'w') # 輕易不要改動(dòng)
for host in root.findall('host'):
if len(host) > 3:
pass
#print '\n', str(len(host)), # 打印host標(biāo)簽中子元素個(gè)數(shù)
if host[0].get('state') == "up": # 判斷IP是否存活
ip = host[1].get('addr') # 提取IP地址
#print ip,
ip_list.append(ip) # 驗(yàn)證存活I(lǐng)P個(gè)數(shù)
ip_ = '\n' + ip + '\t'
res.writelines(ip_)
# 提取端口
if len(host) == 6:
for port in host[4][1:]: # 若確認(rèn)端口開放，但沒有提取出端口請(qǐng)修改host[4][此處+1試試:]，加一后為host[4][2:]，下面的幾處方法一樣
#print port.get('portid'),
port_ = str(port.get('portid')) + ','
res.write(port_)
elif len(host) == 5:
for port in host[3][2:]:
# print port.tag,
#print port.get('portid'),
port_ = str(port.get('portid')) + ','
res.write(port_)
elif len(host) == 4:
for port in host[3][1:]:
#print port.get('portid'),
port_ = str(port.get('portid')) + ','
res.write(port_)
elif len(host) < 4:
print host[0].get('state')
res.close()
print 'Alive IP Total:{} '.format(len(ip_list))
#根據(jù)端口提取相應(yīng)的IP
def get_ip(save_name, port):
with open(save_name+'.txt', 'r') as xxx:
port_info = xxx.readlines()
res_name = save_name+'-out'
with open(res_name + '.txt', 'w+') as save_file:
for x in port.split(','):
for line_info in port_info:
try:
if x in line_info.split('\t')[1].split(','):
line =line_info.split('\t')[0] +":"+x+ '\n'
#print line,
save_file.writelines(line)
except:
pass
print "Done!"
def main(path, port):
if os.path.isfile(path):
save_name = path.split('.')[0]
print save_name
bd = open(path, 'r').read() # nmap的掃描結(jié)果文件
print 'Start extracting IP&Port'
deal_nmap_xml(bd, save_name)
get_ip(save_name, port)
else:
for i in os.listdir(path.strip()):
if i.endswith('.xml'):
#print i
xml_path = path + os.path.sep + i
save_name = i.split('.')[0]
print save_name
bd = open(xml_path, 'r').read() # nmap的掃描結(jié)果文件
print 'Start extracting IP&Port'
deal_nmap_xml(bd, save_name)
get_ip(save_name, port)
if __name__ == "__main__":
#批量提取
#123.py "C:\Users\Desktop\portscan\xml" 21
#123.py "C:\Users\Desktop\portscan\xml" "21,22,23"
#單個(gè)文件提取，將nmap掃描文件放在123.py同一目錄
#123.py nmap.xml 21
#123.py nmap.xml "21,22,23"
path = sys.argv[1] #1.若是單個(gè)文件，需輸入完整路徑(文件和此腳本在同一目錄可不寫路徑) 2.可以輸入xml文件所在目錄
port = sys.argv[2] #提取的端口，多個(gè)使用','逗號(hào)隔開。如："21,22,23,25,110"
main(path, port)
批量提取單個(gè)端口
123.py "C:\Users\^__^\Desktop\portscan\xml" 21
圖片.png
批量提取多個(gè)端口
123.py "C:\Users\^__^\Desktop\portscan\xml" "21,22,23,1521"
圖片.png
單個(gè)文件提取多個(gè)端口
nmap.xml文件需與123.py在同一目錄下，
123.py "nmap.xml" "21,22,23,1433,"
若不在則如下：
123.py ""C:\Users\^__^\Desktop\portscan\nmap.xml" "21,22,23,1433,"
提取單個(gè)端口的也同樣。
圖片.png
單個(gè)文件提取單個(gè)端口
123.py "nmap.xml" 21
圖片.png
