Merge pull request #167 from UnfamiliarLegacy/fix/nitro-packet-size

Set correct maximum on websocket policy
sirjonasxx · Jul 11, 2023 · 8a717ad · 8a717ad
2 parents feb1f2a + 2c5c3f7
commit 8a717ad
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.../src/main/java/gearth/protocol/connection/proxy/nitro/websocket/NitroWebsocketClient.java b/.../src/main/java/gearth/protocol/connection/proxy/nitro/websocket/NitroWebsocketClient.java
@@ -72,8 +72,8 @@ public void stateChanged(HState oldState, HState newState) {
         });
 
         activeSession = (JsrSession) session;
-        activeSession.setMaxBinaryMessageBufferSize(NitroConstants.WEBSOCKET_BUFFER_SIZE);
-        activeSession.setMaxTextMessageBufferSize(NitroConstants.WEBSOCKET_BUFFER_SIZE);
+        activeSession.getPolicy().setMaxBinaryMessageSize(NitroConstants.WEBSOCKET_BUFFER_SIZE);
+        activeSession.getPolicy().setMaxTextMessageSize(NitroConstants.WEBSOCKET_BUFFER_SIZE);
 
         // Set proper headers to spoof being a real client.
         final Map<String, List<String>> headers = new HashMap<>(activeSession.getUpgradeRequest().getHeaders());

diff --git a/.../src/main/java/gearth/protocol/connection/proxy/nitro/websocket/NitroWebsocketServer.java b/.../src/main/java/gearth/protocol/connection/proxy/nitro/websocket/NitroWebsocketServer.java
@@ -188,8 +188,8 @@ private HttpClient createHttpClient() {
     private WebSocketClient createWebSocketClient() {
         final WebSocketClient client = new WebSocketClient(createHttpClient());
 
-        client.setMaxBinaryMessageBufferSize(NitroConstants.WEBSOCKET_BUFFER_SIZE);
-        client.setMaxTextMessageBufferSize(NitroConstants.WEBSOCKET_BUFFER_SIZE);
+        client.getPolicy().setMaxBinaryMessageSize(NitroConstants.WEBSOCKET_BUFFER_SIZE);
+        client.getPolicy().setMaxTextMessageSize(NitroConstants.WEBSOCKET_BUFFER_SIZE);
 
         return client;
     }