#include <qpdf/SHA2_native.hh>

#include <qpdf/QUtil.hh>
#include <cstdio>
#include <stdexcept>

SHA2_native::SHA2_native(int bits) :
    bits(bits)
{
    switch (bits) {
    case 256:
        sph_sha256_init(&ctx256);
        break;
    case 384:
        sph_sha384_init(&ctx384);
        break;
    case 512:
        sph_sha512_init(&ctx512);
        break;
    default:
        badBits();
        break;
    }
}

void
SHA2_native::badBits()
{
    throw std::logic_error("SHA2_native has bits != 256, 384, or 512");
}

void
SHA2_native::update(unsigned char const* buf, size_t len)
{
    switch (bits) {
    case 256:
        sph_sha256(&ctx256, buf, len);
        break;
    case 384:
        sph_sha384(&ctx384, buf, len);
        break;
    case 512:
        sph_sha512(&ctx512, buf, len);
        break;
    default:
        badBits();
        break;
    }
}

void
SHA2_native::finalize()
{
    switch (bits) {
    case 256:
        sph_sha256_close(&ctx256, sha256sum);
        break;
    case 384:
        sph_sha384_close(&ctx384, sha384sum);
        break;
    case 512:
        sph_sha512_close(&ctx512, sha512sum);
        break;
    default:
        badBits();
        break;
    }
}

std::string
SHA2_native::getRawDigest()
{
    std::string result;
    switch (bits) {
    case 256:
        result = std::string(reinterpret_cast<char*>(sha256sum), sizeof(sha256sum));
        break;
    case 384:
        result = std::string(reinterpret_cast<char*>(sha384sum), sizeof(sha384sum));
        break;
    case 512:
        result = std::string(reinterpret_cast<char*>(sha512sum), sizeof(sha512sum));
        break;
    default:
        badBits();
        break;
    }
    return result;
}