// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

#include "util/sha.h"

#include <openssl/sha.h>

#include <string_view>

namespace doris {

constexpr static char dig_vec_lower[] = "0123456789abcdef";

void SHA1Digest::reset(const void* data, size_t length) {
    SHA1_Init(&_sha_ctx);
    SHA1_Update(&_sha_ctx, data, length);
}

std::string_view SHA1Digest::digest() {
    unsigned char buf[SHA_DIGEST_LENGTH];
    SHA1_Final(buf, &_sha_ctx);

    char* to = _reuse_hex;
    for (int i = 0; i < SHA_DIGEST_LENGTH; ++i) {
        *to++ = dig_vec_lower[buf[i] >> 4];
        *to++ = dig_vec_lower[buf[i] & 0x0F];
    }

    return std::string_view {_reuse_hex, _reuse_hex + 2 * SHA_DIGEST_LENGTH};
}

void SHA224Digest::reset(const void* data, size_t length) {
    SHA224_Init(&_sha224_ctx);
    SHA224_Update(&_sha224_ctx, data, length);
}

std::string_view SHA224Digest::digest() {
    unsigned char buf[SHA224_DIGEST_LENGTH];
    SHA224_Final(buf, &_sha224_ctx);

    char* to = _reuse_hex;
    for (int i = 0; i < SHA224_DIGEST_LENGTH; ++i) {
        *to++ = dig_vec_lower[buf[i] >> 4];
        *to++ = dig_vec_lower[buf[i] & 0x0F];
    }

    return std::string_view {_reuse_hex, _reuse_hex + 2 * SHA224_DIGEST_LENGTH};
}

void SHA256Digest::reset(const void* data, size_t length) {
    SHA256_Init(&_sha256_ctx);
    SHA256_Update(&_sha256_ctx, data, length);
}

std::string_view SHA256Digest::digest() {
    unsigned char buf[SHA256_DIGEST_LENGTH];
    SHA256_Final(buf, &_sha256_ctx);

    char* to = _reuse_hex;
    for (int i = 0; i < SHA256_DIGEST_LENGTH; ++i) {
        *to++ = dig_vec_lower[buf[i] >> 4];
        *to++ = dig_vec_lower[buf[i] & 0x0F];
    }

    return std::string_view {_reuse_hex, _reuse_hex + 2 * SHA256_DIGEST_LENGTH};
}

void SHA384Digest::reset(const void* data, size_t length) {
    SHA384_Init(&_sha384_ctx);
    SHA384_Update(&_sha384_ctx, data, length);
}

std::string_view SHA384Digest::digest() {
    unsigned char buf[SHA384_DIGEST_LENGTH];
    SHA384_Final(buf, &_sha384_ctx);

    char* to = _reuse_hex;
    for (int i = 0; i < SHA384_DIGEST_LENGTH; ++i) {
        *to++ = dig_vec_lower[buf[i] >> 4];
        *to++ = dig_vec_lower[buf[i] & 0x0F];
    }

    return std::string_view {_reuse_hex, _reuse_hex + 2 * SHA384_DIGEST_LENGTH};
}

void SHA512Digest::reset(const void* data, size_t length) {
    SHA512_Init(&_sha512_ctx);
    SHA512_Update(&_sha512_ctx, data, length);
}

std::string_view SHA512Digest::digest() {
    unsigned char buf[SHA512_DIGEST_LENGTH];
    SHA512_Final(buf, &_sha512_ctx);

    char* to = _reuse_hex;
    for (int i = 0; i < SHA512_DIGEST_LENGTH; ++i) {
        *to++ = dig_vec_lower[buf[i] >> 4];
        *to++ = dig_vec_lower[buf[i] & 0x0F];
    }

    return std::string_view {_reuse_hex, _reuse_hex + 2 * SHA512_DIGEST_LENGTH};
}

} // namespace doris