/*
 * Copyright 2019-2023 The OpenSSL Project Authors. All Rights Reserved.
 *
 * Licensed under the Apache License 2.0 (the "License").  You may not use
 * this file except in compliance with the License.  You can obtain a copy
 * in the file LICENSE in the source distribution or at
 * https://www.openssl.org/source/license.html
 */

#include <openssl/crypto.h>
#include <openssl/proverr.h>
#include "prov/blake2.h"
#include "prov/digestcommon.h"
#include "prov/implementations.h"

static int ossl_blake2s256_init(void *ctx)
{
    BLAKE2S_PARAM P;

    ossl_blake2s_param_init(&P);
    return ossl_blake2s_init((BLAKE2S_CTX *)ctx, &P);
}

static int ossl_blake2b512_init(void *ctx)
{
    struct blake2b_md_data_st *mdctx = ctx;
    uint8_t digest_length = mdctx->params.digest_length;

    ossl_blake2b_param_init(&mdctx->params);
    if (digest_length != 0)
        mdctx->params.digest_length = digest_length;
    return ossl_blake2b_init(&mdctx->ctx, &mdctx->params);
}

/* ossl_blake2s256_functions */
IMPLEMENT_digest_functions(blake2s256, BLAKE2S_CTX,
                           BLAKE2S_BLOCKBYTES, BLAKE2S_DIGEST_LENGTH, 0,
                           ossl_blake2s256_init, ossl_blake2s_update,
                           ossl_blake2s_final)

/* ossl_blake2b512_functions */

static OSSL_FUNC_digest_init_fn blake2b512_internal_init;
static OSSL_FUNC_digest_newctx_fn blake2b512_newctx;
static OSSL_FUNC_digest_freectx_fn blake2b512_freectx;
static OSSL_FUNC_digest_dupctx_fn blake2b512_dupctx;
static OSSL_FUNC_digest_final_fn blake2b512_internal_final;
static OSSL_FUNC_digest_get_params_fn blake2b512_get_params;

static int blake2b512_internal_init(void *ctx, const OSSL_PARAM params[])
{
    return ossl_prov_is_running() && ossl_blake2b_set_ctx_params(ctx, params)
        && ossl_blake2b512_init(ctx);
}

static void *blake2b512_newctx(void *prov_ctx)
{
    struct blake2b_md_data_st *ctx;

    ctx = ossl_prov_is_running() ? OPENSSL_zalloc(sizeof(*ctx)) : NULL;
    return ctx;
}

static void blake2b512_freectx(void *vctx)
{
    struct blake2b_md_data_st *ctx;

    ctx = (struct blake2b_md_data_st *)vctx;
    OPENSSL_clear_free(ctx, sizeof(*ctx));
}

static void *blake2b512_dupctx(void *ctx)
{
    struct blake2b_md_data_st *in, *ret;

    in = (struct blake2b_md_data_st *)ctx;
    ret = ossl_prov_is_running()? OPENSSL_malloc(sizeof(*ret)) : NULL;
    if (ret != NULL)
        *ret = *in;
    return ret;
}

static int blake2b512_internal_final(void *ctx, unsigned char *out,
                                     size_t *outl, size_t outsz)
{
    struct blake2b_md_data_st *b_ctx;

    b_ctx = (struct blake2b_md_data_st *)ctx;

    if (!ossl_prov_is_running())
        return 0;

    *outl = b_ctx->ctx.outlen;

    if (outsz == 0)
       return 1;

    if (outsz < *outl) {
        ERR_raise(ERR_LIB_PROV, PROV_R_INVALID_DIGEST_SIZE);
        return 0;
    }

    return ossl_blake2b_final(out, ctx);
}

static int blake2b512_get_params(OSSL_PARAM params[])
{
    return ossl_digest_default_get_params(params, BLAKE2B_BLOCKBYTES, 64, 0);
}

const OSSL_DISPATCH ossl_blake2b512_functions[] = {
    {OSSL_FUNC_DIGEST_NEWCTX, (void (*)(void))blake2b512_newctx},
    {OSSL_FUNC_DIGEST_UPDATE, (void (*)(void))ossl_blake2b_update},
    {OSSL_FUNC_DIGEST_FINAL, (void (*)(void))blake2b512_internal_final},
    {OSSL_FUNC_DIGEST_FREECTX, (void (*)(void))blake2b512_freectx},
    {OSSL_FUNC_DIGEST_DUPCTX, (void (*)(void))blake2b512_dupctx},
    {OSSL_FUNC_DIGEST_GET_PARAMS, (void (*)(void))blake2b512_get_params},
    {OSSL_FUNC_DIGEST_GETTABLE_PARAMS,
     (void (*)(void))ossl_digest_default_gettable_params},
    {OSSL_FUNC_DIGEST_INIT, (void (*)(void))blake2b512_internal_init},
    {OSSL_FUNC_DIGEST_GETTABLE_CTX_PARAMS,
     (void (*)(void))ossl_blake2b_gettable_ctx_params},
    {OSSL_FUNC_DIGEST_SETTABLE_CTX_PARAMS,
     (void (*)(void))ossl_blake2b_settable_ctx_params},
    {OSSL_FUNC_DIGEST_GET_CTX_PARAMS,
     (void (*)(void))ossl_blake2b_get_ctx_params},
    {OSSL_FUNC_DIGEST_SET_CTX_PARAMS,
     (void (*)(void))ossl_blake2b_set_ctx_params},
    {0, NULL}
};