disconnect session on bad ping

tlsnotary · Nov 6, 2024 · 190777f · 190777f
1 parent 10ebede
commit 190777f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/entries/Background/ws.ts b/src/entries/Background/ws.ts
@@ -82,6 +82,7 @@ export const connectSession = async () => {
         // Check if connection is open
         socket.send(bufferify({ method: 'ping' }));
       } else {
+        disconnectSession();
         clearInterval(heartbeatInterval); // Stop heartbeat if connection is closed
       }
     }, 55000);
@@ -352,7 +353,7 @@ async function handleRemoveIncomingProofRequest(message: {
 
 export const disconnectSession = async () => {
   if (!state.socket) return;
-  await state.socket.close();
+  const socket = state.socket;
   state.socket = null;
   state.clientId = '';
   state.pairing = '';
@@ -368,6 +369,7 @@ export const disconnectSession = async () => {
   pushToRedux(setOutgoingPairingRequest([]));
   pushToRedux(setIncomingProofRequest([]));
   pushToRedux(setOutgoingProofRequest([]));
+  await socket.close();
 };
 
 function sendMessage(target: string, method: string, params?: any) {