From b1322955cf986975ba2fafc5013c8f7bf9558801 Mon Sep 17 00:00:00 2001
From: Hollow Man <hollowman@opensuse.org>
Date: Sat, 25 Mar 2023 14:46:53 +0200
Subject: [PATCH] Use max frame size and max message size

Signed-off-by: Hollow Man <hollowman@opensuse.org>
---
 src/browser/transport/web_socket_connection.rs | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/browser/transport/web_socket_connection.rs b/src/browser/transport/web_socket_connection.rs
index 9c556487..157d59a8 100644
--- a/src/browser/transport/web_socket_connection.rs
+++ b/src/browser/transport/web_socket_connection.rs
@@ -6,6 +6,7 @@ use std::sync::Mutex;
 use anyhow::Result;
 use log::{debug, info, trace, warn};
 use tungstenite::http::Response;
+use tungstenite::protocol::WebSocketConfig;
 use tungstenite::stream::MaybeTlsStream;
 use url::Url;
 
@@ -131,7 +132,16 @@ impl WebSocketConnection {
         tungstenite::WebSocket<MaybeTlsStream<TcpStream>>,
         Response<Option<Vec<u8>>>,
     )> {
-        let mut client = tungstenite::connect(ws_url)?;
+        let mut client = tungstenite::client::connect_with_config(
+            ws_url,
+            Some(WebSocketConfig {
+                max_send_queue: None,
+                max_message_size: None,
+                max_frame_size: None,
+                accept_unmasked_frames: true,
+            }),
+            u8::MAX - 1,
+        )?;
 
         let stream = client.0.get_mut();