一步一步实现一个简单的OS(试验任务切换)

............

extern void swtch(struct context *old, struct context *new);

struct context kmainContext; 	// 内核任务
struct context testContext0; 	// 任务0
struct context testContext1; 	// 任务1

// 任务0堆栈
uchar stack0[4096];
// 任务1堆栈
uchar stack1[4096];

void test0();
void test1();

/**
* 内核入口
*/
int kmain()
{
..........

// 上下文切换
// 初始化各任务上下文
testContext0.eip = (uint)test0;
testContext0.esp = (uint)stack0 + 4096;

testContext1.eip = (uint)test1;
testContext1.esp = (uint)stack1 + 4096;

// 由kmain，切换到test0
// 这个只是做一个测试，就不加特权级了。
// 这样看起来简单写
// 咱以后的进程切换就用这个
swtch(&kmainContext, &testContext0);

for(;;);
}

void test0()
{
while(1){
printf("A");
swtch(&testContext0, &testContext1);
printf("a");
}
}

void test1()
{
while(1){
printf("B");
swtch(&testContext1, &testContext0);
printf("b");
}
}

#   void swtch(struct context *old, struct context *new);
# 切换进程上下文
.globl swtch
swtch:
# 旧的寄存器地址
movl 4(%esp), %eax

popl 0(%eax)  # 保存旧的EIP（返回地址）
movl %esp, 4(%eax)  # 保存esp
movl %ebx, 8(%eax)  # 保存ebx
movl %ecx, 12(%eax)   # 保存ecx
movl %edx, 16(%eax)   # 保存edx
movl %esi, 20(%eax)   # 保存esi
movl %edi, 24(%eax)   # 保存edi
movl %ebp, 28(%eax)   # 保存ebp

# 新的寄存器地址
movl 4(%esp), %eax

movl 28(%eax), %ebp   # 载入ebp
movl 24(%eax), %edi   # 载入edi
movl 20(%eax), %esi   # 载入esi
movl 16(%eax), %edx   # 载入edx
movl 12(%eax), %ecx   # 载入ecx
movl 8(%eax), %ebx    # 载入ebx
movl 4(%eax), %esp    # 载入esp
pushl 0(%eax)  # 载入新的EIP（返回地址）

ret