forked from juerson/wireguard_converted_nekoray
-
Notifications
You must be signed in to change notification settings - Fork 0
/
1.生成单个nekoray节点.py
112 lines (102 loc) · 6.29 KB
/
1.生成单个nekoray节点.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
108
109
110
111
112
import pyperclip # 将指定的运行结果自动复制到剪切板
import base64
import os
import sys
import re
# 检查文件是否存在或大小为0，即文件无效
def check_file_exist_or_zero_size(file):
if not os.path.exists(file) or os.stat(file).st_size == 0:
sys.exit()
# 读取配置文件的信息
def read_wireguard_key_parameters(conf_file):
with open(file=conf_file, mode='r', encoding='utf-8') as f:
wireguard_param = dict()
for line in f:
if line:
if line.startswith("PrivateKey"):
wireguard_param["PrivateKey"] = line.strip().replace(' ', '').replace("PrivateKey=", '')
if line.startswith("PublicKey"):
wireguard_param["PublicKey"] = line.strip().replace(' ', '').replace("PublicKey=", '')
if line.startswith("Address"):
wireguard_param["Address"] = line.strip().replace(' ', '').replace("Address=", '').split(',')
if line.startswith("MTU"):
wireguard_param["MTU"] = line.strip().replace(' ', '').replace("MTU=", '')
return wireguard_param
# 将从配置文件中读取到信息，写入到指定的JSON字符串中
def update_base_info(conf_file, MTU=None):
param = read_wireguard_key_parameters(conf_file)
peer_public_key = param["PublicKey"].strip()
private_key = param["PrivateKey"].strip()
Address = param.get("Address")
# if条件针对wireguard配置中，Address行的值有多个值或一个值的情况
if len(Address) > 1: # 多个值（一般是IPv4 CIDR 和 IPv6 CIDR）
local_address = r"[\n" + ','.join([r' \"{ip}\"'.format(ip=item) for item in Address]) + r"\n ]"
else: # 只有一个值
local_address = r'\"{ip}\"'.format(ip=Address[0])
MTU = param.get('MTU', 1408) if MTU is None else MTU
nekoray_str_json = '{"_v":0,"addr":"127.0.0.1","cmd":[""],"core":"internal","cs":"{\\n \\"interface_name\\": ' \
'\\"WARP\\",\\n \\"local_address\\": #local_address,\\n' \
' \\"mtu\\": #MUT值,\\n \\"peer_public_key\\": \\"#peer_public_key\\",\\n \\"private_key\\":' \
' \\"#private_key\\",\\n \\"server\\": \\"IP地址\\",\\n \\"server_port\\": 端口,\\n ' \
'\\"system_interface\\": false,\\n \\"tag\\": \\"proxy\\",\\n \\"type\\": \\"wireguard' \
'\\"\\n}","mapping_port":0,"name":"别名","port":1080,"socks_port":0}'
update_key = nekoray_str_json.replace('#peer_public_key', peer_public_key).replace('#private_key', private_key)
update_address = update_key.replace('#local_address', local_address).replace('#MUT值', str(MTU))
return update_address
# 判断是否为IP地址（IPv4或IPv6）
def is_ip_address(ip_addr):
""" 匹配 IPv4 和 IPv6 地址 """
ipv4_pattern = r'^(?:(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.){3}(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$'
ipv6_pattern = r'^(?:(?:[0-9a-fA-F]{1,4}:){7}[0-9a-fA-F]{1,4}|(?:[0-9a-fA-F]{1,4}:){1,7}:|(?:[0-9a-fA-F]{1,4}:){1,6}:[0-9a-fA-F]{1,4}|(?:[0-9a-fA-F]{1,4}:){1,5}(?::[0-9a-fA-F]{1,4}){1,2}|(?:[0-9a-fA-F]{1,4}:){1,4}(?::[0-9a-fA-F]{1,4}){1,3}|(?:[0-9a-fA-F]{1,4}:){1,3}(?::[0-9a-fA-F]{1,4}){1,4}|(?:[0-9a-fA-F]{1,4}:){1,2}(?::[0-9a-fA-F]{1,4}){1,5}|[0-9a-fA-F]{1,4}:(?:(?::[0-9a-fA-F]{1,4}){1,6})|:(?:(?::[0-9a-fA-F]{1,4}){1,7}|:))$'
try:
addr = ip_addr.rsplit(":", 1)[0]
ip = addr[1:-1] if addr.startswith('[') and addr.endswith(']') else addr
port = ip_addr.rsplit(':', 1)[1] if ip_addr.count(":") == 1 or (
ip_addr.count(":") > 3 and "]:" in ip_addr) else None
if re.match(ipv4_pattern, ip) and (port.isdigit() and int(port) >= 80):
ipv4 = re.match(ipv4_pattern, ip).group(0)
return True
elif re.match(ipv6_pattern, ip) and (port.isdigit() and int(port) >= 80):
ipv6 = re.match(ipv6_pattern, ip).group(0)
return True
except Exception as e:
pass
if __name__ == '__main__':
"""判断文件是否存在或文件的大小为0"""
file = "配置文件/wg-config.conf"
check_file_exist_or_zero_size(file) # 检查文件是否存在
"""MTU值的修改"""
print("是否修改MTU值？输入内容为空时，就默认为配置文件的值，配置文件中没有MTU值，就使用1408；")
while True:
input_mtu = input("这里输入MTU值，取值范围为1280~1500：")
if (input_mtu.isdigit() and 1280 <= int(input_mtu) <= 1500) or input_mtu.strip() == '':
break
base_str = None
if input_mtu.isdigit():
base_str = update_base_info(file, MTU=input_mtu) # 调用函数
else:
base_str = update_base_info(file) # 调用函数
print()
while True:
while True:
input_endpoint = input("输入优选IP(格式：162.159.192.10:891)，输入q、quit、exit退出程序：").strip()
state = is_ip_address(input_endpoint)
if state:
break
if input_endpoint.lower() in ["q", "quit", "exit"]:
sys.exit()
addr = input_endpoint.rsplit(":", 1)[0]
ip = addr[1:-1] if addr.startswith('[') and addr.endswith(']') else addr # 去掉IPv6的中括号
port = input_endpoint.rsplit(':', 1)[1]
input_country = input('添加节点名称或别名的前缀吗？(比如，CN)：')
country = f'{input_country.strip()}_' if input_country != '' else '' # 是否添加前缀，通常使用国家或地区的的2个字母
remarks = f"{ip}:{port}" if ip.count(":") == 0 else f"[{ip}]:{port}" # 节点的别名、节点的名称（不重要）
node = base_str.replace('别名', f'{country}{remarks}').replace('IP地址', ip).replace('端口', port)
encoded = base64.b64encode(node.encode('utf-8'), altchars=b'-_')
encoded_str = str(encoded, encoding='utf-8')
transport_protocol = "nekoray://custom#" # 在base64编码好的字符串的前面添加这个前缀（NekoBox软件专用的前缀）
nekoray_node = transport_protocol + encoded_str
pyperclip.copy(nekoray_node)
print(f"{'-' * 52}NekoRAY节点如下:{'-' * 52}\n{nekoray_node}\n{'-' * 120}")
print("节点已经复制到剪切板，可以黏贴到其它地方！")
print(f"{'-' * 120}")