mirror of
https://github.com/bitcoin/bitcoin.git
synced 2026-01-19 23:03:45 +01:00
crypto: add Poly1305 class with std::byte Span interface
This commit is contained in:
@@ -191,6 +191,22 @@ static void TestPoly1305(const std::string &hexmessage, const std::string &hexke
|
||||
tagres.resize(POLY1305_TAGLEN);
|
||||
poly1305_auth(tagres.data(), m.data(), m.size(), key.data());
|
||||
BOOST_CHECK(tag == tagres);
|
||||
|
||||
// Test incremental interface
|
||||
for (int splits = 0; splits < 10; ++splits) {
|
||||
for (int iter = 0; iter < 10; ++iter) {
|
||||
auto data = MakeByteSpan(m);
|
||||
Poly1305 poly1305{MakeByteSpan(key)};
|
||||
for (int chunk = 0; chunk < splits; ++chunk) {
|
||||
size_t now = InsecureRandRange(data.size() + 1);
|
||||
poly1305.Update(data.first(now));
|
||||
data = data.subspan(now);
|
||||
}
|
||||
tagres.assign(POLY1305_TAGLEN, 0);
|
||||
poly1305.Update(data).Finalize(MakeWritableByteSpan(tagres));
|
||||
BOOST_CHECK(tag == tagres);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void TestHKDF_SHA256_32(const std::string &ikm_hex, const std::string &salt_hex, const std::string &info_hex, const std::string &okm_check_hex) {
|
||||
|
||||
@@ -20,3 +20,34 @@ FUZZ_TARGET(crypto_poly1305)
|
||||
std::vector<uint8_t> tag_out(POLY1305_TAGLEN);
|
||||
poly1305_auth(tag_out.data(), in.data(), in.size(), key.data());
|
||||
}
|
||||
|
||||
|
||||
FUZZ_TARGET(crypto_poly1305_split)
|
||||
{
|
||||
FuzzedDataProvider provider{buffer.data(), buffer.size()};
|
||||
|
||||
// Read key and instantiate two Poly1305 objects with it.
|
||||
auto key = provider.ConsumeBytes<std::byte>(Poly1305::KEYLEN);
|
||||
key.resize(Poly1305::KEYLEN);
|
||||
Poly1305 poly_full{key}, poly_split{key};
|
||||
|
||||
// Vector that holds all bytes processed so far.
|
||||
std::vector<std::byte> total_input;
|
||||
|
||||
// Process input in pieces.
|
||||
LIMITED_WHILE(provider.remaining_bytes(), 100) {
|
||||
auto in = provider.ConsumeRandomLengthString();
|
||||
poly_split.Update(MakeByteSpan(in));
|
||||
// Update total_input to match what was processed.
|
||||
total_input.insert(total_input.end(), MakeByteSpan(in).begin(), MakeByteSpan(in).end());
|
||||
}
|
||||
|
||||
// Process entire input at once.
|
||||
poly_full.Update(total_input);
|
||||
|
||||
// Verify both agree.
|
||||
std::array<std::byte, Poly1305::TAGLEN> tag_split, tag_full;
|
||||
poly_split.Finalize(tag_split);
|
||||
poly_full.Finalize(tag_full);
|
||||
assert(tag_full == tag_split);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user