/*-
 * Copyright (c) 2024 Robert Clausecker <fuz@FreeBSD.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <sys/types.h>
#include <sys/md5.h>

#include <machine/cpufunc.h>
#include <machine/specialreg.h>
#include <stdint.h>
#include <string.h>
#include <x86/ifunc.h>

extern void _libmd_md5block_baseline(MD5_CTX *, const void *, size_t);
extern void _libmd_md5block_bmi1(MD5_CTX *, const void *, size_t);
extern void _libmd_md5block_avx512(MD5_CTX *, const void *, size_t);

DEFINE_UIFUNC(, void, _libmd_md5block, (MD5_CTX *, const void *, size_t))
{
        if ((cpu_stdext_feature & (CPUID_STDEXT_AVX512F | CPUID_STDEXT_AVX512VL))
            == (CPUID_STDEXT_AVX512F | CPUID_STDEXT_AVX512VL)) {
                u_int regs[4];
                char cpu_vendor[12];

                do_cpuid(0, regs);
                ((u_int *)&cpu_vendor)[0] = regs[1];
                ((u_int *)&cpu_vendor)[1] = regs[3];
                ((u_int *)&cpu_vendor)[2] = regs[2];

                /* the AVX-512 kernel performs poorly on AMD */
                if (memcmp(cpu_vendor, AMD_VENDOR_ID, sizeof(cpu_vendor)) != 0)
                        return (_libmd_md5block_avx512);
        }

        if (cpu_stdext_feature & CPUID_STDEXT_BMI1)
                return (_libmd_md5block_bmi1);
        else
                return (_libmd_md5block_baseline);
}