diff --git a/src/main.rs b/src/main.rs index 13b19a9..7185043 100644 --- a/src/main.rs +++ b/src/main.rs @@ -193,7 +193,9 @@ fn write_config( writeln!(output, "[Peer]")?; writeln!(output, "PublicKey = {}", &config.data.public_key)?; writeln!(output, "Endpoint = {}", &endpoint_addr)?; - writeln!(output, "AllowedIPs = 0.0.0.0/0 #, ::/0")?; + + let allowed_ips: &[&str] = if config_opts.no_ipv6 { &["0.0.0.0", "::0"] } else { &["0.0.0.0"] }; + write_list(output, "AllowedIPs = ", allowed_ips)?; Ok(()) }