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
1 changed files with 2 additions and 1 deletions

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