python nmap模塊使用進行主機探測（ICMP）

終於審核通過了......第一次用博客，想記錄自己的學習情況，分享知識。

廢話不多說，第一篇blog，大牛請輕噴。

資產清點首先需要進行主機探測，將存活主機統計下來再進行進一步的指紋識別及端口探測。若直接進行全網段IP掃描是很浪費時間的，不如先使用ICMP進行主機存活狀態的探測后，再針對的進行掃描。（基於ping的方式有可能有誤差，但是這里主要考慮的是效率）

python的第三方nmap模塊的安裝，網上有相關教程，不多贅述。

由於需要跟蹤累計PC存活總數，使用了類方法。但同時遇到的問題就是：使用多進程后會出現以下報錯信息：

cPickle.PicklingError: Can't pickle <type 'instancemethod'>: attribute lookup __builtin__.instancemethod failed

查了一下解決方法大概有三種：

1. 用線程替換進程
2. 可以使用copy_reg來規避上面的異常.
3. dill 或pathos.multiprocesssing ：use pathos.multiprocesssing, instead of multiprocessing. pathos.multiprocessing is a fork of multiprocessing that uses dill. dill can serialize almost anything in python, so you are able to send a lot more around in parallel.

這里使用第一種方法進行修改，直接上代碼：

import nmap
from multiprocessing.pool import ThreadPool as Pool
import time
import csv


class Scan(object):
    def __init__(self):
        self.count = 0

    def scan(self, ip):
        csv_write =[]
        nm = nmap.PortScanner()
        nm.scan(hosts=ip, arguments='-n -sn -PE')
        hosts_list = [(x, nm[x]['status']['state']) for x in nm.all_hosts()]
        for host, status in hosts_list:
            # print '{0}:{1}'.format(host, status)
            csv_write.append([host, status])
        name = ip.split('/')[0] + '.csv'
        # print name
        with open(name, 'wb') as csvfile:
            fieldnames = ['ip', 'status']
            w = csv.writer(csvfile, dialect='excel')
            w.writerow(fieldnames)
            w.writerows(csv_write)
        print '[+]\"%s\" have saved, %d PC is up' % (name, len(hosts_list))
        self.count += len(hosts_list)
            
    def get_ips(self, file):
        ipList = []
        with open(file) as f:
            for line in f.readlines():
                ipList.append(line.strip())
        return ipList

    def run(self):
        startTime = time.time()
        print '[+]Scanning...'
        ip_txt = 'C:\zichan.txt'
        ipList = self.get_ips(ip_txt)
        # print ipList
        pool = Pool(5)
        resultList = pool.map(self.scan, ipList)
        pool.close()
        pool.join()
        endTime = time.time()
        print '------------------------------------------------------'
        print '[+]Scanning cost %ss, %d PC is up.' % (endTime - startTime, self.count)
        print '[+]Done.'

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

比較簡單的方法實現，實際情況下掃描7個內網B段大概不到一個半小時，講道理比直接使用nmap快很多（對比的nmap現在還沒有跑完，嗯）

保存格式為CSV文件，控制台輸出如下：

后面會嘗試優化，並添加新的功能如端口掃描、系統指紋、漏洞探測等，這次就這樣！水平有限，嘻嘻。