At this point the threads are really processes...

This commit is contained in:
Thomas Lovén 2018-02-20 20:46:24 +01:00
parent 6523b10984
commit e868b83f50
6 changed files with 93 additions and 93 deletions

View File

@ -5,17 +5,17 @@
#include <multiboot.h>
#include <cpu.h>
#include <interrupts.h>
#include <thread.h>
#include <process.h>
#include <scheduler.h>
int *thread_id = (int *)0x20000;
int *pid = (int *)0x20000;
void thread_function()
{
*thread_id = thread()->tid;
*pid = process()->pid;
while(1)
{
debug("Thread %d\n", *thread_id);
debug("Process %d\n", *pid);
yield();
}
}
@ -35,16 +35,16 @@ void kmain(uint64_t multiboot_magic, void *multiboot_data)
cpu_init();
struct thread *th1 = new_thread(thread_function);
vmm_set_page(th1->P4, 0x20000, pmm_alloc(), PAGE_WRITE | PAGE_PRESENT);
struct thread *th2 = new_thread(thread_function);
vmm_set_page(th2->P4, 0x20000, pmm_alloc(), PAGE_WRITE | PAGE_PRESENT);
struct thread *th3 = new_thread(thread_function);
vmm_set_page(th3->P4, 0x20000, pmm_alloc(), PAGE_WRITE | PAGE_PRESENT);
struct process *p1 = new_process(thread_function);
vmm_set_page(p1->P4, 0x20000, pmm_alloc(), PAGE_WRITE | PAGE_PRESENT);
struct process *p2 = new_process(thread_function);
vmm_set_page(p2->P4, 0x20000, pmm_alloc(), PAGE_WRITE | PAGE_PRESENT);
struct process *p3 = new_process(thread_function);
vmm_set_page(p3->P4, 0x20000, pmm_alloc(), PAGE_WRITE | PAGE_PRESENT);
ready(th1);
ready(th2);
ready(th3);
ready(p1);
ready(p2);
ready(p3);
debug_ok("Boot \"Complete\"\n");

View File

@ -3,9 +3,9 @@
#include <scheduler.h>
#include <stdint.h>
struct thread
struct process
{
uint64_t tid;
uint64_t pid;
void *stack_ptr;
uint64_t state;
uint64_t P4;
@ -13,9 +13,9 @@ struct thread
uint8_t stack[];
};
struct thread *thread();
struct process *process();
struct thread *new_thread(void (*function)(void));
struct process *new_process(void (*function)(void));
void yield();
void start_scheduler();

View File

@ -1,8 +1,8 @@
#pragma once
#include <queue.h>
#define runQ readyQ, readyQ_next, struct thread
#define runQ readyQ, readyQ_next, struct process
QUEUE_DECLARE(runQ);
void ready(struct thread *th);
struct thread *scheduler_next();
void ready(struct process *proc);
struct process *scheduler_next();

69
src/kernel/proc/process.c Normal file
View File

@ -0,0 +1,69 @@
#include <process.h>
#include <scheduler.h>
#include <memory.h>
#include <cpu.h>
struct swtch_stack
{
uint64_t RBP;
uint64_t RBX;
uint64_t R12;
uint64_t R13;
uint64_t R14;
uint64_t R15;
uint64_t RBP2;
uint64_t ret;
}__attribute__((packed));
struct process *sched_proc = 0;
struct process *_proc = 0;
uint64_t next_pid = 1;
struct process *new_process(void (*function)(void))
{
struct process *proc = P2V(pmm_calloc());
proc->pid = next_pid++;
proc->stack_ptr = incptr(proc, PAGE_SIZE - sizeof(struct swtch_stack));
proc->P4 = new_P4();
struct swtch_stack *stk = proc->stack_ptr;
stk->RBP = (uint64_t)&stk->RBP2;
stk->ret = (uint64_t)function;
return proc;
}
struct process *process()
{
return _proc;
}
void yield()
{
switch_stack(&_proc->stack_ptr, &sched_proc->stack_ptr);
}
void scheduler()
{
while(1)
{
struct process *new = 0;
while(!(new = scheduler_next()));
_proc = new;
write_cr3(new->P4);
switch_stack(&sched_proc->stack_ptr, &new->stack_ptr);
ready(_proc);
_proc = 0;
}
}
void start_scheduler()
{
sched_proc = new_process(scheduler);
sched_proc->pid = (uint64_t)-1;
uint64_t stack;
switch_stack(&stack, &sched_proc->stack_ptr);
}

View File

@ -1,15 +1,15 @@
#include <scheduler.h>
#include <queue.h>
#include <thread.h>
#include <process.h>
QUEUE_DEFINE(runQ);
void ready(struct thread *th)
void ready(struct process *proc)
{
queue_add(runQ, th);
queue_add(runQ, proc);
}
struct thread *scheduler_next()
struct process *scheduler_next()
{
return queue_pop(runQ);
}

View File

@ -1,69 +0,0 @@
#include <thread.h>
#include <scheduler.h>
#include <memory.h>
#include <cpu.h>
struct swtch_stack
{
uint64_t RBP;
uint64_t RBX;
uint64_t R12;
uint64_t R13;
uint64_t R14;
uint64_t R15;
uint64_t RBP2;
uint64_t ret;
}__attribute__((packed));
struct thread *sched_th = 0;
struct thread *_thread = 0;
uint64_t next_tid = 1;
struct thread *new_thread(void (*function)(void))
{
struct thread *th = P2V(pmm_calloc());
th->tid = next_tid++;
th->stack_ptr = incptr(th, PAGE_SIZE - sizeof(struct swtch_stack));
th->P4 = new_P4();
struct swtch_stack *stk = th->stack_ptr;
stk->RBP = (uint64_t)&stk->RBP2;
stk->ret = (uint64_t)function;
return th;
}
struct thread *thread()
{
return _thread;
}
void yield()
{
switch_stack(&_thread->stack_ptr, &sched_th->stack_ptr);
}
void scheduler()
{
while(1)
{
struct thread *new = 0;
while(!(new = scheduler_next()));
_thread = new;
write_cr3(new->P4);
switch_stack(&sched_th->stack_ptr, &new->stack_ptr);
ready(_thread);
_thread = 0;
}
}
void start_scheduler()
{
sched_th = new_thread(scheduler);
sched_th->tid = (uint64_t)-1;
uint64_t stack;
switch_stack(&stack, &sched_th->stack_ptr);
}