blob: d89381325182952e32857b8140a6ff9d3e3c9320 (
plain)
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
|
#include <kernel/sched.h>
#include <kernel/container_of.h>
#include <kernel/data/list.h>
#include <kernel/string.h>
#include <kernel/kmalloc.h>
#include <stddef.h>
static struct list_head *ready_queue;
static struct list_head *wait_queue;
static struct task_block *cur = NULL;
static long int next_id = 1;
static int scheduler_enabled = 0;
static void _enqueue(struct list_head *queue, struct task_block *task) {
struct list_head *temp = queue;
while (temp->next != NULL)
temp = temp->next;
list_add(&task->list, temp);
}
static struct task_block* _dequeue(struct list_head *queue) {
if (queue->next == NULL)
return NULL;
struct list_head *temp = queue->next;
list_del(temp);
struct task_block *ret = container_of(temp, struct task_block, list);
return ret;
}
void sched_init(void) {
ready_queue = kmalloc(sizeof(struct list_head));
wait_queue = kmalloc(sizeof(struct list_head));
struct task_block *boot_task = init_task(0, 0);
boot_task->threads->state = THREAD_RUNNING;
cur = boot_task;
switch_thread(boot_task->threads, boot_task->threads);
scheduler_enabled = 1;
}
void schedule_next(void) {
if (scheduler_enabled == 0)
return;
struct task_block *task = _dequeue(ready_queue);
if (task == NULL)
return;
if (cur != NULL)
_enqueue(ready_queue, cur);
disable_ints();
switch_thread(cur->threads, task->threads);
enable_ints();
task->threads->state = THREAD_RUNNING;
cur->threads->state = THREAD_READY;
cur = task;
}
struct task_block* init_task(int nice, unsigned int ppid) {
struct task_block *ret = kmalloc(sizeof(struct task_block));
ret->pid = next_id++;
ret->parent_pid = ppid;
ret->nice = nice;
ret->num_threads = 1;
ret->threads = kmalloc(sizeof(struct thread_block));
ret->threads->tid = ret->pid;
ret->threads->tgid = ret->pid;
return ret;
}
|
