RanchiMall/electrumx
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Have proxy.create_connection use kwargs

Browse Source 
Fixes #227
This commit is contained in:
Neil Booth 2017-08-21 23:38:54 +09:00
parent ab62092afc
commit fab3f5affe
2 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lib/socks.py
View File

@ -224,7 +224,7 @@ class SocksProxy(util.LoggedClass):
self.logger.info('detected proxy at {} ({})'
.format(util.address_string(paddress), self.ip_addr))
async def create_connection(self, protocol_factory, host, port, ssl=None):
async def create_connection(self, protocol_factory, host, port, **kwargs):
'''All arguments are as to asyncio's create_connection method.'''
try:
sock = await self.connect_via_proxy(host, port)
@ -236,6 +236,6 @@ class SocksProxy(util.LoggedClass):
self.set_lost()
raise
hostname = host if ssl else None
hostname = host if kwargs.get('ssl') else None
return await self.loop.create_connection(
protocol_factory, ssl=ssl, sock=sock, server_hostname=hostname)
protocol_factory, sock=sock, server_hostname=hostname, **kwargs)

7
server/peers.py
View File

@ -560,11 +560,12 @@ class PeerManager(util.LoggedClass):
create_connection = self.proxy.create_connection
else:
create_connection = self.loop.create_connection
local_addr = (self.env.host, None) if self.env.host else None
local_addr = (self.env.host, None) if self.env.host else None
protocol_factory = partial(PeerSession, peer, self, kind)
coro = create_connection(protocol_factory, peer.host, port, ssl=sslc, local_addr=local_addr)
coro = create_connection(protocol_factory, peer.host, port, ssl=sslc,
local_addr=local_addr)
callback = partial(self.connection_done, peer, port_pairs)
self.ensure_future(coro, callback)