From f8976434ed17a9d5ff6c1bf9e78ee1b58ddac3f4 Mon Sep 17 00:00:00 2001
From: Gleb Naumenko <naumenko.gs@gmail.com>
Date: Thu, 11 Nov 2021 09:14:09 +0300
Subject: [PATCH] Add functional test support for sendrecon message

This is needed to pass existing functional tests.
---
 test/functional/test_framework/messages.py | 28 ++++++++++++++++++++++
 test/functional/test_framework/p2p.py      |  3 +++
 2 files changed, 31 insertions(+)

diff --git a/test/functional/test_framework/messages.py b/test/functional/test_framework/messages.py
index f57b6e74944d7..a261b944ef3c0 100755
--- a/test/functional/test_framework/messages.py
+++ b/test/functional/test_framework/messages.py
@@ -1817,3 +1817,31 @@ def serialize(self):
     def __repr__(self):
         return "msg_cfcheckpt(filter_type={:#x}, stop_hash={:x})".format(
             self.filter_type, self.stop_hash)
+
+class msg_sendrecon:
+    __slots__ = ("initiator", "responder", "version", "salt")
+    msgtype = b"sendrecon"
+
+    def __init__(self):
+        self.initiator = False
+        self.responder = False
+        self.version = 0
+        self.salt = 0
+
+    def deserialize(self, f):
+        self.initiator = struct.unpack("<?", f.read(1))[0]
+        self.responder = struct.unpack("<?", f.read(1))[0]
+        self.version = struct.unpack("<I", f.read(4))[0]
+        self.salt = struct.unpack("<Q", f.read(8))[0]
+
+    def serialize(self):
+        r = b""
+        r += struct.pack("<?", self.initiator)
+        r += struct.pack("<?", self.responder)
+        r += struct.pack("<I", self.version)
+        r += struct.pack("<Q", self.salt)
+        return r
+
+    def __repr__(self):
+        return "msg_sendrecon(initiator=%i, responder=%i, version=%lu, salt=%lu)" %\
+            (self.initiator, self.responder, self.version, self.salt)
diff --git a/test/functional/test_framework/p2p.py b/test/functional/test_framework/p2p.py
index 251d3d5eae47e..b3bef89a8d362 100755
--- a/test/functional/test_framework/p2p.py
+++ b/test/functional/test_framework/p2p.py
@@ -59,6 +59,7 @@
     msg_sendaddrv2,
     msg_sendcmpct,
     msg_sendheaders,
+    msg_sendrecon,
     msg_tx,
     MSG_TX,
     MSG_TYPE_MASK,
@@ -120,6 +121,7 @@
     b"sendaddrv2": msg_sendaddrv2,
     b"sendcmpct": msg_sendcmpct,
     b"sendheaders": msg_sendheaders,
+    b"sendrecon": msg_sendrecon,
     b"tx": msg_tx,
     b"verack": msg_verack,
     b"version": msg_version,
@@ -415,6 +417,7 @@ def on_pong(self, message): pass
     def on_sendaddrv2(self, message): pass
     def on_sendcmpct(self, message): pass
     def on_sendheaders(self, message): pass
+    def on_sendrecon(self, message): pass
     def on_tx(self, message): pass
     def on_wtxidrelay(self, message): pass