big commit

kernel/process.c

@@ -0,0 +1,189 @@
+#include <process.h>
+#include <string.h>
+#include <errno.h>
+#include <kmalloc.h>
+#include <cls.h>
+#include <aal/debug.h>
+#include <page.h>
+
+struct process *create_process(unsigned long user_pc)
+{
+	struct process *proc;
+
+	proc = aal_mc_alloc_pages(1, 0);
+
+	memset(proc, 0, sizeof(struct process));
+
+	aal_mc_init_user_process(&proc->ctx, &proc->uctx,
+	                         ((char *)proc) + PAGE_SIZE, user_pc, 0);
+
+	INIT_LIST_HEAD(&proc->vm_range_list);
+	proc->page_table = aal_mc_pt_create();
+
+	return proc;
+}
+
+void update_process_page_table(struct process *process, struct vm_range *range)
+{
+	unsigned long p, pa = range->phys;
+
+	p = range->start;
+	while (p < range->end) {
+		aal_mc_pt_set_page(process->page_table, (void *)p,
+		                   pa, PTATTR_WRITABLE | PTATTR_USER);
+
+		pa += PAGE_SIZE;
+		p += PAGE_SIZE;
+	}
+}
+
+int add_process_memory_range(struct process *process,
+                             unsigned long start, unsigned long end,
+                             unsigned long phys, unsigned long flag)
+{
+	struct vm_range *range;
+
+	range = kmalloc(sizeof(struct vm_range), 0);
+	if (!range) {
+		return -ENOMEM;
+	}
+	INIT_LIST_HEAD(&range->list);
+	range->start = start;
+	range->end = end;
+	range->phys = phys;
+	range->flag = flag;
+
+	kprintf("range: %lx - %lx => %lx - %lx\n",
+	        range->start, range->end, range->phys, range->phys + 
+	        range->end - range->start);
+
+	update_process_page_table(process, range);
+
+	list_add_tail(&range->list, &process->vm_range_list);
+
+	return 0;
+}
+
+void init_process_stack(struct process *process)
+{
+	char *stack = aal_mc_alloc_pages(1, 0);
+	unsigned long *p = (unsigned long *)(stack + PAGE_SIZE);
+
+	memset(stack, 0, PAGE_SIZE);
+
+	add_process_memory_range(process, USER_END - PAGE_SIZE,
+	                         USER_END,
+	                         virt_to_phys(p), VR_STACK);
+
+	/* TODO: fill with actual value of argc, argv, envp */
+	
+	p[-1] = 0;     /* env: "" */
+	p[-2] = 0x41;  /* argv(0): "a" */
+	p[-3] = USER_END - sizeof(unsigned long); /* envp: END - 8 */
+	p[-4] = 0;     /* argv[1] = NULL */
+	p[-5] = USER_END - sizeof(unsigned long) * 2; /* argv[0] = END - 16 */
+	p[-6] = 1;     /* argc */
+
+	aal_mc_modify_user_context(process->uctx, AAL_UCR_STACK_POINTER,
+	                           USER_END - sizeof(unsigned long) * 6);
+	process->region.stack_end = USER_END;
+	process->region.stack_start = USER_END - PAGE_SIZE;
+}
+
+
+unsigned long extend_process_region(struct process *proc,
+                                    unsigned long start, unsigned long end,
+                                    unsigned long address)
+{
+	unsigned long aligned_end, aligned_new_end;
+	void *p;
+
+	if (!address || address < start || address >= USER_END) {
+		return end;
+	}
+
+	aligned_end = ((end + PAGE_SIZE - 1) & PAGE_MASK);
+
+	if (aligned_end >= address) {
+		return address;
+	}
+
+	aligned_new_end = (address + PAGE_SIZE - 1) & PAGE_MASK;
+	
+	p = allocate_pages((aligned_new_end - aligned_end) >> PAGE_SHIFT,
+	                   0);
+	if (!p) {
+		return end;
+	}
+
+	add_process_memory_range(proc, aligned_end, aligned_new_end,
+	                         virt_to_phys(p), 0);
+	return address;
+}
+
+int remove_process_region(struct process *proc,
+                          unsigned long start, unsigned long end)
+{
+	if ((start & (PAGE_SIZE - 1)) || (end & (PAGE_SIZE - 1))) {
+		return -EINVAL;
+	}
+
+	while (start < end) {
+		aal_mc_pt_clear_page(proc->page_table, (void *)start);
+		start += PAGE_SIZE;
+	}
+
+	return 0;
+}
+
+static void idle(void)
+{
+	while (1) {
+		cpu_enable_interrupt();
+		schedule();
+		cpu_halt();
+	}
+}
+
+void sched_init(void)
+{
+	struct process *idle_process = &cpu_local_var(idle);
+
+	memset(idle_process, 0, sizeof(struct process));
+
+	aal_mc_init_context(&idle_process->ctx, NULL, idle);
+
+	cpu_local_var(next) = idle_process;
+}
+
+void schedule(void)
+{
+	struct cpu_local_var *v = get_this_cpu_local_var();
+	struct process *next, *prev;
+	int switch_ctx = 0;
+
+	cpu_disable_interrupt();
+	if (v->next && v->next != v->current) {
+		prev = v->current;
+		next = v->next;
+
+		switch_ctx = 1;
+
+		v->current = next;
+		v->next = NULL;
+	}
+	cpu_enable_interrupt();
+
+	if (switch_ctx) {
+		kprintf("schedule: %p => %p \n", prev, next);
+		aal_mc_load_page_table(next->page_table);
+
+		if (prev) {
+			aal_mc_switch_context(&prev->ctx, &next->ctx);
+		} else {
+			aal_mc_switch_context(NULL, &next->ctx);
+		}
+	}
+}
+
+