From 3c11838b26bd4e70abaa8b45f7352249e1f6556f Mon Sep 17 00:00:00 2001 From: Pierre Krieger Date: Wed, 6 Feb 2019 13:38:12 +0100 Subject: [PATCH] Limit Kademlia messages to 4kiB --- protocols/kad/Cargo.toml | 2 +- protocols/kad/src/protocol.rs | 10 ++++++++-- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/protocols/kad/Cargo.toml b/protocols/kad/Cargo.toml index a096890953a..3ff31f22d45 100644 --- a/protocols/kad/Cargo.toml +++ b/protocols/kad/Cargo.toml @@ -2,7 +2,7 @@ name = "libp2p-kad" edition = "2018" description = "Kademlia protocol for libp2p" -version = "0.3.2" +version = "0.3.3" authors = ["Parity Technologies "] license = "MIT" repository = "https://github.com/libp2p/rust-libp2p" diff --git a/protocols/kad/src/protocol.rs b/protocols/kad/src/protocol.rs index 8288c55cce8..75fc4482b11 100644 --- a/protocols/kad/src/protocol.rs +++ b/protocols/kad/src/protocol.rs @@ -159,8 +159,11 @@ where #[inline] fn upgrade_inbound(self, incoming: C, _: Self::Info) -> Self::Future { + let mut codec = codec::UviBytes::default(); + codec.set_max_len(4096); + future::ok( - Framed::new(incoming, codec::UviBytes::default()) + Framed::new(incoming, codec) .from_err::() .with::<_, fn(_) -> _, _>(|response| -> Result<_, IoError> { let proto_struct = resp_msg_to_proto(response); @@ -185,8 +188,11 @@ where #[inline] fn upgrade_outbound(self, incoming: C, _: Self::Info) -> Self::Future { + let mut codec = codec::UviBytes::default(); + codec.set_max_len(4096); + future::ok( - Framed::new(incoming, codec::UviBytes::default()) + Framed::new(incoming, codec) .from_err::() .with::<_, fn(_) -> _, _>(|request| -> Result<_, IoError> { let proto_struct = req_msg_to_proto(request);