fix(rpc): Increase max response size (#336)

eigerco · Jul 16, 2024 · faa2533 · faa2533
1 parent e76ef1c
commit faa2533
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/rpc/src/client.rs b/rpc/src/client.rs
@@ -16,6 +16,8 @@ mod native {
     use std::result::Result;
 
     use async_trait::async_trait;
+    use celestia_types::consts::appconsts::SHARE_SIZE;
+    use celestia_types::consts::data_availability_header::MAX_EXTENDED_SQUARE_WIDTH;
     use http::{header, HeaderValue};
     use jsonrpsee::core::client::{BatchResponse, ClientT, Subscription, SubscriptionClientT};
     use jsonrpsee::core::params::BatchRequestBuilder;
@@ -27,6 +29,13 @@ mod native {
 
     use crate::Error;
 
+    const MAX_EDS_SIZE_BYTES: usize =
+        MAX_EXTENDED_SQUARE_WIDTH * MAX_EXTENDED_SQUARE_WIDTH * SHARE_SIZE;
+
+    // The biggest response we might get is for requesting an EDS.
+    // Also, we allow 1 MB extra for any metadata they come with it.
+    const MAX_RESPONSE_SIZE: usize = MAX_EDS_SIZE_BYTES + 1024 * 1024;
+
     /// Json RPC client.
     pub enum Client {
         /// A client using 'http\[s\]' protocol.
@@ -56,11 +65,13 @@ mod native {
             let client = match protocol {
                 Some("http") | Some("https") => Client::Http(
                     HttpClientBuilder::default()
+                        .max_response_size(MAX_RESPONSE_SIZE as u32)
                         .set_headers(headers)
                         .build(conn_str)?,
                 ),
                 Some("ws") | Some("wss") => Client::Ws(
                     WsClientBuilder::default()
+                        .max_response_size(MAX_RESPONSE_SIZE as u32)
                         .set_headers(headers)
                         .build(conn_str)
                         .await?,