Merge branch 'master' into fix/refactor-mplex-swarm-host

libp2p/host/basic_host.py

@@ -1,5 +1,6 @@
 import asyncio
 from typing import List, Sequence
+import logging
 
 import multiaddr
 
@@ -13,6 +14,8 @@ from libp2p.protocol_muxer.multiselect_client import MultiselectClient
 from libp2p.protocol_muxer.multiselect_communicator import MultiselectCommunicator
 from libp2p.routing.kademlia.kademlia_peer_router import KadmeliaPeerRouter
 from libp2p.typing import StreamHandlerFn, TProtocol
+from libp2p.protocol_muxer.exceptions import MultiselectClientError, MultiselectError
+from libp2p.host.exceptions import StreamFailure
 
 from .host_interface import IHost
 
@@ -22,6 +25,10 @@ from .host_interface import IHost
 # telling it to listen on the given listen addresses.
 
 
+logger = logging.getLogger("libp2p.network.basic_host")
+logger.setLevel(logging.DEBUG)
+
+
 class BasicHost(IHost):
 
     _network: INetwork
@@ -101,9 +108,14 @@ class BasicHost(IHost):
         net_stream = await self._network.new_stream(peer_id, protocol_ids)
 
         # Perform protocol muxing to determine protocol to use
-        selected_protocol = await self.multiselect_client.select_one_of(
-            list(protocol_ids), MultiselectCommunicator(net_stream)
-        )
+        try:
+            selected_protocol = await self.multiselect_client.select_one_of(
+                list(protocol_ids), MultiselectCommunicator(net_stream)
+            )
+        except MultiselectClientError as error:
+            logger.debug("fail to open a stream to peer %s, error=%s", peer_id, error)
+            await net_stream.reset()
+            raise StreamFailure("failt to open a stream to peer %s", peer_id) from error
 
         net_stream.set_protocol(selected_protocol)
         return net_stream
@@ -135,8 +147,12 @@ class BasicHost(IHost):
     # Reference: `BasicHost.newStreamHandler` in Go.
     async def _swarm_stream_handler(self, net_stream: INetStream) -> None:
         # Perform protocol muxing to determine protocol to use
-        protocol, handler = await self.multiselect.negotiate(
-            MultiselectCommunicator(net_stream)
-        )
+        try:
+            protocol, handler = await self.multiselect.negotiate(
+                MultiselectCommunicator(net_stream)
+            )
+        except MultiselectError:
+            await net_stream.reset()
+            return
         net_stream.set_protocol(protocol)
         asyncio.ensure_future(handler(net_stream))