Quelle  zt_regs.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2021 ARM Limited
 *
 * Verify that using an instruction not supported in streaming mode
 * traps when in streaming mode.
 */

#include <signal.h>
#include <ucontext.h>
#include <sys/prctl.h>

#include "test_signals_utils.h"
#include "testcases.h"

static union {
 ucontext_t uc;
 char buf[1024 * 128];
} context;

static void enable_za(void)
{
 /* smstart za; real data is TODO */
 asm volatile(".inst 0xd503457f" : : : );
}

int zt_regs_run(struct tdescr *td, siginfo_t *si, ucontext_t *uc)
{
 size_t offset;
 struct _aarch64_ctx *head = GET_BUF_RESV_HEAD(context);
 struct zt_context *zt;
 char *zeros;

 /*
 * Get a signal context which should have a ZT frame and registers
 * in it.
 */
 enable_za();
 if (!get_current_context(td, &context.uc, sizeof(context)))
  return 1;

 head = get_header(head, ZT_MAGIC, GET_BUF_RESV_SIZE(context), &offset);
 if (!head) {
  fprintf(stderr, "No ZT context\n");
  return 1;
 }

 zt = (struct zt_context *)head;
 if (zt->nregs == 0) {
  fprintf(stderr, "Got context with no registers\n");
  return 1;
 }

 fprintf(stderr, "Got expected size %u for %d registers\n",
  head->size, zt->nregs);

 /* We didn't load any data into ZT so it should be all zeros */
 zeros = malloc(ZT_SIG_REGS_SIZE(zt->nregs));
 if (!zeros) {
  fprintf(stderr, "Out of memory, nregs=%u\n", zt->nregs);
  return 1;
 }
 memset(zeros, 0, ZT_SIG_REGS_SIZE(zt->nregs));

 if (memcmp(zeros, (char *)zt + ZT_SIG_REGS_OFFSET,
     ZT_SIG_REGS_SIZE(zt->nregs)) != 0) {
  fprintf(stderr, "ZT data invalid\n");
  free(zeros);
  return 1;
 }

 free(zeros);

 td->pass = 1;

 return 0;
}

struct tdescr tde = {
 .name = "ZT register data",
 .descr = "Validate that ZT is present and has data when ZA is enabled",
 .feats_required = FEAT_SME2,
 .timeout = 3,
 .sanity_disabled = true,
 .run = zt_regs_run,
};