仅使用内置模块的Python 2.7计算子网中的所有IP

时间:2016-12-29 17:54:23

标签: python python-2.7 ip subnet cidr

我想知道如何创建一个带有这样的cidr表示法的脚本:

204.15.20.0/22

并计算该范围内的所有可能IP地址。

204.15.20.0
...
...
204.15.23.255

现在我终于知道了这背后的数学,我想知道如何在python中创建这样的模块。

我知道netaddr.IPNetwork()等现成的模块,但我想知道如何使用Python 2.7中的内置模块

谢谢:)

1 个答案:

答案 0 :(得分:3)

您可以使用socketstruct模块的组合在IPv4地址和整数之间进行转换。之后是一个屏蔽整数地址并从cidr值中计算出子网中值范围的问题。

import re
import socket
import struct

def inet_atoi(ipv4_str):
    """Convert dotted ipv4 string to int"""
    # note: use socket for packed binary then struct to unpack
    return struct.unpack("!I", socket.inet_aton(ipv4_str))[0]

def inet_itoa(ipv4_int):
    """Convert int to dotted ipv4 string"""
    # note: use struct to pack then socket to string
    return socket.inet_ntoa(struct.pack("!I", ipv4_int))

def ipv4_range(ipaddr):
    """Return a list of IPv4 address contianed in a cidr address range"""
    # split out for example 192.168.1.1:22/24
    ipv4_str, port_str, cidr_str = re.match(
        r'([\d\.]+)(:\d+)?(/\d+)?', ipaddr).groups()

    # convert as needed
    ipv4_int = inet_atoi(ipv4_str)
    port_str = port_str or ''
    cidr_str = cidr_str or ''
    cidr_int = int(cidr_str[1:]) if cidr_str else 0

    # mask ipv4
    ipv4_base = ipv4_int & (0xffffffff << (32 - cidr_int))

    # generate list
    addrs = [inet_itoa(ipv4_base + val)
        for val in range(1 << (32 - cidr_int) + 2)]
    return addrs

print(ipv4_range('204.15.20.0/22'))