Ensure tunnel is running after updating wireguard config

1 files changed, 7 insertions, 2 deletions

diff --git a/talpid-wireguard/src/wireguard_go/mod.rs b/talpid-wireguard/src/wireguard_go/mod.rs
index 418462e7d1..2bfd8ef987 100644
--- a/talpid-wireguard/src/wireguard_go/mod.rs
+++ b/talpid-wireguard/src/wireguard_go/mod.rs
@@ -137,11 +137,16 @@ impl WgGoTunnel {
             }
             WgGoTunnel::Singlehop(mut state) => {
                 state.set_config(config.clone())?;
-                Ok(WgGoTunnel::Singlehop(state))
+                // HACK: Check if the tunnel is working by sending a ping in the tunnel.
+                let new_state = WgGoTunnel::Singlehop(state);
+                new_state.ensure_tunnel_is_running().await?;
+                Ok(new_state)
             }
             WgGoTunnel::Multihop(mut state) => {
                 state.set_config(config.clone())?;
-                Ok(WgGoTunnel::Multihop(state))
+                let new_state = WgGoTunnel::Multihop(state);
+                new_state.ensure_tunnel_is_running().await?;
+                Ok(new_state)
             }
         }
     }