Catch and close() after UninitializedMessageException in websocket messages

This commit is contained in:
Chris Eager
2023-05-02 13:42:43 -05:00
committed by Chris Eager
parent 6181d439f6
commit 4fd2422e4d

View File

@@ -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");
}