From 2b06fff6ea1e59ee5f66bb6c290bcaea11352f4b Mon Sep 17 00:00:00 2001
From: Andrey Golovizin <ag@sologoc.com>
Date: Wed, 14 Jul 2021 15:34:52 +0200
Subject: [PATCH] Fix AllowedIPs for IPv6

---
 src/main.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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(())
 }