Astral/kernel-src/arch/x86-64/cpu.c

#include <arch/cpu.h>
#include <arch/apic.h>
#include <cpuid.h>
#include <logging.h>

#define CPUID_SYSCALL (1 << 11)
#define EFER_SYSCALLENABLE 1

void arch_syscall_entry();
static void illisr(isr_t *self, context_t *ctx) {
	if (ARCH_CONTEXT_ISUSER(ctx)) {
		signal_signalthread(_cpu()->thread, SIGILL, true);
	} else {
		_panic("Invalid Opcode", ctx);
	}
}

static void div0isr(isr_t *self, context_t *ctx) {
	if (ARCH_CONTEXT_ISUSER(ctx)) {
		signal_signalthread(_cpu()->thread, SIGFPE, true);
	} else {
		_panic("Division by 0", ctx);
	}
}

void cpu_initstate() {
	arch_apic_initap();

	// syscall instruction
	unsigned int eax = 0,ebx = 0,ecx = 0,edx = 0;
	__get_cpuid(0x80000001, &eax, &ebx, &ecx, &edx);

	// TODO in the future, have an alternative way of handling syscalls
	__assert(edx & CPUID_SYSCALL);
	uint64_t efer = rdmsr(MSR_EFER);
	efer |= EFER_SYSCALLENABLE;
	wrmsr(MSR_EFER, efer);

	uint64_t star = 0;
	star |= (uint64_t)0x13 << 48;
	star |= (uint64_t)0x08 << 32;

	wrmsr(MSR_STAR, star);
	wrmsr(MSR_LSTAR, (uint64_t)arch_syscall_entry);
	wrmsr(MSR_CSTAR, 0); // no compatibility mode syscall handler
	wrmsr(MSR_FMASK, 0x200); // disable interrupts on syscall

	// enable SSE
	asm volatile(
		"mov %%cr0, %%rax;"
		"and $0xFFFB, %%ax;"
		"or  $2, %%eax;"
		"mov %%rax, %%cr0;"
		"mov %%cr4, %%rax;"
		"or  $0b11000000000, %%rax;"
		"mov %%rax, %%cr4;"
		: : : "rax");

	// set NE in cr0 and reset x87 fpu
	asm volatile(
		"fninit;"
		"mov %%cr0, %%rax;"
		"or $0b100000, %%rax;"
		"mov %%rax, %%cr0;"
		: : : "rax"
	);

	interrupt_register(0, div0isr, NULL, IPL_IGNORE);
	interrupt_register(6, illisr, NULL, IPL_IGNORE);
}