From 4fd2422e4dedf0fdcc363d75a206f7e3bd1c626c Mon Sep 17 00:00:00 2001
From: Chris Eager <chris@signal.org>
Date: Tue, 2 May 2023 13:42:43 -0500
Subject: [PATCH] Catch  and `close()` after `UninitializedMessageException` in
 websocket messages

---
 .../whispersystems/websocket/WebSocketResourceProvider.java    | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/websocket-resources/src/main/java/org/whispersystems/websocket/WebSocketResourceProvider.java b/websocket-resources/src/main/java/org/whispersystems/websocket/WebSocketResourceProvider.java
index 12e37fba0..021cf2a92 100644
--- a/websocket-resources/src/main/java/org/whispersystems/websocket/WebSocketResourceProvider.java
+++ b/websocket-resources/src/main/java/org/whispersystems/websocket/WebSocketResourceProvider.java
@@ -5,6 +5,7 @@
 package org.whispersystems.websocket;
 
 import com.google.common.annotations.VisibleForTesting;
+import com.google.protobuf.UninitializedMessageException;
 import org.eclipse.jetty.websocket.api.RemoteEndpoint;
 import org.eclipse.jetty.websocket.api.Session;
 import org.eclipse.jetty.websocket.api.WebSocketListener;
@@ -113,7 +114,7 @@ public class WebSocketResourceProvider<T extends Principal> implements WebSocket
           close(session, 1018, "Badly formatted");
           break;
       }
-    } catch (InvalidMessageException e) {
+    } catch (UninitializedMessageException | InvalidMessageException e) {
       logger.debug("Parsing", e);
       close(session, 1018, "Badly formatted");
     }