jleony/nanoreth
1
0
Fork 0
mirror of https://github.com/hl-archive-node/nanoreth.git synced 2025-12-06 10:59:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(net): use real tcp port for local node record (#8853)

Browse Source
This commit is contained in:
Matthias Seitz
2024-06-14 23:38:19 +02:00
committed by GitHub
parent 343ff13fb5
commit 22e6afe3f1
5 changed files with 78 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
crates/net/peers/src/node_record.rs
View File

@ -42,8 +42,10 @@ pub struct NodeRecord {
}
impl NodeRecord {
/// Derive the [`NodeRecord`] from the secret key and addr.
///
/// Note: this will set both the TCP and UDP ports to the port of the addr.
#[cfg(feature = "secp256k1")]
/// Derive the [`NodeRecord`] from the secret key and addr
pub fn from_secret_key(addr: SocketAddr, sk: &secp256k1::SecretKey) -> Self {
let pk = secp256k1::PublicKey::from_secret_key(secp256k1::SECP256K1, sk);
let id = PeerId::from_slice(&pk.serialize_uncompressed()[1..]);
@ -73,8 +75,19 @@ impl NodeRecord {
self
}
/// Sets the tcp port
pub const fn with_tcp_port(mut self, port: u16) -> Self {
self.tcp_port = port;
self
}
/// Sets the udp port
pub const fn with_udp_port(mut self, port: u16) -> Self {
self.udp_port = port;
self
}
/// Creates a new record from a socket addr and peer id.
#[allow(dead_code)]
pub const fn new(addr: SocketAddr, id: PeerId) -> Self {
Self { address: addr.ip(), tcp_port: addr.port(), udp_port: addr.port(), id }
}