在一個(gè)網(wǎng)段里,想知道哪個(gè)被占用,哪個(gè)沒(méi)占用。最初筆者非常low的一個(gè)一個(gè)ping去測(cè)試,隨后會(huì)寫(xiě)py腳本了,便寫(xiě)腳本去做??墒敲看螔呙瓒夹枰鹊胶荛L(zhǎng)時(shí)間。因?yàn)樽钌俚木W(wǎng)段是1-255,時(shí)間長(zhǎng)是肯定的。今天給大家分享個(gè)高大上的方法,多進(jìn)程網(wǎng)段掃描-速度搜搜的
這里用到兩個(gè)模塊,multiprocessing、subprocess,具體功能大家可以Google細(xì)細(xì)了解。
multiprocessing
python中的多線程其實(shí)并不是真正的多線程,如果想要充分地使用多核CPU的資源,在python中大部分情況需要使用多進(jìn)程。Python提供了非常好用的多進(jìn)程包multiprocessing,只需要定義一個(gè)函數(shù),Python會(huì)完成其他所有事情。借助這個(gè)包,可以輕松完成從單進(jìn)程到并發(fā)執(zhí)行的轉(zhuǎn)換。multiprocessing支持子進(jìn)程、通信和共享數(shù)據(jù)、執(zhí)行不同形式的同步,提供了Process、Queue、Pipe、Lock等組件。
subprocess
subprocess.call()父進(jìn)程等待子進(jìn)程完成返回退出信息returncode,相當(dāng)于exit code.
#!/usr/bin/python
# -*- coding: utf-8 -*-
#多進(jìn)程掃描網(wǎng)段
import multiprocessing
import subprocess
# ping 網(wǎng)段ip
def ping_host(activeq,notactiveq,ipaddr):
#ping -c1 -w1 中-c1是指ping的次數(shù),-w是指執(zhí)行的最后期限,也就是執(zhí)行的時(shí)間,單位為秒
if subprocess.call('ping -c1 -W 1 %s > /dev/null' % ipaddr, shell=True) != 0:
activeq.put(ipaddr)
else:
notactiveq.put(ipaddr)
#讀取隊(duì)列數(shù)據(jù)
def read(q):
while True:
if not q.empty():
value = q.get(True)
print value
else:
break
#掃描ip主函數(shù)
if __name__ == '__main__':
#創(chuàng)建進(jìn)程間通信隊(duì)列
manager = multiprocessing.Manager()
activeq = manager.Queue()
notactiveq = manager.Queue()
process_number = 4
host_list = []
for ipnum in range(1, 255):
host_list.append('192.168.1.' + str(ipnum))
#創(chuàng)建進(jìn)程池
pool = multiprocessing.Pool(processes=process_number)
for ipaddr in host_list:
pool.apply_async(ping_host,args=[activeq,notactiveq,ipaddr])
pool.close()
pool.join()
#輸出正在使用ip
read(activeq)
#輸出未被使用ip
read(notactiveq)
擴(kuò)展版本
初版在傳入一個(gè)C類(lèi)地址,使用了很low的for循環(huán)去創(chuàng)造一個(gè)地址池。后經(jīng)過(guò)高人點(diǎn)撥,有個(gè)現(xiàn)成的模塊可以直接生成,而且支持掩碼很是方便。
首先安裝pip
pip install netaddr
from netaddr import IPNetwork
host_list=IPNetwork('192.168.1.0/24')