Kylin/Encrypt/SecureHashAlgorithm.cpp
2024-10-29 13:51:37 +00:00

40 lines
1.3 KiB
C++

#include "SecureHashAlgorithm.h"
#include <mbedtls/sha256.h>
#include <iomanip>
#include <sstream>
class SecureHashAlgorithmPrivate {
public:
mbedtls_sha256_context context;
};
SecureHashAlgorithm::SecureHashAlgorithm() : m_d(new SecureHashAlgorithmPrivate()) {
mbedtls_sha256_init(&m_d->context);
mbedtls_sha256_starts(&m_d->context, 0);
}
SecureHashAlgorithm::~SecureHashAlgorithm() {
mbedtls_sha256_free(&m_d->context);
delete m_d;
}
int SecureHashAlgorithm::update(const unsigned char *input, size_t ilen) {
return mbedtls_sha256_update(&m_d->context, input, ilen);
}
int SecureHashAlgorithm::finish(unsigned char *output) {
return mbedtls_sha256_finish(&m_d->context, output);
}
int SecureHashAlgorithm::sha256(const unsigned char *input, size_t ilen, unsigned char *output) {
return mbedtls_sha256(input, ilen, output, 0);
}
std::string SecureHashAlgorithm::sha256(const std::string_view &input, Sha256DigestType &digest) {
sha256(reinterpret_cast<const uint8_t *>(input.data()), input.size(), digest);
std::ostringstream oss;
for (int i = 0; i < sizeof(Sha256DigestType); ++i)
oss << std::hex << std::setfill('0') << std::setw(2) << (((uint16_t)digest[i]) & 0xFF);
return oss.str();
}