#include <stdio.h>
#include <stdlib.h>
#include <assert.h>

#include <fbos/init.h>
#include <fbos/sched.h>
#include <fbos/mm.h>

unsigned long stack[4][THREAD_SIZE / sizeof(unsigned long)];

struct task_struct tasks[4] = {
	[TASK_INIT] = { .stack = stack[0], .entry_addr = NULL, },
	[TASK_FIZZ] = { .stack = stack[1], .entry_addr = NULL, },
	[TASK_BUZZ] = { .stack = stack[2], .entry_addr = NULL, },
	[TASK_FIZZBUZZ] = { .stack = stack[3], .entry_addr = NULL, },
};

int main(void)
{
	FILE *fh = fopen("./usr/initramfs.cpio", "rb");
	assert(fh);

	fseek(fh, 0, SEEK_END);
	long fsize = ftell(fh);
	rewind(fh);

	unsigned char *contents = malloc((unsigned long)fsize + 1);
	fread(contents, (unsigned long)fsize, 1, fh);
	fclose(fh);

	contents[fsize] = 0;

	extract_initrd(contents, (uint64_t)fsize, tasks);
	free(contents);

	assert(((uintptr_t)tasks[0].entry_addr & 0xff) == 0xe8);
	assert(((uintptr_t)tasks[1].entry_addr & 0xff) == 0xbc);
	assert(((uintptr_t)tasks[2].entry_addr & 0xff) == 0xb4);
	assert(((uintptr_t)tasks[3].entry_addr & 0xff) == 0x80);

	exit(0);
}