python 多进程 | ip段扫描

在一个网段里,想知道哪个被占用,哪个没占用。最初笔者非常low的一个一个ping去测试,随后会写py脚本了,便写脚本去做。可是每次扫描都需要等到很长时间。因为最少的网段是1-255,时间长是肯定的。今天给大家分享个高大上的方法,多进程网段扫描-速度搜搜的

这里用到两个模块,multiprocessing、subprocess,具体功能大家可以Google细细了解。

multiprocessing
python中的多线程其实并不是真正的多线程,如果想要充分地使用多核CPU的资源,在python中大部分情况需要使用多进程。Python提供了非常好用的多进程包multiprocessing,只需要定义一个函数,Python会完成其他所有事情。借助这个包,可以轻松完成从单进程到并发执行的转换。multiprocessing支持子进程、通信和共享数据、执行不同形式的同步,提供了Process、Queue、Pipe、Lock等组件。

subprocess
subprocess.call()父进程等待子进程完成返回退出信息returncode,相当于exit code.

#!/usr/bin/python 
# -*- coding: utf-8 -*-
#多进程扫描网段
import multiprocessing
import subprocess

# ping 网段ip
def ping_host(activeq,notactiveq,ipaddr):
    #ping -c1 -w1 中-c1是指ping的次数,-w是指执行的最后期限,也就是执行的时间,单位为秒
    if subprocess.call('ping -c1 -W 1 %s > /dev/null' % ipaddr, shell=True) != 0:
        activeq.put(ipaddr)
    else:
        notactiveq.put(ipaddr)

#读取队列数据
def read(q):
    while True:
        if not q.empty():
            value = q.get(True)
            print value
        else:
            break
#扫描ip主函数
if __name__ == '__main__':
    #创建进程间通信队列
    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))
     #创建进程池
    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)

扩展版本

初版在传入一个C类地址,使用了很low的for循环去创造一个地址池。后经过高人点拨,有个现成的模块可以直接生成,而且支持掩码很是方便。
首先安装pip

pip install netaddr
from netaddr import IPNetwork
host_list=IPNetwork('192.168.1.0/24')
    原文作者:_简_述_
    原文地址: https://www.jianshu.com/p/66b8d1efdeb3
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞