Skip to content

Instantly share code, notes, and snippets.

@2xyo
Forked from waffle2k/cidr2regex.py
Last active August 29, 2015 14:09
Show Gist options
  • Save 2xyo/f4e45b26bd29656e04f1 to your computer and use it in GitHub Desktop.
Save 2xyo/f4e45b26bd29656e04f1 to your computer and use it in GitHub Desktop.
#!/usr/bin/python
''' Not my script, found on the Internet, and rediscovered on my hard drive
'''
import sys
def cidr_to_regex(cidr):
ip, prefix = cidr.split('/')
base = 0
for val in map(int, ip.split('.')):
base = (base << 8) | val
shift = 32 - int(prefix)
start = base >> shift << shift
end = start | (1 << shift) - 1
def regex(lower, upper):
if lower == upper:
return str(lower)
from math import log10
exp = int(log10(upper - lower))
delta = 10 ** exp
if lower == 0 and upper == 255:
return "\d+"
if delta == 1:
val = ""
for a, b in zip(str(lower), str(upper)):
if a == b:
val += str(a)
elif (a, b) == ("0", "9"):
val += '\d'
elif int(b) - int(a) == 1:
val += '[%s%s]' % (a, b)
elif int(a) < int(b):
val += '[%s-%s]' % (a, b)
else:
val += '[%s-%s]' % (b, a)
return val
def gen_classes():
floor_ = lambda x: int(round(x / delta, 0) * delta)
xs = range(floor_(upper) - delta, floor_(lower), -delta)
for x in map(str, xs):
yield '%s%s' % (x[:-exp], r'\d' * exp)
yield regex(lower, floor_(lower) + (delta - 1))
yield regex(floor_(upper), upper)
return "(?:{})".format('|'.join(gen_classes()))
def get_parts():
for x in range(24, -1, -8):
yield regex(start >> x & 255, end >> x & 255)
return '^%s$' % r'\.'.join(get_parts())
for line in sys.stdin.readlines():
print cidr_to_regex( line )
print
@mordyovits
Copy link

This is still wrong. You can't simply reverse the outputed order when b > a. From 8 to 1 is not the same as from 1 to 8. The correct output is not [1-8], the correct output is [8901].

@mordyovits
Copy link

You can see a corrected version at my fork: https://gist.github.com/mordyovits/ab2ba4635a1ed9d2065b

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment