serenity/Userland/Libraries/LibCrypto/Hash/HashFunction.h

/*
 * Copyright (c) 2020, Ali Mohammad Pur <mpfard@serenityos.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <AK/ByteBuffer.h>
#include <AK/StringView.h>
#include <AK/Types.h>

namespace Crypto {
namespace Hash {

template<size_t BlockS, typename DigestT>
class HashFunction {
public:
    static constexpr auto BlockSize = BlockS / 8;
    static constexpr auto DigestSize = DigestT::Size;

    using DigestType = DigestT;

    static size_t block_size() { return BlockSize; };
    static size_t digest_size() { return DigestSize; };

    virtual void update(const u8*, size_t) = 0;

    void update(const Bytes& buffer) { update(buffer.data(), buffer.size()); };
    void update(const ReadonlyBytes& buffer) { update(buffer.data(), buffer.size()); };
    void update(const ByteBuffer& buffer) { update(buffer.data(), buffer.size()); };
    void update(const StringView& string) { update((const u8*)string.characters_without_null_termination(), string.length()); };

    virtual DigestType peek() = 0;
    virtual DigestType digest() = 0;

    virtual void reset() = 0;

    virtual String class_name() const = 0;

protected:
    virtual ~HashFunction() = default;
};
}
}