From 667af585e7a1308d28c52d07c8128b8083aec2b4 Mon Sep 17 00:00:00 2001
From: mempirate <jonas.bostoen@hotmail.com>
Date: Mon, 24 Apr 2023 19:16:47 +0200
Subject: [PATCH] feat(net): add `remote_addr` to session established event
 (#2373)

---
 crates/net/network/src/manager.rs      | 3 +++
 crates/net/network/src/transactions.rs | 4 ++++
 2 files changed, 7 insertions(+)

diff --git a/crates/net/network/src/manager.rs b/crates/net/network/src/manager.rs
index 55625a30d3..bc15b44b95 100644
--- a/crates/net/network/src/manager.rs
+++ b/crates/net/network/src/manager.rs
@@ -663,6 +663,7 @@ where
                             }
                             this.event_listeners.send(NetworkEvent::SessionEstablished {
                                 peer_id,
+                                remote_addr,
                                 capabilities,
                                 version,
                                 status,
@@ -855,6 +856,8 @@ pub enum NetworkEvent {
     SessionEstablished {
         /// The identifier of the peer to which a session was established.
         peer_id: PeerId,
+        /// The remote addr of the peer to which a session was established.
+        remote_addr: SocketAddr,
         /// Capabilities the peer announced
         capabilities: Arc<Capabilities>,
         /// A request channel to the session task.
diff --git a/crates/net/network/src/transactions.rs b/crates/net/network/src/transactions.rs
index 3794eab1e6..38c3a14f7a 100644
--- a/crates/net/network/src/transactions.rs
+++ b/crates/net/network/src/transactions.rs
@@ -795,6 +795,7 @@ mod tests {
             match ev {
                 NetworkEvent::SessionEstablished {
                     peer_id,
+                    remote_addr,
                     capabilities,
                     messages,
                     status,
@@ -803,6 +804,7 @@ mod tests {
                     // to insert a new peer in transactions peerset
                     transactions.on_network_event(NetworkEvent::SessionEstablished {
                         peer_id,
+                        remote_addr,
                         capabilities,
                         messages,
                         status,
@@ -870,12 +872,14 @@ mod tests {
             match ev {
                 NetworkEvent::SessionEstablished {
                     peer_id,
+                    remote_addr,
                     capabilities,
                     messages,
                     status,
                     version,
                 } => transactions.on_network_event(NetworkEvent::SessionEstablished {
                     peer_id,
+                    remote_addr,
                     capabilities,
                     messages,
                     status,