/*-
 * Copyright (c) 2025 Raptor Computing Systems, LLC
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <stdio.h>
#include <stdlib.h>
#include <ucontext.h>
#include <errno.h>

#include <atf-c.h>

#define STACK_SIZE      (64ull << 10)

static volatile int callback_reached = 0;

static ucontext_t uctx_save, uctx_switch;

static void swapcontext_callback()
{
        // Increment callback reached variable
        // If this is called multiple times, we will fail the test
        // If this is not called at all, we will fail the test
        callback_reached++;
}

ATF_TC(swapcontext_basic);
ATF_TC_HEAD(swapcontext_basic, tc)
{
        atf_tc_set_md_var(tc, "descr",
                "Verify basic functionality of swapcontext");
}

ATF_TC_BODY(swapcontext_basic, tc)
{
        char *stack;
        int res;

        stack = malloc(STACK_SIZE);
        ATF_REQUIRE_MSG(stack != NULL, "malloc failed: %s", strerror(errno));
        res = getcontext(&uctx_switch);
        ATF_REQUIRE_MSG(res == 0, "getcontext failed: %s", strerror(errno));

        uctx_switch.uc_stack.ss_sp = stack;
        uctx_switch.uc_stack.ss_size = STACK_SIZE;
        uctx_switch.uc_link = &uctx_save;
        makecontext(&uctx_switch, swapcontext_callback, 0);

        res = swapcontext(&uctx_save, &uctx_switch);

        ATF_REQUIRE_MSG(res == 0, "swapcontext failed: %s", strerror(errno));
        ATF_REQUIRE_MSG(callback_reached == 1,
                "callback failed, reached %d times", callback_reached);
}

ATF_TP_ADD_TCS(tp)
{
        ATF_TP_ADD_TC(tp, swapcontext_basic);

        return (atf_no_error());
}