任务切换

首先我们的中断处理函数过于简单,起码应该保持相关的寄存器,所以System_call.S修改如下:

.globl timer_interrupt
timer_interrupt:
	push %ds
	pushl %edx
	pushl %ecx
	pushl %ebx
	pushl %eax
	movl $0x10, %eax
	mov %ax, %ds
	incl jiffies
	movb $0x20,%al	
	outb %al,$0x20
	call do_timer
	popl %eax
	popl %ebx
	popl %ecx
	popl %edx
	pop %ds
	iret

通过前面的实验我们知道一个任务应该有自己的ldt和tss,另外还应该记录当前的状态,以及标识自己的pid和father

struct task_struct {
	long state;
	long pid,father;
	struct desc_struct ldt[3];
/* tss for this task */
	struct tss_struct tss;
};

下面我们开始构造任务,init_task是任务0,task1是任务1,INIT_TASK纯粹是为了填充结构体方便,task_unio中的stack是任务在内核态的堆栈。

#define INIT_TASK \
{  0,0,0,\
	{ \
		{0,0}, \
/* ldt */	{0x9f,0xc0fa00}, \
		{0x9f,0xc0f200}, \
	}, \
/*tss*/	{\
	}, \
}

#define PAGE_SIZE 4096
union task_union {
	struct task_struct task;
	char stack[PAGE_SIZE];
};
static union task_union init_task = {INIT_TASK,};
static union task_union task1 = {INIT_TASK,};


下面是任务在ring3的堆栈

struct stack_struct{
	char stack[256];
	int  top;
};
struct stack_struct stack0;
struct stack_struct stack1;

在下面我重新整理了gdt,然后填充ldt和tss

void main(void)
{

	disp_str("How old are you?\n");
	set_intr_gate(32,&timer_interrupt);
	_set_gdt_desc(&gdt[1],0x0000,0x7ff,DA_CR+DA_32+DA_LIMIT_4K);
	//堆栈段 0x10
	_set_gdt_desc(&gdt[2],0x0000,0x7ff,DA_DRW+DA_32+DA_LIMIT_4K);
	//视频段 0x18
	_set_gdt_desc(&gdt[3],0xb8000,0x2,DA_DRW+DA_32+DA_DPL3+DA_LIMIT_4K);
	//init_task的LDT  0x28
	_set_gdt_desc(&gdt[5],&init_task.task.ldt[0],0x40,DA_LDT);
	//task1的LDT  0x38
	_set_gdt_desc(&gdt[7],&task1.task.ldt[0],0x40,DA_LDT);

	//填充init_task的TSS
	init_task.task.tss.backlink = 0;
	init_task.task.tss.esp0=PAGE_SIZE+(long)&init_task;
	init_task.task.tss.ss0=0x10;
	init_task.task.tss.eip=&testA;
	init_task.task.tss.esp = &stack0.top;
	init_task.task.tss.flags=0x200;
	init_task.task.tss.es=0x17;
	init_task.task.tss.cs=0xf;
	init_task.task.tss.ss=0x17;
	init_task.task.tss.ds=0x17;
	init_task.task.tss.fs=0x17;
	init_task.task.tss.gs=0x18;
	init_task.task.tss.ldt=0x28;
	init_task.task.tss.trap=0x8000;
	init_task.task.tss.iobase=0x0;
	//init_task的TSS  0x20
	_set_gdt_desc(&gdt[4],&init_task.task.tss,0x68,DA_386TSS+DA_DPL3);
	
	//填充task1的TSS
	task1.task.tss.backlink = 0;
	task1.task.tss.esp0=PAGE_SIZE+(long)&task1;//&stack1_krn_ptr;
	task1.task.tss.ss0=0x10;
	task1.task.tss.eip=&testC;
	task1.task.tss.esp = &stack1.top;
      
	task1.task.tss.flags=0x200;
	task1.task.tss.es=0x17;
	task1.task.tss.cs=0xf;
	task1.task.tss.ss=0x17;
	task1.task.tss.ds=0x17;
	task1.task.tss.fs=0x17;
	task1.task.tss.gs=0x18;
	    
	task1.task.tss.ldt=0x38;
	task1.task.tss.trap=0x8000;
	task1.task.tss.iobase=0x0;
	//task1的TSS  0x30
	_set_gdt_desc(&gdt[6],&task1.task.tss,0x68,DA_386TSS+DA_DPL3);
	//task[1] = &(task1.task);
	ltr();//加载tss
	lldt();//加载ldt
	sti();//开中断
	move_to_user_mode();

	while(1);
}
任务有了就该任务切换了,这个动作我们在do_timer函数中做

int current_task=0;
//task0的tss选择子为0x20;task1的tss选择子为0x30
#define enter_task0() __asm__ ("ljmp $0x20, $0\n\t" :::)
#define enter_task1() __asm__ ("ljmp $0x30, $0\n\t" :::)

void do_timer(void)
{
	if(current_task)
	{
		current_task = 0;
		enter_task0();
	}
	else
	{
		current_task = 1;
		enter_task1();
	}
	disp_str("T");
}


由于重新整理了gdt,所以一些选择子要相应改变,并且添加了相应的宏定义

#define	DA_CR		0x9A
#define	DA_LIMIT_4K	0x8000
#define	DA_386TSS 0x89	
#define DA_LDT  0x82

#define set_registers() \
__asm__ ("movl $0x17,%%eax\n\t" \
	"movw %%ax,%%ds\n\t" \
	"movw %%ax,%%es\n\t" \
	"movw %%ax,%%fs\n\t" \
	"movw $0x18,%%ax\n\t" \
	"movw %%ax,%%gs" \
	:::"ax")

#define move_to_user_mode() \
__asm__ ("movl %%esp,%%eax\n\t" \
	"pushl $0x17\n\t" \
	"pushl %%eax\n\t" \
	"pushfl\n\t"\
	"pushl $0xf\n\t" \
	"pushl $testA\n\t" \
	"iret\n" \
	:::"ax")
		
#define lldt() \
__asm__ ("movw $0x28,%%ax\n\t" \
	"lldt     %%ax\n\t" \
	:::"ax")
		
#define ltr() \
__asm__ ("movw $0x20,%%ax\n\t" \
	"ltr     %%ax\n\t" \
	:::"ax")
中循环打印一个‘C’
void testC(void)		
{
	while(1){
		disp_str("C");
		delay(2);
	}
}

执行效果如下图,A是任务0打印的,C是任务1打印的,T是do_timer中打印的



main.c的完整代码如下:

#include <linux/head.h>
#include <asm/system.h>
#include <asm/io.h>
extern void divide_error(void);
void disp_str(char *info);

#define DA_C    	0x98    
#define DA_32   	0x4000 	
#define	DA_DPL3		0x60
#define	DA_DPL0		0x00
#define	DA_DRWA 	0x93
#define	DA_DRW 		0x92
#define	SA_RPL3  	3 
#define	DA_CR		0x9A
#define	DA_LIMIT_4K	0x8000
#define	DA_386TSS 0x89	
#define DA_LDT  0x82

#define set_registers() \
__asm__ ("movl $0x17,%%eax\n\t" \
	"movw %%ax,%%ds\n\t" \
	"movw %%ax,%%es\n\t" \
	"movw %%ax,%%fs\n\t" \
	"movw $0x18,%%ax\n\t" \
	"movw %%ax,%%gs" \
	:::"ax")

#define move_to_user_mode() \
__asm__ ("movl %%esp,%%eax\n\t" \
	"pushl $0x17\n\t" \
	"pushl %%eax\n\t" \
	"pushfl\n\t"\
	"pushl $0xf\n\t" \
	"pushl $testA\n\t" \
	"iret\n" \
	:::"ax")
		
#define lldt() \
__asm__ ("movw $0x28,%%ax\n\t" \
	"lldt     %%ax\n\t" \
	:::"ax")
		
#define ltr() \
__asm__ ("movw $0x20,%%ax\n\t" \
	"ltr     %%ax\n\t" \
	:::"ax")				
void delay(int time)
{
	int i, j, k;
	for (k = 0; k < time; k++) {
		for (i = 0; i < 10; i++) {
			for (j = 0; j < 10000; j++) {}
		}
	}
}
void testA(void)		
{

	set_registers();
	while(1){
		disp_str("A");
		delay(2);
	}
}
void testC(void)		
{
	while(1){
		disp_str("C");
		delay(2);
	}
}

typedef	unsigned int		u32;
typedef	unsigned short	u16;
typedef	unsigned char		u8;
typedef struct Descriptor		/* ???8 ??a?-?è??*/
{
	u16	limit_low;		/* Limit */
	u16	base_low;		/* Base */
	u8	base_mid;		/* Base */
	u8	attr1;			/* P(1) DPL(2) DT(1) TYPE(4) */
	u8	limit_high_attr2;	/* G(1) D(1) 0(1) AVL(1) LimitHigh(4) */
	u8	base_high;		/* Base */
}DESCRIPTOR;
void _set_gdt_desc(struct desc_struct *descriptor_addr,u32 base,u32 limit,u16 attr)
{
	DESCRIPTOR *descriptor = (DESCRIPTOR *)descriptor_addr;
	descriptor->limit_low	= limit & 0x0FFFF;
	descriptor->base_low		= base & 0x0FFFF;
	descriptor->base_mid		= (base >> 16) & 0x0FF;
	descriptor->attr1			= attr & 0xFF;
	descriptor->limit_high_attr2= ((limit>>16) & 0x0F) | (attr>>8) & 0xF0;
	descriptor->base_high	= (base >> 24) & 0x0FF;
}



typedef struct tss_struct {
	u32	backlink;
	u32	esp0;	/* stack pointer to use during interrupt */
	u32	ss0;	/*   "   segment  "  "    "        "     */
	u32	esp1;
	u32	ss1;
	u32	esp2;
	u32	ss2;
	u32	cr3;
	u32	eip;
	u32	flags;
	u32	eax;
	u32	ecx;
	u32	edx;
	u32	ebx;
	u32	esp;
	u32	ebp;
	u32	esi;
	u32	edi;
	u32	es;
	u32	cs;
	u32	ss;
	u32	ds;
	u32	fs;
	u32	gs;
	u32	ldt;
	u16	trap;
	u16	iobase;		/* I/O位图基址大于或等于TSS段界限,就表示没有I/O许可位图 */
}TSS;

extern void timer_interrupt(void);
long volatile jiffies=0;
int current_task=0;
//task0的tss选择子为0x20;task1的tss选择子为0x30
#define enter_task0() __asm__ ("ljmp $0x20, $0\n\t" :::)
#define enter_task1() __asm__ ("ljmp $0x30, $0\n\t" :::)

void do_timer(void)
{
	if(current_task)
	{
		current_task = 0;
		enter_task0();
	}
	else
	{
		current_task = 1;
		enter_task1();
	}
	disp_str("T");
}


struct task_struct {
	long state;
	long pid,father;
	struct desc_struct ldt[3];
/* tss for this task */
	struct tss_struct tss;
};

#define INIT_TASK \
{  0,0,0,\
	{ \
		{0,0}, \
/* ldt */	{0x9f,0xc0fa00}, \
		{0x9f,0xc0f200}, \
	}, \
/*tss*/	{\
	}, \
}

#define PAGE_SIZE 4096
#define NR_TASKS 64
union task_union {
	struct task_struct task;
	char stack[PAGE_SIZE];
};
static union task_union init_task = {INIT_TASK,};
//struct task_struct *current = &(init_task.task);
//struct task_struct * task[NR_TASKS] = {&(init_task.task), };
static union task_union task1 = {INIT_TASK,};

struct stack_struct{
	char stack[256];
	int  top;
};
struct stack_struct stack0;
struct stack_struct stack1;
void main(void)
{

	disp_str("How old are you?\n");
	set_intr_gate(32,&timer_interrupt);
	_set_gdt_desc(&gdt[1],0x0000,0x7ff,DA_CR+DA_32+DA_LIMIT_4K);
	//堆栈段 0x10
	_set_gdt_desc(&gdt[2],0x0000,0x7ff,DA_DRW+DA_32+DA_LIMIT_4K);
	//视频段 0x18
	_set_gdt_desc(&gdt[3],0xb8000,0x2,DA_DRW+DA_32+DA_DPL3+DA_LIMIT_4K);
	//init_task的LDT  0x28
	_set_gdt_desc(&gdt[5],&init_task.task.ldt[0],0x40,DA_LDT);
	//task1的LDT  0x38
	_set_gdt_desc(&gdt[7],&task1.task.ldt[0],0x40,DA_LDT);

	//填充init_task的TSS
	init_task.task.tss.backlink = 0;
	init_task.task.tss.esp0=PAGE_SIZE+(long)&init_task;
	init_task.task.tss.ss0=0x10;
	init_task.task.tss.eip=&testA;
	init_task.task.tss.esp = &stack0.top;
	init_task.task.tss.flags=0x200;
	init_task.task.tss.es=0x17;
	init_task.task.tss.cs=0xf;
	init_task.task.tss.ss=0x17;
	init_task.task.tss.ds=0x17;
	init_task.task.tss.fs=0x17;
	init_task.task.tss.gs=0x18;
	init_task.task.tss.ldt=0x28;
	init_task.task.tss.trap=0x8000;
	init_task.task.tss.iobase=0x0;
	//init_task的TSS  0x20
	_set_gdt_desc(&gdt[4],&init_task.task.tss,0x68,DA_386TSS+DA_DPL3);
	
	//填充task1的TSS
	task1.task.tss.backlink = 0;
	task1.task.tss.esp0=PAGE_SIZE+(long)&task1;//&stack1_krn_ptr;
	task1.task.tss.ss0=0x10;
	task1.task.tss.eip=&testC;
	task1.task.tss.esp = &stack1.top;
      
	task1.task.tss.flags=0x200;
	task1.task.tss.es=0x17;
	task1.task.tss.cs=0xf;
	task1.task.tss.ss=0x17;
	task1.task.tss.ds=0x17;
	task1.task.tss.fs=0x17;
	task1.task.tss.gs=0x18;
	    
	task1.task.tss.ldt=0x38;
	task1.task.tss.trap=0x8000;
	task1.task.tss.iobase=0x0;
	//task1的TSS  0x30
	_set_gdt_desc(&gdt[6],&task1.task.tss,0x68,DA_386TSS+DA_DPL3);
	//task[1] = &(task1.task);
	ltr();//加载tss
	lldt();//加载ldt
	sti();//开中断
	move_to_user_mode();

	while(1);
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值