#coding:utf-8

import os,sys,xml.dom.minidom,multiprocessing

from sendmail import *

from time import sleep

def dnsbrute():

try:

domain ='XXX.com' #指定要进行域名爆破的域

dnsserver ='114.114.114.114' #指定dns服务器

os.system('rm-rf *.xml')

os.system('dnsenum-f wydomain.csv --noreverse --dnsserver '+str(dnsserver)+' '+str(domain))

while True:

ifos.system('ps -ef |grep dnsenum |grep -v grep') != 0:

IPlist= []

fori in os.popen('cut -d/ -f 1 '+str(domain)+'_ips.txt'):

IPlist.append(str(i).strip())

break

except Exception,e:

print e

return IPlist

def nmap_port_scan(i):

try:

cmd = 'nmap -sV--open -n -Pn -oX '+i+'.xml '+i+'1>/dev/null 2>&1'

os.system(cmd)

except Exception,e:

print e

def multiprocess_scan(IPlist):

try:

filename = []

process = []

for i in IPlist:

filename.append(i+'.xml')

process.append(multiprocessing.Process(target=nmap_port_scan,args=(i,),name="process-"+ i))

for x inprocess:

x.start()

x.join()

except Exception,e:

print e

return filename

def port_xml_parser(filename):

try:

total =len(filename)

finished = 0

remainder =len(filename)

print '正在进行nmap扫描,请稍等:'

while True:

res =os.system('ps -ef |grep nmap |grep -v grep |grep -v "/dev/null"1>/dev/null 2>&1')

ifres==0:

sleep(1)

status= int(os.popen('ps -ef |grep nmap |grep -v grep |grep -v "/dev/null"|wc -l 1>/dev/null 2>&1'))

ifstatus != remainder:

remainder= status

finished= total - remainder

print'扫描进度:'+str(finished)+'/'+str(remainder)+' 请等待.n'

else:

print '#',

else:

print'扫描进度:'+str(total)+'/'+str(total)+' 扫描已完成.n'

break

results = {}

IPs = []

for i infilename:

ports ={}

dom =xml.dom.minidom.parse(i)

root =dom.documentElement

ifroot.getElementsByTagName('address'):

ip= str(root.getElementsByTagName('address')[0].getAttribute("addr"))

port= root.getElementsByTagName('port')

service= root.getElementsByTagName('service')

forj in range(len(port)):

ports[str(port[j].getAttribute("portid"))]=str(service[j].getAttribute("name"))

ifports!=None:

IPs.append(ip)

results[ip]=ports

mail('results','results:'+str(results))

except Exception,e:

print e

return results,IPs

def run():

port_xml_parser(multiprocess_scan(dnsbrute()))

if __name__ == '__main__':

run()