Keep sessions alive through UDP churn

src/bin/dosh-client.rs

@@ -36,7 +36,7 @@ use dosh::transport::{
     DoshTransport, SessionEvent, SessionRole, SessionTransportConfig, TransportConfig,
     TransportEvent,
 };
-use dosh::udp::is_transient_udp_send_error;
+use dosh::udp::{is_transient_udp_error, is_transient_udp_send_error};
 use serde::{Deserialize, Serialize};
 use sha2::{Digest, Sha256};
 use std::collections::BTreeMap;
@@ -4440,9 +4440,14 @@ where
             return Err(anyhow!("timed out waiting for server response"));
         }
         let remaining = wait - elapsed;
-        let (n, _) = tokio::time::timeout(remaining, socket.recv_from(&mut buf))
+        let (n, _) = match tokio::time::timeout(remaining, socket.recv_from(&mut buf))
             .await
-            .context("timed out waiting for server response")??;
+            .context("timed out waiting for server response")?
+        {
+            Ok(value) => value,
+            Err(err) if is_transient_udp_error(&err) => continue,
+            Err(err) => return Err(err.into()),
+        };
         let Ok(packet) = protocol::decode(&buf[..n]) else {
             continue;
         };
@@ -4832,7 +4837,11 @@ async fn run_terminal(
                 }
             }
             recv = socket.recv_from(&mut recv_buf) => {
-                let (n, _) = recv?;
+                let (n, _) = match recv {
+                    Ok(value) => value,
+                    Err(err) if is_transient_udp_error(&err) => continue,
+                    Err(err) => return Err(err.into()),
+                };
                 let Ok(packet) = protocol::decode(&recv_buf[..n]) else {
                     continue;
                 };
@@ -7770,8 +7779,8 @@ mod tests {
         retransmit_stream_opens, rewrite_forward_command, selected_predict_mode, selected_udp_host,
         server_version_mismatch, should_flush_terminal_input_after_contact,
         should_hold_during_startup_gate, should_hold_post_submit_input,
-        should_repaint_idle_alternate_screen, split_after_command_submit, ssh_destination_host,
-        ssh_username, ssh_with_user, ssh_command_target, startup_command, status_ssh_target,
+        should_repaint_idle_alternate_screen, split_after_command_submit, ssh_command_target,
+        ssh_destination_host, ssh_username, ssh_with_user, startup_command, status_ssh_target,
         strip_stale_mouse_reports, strip_terminal_focus_reports, terminal_private_mode_transition,
         toml_bare_key_or_quoted, unix_update_script, update_installer_url, update_version_status,
         upsert_managed_block, valid_forward_host, vscode_safe_alias, windows_update_script,