From 0d2b59370bcb9887b947527a276b370528d12271 Mon Sep 17 00:00:00 2001
From: Markus Pettersson <markus.pettersson@mullvad.net>
Date: Tue, 17 Sep 2024 16:32:41 +0200
Subject: [PATCH] Fix bug where allowed ips would not be patched if DAITA was
 enabled

---
 talpid-wireguard/src/lib.rs | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/talpid-wireguard/src/lib.rs b/talpid-wireguard/src/lib.rs
index 634c265f9c08..0c918e1fc719 100644
--- a/talpid-wireguard/src/lib.rs
+++ b/talpid-wireguard/src/lib.rs
@@ -488,15 +488,17 @@ impl WireguardMonitor {
         log_path: Option<&Path>,
         args: TunnelArgs<'_, F>,
     ) -> Result<WireguardMonitor> {
+        let should_negotiate_ephemeral_peer = config.quantum_resistant || config.daita;
         let tunnel = Self::open_tunnel(
             args.runtime.clone(),
             &config,
             log_path,
             args.resource_dir,
             args.tun_provider.clone(),
-            // TODO: This seems like a bug! Should `config.quantum_resistant` really be the
-            // argument for `gateway_only` parameter?
-            config.quantum_resistant,
+            // In case we should negotiate an ephemeral peer, we should specify via AllowedIPs
+            // that we only allows traffic to/from the gateway. This is only needed on Android
+            // since we lack a firewall there.
+            should_negotiate_ephemeral_peer,
         )?;
 
         let (close_obfs_sender, close_obfs_listener) = sync_mpsc::channel();
@@ -578,7 +580,7 @@ impl WireguardMonitor {
                 }
             };
 
-            if config.quantum_resistant || config.daita {
+            if should_negotiate_ephemeral_peer {
                 // Ping before negotiating the ephemeral peer to make sure that the tunnel works.
                 tokio::task::spawn_blocking(ping()).await.unwrap()?;
                 let ephemeral_obfs_sender = close_obfs_sender.clone();