Set correct maximum on websocket policy

This commit is contained in:
UnfamiliarLegacy 2023-06-28 22:59:28 +02:00
parent feb1f2a527
commit 2c5c3f7db8
2 changed files with 4 additions and 4 deletions

View File

@ -72,8 +72,8 @@ public class NitroWebsocketClient implements NitroSession {
}); });
activeSession = (JsrSession) session; activeSession = (JsrSession) session;
activeSession.setMaxBinaryMessageBufferSize(NitroConstants.WEBSOCKET_BUFFER_SIZE); activeSession.getPolicy().setMaxBinaryMessageSize(NitroConstants.WEBSOCKET_BUFFER_SIZE);
activeSession.setMaxTextMessageBufferSize(NitroConstants.WEBSOCKET_BUFFER_SIZE); activeSession.getPolicy().setMaxTextMessageSize(NitroConstants.WEBSOCKET_BUFFER_SIZE);
// Set proper headers to spoof being a real client. // Set proper headers to spoof being a real client.
final Map<String, List<String>> headers = new HashMap<>(activeSession.getUpgradeRequest().getHeaders()); final Map<String, List<String>> headers = new HashMap<>(activeSession.getUpgradeRequest().getHeaders());

View File

@ -188,8 +188,8 @@ public class NitroWebsocketServer implements WebSocketListener, NitroSession {
private WebSocketClient createWebSocketClient() { private WebSocketClient createWebSocketClient() {
final WebSocketClient client = new WebSocketClient(createHttpClient()); final WebSocketClient client = new WebSocketClient(createHttpClient());
client.setMaxBinaryMessageBufferSize(NitroConstants.WEBSOCKET_BUFFER_SIZE); client.getPolicy().setMaxBinaryMessageSize(NitroConstants.WEBSOCKET_BUFFER_SIZE);
client.setMaxTextMessageBufferSize(NitroConstants.WEBSOCKET_BUFFER_SIZE); client.getPolicy().setMaxTextMessageSize(NitroConstants.WEBSOCKET_BUFFER_SIZE);
return client; return client;
} }