88 lines
3.5 KiB
C++
88 lines
3.5 KiB
C++
/*
|
|
* Copyright (C) 2019 ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* This file is part of Scylla.
|
|
*
|
|
* Scylla is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Scylla is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with Scylla. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "hashers.hh"
|
|
|
|
#define CRYPTOPP_ENABLE_NAMESPACE_WEAK 1
|
|
#include <cryptopp/md5.h>
|
|
#include <cryptopp/sha.h>
|
|
|
|
template <typename T> struct hasher_traits;
|
|
template <> struct hasher_traits<md5_hasher> { using impl_type = CryptoPP::Weak::MD5; };
|
|
template <> struct hasher_traits<sha256_hasher> { using impl_type = CryptoPP::SHA256; };
|
|
|
|
template <typename T, size_t size> struct cryptopp_hasher<T, size>::impl {
|
|
using impl_type = typename hasher_traits<T>::impl_type;
|
|
|
|
impl_type hash{};
|
|
|
|
void update(const char* ptr, size_t length) {
|
|
using namespace CryptoPP;
|
|
static_assert(sizeof(char) == sizeof(byte), "Assuming lengths will be the same");
|
|
hash.Update(reinterpret_cast<const byte*>(ptr), length * sizeof(byte));
|
|
}
|
|
|
|
bytes finalize() {
|
|
bytes digest{bytes::initialized_later(), size};
|
|
hash.Final(reinterpret_cast<unsigned char*>(digest.begin()));
|
|
return digest;
|
|
}
|
|
|
|
std::array<uint8_t, size> finalize_array() {
|
|
std::array<uint8_t, size> array;
|
|
hash.Final(reinterpret_cast<unsigned char*>(array.data()));
|
|
return array;
|
|
}
|
|
};
|
|
|
|
template <typename T, size_t size> cryptopp_hasher<T, size>::cryptopp_hasher() : _impl(std::make_unique<impl>()) {}
|
|
|
|
template <typename T, size_t size> cryptopp_hasher<T, size>::~cryptopp_hasher() = default;
|
|
|
|
template <typename T, size_t size> cryptopp_hasher<T, size>::cryptopp_hasher(cryptopp_hasher&& o) noexcept = default;
|
|
|
|
template <typename T, size_t size> cryptopp_hasher<T, size>::cryptopp_hasher(const cryptopp_hasher& o) : _impl(std::make_unique<cryptopp_hasher<T, size>::impl>(*o._impl)) {}
|
|
|
|
template <typename T, size_t size> cryptopp_hasher<T, size>& cryptopp_hasher<T, size>::operator=(cryptopp_hasher&& o) noexcept = default;
|
|
|
|
template <typename T, size_t size> cryptopp_hasher<T, size>& cryptopp_hasher<T, size>::operator=(const cryptopp_hasher& o) {
|
|
_impl = std::make_unique<cryptopp_hasher<T, size>::impl>(*o._impl);
|
|
return *this;
|
|
}
|
|
|
|
template <typename T, size_t size> bytes cryptopp_hasher<T, size>::finalize() { return _impl->finalize(); }
|
|
|
|
template <typename T, size_t size> std::array<uint8_t, size> cryptopp_hasher<T, size>::finalize_array() {
|
|
return _impl->finalize_array();
|
|
}
|
|
|
|
template <typename T, size_t size> void cryptopp_hasher<T, size>::update(const char* ptr, size_t length) { _impl->update(ptr, length); }
|
|
|
|
template <typename T, size_t size> bytes cryptopp_hasher<T, size>::calculate(const std::string_view& s) {
|
|
typename cryptopp_hasher<T, size>::impl::impl_type hash;
|
|
unsigned char digest[size];
|
|
hash.CalculateDigest(digest, reinterpret_cast<const unsigned char*>(s.data()), s.size());
|
|
return std::move(bytes{reinterpret_cast<const int8_t*>(digest), size});
|
|
}
|
|
|
|
template class cryptopp_hasher<md5_hasher, 16>;
|
|
template class cryptopp_hasher<sha256_hasher, 32>;
|