mirror of https://github.com/jumpserver/jumpserver
92 lines
2.2 KiB
Python
92 lines
2.2 KiB
Python
from ipaddress import ip_network, ip_address
|
|
from django.conf import settings
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from .ipip import get_ip_city_by_ipip
|
|
from .geoip import get_ip_city_by_geoip
|
|
|
|
|
|
def is_ip_address(address):
|
|
""" 192.168.10.1 """
|
|
try:
|
|
ip_address(address)
|
|
except ValueError:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
|
|
def is_ip_network(ip):
|
|
""" 192.168.1.0/24 """
|
|
try:
|
|
ip_network(ip)
|
|
except ValueError:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
|
|
def is_ip_segment(ip):
|
|
""" 10.1.1.1-10.1.1.20 """
|
|
if '-' not in ip:
|
|
return False
|
|
ip_address1, ip_address2 = ip.split('-')
|
|
return is_ip_address(ip_address1) and is_ip_address(ip_address2)
|
|
|
|
|
|
def in_ip_segment(ip, ip_segment):
|
|
ip1, ip2 = ip_segment.split('-')
|
|
ip1 = int(ip_address(ip1))
|
|
ip2 = int(ip_address(ip2))
|
|
ip = int(ip_address(ip))
|
|
return min(ip1, ip2) <= ip <= max(ip1, ip2)
|
|
|
|
|
|
def contains_ip(ip, ip_group):
|
|
"""
|
|
ip_group:
|
|
[192.168.10.1, 192.168.1.0/24, 10.1.1.1-10.1.1.20, 2001:db8:2de::e13, 2001:db8:1a:1110::/64.]
|
|
|
|
"""
|
|
|
|
if '*' in ip_group:
|
|
return True
|
|
|
|
for _ip in ip_group:
|
|
if is_ip_address(_ip):
|
|
# 192.168.10.1
|
|
if ip == _ip:
|
|
return True
|
|
elif is_ip_network(_ip) and is_ip_address(ip):
|
|
# 192.168.1.0/24
|
|
if ip_address(ip) in ip_network(_ip):
|
|
return True
|
|
elif is_ip_segment(_ip) and is_ip_address(ip):
|
|
# 10.1.1.1-10.1.1.20
|
|
if in_ip_segment(ip, _ip):
|
|
return True
|
|
else:
|
|
# is domain name
|
|
if ip == _ip:
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
def get_ip_city(ip):
|
|
if not ip or not isinstance(ip, str):
|
|
return _("Invalid ip")
|
|
if ':' in ip:
|
|
return 'IPv6'
|
|
|
|
info = get_ip_city_by_ipip(ip)
|
|
if info:
|
|
city = info.get('city', _("Unknown"))
|
|
country = info.get('country')
|
|
|
|
# 国内城市 并且 语言是中文就使用国内
|
|
is_zh = settings.LANGUAGE_CODE.startswith('zh')
|
|
if country == '中国' and is_zh:
|
|
return city
|
|
return get_ip_city_by_geoip(ip)
|