/* SPDX-License-Identifier: GPL-2.0-or-later */
/*
 * GHASH, arm optimized
 *
 * Copyright 2026 Google LLC
 */

#include <asm/hwcap.h>
#include <asm/neon.h>
#include <asm/simd.h>

static __ro_after_init DEFINE_STATIC_KEY_FALSE(have_neon);

void pmull_ghash_update_p8(size_t blocks, struct polyval_elem *dg,
                           const u8 *src, const struct polyval_elem *h);

#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_neon) && may_use_simd()) {
                do {
                        /* Allow rescheduling every 4 KiB. */
                        size_t n =
                                min_t(size_t, nblocks, 4096 / GHASH_BLOCK_SIZE);

                        scoped_ksimd()
                                pmull_ghash_update_p8(n, acc, data, &key->h);
                        data += n * GHASH_BLOCK_SIZE;
                        nblocks -= n;
                } while (nblocks);
        } 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 (elf_hwcap & HWCAP_NEON)
                static_branch_enable(&have_neon);
}