adding additional bugfixed files

network/multiaddr.py

@@ -2,6 +2,8 @@ class MultiAddr:
 
     # Validates input string and constructs internal representation.
     def __init__(self, addr):
+        self.protocol_map = dict()
+
         # Empty multiaddrs are valid.
         if not addr:
             self.protocol_map = dict()

network/stream/net_stream.py

@@ -19,19 +19,19 @@ class NetStream(INetStream):
         """
         self.protocol_id = protocol_id
 
-    def read(self):
+    async def read(self):
         """
         read from stream
         :return: bytes of input until EOF
         """
-        return self.muxed_stream.read()
+        return await self.muxed_stream.read()
 
-    def write(self, bytes):
+    async def write(self, bytes):
         """
         write to stream
         :return: number of bytes written
         """
-        return self.muxed_stream.write(bytes)
+        return await self.muxed_stream.write(bytes)
 
     def close(self):
         """

network/swarm.py

@@ -22,13 +22,20 @@ class Swarm(INetwork):
         """
         self.stream_handlers[protocol_id] = stream_handler
 
-    def new_stream(self, peer_id, protocol_id):
+    async def new_stream(self, peer_id, protocol_id):
         """
         :param peer_id: peer_id of destination
         :param protocol_id: protocol id
         :return: net stream instance
         """
-        muxed_conn = None
+        # Get peer info from peer store
+        addrs = self.peerstore.addrs(peer_id)
+
+        if not addrs:
+            raise SwarmException("No known addresses to peer")
+
+        multiaddr = addrs[0]
+
         if peer_id in self.connections:
             """
             If muxed connection already exists for peer_id,
@@ -37,14 +44,8 @@ class Swarm(INetwork):
             """
             muxed_conn = self.connections[peer_id]
         else:
-            # Get peer info from peer store
-            addrs = self.peerstore.addrs(peer_id)
-
             # Transport dials peer (gets back a raw conn)
-            if not addrs:
-                raise SwarmException("No known addresses to peer")
-            first_addr = addrs[0]
-            raw_conn = self.transport.dial(first_addr)
+            raw_conn = await self.transport.dial(MultiAddr(multiaddr))
 
             # Use upgrader to upgrade raw conn to muxed conn
             muxed_conn = self.upgrader.upgrade_connection(raw_conn, True)
@@ -54,28 +55,30 @@ class Swarm(INetwork):
 
         # Use muxed conn to open stream, which returns
         # a muxed stream
-        stream_id = str(uuid.uuid4())
-        muxed_stream = muxed_conn.open_stream(protocol_id, stream_id, peer_id, first_addr)
+        # TODO: use better stream IDs
+        stream_id = (uuid.uuid4().int & (1<<64)-1) >> 3
+        muxed_stream = muxed_conn.open_stream(protocol_id, stream_id, peer_id, multiaddr)
 
         # Create a net stream
         net_stream = NetStream(muxed_stream)
 
         return net_stream
 
-    def listen(self, *args):
+    async def listen(self, *args):
         """
         :param *args: one or many multiaddrs to start listening on
         :return: true if at least one success
         """
-
-        # For each multiaddr in args
-            # Check if a listener for multiaddr exists already
-            # If listener already exists, continue
-            # Otherwise, do the following:
-                # Pass multiaddr into conn handler
-                # Have conn handler delegate to stream handler
-                # Call listener listen with the multiaddr
-                # Map multiaddr to listener
+        """
+        For each multiaddr in args
+            Check if a listener for multiaddr exists already
+            If listener already exists, continue
+            Otherwise:
+                Capture multiaddr in conn handler
+                Have conn handler delegate to stream handler
+                Call listener listen with the multiaddr
+                Map multiaddr to listener
+        """
         for multiaddr_str in args:
             if multiaddr_str in self.listeners:
                 return True
@@ -83,27 +86,28 @@ class Swarm(INetwork):
             multiaddr = MultiAddr(multiaddr_str)
             multiaddr_dict = multiaddr.to_options()
 
-            def conn_handler(reader, writer):
+            async def conn_handler(reader, writer):
                 # Upgrade reader/write to a net_stream and pass to appropriate stream handler (using multiaddr)
-                raw_conn = RawConnection(multiaddr_dict.host, multiaddr_dict.port, reader, writer)
+                raw_conn = RawConnection(multiaddr_dict['host'], multiaddr_dict['port'], reader, writer)
                 muxed_conn = self.upgrader.upgrade_connection(raw_conn, False)
 
-                muxed_stream, stream_id, protocol_id = muxed_conn.accept_stream()
+                muxed_stream, stream_id, protocol_id = await muxed_conn.accept_stream()
                 net_stream = NetStream(muxed_stream)
                 net_stream.set_protocol(protocol_id)
 
                 # Give to stream handler
                 # TODO: handle case when stream handler is set
-                self.stream_handlers[protocol_id](net_stream)
+                # TODO: handle case of multiple protocols over same raw connection
+                await self.stream_handlers[protocol_id](net_stream)
 
             try:
                 # Success
                 listener = self.transport.create_listener(conn_handler)
-                listener.listen(multiaddr)
+                await listener.listen(multiaddr)
                 return True
             except IOError:
                 # Failed. Continue looping.
-                print("Failed to connect to: " + multiaddr)
+                print("Failed to connect to: " + str(multiaddr))
 
         # No multiaddr succeeded
         return False