1 #!/usr/bin/env python
2 # -*- coding: utf_8 -*-
3 # Date: 2015年10月23日
4 # Author:蔚蓝行
5 # 博客 http://www.cnblogs.com/duanv/
6
7 from IPy import IP
8 import threading
9 import nmap
10 import time
11 import sys
12 import subprocess
13 from xml.dom import minidom
14
15 def usage():
16 print 'The script requires root privileges!'
17 print 'example:python scan.py 192.168.0.1/24'
18
19 #生成xml文件的模板函数
20 def addResult(newresult):
21 global doc
22 global scan_result
23
24 ip = doc.createElement("ip")
25 ip.setAttribute("address", newresult["address"])
26
27 osclass = doc.createElement("osclass")
28 osclass.appendChild(doc.createTextNode(newresult["osclass"]))
29 ip.appendChild(osclass)
30
31 port = doc.createElement("port")
32
33 tcp = doc.createElement("tcp")
34 tcp.appendChild(doc.createTextNode(newresult["tcp"]))
35 port.appendChild(tcp)
36
37 udp = doc.createElement("udp")
38 udp.appendChild(doc.createTextNode(newresult["udp"]))
39 port.appendChild(udp)
40
41 ip.appendChild(port)
42 scan_result.appendChild(ip)
43
44 #扫描函数，调用nmap库
45 def ip_scan(ip):
46 global nm
47 #这里调用系统ping命令来判断主机存活
48 p = subprocess.Popen("ping -c 1 -t 1 "+ip,stdin = subprocess.PIPE, stdout = subprocess.PIPE, stderr = subprocess.PIPE, shell = True)
49 out = p.stdout.read()
50 #如过没有100%丢包则主机存活，对是否丢包的判断是抓取系统回显内容，测试用的是‘MAC OS X’系统，抓取内容为‘100.0% packet loss’
51 if '100.0% packet loss' not in out:
52 try:
53 #调用nmap扫描主机操作系统，同时进行SYN扫描和UDP扫描探测开放的端口
54 nm.scan(ip,arguments='-O -sS -sU -F')
55 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]}
56 addResult(sr)
57 except:
58 pass
59
60 #循环，遍历未扫描的IP
61 def loop():
62 global mutex
63 global ipx
64
65 while 1:
66 #线程锁，扫描一个IP就将IPX列表中的该IP移除
67 mutex.acquire()
68 #如果列表中没有IP，则跳出循环结束该线程
69 if len(ipx)<=0:
70 mutex.release()
71 break
72 ip=ipx[0]
73 ipx.remove(ipx[0])
74 mutex.release()
75 #调用扫描函数
76 ip_scan(str(ip))
77
78 #创建线程的函数，默认创建40个
79 def creat_threads():
80 threads=[]
81 for i in range(40):
82 threads.append(threading.Thread(target=loop,))
83 for t in threads:
84 t.start()
85 for t in threads:
86 t.join()
87
88
89 def start():
90 #mutex:线程锁
91 global mutex
92 #ipx:存储要扫描的IP地址段列表
93 global ipx
94 #nm:nmap模块扫描对象
95 global nm
96 #doc:xml文档对象
97 global doc
98 #scan_result:xml文档的根元素
99 global scan_result
100
101 if '-h' == sys.argv[1]:
102 usage()
103 exit()
104 else:
105 #获取命令行输入的要扫描的IP段
106 ip=sys.argv[1]
107 #xml文档一些对象的初始化
108 doc = minidom.Document()
109 doc.appendChild(doc.createComment("scan_result xml."))
110 scan_result = doc.createElement("scan_result")
111 doc.appendChild(scan_result)
112
113 #初始化参数
114 ipx=[]
115 nm=nmap.PortScanner()
116 mutex=threading.Lock()
117
118 #调用IPy模块的IP函数，将IP地址段的每个IP存入列表
119 ipp=IP(ip, make_net=True)
120 for x in ipp:
121 ipx.append(x)
122 #去掉首尾代表子网和全部主机的IP
123 ipx=ipx[1:-1]
124
125 print("please wait...")
126 #计算时间
127 time_start=time.time()
128 #创建线程
129 creat_threads()
130
131 time_end=time.time()
132 t=time_end-time_start
133 print '*'*48
134 print '\nTime:'+str(t)+'s'
135 print 'Scan results have been saved to scan_result.xml.\n'
136 print '*'*48
137
138 #xml文件操作
139 f = file("scan_result.xml","w")
140 f.write(doc.toprettyxml(indent = "\t", newl = "\n", encoding = "utf-8"))
141 f.close()
142
143 if __name__=='__main__':
144 start()
