79 lines
2.1 KiB
Python
79 lines
2.1 KiB
Python
import socket
|
|
|
|
# CAPS R OK BCUZ WE HAZ CIDict
|
|
HEADER_PRECEDENCE_ORDER = (
|
|
'X_FORWARDED_FOR',
|
|
'CLIENT_IP',
|
|
'X_REAL_IP',
|
|
'X_FORWARDED',
|
|
'X_CLUSTER_CLIENT_IP',
|
|
'FORWARDED_FOR',
|
|
'FORWARDED',
|
|
'VIA',
|
|
'REMOTE_ADDR',
|
|
)
|
|
|
|
# Private IP addresses
|
|
# http://en.wikipedia.org/wiki/List_of_assigned_/8_IPv4_address_blocks
|
|
# http://www.ietf.org/rfc/rfc3330.txt (IPv4)
|
|
# http://www.ietf.org/rfc/rfc5156.txt (IPv6)
|
|
# Regex would be ideal here, but this is keeping it simple
|
|
PRIVATE_IP_PREFIX = (
|
|
'0.', # externally non-routable
|
|
'10.', # class A private block
|
|
'169.254.', # link-local block
|
|
'172.16.', '172.17.', '172.18.', '172.19.',
|
|
'172.20.', '172.21.', '172.22.', '172.23.',
|
|
'172.24.', '172.25.', '172.26.', '172.27.',
|
|
'172.28.', '172.29.', '172.30.', '172.31.', # class B private blocks
|
|
'192.0.2.', # reserved for documentation and example code
|
|
'192.168.', # class C private block
|
|
'255.255.255.', # IPv4 broadcast address
|
|
'2001:db8:', # reserved for documentation and example code
|
|
'fc00:', # IPv6 private block
|
|
'fe80:', # link-local unicast
|
|
'ff00:', # IPv6 multicast
|
|
)
|
|
|
|
LOOPBACK_PREFIX = (
|
|
'127.', # IPv4 loopback device
|
|
'::1', # IPv6 loopback device
|
|
)
|
|
|
|
NON_PUBLIC_IP_PREFIX = PRIVATE_IP_PREFIX + LOOPBACK_PREFIX
|
|
|
|
|
|
def is_valid_ipv4(ip_str):
|
|
"""
|
|
Check the validity of an IPv4 address
|
|
"""
|
|
try:
|
|
socket.inet_pton(socket.AF_INET, ip_str)
|
|
except AttributeError: # pragma: no cover
|
|
try: # Fall-back on legacy API or False
|
|
socket.inet_aton(ip_str)
|
|
except (AttributeError, socket.error):
|
|
return False
|
|
return ip_str.count('.') == 3
|
|
except socket.error:
|
|
return False
|
|
return True
|
|
|
|
|
|
def is_valid_ipv6(ip_str):
|
|
"""
|
|
Check the validity of an IPv6 address
|
|
"""
|
|
try:
|
|
socket.inet_pton(socket.AF_INET6, ip_str)
|
|
except socket.error:
|
|
return False
|
|
return True
|
|
|
|
|
|
def is_valid_ip(ip_str):
|
|
"""
|
|
Check the validity of an IP address
|
|
"""
|
|
return is_valid_ipv4(ip_str) or is_valid_ipv6(ip_str)
|