From 2ed2587fc90b6d987f208740dd6a6d113b1c565e Mon Sep 17 00:00:00 2001
From: Abhinav Agarwalla <120122716+lla-dane@users.noreply.github.com>
Date: Tue, 17 Jun 2025 17:55:50 +0530
Subject: [PATCH] fix: removed dummy ID(b) from upgrade_security for inbound
 connections (#681)

* fix: removed dummy ID(b) from upgrade_security for inbound connections

* added newsfragment

* updated newsfragment
---
 libp2p/network/swarm.py        | 7 ++-----
 libp2p/transport/upgrader.py   | 7 ++++++-
 newsfragments/681.breaking.rst | 2 ++
 3 files changed, 10 insertions(+), 6 deletions(-)
 create mode 100644 newsfragments/681.breaking.rst

diff --git a/libp2p/network/swarm.py b/libp2p/network/swarm.py
index d19b8177..706d649a 100644
--- a/libp2p/network/swarm.py
+++ b/libp2p/network/swarm.py
@@ -187,7 +187,7 @@ class Swarm(Service, INetworkService):
         # Per, https://discuss.libp2p.io/t/multistream-security/130, we first secure
         # the conn and then mux the conn
         try:
-            secured_conn = await self.upgrader.upgrade_security(raw_conn, peer_id, True)
+            secured_conn = await self.upgrader.upgrade_security(raw_conn, True, peer_id)
         except SecurityUpgradeFailure as error:
             logger.debug("failed to upgrade security for peer %s", peer_id)
             await raw_conn.close()
@@ -257,10 +257,7 @@ class Swarm(Service, INetworkService):
                 # Per, https://discuss.libp2p.io/t/multistream-security/130, we first
                 # secure the conn and then mux the conn
                 try:
-                    # FIXME: This dummy `ID(b"")` for the remote peer is useless.
-                    secured_conn = await self.upgrader.upgrade_security(
-                        raw_conn, ID(b""), False
-                    )
+                    secured_conn = await self.upgrader.upgrade_security(raw_conn, False)
                 except SecurityUpgradeFailure as error:
                     logger.debug("failed to upgrade security for peer at %s", maddr)
                     await raw_conn.close()
diff --git a/libp2p/transport/upgrader.py b/libp2p/transport/upgrader.py
index f1c49af1..8b47fff4 100644
--- a/libp2p/transport/upgrader.py
+++ b/libp2p/transport/upgrader.py
@@ -48,11 +48,16 @@ class TransportUpgrader:
         # TODO: Figure out what to do with this function.
 
     async def upgrade_security(
-        self, raw_conn: IRawConnection, peer_id: ID, is_initiator: bool
+        self,
+        raw_conn: IRawConnection,
+        is_initiator: bool,
+        peer_id: ID | None = None,
     ) -> ISecureConn:
         """Upgrade conn to a secured connection."""
         try:
             if is_initiator:
+                if peer_id is None:
+                    raise ValueError("peer_id must be provided for outbout connection")
                 return await self.security_multistream.secure_outbound(
                     raw_conn, peer_id
                 )
diff --git a/newsfragments/681.breaking.rst b/newsfragments/681.breaking.rst
new file mode 100644
index 00000000..55050d65
--- /dev/null
+++ b/newsfragments/681.breaking.rst
@@ -0,0 +1,2 @@
+Reordered the arguments to `upgrade_security` to place `is_initiator` before `peer_id`, and made `peer_id` optional.
+This allows the method to reflect the fact that peer identity is not required for inbound connections.