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

Fill yourip in extended handshake

Browse source
This commit is contained in:
Ivan 2024-08-18 22:55:32 +02:00
parent c697809e50
commit 3c0baacaa7
2 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
crates/librqbit/src/peer_connection.rs
View file

@ -14,7 +14,7 @@ use librqbit_core::{
};
use parking_lot::RwLock;
use peer_binary_protocol::{
extended::{handshake::ExtendedHandshake, ExtendedMessage},
extended::{handshake::{ExtendedHandshake, YourIP}, ExtendedMessage},
serialize_piece_preamble, Handshake, Message, MessageOwned, PIECE_MESSAGE_DEFAULT_LEN,
};
use serde::{Deserialize, Serialize};
@ -245,7 +245,9 @@ impl<H: PeerConnectionHandler> PeerConnection<H> {
let supports_extended = handshake_supports_extended;
if supports_extended {
let your_ip = self.addr.ip();
let mut my_extended = ExtendedHandshake::new();
my_extended.yourip = Some(YourIP(your_ip));
self.handler
.update_my_extended_handshake(&mut my_extended)?;
let my_extended = Message::Extended(ExtendedMessage::Handshake(my_extended));

5
crates/peer_binary_protocol/src/extended/handshake.rs
View file

@ -96,7 +96,10 @@ impl Serialize for YourIP {
let buf = ipv4.octets();
serializer.serialize_bytes(&buf)
}
IpAddr::V6(_) => todo!(),
IpAddr::V6(ipv6) => {
let buf = ipv6.octets();
serializer.serialize_bytes(&buf)
},
}
}
}