#!/usr/bin/env python

 # -*- coding: utf_8 -*-

 # Date: 2015年10月23日

 # Author:蔚蓝行

 # 博客 http://www.cnblogs.com/duanv/

 from IPy import IP

 import threading

 import nmap

 import time

 import sys

 import subprocess

 from xml.dom import minidom

 def usage():

     print 'The script requires root privileges!'

     print 'example:python scan.py 192.168.0.1/24'

 #生成xml文件的模板函数

 def addResult(newresult):

     global doc

     global scan_result

     ip = doc.createElement("ip")

     ip.setAttribute("address", newresult["address"])

     osclass = doc.createElement("osclass")

     osclass.appendChild(doc.createTextNode(newresult["osclass"]))

     ip.appendChild(osclass)

     port = doc.createElement("port")

     tcp = doc.createElement("tcp")

     tcp.appendChild(doc.createTextNode(newresult["tcp"]))

     port.appendChild(tcp)

     udp = doc.createElement("udp")

     udp.appendChild(doc.createTextNode(newresult["udp"]))

     port.appendChild(udp)

     ip.appendChild(port)

     scan_result.appendChild(ip)

 #扫描函数，调用nmap库

 def ip_scan(ip):

     global nm

     #这里调用系统ping命令来判断主机存活

     p = subprocess.Popen("ping -c 1 -t 1 "+ip,stdin = subprocess.PIPE, stdout = subprocess.PIPE, stderr = subprocess.PIPE, shell = True)

     out = p.stdout.read()

     #如过没有100%丢包则主机存活，对是否丢包的判断是抓取系统回显内容，测试用的是‘MAC OS X’系统，抓取内容为‘100.0% packet loss’

     if '100.0% packet loss' not in out:

         try:

             #调用nmap扫描主机操作系统，同时进行SYN扫描和UDP扫描探测开放的端口

             nm.scan(ip,arguments='-O -sS -sU -F')

             sr={'address':ip,'osclass':str(nm[ip]['osclass'])[1:-1],'tcp':str(nm[ip].all_tcp())[1:-1],'udp':str(nm[ip].all_udp())[1:-1]}

             addResult(sr)

         except:

             pass

 #循环，遍历未扫描的IP

 def loop():

     global mutex

     global ipx

     while 1:

         #线程锁，扫描一个IP就将IPX列表中的该IP移除

         mutex.acquire()

         #如果列表中没有IP，则跳出循环结束该线程

         if len(ipx)<=0:

             mutex.release()

             break

         ip=ipx[0]

         ipx.remove(ipx[0])

         mutex.release()

         #调用扫描函数

         ip_scan(str(ip))

 #创建线程的函数，默认创建40个

 def creat_threads():

     threads=[]

     for i in range(40):

         threads.append(threading.Thread(target=loop,))

     for t in threads:

         t.start()

     for t in threads:

         t.join()

 def start():

     #mutex:线程锁

     global mutex

     #ipx:存储要扫描的IP地址段列表

     global ipx

     #nm:nmap模块扫描对象

     global nm

     #doc:xml文档对象

     global doc

     #scan_result:xml文档的根元素

     global scan_result

     if '-h' == sys.argv[1]:

         usage()

         exit()

     else:

         #获取命令行输入的要扫描的IP段

         ip=sys.argv[1]

         #xml文档一些对象的初始化

         doc = minidom.Document()

         doc.appendChild(doc.createComment("scan_result xml."))

         scan_result = doc.createElement("scan_result")

         doc.appendChild(scan_result)

         #初始化参数

         ipx=[]

         nm=nmap.PortScanner()

         mutex=threading.Lock()

         #调用IPy模块的IP函数，将IP地址段的每个IP存入列表

         ipp=IP(ip, make_net=True)

         for x in ipp:

             ipx.append(x)

         #去掉首尾代表子网和全部主机的IP

         ipx=ipx[1:-1]

         print("please wait...")

         #计算时间

         time_start=time.time()

         #创建线程

         creat_threads()

         time_end=time.time()

         t=time_end-time_start

         print '*'*48

         print '\nTime:'+str(t)+'s'

         print 'Scan results have been saved to scan_result.xml.\n'

         print '*'*48

         #xml文件操作

         f = file("scan_result.xml","w")

         f.write(doc.toprettyxml(indent = "\t", newl = "\n", encoding = "utf-8"))

         f.close()

 if __name__=='__main__':

     start()