python 多进程 | ip段扫描

作者: _简_述_ | 来源:发表于2017-04-20 16:41 被阅读391次

    在一个网段里,想知道哪个被占用,哪个没占用。最初笔者非常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')
    

    相关文章

      网友评论

      本文标题:python 多进程 | ip段扫描

      本文链接:https://www.haomeiwen.com/subject/qsdlzttx.html