[avalanche-types] Set max message size to usize.MAX for gRPC client/s…

…evers (#77)

* [avalanche-types] Set max message size to usize.MAX for gRPC client/servers
* Add coverage for HTTP client

---------

Signed-off-by: Sam Batschelet <sam.batschelet@avalabs.org>

crates/avalanche-types/src/subnet/rpc/database/rpcdb/client/mod.rs

@@ -43,7 +43,9 @@ pub struct DatabaseClient {
 impl DatabaseClient {
     pub fn new(client_conn: Channel) -> BoxedDatabase {
         Box::new(Self {
-            inner: RpcDbDatabaseClient::new(client_conn),
+            inner: RpcDbDatabaseClient::new(client_conn)
+                .max_decoding_message_size(usize::MAX)
+                .max_encoding_message_size(usize::MAX),
             closed: Arc::new(AtomicBool::new(false)),
         })
     }

crates/avalanche-types/src/subnet/rpc/http/client.rs

@@ -12,7 +12,9 @@ pub struct Client {
 impl Client {
     pub fn new(client_conn: Channel) -> Box<dyn subnet::rpc::http::Handler + Send + Sync> {
         Box::new(Client {
-            inner: pb::http::http_client::HttpClient::new(client_conn),
+            inner: pb::http::http_client::HttpClient::new(client_conn)
+                .max_decoding_message_size(usize::MAX)
+                .max_encoding_message_size(usize::MAX),
         })
     }
 }

crates/avalanche-types/src/subnet/rpc/plugin.rs

@@ -92,6 +92,8 @@ where
         .add_service(reflection_service)
         .add_service(VmServer::new(vm))
         .serve_with_shutdown(addr, stop_ch.recv().map(|_| ()))
+        .max_decoding_message_size(usize::MAX)
+        .max_encoding_message_size(usize::MAX)
         .await
         .map_err(|e| Error::new(ErrorKind::Other, format!("grpc server failed: {:?}", e)))?;
     log::info!("grpc server shutdown complete: {}", addr);

crates/avalanche-types/src/subnet/rpc/runtime/client.rs

@@ -17,7 +17,9 @@ pub struct Client {
 impl Client {
     pub fn new(client_conn: Channel) -> Self {
         Self {
-            inner: RuntimeClient::new(client_conn),
+            inner: RuntimeClient::new(client_conn)
+                .max_decoding_message_size(usize::MAX)
+                .max_encoding_message_size(usize::MAX),
         }
     }
 }

crates/avalanche-types/src/subnet/rpc/snow/engine/common/appsender/client.rs

@@ -19,7 +19,9 @@ pub struct AppSenderClient {
 impl AppSenderClient {
     pub fn new(client_conn: Channel) -> Self {
         Self {
-            inner: app_sender_client::AppSenderClient::new(client_conn),
+            inner: app_sender_client::AppSenderClient::new(client_conn)
+                .max_decoding_message_size(usize::MAX)
+                .max_encoding_message_size(usize::MAX),
         }
     }
 }

crates/avalanche-types/src/subnet/rpc/snow/validators/client.rs

@@ -23,7 +23,9 @@ pub struct ValidatorStateClient {
 impl ValidatorStateClient {
     pub fn new(client_conn: Channel) -> Self {
         Self {
-            inner: validator_state_client::ValidatorStateClient::new(client_conn),
+            inner: validator_state_client::ValidatorStateClient::new(client_conn)
+                .max_decoding_message_size(usize::MAX)
+                .max_encoding_message_size(usize::MAX),
         }
     }
 }