Fuzz tests for CPubKey<->EllSwift

dhruv · Jul 20, 2022 · a58c16b · a58c16b
1 parent 5072b2c
commit a58c16b
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/test/fuzz/key.cpp b/src/test/fuzz/key.cpp
@@ -16,9 +16,11 @@
 #include <script/signingprovider.h>
 #include <script/standard.h>
 #include <streams.h>
+#include <test/fuzz/FuzzedDataProvider.h>
 #include <test/fuzz/fuzz.h>
 #include <util/strencodings.h>
 
+#include <array>
 #include <cassert>
 #include <cstdint>
 #include <numeric>
@@ -306,3 +308,21 @@ FUZZ_TARGET_INIT(key, initialize_key)
         }
     }
 }
+
+FUZZ_TARGET_INIT(ellswift, initialize_key)
+{
+    FuzzedDataProvider fuzzed_data_provider{buffer.data(), buffer.size()};
+    auto pubkey_bytes = fuzzed_data_provider.ConsumeBytes<uint8_t>(CPubKey::COMPRESSED_SIZE);
+    pubkey_bytes.resize(CPubKey::COMPRESSED_SIZE);
+    CPubKey pubkey(pubkey_bytes.begin(), pubkey_bytes.end());
+
+    if (!pubkey.IsFullyValid()) {
+        return;
+    }
+
+    auto rnd32 = fuzzed_data_provider.ConsumeBytes<uint8_t>(32);
+    rnd32.resize(32);
+    std::array<uint8_t, 32> rnd32_array;
+    std::copy(rnd32.begin(), rnd32.end(), rnd32_array.begin());
+    (void)pubkey.EllSwiftEncode(rnd32_array);
+}