Bitmessage/MiNode
3
0
Fork 1
Code Pull Requests 9 Actions Activity

Tweaks to connection logic

Browse Source
This commit is contained in:
TheKysek 2016-07-19 12:48:04 +02:00
parent dd3e93812a
commit 89baf6258b
1 changed files with 21 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
src/connection.py
View File

@ -23,8 +23,11 @@ class Connection(threading.Thread):
self.vectors_to_get = set() self.vectors_to_get = set()
self.status = 'ready' self.status = 'ready'
self.sent_verack = False
self.sent_big_inv_message = False self.tls = False
self.verack_received = False
self.verack_sent = False
self.host = host self.host = host
self.port = int(port) self.port = int(port)
@ -67,9 +70,8 @@ class Connection(threading.Thread):
data = None data = None
if time.time() - self.last_message_sent > 300 and self.status == 'fully_established': if time.time() - self.last_message_sent > 300 and self.status == 'fully_established':
self.send_queue.put(message.Message(b'pong', b'')) self.send_queue.put(message.Message(b'pong', b''))
if not self.sent_big_inv_message and self.status == 'verack_received' and self.sent_verack:
self._on_connection_fully_established()
except ConnectionResetError: except ConnectionResetError:
logging.debug('Disconnecting from {};{}. Reason: ConnectionResetError'.format(self.host, self.port))
data = None data = None
self._process_buffer() self._process_buffer()
self._request_objects() self._request_objects()
@ -117,6 +119,7 @@ class Connection(threading.Thread):
print(e) print(e)
break break
self.s.settimeout(0.5) self.s.settimeout(0.5)
self.tls = True
logging.debug('Established TLS connection with {}:{}'.format(self.host, self.port)) logging.debug('Established TLS connection with {}:{}'.format(self.host, self.port))
def _send_message(self, m): def _send_message(self, m):
@ -130,18 +133,22 @@ class Connection(threading.Thread):
def _on_connection_fully_established(self): def _on_connection_fully_established(self):
self.status = 'fully_established' self.status = 'fully_established'
time.sleep(2) logging.info('Established Bitmessage protocol connection to {}:{}'.format(self.host, self.port))
if self.remote_version.services & 2: # NODE_SSL
self._do_tls_handshake()
with shared.objects_lock: with shared.objects_lock:
self.send_queue.put(message.Inv({vector for vector in shared.objects.keys() if shared.objects[vector].expires_time > time.time()})) self.send_queue.put(message.Inv({vector for vector in shared.objects.keys() if shared.objects[vector].expires_time > time.time()}))
addr = {structure.NetAddr(1, c.host, c.port) for c in shared.connections.copy() if not c.server and c.status == 'fully_established'} addr = {structure.NetAddr(1, c.host, c.port) for c in shared.connections.copy() if not c.server and c.status == 'fully_established'}
if len(addr) != 0: if len(addr) != 0:
self.send_queue.put(message.Addr(addr)) self.send_queue.put(message.Addr(addr))
self.sent_big_inv_message = True
def _process_queue(self): def _process_queue(self):
while not self.send_queue.empty(): while not self.send_queue.empty():
m = self.send_queue.get() m = self.send_queue.get()
if m: if m:
if m == 'fully_established':
self._on_connection_fully_established()
else:
self._send_message(m) self._send_message(m)
self.last_message_sent = time.time() self.last_message_sent = time.time()
else: else:
@ -172,20 +179,21 @@ class Connection(threading.Thread):
self.send_queue.put(None) self.send_queue.put(None)
else: else:
self.send_queue.put(message.Message(b'verack', b'')) self.send_queue.put(message.Message(b'verack', b''))
self.sent_verack = True self.verack_sent = True
self.remote_version = version self.remote_version = version
if not self.server: if not self.server:
self._do_tls_handshake() self.send_queue.put('fully_established')
shared.address_advertise_queue.put(structure.NetAddr(version.services, self.host, self.port)) shared.address_advertise_queue.put(structure.NetAddr(version.services, self.host, self.port))
shared.node_pool.add((self.host, self.port)) shared.node_pool.add((self.host, self.port))
shared.address_advertise_queue.put(structure.NetAddr(shared.services, version.host, shared.listening_port)) shared.address_advertise_queue.put(structure.NetAddr(shared.services, version.host, shared.listening_port))
if self.server: if self.server:
self.send_queue.put(message.Version(self.host, self.port)) self.send_queue.put(message.Version(self.host, self.port))
self._do_tls_handshake()
elif m.command == b'verack': elif m.command == b'verack':
self.status = 'verack_received' self.verack_received = True
logging.debug('{}:{} -> {}'.format(self.host, self.port, 'verack')) logging.debug('{}:{} -> {}'.format(self.host, self.port, 'verack'))
logging.info('Established Bitmessage protocol connection to {}:{}'.format(self.host, self.port)) if self.server:
self.send_queue.put('fully_established')
elif m.command == b'inv': elif m.command == b'inv':
inv = message.Inv.from_message(m) inv = message.Inv.from_message(m)
logging.debug('{}:{} -> {}'.format(self.host, self.port, inv)) logging.debug('{}:{} -> {}'.format(self.host, self.port, inv))