Source code

Revision control

Copy as Markdown

Other Tools

/*
* BLAKE2s
* (C) 2023 Richard Huveneers
*
* Based on the RFC7693 reference implementation
*
* Botan is released under the Simplified BSD License (see license.txt)
*/
#include <botan/internal/blake2s.h>
#include <botan/exceptn.h>
#include <botan/internal/fmt.h>
#include <botan/internal/loadstor.h>
#include <botan/internal/rotate.h>
namespace Botan {
namespace {
// Initialization Vector.
const uint32_t blake2s_iv[8] = {
0x6A09E667, 0xBB67AE85, 0x3C6EF372, 0xA54FF53A, 0x510E527F, 0x9B05688C, 0x1F83D9AB, 0x5BE0CD19};
// Mixing function G.
inline void B2S_G(uint8_t a, uint8_t b, uint8_t c, uint8_t d, uint32_t x, uint32_t y, uint32_t* v) {
v[a] = v[a] + v[b] + x;
v[d] = rotr<16>(v[d] ^ v[a]);
v[c] = v[c] + v[d];
v[b] = rotr<12>(v[b] ^ v[c]);
v[a] = v[a] + v[b] + y;
v[d] = rotr<8>(v[d] ^ v[a]);
v[c] = v[c] + v[d];
v[b] = rotr<7>(v[b] ^ v[c]);
}
} // namespace
std::string BLAKE2s::name() const {
return fmt("BLAKE2s({})", m_outlen << 3);
}
// Secret key (also <= 32 bytes) is optional (keylen = 0).
// (keylen=0: no key)
void BLAKE2s::state_init(size_t outlen, const uint8_t* key, size_t keylen) {
for(size_t i = 0; i < 8; i++) { // state, "param block"
m_h[i] = blake2s_iv[i];
}
m_h[0] ^= 0x01010000 ^ (keylen << 8) ^ outlen;
m_t[0] = 0; // input count low word
m_t[1] = 0; // input count high word
m_c = 0; // pointer within buffer
m_outlen = outlen;
for(size_t i = keylen; i < 64; i++) { // zero input block
m_b[i] = 0;
}
if(keylen > 0) {
add_data(std::span<const uint8_t>(key, keylen));
m_c = 64; // at the end
}
}
// Compression function. "last" flag indicates last block.
void BLAKE2s::compress(bool last) {
const uint8_t sigma[10][16] = {{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15},
{14, 10, 4, 8, 9, 15, 13, 6, 1, 12, 0, 2, 11, 7, 5, 3},
{11, 8, 12, 0, 5, 2, 15, 13, 10, 14, 3, 6, 7, 1, 9, 4},
{7, 9, 3, 1, 13, 12, 11, 14, 2, 6, 5, 10, 4, 0, 15, 8},
{9, 0, 5, 7, 2, 4, 10, 15, 14, 1, 11, 12, 6, 8, 3, 13},
{2, 12, 6, 10, 0, 11, 8, 3, 4, 13, 7, 5, 15, 14, 1, 9},
{12, 5, 1, 15, 14, 13, 4, 10, 0, 7, 6, 3, 9, 2, 8, 11},
{13, 11, 7, 14, 12, 1, 3, 9, 5, 0, 15, 4, 8, 6, 2, 10},
{6, 15, 14, 9, 11, 3, 0, 8, 12, 2, 13, 7, 1, 4, 10, 5},
{10, 2, 8, 4, 7, 6, 1, 5, 15, 11, 9, 14, 3, 12, 13, 0}};
uint32_t v[16], m[16];
for(size_t i = 0; i < 8; i++) { // init work variables
v[i] = m_h[i];
v[i + 8] = blake2s_iv[i];
}
v[12] ^= m_t[0]; // low 32 bits of offset
v[13] ^= m_t[1]; // high 32 bits
if(last) { // last block flag set ?
v[14] = ~v[14];
}
load_le<uint32_t>(m, m_b, 16); // get little-endian words
for(size_t i = 0; i < 10; i++) { // ten rounds
B2S_G(0, 4, 8, 12, m[sigma[i][0]], m[sigma[i][1]], v);
B2S_G(1, 5, 9, 13, m[sigma[i][2]], m[sigma[i][3]], v);
B2S_G(2, 6, 10, 14, m[sigma[i][4]], m[sigma[i][5]], v);
B2S_G(3, 7, 11, 15, m[sigma[i][6]], m[sigma[i][7]], v);
B2S_G(0, 5, 10, 15, m[sigma[i][8]], m[sigma[i][9]], v);
B2S_G(1, 6, 11, 12, m[sigma[i][10]], m[sigma[i][11]], v);
B2S_G(2, 7, 8, 13, m[sigma[i][12]], m[sigma[i][13]], v);
B2S_G(3, 4, 9, 14, m[sigma[i][14]], m[sigma[i][15]], v);
}
for(size_t i = 0; i < 8; ++i) {
m_h[i] ^= v[i] ^ v[i + 8];
}
}
/*
* Clear memory of sensitive data
*/
void BLAKE2s::clear() {
state_init(m_outlen, nullptr, 0);
}
void BLAKE2s::add_data(std::span<const uint8_t> in) {
for(size_t i = 0; i < in.size(); i++) {
if(m_c == 64) { // buffer full ?
m_t[0] += m_c; // add counters
if(m_t[0] < m_c) { // carry overflow ?
m_t[1]++; // high word
}
compress(false); // compress (not last)
m_c = 0; // counter to zero
}
m_b[m_c++] = in[i];
}
}
void BLAKE2s::final_result(std::span<uint8_t> out) {
m_t[0] += m_c; // mark last block offset
if(m_t[0] < m_c) { // carry overflow
m_t[1]++; // high word
}
while(m_c < 64) { // fill up with zeros
m_b[m_c++] = 0;
}
compress(true); // final block flag = 1
// little endian convert and store
copy_out_le(out.first(output_length()), m_h);
clear();
}
std::unique_ptr<HashFunction> BLAKE2s::copy_state() const {
std::unique_ptr<BLAKE2s> h = std::make_unique<BLAKE2s>(m_outlen << 3);
memcpy(h->m_b, m_b, sizeof(m_b));
memcpy(h->m_h, m_h, sizeof(m_h));
memcpy(h->m_t, m_t, sizeof(m_t));
h->m_c = m_c;
return h;
}
/*
* BLAKE2s Constructor
*/
BLAKE2s::BLAKE2s(size_t output_bits) {
if(output_bits == 0 || output_bits > 256 || output_bits % 8 != 0) {
throw Invalid_Argument("Bad output bits size for BLAKE2s");
};
state_init(output_bits >> 3, nullptr, 0);
}
BLAKE2s::~BLAKE2s() {
secure_scrub_memory(m_b, sizeof(m_b));
secure_scrub_memory(m_h, sizeof(m_h));
secure_scrub_memory(m_t, sizeof(m_t));
}
} // namespace Botan