-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathget_ip_list.py
108 lines (92 loc) · 3.6 KB
/
get_ip_list.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
import socket
import struct
import ipaddress
# A类地址:10.0.0.0至10.255.255.255
# B类地址:172.16.0.0 至172.31.255.255
# C类地址:192.168.0.0至192.168.255.255
# 检查输入的ip地址的范围格式是否合法
def check_ip_range_format(ip_range_str: str):
if len(ip_range_str.split("-")) == 2:
start_ip = ip_range_str.split("-")[0]
last_ip = ip_range_str.split("-")[1]
if ipaddress.IPv4Network(start_ip) and ipaddress.IPv4Network(last_ip):
start_ip_a = int(start_ip.split(".")[0])
start_ip_b = int(start_ip.split(".")[1])
start_ip_c = int(start_ip.split(".")[2])
start_ip_d = int(start_ip.split(".")[3])
last_ip_a = int(last_ip.split(".")[0])
last_ip_b = int(last_ip.split(".")[1])
last_ip_c = int(last_ip.split(".")[2])
last_ip_d = int(last_ip.split(".")[3])
# 开始的ip要比结束的ip要小，
if start_ip_a < last_ip_a:
return True
elif start_ip_a == last_ip_a:
if start_ip_b < last_ip_b:
return True
elif start_ip_b == last_ip_b:
if start_ip_c < last_ip_c:
return True
elif start_ip_c == last_ip_c:
if start_ip_d < last_ip_d:
return True
elif start_ip_d == last_ip_d:
return True
else:
return False
else:
return False
else:
return False
else:
return False
else:
return False
else:
return False
# 检查输入的ip地址的范围格式是否合法
def check_ip_cidr_format(ip_cidr_str: str):
if len(ip_cidr_str.split("/")) == 2:
ip_address = ip_cidr_str.split("/")[0]
subnet_mask = int(ip_cidr_str.split("/")[1])
if 8 <= subnet_mask <= 30:
if ipaddress.IPv4Network(ip_address):
return True
else:
return False
else:
return False
# 输入ip地址的起始范围，列表形式输出这个范围的所有ip地址
def get_ip_range(start_ip: str, end_ip: str):
start = struct.unpack('!I', socket.inet_aton(start_ip))[0]
end = struct.unpack('!I', socket.inet_aton(end_ip))[0]
ip_list = []
for ip in range(start, end + 1):
ip_list.append(socket.inet_ntoa(struct.pack('!I', ip)))
return ip_list
# 给定一个范围的ip,输出这个范围里面所有的ip地址
def ip_range_list(ip_str: str):
first_ip = ip_str.split("-")[0]
last_ip = ip_str.split("-")[1]
ip_list = get_ip_range(first_ip, last_ip)
return ip_list
def ip_cidr_list(ip_address, subnet_mask):
try:
# 将IP地址和子网掩码转换为ipaddress对象
ip_network = ipaddress.IPv4Network(f"{ip_address}/{subnet_mask}", strict=False)
# 获取第一个主机位和最后一个主机位
first_ip = str(ip_network.network_address + 1)
last_ip = str(ip_network.broadcast_address - 1)
ip_list = get_ip_range(first_ip, last_ip)
return ip_list
except ipaddress.AddressValueError:
return None, None
def get_ip_list(ip_str: str):
if check_ip_range_format(ip_str):
return ip_range_list(ip_str)
elif check_ip_cidr_format(ip_str):
ip_address = ip_str.split("/")[0]
subnet_mask = ip_str.split("/")[1]
return ip_cidr_list(ip_address, subnet_mask)
else:
return False