Add eth entry to local enr record.

Signed-off-by: lakshya-sky <[email protected]>

Author: lakshya-sky

cmd/ethrex/initializers.rs

@@ -331,15 +331,15 @@ pub fn get_local_node_record(
 ) -> NodeRecord {
     match read_node_config_file(datadir) {
         Ok(Some(ref mut config)) => {
-            NodeRecord::from_node(local_p2p_node, config.node_record.seq + 1, signer)
+            NodeRecord::from_node(local_p2p_node, config.node_record.seq + 1, signer, None)
                 .expect("Node record could not be created from local node")
         }
         _ => {
             let timestamp = SystemTime::now()
                 .duration_since(UNIX_EPOCH)
                 .unwrap_or_default()
                 .as_secs();
-            NodeRecord::from_node(local_p2p_node, timestamp, signer)
+            NodeRecord::from_node(local_p2p_node, timestamp, signer, None)
                 .expect("Node record could not be created from local node")
         }
     }

crates/networking/p2p/discv4/server.rs

@@ -15,6 +15,7 @@ use crate::{
 };
 use bytes::BytesMut;
 use ethrex_common::{H256, H512};
+use ethrex_storage::Store;
 use futures::StreamExt;
 use rand::rngs::OsRng;
 use secp256k1::SecretKey;
@@ -77,6 +78,7 @@ pub enum OutMessage {
 
 #[derive(Debug)]
 pub struct DiscoveryServer {
+    storage: Store,
     local_node: Node,
     local_node_record: NodeRecord,
     signer: SecretKey,
@@ -89,6 +91,7 @@ pub struct DiscoveryServer {
 
 impl DiscoveryServer {
     pub async fn spawn(
+        storage: Store,
         local_node: Node,
         signer: SecretKey,
         udp_socket: Arc<UdpSocket>,
@@ -97,9 +100,11 @@ impl DiscoveryServer {
     ) -> Result<(), DiscoveryServerError> {
         info!("Starting Discovery Server");
 
-        let local_node_record = NodeRecord::from_node(&local_node, 1, &signer)
+        let fork_id = storage.get_fork_id().await.ok();
+        let local_node_record = NodeRecord::from_node(&local_node, 1, &signer, fork_id)
             .expect("Failed to create local node record");
         let mut discovery_server = Self {
+            storage,
             local_node: local_node.clone(),
             local_node_record,
             signer,

crates/networking/p2p/network.rs

@@ -111,6 +111,7 @@ pub async fn start_network(context: P2PContext, bootnodes: Vec<Node>) -> Result<
     );
 
     DiscoveryServer::spawn(
+        context.storage.clone(),
         context.local_node.clone(),
         context.signer,
         udp_socket.clone(),

crates/networking/p2p/types.rs

@@ -347,7 +347,12 @@ impl NodeRecord {
         Ok(result)
     }
 
-    pub fn from_node(node: &Node, seq: u64, signer: &SecretKey) -> Result<Self, NodeError> {
+    pub fn from_node(
+        node: &Node,
+        seq: u64,
+        signer: &SecretKey,
+        fork_id: Option<ForkId>,
+    ) -> Result<Self, NodeError> {
         let mut record = NodeRecord {
             seq,
             ..Default::default()
@@ -372,6 +377,12 @@ impl NodeRecord {
             .pairs
             .push(("udp".into(), node.udp_port.encode_to_vec().into()));
 
+        if let Some(fork_id) = fork_id {
+            record
+                .pairs
+                .push(("eth".into(), fork_id.encode_to_vec().into()));
+        }
+
         record.signature = record.sign_record(signer)?;
 
         Ok(record)
@@ -581,7 +592,7 @@ mod tests {
             addr.port(),
             public_key_from_signing_key(&signer),
         );
-        let mut record = NodeRecord::from_node(&node, 1, &signer).unwrap();
+        let mut record = NodeRecord::from_node(&node, 1, &signer, None).unwrap();
         // Drop fork ID since the test doesn't use it
         record.pairs.retain(|(k, _)| k != "eth");
         record.sign_record(&signer).unwrap();

crates/networking/rpc/admin/peers.rs

@@ -167,7 +167,7 @@ mod tests {
     fn test_peer_data_to_serialized_peer() {
         // Test that we can correctly serialize an active Peer
         let node = Node::from_enode_url("enode://4aeb4ab6c14b23e2c4cfdce879c04b0748a20d8e9b59e25ded2a08143e265c6c25936e74cbc8e641e3312ca288673d91f2f93f8e277de3cfa444ecdaaf982052@157.90.35.166:30303").unwrap();
-        let record = NodeRecord::from_node(&node, 17, &SecretKey::new(&mut OsRng)).unwrap();
+        let record = NodeRecord::from_node(&node, 17, &SecretKey::new(&mut OsRng), None).unwrap();
         let mut peer = PeerData::new(
             node,
             Some(record),

crates/networking/rpc/test_utils.rs

@@ -208,7 +208,7 @@ pub fn example_local_node_record() -> NodeRecord {
     let node = Node::new("127.0.0.1".parse().unwrap(), 30303, 30303, public_key_1);
     let signer = SecretKey::new(&mut rand::rngs::OsRng);
 
-    NodeRecord::from_node(&node, 1, &signer).unwrap()
+    NodeRecord::from_node(&node, 1, &signer, None).unwrap()
 }
 
 // Util to start an api for testing on ports 8500 and 8501,