Update main.py

aidecentralized · Mar 7, 2025 · 3200465 · 3200465
1 parent 3e221d0
commit 3200465
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/utils/communication/grpc/main.py b/src/utils/communication/grpc/main.py
@@ -172,7 +172,7 @@ def update_port(
         with self.lock:
             # FIXME: This is a security vulnerability because
             # any node can update the ip and port of any other node
-            self.peer_ids[request.rank.rank]["ip"] = request.ip  # type: ignore
+            # self.peer_ids[request.rank.rank]["ip"] = request.ip  # type: ignore
             self.peer_ids[request.rank.rank]["port"] = request.port.port  # type: ignore
             return comm_pb2.Empty()  # type: ignore
 
@@ -287,7 +287,7 @@ def start_listener(self):
             ],
         )
         comm_pb2_grpc.add_CommunicationServerServicer_to_server(self.servicer, self.listener)  # type: ignore
-        address = f"{self.host}:{self.port}"
+        address = f"*:{self.port}"
         self.listener.add_insecure_port(address)
         self.listener.start()
         print(f"Started listener on {address}")