kumi/wireguard-vpn-setup
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(config): handle missing wireguard keys gracefully

Browse source 
Switch to `config.get` for retrieving Wireguard keys to safely handle cases where keys might not be present in the configuration. This change prevents potential crashes due to missing keys and ensures that defaults are generated as needed.
This commit is contained in:
Kumi 2024-07-18 18:00:20 +02:00
parent eb575e8f49
commit b7656a46be
Signed by: kumi
GPG key ID: ECBCC9082395383F
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
worker.py
View file

@ -354,12 +354,12 @@ def run(config, provider, location, server_type):
ssh_execute_command(server_ip, command) ssh_execute_command(server_ip, command)
# Check if private key and preshared key are provided # Check if private key and preshared key are provided
if config["wireguard"]["private_key"]: if config.get("wireguard", "private_key"):
private_key = config["wireguard"]["private_key"] private_key = config["wireguard"]["private_key"]
else: else:
private_key = generate_private_key() private_key = generate_private_key()
if config["wireguard"]["preshared_key"]: if config.get("wireguard", "preshared_key"):
preshared_key = config["wireguard"]["preshared_key"] preshared_key = config["wireguard"]["preshared_key"]
else: else:
preshared_key = generate_preshared_key() preshared_key = generate_preshared_key()