一个简化CIDR表的脚本

r2007

最近调整路由器的静态路由，想把收集到的cidr表归集缩短一下，google一番发现只有一个freebsd的ports程序可以实现。没有现成的freebsd，总不能现装吧。用python编了一个，发现挺好用。丢上来分享给CU的朋友。

欢迎debug.
usage:
[python] agg.py filename
or
cmd | [python] agg.py
文件格式
172.16.0.0/16
每行一个cidr地址

用到了defaultdict，所以仅支持2.5。

1. import sys
   
2. from collections import defaultdict
   
3. route_map=defaultdict(set)
   
4. try:
   
5.     input_file=open(sys.argv[1],'r')
   
6. except:
   
7.     input_file=sys.stdin
   
8. iplist=set()
   
9. ln=0
   
10. for line in input_file:
    
11.     ln+=1
    
12.     try:
    
13.         ip,mask=line.split('/')
    
14.         mask=32-int(mask)
    
15.         ip=reduce(lambda a,b:(a<<8)+b,[int(i) for i in ip.split('.')])>>mask
    
16.     except:
    
17.         print >>sys.stderr,'%d invaild data' % ln
    
18.         continue
    
19.     iplist.add((mask,ip))
    
20. for mask,ip in sorted(iplist,reverse=True):
    
21.     for i in sorted((k for k in route_map.iterkeys() if k>=mask)):
    
22.         ip_set=route_map[i]
    
23.         if i==mask:
    
24.             if ip in ip_set:
    
25.                 break
    
26.             if ip^1 in ip_set:
    
27.                 ip_set.remove(ip^1)
    
28.                 ip=ip/2
    
29.                 mask+=1
    
30.         else:
    
31.             if ip>>i-mask in ip_set:
    
32.                 break
    
33.     else:
    
34.         route_map[mask].add(ip)
    
35. for mask in route_map.iterkeys():
    
36.     for ip in route_map[mask]:
    
37.         ip=ip<<mask
    
38.         first = int((ip >> 24) & 255)
    
39.         second = int((ip >> 16) & 255)
    
40.         third = int((ip >> 8) & 255)
    
41.         fourth = int(ip & 255)
    
42.         print '%d.%d.%d.%d/%d' % (first, second, third, fourth,32-mask)
    

复制代码

可可熊

好东西，收藏了。

xiaoyu9805119

不错。顶一个。实用小脚本

wstar

没用过Python，不过还是顶啊。