From 37510526979c64cb08c315582ec5b7ba224c243e Mon Sep 17 00:00:00 2001
From: Cody Henthorne <cody@signal.org>
Date: Tue, 20 May 2025 14:42:34 -0400
Subject: [PATCH] Force new auth WebSocket during various IMO recovery flows.

---
 .../securesms/messages/IncomingMessageObserver.kt             | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/app/src/main/java/org/thoughtcrime/securesms/messages/IncomingMessageObserver.kt b/app/src/main/java/org/thoughtcrime/securesms/messages/IncomingMessageObserver.kt
index 10963e1660..ea5f3bedfa 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/messages/IncomingMessageObserver.kt
+++ b/app/src/main/java/org/thoughtcrime/securesms/messages/IncomingMessageObserver.kt
@@ -266,6 +266,8 @@ class IncomingMessageObserver(
     SignalExecutors.BOUNDED.execute {
       Log.w(TAG, "Beginning termination. ${this.hashCode()}")
       terminated = true
+      Log.w(TAG, "Disconnecting auth socket as part of termination")
+      authWebSocket.disconnect()
     }
   }
 
@@ -478,6 +480,8 @@ class IncomingMessageObserver(
           attempts++
           Log.w(TAG, e)
         } finally {
+          Log.w(TAG, "Disconnecting auth websocket")
+          authWebSocket.disconnect()
           webSocketDisposable.dispose()
           decryptionDrained = false
         }