290 lines
7.3 KiB
C
290 lines
7.3 KiB
C
#include <stdint.h>
|
|
#include <stdlib.h>
|
|
#include <stdbool.h>
|
|
#include "hal/hal.h"
|
|
#include "spinlock/spinlock.h"
|
|
#include "proc.h"
|
|
#include "dlmalloc/malloc.h"
|
|
#include "pmm/pmm.h"
|
|
#include "util/util.h"
|
|
#include "kprintf.h"
|
|
#include "elf.h"
|
|
#include "errors.h"
|
|
#include "vfs/vfs.h"
|
|
#include "bootinfo/bootinfo.h"
|
|
|
|
#define PROC_REAPER_FREQ 30
|
|
|
|
uint64_t pids = 0;
|
|
uint64_t sched_ticks = 0;
|
|
|
|
Procs PROCS;
|
|
|
|
bool proc_checkelf(uint8_t *elf) {
|
|
if (elf[0] != 0x7f || elf[1] != 'E' || elf[2] != 'L' || elf[3] != 'F') {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
ElfAuxval proc_load_elf_segs(Proc *proc, uint8_t *data) {
|
|
PgTable *vas = proc->platformdata.cr3;
|
|
ElfAuxval aux = {0};
|
|
|
|
Elf64_Ehdr *elfhdr = (Elf64_Ehdr *)data;
|
|
aux.entry = elfhdr->e_entry;
|
|
aux.phnum = elfhdr->e_phnum;
|
|
aux.phent = elfhdr->e_phentsize;
|
|
|
|
for (uint64_t seg = 0; seg < elfhdr->e_phnum; seg++) {
|
|
Elf64_Phdr *phdr = (Elf64_Phdr *)(data + elfhdr->e_phoff + (elfhdr->e_phentsize * seg));
|
|
|
|
switch (phdr->p_type) {
|
|
case PT_PHDR: {
|
|
aux.phdr = (uint64_t)phdr->p_vaddr;
|
|
} break;
|
|
case PT_LOAD: {
|
|
uint64_t off = phdr->p_vaddr & (HAL_PAGE_SIZE - 1);
|
|
uint64_t blocks = (phdr->p_memsz / HAL_PAGE_SIZE) + 1;
|
|
|
|
uint8_t *physaddr = pmm_alloc(blocks);
|
|
uint8_t *virtaddr = (uint8_t *)(phdr->p_vaddr - off);
|
|
|
|
hal_memset(VIRT(physaddr), 0, phdr->p_memsz);
|
|
hal_memcpy(VIRT(physaddr) + off, (data + phdr->p_offset), phdr->p_filesz);
|
|
|
|
uint32_t pgflags = HAL_PG_USER | HAL_PG_RW | HAL_PG_PRESENT;
|
|
hal_vmm_map_range(VIRT(vas), virtaddr, physaddr, blocks * HAL_PAGE_SIZE, pgflags);
|
|
|
|
VasRange *range = dlmalloc(sizeof(*range));
|
|
range->virtstart = virtaddr;
|
|
range->physstart = physaddr;
|
|
range->size = blocks * HAL_PAGE_SIZE;
|
|
range->pgflags = pgflags;
|
|
|
|
LL_APPEND(proc->vas, range);
|
|
} break;
|
|
}
|
|
}
|
|
return aux;
|
|
}
|
|
|
|
Proc *proc_spawnkern(void (*ent)(void), char *name) {
|
|
if (pids >= PROC_MAX) {
|
|
return NULL;
|
|
}
|
|
|
|
Proc *proc = dlmalloc(sizeof(*proc));
|
|
if (proc == NULL) {
|
|
return NULL;
|
|
}
|
|
hal_memset(proc, 0, sizeof(*proc));
|
|
hal_memcpy(proc->name, name, PROC_NAME_MAX);
|
|
|
|
proc->kern = true;
|
|
|
|
uint8_t *sp = (uint8_t *)pmm_alloc(PROC_STACKBLOCKS) + PROC_STACKSIZE;
|
|
|
|
proc->platformdata.kstack = sp;
|
|
hal_memset(&proc->platformdata.trapframe, 0, sizeof(proc->platformdata.trapframe));
|
|
|
|
proc->platformdata.trapframe.ss = 0x10;
|
|
proc->platformdata.trapframe.rsp = (uint64_t)VIRT(sp);
|
|
proc->platformdata.trapframe.rflags = 0x202;
|
|
proc->platformdata.trapframe.cs = 0x08;
|
|
proc->platformdata.trapframe.rip = (uint64_t)ent;
|
|
proc->platformdata.cr3 = hal_vmm_current_cr3();
|
|
proc->state = PROC_READY;
|
|
proc->pid = pids++;
|
|
|
|
return proc;
|
|
}
|
|
|
|
Proc *proc_spawnuser(char *mountpoint, char *path) {
|
|
VfsObj *vobj = vfs_open(mountpoint, path, VFS_FLAG_READ);
|
|
if (vobj == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
VfsStat stat;
|
|
if (vobj->stat(vobj, &stat) != E_OK) {
|
|
vfs_close(vobj);
|
|
return NULL;
|
|
}
|
|
|
|
if (stat.type != VFS_TYPE_FILE) {
|
|
vfs_close(vobj);
|
|
return NULL;
|
|
}
|
|
|
|
uint8_t *data = dlmalloc(stat.size);
|
|
if (data == NULL) {
|
|
vfs_close(vobj);
|
|
return NULL;
|
|
}
|
|
|
|
if (vobj->read(vobj, data, stat.size, 0) != E_OK) {
|
|
dlfree(data);
|
|
vfs_close(vobj);
|
|
return NULL;
|
|
}
|
|
|
|
vfs_close(vobj);
|
|
|
|
Proc *proc = dlmalloc(sizeof(*proc));
|
|
hal_memset(proc, 0, sizeof(*proc));
|
|
ksprintf(proc->name, "%s:%s", mountpoint, path);
|
|
|
|
uint8_t *sp = (uint8_t *)pmm_alloc(PROC_STACKBLOCKS) + PROC_STACKSIZE;
|
|
uint8_t *spbase = sp - PROC_STACKSIZE;
|
|
|
|
proc->platformdata.kstack = sp;
|
|
hal_memset(&proc->platformdata.trapframe, 0, sizeof(proc->platformdata.trapframe));
|
|
|
|
proc->platformdata.cr3 = hal_vmm_userproc_pml4(proc);
|
|
uint32_t flags = HAL_PG_RW | HAL_PG_USER | HAL_PG_PRESENT;
|
|
hal_vmm_map_range(VIRT(proc->platformdata.cr3), spbase, spbase, PROC_STACKSIZE, flags);
|
|
VasRange *range = dlmalloc(sizeof(*range));
|
|
range->virtstart = spbase;
|
|
range->physstart = spbase;
|
|
range->size = PROC_STACKSIZE;
|
|
range->pgflags = flags;
|
|
|
|
LL_APPEND(proc->vas, range);
|
|
|
|
ElfAuxval aux = proc_load_elf_segs(proc, data);
|
|
|
|
proc->platformdata.trapframe.ss = 0x20 | 0x3;
|
|
proc->platformdata.trapframe.rsp = (uint64_t)VIRT(sp);
|
|
proc->platformdata.trapframe.rflags = 0x202;
|
|
proc->platformdata.trapframe.cs = 0x18 | 0x3;
|
|
proc->platformdata.trapframe.rip = aux.entry;
|
|
proc->state = PROC_READY;
|
|
proc->pid = pids++;
|
|
|
|
return proc;
|
|
}
|
|
|
|
void proc_register(Proc *proc) {
|
|
spinlock_acquire(&PROCS.spinlock);
|
|
LL_APPEND(PROCS.procs, proc);
|
|
spinlock_release(&PROCS.spinlock);
|
|
}
|
|
|
|
Proc *proc_nextready(void) {
|
|
Proc *proc = PROCS.current->next;
|
|
for (;;) {
|
|
if (proc == NULL) {
|
|
proc = PROCS.procs;
|
|
}
|
|
if (proc->state != PROC_ZOMBIE) {
|
|
return proc;
|
|
}
|
|
proc = proc->next;
|
|
}
|
|
return proc;
|
|
}
|
|
|
|
void proc_reaper(void) {
|
|
Proc *head = PROCS.procs;
|
|
while (head) {
|
|
if (head->state == PROC_ZOMBIE) {
|
|
Proc *zombie = head;
|
|
head = head->next;
|
|
|
|
LL_REMOVE(PROCS.procs, zombie);
|
|
|
|
pmm_free((uintptr_t)(zombie->platformdata.kstack - PROC_STACKSIZE), PROC_STACKBLOCKS);
|
|
|
|
if (!zombie->kern) {
|
|
VasRange *vashead = zombie->vas;
|
|
size_t i = 0;
|
|
while (vashead) {
|
|
VasRange *tmp = vashead;
|
|
vashead = vashead->next;
|
|
hal_vmm_unmap_range(VIRT(zombie->platformdata.cr3), tmp->virtstart, tmp->physstart, tmp->size);
|
|
// first pmm mapping is for the elf itself
|
|
if (i == 0) {
|
|
pmm_free((uintptr_t)tmp->physstart, tmp->size / HAL_PAGE_SIZE);
|
|
}
|
|
dlfree(tmp);
|
|
i++;
|
|
}
|
|
|
|
pmm_free((uintptr_t)zombie->platformdata.cr3, 1);
|
|
}
|
|
dlfree(zombie);
|
|
} else {
|
|
head = head->next;
|
|
}
|
|
}
|
|
}
|
|
|
|
void proc_sched(void *cpustate) {
|
|
hal_intr_disable();
|
|
sched_ticks++;
|
|
|
|
if (sched_ticks % PROC_REAPER_FREQ == 0) {
|
|
proc_reaper();
|
|
}
|
|
|
|
IntrStackFrame *frame = cpustate;
|
|
|
|
PROCS.current->platformdata.trapframe = *frame;
|
|
PROCS.current->platformdata.fsbase = hal_cpu_rdmsr(HAL_CPU_FSBASE);
|
|
|
|
PROCS.current = proc_nextready();
|
|
PROCS.current->state = PROC_RUNNING;
|
|
|
|
hal_cpu_wrmsr(HAL_CPU_FSBASE, PROCS.current->platformdata.fsbase);
|
|
HAL_CPUS[0].syscall_kstack = VIRT(PROCS.current->platformdata.kstack);
|
|
hal_switchproc(&PROCS.current->platformdata.trapframe, (void *)PROCS.current->platformdata.cr3);
|
|
}
|
|
|
|
void proc_kill(Proc *proc) {
|
|
proc->state = PROC_ZOMBIE;
|
|
}
|
|
|
|
void proc_killself(void) {
|
|
spinlock_acquire(&PROCS.spinlock);
|
|
Proc *proc = PROCS.current;
|
|
proc_kill(proc);
|
|
spinlock_release(&PROCS.spinlock);
|
|
}
|
|
|
|
void proc_idle(void) {
|
|
for(;;) {
|
|
// .
|
|
}
|
|
}
|
|
|
|
void proc_status(void) {
|
|
static const char *statuses[] = {"ready", "running", "zombie", "waiting"};
|
|
for (;;) {
|
|
spinlock_acquire(&PROCS.spinlock);
|
|
Proc *head = PROCS.procs;
|
|
while (head) {
|
|
kprintf("%s %s %s\n", head->kern ? "kern" : "user", statuses[head->state], head->name);
|
|
head = head->next;
|
|
}
|
|
kprintf("\n\n");
|
|
spinlock_release(&PROCS.spinlock);
|
|
|
|
hal_wait(3 * 1000);
|
|
}
|
|
}
|
|
|
|
void proc_init(void) {
|
|
spinlock_init(&PROCS.spinlock);
|
|
PROCS.procs = NULL;
|
|
|
|
Proc *idle = proc_spawnkern(&proc_idle, "kIDLE");
|
|
proc_register(idle);
|
|
PROCS.current = idle;
|
|
|
|
proc_register(proc_spawnkern(&proc_status, "status"));
|
|
proc_register(proc_spawnuser("base", "/bin/hello"));
|
|
|
|
hal_switchproc(&PROCS.current->platformdata.trapframe, (void *)PROCS.current->platformdata.cr3);
|
|
}
|