leyoda/rqbit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update handshake deserialize to error on wrong messages

Browse source
This commit is contained in:
Igor Katson 2024-03-09 19:50:28 +00:00
parent 56527ae8d7
commit 5a12095959
No known key found for this signature in database
GPG key ID: B4EC22B66D61A3F5
1 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

15
crates/peer_binary_protocol/src/lib.rs
View file

@ -531,12 +531,15 @@ impl Handshake<ByteBuf<'static>> {
expected_len, expected_len,
"handshake", "handshake",
))?; ))?;
Ok(( let h = Self::bopts()
Self::bopts() .deserialize::<Handshake<ByteBuf<'_>>>(hbuf)
.deserialize(hbuf) .map_err(|e| MessageDeserializeError::Other(e.into()))?;
.map_err(|e| MessageDeserializeError::Other(e.into()))?, if h.pstr.0 != PSTR_BT1.as_bytes() {
expected_len, return Err(MessageDeserializeError::Other(anyhow::anyhow!(
)) "pstr doesn't match bittorrent V1"
)));
}
Ok((h, expected_len))
} }
} }