// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2024 ARM Limited
 *
 * Common helper functions for SVE and SME functionality.
 */

#include <stdbool.h>
#include <kselftest.h>
#include <asm/sigcontext.h>
#include <sys/prctl.h>

unsigned int vls[SVE_VQ_MAX];
unsigned int nvls;

int sve_fill_vls(bool use_sme, int min_vls)
{
        int vq, vl;
        int pr_set_vl = use_sme ? PR_SME_SET_VL : PR_SVE_SET_VL;
        int len_mask = use_sme ? PR_SME_VL_LEN_MASK : PR_SVE_VL_LEN_MASK;

        /*
         * Enumerate up to SVE_VQ_MAX vector lengths
         */
        for (vq = SVE_VQ_MAX; vq > 0; --vq) {
                vl = prctl(pr_set_vl, vq * 16);
                if (vl == -1)
                        return KSFT_FAIL;

                vl &= len_mask;

                /*
                 * Unlike SVE, SME does not require the minimum vector length
                 * to be implemented, or the VLs to be consecutive, so any call
                 * to the prctl might return the single implemented VL, which
                 * might be larger than 16. So to avoid this loop never
                 * terminating,  bail out here when we find a higher VL than
                 * we asked for.
                 * See the ARM ARM, DDI 0487K.a, B1.4.2: I_QQRNR and I_NWYBP.
                 */
                if (vq < sve_vq_from_vl(vl))
                        break;

                /* Skip missing VLs */
                vq = sve_vq_from_vl(vl);

                vls[nvls++] = vl;
        }

        if (nvls < min_vls) {
                fprintf(stderr, "Only %d VL supported\n", nvls);
                return KSFT_SKIP;
        }

        return KSFT_PASS;
}