MirrorMirror
/
Signal-Server
mirror of https://github.com/signalapp/Signal-Server.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
websocket-resources/src/main/java/org/whispersystems/websocket/WebSocketResourceProvider.java
View File

@ -5,6 +5,7 @@
package org.whispersystems.websocket; package org.whispersystems.websocket;
import com.google.common.annotations.VisibleForTesting; import com.google.common.annotations.VisibleForTesting;
import com.google.protobuf.UninitializedMessageException;
import org.eclipse.jetty.websocket.api.RemoteEndpoint; import org.eclipse.jetty.websocket.api.RemoteEndpoint;
import org.eclipse.jetty.websocket.api.Session; import org.eclipse.jetty.websocket.api.Session;
import org.eclipse.jetty.websocket.api.WebSocketListener; import org.eclipse.jetty.websocket.api.WebSocketListener;
@ -113,7 +114,7 @@ public class WebSocketResourceProvider<T extends Principal> implements WebSocket
close(session, 1018, "Badly formatted"); close(session, 1018, "Badly formatted");
break; break;
} }
} catch (InvalidMessageException e) { } catch (UninitializedMessageException | InvalidMessageException e) {
logger.debug("Parsing", e); logger.debug("Parsing", e);
close(session, 1018, "Badly formatted"); close(session, 1018, "Badly formatted");
} }