From f608849fcd8daec53ce20dc1fcbeead7e1cae7ac Mon Sep 17 00:00:00 2001 From: Lee Smet Date: Mon, 6 Nov 2023 11:46:01 +0100 Subject: [PATCH] Fix #41: Harden peer exchange handshake Signed-off-by: Lee Smet --- src/peer_manager.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/peer_manager.rs b/src/peer_manager.rs index 21fa990..f10f303 100644 --- a/src/peer_manager.rs +++ b/src/peer_manager.rs @@ -140,9 +140,9 @@ impl PeerManager { buf[1..].copy_from_slice(&node_tun_addr.octets()[..]); // Step 1 - stream.write_all(&buf).await.unwrap(); + stream.write_all(&buf).await?; // Step 2 - stream.read_exact(&mut buf).await.unwrap(); + stream.read_exact(&mut buf).await?; let received_overlay_ip = match buf[0] { 0 => IpAddr::from(<&[u8] as TryInto<[u8; 4]>>::try_into(&buf[1..5]).unwrap()), 1 => IpAddr::from(<&[u8] as TryInto<[u8; 16]>>::try_into(&buf[1..]).unwrap()),