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，為什么中間會有空格呢，奇怪，慢慢改吧