Allow Connecting -> Connecting transition

oxidecomputer · Dec 12, 2024 · 6a1bde5 · 6a1bde5
1 parent a033afb
commit 6a1bde5
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/upstairs/src/client.rs b/upstairs/src/client.rs
@@ -972,6 +972,16 @@ impl DownstairsClient {
         use NegotiationState as N;
         use UpstairsState as U;
         match (prev_state, next_state) {
+            (
+                D::Connecting { .. },
+                D::Connecting {
+                    state: N::Start { .. },
+                    ..
+                },
+            ) => {
+                // restarting negotiation is allowed
+                true
+            }
             (
                 D::Connecting {
                     state: prev_state,
@@ -982,6 +992,7 @@ impl DownstairsClient {
                     mode: next_mode,
                 },
             ) => {
+                // Check normal negotiation path
                 if next_mode == C::New && matches!(up_state, U::Active) {
                     return false;
                 }