refactor(network): Modify max frame length for transmission. (nervosn…

…etwork#270)
homura · Jun 10, 2019 · de57a5e · de57a5e
1 parent a5cdf59
commit de57a5e
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/core/network/Cargo.toml b/core/network/Cargo.toml
@@ -13,6 +13,7 @@ core-serialization = { path = "../serialization" }
 core-runtime = { path = "../runtime" }
 core-types = { path = "../types" }
 
+tokio = "0.1"
 tentacle = "0.2"
 tentacle-discovery = "0.2"
 log = "0.4"

diff --git a/core/network/src/p2p/protocol/transmission.rs b/core/network/src/p2p/protocol/transmission.rs
@@ -3,6 +3,7 @@ use tentacle::context::{ProtocolContext, ProtocolContextMutRef};
 use tentacle::service::{ProtocolHandle, ProtocolMeta};
 use tentacle::{builder::MetaBuilder, bytes::Bytes, traits::ServiceProtocol};
 use tentacle::{multiaddr::Multiaddr, ProtocolId, SessionId};
+use tokio::codec::LengthDelimitedCodec;
 
 use common_channel::Sender;
 
@@ -24,6 +25,8 @@ pub const PROTOCOL_NAME: &str = "transmission";
 /// Protocol support versions
 pub const SUPPORT_VERSIONS: [&str; 1] = ["0.2"];
 
+const MAX_FRAME_LENGTH: usize = 1024 * 1024 * 100;
+
 pub struct SessionMessage {
     pub id:   SessionId,
     pub addr: Multiaddr,
@@ -49,6 +52,12 @@ impl TransmissionProtocol {
             .name(name!(PROTOCOL_NAME))
             .support_versions(support_versions!(SUPPORT_VERSIONS))
             .service_handle(service_handle)
+            .codec(|| {
+                let mut codec = LengthDelimitedCodec::new();
+                codec.set_max_frame_length(MAX_FRAME_LENGTH);
+
+                Box::new(codec)
+            })
             .build()
     }
 }