การใช้โมดูล "struct" อาจทำให้เกิดปัญหากับ endian-ness และขนาดประเภทและไม่จำเป็น หรือคือ socket.inet_aton () Python ทำงานได้ดีกับที่อยู่ IP แบบจุดสี่เหลี่ยม:
def ip_to_u32(ip):
return int(''.join('%02x' % int(d) for d in ip.split('.')), 16)
ฉันจำเป็นต้องทำการจับคู่ IP ในแต่ละซ็อกเก็ตการโทรยอมรับ () กับเครือข่ายต้นทางที่อนุญาตทั้งหมดดังนั้นฉันจึงคำนวณมาสก์และเครือข่ายล่วงหน้าเป็นจำนวนเต็ม:
SNS_SOURCES = [
'207.171.167.101',
'207.171.167.25',
'207.171.167.26',
'207.171.172.6',
'54.239.98.0/24',
'54.240.217.16/29',
'54.240.217.8/29',
'54.240.217.64/28',
'54.240.217.80/29',
'72.21.196.64/29',
'72.21.198.64/29',
'72.21.198.72',
'72.21.217.0/24',
]
def build_masks():
masks = [ ]
for cidr in SNS_SOURCES:
if '/' in cidr:
netstr, bits = cidr.split('/')
mask = (0xffffffff << (32 - int(bits))) & 0xffffffff
net = ip_to_u32(netstr) & mask
else:
mask = 0xffffffff
net = ip_to_u32(cidr)
masks.append((mask, net))
return masks
จากนั้นฉันสามารถดูได้อย่างรวดเร็วว่า IP ที่ระบุอยู่ในเครือข่ายใดเครือข่ายหนึ่ง:
ip = ip_to_u32(ipstr)
for mask, net in cached_masks:
if ip & mask == net:
break
else:
raise BadClientIP(ipstr)
ไม่จำเป็นต้องมีการนำเข้าโมดูลและโค้ดนั้นเร็วมากในการจับคู่