dandri/mycelium
1
0
Fork 0
mirror of https://github.com/threefoldtech/mycelium.git synced 2026-05-25 09:54:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix #41: Harden peer exchange handshake

Browse Source 
Signed-off-by: Lee Smet <lee.smet@hotmail.com>
This commit is contained in:
Lee Smet
2023-11-06 11:46:01 +01:00
parent 880160673b
commit f608849fcd
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/peer_manager.rs
+2 -2
View File
@@ -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()),