/*
 * 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/core_names.h>
#include <openssl/proverr.h>
#include <openssl/err.h>
#include "prov/blake2.h"
#include "prov/digestcommon.h"
#include "prov/implementations.h"

#define IMPLEMENT_BLAKE_functions(variant, VARIANT, variantsize) \
static const OSSL_PARAM known_blake##variant##_ctx_params[] = { \
    {OSSL_DIGEST_PARAM_SIZE, OSSL_PARAM_UNSIGNED_INTEGER, NULL, 0, 0}, \
    OSSL_PARAM_END \
}; \
 \
const OSSL_PARAM *ossl_blake##variant##_gettable_ctx_params(ossl_unused void *ctx, \
                                                   ossl_unused void *pctx) \
{ \
    return known_blake##variant##_ctx_params; \
} \
 \
const OSSL_PARAM *ossl_blake##variant##_settable_ctx_params(ossl_unused void *ctx, \
                                                   ossl_unused void *pctx) \
{ \
    return known_blake##variant##_ctx_params; \
} \
 \
int ossl_blake##variant##_get_ctx_params(void *vctx, OSSL_PARAM params[]) \
{ \
    struct blake##variant##_md_data_st *mdctx = vctx; \
    OSSL_PARAM *p; \
 \
    BLAKE##VARIANT##_CTX *ctx = &mdctx->ctx; \
 \
    if (ctx == NULL) \
        return 0; \
    if (params == NULL) \
        return 1; \
 \
    p = OSSL_PARAM_locate(params, OSSL_DIGEST_PARAM_SIZE); \
    if (p != NULL \
        && !OSSL_PARAM_set_uint(p, (unsigned int)mdctx->params.digest_length)) { \
        ERR_raise(ERR_LIB_PROV, PROV_R_FAILED_TO_SET_PARAMETER); \
        return 0; \
    } \
 \
    return 1; \
} \
 \
int ossl_blake##variant##_set_ctx_params(void *vctx, const OSSL_PARAM params[]) \
{ \
    size_t size; \
    struct blake##variant##_md_data_st *mdctx = vctx; \
    const OSSL_PARAM *p; \
 \
    BLAKE##VARIANT##_CTX *ctx = &mdctx->ctx; \
 \
    if (ctx == NULL) \
        return 0; \
    if (params == NULL) \
        return 1; \
 \
    p = OSSL_PARAM_locate_const(params, OSSL_DIGEST_PARAM_SIZE); \
    if (p != NULL) { \
        if (!OSSL_PARAM_get_size_t(p, &size)) { \
            ERR_raise(ERR_LIB_PROV, PROV_R_FAILED_TO_GET_PARAMETER); \
            return 0; \
        } \
        if (size < 1 || size > BLAKE##VARIANT##_OUTBYTES) { \
            ERR_raise(ERR_LIB_PROV, PROV_R_INVALID_DIGEST_SIZE); \
            return 0; \
        } \
        ossl_blake##variant##_param_set_digest_length(&mdctx->params, (uint8_t)size); \
    } \
 \
    return 1; \
} \
 \
static int ossl_blake##variantsize##_init(void *ctx) \
{ \
    struct blake##variant##_md_data_st *mdctx = ctx; \
    uint8_t digest_length = mdctx->params.digest_length; \
 \
    ossl_blake##variant##_param_init(&mdctx->params); \
    if (digest_length != 0) \
        mdctx->params.digest_length = digest_length; \
    return ossl_blake##variant##_init(&mdctx->ctx, &mdctx->params); \
} \
 \
static OSSL_FUNC_digest_init_fn blake##variantsize##_internal_init; \
static OSSL_FUNC_digest_newctx_fn blake##variantsize##_newctx; \
static OSSL_FUNC_digest_freectx_fn blake##variantsize##_freectx; \
static OSSL_FUNC_digest_dupctx_fn blake##variantsize##_dupctx; \
static OSSL_FUNC_digest_final_fn blake##variantsize##_internal_final; \
static OSSL_FUNC_digest_get_params_fn blake##variantsize##_get_params; \
 \
static int blake##variantsize##_internal_init(void *ctx, const OSSL_PARAM params[]) \
{ \
    return ossl_prov_is_running() && ossl_blake##variant##_set_ctx_params(ctx, params) \
        && ossl_blake##variantsize##_init(ctx); \
} \
 \
static void *blake##variantsize##_newctx(void *prov_ctx) \
{ \
    struct blake##variant##_md_data_st *ctx; \
 \
    ctx = ossl_prov_is_running() ? OPENSSL_zalloc(sizeof(*ctx)) : NULL; \
    return ctx; \
} \
 \
static void blake##variantsize##_freectx(void *vctx) \
{ \
    struct blake##variant##_md_data_st *ctx; \
 \
    ctx = (struct blake##variant##_md_data_st *)vctx; \
    OPENSSL_clear_free(ctx, sizeof(*ctx)); \
} \
 \
static void *blake##variantsize##_dupctx(void *ctx) \
{ \
    struct blake##variant##_md_data_st *in, *ret; \
 \
    in = (struct blake##variant##_md_data_st *)ctx; \
    ret = ossl_prov_is_running()? OPENSSL_malloc(sizeof(*ret)) : NULL; \
    if (ret != NULL) \
        *ret = *in; \
    return ret; \
} \
 \
static int blake##variantsize##_internal_final(void *ctx, unsigned char *out, \
                                     size_t *outl, size_t outsz) \
{ \
    struct blake##variant##_md_data_st *b_ctx; \
 \
    b_ctx = (struct blake##variant##_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_blake##variant##_final(out, ctx); \
} \
 \
static int blake##variantsize##_get_params(OSSL_PARAM params[]) \
{ \
    return ossl_digest_default_get_params(params, BLAKE##VARIANT##_BLOCKBYTES, BLAKE##VARIANT##_OUTBYTES, 0); \
} \
 \
const OSSL_DISPATCH ossl_blake##variantsize##_functions[] = { \
    {OSSL_FUNC_DIGEST_NEWCTX, (void (*)(void))blake##variantsize##_newctx}, \
    {OSSL_FUNC_DIGEST_UPDATE, (void (*)(void))ossl_blake##variant##_update}, \
    {OSSL_FUNC_DIGEST_FINAL, (void (*)(void))blake##variantsize##_internal_final}, \
    {OSSL_FUNC_DIGEST_FREECTX, (void (*)(void))blake##variantsize##_freectx}, \
    {OSSL_FUNC_DIGEST_DUPCTX, (void (*)(void))blake##variantsize##_dupctx}, \
    {OSSL_FUNC_DIGEST_GET_PARAMS, (void (*)(void))blake##variantsize##_get_params}, \
    {OSSL_FUNC_DIGEST_GETTABLE_PARAMS, \
     (void (*)(void))ossl_digest_default_gettable_params}, \
    {OSSL_FUNC_DIGEST_INIT, (void (*)(void))blake##variantsize##_internal_init}, \
    {OSSL_FUNC_DIGEST_GETTABLE_CTX_PARAMS, \
     (void (*)(void))ossl_blake##variant##_gettable_ctx_params}, \
    {OSSL_FUNC_DIGEST_SETTABLE_CTX_PARAMS, \
     (void (*)(void))ossl_blake##variant##_settable_ctx_params}, \
    {OSSL_FUNC_DIGEST_GET_CTX_PARAMS, \
     (void (*)(void))ossl_blake##variant##_get_ctx_params}, \
    {OSSL_FUNC_DIGEST_SET_CTX_PARAMS, \
     (void (*)(void))ossl_blake##variant##_set_ctx_params}, \
    {0, NULL} \
};

IMPLEMENT_BLAKE_functions(2s, 2S, 2s256)
IMPLEMENT_BLAKE_functions(2b, 2B, 2b512)