reth_discv5/
enr.rs

Help
1//! Interface between node identification on protocol version 5 and 4. Specifically, between types
2//! [`discv5::enr::NodeId`] and [`PeerId`].
3
4use discv5::enr::{CombinedPublicKey, EnrPublicKey, NodeId};
5use enr::Enr;
6use reth_network_peers::{id2pk, pk2id, PeerId};
7use secp256k1::{PublicKey, SecretKey};
8
9/// Extracts a [`CombinedPublicKey::Secp256k1`] from a [`discv5::Enr`] and converts it to a
10/// [`PeerId`]. Note: conversion from discv5 ID to discv4 ID is not possible.
11pub fn enr_to_discv4_id(enr: &discv5::Enr) -> Option<PeerId> {
12    let pk = enr.public_key();
13    if !matches!(pk, CombinedPublicKey::Secp256k1(_)) {
14        return None
15    }
16
17    let pk = PublicKey::from_slice(&pk.encode()).unwrap();
18
19    Some(pk2id(&pk))
20}
21
22/// Converts a [`PeerId`] to a [`discv5::enr::NodeId`].
23pub fn discv4_id_to_discv5_id(peer_id: PeerId) -> Result<NodeId, secp256k1::Error> {
24    Ok(id2pk(peer_id)?.into())
25}
26
27/// Converts a [`PeerId`] to a [`reth_network_peers::PeerId`].
28pub fn discv4_id_to_multiaddr_id(
29    peer_id: PeerId,
30) -> Result<discv5::libp2p_identity::PeerId, secp256k1::Error> {
31    let pk = id2pk(peer_id)?.encode();
32    let pk: discv5::libp2p_identity::PublicKey =
33        discv5::libp2p_identity::secp256k1::PublicKey::try_from_bytes(&pk).unwrap().into();
34
35    Ok(pk.to_peer_id())
36}
37
38/// Wrapper around [`discv5::Enr`] ([`Enr<CombinedKey>`]).
39#[derive(Debug, Clone)]
40pub struct EnrCombinedKeyWrapper(pub discv5::Enr);
41
42impl From<Enr<SecretKey>> for EnrCombinedKeyWrapper {
43    fn from(value: Enr<SecretKey>) -> Self {
44        let encoded_enr = alloy_rlp::encode(&value);
45        Self(alloy_rlp::Decodable::decode(&mut &encoded_enr[..]).unwrap())
46    }
47}
48
49impl From<EnrCombinedKeyWrapper> for Enr<SecretKey> {
50    fn from(val: EnrCombinedKeyWrapper) -> Self {
51        let encoded_enr = alloy_rlp::encode(&val.0);
52        alloy_rlp::Decodable::decode(&mut &encoded_enr[..]).unwrap()
53    }
54}
55
56#[cfg(test)]
57mod tests {
58    use super::*;
59    use alloy_rlp::Encodable;
60    use discv5::enr::{CombinedKey, EnrKey};
61    use reth_chainspec::{EthereumHardfork, MAINNET};
62    use reth_network_peers::NodeRecord;
63
64    #[test]
65    fn discv5_discv4_id_conversion() {
66        let discv5_pk = CombinedKey::generate_secp256k1().public();
67        let discv5_peer_id = NodeId::from(discv5_pk.clone());
68
69        // convert to discv4 id
70        let pk = secp256k1::PublicKey::from_slice(&discv5_pk.encode()).unwrap();
71        let discv4_peer_id = pk2id(&pk);
72        // convert back to discv5 id
73        let discv5_peer_id_from_discv4_peer_id = discv4_id_to_discv5_id(discv4_peer_id).unwrap();
74
75        assert_eq!(discv5_peer_id, discv5_peer_id_from_discv4_peer_id)
76    }
77
78    #[test]
79    fn conversion_to_node_record_from_enr() {
80        const IP: &str = "::";
81        const TCP_PORT: u16 = 30303;
82        const UDP_PORT: u16 = 9000;
83
84        let key = CombinedKey::generate_secp256k1();
85
86        let mut buf = Vec::new();
87        let fork_id = MAINNET.hardfork_fork_id(EthereumHardfork::Frontier);
88        fork_id.unwrap().encode(&mut buf);
89
90        let enr = Enr::builder()
91            .ip6(IP.parse().unwrap())
92            .udp6(UDP_PORT)
93            .tcp6(TCP_PORT)
94            .build(&key)
95            .unwrap();
96
97        let enr = EnrCombinedKeyWrapper(enr).into();
98        let node_record = NodeRecord::try_from(&enr).unwrap();
99
100        assert_eq!(
101            NodeRecord {
102                address: IP.parse().unwrap(),
103                tcp_port: TCP_PORT,
104                udp_port: UDP_PORT,
105                id: pk2id(&enr.public_key())
106            },
107            node_record
108        )
109    }
110}
reth_discv5/enr.rs

reth_discv5/
enr.rs
