14963aeb3SMark Brown // SPDX-License-Identifier: GPL-2.0
24963aeb3SMark Brown /*
34963aeb3SMark Brown * Copyright (C) 2021 ARM Limited
44963aeb3SMark Brown *
54963aeb3SMark Brown * Verify that the ZA register context in signal frames is set up as
64963aeb3SMark Brown * expected.
74963aeb3SMark Brown */
84963aeb3SMark Brown
9*be158b7eSAndre Przywara #include <kselftest.h>
104963aeb3SMark Brown #include <signal.h>
114963aeb3SMark Brown #include <ucontext.h>
124963aeb3SMark Brown #include <sys/prctl.h>
134963aeb3SMark Brown
144963aeb3SMark Brown #include "test_signals_utils.h"
15*be158b7eSAndre Przywara #include "sve_helpers.h"
164963aeb3SMark Brown #include "testcases.h"
174963aeb3SMark Brown
1831ba6342SMark Brown static union {
1931ba6342SMark Brown ucontext_t uc;
2031ba6342SMark Brown char buf[1024 * 128];
2131ba6342SMark Brown } context;
224963aeb3SMark Brown
sme_get_vls(struct tdescr * td)234963aeb3SMark Brown static bool sme_get_vls(struct tdescr *td)
244963aeb3SMark Brown {
25*be158b7eSAndre Przywara int res = sve_fill_vls(VLS_USE_SME, 1);
264963aeb3SMark Brown
27*be158b7eSAndre Przywara if (!res)
284963aeb3SMark Brown return true;
29*be158b7eSAndre Przywara
30*be158b7eSAndre Przywara if (res == KSFT_SKIP)
31*be158b7eSAndre Przywara td->result = KSFT_SKIP;
32*be158b7eSAndre Przywara
33*be158b7eSAndre Przywara return false;
344963aeb3SMark Brown }
354963aeb3SMark Brown
setup_za_regs(void)364963aeb3SMark Brown static void setup_za_regs(void)
374963aeb3SMark Brown {
384963aeb3SMark Brown /* smstart za; real data is TODO */
394963aeb3SMark Brown asm volatile(".inst 0xd503457f" : : : );
404963aeb3SMark Brown }
414963aeb3SMark Brown
42b95b07d1SMark Brown static char zeros[ZA_SIG_REGS_SIZE(SVE_VQ_MAX)];
43b95b07d1SMark Brown
do_one_sme_vl(struct tdescr * td,siginfo_t * si,ucontext_t * uc,unsigned int vl)444963aeb3SMark Brown static int do_one_sme_vl(struct tdescr *td, siginfo_t *si, ucontext_t *uc,
454963aeb3SMark Brown unsigned int vl)
464963aeb3SMark Brown {
4731ba6342SMark Brown size_t offset;
4831ba6342SMark Brown struct _aarch64_ctx *head = GET_BUF_RESV_HEAD(context);
494963aeb3SMark Brown struct za_context *za;
504963aeb3SMark Brown
514963aeb3SMark Brown fprintf(stderr, "Testing VL %d\n", vl);
524963aeb3SMark Brown
534963aeb3SMark Brown if (prctl(PR_SME_SET_VL, vl) != vl) {
544963aeb3SMark Brown fprintf(stderr, "Failed to set VL\n");
554963aeb3SMark Brown return 1;
564963aeb3SMark Brown }
574963aeb3SMark Brown
584963aeb3SMark Brown /*
594963aeb3SMark Brown * Get a signal context which should have a SVE frame and registers
604963aeb3SMark Brown * in it.
614963aeb3SMark Brown */
624963aeb3SMark Brown setup_za_regs();
6331ba6342SMark Brown if (!get_current_context(td, &context.uc, sizeof(context)))
644963aeb3SMark Brown return 1;
654963aeb3SMark Brown
6631ba6342SMark Brown head = get_header(head, ZA_MAGIC, GET_BUF_RESV_SIZE(context), &offset);
674963aeb3SMark Brown if (!head) {
684963aeb3SMark Brown fprintf(stderr, "No ZA context\n");
694963aeb3SMark Brown return 1;
704963aeb3SMark Brown }
714963aeb3SMark Brown
724963aeb3SMark Brown za = (struct za_context *)head;
734963aeb3SMark Brown if (za->vl != vl) {
744963aeb3SMark Brown fprintf(stderr, "Got VL %d, expected %d\n", za->vl, vl);
754963aeb3SMark Brown return 1;
764963aeb3SMark Brown }
774963aeb3SMark Brown
78b95b07d1SMark Brown if (head->size != ZA_SIG_CONTEXT_SIZE(sve_vq_from_vl(vl))) {
79b95b07d1SMark Brown fprintf(stderr, "ZA context size %u, expected %lu\n",
80b95b07d1SMark Brown head->size, ZA_SIG_CONTEXT_SIZE(sve_vq_from_vl(vl)));
81b95b07d1SMark Brown return 1;
82b95b07d1SMark Brown }
83b95b07d1SMark Brown
844963aeb3SMark Brown fprintf(stderr, "Got expected size %u and VL %d\n",
854963aeb3SMark Brown head->size, za->vl);
864963aeb3SMark Brown
87b95b07d1SMark Brown /* We didn't load any data into ZA so it should be all zeros */
88b95b07d1SMark Brown if (memcmp(zeros, (char *)za + ZA_SIG_REGS_OFFSET,
89b95b07d1SMark Brown ZA_SIG_REGS_SIZE(sve_vq_from_vl(za->vl))) != 0) {
90b95b07d1SMark Brown fprintf(stderr, "ZA data invalid\n");
91b95b07d1SMark Brown return 1;
92b95b07d1SMark Brown }
93b95b07d1SMark Brown
944963aeb3SMark Brown return 0;
954963aeb3SMark Brown }
964963aeb3SMark Brown
sme_regs(struct tdescr * td,siginfo_t * si,ucontext_t * uc)974963aeb3SMark Brown static int sme_regs(struct tdescr *td, siginfo_t *si, ucontext_t *uc)
984963aeb3SMark Brown {
994963aeb3SMark Brown int i;
1004963aeb3SMark Brown
1014963aeb3SMark Brown for (i = 0; i < nvls; i++) {
1024963aeb3SMark Brown if (do_one_sme_vl(td, si, uc, vls[i]))
1034963aeb3SMark Brown return 1;
1044963aeb3SMark Brown }
1054963aeb3SMark Brown
1064963aeb3SMark Brown td->pass = 1;
1074963aeb3SMark Brown
1084963aeb3SMark Brown return 0;
1094963aeb3SMark Brown }
1104963aeb3SMark Brown
1114963aeb3SMark Brown struct tdescr tde = {
1124963aeb3SMark Brown .name = "ZA register",
1134963aeb3SMark Brown .descr = "Check that we get the right ZA registers reported",
1144963aeb3SMark Brown .feats_required = FEAT_SME,
1154963aeb3SMark Brown .timeout = 3,
1164963aeb3SMark Brown .init = sme_get_vls,
1174963aeb3SMark Brown .run = sme_regs,
1184963aeb3SMark Brown };
119