/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright 2023 Oxide Computer Company
 */

/*
 * The purpose of this program is to go through and replace all the FPU
 * registers in the floating point state in a ucontext_t and verify that we see
 * what we expected here.
 */

#include <err.h>
#include <stdlib.h>
#include <ucontext.h>

#include "xsave_util.h"

xsu_fpu_t to_set, found;

static void
setcontext_replace_check(uint32_t hwsup)
{
	xsu_getfpu(&found, hwsup);
	if (xsu_same(&to_set, &found, hwsup)) {
		(void) printf("TEST PASSED: setcontext() correctly wrote FPU "
		    "contents\n");
		exit(EXIT_SUCCESS);
	} else {
		errx(EXIT_FAILURE, "TEST_FAILED: setcontext() did not write "
		    "full FPU state");
	}
}

int
main(void)
{
	ucontext_t *ctx;
	uint32_t start = arc4random();
	uint32_t hwsup = xsu_hwsupport();

	ctx = ucontext_alloc(0);
	if (ctx == NULL) {
		err(EXIT_FAILURE, "failed to get allocate ucontext_t");
	}
	(void) printf("filling starting at 0x%x\n", start);
	xsu_fill(&to_set, hwsup, start);
	if (getcontext_extd(ctx, 0) != 0) {
		err(EXIT_FAILURE, "failed to get extended context");
	}

	xsu_overwrite_uctx(ctx, &to_set, hwsup);
	xsu_ustack_alloc(ctx);
	makecontext(ctx, setcontext_replace_check, 1, hwsup);
	(void) setcontext(ctx);

	err(EXIT_FAILURE, "TEST FAILED: set context did not work!");
}