Kiritow/wg-ops
1
0
Fork 0
mirror of https://github.com/Kiritow/wg-ops.git synced 2024-03-22 13:11:37 +08:00
Code Issues Projects Releases Wiki Activity

Update config generator

Browse Source
This commit is contained in:
Kirigaya Kazuto 2021-08-22 17:15:40 +00:00
parent 75a6b0106b
commit 1575472880
2 changed files with 69 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tool_common.py
View File

@ -75,4 +75,4 @@ def base64_to_json(content):
def get_sha256(content): def get_sha256(content):
return hashlib.sha256(content).hexdigest() return hashlib.sha256(content.encode('utf-8')).hexdigest()

109
tool_generate.py
View File

@ -37,7 +37,6 @@ if "version" not in config or int(config["version"]) < 1:
op_mode = config["mode"] op_mode = config["mode"]
udp_clients = config["udp2raw"]["client"] udp_clients = config["udp2raw"]["client"]
udp_servers = config["udp2raw"]["server"] udp_servers = config["udp2raw"]["server"]
udp_demuxer = config["udp2raw"]["demuxer"]
logger.info("Generating WireGuard config...") logger.info("Generating WireGuard config...")
@ -59,65 +58,93 @@ PostUp=sysctl net.ipv4.tcp_congestion_control=bbr
f.write("PostUp=sysctl net.ipv4.ip_forward=1\n") f.write("PostUp=sysctl net.ipv4.ip_forward=1\n")
current_dir = os.getcwd() current_dir = os.getcwd()
path_tunnel = os.path.join(current_dir, "bin", "udp2raw_amd64") bin_tunnel = os.path.join(current_dir, "bin", "udp2raw_amd64")
path_speeder = os.path.join(current_dir, "bin", "speederv2_amd64") bin_speeder = os.path.join(current_dir, "bin", "speederv2_amd64")
path_demuxer = os.path.join(current_dir, "bin", "w2u") bin_demuxer = os.path.join(current_dir, "bin", "w2u")
cache_nb_config = []
cache_config = [] cache_config = []
for info in udp_clients: for client_info in udp_clients:
if info["speeder"]["enable"]: speeder_info = client_info["speeder"]
# WG --> Speeder --> RawTunnel balancer_info = client_info["demuxer"]
speeder = info["speeder"]
cache_config.append("PostUp={} new-window -t tunnel -d '{} -c -l127.0.0.1:{} -r 127.0.0.1:{} -f{} --mode 0'".format(tmux_path, path_speeder, speeder["port"], info["port"], speeder["ratio"]))
filename = write_tunnel_config("c", "127.0.0.1:{}".format(info["port"]), info["remote"], info["password"]) if balancer_info:
filepath = os.path.join(current_dir, "local", "tunnel", filename) # ... => Balancer => Tunnels
cache_config.append("PostUp={} new-window -t tunnel -d '{} --conf-file {}'".format(tmux_path, path_tunnel, filepath)) cache_nb_config.append("PostUp={} new-window -t tunnel -d '{} -f {} -l {} -t {} -s {}'".format(tmux_path, bin_demuxer, config["listen"], balancer_info["port"], client_info["port"], balancer_info["size"]))
for info in udp_demuxer: if speeder_info:
cache_config.append("PostUp={} new-window -t tunnel -d '{} -f {} -l {} -t {} -s {}'".format(tmux_path, path_demuxer, config["listen"], info["port"], info["forward"], info["size"])) if balancer_info:
# WG => Speeder => Balancer => Tunnels
cache_config.append("PostUp={} new-window -t tunnel -d '{} -c -l127.0.0.1:{} -r 127.0.0.1:{} -f{} --mode 0'".format(tmux_path, bin_speeder, speeder_info["port"], balancer_info["port"], speeder_info["ratio"]))
else:
# WG => Speeder => Tunnel
cache_config.append("PostUp={} new-window -t tunnel -d '{} -c -l127.0.0.1:{} -r 127.0.0.1:{} -f{} --mode 0'".format(tmux_path, bin_speeder, speeder_info["port"], client_info["port"], speeder_info["ratio"]))
for info in udp_servers: if balancer_info:
if info["speeder"]["enable"]: # Generate multiple tunnels
# RawTunnel --> Speeder --> WG for offset in range(balancer_info["size"]):
speeder = info["speeder"] config_filename = write_tunnel_config("c", "127.0.0.1:{}".format(client_info["port"] + offset), client_info["remote"], client_info["password"])
cache_config.append("PostUp={} new-window -t tunnel -d '{} -s -l127.0.0.1:{} -r 127.0.0.1:{} -f{} --mode 0'".format(tmux_path, path_speeder, speeder["port"], config["listen"], speeder["ratio"])) filepath = os.path.join(current_dir, "local", "tunnel", config_filename)
cache_config.append("PostUp={} new-window -t tunnel -d '{} --conf-file {}'".format(tmux_path, bin_tunnel, filepath))
filename = write_tunnel_config("s", "0.0.0.0:{}".format(info["port"]), "127.0.0.1:{}".format(speeder["port"]), info["password"])
filepath = os.path.join(current_dir, "local", "tunnel", filename)
cache_config.append("PostUp={} new-window -t tunnel -d '{} --conf-file {}'".format(tmux_path, path_tunnel, filepath))
else: else:
# RawTunnel --> WG config_filename = write_tunnel_config("c", "127.0.0.1:{}".format(client_info["port"]), client_info["remote"], client_info["password"])
filename = write_tunnel_config("s", "0.0.0.0:{}".format(info["port"]), "127.0.0.1:{}".format(config["listen"]), info["password"]) filepath = os.path.join(current_dir, "local", "tunnel", config_filename)
filepath = os.path.join(current_dir, "local", "tunnel", filename) cache_config.append("PostUp={} new-window -t tunnel -d '{} --conf-file {}'".format(tmux_path, bin_tunnel, filepath))
cache_config.append("PostUp={} new-window -t tunnel -d '{} --conf-file {}'".format(tmux_path, path_tunnel, filepath))
# Remove last sleep for server_info in udp_servers:
speeder_info = client_info["speeder"]
if speeder_info:
# RawTunnel => Speeder => WG
speeder = server_info["speeder"]
cache_config.append("PostUp={} new-window -t tunnel -d '{} -s -l127.0.0.1:{} -r 127.0.0.1:{} -f{} --mode 0'".format(tmux_path, bin_speeder, speeder["port"], config["listen"], speeder["ratio"]))
config_filename = write_tunnel_config("s", "0.0.0.0:{}".format(server_info["port"]), "127.0.0.1:{}".format(speeder["port"]), server_info["password"])
filepath = os.path.join(current_dir, "local", "tunnel", config_filename)
cache_config.append("PostUp={} new-window -t tunnel -d '{} --conf-file {}'".format(tmux_path, bin_tunnel, filepath))
else:
# RawTunnel => WG
config_filename = write_tunnel_config("s", "0.0.0.0:{}".format(server_info["port"]), "127.0.0.1:{}".format(config["listen"]), server_info["password"])
filepath = os.path.join(current_dir, "local", "tunnel", config_filename)
cache_config.append("PostUp={} new-window -t tunnel -d '{} --conf-file {}'".format(tmux_path, bin_tunnel, filepath))
# Add sleep interval
if cache_config: if cache_config:
for i in range(len(cache_config) - 1): for i in range(len(cache_config) - 1):
cache_config[i] = "{}; sleep 2".format(cache_config[i]) cache_config[i] = "{}; sleep 2".format(cache_config[i])
cache_config.append("") cache_config.append("")
f.write('\n'.join(cache_config)) f.write('\n'.join(cache_config))
if cache_nb_config:
cache_nb_config.append("")
f.write('\n'.join(cache_nb_config))
# Generate PostDown # Generate PostDown
f.write("PostDown={} kill-session -t tunnel\n".format(tmux_path)) f.write("PostDown={} kill-session -t tunnel\n".format(tmux_path))
for info in config["peers"]: for peer_info in config["peers"]:
f.write(''' f.write('''
[Peer] [Peer]
PublicKey = {} PublicKey = {}
AllowedIPs = {} AllowedIPs = {}
'''.format(info["pubkey"], info["allowed"])) '''.format(peer_info["pubkey"], peer_info["allowed"]))
if info["endpoint"]: if peer_info["endpoint"]:
udp_info = udp_clients[int(info["endpoint"]) - 1] client_info = udp_clients[int(peer_info["endpoint"]) - 1]
if udp_info["speeder"]["enable"]: speeder_info = client_info["speeder"]
# WG --> Speeder balancer_info = client_info["demuxer"]
f.write("Endpoint = 127.0.0.1:{}\n".format(udp_info["speeder"]["port"]))
if speeder_info:
# WG => Speeder => ...
f.write("Endpoint = 127.0.0.1:{}\n".format(speeder_info["port"]))
elif balancer_info:
# WG => Balancer => ...
f.write("Endpoint = 127.0.0.1:{}\n".format(balancer_info["port"]))
else: else:
# WG --> Tunnel # WG => ...
f.write("Endpoint = 127.0.0.1:{}\n".format(udp_info["port"])) f.write("Endpoint = 127.0.0.1:{}\n".format(client_info["port"]))
if info["keepalive"]:
f.write("PersistentKeepalive = {}\n".format(info["keepalive"])) if peer_info["keepalive"]:
f.write("PersistentKeepalive = {}\n".format(peer_info["keepalive"]))
os.system("chmod 600 local/{}.conf".format(config["interface"])) os.system("chmod 600 local/{}.conf".format(config["interface"]))
@ -156,8 +183,8 @@ sudo cp local/{}.conf /etc/wireguard/
sudo -- bash -c "wg syncconf {} <(wg-quick strip {})" sudo -- bash -c "wg syncconf {} <(wg-quick strip {})"
'''.format(config["interface"], config["interface"], config["interface"])) '''.format(config["interface"], config["interface"], config["interface"]))
for info in config["peers"]: for peer_info in config["peers"]:
f.write("sudo ip -4 route add {} dev {}\n".format(info["allowed"], config["interface"])) f.write("sudo ip -4 route add {} dev {}\n".format(peer_info["allowed"], config["interface"]))
logger.info('''[Done] Config generated. Before you run start.sh, besure to: logger.info('''[Done] Config generated. Before you run start.sh, besure to: