70 lines
1.6 KiB
Python
70 lines
1.6 KiB
Python
|
# analyzeutl 是分析器工具集合
|
|||
|
|
import awdb
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 过滤所有空值
|
|||
|
|
def filterNull(s):
|
|||
|
|
if s == "":
|
|||
|
|
return "UnKnown"
|
|||
|
|
return s
|
|||
|
|
|
|||
|
|
|
|||
|
|
def IP46(IP: str):
|
|||
|
|
if ':' in IP:
|
|||
|
|
return "v6"
|
|||
|
|
if '.' in IP:
|
|||
|
|
return "v4"
|
|||
|
|
return "Unknown"
|
|||
|
|
|
|||
|
|
|
|||
|
|
path_ip4app = "./data/IP_scene_all_cn.awdb"
|
|||
|
|
path_ip6 = "./data/IP_city_single_BD09_WGS84_ipv6.awdb"
|
|||
|
|
path_ip4qvxian = "./data/IP_basic_single_WGS84.awdb"
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 实例化数据读取器
|
|||
|
|
def makereader(arg=0):
|
|||
|
|
# 默认加载所有离线数据
|
|||
|
|
dloader_ip4app = awdb.open_database(path_ip4app)
|
|||
|
|
dloader_ip6 = awdb.open_database(path_ip6)
|
|||
|
|
dloader_ip4qx = awdb.open_database(path_ip4qvxian)
|
|||
|
|
return dloader_ip4app, dloader_ip4qx, dloader_ip6
|
|||
|
|
|
|||
|
|
|
|||
|
|
reader_ip4app, reader_ip4qx, reader_ip6 = makereader()
|
|||
|
|
|
|||
|
|
|
|||
|
|
# # 传入一个ip地址，根据对象中的数据进行IP关联分析
|
|||
|
|
# def make_IPinfo(ip):
|
|||
|
|
# record=getrecord(ip)
|
|||
|
|
# cou = record.get('areacode', b'').decode("utf-8")
|
|||
|
|
# multiarea = record.get('multiAreas', {})
|
|||
|
|
# if multiarea:
|
|||
|
|
# print("有多组记录")
|
|||
|
|
# else:
|
|||
|
|
# pass
|
|||
|
|
# return record
|
|||
|
|
|
|||
|
|
# 返回IP离线库中与ip相关的记录，记录中不包含应用场景
|
|||
|
|
def getrecord(ip):
|
|||
|
|
if (IP46(ip) == "v4"):
|
|||
|
|
return IP4_info(ip)
|
|||
|
|
elif (IP46(ip) == "v6"):
|
|||
|
|
return IP6_info(ip)
|
|||
|
|
else:
|
|||
|
|
print("地址存在问题")
|
|||
|
|
print(ip)
|
|||
|
|
return 1
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 返回IPv4记录
|
|||
|
|
def IP4_info(ip):
|
|||
|
|
(record, prefix_len) = reader_ip4qx.get_with_prefix_len(ip)
|
|||
|
|
return record
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 返回IPv6记录
|
|||
|
|
def IP6_info(ip):
|
|||
|
|
(record, prefix_len) = reader_ip6.get_with_prefix_len(ip)
|
|||
|
|
return record
|