python nmap模块 端口探测

今天添加端口探测功能，主要实现方式是通过nmap模块调用，扫描1-65535端口。上一篇中已经将UP的PC机全部获取到，这里直接从已知在线的PC中进行端口扫描就可以了，会节省很多时间。

代码如下，还是非常简单的：

 import nmap
 import time
 import csv
 import os

 class Port_Scan(object):
     def __init__(self):
         self.rootpath = 'C:/cunhuo/'
         self.dirpath = 'C:/portScan/'
         self.savepath = 'C:/portScan/result.csv'
         self.csv_write = []

     def get_ip(self, file):
         ipList = []
         with open(file, 'r') as f:
             for ip in f.readlines()[1:]:
                 ipList.append(ip.strip()[:-3])
         return ipList

     def scan(self, ip):
         if not os.path.exists(self.dirpath):
             os.mkdir(self.dirpath)
         nm = nmap.PortScanner()
         try:
             nm.scan(ip, arguments='1-65535')
             print '[*]Host:%s' % ip
             for proto in nm[ip].all_protocols():
                 lport = nm[ip][proto].keys()
                 lport.sort()
                 for port in lport:
                     self.csv_write.append([ip, port, nm[ip][proto][port]['state']])
                     print 'port : %s\tstate : %s' % (port, nm[ip][proto][port]['state'])
         except:
             self.csv_write.append([ip, 'unknown', 'unknown'])
             print 'port : unknown\tstate : unknown'

     def run(self):
         print '[+]Scanning...'
         startTime = time.time()
         files = os.listdir(self.rootpath)
         for file in files:
             ips = self.get_ip(self.rootpath + file)

             for ip in ips:
                 self.scan(ip)
         with open(self.savepath,'w') as f:
             fieldnames = ['ip', 'port', 'state']
             w = csv.writer(f, dialect='excel')
             w.writerow(fieldnames)
             w.writerows(self.csv_write)
         endTime = time.time()
         print '------------------------------------------------------'
         print '[+]Scanning cost %ss, result have saved %s' % (endTime - startTime, self.savepath)
         print '[+]Done.'

 if __name__ == '__main__':
     myScan = Port_Scan()
     myScan.run()

这里对于错误捕获的实现有点偷懒了，我们之前扫描出的存活主机可能在端口探测这个脚本执行时离线了，所以如果不加try...except判断的话会跑出keyerror的错误，说结果中的port、state不在字典中。但是这样加判断也有失偏颇...除了离线主机无法探测到，其他情况导致的也一并将port和state设置为unknown了，不知道以后的实验中会不会发生误报，到时候会进行修改调整。

扫描用时比较长，我就展示一下运行中的截图了：

result.csv，为什么中间会有空格呢，奇怪，慢慢改吧