Fuzz test for BIP324 key derivation

dhruv · Apr 13, 2022 · 98038b5 · 98038b5
1 parent 90e9f37
commit 98038b5
Showing 1 changed file with 33 additions and 0 deletions.
diff --git 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));
+}