Increase the max builder message size (#3599)

* increase the max builder message size

* do the same with the generator

* add the correct imports

Author: rob-maron

crates/hotshot-builder/shared/src/utils/event_service_wrapper.rs

@@ -6,7 +6,7 @@ use futures::{stream::unfold, Stream, StreamExt};
 use hotshot::types::Event;
 use hotshot_events_service::events::Error as EventStreamError;
 use hotshot_types::traits::node_implementation::NodeType;
-use surf_disco::{client::HealthStatus, Client};
+use surf_disco::{client::HealthStatus, reexports::WebSocketConfig, Client};
 use tokio::time::{sleep, timeout};
 use tracing::{error, warn};
 use url::Url;
@@ -64,8 +64,16 @@ impl<Types: NodeType, ApiVer: StaticVersionType + 'static> EventServiceStream<Ty
 
         tracing::info!("Builder client connected to the hotshot events API");
 
+        // Create a new [`WebSocketConfig`]. We trust the events service on our nodes to not
+        // send us malicious messages.
+        let websocket_config = WebSocketConfig {
+            max_message_size: None,
+            max_frame_size: None,
+            ..Default::default()
+        };
+
         Ok(client
-            .socket("hotshot-events/events")
+            .socket_with_config("hotshot-events/events", websocket_config)
             .subscribe::<Event<Types>>()
             .await?)
     }

hotshot-events-service/Cargo.toml

@@ -28,7 +28,7 @@ vbs = { workspace = true }
 [dev-dependencies]
 hotshot-example-types = { workspace = true }
 portpicker = "0.1.1"
-surf-disco = "0.9"
+surf-disco = { workspace = true }
 
 [lints]
 workspace = true

sequencer/src/bin/submit-transactions.rs

@@ -25,7 +25,7 @@ use rand_chacha::ChaChaRng;
 use rand_distr::Distribution;
 use sequencer::SequencerApiVersion;
 use sequencer_utils::logging;
-use surf_disco::{Client, Url};
+use surf_disco::{reexports::WebSocketConfig, Client, Url};
 use tide_disco::{error::ServerError, App};
 use tokio::{task::spawn, time::sleep};
 use vbs::version::StaticVersionType;
@@ -192,8 +192,20 @@ async fn main() {
     // Subscribe to block stream so we can check that our transactions are getting sequenced.
     let client = Client::<Error, SequencerApiVersion>::new(opt.urls[0].clone());
     let block_height: usize = client.get("status/block-height").send().await.unwrap();
+
+    // Create a new [`WebSocketConfig`]. We trust the events service on our nodes to not
+    // send us malicious messages.
+    let websocket_config = WebSocketConfig {
+        max_message_size: None,
+        max_frame_size: None,
+        ..Default::default()
+    };
+
     let mut blocks = client
-        .socket(&format!("availability/stream/blocks/{}", block_height - 1))
+        .socket_with_config(
+            &format!("availability/stream/blocks/{}", block_height - 1),
+            websocket_config,
+        )
         .subscribe()
         .await
         .unwrap();