/* SPDX-License-Identifier: GPL-2.0-only */
/*
 * GHASH, RISC-V optimized
 *
 * Copyright (C) 2023 VRULL GmbH
 * Copyright (C) 2023 SiFive, Inc.
 * Copyright 2026 Google LLC
 */

#include <asm/simd.h>
#include <asm/vector.h>

static __ro_after_init DEFINE_STATIC_KEY_FALSE(have_zvkg);

asmlinkage void ghash_zvkg(u8 accumulator[GHASH_BLOCK_SIZE],
                           const u8 key[GHASH_BLOCK_SIZE],
                           const u8 *data, size_t nblocks);

#define ghash_preparekey_arch ghash_preparekey_arch
static void ghash_preparekey_arch(struct ghash_key *key,
                                  const u8 raw_key[GHASH_BLOCK_SIZE])
{
        /* Save key in POLYVAL format for fallback */
        ghash_key_to_polyval(raw_key, &key->h);

        /* Save key in GHASH format for zvkg */
        memcpy(key->h_raw, raw_key, GHASH_BLOCK_SIZE);
}

#define ghash_blocks_arch ghash_blocks_arch
static void ghash_blocks_arch(struct polyval_elem *acc,
                              const struct ghash_key *key,
                              const u8 *data, size_t nblocks)
{
        if (static_branch_likely(&have_zvkg) && likely(may_use_simd())) {
                u8 ghash_acc[GHASH_BLOCK_SIZE];

                polyval_acc_to_ghash(acc, ghash_acc);

                kernel_vector_begin();
                ghash_zvkg(ghash_acc, key->h_raw, data, nblocks);
                kernel_vector_end();

                ghash_acc_to_polyval(ghash_acc, acc);
                memzero_explicit(ghash_acc, sizeof(ghash_acc));
        } else {
                ghash_blocks_generic(acc, &key->h, data, nblocks);
        }
}

#define gf128hash_mod_init_arch gf128hash_mod_init_arch
static void gf128hash_mod_init_arch(void)
{
        if (riscv_isa_extension_available(NULL, ZVKG) &&
            riscv_vector_vlen() >= 128)
                static_branch_enable(&have_zvkg);
}