首页 文章

Python 3.3中的ip地址网络

提问于
浏览
0

我试图转换一组可以在任何给定地址上开始和结束的IP范围,但不一定是在.0,.127或.255等上 . 我的代码大部分都在工作;但是,对于大范围来说,它可能会很慢 .

例如 find_range("1.40.0.0","1.44.255.255") 将花费一分钟来返回 1.40.0.0/141.44.0.0/16 的正确结果 .

另外,当起始范围没有以.0结尾时,我遇到了麻烦 . 如何解决这两个问题:大型IP范围的缓慢以及起始范围何时不以.0结束?

对于缓慢的问题,我尝试一次跳过多个地址,但这会错过较小的范围 .

import ipaddress, socket, struct

def ip2int(addr):                                                               
    return struct.unpack("!I", socket.inet_aton(addr))[0]                       

def int2ip(addr):                                                               
    return socket.inet_ntoa(struct.pack("!I", addr))        

def ipminus(ip, amount=1):
    tmp = ip2int(ip)
    return int2ip(tmp - amount)

def ipplus(ip):
    tmp = ip2int(ip)
    return int2ip(tmp + 1)

def cidr_notation(a,b):
    for mask in range(32, 6, -1):
        test = "%s/%s" % (a,mask)
        try:
            n = ipaddress.IPv4Network(test,False)
            if b == "%s" % (n.broadcast_address):
                return test
        except:
            pass
    return None

def split_range(a,b):
    a1 = ip2int(a)
    b1 = ip2int(b)

    needed = 1
    while needed:
        result = cidr_notation(a,b)
        if result:
            print( "* %16s\t%16s\t%16s" % (result, a, b))
            if ip2int(b) > b1:
                needed = 0
            else:
                a = ipplus(b)
                b = int2ip(b1)
        else:
            b = ipminus(b)

    return result

def find_range(x,y):
    result = cidr_notation(x,y)
    if result:
        print( "# %16s\t%16s\t%16s" % (result, x, y))
    else:
        split_range(x,y)

# main...
print("%16s\t%16s\t%16s" % ("mask","start","end"))
print("%16s\t%16s\t%16s" % ("----","-----","---"))

find_range("128.191.0.0","128.191.255.255") #fast
find_range("10.55.96.106","10.55.96.106") #fast
find_range("5.135.14.0","5.135.61.11") #slow
find_range("4.31.64.72","4.59.175.255") #does not work, how to fix?
find_range("1.40.0.0","1.44.255.255") #very slow
# 5000 more find_range() calls...

1 回答

  • 1

    基于bsdlp的评论,这段代码变得更加简单和快捷!

    def find_range(x,y,c_code="",c_name=""):
        print()
        print("%29s\t%23s" % (x,y))
        print("%16s\t%16s\t%16s\t%4s\t%s" % ("mask","start","end","code","name"))
        print("%16s\t%16s\t%16s\t%4s\t%s" % ("----","-----","---","----","----"))
    
        result = ipaddress.summarize_address_range( ipaddress.IPv4Address(x), ipaddress.IPv4Address(y) )
        for entry in result:
            net = str( entry.network_address )
            bcast = str( entry.broadcast_address )
            print( "%16s\t%16s\t%16s\t%4s\t%s" % (entry, net, bcast,c_code,c_name))
    

相关问题