1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
#include <types.h>
#include <regs.h>
#include <heap.h>
#include <scheduler.h>
#include <libk/list.h>
#include <libk/stdio.h>
#include <process.h>
#include <timer.h>
#include <panic.h>
process_t process_queue;
process_t *curr_process;
process_t *init_process(uint64_t priv_lvl, uint64_t rip, uint64_t argc,
uint64_t *argv)
{
uint32_t stack_size = 32 * 1024;
uint64_t rsp = (uint64_t)kalloc(stack_size) + stack_size - 8;
regs_t *regs = (regs_t *)(rsp - sizeof(regs_t));
regs->rax = 0;
regs->rbx = 0;
regs->rcx = 0;
regs->rdx = (uint64_t)argv;
regs->rdi = rip;
regs->rsi = argc;
regs->rsp = rsp;
regs->rbp = 0;
regs->r8 = 0;
regs->r9 = 0;
regs->r10 = 0;
regs->r11 = 0;
regs->r12 = 0;
regs->r13 = 0;
regs->r14 = 0;
regs->r15 = 0;
regs->rip = (uint64_t)process_init_wrapper;
regs->rflags = 0x202;
regs->error = 0;
if (priv_lvl == 0) {
regs->cs = 0x8;
regs->ss = 0x10;
regs->seg = 0x0010001000100010;
} else {
regs->cs = 0x23;
regs->ss = 0x1b;
regs->seg = 0x001b001b001b001b;
}
process_t *new_process = (process_t *)kalloc(sizeof(process_t));
new_process->rsp = rsp - sizeof(regs_t);
add_to_list(&new_process->list, &process_queue.list,
process_queue.list.next);
return new_process;
}
void process_init_wrapper(uint64_t rip, uint64_t argc, uint64_t *argv)
{
void (*f)(uint64_t rdi, ...) = (void (*)(uint64_t rdi, ...))rip;
switch (argc) {
case 0:
(*f)(argv[0]);
break;
case 1:
(*f)(argv[0]);
break;
case 2:
(*f)(argv[0], argv[1]);
break;
case 3:
(*f)(argv[0], argv[1], argv[2]);
break;
case 4:
(*f)(argv[0], argv[1], argv[2], argv[3]);
break;
case 5:
(*f)(argv[0], argv[1], argv[2], argv[3], argv[4]);
break;
default:
(*f)(argv[0], argv[1], argv[2], argv[3], argv[4], argv[5]);
break;
}
kfree(argv);
remove_current_process();
}
__attribute__((noreturn)) void idle_thread()
{
for (;;) {
__asm__ __volatile__("pause; hlt;");
}
}
__attribute__((noreturn)) void remove_current_process()
{
free_node(&curr_process->list);
kfree(curr_process);
curr_process = scheduler();
if (curr_process == NULL) {
panic(0, "no processes left\n");
}
restore_context_from_rsp(curr_process->rsp);
}
__attribute__((noreturn)) void context_switch(uint64_t irq_rsp)
{
curr_process->rsp = save_context_from_rsp(irq_rsp);
curr_process = scheduler();
restore_context_from_rsp(curr_process->rsp);
}
|
