#pragma once

#include <assert.h>
#include <stdint.h>
#include <string.h>
#include <ucontext.h>
#include <unistd.h>

#include "software_interrupt.h"

/*
 * This file contains the common dependencies of the architecture-dependent code
 */

typedef uint64_t reg_t;

/* Userspace Registers. */
enum UREGS
{
	UREG_RSP = 0,
	UREG_RIP = 1,
	UREG_COUNT
};

/* The enum is compared directly in assembly, so maintain integral values! */
enum ARCH_CONTEXT
{
	ARCH_CONTEXT_UNUSED  = 0,
	ARCH_CONTEXT_FAST    = 1,
	ARCH_CONTEXT_SLOW    = 2,
	ARCH_CONTEXT_RUNNING = 3
};


struct arch_context {
	enum ARCH_CONTEXT variant;
	reg_t             regs[UREG_COUNT];
	mcontext_t        mctx;
};

/*
 * This is the slowpath switch to a preempted sandbox!
 * SIGUSR1 on the current thread and restore mcontext there!
 */
extern __thread struct arch_context worker_thread_base_context;

/* Cannot be inlined because called in Assembly */
void __attribute__((noinline)) __attribute__((noreturn)) arch_context_mcontext_restore(void);

extern __thread volatile bool worker_thread_is_switching_context;