// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2023 ARM Limited
 */

#include <errno.h>
#include <signal.h>
#include <unistd.h>

#include <sys/mman.h>
#include <sys/prctl.h>

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

static uint64_t *gcs_page;

#ifndef __NR_map_shadow_stack
#define __NR_map_shadow_stack 453
#endif

static bool alloc_gcs(struct tdescr *td)
{
        long page_size = sysconf(_SC_PAGE_SIZE);

        gcs_page = (void *)syscall(__NR_map_shadow_stack, 0,
                                   page_size, 0);
        if (gcs_page == MAP_FAILED) {
                fprintf(stderr, "Failed to map %ld byte GCS: %d\n",
                        page_size, errno);
                return false;
        }

        return true;
}

static int gcs_write_fault_trigger(struct tdescr *td)
{
        /* Verify that the page is readable (ie, not completely unmapped) */
        fprintf(stderr, "Read value 0x%lx\n", gcs_page[0]);

        /* A regular write should trigger a fault */
        gcs_page[0] = EINVAL;

        return 0;
}

static int gcs_write_fault_signal(struct tdescr *td, siginfo_t *si,
                                  ucontext_t *uc)
{
        ASSERT_GOOD_CONTEXT(uc);

        return 1;
}


struct tdescr tde = {
        .name = "GCS write fault",
        .descr = "Normal writes to a GCS segfault",
        .feats_required = FEAT_GCS,
        .timeout = 3,
        .sig_ok = SIGSEGV,
        .sanity_disabled = true,
        .init = alloc_gcs,
        .trigger = gcs_write_fault_trigger,
        .run = gcs_write_fault_signal,
};