def msg_version(self, msg: VersionBtcMessage) -> None:
"""
Handle version message.
Gateway initiates connection, so do not check for misbehavior. Record that we received the version message,
send a verack, and synchronize chains if need be.
:param msg: VERSION message
"""
self.request_witness_data = msg.services(
) & NODE_WITNESS_SERVICE_FLAG > 0
if self.request_witness_data:
self.connection.log_debug(
"Connection with Bitcoin node supports SegWit.")
self.connection.on_connection_established()
reply = VerAckBtcMessage(self.magic)
self.connection.enqueue_msg(reply)
send_compact_msg = SendCompactBtcMessage(
self.magic, on_flag=self.node.opts.compact_block, version=1)
self.node.alarm_queue.register_alarm(2, self.connection.enqueue_msg,
send_compact_msg)
self.node.alarm_queue.register_alarm(self.ping_interval_s,
self.connection.send_ping)
if self.connection.is_active():
self.node.on_blockchain_connection_ready(self.connection)
def test_parse_message_success_all_types(self):
# TODO: pull these numbers into constants, along with all the BTC messages
self.create_message_successfully(self.VERSION_BTC_MESSAGE,
VersionBtcMessage)
self.create_message_successfully(VerAckBtcMessage(self.MAGIC),
VerAckBtcMessage)
self.create_message_successfully(PingBtcMessage(self.MAGIC),
PingBtcMessage)
self.create_message_successfully(PongBtcMessage(self.MAGIC, 123),
PongBtcMessage)
self.create_message_successfully(GetAddrBtcMessage(self.MAGIC),
GetAddrBtcMessage)
self.create_message_successfully(
AddrBtcMessage(self.MAGIC,
[(int(time.time()), "127.0.0.1", 8000)]),
AddrBtcMessage)
inv_vector = [(1, self.HASH), (2, self.HASH)]
self.create_message_successfully(InvBtcMessage(self.MAGIC, inv_vector),
InvBtcMessage)
self.create_message_successfully(
GetDataBtcMessage(self.MAGIC, inv_vector), GetDataBtcMessage)
self.create_message_successfully(
NotFoundBtcMessage(self.MAGIC, inv_vector), NotFoundBtcMessage)
hashes = [self.HASH, self.HASH]
self.create_message_successfully(
GetHeadersBtcMessage(self.MAGIC, self.VERSION, hashes, self.HASH),
GetHeadersBtcMessage)
self.create_message_successfully(
GetBlocksBtcMessage(self.MAGIC, self.VERSION, hashes, self.HASH),
GetBlocksBtcMessage)
self.create_message_successfully(
TxBtcMessage(self.MAGIC, self.VERSION, [], [], 0), TxBtcMessage)
txs = [TxIn(buf=bytearray(10), length=10, off=0).rawbytes()] * 5
self.create_message_successfully(
BlockBtcMessage(self.MAGIC, self.VERSION, self.HASH, self.HASH, 0,
0, 0, txs), BlockBtcMessage)
self.create_message_successfully(
HeadersBtcMessage(self.MAGIC,
[helpers.generate_bytearray(81)] * 2),
HeadersBtcMessage)
self.create_message_successfully(
RejectBtcMessage(self.MAGIC, b"a message",
RejectBtcMessage.REJECT_MALFORMED, b"test break",
helpers.generate_bytearray(10)), RejectBtcMessage)
self.create_message_successfully(SendHeadersBtcMessage(self.MAGIC),
SendHeadersBtcMessage)
self.create_message_successfully(
FeeFilterBtcMessage(self.MAGIC, fee_rate=100), FeeFilterBtcMessage)
self.create_message_successfully(
BtcMessage(self.MAGIC, b'xversion', 0, bytearray(30)),
XversionBtcMessage)
def setUp(self):
self.local_node_fileno = 1
self.remote_node_fileno = 2
self.gateway_node = spies.make_spy_node(BtcGatewayNode,
8000,
include_default_btc_args=True)
self.btc_node_connection = spies.make_spy_connection(
BtcNodeConnection, self.local_node_fileno, 8001, self.gateway_node)
self.btc_remote_node_connection = spies.make_spy_connection(
BtcRemoteConnection, self.remote_node_fileno, 8002,
self.gateway_node)
self.gateway_node.sdn_connection = MagicMock()
self.gateway_node.node_conn = self.btc_node_connection
self.gateway_node.remote_node_conn = self.btc_remote_node_connection
self.gateway_node.connection_pool.add(self.local_node_fileno,
LOCALHOST, 8001,
self.btc_node_connection)
self.gateway_node.connection_pool.add(self.remote_node_fileno,
LOCALHOST, 8002,
self.btc_remote_node_connection)
self.gateway_node.account_id = "1234"
self.handshake = \
bytearray(VersionBtcMessage(100, 1000, "127.0.0.1", 8333, "0.0.0.0", 8333,
13, 0, bytearray("hello", "utf-8"), 0).rawbytes()) + \
bytearray(VerAckBtcMessage(100).rawbytes())
inps = [
TxIn(prev_outpoint_hash=bytearray("0" * 32, "utf-8"),
prev_out_index=10,
sig_script=bytearray(1000),
sequence=32)
]
outs = [TxOut(value=10, pk_script=bytearray(1000))]
self.txmsg = bytearray(
TxBtcMessage(magic=100,
version=0,
tx_in=inps,
tx_out=outs,
lock_time=12345).rawbytes())
self.blockmsg = bytearray(
BlockBtcMessage(magic=100,
version=0,
prev_block=get_object_hash(bytearray(32)),
merkle_root=get_object_hash(bytearray(32)),
timestamp=1023,
bits=1,
txns=[bytearray(64)] * 32000,
nonce=34).rawbytes())
self.transform = MessageTransformations()
def msg_version(self, _msg):
"""
Handle version message.
Gateway initiates connection, so do not check for misbehavior. Record that we received the version message,
send a verack, and synchronize chains if need be.
"""
self.connection.on_connection_established()
reply = VerAckBtcMessage(self.magic)
self.connection.enqueue_msg(reply)
self.connection.node.alarm_queue.register_alarm(
self.ping_interval_s, self.connection.send_ping)
if self.connection.is_active():
self.connection.node.on_remote_blockchain_connection_ready(
self.connection)