toaruos/kernel/arch/x86_64/idt.c
2021-06-06 15:54:30 +09:00

272 lines
7.9 KiB
C

#include <kernel/types.h>
#include <kernel/string.h>
#include <kernel/printf.h>
#include <kernel/vfs.h>
#include <kernel/version.h>
#include <kernel/process.h>
#include <kernel/signal.h>
#include <kernel/misc.h>
#include <sys/time.h>
#include <sys/utsname.h>
#include <kernel/arch/x86_64/mmu.h>
#include <kernel/arch/x86_64/ports.h>
#include <kernel/arch/x86_64/pml.h>
#include <kernel/arch/x86_64/regs.h>
#include <kernel/arch/x86_64/irq.h>
#undef DEBUG_FAULTS
#define LOUD_SEGFAULTS
static struct idt_pointer idtp;
static idt_entry_t idt[256];
void idt_set_gate(uint8_t num, interrupt_handler_t handler, uint16_t selector, uint8_t flags, int userspace) {
uintptr_t base = (uintptr_t)handler;
idt[num].base_low = (base & 0xFFFF);
idt[num].base_mid = (base >> 16) & 0xFFFF;
idt[num].base_high = (base >> 32) & 0xFFFFFFFF;
idt[num].selector = selector;
idt[num].zero = 0;
idt[num].pad = 0;
idt[num].flags = flags | (userspace ? 0x60 : 0);
}
void idt_install(void) {
idtp.limit = sizeof(idt);
idtp.base = (uintptr_t)&idt;
/** ISRs */
idt_set_gate(0, _isr0, 0x08, 0x8E, 0);
idt_set_gate(1, _isr1, 0x08, 0x8E, 0);
idt_set_gate(2, _isr2, 0x08, 0x8E, 0);
idt_set_gate(3, _isr3, 0x08, 0x8E, 0);
idt_set_gate(4, _isr4, 0x08, 0x8E, 0);
idt_set_gate(5, _isr5, 0x08, 0x8E, 0);
idt_set_gate(6, _isr6, 0x08, 0x8E, 0);
idt_set_gate(7, _isr7, 0x08, 0x8E, 0);
idt_set_gate(8, _isr8, 0x08, 0x8E, 0);
idt_set_gate(9, _isr9, 0x08, 0x8E, 0);
idt_set_gate(10, _isr10, 0x08, 0x8E, 0);
idt_set_gate(11, _isr11, 0x08, 0x8E, 0);
idt_set_gate(12, _isr12, 0x08, 0x8E, 0);
idt_set_gate(13, _isr13, 0x08, 0x8E, 0);
idt_set_gate(14, _isr14, 0x08, 0x8E, 0);
idt_set_gate(15, _isr15, 0x08, 0x8E, 0);
idt_set_gate(16, _isr16, 0x08, 0x8E, 0);
idt_set_gate(17, _isr17, 0x08, 0x8E, 0);
idt_set_gate(18, _isr18, 0x08, 0x8E, 0);
idt_set_gate(19, _isr19, 0x08, 0x8E, 0);
idt_set_gate(20, _isr20, 0x08, 0x8E, 0);
idt_set_gate(21, _isr21, 0x08, 0x8E, 0);
idt_set_gate(22, _isr22, 0x08, 0x8E, 0);
idt_set_gate(23, _isr23, 0x08, 0x8E, 0);
idt_set_gate(24, _isr24, 0x08, 0x8E, 0);
idt_set_gate(25, _isr25, 0x08, 0x8E, 0);
idt_set_gate(26, _isr26, 0x08, 0x8E, 0);
idt_set_gate(27, _isr27, 0x08, 0x8E, 0);
idt_set_gate(28, _isr28, 0x08, 0x8E, 0);
idt_set_gate(29, _isr29, 0x08, 0x8E, 0);
idt_set_gate(30, _isr30, 0x08, 0x8E, 0);
idt_set_gate(31, _isr31, 0x08, 0x8E, 0);
idt_set_gate(32, _irq0, 0x08, 0x8E, 0);
idt_set_gate(33, _irq1, 0x08, 0x8E, 0);
idt_set_gate(34, _irq2, 0x08, 0x8E, 0);
idt_set_gate(35, _irq3, 0x08, 0x8E, 0);
idt_set_gate(36, _irq4, 0x08, 0x8E, 0);
idt_set_gate(37, _irq5, 0x08, 0x8E, 0);
idt_set_gate(38, _irq6, 0x08, 0x8E, 0);
idt_set_gate(39, _irq7, 0x08, 0x8E, 0);
idt_set_gate(40, _irq8, 0x08, 0x8E, 0);
idt_set_gate(41, _irq9, 0x08, 0x8E, 0);
idt_set_gate(42, _irq10, 0x08, 0x8E, 0);
idt_set_gate(43, _irq11, 0x08, 0x8E, 0);
idt_set_gate(44, _irq12, 0x08, 0x8E, 0);
idt_set_gate(45, _irq13, 0x08, 0x8E, 0);
idt_set_gate(46, _irq14, 0x08, 0x8E, 0);
idt_set_gate(47, _irq15, 0x08, 0x8E, 0);
idt_set_gate(124, _isr124, 0x08, 0x8E, 0); /* Bad TLB shootdown. */
idt_set_gate(125, _isr125, 0x08, 0x8E, 0); /* Halts everyone. */
idt_set_gate(126, _isr126, 0x08, 0x8E, 0); /* Does nothing, used to exit wait-for-interrupt sleep. */
idt_set_gate(127, _isr127, 0x08, 0x8E, 1); /* Legacy system call entry point, called by userspace. */
asm volatile (
"lidt %0"
: : "m"(idtp)
);
}
void idt_ap_install(void) {
idtp.limit = sizeof(idt);
idtp.base = (uintptr_t)&idt;
asm volatile (
"lidt %0"
: : "m"(idtp)
);
}
static spin_lock_t dump_lock = {0};
static void dump_regs(struct regs * r) {
spin_lock(dump_lock);
printf(
"Registers at interrupt:\n"
" $rip=0x%016lx\n"
" $rsi=0x%016lx,$rdi=0x%016lx,$rbp=0x%016lx,$rsp=0x%016lx\n"
" $rax=0x%016lx,$rbx=0x%016lx,$rcx=0x%016lx,$rdx=0x%016lx\n"
" $r8= 0x%016lx,$r9= 0x%016lx,$r10=0x%016lx,$r11=0x%016lx\n"
" $r12=0x%016lx,$r13=0x%016lx,$r14=0x%016lx,$r15=0x%016lx\n"
" cs=0x%016lx ss=0x%016lx rflags=0x%016lx int=0x%02lx err=0x%02lx\n",
r->rip,
r->rsi, r->rdi, r->rbp, r->rsp,
r->rax, r->rbx, r->rcx, r->rdx,
r->r8, r->r9, r->r10, r->r11,
r->r12, r->r13, r->r14, r->r15,
r->cs, r->ss, r->rflags, r->int_no, r->err_code
);
spin_unlock(dump_lock);
}
extern void syscall_handler(struct regs *);
#define IRQ_CHAIN_SIZE 16
#define IRQ_CHAIN_DEPTH 4
static irq_handler_chain_t irq_routines[IRQ_CHAIN_SIZE * IRQ_CHAIN_DEPTH] = { NULL };
static const char * _irq_handler_descriptions[IRQ_CHAIN_SIZE * IRQ_CHAIN_DEPTH] = { NULL };
const char * get_irq_handler(int irq, int chain) {
if (irq >= IRQ_CHAIN_SIZE) return NULL;
if (chain >= IRQ_CHAIN_DEPTH) return NULL;
return _irq_handler_descriptions[IRQ_CHAIN_SIZE * chain + irq];
}
void irq_install_handler(size_t irq, irq_handler_chain_t handler, const char * desc) {
for (size_t i = 0; i < IRQ_CHAIN_DEPTH; i++) {
if (irq_routines[i * IRQ_CHAIN_SIZE + irq])
continue;
irq_routines[i * IRQ_CHAIN_SIZE + irq] = handler;
_irq_handler_descriptions[i * IRQ_CHAIN_SIZE + irq ] = desc;
break;
}
}
void irq_uninstall_handler(size_t irq) {
for (size_t i = 0; i < IRQ_CHAIN_DEPTH; i++)
irq_routines[i * IRQ_CHAIN_SIZE + irq] = NULL;
}
static void map_more_stack(uintptr_t fromAddr) {
volatile process_t * volatile proc = this_core->current_process;
if (proc->group != 0) {
proc = process_from_pid(proc->group);
}
spin_lock(proc->image.lock);
for (uintptr_t i = fromAddr; i < proc->image.userstack; i += 0x1000) {
union PML * page = mmu_get_page(i, MMU_GET_MAKE);
mmu_frame_allocate(page, MMU_FLAG_WRITABLE);
mmu_invalidate(i);
}
proc->image.userstack = fromAddr;
spin_unlock(proc->image.lock);
}
struct regs * isr_handler(struct regs * r) {
this_core->interrupt_registers = r;
switch (r->int_no) {
case 14: /* Page fault */ {
uintptr_t faulting_address;
asm volatile("mov %%cr2, %0" : "=r"(faulting_address));
if (!this_core->current_process || r->cs == 0x08) {
arch_fatal();
}
if (faulting_address == 0xFFFFB00F) {
/* Thread exit */
task_exit(0);
break;
}
if (faulting_address == 0x8DEADBEEF) {
return_from_signal_handler();
break;
}
if (faulting_address < 0x800000000000 && faulting_address > 0x700000000000) {
map_more_stack(faulting_address & 0xFFFFffffFFFFf000);
break;
}
#ifdef DEBUG_FAULTS
arch_fatal();
#else
# ifdef LOUD_SEGFAULTS
printf("Page fault in pid=%d (%s; cpu=%d) at %#zx\n", (int)this_core->current_process->id, this_core->current_process->name, this_core->cpu_id, faulting_address);
dump_regs(r);
# endif
send_signal(this_core->current_process->id, SIGSEGV, 1);
#endif
break;
}
case 13: /* GPF */ {
#ifdef DEBUG_FAULTS
arch_fatal();
#else
if (!this_core->current_process || r->cs == 0x08) {
arch_fatal();
}
# ifdef LOUD_SEGFAULTS
printf("GPF in userspace on CPU %d\n", this_core->cpu_id);
dump_regs(r);
# endif
send_signal(this_core->current_process->id, SIGSEGV, 1);
#endif
break;
}
case 8: /* Double fault */ {
arch_fatal();
break;
}
case 127: /* syscall */ {
syscall_handler(r);
asm volatile("sti");
return r;
}
case 39: {
/* Spurious interrupt */
break;
}
default: {
if (r->int_no < 32) {
#ifdef DEBUG_FAULTS
arch_fatal();
#else
if (!this_core->current_process || r->cs == 0x08) {
arch_fatal();
}
send_signal(this_core->current_process->id, SIGILL, 1);
#endif
} else {
for (size_t i = 0; i < IRQ_CHAIN_DEPTH; i++) {
irq_handler_chain_t handler = irq_routines[i * IRQ_CHAIN_SIZE + (r->int_no - 32)];
if (!handler) break;
if (handler(r)) {
goto done;
}
}
irq_ack(r->int_no - 32);
break;
}
}
}
done:
if (this_core->current_process == this_core->kernel_idle_task && process_queue && process_queue->head) {
/* If this is kidle and we got here, instead of finishing the interrupt
* we can just switch task and there will probably be something else
* to run that was awoken by the interrupt. */
switch_next();
}
return r;
}