Fix linting issues'

This commit is contained in:
Stuckinaboot
2019-05-01 13:54:19 -04:00
parent e555f17a7b
commit 4333c2d061
10 changed files with 31 additions and 31 deletions

View File

@ -2,6 +2,7 @@ import asyncio
import multiaddr
from Crypto.PublicKey import RSA
from libp2p.security.insecure_security import InsecureTransport
from .peer.peerstore import PeerStore
from .peer.id import id_from_public_key
from .network.swarm import Swarm
@ -10,7 +11,6 @@ from .kademlia.routed_host import RoutedHost
from .transport.upgrader import TransportUpgrader
from .transport.tcp.tcp import TCP
from .kademlia.network import KademliaServer
from libp2p.security.insecure_security import InsecureTransport
async def cleanup_done_tasks():

View File

@ -1,7 +1,5 @@
import asyncio
from .raw_connection_interface import IRawConnection
class RawConnection(IRawConnection):
def __init__(self, ip, port, reader, writer, initiator):

View File

@ -147,7 +147,7 @@ class Swarm(INetwork):
# to appropriate stream handler (using multiaddr)
raw_conn = RawConnection(multiaddr.value_for_protocol('ip4'),
multiaddr.value_for_protocol('tcp'), reader, writer, False)
# Per, https://discuss.libp2p.io/t/multistream-security/130, we first secure
# the conn and then mux the conn
secured_conn = await self.upgrader.upgrade_security(raw_conn, peer_id, False)

View File

@ -5,7 +5,7 @@ class InsecureTransport(ISecureTransport):
def __init__(self, transport_id):
self.transport_id = transport_id
async def secure_inbound(self, conn):
"""
Secure the connection, either locally or by communicating with opposing node via conn,

View File

@ -1,5 +1,7 @@
from abc import ABC, abstractmethod
# pylint: disable=W0105
"""
Represents a secured connection object, which includes a connection and details about the security
involved in the secured connection
@ -19,4 +21,3 @@ class ISecureConn(ABC):
"""
:return: map containing details about the connections security
"""

View File

@ -1,9 +1,9 @@
import asyncio
from abc import ABC, abstractmethod
# pylint: disable=W0105
"""
Transport that is used to secure a connection. This transport is
Transport that is used to secure a connection. This transport is
chosen by a security transport multistream module.
Relevant go repo: https://github.com/libp2p/go-conn-security/blob/master/interface.go

View File

@ -1,9 +1,9 @@
import asyncio
from abc import ABC, abstractmethod
from abc import ABC
from libp2p.protocol_muxer.multiselect_client import MultiselectClient
from libp2p.protocol_muxer.multiselect import Multiselect
# pylint: disable=W0105
"""
Represents a secured connection object, which includes a connection and details about the security
involved in the secured connection
@ -13,7 +13,7 @@ Relevant go repo: https://github.com/libp2p/go-conn-security/blob/master/interfa
class SecurityMultistream(ABC):
def __init__(self):
# Map protocol to secure transport
# Map protocol to secure transport
self.transports = {}
# Create multiselect
@ -31,7 +31,7 @@ class SecurityMultistream(ABC):
# we only care about selecting the protocol, not any handler function
self.multiselect.add_handler(protocol, None)
async def secure_inbound(self, conn):
"""
Secure the connection, either locally or by communicating with opposing node via conn,
@ -66,19 +66,21 @@ class SecurityMultistream(ABC):
async def select_transport(self, conn, initiator):
"""
Select a transport that both us and the node on the
Select a transport that both us and the node on the
other end of conn support and agree on
:param conn: conn to choose a transport over
:param initiator: true if we are the initiator, false otherwise
:return: selected secure transport
"""
# TODO: Is conn acceptable to multiselect/multiselect_client instead of stream? In go repo,
# they pass in a raw conn (https://raw.githubusercontent.com/libp2p/go-conn-security-multistream/master/ssms.go)
# TODO: Is conn acceptable to multiselect/multiselect_client
# instead of stream? In go repo, they pass in a raw conn
# (https://raw.githubusercontent.com/libp2p/go-conn-security-multistream/master/ssms.go)
protocol = None
if initiator:
# Select protocol if initiator
protocol = await self.multiselect_client.select_one_of(list(self.transports.keys()), conn)
protocol = \
await self.multiselect_client.select_one_of(list(self.transports.keys()), conn)
else:
# Select protocol if non-initiator
protocol, _ = await self.multiselect.negotiate(conn)

View File

@ -25,8 +25,8 @@ class TransportUpgrader:
"""
if initiator:
return await self.security_multistream.secure_outbound(raw_conn, peer_id)
else:
return await self.security_multistream.secure_inbound(raw_conn)
return await self.security_multistream.secure_inbound(raw_conn)
def upgrade_connection(self, conn, generic_protocol_handler, peer_id):
"""