blake2s.cpp

Go to the documentation of this file.

// === AUDIT STATUS ===
// internal:    { status: not started, auditors: [], date: YYYY-MM-DD }
// external_1:  { status: not started, auditors: [], date: YYYY-MM-DD }
// external_2:  { status: not started, auditors: [], date: YYYY-MM-DD }
// =====================
 
#include "blake_util.hpp"
 
#include "barretenberg/stdlib/hash/blake2s/blake2s.hpp"
#include "barretenberg/stdlib/primitives/plookup/plookup.hpp"
 
namespace bb::stdlib {
 
template <typename Builder> void Blake2s<Builder>::increment_counter(blake2s_state& S, const uint32_t inc)
{
    field_ct inc_scalar(static_cast<uint256_t>(inc));
 
    // Note that the initial blake2s_state values are circuit constants.
    S.t[0] = S.t[0] + inc_scalar;
 
    // Note that although blake2s_state is a circuit constant, we use designated functions such as
    // `ranged_less_than` to enforce constraints as appropriate.
    bool_ct to_inc = S.t[0].template ranged_less_than<32>(inc_scalar);
    S.t[1] = S.t[1] + field_ct(to_inc);
}
 
template <typename Builder> void Blake2s<Builder>::compress(blake2s_state& S, byte_array_ct const& in)
{
    using plookup::ColumnIdx;
    using namespace blake_util;
    field_ct m[BLAKE2S_STATE_SIZE];
    field_ct v[BLAKE2S_STATE_SIZE];
 
    for (size_t i = 0; i < BLAKE2S_STATE_SIZE; ++i) {
        m[i] = static_cast<field_ct>(in.slice(i * 4, 4).reverse());
    }
 
    for (size_t i = 0; i < 8; ++i) {
        v[i] = S.h[i];
    }
 
    v[8] = field_ct(uint256_t(blake2s_IV[0]));
    v[9] = field_ct(uint256_t(blake2s_IV[1]));
    v[10] = field_ct(uint256_t(blake2s_IV[2]));
    v[11] = field_ct(uint256_t(blake2s_IV[3]));
 
    // Use the lookup tables to perform XORs
    const auto lookup_1 =
        plookup_read<Builder>::get_lookup_accumulators(BLAKE_XOR, S.t[0], field_ct(uint256_t(blake2s_IV[4])), true);
    v[12] = lookup_1[ColumnIdx::C3][0];
    const auto lookup_2 =
        plookup_read<Builder>::get_lookup_accumulators(BLAKE_XOR, S.t[1], field_ct(uint256_t(blake2s_IV[5])), true);
    v[13] = lookup_2[ColumnIdx::C3][0];
    const auto lookup_3 =
        plookup_read<Builder>::get_lookup_accumulators(BLAKE_XOR, S.f[0], field_ct(uint256_t(blake2s_IV[6])), true);
    v[14] = lookup_3[ColumnIdx::C3][0];
    const auto lookup_4 =
        plookup_read<Builder>::get_lookup_accumulators(BLAKE_XOR, S.f[1], field_ct(uint256_t(blake2s_IV[7])), true);
    v[15] = lookup_4[ColumnIdx::C3][0];
 
    for (size_t idx = 0; idx < 10; idx++) {
        blake_util::round_fn(v, m, idx);
    }
 
    // At this point in the algorithm, the elements (v0, v1, v2, v3) and (v8, v9, v10, v11) in the state matrix 'v' can
    // be 'overflowed' i.e. contain values > 2^{32}. However we do NOT need to normalize them to be < 2^{32}, the
    // following `read_sequence_from_table` calls correctly constrain the output to be 32-bits
    for (size_t i = 0; i < 8; ++i) {
        const auto lookup_a = plookup_read<Builder>::get_lookup_accumulators(BLAKE_XOR, S.h[i], v[i], true);
        const auto lookup_b =
            plookup_read<Builder>::get_lookup_accumulators(BLAKE_XOR, lookup_a[ColumnIdx::C3][0], v[i + 8], true);
        S.h[i] = lookup_b[ColumnIdx::C3][0];
    }
}
 
template <typename Builder> void Blake2s<Builder>::blake2s(blake2s_state& S, byte_array_ct const& in)
{
    using plookup::ColumnIdx;
    using namespace blake_util;
 
    size_t offset = 0;
    size_t size = in.size();
 
    while (size > BLAKE2S_BLOCKBYTES) {
        increment_counter(S, BLAKE2S_BLOCKBYTES);
        compress(S, in.slice(offset, BLAKE2S_BLOCKBYTES));
        offset += BLAKE2S_BLOCKBYTES;
        size -= BLAKE2S_BLOCKBYTES;
    }
 
    // Set last block.
    S.f[0] = field_t<Builder>(uint256_t((uint32_t)-1));
 
    // Build final block: remaining input + constant padding
    Builder* ctx = in.get_context();
    auto remaining = in.slice(offset);
 
    // Combine remaining bytes and constant padding (no constraints needed for constants)
    byte_array_ct final = remaining; // Copy constrained remaining bytes
    byte_array_ct padding = byte_array_ct::constant_padding(ctx, BLAKE2S_BLOCKBYTES - size);
    final.write(padding);
 
    increment_counter(S, static_cast<uint32_t>(size));
    compress(S, final);
}
 
template <typename Builder> byte_array<Builder> Blake2s<Builder>::hash(const byte_array_ct& input)
{
    blake2s_state S;
 
    for (size_t i = 0; i < 8; i++) {
        S.h[i] = field_ct(uint256_t(initial_H[i]));
    }
 
    blake2s(S, input);
 
    // Build result from state values
    byte_array_ct result = byte_array_ct::constant_padding(input.get_context(), 0);
    for (const auto& h : S.h) {
        // byte_array_ct(field, num_bytes) constructor adds range constraints for each byte
        byte_array_ct v(h, 4);
        auto reversed = v.reverse();
        result.write(reversed);
    }
    return result;
}
 
template class Blake2s<UltraCircuitBuilder>;
template class Blake2s<MegaCircuitBuilder>;
 
} // namespace bb::stdlib