diff --git a/src/transport/webrtc/util.rs b/src/transport/webrtc/util.rs index 55917afc6..cbc5fcb36 100644 --- a/src/transport/webrtc/util.rs +++ b/src/transport/webrtc/util.rs @@ -23,6 +23,8 @@ use crate::{codec::unsigned_varint::UnsignedVarint, error::ParseError, transport use prost::Message; use tokio_util::codec::{Decoder, Encoder}; +const MAX_MESSAGE_SIZE: usize = 16 * 1024; + /// WebRTC mesage. #[derive(Debug)] pub struct WebRtcMessage { @@ -74,7 +76,7 @@ impl WebRtcMessage { /// Decode payload into [`WebRtcMessage`]. pub fn decode(payload: &[u8]) -> Result { // TODO: https://github.com/paritytech/litep2p/issues/352 set correct size - let mut codec = UnsignedVarint::new(None); + let mut codec = UnsignedVarint::new(Some(MAX_MESSAGE_SIZE)); let mut data = bytes::BytesMut::from(payload); let result = codec .decode(&mut data)