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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
// http://hosted.cjmovie.net/TutMultitask.htm
//
//
#include "kernel.h"
#include "lib/logger/log.h" // logger facilities
#include "lib/buffer/ringbuffer.h"
#include "mem.h"
#include "timer.h"
#include "console.h"
#include "x86.h"
#include "syscalls.h"
#include "fs/fs.h"
#include "fs/ext2.h"
#define FOOLOS_MODULE_NAME "task"
#define MAX_TASKS 10
static volatile int current_task=-2;
static volatile struct task_list_struct
{
bool active;
uint32_t esp; // stack pointer of the task;
uint32_t vmem; // number of virtual memory table to switch to
}task_list[MAX_TASKS];
void task_create(int pid,void(*thread)())
{
task_list[pid].vmem=vmem_new_space_dir();
if(pid==0)x86_set_pdbr(0x0D08000);
if(pid==1)x86_set_pdbr(0x150E000);
unsigned int *stack;
task_list[pid].esp = 0x8248000-4095;
stack = (unsigned int*)task_list[pid].esp+4095; //This makes a pointer to the stack for us
log(FOOLOS_MODULE_NAME,FOOLOS_LOG_INFO,"new task: stack: 0x%08X thread: 0x%08X", stack, thread);
//First, this stuff is pushed by the processor
*--stack = 0x0202; //This is EFLAGS
*--stack = 0x08; //This is CS, our code segment
*--stack = (uint32_t)thread; //This is EIP
//Next, the stuff pushed by 'pusha'
*--stack = 0; //EDI
*--stack = 0; //ESI
*--stack = 0; //EBP
*--stack = 0; //Just an offset, no value
*--stack = 0; //EBX
*--stack = 0; //EDX
*--stack = 0; //ECX
*--stack = 0; //EAX
//Now these are the data segments pushed by the IRQ handler
/*
*--stack = 0x10; //DS
*--stack = 0x10; //ES
*--stack = 0x10; //FS
*--stack = 0x10; //GS
*/
task_list[pid].esp = (uint32_t)stack; //Update the stack pointer
task_list[pid].active=true;
x86_set_pdbr(0x0502000);
};
// this gets called by our clock interrupt regularly!
uint32_t task_switch_next(uint32_t oldesp)
{
timer_tick();
if(current_task==-2)return oldesp;
if(current_task!=-1)task_list[current_task].esp=oldesp;
for(int i=0;i<MAX_TASKS;i++)
{
int pid=(current_task+1+i)%MAX_TASKS; // schedule round robin style
if(task_list[pid].active)
{
log(FOOLOS_MODULE_NAME,FOOLOS_LOG_INFO,"switch from %d to pid: %d (0x%08X) vmem: %d ", current_task,pid,task_list[pid].esp,task_list[pid].vmem);
current_task=pid;
if(pid==0)x86_set_pdbr(0x0D08000);
if(pid==1)x86_set_pdbr(0x150E000);
return task_list[pid].esp;
}
}
return oldesp;
}
// task testing //
volatile void test1()
{
static char *argv[]={"/bin/foolshell",NULL};
static char *env[]={"PATH=/bin","PWD=/home/miguel","PS1=$ ",NULL};
syscall_execve("/bin/task1",argv,env);
while(1)
{
syscall_write(1,">",1);
}
}
volatile void test2()
{
static char *argv[]={"/bin/foolshell",NULL};
static char *env[]={"PATH=/bin","PWD=/home/miguel","PS1=$ ",NULL};
syscall_execve("/bin/task2",argv,env);
while(1)
{
for(int i='a';i<'z';i++)
{
syscall_write(1,&i,1);
}
}
}
volatile void test3()
{
while(1)
{
syscall_write(1," ",1);
}
}
void task_init()
{
task_create(0,test2);
task_create(1,test1);
// task_create(2,test3);
current_task=-1;
}
|
