Merge pull request #9 from ngiddings/pre-alpha

Pre alpha
This commit is contained in:
Nathan Giddings
2022-08-19 03:57:21 -05:00
committed by GitHub
19 changed files with 557 additions and 214 deletions

View File

@@ -1,6 +1,5 @@
#pragma once #pragma once
#include "mmgr.h"
#include "types/physaddr.h" #include "types/physaddr.h"
#include <stdint.h> #include <stdint.h>

View File

@@ -1,6 +1,5 @@
#pragma once #pragma once
#include "mmgr.h"
#include <stddef.h> #include <stddef.h>
/** /**
@@ -15,7 +14,7 @@
* @param block_size Size in bytes of a single unit of allocation * @param block_size Size in bytes of a single unit of allocation
* @return a status code * @return a status code
*/ */
int kminit(void *base, size_t heap_size, size_t block_size); int kminit(void *base, void* start, size_t heap_size, size_t block_size);
/** /**
* @brief Allocates a block of memory containing at least `size` bytes. * @brief Allocates a block of memory containing at least `size` bytes.

View File

@@ -3,6 +3,7 @@
#include "avltree.h" #include "avltree.h"
#include "memmap.h" #include "memmap.h"
#include "priorityqueue.h" #include "priorityqueue.h"
#include "queue.h"
#include "mmgr.h" #include "mmgr.h"
#include "syscalls.h" #include "syscalls.h"
#include <stddef.h> #include <stddef.h>
@@ -10,11 +11,14 @@
#define MAX_SYSCALL_ID 256 #define MAX_SYSCALL_ID 256
#define module_limit 8 #define module_limit 8
enum process_state_t #define IO_OP 1 << 0
{ #define IO_SYNC 0 << 0
PROCESS_ACTIVE, #define IO_ASYNC 1 << 0
PROCESS_WAITING
}; #define IO_RECIPIENT_TYPE 3 << 1
#define IO_PID 0 << 1
#define IO_PORT 1 << 1
#define IO_MAILBOX 2 << 1
struct process_context_t; struct process_context_t;
@@ -36,23 +40,41 @@ struct boot_info_t
struct message_t struct message_t
{ {
uint16_t sender, type; unsigned long sender;
uint32_t param1, param2, param3; unsigned long code;
unsigned long args[6];
};
enum process_state_t
{
PROCESS_ACTIVE,
PROCESS_REQUESTING,
PROCESS_SENDING
}; };
struct process_t struct process_t
{ {
size_t priority; unsigned long pid;
size_t resource_id; int priority;
physaddr_t page_table; physaddr_t page_table;
struct process_context_t *state; enum process_state_t state;
struct message_t *message; struct queue_t sending_queue;
struct queue_t message_queue;
struct message_t *message_buffer;
struct process_context_t *ctx;
};
struct port_t
{
unsigned long id;
unsigned long owner_pid;
}; };
struct kernel_t struct kernel_t
{ {
struct syscall_t syscall_table[MAX_SYSCALL_ID]; struct syscall_t syscall_table[MAX_SYSCALL_ID];
struct priority_queue_t priority_queue; struct priority_queue_t priority_queue;
struct avltree_t *port_table;
struct avltree_t *process_table; struct avltree_t *process_table;
struct process_t *active_process; struct process_t *active_process;
int next_pid; int next_pid;
@@ -62,7 +84,7 @@ void kernel_initialize(struct boot_info_t *boot_info);
int set_syscall(int id, int arg_count, int pid, void *func_ptr); int set_syscall(int id, int arg_count, int pid, void *func_ptr);
size_t do_syscall(enum syscall_id_t id, syscall_arg_t arg1, syscall_arg_t arg2, syscall_arg_t arg3); size_t do_syscall(enum syscall_id_t id, syscall_arg_t arg1, syscall_arg_t arg2, syscall_arg_t arg3, void *pc, void *stack, unsigned long flags);
int load_module(struct module_t *module); int load_module(struct module_t *module);
@@ -70,14 +92,20 @@ int active_process();
int add_process(void *program_entry, int priority, physaddr_t address_space); int add_process(void *program_entry, int priority, physaddr_t address_space);
struct process_context_t *next_process(struct process_context_t *prev_state); struct process_context_t *next_process();
int terminate_process(size_t process_id); int terminate_process(size_t process_id);
/* int store_active_context(struct process_context_t *context, size_t size);
int accept_message(size_t process_id, struct message_t *message);
int send_message(size_t process_id, const struct message_t *message); struct process_context_t *get_active_context();
*/
int open_port(unsigned long id);
int close_port(unsigned long id);
int send_message(int recipient, struct message_t *message, int flags);
int receive_message(struct message_t *buffer, int flags);
void panic(const char *message) __attribute__ ((noreturn)); void panic(const char *message) __attribute__ ((noreturn));

View File

@@ -1,9 +1,26 @@
#pragma once #pragma once
void *initialize_context(void *task_entry); struct process_context_t;
/**
* @brief Allocates a new process context and initializes it with the given
* program counter.
*
* @param task_entry
* @return void*
*/
void *initialize_context(void *pc);
void destroy_context(void *ctx); void destroy_context(void *ctx);
void save_context(struct process_context_t *context, void *ptr); void save_context(struct process_context_t *context);
void load_context(struct process_context_t *context); void load_context(struct process_context_t *context);
void set_context_pc(struct process_context_t *context, void *pc);
void set_context_stack(struct process_context_t *context, void *stack);
void set_context_flags(struct process_context_t *context, unsigned long flags);
void set_context_return(struct process_context_t *context, unsigned long value);

View File

@@ -13,7 +13,7 @@ enum page_flag_t
PAGE_COPY_ON_WRITE = 1 << 18 PAGE_COPY_ON_WRITE = 1 << 18
}; };
const size_t page_table_levels; extern const size_t page_table_levels;
/** /**
* @brief * @brief

View File

@@ -70,7 +70,7 @@ int construct_priority_queue(struct priority_queue_t *queue, int capacity);
* @param queue * @param queue
* @return void* * @return void*
*/ */
void *extract_min(struct priority_queue_t *queue); void *priorityqueue_extract_min(struct priority_queue_t *queue);
/** /**
* @brief Inserts a new object onto the queue. * @brief Inserts a new object onto the queue.
@@ -80,7 +80,7 @@ void *extract_min(struct priority_queue_t *queue);
* @param priority * @param priority
* @return int * @return int
*/ */
int queue_insert(struct priority_queue_t *queue, void *value, int priority); int priorityqueue_insert(struct priority_queue_t *queue, void *value, int priority);
/** /**
* @brief Removes the object with a matching value from the queue. * @brief Removes the object with a matching value from the queue.
@@ -89,4 +89,4 @@ int queue_insert(struct priority_queue_t *queue, void *value, int priority);
* @param value * @param value
* @return int * @return int
*/ */
int queue_remove(struct priority_queue_t *queue, void *value); int priorityqueue_remove(struct priority_queue_t *queue, void *value);

60
include/queue.h Normal file
View File

@@ -0,0 +1,60 @@
#pragma once
struct queue_node_t;
/**
* @brief A last-in-first-out queue.
*
*/
struct queue_t
{
/**
* @brief The number of objects stored in this queue.
*
*/
unsigned int count;
/**
* @brief A link to the first node stored in this queue.
*
*/
struct queue_node_t *first;
/**
* @brief A link to the last node stored in this queue.
*
*/
struct queue_node_t *last;
};
/**
* @brief Initializes a queue structure.
*
* @param queue
*/
void queue_construct(struct queue_t *queue);
/**
* @brief Inserts a new item at the end of the queue.
*
* @param queue
* @param ptr
*/
void queue_insert(struct queue_t *queue, void *ptr);
/**
* @brief Removes the next item from the queue and returns it.
*
* @param queue
* @return void*
*/
void *queue_get_next(struct queue_t *queue);
/**
* @brief Returns the next item on the queue without removing it.
*
* @param queue
* @return void*
*/
void *queue_peek(struct queue_t *queue);

View File

@@ -9,7 +9,11 @@ enum syscall_id_t
SYSCALL_YIELD, SYSCALL_YIELD,
SYSCALL_MMAP, SYSCALL_MMAP,
SYSCALL_MUNMAP, SYSCALL_MUNMAP,
SYSCALL_TERMINATE_SELF SYSCALL_TERMINATE_SELF,
SYSCALL_SEND,
SYSCALL_RECEIVE,
SYSCALL_OPEN_PORT,
SYSCALL_CLOSE_PORT
}; };
typedef union typedef union
@@ -48,3 +52,11 @@ size_t mmap(syscall_arg_t location, syscall_arg_t length, syscall_arg_t flags);
size_t munmap(syscall_arg_t location, syscall_arg_t length); size_t munmap(syscall_arg_t location, syscall_arg_t length);
size_t terminate_self(); size_t terminate_self();
size_t send(syscall_arg_t recipient, syscall_arg_t message, syscall_arg_t flags);
size_t receive(syscall_arg_t buffer, syscall_arg_t flags);
size_t openport(syscall_arg_t id);
size_t closeport(syscall_arg_t id);

View File

@@ -1,6 +1,6 @@
menuentry "Quark OS" { menuentry "Quark OS" {
multiboot2 /apps/quark-kernel multiboot2 /apps/quark-kernel
module2 /apps/quark-testmod module2 /apps/quark-testserver
module2 /apps/quark-testmod module2 /apps/quark-testmod
module2 /apps/quark-testmod module2 /apps/quark-testmod
module2 /apps/quark-testmod module2 /apps/quark-testmod

View File

@@ -1,5 +1,5 @@
noinst_PROGRAMS = quark-kernel noinst_PROGRAMS = quark-kernel
quark_kernel_SOURCES = kernel.c mmgr.c priorityqueue.c stdio.c string.c elf.c syscalls.c heap.c memmap.c avltree.c quark_kernel_SOURCES = kernel.c mmgr.c priorityqueue.c stdio.c string.c elf.c syscalls.c heap.c memmap.c avltree.c queue.c
quark_kernel_LDADD = -lgcc quark_kernel_LDADD = -lgcc
quark_kernel_CFLAGS = -I$(top_srcdir)/include -ffreestanding -mgeneral-regs-only -O0 -Wall -ggdb quark_kernel_CFLAGS = -I$(top_srcdir)/include -ffreestanding -mgeneral-regs-only -O0 -Wall -ggdb
quark_kernel_LDFLAGS = -nostdlib quark_kernel_LDFLAGS = -nostdlib
@@ -14,7 +14,7 @@ quark_kernel_SOURCES += x86/paging.c \
x86/isr.c \ x86/isr.c \
x86/msr.c \ x86/msr.c \
x86/context.c \ x86/context.c \
x86/contextswitch.S \ x86/load_context.S \
x86/preempt.S \ x86/preempt.S \
x86/quark_x86.c \ x86/quark_x86.c \
x86/entry.S x86/entry.S

View File

@@ -9,7 +9,8 @@
struct heap_t struct heap_t
{ {
struct heap_node_t *base; void *base;
struct heap_node_t *header;
size_t heap_size; size_t heap_size;
size_t block_size; size_t block_size;
size_t tree_height; size_t tree_height;
@@ -47,7 +48,7 @@ size_t find_free_block(struct heap_t *heap, size_t height)
} }
for(size_t index = 1 << (heap->tree_height - height); index < 1 << (heap->tree_height - height + 1); index++) for(size_t index = 1 << (heap->tree_height - height); index < 1 << (heap->tree_height - height + 1); index++)
{ {
if(heap->base[index].state == AVAIL) if(heap->header[index].state == AVAIL)
{ {
return index; return index;
} }
@@ -55,9 +56,9 @@ size_t find_free_block(struct heap_t *heap, size_t height)
size_t index = find_free_block(heap, height + 1); size_t index = find_free_block(heap, height + 1);
if(index) if(index)
{ {
heap->base[index].state = UNAVAIL; heap->header[index].state = UNAVAIL;
heap->base[index << 1].state = AVAIL; heap->header[index << 1].state = AVAIL;
heap->base[(index << 1) ^ 1].state = AVAIL; heap->header[(index << 1) ^ 1].state = AVAIL;
} }
return index << 1; return index << 1;
} }
@@ -65,7 +66,7 @@ size_t find_free_block(struct heap_t *heap, size_t height)
int map_region(struct heap_t *heap, size_t height, size_t index) int map_region(struct heap_t *heap, size_t height, size_t index)
{ {
int status = 0; int status = 0;
if(heap->base[index].mapped == 0) if(heap->header[index].mapped == 0)
{ {
if(height > 0) if(height > 0)
{ {
@@ -83,48 +84,49 @@ int map_region(struct heap_t *heap, size_t height, size_t index)
status = map_page(ptr, reserve_page(), PAGE_RW); status = map_page(ptr, reserve_page(), PAGE_RW);
} }
} }
heap->base[index].mapped = 1; heap->header[index].mapped = 1;
} }
return status; return status;
} }
int heap_contruct(struct heap_t *heap, void *base, size_t heap_size, size_t block_size) int heap_contruct(struct heap_t *heap, void *base, void *start, size_t heap_size, size_t block_size)
{ {
heap->base = (struct heap_node_t*) base; heap->base = base;
heap->header = (struct heap_node_t*) start;
heap->heap_size = heap_size; heap->heap_size = heap_size;
heap->block_size = block_size; heap->block_size = block_size;
heap->tree_height = ilog2(heap_size / block_size); heap->tree_height = ilog2(heap_size / block_size);
size_t header_size = (heap_size / block_size) << 1; size_t header_size = (heap_size / block_size) << 1;
for(size_t i = 1; i < (heap_size / block_size) * 2; i++) for(size_t i = 1; i <= (heap_size / block_size) * 2; i++)
{ {
int flags = page_type((void*) heap->base + i); int flags = page_type((void*) heap->header + i);
if((flags & PAGE_PRESENT) == 0) if((flags & PAGE_PRESENT) == 0)
{ {
int status = map_page((void*)heap->base + i, reserve_page(), PAGE_RW); int status = map_page((void*)heap->header + i, reserve_page(), PAGE_RW);
if(status != S_OK) if(status != S_OK)
{ {
return status; return status;
} }
} }
heap->base[i].state = UNAVAIL; heap->header[i].state = UNAVAIL;
heap->base[i].mapped = 0; heap->header[i].mapped = 0;
} }
for(size_t i = 0; i < heap_size / block_size; i++) for(size_t i = 0; i < heap_size / block_size; i++)
{ {
if(block_size * i >= header_size) if(block_size * i >= header_size + (start - base))
{ {
size_t index = i + (1 << heap->tree_height); size_t index = i + (1 << heap->tree_height);
heap->base[index].state = AVAIL; heap->header[index].state = AVAIL;
for(; index > 1 && heap->base[index ^ 1].state == 0; index >>= 1) for(; index > 1 && heap->header[index ^ 1].state == 0; index >>= 1)
{ {
heap->base[index].state = UNAVAIL; heap->header[index].state = UNAVAIL;
heap->base[index ^ 1].state = UNAVAIL; heap->header[index ^ 1].state = UNAVAIL;
heap->base[index >> 1].state = AVAIL; heap->header[index >> 1].state = AVAIL;
} }
} }
else else
{ {
heap->base[i + (1 << heap->tree_height)].state = UNAVAIL; heap->header[i + (1 << heap->tree_height)].state = UNAVAIL;
} }
} }
return S_OK; return S_OK;
@@ -138,7 +140,7 @@ void *heap_allocate(struct heap_t *heap, size_t size)
size_t index = find_free_block(heap, height); size_t index = find_free_block(heap, height);
if(index) if(index)
{ {
heap->base[index].state = ALLOCATED; heap->header[index].state = ALLOCATED;
void *ptr = (void*) ((size_t) heap->base + (heap->block_size << height) * (index - (1 << (heap->tree_height - height)))); void *ptr = (void*) ((size_t) heap->base + (heap->block_size << height) * (index - (1 << (heap->tree_height - height))));
map_region(heap, height, index); map_region(heap, height, index);
return ptr; return ptr;
@@ -150,19 +152,19 @@ void heap_free(struct heap_t *heap, void *ptr)
{ {
size_t offset = (size_t) ptr - (size_t) heap->base; size_t offset = (size_t) ptr - (size_t) heap->base;
size_t index = (offset / heap->block_size) + (1 << heap->tree_height); size_t index = (offset / heap->block_size) + (1 << heap->tree_height);
for(; index > 0 && heap->base[index].state == UNAVAIL; index >>= 1); for(; index > 0 && heap->header[index].state == UNAVAIL; index >>= 1);
heap->base[index].state = AVAIL; heap->header[index].state = AVAIL;
for(; index > 1 && heap->base[index ^ 1].state == AVAIL; index >>= 1) for(; index > 1 && heap->header[index ^ 1].state == AVAIL; index >>= 1)
{ {
heap->base[index].state = UNAVAIL; heap->header[index].state = UNAVAIL;
heap->base[index ^ 1].state = UNAVAIL; heap->header[index ^ 1].state = UNAVAIL;
heap->base[index >> 1].state = AVAIL; heap->header[index >> 1].state = AVAIL;
} }
} }
int kminit(void *base, size_t heap_size, size_t block_size) int kminit(void *base, void* start, size_t heap_size, size_t block_size)
{ {
return heap_contruct(&system_heap, base, heap_size, block_size); return heap_contruct(&system_heap, base, start, heap_size, block_size);
} }
void *kmalloc(size_t size) void *kmalloc(size_t size)

View File

@@ -18,7 +18,7 @@ void kernel_initialize(struct boot_info_t *boot_info)
{ {
insert_region(&boot_info->map, (physaddr_t)&_kernel_pstart, (physaddr_t)&_kernel_pend - (physaddr_t)&_kernel_pstart, M_UNAVAILABLE); insert_region(&boot_info->map, (physaddr_t)&_kernel_pstart, (physaddr_t)&_kernel_pend - (physaddr_t)&_kernel_pstart, M_UNAVAILABLE);
initialize_page_stack(&boot_info->map, (physaddr_t*)&_kernel_end); initialize_page_stack(&boot_info->map, (physaddr_t*)&_kernel_end);
kminit(page_stack_top(), 0xFFC00000 - (size_t)page_stack_top(), page_size); kminit(&_kernel_start, page_stack_top(), 0xFFC00000 - (size_t)&_kernel_start, 64);
initialize_screen(); initialize_screen();
printf("***%s***\n", PACKAGE_STRING); printf("***%s***\n", PACKAGE_STRING);
printf("Type\t\tLocation\t\tSize\n"); printf("Type\t\tLocation\t\tSize\n");
@@ -30,6 +30,7 @@ void kernel_initialize(struct boot_info_t *boot_info)
kernel.active_process = NULL; kernel.active_process = NULL;
kernel.next_pid = 1; kernel.next_pid = 1;
kernel.process_table = NULL; kernel.process_table = NULL;
kernel.port_table = NULL;
if(construct_priority_queue(&kernel.priority_queue, 512) != S_OK) if(construct_priority_queue(&kernel.priority_queue, 512) != S_OK)
{ {
panic("Failed to construct priority queue."); panic("Failed to construct priority queue.");
@@ -38,6 +39,10 @@ void kernel_initialize(struct boot_info_t *boot_info)
set_syscall(SYSCALL_TEST, 1, 0, test_syscall); set_syscall(SYSCALL_TEST, 1, 0, test_syscall);
set_syscall(SYSCALL_MMAP, 3, 0, mmap); set_syscall(SYSCALL_MMAP, 3, 0, mmap);
set_syscall(SYSCALL_MUNMAP, 2, 0, munmap); set_syscall(SYSCALL_MUNMAP, 2, 0, munmap);
set_syscall(SYSCALL_SEND, 3, 0, send);
set_syscall(SYSCALL_RECEIVE, 2, 0, receive);
set_syscall(SYSCALL_OPEN_PORT, 1, 0, openport);
set_syscall(SYSCALL_CLOSE_PORT, 1, 0, closeport);
for(int i = 0; i < boot_info->module_count; i++) for(int i = 0; i < boot_info->module_count; i++)
{ {
if(load_module(&boot_info->modules[i]) != S_OK) if(load_module(&boot_info->modules[i]) != S_OK)
@@ -51,13 +56,8 @@ void kernel_initialize(struct boot_info_t *boot_info)
panic("Failed to initialize interrupts."); panic("Failed to initialize interrupts.");
} }
/*asm("mov $281, %ax;"
"mov %ax, %ds");
asm("hlt");*/
irq_enable(); irq_enable();
load_context(next_process(NULL)); load_context(next_process());
} }
int set_syscall(int id, int arg_count, int pid, void *func_ptr) int set_syscall(int id, int arg_count, int pid, void *func_ptr)
@@ -89,7 +89,7 @@ int set_syscall(int id, int arg_count, int pid, void *func_ptr)
return S_OK; return S_OK;
} }
size_t do_syscall(enum syscall_id_t id, syscall_arg_t arg1, syscall_arg_t arg2, syscall_arg_t arg3) size_t do_syscall(enum syscall_id_t id, syscall_arg_t arg1, syscall_arg_t arg2, syscall_arg_t arg3, void *pc, void *stack, unsigned long flags)
{ {
if(id < 0 || id > MAX_SYSCALL_ID) if(id < 0 || id > MAX_SYSCALL_ID)
{ {
@@ -111,6 +111,9 @@ size_t do_syscall(enum syscall_id_t id, syscall_arg_t arg1, syscall_arg_t arg2,
paging_load_address_space(callee->page_table); paging_load_address_space(callee->page_table);
switched_address_space = true; switched_address_space = true;
} }
set_context_pc(kernel.active_process->ctx, pc);
set_context_stack(kernel.active_process->ctx, stack);
set_context_flags(kernel.active_process->ctx, flags);
size_t result; size_t result;
switch(kernel.syscall_table[id].arg_count) switch(kernel.syscall_table[id].arg_count)
{ {
@@ -197,7 +200,7 @@ int active_process()
} }
else else
{ {
return kernel.active_process->resource_id; return kernel.active_process->pid;
} }
} }
@@ -210,28 +213,31 @@ int add_process(void *program_entry, int priority, physaddr_t address_space)
} }
struct process_context_t *initial_context = initialize_context(program_entry); struct process_context_t *initial_context = initialize_context(program_entry);
new_process->priority = priority; new_process->priority = priority;
new_process->resource_id = kernel.next_pid; new_process->pid = kernel.next_pid;
new_process->page_table = address_space; new_process->page_table = address_space;
new_process->state = initial_context; new_process->state = PROCESS_ACTIVE;
kernel.process_table = avl_insert(kernel.process_table, new_process->resource_id, new_process); new_process->message_buffer = NULL;
queue_insert(&kernel.priority_queue, new_process, new_process->priority); new_process->ctx = initial_context;
queue_construct(&new_process->sending_queue);
queue_construct(&new_process->message_queue);
kernel.process_table = avl_insert(kernel.process_table, new_process->pid, new_process);
priorityqueue_insert(&kernel.priority_queue, new_process, new_process->priority);
kernel.next_pid++; kernel.next_pid++;
return new_process->resource_id; return new_process->pid;
} }
struct process_context_t *next_process(struct process_context_t *prev_state) struct process_context_t *next_process()
{ {
if(prev_state != NULL) if(kernel.active_process != NULL)
{ {
kernel.active_process->state = prev_state; priorityqueue_insert(&kernel.priority_queue, kernel.active_process, kernel.active_process->priority);
queue_insert(&kernel.priority_queue, kernel.active_process, kernel.active_process->priority);
} }
kernel.active_process = extract_min(&kernel.priority_queue); kernel.active_process = priorityqueue_extract_min(&kernel.priority_queue);
if(kernel.active_process != NULL) if(kernel.active_process != NULL)
{ {
paging_load_address_space(kernel.active_process->page_table); paging_load_address_space(kernel.active_process->page_table);
printf("entering process %08x cr3=%08x state=%08x.\n", kernel.active_process, kernel.active_process->page_table, kernel.active_process->state); printf("entering process %08x cr3=%08x ctx=%08x.\n", kernel.active_process->pid, kernel.active_process->page_table, kernel.active_process->ctx);
return kernel.active_process->state; return kernel.active_process->ctx;
} }
panic("no processes available to enter!"); panic("no processes available to enter!");
} }
@@ -248,62 +254,155 @@ int terminate_process(size_t process_id)
kernel.active_process = NULL; kernel.active_process = NULL;
} }
kernel.process_table = avl_remove(kernel.process_table, process_id); kernel.process_table = avl_remove(kernel.process_table, process_id);
queue_remove(&kernel.priority_queue, process); priorityqueue_remove(&kernel.priority_queue, process);
destroy_context(process->state); destroy_context(process->ctx);
kfree(process); kfree(process);
return S_OK; return S_OK;
} }
/* int store_active_context(struct process_context_t *context, size_t size)
int accept_message(size_t process_id, struct message_t *message)
{ {
if(message == NULL) if(kernel.active_process != NULL && kernel.active_process->ctx != NULL)
{ {
return S_NULL_POINTER; memcpy(kernel.active_process->ctx, context, size);
return S_OK;
} }
else if(kernel->resource_table->limit >= process_id) else
{ {
return S_OUT_OF_BOUNDS; return S_DOESNT_EXIST;
} }
else if(kernel->resource_table->array[process_id].type != RESOURCE_PROCESS) }
int open_port(unsigned long id)
{
if(avl_get(kernel.port_table, id) != NULL)
{ {
return S_INVALID_ARGUMENT; return S_EXISTS;
} }
struct process_t *process = &kernel->resource_table->array[process_id].process; printf("opening port %i -> %i\n", id, kernel.active_process->pid);
process->state = PROCESS_WAITING; struct port_t *port = kmalloc(sizeof(struct port_t));
process->message = message; port->id = id;
if(kernel->active_process == process) port->owner_pid = kernel.active_process->pid;
{ kernel.port_table = avl_insert(kernel.port_table, id, port);
kernel->active_process = NULL;
}
queue_remove(kernel->priority_queue, process);
return S_OK; return S_OK;
} }
int send_message(size_t process_id, const struct message_t *message) int close_port(unsigned long id)
{ {
if(kernel == NULL || message == NULL) struct port_t *port = avl_get(kernel.port_table, id);
if(port == NULL)
{ {
return S_NULL_POINTER; return S_DOESNT_EXIST;
} }
else if(kernel->resource_table->limit >= process_id) else if(port->owner_pid != kernel.active_process->pid)
{
return S_OUT_OF_BOUNDS;
}
else if(kernel->resource_table->array[process_id].type != RESOURCE_PROCESS)
{ {
return S_INVALID_ARGUMENT; return S_INVALID_ARGUMENT;
} }
struct process_t *process = &kernel->resource_table->array[process_id].process; printf("closing port %i attached to %i\n", id, kernel.active_process->pid);
if(process->state != PROCESS_WAITING || process->message == NULL) kernel.port_table = avl_remove(kernel.port_table, id);
{ kfree(port);
return S_BUSY; return S_OK;
} }
queue_insert(kernel->priority_queue, kernel->active_process);
struct message_t buffer = *message; int send_message(int recipient, struct message_t *message, int flags)
{
int op_type = flags & IO_OP;
int dest_type = flags & IO_RECIPIENT_TYPE;
if((flags & ~(IO_OP | IO_RECIPIENT_TYPE)) != 0 || dest_type >= IO_MAILBOX)
{
printf("Invalid flags on send_message\n");
return S_INVALID_ARGUMENT;
}
if(dest_type == IO_PORT)
{
struct port_t *port = avl_get(kernel.port_table, recipient);
if(port != NULL)
{
recipient = port->owner_pid;
}
else
{
printf("Port %i does not exist\n", recipient);
return S_DOESNT_EXIST;
}
}
struct process_t *dest = avl_get(kernel.process_table, recipient);
if(dest == NULL)
{
return S_DOESNT_EXIST;
}
else if(dest->message_buffer != NULL)
{
printf("Sending message directly from %i to %i\n", kernel.active_process->pid, dest->pid);
struct message_t kernel_buffer;
memcpy(&kernel_buffer, message, sizeof(struct message_t));
kernel_buffer.sender = kernel.active_process->pid;
paging_load_address_space(dest->page_table);
memcpy(dest->message_buffer, &kernel_buffer, sizeof(struct message_t));
paging_load_address_space(kernel.active_process->page_table);
dest->message_buffer = NULL;
dest->state = PROCESS_ACTIVE;
set_context_return(dest->ctx, S_OK);
priorityqueue_insert(&kernel.priority_queue, dest, dest->priority);
return S_OK;
}
else if(op_type == IO_ASYNC)
{
printf("Queueing message from %i to %i\n", kernel.active_process->pid, dest->pid);
struct message_t *queued_msg = kmalloc(sizeof(struct message_t));
if(queued_msg == NULL)
{
return S_OUT_OF_MEMORY;
}
memcpy(queued_msg, message, sizeof(struct message_t));
queue_insert(&dest->message_queue, queued_msg);
return S_OK;
}
else
{
printf("Queueing process %i to %i\n", kernel.active_process->pid, dest->pid);
queue_insert(&dest->sending_queue, kernel.active_process);
kernel.active_process->state = PROCESS_SENDING;
kernel.active_process = NULL;
load_context(next_process());
}
}
int receive_message(struct message_t *buffer, int flags)
{
if(kernel.active_process->sending_queue.count > 0)
{
struct message_t kernel_buffer;
struct process_t *sender = queue_get_next(&kernel.active_process->sending_queue);
paging_load_address_space(sender->page_table);
memcpy(&kernel_buffer, &sender->message_buffer, sizeof(struct message_t));
kernel_buffer.sender = sender->pid;
paging_load_address_space(kernel.active_process->page_table);
memcpy(buffer, &kernel_buffer, sizeof(struct message_t));
sender->state = PROCESS_ACTIVE;
set_context_return(sender->ctx, S_OK);
priorityqueue_insert(&kernel.priority_queue, sender, sender->priority);
return S_OK;
}
else if(kernel.active_process->message_queue.count > 0)
{
struct message_t *queued_msg = queue_get_next(&kernel.active_process->message_queue);
memcpy(buffer, queued_msg, sizeof(struct message_t));
kfree(queued_msg);
return S_OK;
}
else if((flags & IO_OP) == IO_ASYNC)
{
return S_DOESNT_EXIST;
}
else
{
kernel.active_process->message_buffer = buffer;
kernel.active_process->state = PROCESS_REQUESTING;
kernel.active_process = NULL;
load_context(next_process());
}
} }
*/
void panic(const char *message) void panic(const char *message)
{ {

View File

@@ -38,7 +38,7 @@ int construct_priority_queue(struct priority_queue_t *queue, int capacity)
return S_OK; return S_OK;
} }
void *extract_min(struct priority_queue_t *queue) void *priorityqueue_extract_min(struct priority_queue_t *queue)
{ {
if(queue->size == 0) if(queue->size == 0)
return NULL; return NULL;
@@ -49,7 +49,7 @@ void *extract_min(struct priority_queue_t *queue)
return value; return value;
} }
int queue_insert(struct priority_queue_t *queue, void *value, int priority) int priorityqueue_insert(struct priority_queue_t *queue, void *value, int priority)
{ {
if(queue->size == queue->capacity) if(queue->size == queue->capacity)
return S_OUT_OF_MEMORY; return S_OUT_OF_MEMORY;
@@ -65,7 +65,7 @@ int queue_insert(struct priority_queue_t *queue, void *value, int priority)
return S_OK; return S_OK;
} }
int queue_remove(struct priority_queue_t *queue, void *value) int priorityqueue_remove(struct priority_queue_t *queue, void *value)
{ {
for(size_t i = 0; i < queue->size; i++) for(size_t i = 0; i < queue->size; i++)
{ {

60
src/queue.c Normal file
View File

@@ -0,0 +1,60 @@
#include "queue.h"
#include "heap.h"
struct queue_node_t
{
/**
* @brief Pointer to the user-defined object this node refers to.
*
*/
void *ptr;
/**
* @brief Link to the next node in the queue.
*
*/
struct queue_node_t *next;
};
void queue_construct(struct queue_t *queue)
{
queue->first = NULL;
queue->last = NULL;
queue->count = 0;
}
void queue_insert(struct queue_t *queue, void *ptr)
{
struct queue_node_t *node = kmalloc(sizeof(struct queue_node_t));
node->ptr = ptr;
node->next = NULL;
if(queue->last == NULL)
{
queue->first = queue->last = node;
}
else
{
queue->last->next = node;
queue->last = node;
}
queue->count++;
}
void *queue_get_next(struct queue_t *queue)
{
struct queue_node_t *node = queue->first;
queue->first = node->next;
if(queue->first == NULL)
{
queue->last = NULL;
}
queue->count--;
void *ptr = node->ptr;
kfree(node);
return ptr;
}
void *queue_peek(struct queue_t *queue)
{
return queue->first->ptr;
}

View File

@@ -91,3 +91,23 @@ size_t terminate_self()
{ {
return terminate_process(active_process()); return terminate_process(active_process());
} }
size_t send(syscall_arg_t recipient, syscall_arg_t message, syscall_arg_t flags)
{
return send_message(recipient.unsigned_int, message.ptr, flags.unsigned_int);
}
size_t receive(syscall_arg_t buffer, syscall_arg_t flags)
{
return receive_message(buffer.ptr, flags.unsigned_int);
}
size_t openport(syscall_arg_t id)
{
return open_port(id.unsigned_int);
}
size_t closeport(syscall_arg_t id)
{
return close_port(id.unsigned_int);
}

View File

@@ -1,35 +1,65 @@
#include "platform/context.h" #include "platform/context.h"
#include "kernel.h"
#include "mmgr.h" #include "mmgr.h"
#include "heap.h"
#include "string.h" #include "string.h"
#include "system.h"
#include "x86/processstate.h" #include "x86/processstate.h"
void *initialize_context(void *task_entry) void *initialize_context(void *task_entry)
{ {
/* physaddr_t stack_frame = reserve_page();
* TODO: this implementation is a goddamn mess. if(stack_frame % page_size != 0)
* Stack pointer is hardcoded, and the stack isn't resizable. {
* PCB pointer is just a constant. return NULL;
*/ }
map_page(NULL, reserve_page(), PAGE_RW); map_page((void*)&_kernel_start - page_size, stack_frame, PAGE_RW | PAGE_USERMODE);
map_page((void*)0xFF3FF000, reserve_page(), PAGE_RW | PAGE_USERMODE); unmap_page((void*)&_kernel_start - (2 * page_size));
unmap_page((void*)0xFF3FE000); struct process_context_t *context = kmalloc(sizeof(struct process_context_t));
if(context != NULL)
{
memset(context, 0, sizeof(struct process_context_t));
uint32_t flags; uint32_t flags;
asm("pushf; " asm("pushf; "
"mov (%%esp), %0; " "mov (%%esp), %0; "
"popf; " "popf; "
: "=r"(flags)); : "=r"(flags));
struct process_context_t *state = (struct process_context_t*)PCB_LOCATION; context->cs = 0x1B;
memset(NULL, 0, page_size); context->eip = (uint32_t)task_entry;
state->cs = 0x1B; context->flags = (flags & ~0xFD) | 0x200;
state->eip = (uint32_t)task_entry; context->ss = 0x23;
state->flags = (flags & ~0xFD) | 0x200; context->esp = &_kernel_start;
state->ss = 0x23; context->ebp = &_kernel_start;
state->esp = 0xFF400000; }
state->ebp = 0xFF400000; return (void*)context;
return (void*)state;
} }
void destroy_context(void *ctx) void destroy_context(void *ctx)
{ {
// Nothing to do... kfree(ctx);
}
void save_context(struct process_context_t *context)
{
store_active_context(context, sizeof(*context));
}
void set_context_pc(struct process_context_t *context, void *pc)
{
context->eip = pc;
}
void set_context_stack(struct process_context_t *context, void *stack)
{
context->esp = stack;
}
void set_context_flags(struct process_context_t *context, unsigned long flags)
{
context->flags = flags;
}
void set_context_return(struct process_context_t *context, unsigned long value)
{
context->eax = value;
} }

View File

@@ -1,67 +0,0 @@
.section .text
/*
* save_context(struct process_state_t *context, struct interrupt_frame_t *frame)
*/
.global save_context
.type save_context, @function
save_context:
push %edi
push %esi
push %eax
mov $0x10, %ax
mov %ax, %ds
mov %ax, %es
mov %ax, %fs
mov %ax, %gs
mov 16(%esp), %eax
mov %ebx, 0x04(%eax)
mov %ecx, 0x08(%eax)
mov %edx, 0x0C(%eax)
mov %edi, 0x10(%eax)
mov %esi, 0x14(%eax)
mov %ebp, 0x18(%eax)
mov %eax, %edi
pop %eax
mov %eax, 0x00(%edi)
mov 16(%esp), %esi
mov 0x10(%esi), %eax
mov %eax, 0x1C(%edi)
mov 0x0C(%esi), %eax
mov %eax, 0x20(%edi)
mov 0x08(%esi), %eax
mov %eax, 0x2C(%edi)
mov 0x04(%esi), %eax
mov %eax, 0x24(%edi)
mov 0x00(%esi), %eax
mov %eax, 0x28(%edi)
pop %esi
pop %edi
ret
/*
* load_context(struct process_state_t *context)
*/
.global load_context
.type load_context, @function
load_context:
mov 4(%esp), %eax
push 0x1C(%eax)
push 0x20(%eax)
push 0x2C(%eax)
push 0x24(%eax)
push 0x28(%eax)
push 0x00(%eax)
mov 0x04(%eax), %ebx
mov 0x08(%eax), %ecx
mov 0x0C(%eax), %edx
mov 0x10(%eax), %edi
mov 0x14(%eax), %esi
mov 0x18(%eax), %ebp
mov 0x1C(%eax), %ax
mov %ax, %ds
mov %ax, %es
mov %ax, %fs
mov %ax, %gs
pop %eax
iret

28
src/x86/load_context.S Normal file
View File

@@ -0,0 +1,28 @@
.section .text
/*
* load_context(struct process_state_t *context)
*/
.global load_context
.type load_context, @function
load_context:
mov 4(%esp), %eax
push 0x1C(%eax)
push 0x20(%eax)
push 0x2C(%eax)
push 0x24(%eax)
push 0x28(%eax)
push 0x00(%eax)
mov 0x04(%eax), %ebx
mov 0x08(%eax), %ecx
mov 0x0C(%eax), %edx
mov 0x10(%eax), %edi
mov 0x14(%eax), %esi
mov 0x18(%eax), %ebp
mov 0x1C(%eax), %ax
mov %ax, %ds
mov %ax, %es
mov %ax, %fs
mov %ax, %gs
pop %eax
iret

View File

@@ -4,6 +4,13 @@
.type isr_syscall, @function .type isr_syscall, @function
isr_syscall: isr_syscall:
cli cli
mov %esp, %ebp
mov 8(%ebp), %esi
push %esi
mov 12(%ebp), %esi
push %esi
mov (%ebp), %esi
push %esi
push %edx push %edx
push %ecx push %ecx
push %ebx push %ebx
@@ -14,7 +21,7 @@ isr_syscall:
mov %ax, %fs mov %ax, %fs
mov %ax, %gs mov %ax, %gs
call do_syscall call do_syscall
add $0x10, %esp mov %ebp, %esp
mov $0x23, %cx mov $0x23, %cx
mov %cx, %ds mov %cx, %ds
mov %cx, %es mov %cx, %es
@@ -25,14 +32,63 @@ isr_syscall:
.global isr_preempt .global isr_preempt
.type isr_preempt, @function .type isr_preempt, @function
isr_preempt: isr_preempt:
// Disable interrupts for now
cli cli
// Save process's base pointer
push %ebp
// Save base pointer; it points to the interrupt stack frame
mov %esp, %ebp
add $4, %ebp
// Save EAX on stack so it can be used as a buffer
push %eax
// Load EFLAGS, then push it onto stack
mov 8(%ebp), %eax
push %eax
// Load EIP, then push it onto stack
mov (%ebp), %eax
push %eax
// Load CS, then push it onto stack
mov 4(%ebp), %eax
push %eax
// Load ESP, then push it onto stack
mov 12(%ebp), %eax
push %eax
// Load SS, then push it onto stack
mov 16(%ebp), %eax
push %eax
// Load EBP, then push it onto stack
mov -4(%ebp), %eax
push %eax
// Push GP registers onto stack
push %esi
push %edi
push %edx
push %ecx
push %ebx
// Restore saved value of EAX, then push it onto stack
mov -8(%ebp), %eax
push %eax
// Push pointer to the process context saved on the stack
push %esp push %esp
push $0x800
call save_context call save_context
add $8, %esp mov %ebp, %esp
push $0x800
call next_process call next_process
add $8, %esp
push %eax push %eax
call load_context call load_context