#include <tce/syscall.h>
#include <stdlib.h>
#include <sched.h>

static size_t call(size_t a, size_t b, size_t c, size_t d, size_t e, size_t f) {
	size_t ret;
	asm volatile("int $64" : "=a"(ret) : "a"(a), "b"(b), "c"(c), "d"(d), "S"(e), "D"(f));
	return ret;
}

void thread_exit() {
	call(SC_THREAD_EXIT, 0, 0, 0, 0, 0);
}

void schedule() {
	call(SC_SCHEDULE, 0, 0,0, 0, 0);
}

void thread_sleep(int time) {
	call(SC_THREAD_SLEEP, time, 0, 0, 0, 0);
}

void process_exit(int retval) {
	call(SC_PROCESS_EXIT, retval, 0, 0, 0, 0);
}

void printk(char* str) {
	call(SC_PRINTK, (unsigned)str, 0, 0, 0, 0);
}

//THREAD CREATION
struct thread_start_data {
	void (*entry)(void*);
	void *data;
	void *stack;
};
void *_stack_to_free = 0;
uint32_t _stack_freeing_mutex = MUTEX_UNLOCKED;

void thread_start(void *data) {
	struct thread_start_data *tsd = data;
	tsd->entry(tsd->data);
	mutex_lock(&_stack_freeing_mutex);
	if (_stack_to_free != 0) free(_stack_to_free);
	_stack_to_free = tsd->stack;
	asm volatile("movl %0, (_stack_freeing_mutex); int $64;" ::
		"a"(SC_THREAD_EXIT), "r"(MUTEX_UNLOCKED));
}
void thread_new(void (*entry)(void*), void *data) {
	struct thread_start_data *tsd = malloc(sizeof(struct thread_start_data));
	tsd->entry = entry;
	tsd->data = data;
	tsd->stack = malloc(NEW_STACK_SIZE);
	call(SC_THREAD_NEW, (unsigned)thread_start, (unsigned)tsd, (unsigned)(tsd->stack + NEW_STACK_SIZE), 0, 0);
}

void irq_wait(int number) {
	call(SC_IRQ_WAIT, number, 0, 0, 0, 0);
}

int proc_priv() {
	return call(SC_PROC_PRIV, 0, 0, 0, 0, 0);
}

void* sbrk(size_t s) {
	return (void*)call(SC_SBRK, s, 0, 0, 0, 0);
}

void brk(void* ptr) {
	return call (SC_BRK, ptr, 0, 0, 0, 0);
}