From 2a6ce59276b1778a3e8ffd79d4cd345a2321ac0a Mon Sep 17 00:00:00 2001
From: dhruv <856960+dhruv@users.noreply.github.com>
Date: Thu, 18 Nov 2021 15:28:52 -0800
Subject: [PATCH] Fuzz test for BIP324 key derivation

---
 src/test/fuzz/net.cpp | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/src/test/fuzz/net.cpp b/src/test/fuzz/net.cpp
index 4981287152..09dac0f17f 100644
--- a/src/test/fuzz/net.cpp
+++ b/src/test/fuzz/net.cpp
@@ -4,6 +4,7 @@
 
 #include <chainparams.h>
 #include <chainparamsbase.h>
+#include <key.h>
 #include <net.h>
 #include <net_permissions.h>
 #include <netaddress.h>
@@ -77,3 +78,35 @@ FUZZ_TARGET_INIT(net, initialize_net)
     (void)node.HasPermission(net_permission_flags);
     (void)node.ConnectedThroughNetwork();
 }
+
+void initialize_chainparams()
+{
+    SelectParams(CBaseChainParams::REGTEST);
+}
+
+FUZZ_TARGET_INIT(bip324, initialize_chainparams)
+{
+    FuzzedDataProvider fuzzed_data_provider{buffer.data(), buffer.size()};
+
+    ECDHSecret ecdh_secret;
+    ecdh_secret.resize(ECDH_SECRET_SIZE);
+    auto ecdh_secret_bytes = fuzzed_data_provider.ConsumeBytes<uint8_t>(ECDH_SECRET_SIZE);
+    ecdh_secret_bytes.resize(ECDH_SECRET_SIZE);
+
+    memcpy(ecdh_secret.data(), ecdh_secret_bytes.data(), ECDH_SECRET_SIZE);
+
+    auto initiator_hdata_len = fuzzed_data_provider.ConsumeIntegralInRange(0, 4096);
+    auto initiator_hdata = fuzzed_data_provider.ConsumeBytes<uint8_t>(initiator_hdata_len);
+
+    auto responder_hdata_len = fuzzed_data_provider.ConsumeIntegralInRange(0, 4096);
+    auto responder_hdata = fuzzed_data_provider.ConsumeBytes<uint8_t>(responder_hdata_len);
+
+    BIP324Keys keys;
+    assert(DeriveBIP324Keys(std::move(ecdh_secret), initiator_hdata, responder_hdata, keys));
+    assert(keys.initiator_F.size() == BIP324_KEY_LEN);
+    assert(keys.initiator_V.size() == BIP324_KEY_LEN);
+    assert(keys.responder_F.size() == BIP324_KEY_LEN);
+    assert(keys.responder_V.size() == BIP324_KEY_LEN);
+    assert(keys.session_id.size() == BIP324_KEY_LEN);
+    assert("0000000000000000000000000000000000000000000000000000000000000000" == HexStr(ecdh_secret));
+}