// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2026 Arm Ltd
 *
 * Verify that the POR_EL0 register is left untouched on sigreturn if the
 * POE frame record is missing.
 */

#include <asm/sigcontext.h>

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

#define POR_EL0_INIT    0x07ul
#define POR_EL0_CUSTOM  0x77ul

static bool failed_check;

static bool modify_por_el0(struct tdescr *td)
{
        set_por_el0(POR_EL0_CUSTOM);

        return true;
}

static int signal_remove_poe_context(struct tdescr *td, siginfo_t *si,
                                      ucontext_t *uc)
{
        struct _aarch64_ctx *ctx = GET_UC_RESV_HEAD(uc);
        size_t resv_size = GET_UCP_RESV_SIZE(uc);
        struct _aarch64_ctx *poe_ctx_head;

        poe_ctx_head = get_header(ctx, POE_MAGIC, resv_size, NULL);
        if (!poe_ctx_head) {
                fprintf(stderr, "Missing poe_context record\n");
                failed_check = true;
                return 0;
        }

        /*
         * Actually removing the record would require moving down the next
         * records. An easier option is to turn it into an ESR record, which is
         * ignored by sigreturn().
         */
        poe_ctx_head->magic = ESR_MAGIC;

        return 0;
}

static void check_por_el0_preserved(struct tdescr *td)
{
        uint64_t por_el0 = get_por_el0();

        if (por_el0 == POR_EL0_INIT) {
                fprintf(stderr, "POR_EL0 preserved\n");
        } else {
                fprintf(stderr, "POR_EL0 unexpectedly set to %lx\n", por_el0);
                failed_check = true;
        }

        td->pass = !failed_check;
}

struct tdescr tde = {
        .name = "POR_EL0 missing poe_context",
        .descr = "Remove poe_context record and check POR_EL0 is preserved",
        .feats_required = FEAT_POE,
        .timeout = 3,
        .sig_trig = SIGUSR1,
        .init = modify_por_el0,
        .run = signal_remove_poe_context,
        .check_result = check_por_el0_preserved,
};