Make tunnel connection errors recoverable

mullvad · Nov 28, 2024 · b3c31e4 · b3c31e4
1 parent 64a6fcb
commit b3c31e4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 12 deletions.
diff --git a/talpid-wireguard/src/lib.rs b/talpid-wireguard/src/lib.rs
@@ -1087,15 +1087,6 @@ pub enum TunnelError {
     #[cfg(daita)]
     #[error("Failed to start DAITA - tunnel implemenation does not support DAITA")]
     DaitaNotSupported,
-
-    /// [connectivity] error.
-    #[error(transparent)]
-    Connectivity(#[from] Box<connectivity::Error>),
-
-    /// Tunnel seemingly does not serve any traffic
-    #[cfg(target_os = "android")]
-    #[error("Tunnel seemingly does not serve any traffic")]
-    TunnelUp,
 }
 
 #[cfg(target_os = "linux")]

diff --git a/talpid-wireguard/src/wireguard_go/mod.rs b/talpid-wireguard/src/wireguard_go/mod.rs
@@ -1,5 +1,7 @@
 #[cfg(target_os = "android")]
 use super::config;
+#[cfg(target_os = "android")]
+use super::Error;
 use super::{
     stats::{Stats, StatsMap},
     Config, Tunnel, TunnelError,
@@ -434,14 +436,15 @@ impl WgGoTunnel {
         &self,
         checker: &mut connectivity::Check<connectivity::Cancellable>,
     ) -> Result<()> {
-        let connectivity_err = |e| TunnelError::Connectivity(Box::new(e));
         let connection_established = checker
             .establish_connectivity(self)
-            .map_err(connectivity_err)?;
+            .map_err(|e| TunnelError::RecoverableStartWireguardError(Box::new(e)))?;
 
         // Timed out
         if !connection_established {
-            return Err(TunnelError::TunnelUp);
+            return Err(TunnelError::RecoverableStartWireguardError(Box::new(
+                Error::TimeoutError,
+            )));
         }
         Ok(())
     }