Skip to content

Commit

Permalink
[tests] Move version message sending from NodeConn to NodeConnCB
Browse files Browse the repository at this point in the history
This commit moves the logic that sends a version message
on connection from NodeConn to NodeConnCB. NodeConn should
not be aware of the semantics or meaning of the P2P payloads.
  • Loading branch information
jnewbery committed Nov 28, 2017
1 parent dad596f commit e9dfa9b
Showing 1 changed file with 14 additions and 11 deletions.
25 changes: 14 additions & 11 deletions test/functional/test_framework/mininode.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@ class NodeConn(asyncore.dispatcher):
def __init__(self):
super().__init__(map=mininode_socket_map)

def peer_connect(self, dstaddr, dstport, net="regtest", services=NODE_NETWORK|NODE_WITNESS, send_version=True):
def peer_connect(self, dstaddr, dstport, net="regtest"):
self.dstaddr = dstaddr
self.dstport = dstport
self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
Expand All @@ -86,16 +86,6 @@ def peer_connect(self, dstaddr, dstport, net="regtest", services=NODE_NETWORK|NO
self.network = net
self.disconnect = False

if send_version:
# stuff version msg into sendbuf
vt = msg_version()
vt.nServices = services
vt.addrTo.ip = self.dstaddr
vt.addrTo.port = self.dstport
vt.addrFrom.ip = "0.0.0.0"
vt.addrFrom.port = 0
self.send_message(vt, True)

logger.info('Connecting to Bitcoin Node: %s:%d' % (self.dstaddr, self.dstport))

try:
Expand Down Expand Up @@ -279,6 +269,19 @@ def __init__(self):
# The network services received from the peer
self.nServices = 0

def peer_connect(self, *args, services=NODE_NETWORK|NODE_WITNESS, send_version=True, **kwargs):
super().peer_connect(*args, **kwargs)

if send_version:
# Send a version msg
vt = msg_version()
vt.nServices = services
vt.addrTo.ip = self.dstaddr
vt.addrTo.port = self.dstport
vt.addrFrom.ip = "0.0.0.0"
vt.addrFrom.port = 0
self.send_message(vt, True)

# Message receiving methods

def on_message(self, message):
Expand Down

0 comments on commit e9dfa9b

Please sign in to comment.