Coder Social home page Coder Social logo

os_challenge_sudo_shell's Introduction

Shell挑战性任务设计文档:MOS_SUDO_SHELL

一.实现不带.b的后缀指令

​ 实现不带.b的后缀指令同时兼容带有.b后缀的指令,这一点可以通过**首先尝试打开不带后缀的指令名,例如ls,失败后再尝试打开带后缀的指令名ls.b实现。**修改spawn.c/spawn函数中的打开文件逻辑即可。

int spawn(char *prog, char **argv) {
	// Step 1: Open the file 'prog' (the path of the program).
	// Return the error if 'open' fails.
	int fd;
	if ((fd = open(prog, O_RDONLY)) < 0) {
		/*Shell Challenge: *.b*/
		char *ext = ".b";
		char prog_b[1024];
		strcpy(prog_b, prog);
		// strcat
		for (int i = strlen(prog_b), j = 0; j < strlen(ext); i++, j++) {
			prog_b[i] = ext[j];
		}
		if ((fd = open(prog_b, O_RDONLY)) < 0) {
			return fd;
		}
	}
    //...
}

二.实现指令条件执行

​ 需要实现Linux Shell中的&&||,需要满足短路原则。

  • 对于command1 && command2commmand2被执行当且仅当command1返回0
  • 对于command1 | command2command2被执行当且仅当command1返回1
  • 并且&&||的优先级相同,从左向右执行

​ 课程组已经给出了测试程序true.cfalse.c,其中true.c返回0,false.c返回1,可以增加适当的输出信息用于测试。

​ 首先应当想到在解析命令时新增两种token&&||,分别用首字母ao表示。

int _gettoken(char *s, char **p1, char **p2) {
    //...
	if (strchr(SYMBOLS, *s)) { // 修改此处逻辑 识别 && || ;
		int t = *s;
		*p1 = s;
		/*Shell Challenge*/
		char *s1 = s + 1;
		if (*s1 == *s && *s1 == '&') {
			s++;
			t = 'a';
		} else if (*s1 == *s && *s1 == '|') {
			s++;
			t = 'o';
		}
		*s++ = 0;
		*p2 = s;
		return t;
	}
    //...
}

​ 当我们在parsecmd中解析到&&||时,我们的思路是先fork一下,首先运行左侧的指令,是否运行右边的指令由左边指令的返回值确定,返回值需要通过进程间通信实现。大致梳理一下这个过程:

  • parent
    • child
      • spawn : grandson

​ 我们可以看出这个通信过程是一个由底向上的过程,我们知道指令的具体执行是由子shell进行spawn后的“孙子”进程完成的,那么就要由“晚辈的”一层层向“长辈”靠拢,这其间的通信我们使用ipc实现。

grandson的返回值在libos.c/libmain中接收,并返回给child,需要注意的是除了这里debugf.c/user_panic中也要进行通信,我选择返回-1表示指令错误执行。

这里需要注意的是,需要新建用于传递返回值的系统调用和新的返回值字段,否则会与文件系统的ipc发生冲突。

  • env结构体添加用于接受返回值的成员return_value和是否处于接受返回值的状态量waiting_return_value

    struct Env {
    	//...
    	// Shell Challenge
    	u_int waiting_return_value;
    	u_int return_value;
    	
    };
  • 仿照ipc完成进行返回值传递和接受的系统调用

    • sys_send_return_value:发送返回值
    • sys_recv_return_value:接受返回值
    int sys_send_return_value(u_int envid, u_int value, u_int srcva, u_int perm) {
    	struct Env *e;
    	struct Page *p;
    	if (srcva != 0 && is_illegal_va(srcva)) {
    		return -E_INVAL;
    	}
    	try(envid2env(envid,&e,0));
    	if (e->waiting_return_value != 1) {
    		return -E_IPC_NOT_RECV;
    	}
    	e->env_ipc_value = value;
    	e->env_ipc_from = curenv->env_id;
    	e->env_ipc_perm = PTE_V | perm;
    	e->env_ipc_recving = 0;
    	e->waiting_return_value = 0;
    	e->return_value = value;
    	e->env_status = ENV_RUNNABLE;
    	TAILQ_INSERT_TAIL(&env_sched_list,e,env_sched_link);
    	if (srcva != 0) {
    		p = page_lookup(curenv->env_pgdir,srcva,NULL);
    		if (!p) {
    			return -E_INVAL;
    		}
    		try(page_insert(e->env_pgdir,e->env_asid,p,e->env_ipc_dstva,perm));
    	}
    	return 0;
    }
    
    int sys_recv_return_value() {
    	curenv->waiting_return_value = 1;
    	curenv->env_status = ENV_NOT_RUNNABLE;
    	TAILQ_REMOVE(&env_sched_list,curenv,env_sched_link);
    	((struct Trapframe *)KSTACKTOP - 1)->regs[2] = 0;
    	schedule(1);
    }
    
  • 传递返回值

    // libos.c
    void libmain(int argc, char **argv) {
    	// set env to point at our env structure in envs[].
    	env = &envs[ENVX(syscall_getenvid())];
    
    	// call user main routine
    	u_int r = main(argc, argv);
    	//syscall_ipc_try_send(env->env_parent_id, r, 0, 0);
    	syscall_send_return_value(env->env_parent_id, r, 0, 0);
    	syscall_set_job_done(env->env_id);
    	// exit gracefully
    	exit();
    }
    
    // debugf.c
    void _user_panic(const char *file, int line, const char *fmt, ...) {
    	debugf("panic at %s:%d: ", file, line);
    	va_list ap;
    	va_start(ap, fmt);
    	vdebugf(fmt, ap);
    	va_end(ap);
    	syscall_send_return_value(env->env_parent_id, -1, 0, 0);
    	debugf("\n");
    	exit();
    }
  • child进程使用ipc_recv接收返回值,这里需要注意的是,使用syscall_recv_return_value后,并不需要再进行wait,因为接收到返回值已经说明子进程执行完毕。

  • 另外需要注意的是,当我们遇到&& ||child需要向parent进行通信,其他时候不需要,我们可以设置一个标志flag进行区分。

void runcmd(char *s) {
	// ...
	int child = spawn(argv[0], argv); // spawn a new process to run the command
	// if succeeds, child is the envid of the new process.
	// if fails, child is the error code. 
	if (child >= 0) {
		if (flag == 1) {
			syscall_recv_return_value();
			syscall_send_return_value(env->env_parent_id, env->return_value, 0, 0);
		} else if (job_flag == 1) { // 需要创建后台任务 
			syscall_create_job(child, cmd);
			syscall_recv_return_value();
		} else {
			syscall_recv_return_value();
		}
	} 
	//...
}

​ 考虑到一种比较复杂的情况是cmd1 || cmd2 || cmd3 等可能省略中间部分指令的情况,我选择给parsecmd增加一个参数mark来表示这条指令是否需要执行,经过梳理运行逻辑,一种可行的解决办法是:当我们遇到短路原则时,下一条指令是一定不需要执行的,我们将mark标记为0,当return时,若mark标记为0,则return 0,否则return argc。在runcmd原有的逻辑中,return 0 说明后面没有命令,会退出命令运行,但在此时,我们不能够退出,我们的目标是跳过一条指令,还需要判断后边的指令是否需要执行我们可以对return 0的情况进行分析

  1. 命令末尾:return 0 && argv[0] = ""
  2. 若是省略命令:return 0 && argv[0] != ""

因此我们可以利用argv[0]是否为空串来对这两种情况进行区分,当是第一种情况时,直接退出;当是第二种情况时,我们需要与代表着cmd3的父进程进行通信,这时我们将前两个已经短路的指令看作一个整体,返回0

  • parsecmd中注意细节,每一个return都要对mark进行判断。
int parsecmd(char **argv, int *rightpipe, int mark) {
	int argc = 0;
	while (1) {
		//...
		case 'a':; // &&
			flag = 1;
			int child2 = fork();
			if (child2 == 0) { // child shell
				return mark ? argc : 0;
			} else { // parent shell
				syscall_recv_return_value();
				if (env->return_value == 0) {
					return parsecmd(argv, rightpipe, 1);
				} else {
					return parsecmd(argv, rightpipe, 0);
				}
			}
			break;
		case 'o':; // ||
			flag = 1;
			int child3 = fork();
			if (child3 == 0) {
				return mark ? argc : 0;
			} else {
				syscall_recv_return_value();
				if (env->return_value != 0) {
					return parsecmd(argv, rightpipe, 1);
				} else {
					return parsecmd(argv, rightpipe, 0);
				}
			}
			break;
		}
	}

	return argc;
}

void runcmd(char *s) {
	//...
	if (argc == 0) {
		if (argv[0]) { // 后边还有指令
			syscall_send_return_value(env->env_parent_id, 0, 0, 0);
		}
		return;
	}
	argv[argc] = 0;
	//...
}
  • 这里有一个很奇怪的点,可能和我的设计有关,我必须将close_all移动到后面,否则不能即使recv返回值,导致返回值丢失进入死锁

    	int child = spawn(argv[0], argv); // spawn a new process to run the command
    	if (child >= 0) {
    		if (flag == 1) {
    			syscall_recv_return_value();
    			syscall_send_return_value(env->env_parent_id, env->return_value, 0, 0);
    		} else if (job_flag == 1) { // 需要创建后台任务 
    			syscall_create_job(child, cmd);
    			syscall_recv_return_value();
    		} else {
    			syscall_recv_return_value();
    		}
    	} else {
    		debugf("spawn %s: %d\n", argv[0], child);
    	}
    	if (rightpipe) {
    		wait(rightpipe);
    	}
    	close_all(); // close all file descriptors !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

三.实现更多指令

3.1 touch

  • touch <file>:不会出现创建多个文件的情况

​ 新建touch.c并在include.mk中加入touch.b检查文件是否存在的方式为先尝试打开,若不能成功打开则进行创建。

// touch.c
#include<lib.h>

void touch(char *path) {
    int fd = open(path, O_RDONLY);
    // 首先检查文件是否存在
    if (fd > 0) {
        close(fd);
        return;
    } else {
        // 文件不存在,创建文件
        fd = open(path, O_CREAT);
        if (fd < 0) {
            printf("touch: cannot touch '%s': No such file or directory\n", path);
            return;
        }
        close(fd);
    
    }
}

int main(int argc, char **argv) {
    int i;
    if (argc < 2) {
        printf("touch: missing file operand\n");
        return 0;
    }
    touch(argv[1]);
}

3.2 mkdir

  • mkdir <dir>
  • mkdir -p <dir>

mkdir的实现方式与touch类似,在open时传入O_MKDIR,需要注意的是要在文件系统服务函数serv.c中打开函数时相应加入对于O_MKDIR的判断。

 void serve_open(u_int envid, struct Fsreq_open *rq) {
	//...
	if ((rq->req_omode & O_CREAT) && (r = file_create(rq->req_omode, rq->req_path, &f)) < 0 &&
	    r != -E_FILE_EXISTS) {
		ipc_send(envid, r, 0, 0);
		return;
	} else if ((rq->req_omode & O_MKDIR) && (r = file_create(rq->req_omode, rq->req_path, &f)) < 0 &&
	    r != -E_FILE_EXISTS) {
		ipc_send(envid, r, 0, 0);
		return;
	}
     //...
}

​ 同时应当修改file_create函数,增加传入参数rq->req_mode,区分创建文件还是目录,为结构体的type字段赋值。

int file_create(u_int req_mode, char *path, struct File **file) {
    //...
	/*Shell Challenge*/
	if (req_mode == O_MKDIR) {
		f->f_type = FTYPE_DIR;
	} else {
		f->f_type = FTYPE_REG;
	}
	//...
}

​ 具体的mkdir逻辑

#include<lib.h>

void mkdir(char *path, int flag) {
    int fd = open(path, O_RDONLY); // 首先检查文件是否存在
    if (fd > 0) {
        close(fd);
        if (!flag) { 
            printf("mkdir: cannot create directory '%s': File exists\n", path);
        }
        return;
    } else { // 文件不存在,创建文件
        fd = open(path, O_MKDIR);
        if (fd < 0) {
            if (!flag) {
                printf("mkdir: cannot create directory '%s': No such file or directory\n", path);
                return;
            } else { // 递归创建目录,例如 "/nonexist/dqr/123"
                char *p = path;
                if (*p == '/') { // 跳过开头的'/'
                    p++;
                }
                while (1) {
                    if (*p == '/') {
                        *p = '\0';
                        mkdir(path, 1);
                        *p = '/';
                    } else if (*p == '\0') {
                        mkdir(path, 1);
                        break;
                    }
                    p++;
                }
            }
        }
        close(fd);
    }
}

int main(int argc, char **argvs) {
    if (argc < 2) {
        printf("mkdir: missing operand\n");
        return 0;
    }
    // 需要考虑参数p忽略错误
    if (strcmp(argvs[1], "-p") == 0) {
        mkdir(argvs[2], 1);
    } else {
        mkdir(argvs[1], 0);
    }
    return 0;
}

3.3 rm

  • rm <file>
  • rm <dir>
  • rm -r <dir>|<file>
  • rm -rf <dir>|<file>

​ 对于rm指令处理的关键在于对于参数的处理,我们知道删除目录时需要-r/-rf参数,所以需要特殊判断无参数删除目录文件的情况。我采用的方法是在文件系统删除请求结构体中加入一个模式字段,用来表示参数情况(无参数为0,-r为1,-rf为2)。

struct Fsreq_remove {
	char req_path[MAXPATHLEN];
	u_int remove_type;
};

​ 对于这个参数的传递,我选择了一种不太“优雅“的方式,由于不能修改remove函数的传参(remove函数在test中有引用,评测时替换为标准文件,若修改会导致编译不过),我选择将模式数字拼接在路径字符串的\0之后,这样我们既可以读取模式数字,又可以保证读取路径的正确性。

​ 在remove函数中将这个数字取出,沿着调用链进行传递。

int remove(const char *path) {
	// Call fsipc_remove.
	// 最后一位为模式
	int len = strlen(path);
	int type = path[++len] - '0';
	/* Exercise 5.13: Your code here. */
	return fsipc_remove(path, type);
}

​ 在fsipc_remove中构建请求结构体时对新增字段赋值,达到通过结构体传参的目的。

int fsipc_remove(const char *path, u_int type) {
	//...
	req->remove_type = type;
	//...
}

​ 在文件系统服务函数中,为file_remove新增模式参数

void serve_remove(u_int envid, struct Fsreq_remove *rq) {
	r = file_remove(rq->req_path, rq->remove_type);
	ipc_send(envid, r, 0, 0);
}

​ 修改具体的文件删除逻辑,当要删除的文件为目录但传递的参数为N(0)时,返回一种新的错误码(E_IS_DIR=14,新增定义在error.h中)

int file_remove(char *path, u_int type) {
	int r;
	struct File *f;

	// Step 1: find the file on the disk.
	if ((r = walk_path(path, 0, &f, 0)) < 0) {
		return r;
	}

	/*Shell Challenge*/
	if (f->f_type == FTYPE_DIR && type == 0) {
		return -E_IS_DIR;
	}
    //...
}

​ 具体的remove逻辑:这里返回小于0值说明一定是N模式下删除了目录。

#include <lib.h>

#define N 0
#define R 1
#define RF 2

void rm(char *path, int flag) {
    int fd = open(path, O_RDONLY);
    if (fd < 0) {
        if (flag != RF) {
            printf("rm: cannot remove '%s': No such file or directory\n", path);
        }
    } else {
        close(fd);
        /*为了避免传参问题 将模式拼接在路径末尾\0之后*/
        char path1[MAXPATHLEN+1];
        strcpy(path1, path);
        int len = strlen(path1);
        path1[++len] = flag + '0';
        /********************/
        fd = remove(path1);
        if (fd < 0) {
            printf("rm: cannot remove '%s': Is a directory\n", path);
        }
    }
}

int main(int argc, char **argv) {
    if (argc < 2) {
        printf("rm: missing operand\n");
    }
    if (strcmp(argv[1], "-r") == 0) {
        rm(argv[2], R);
    } else if (strcmp(argv[1], "-rf") == 0) {
        rm(argv[2], RF);
    } else {
        rm(argv[1], N);
    }
    return 0;
}

四.实现反引号

​ 使用反引号实现指令替换,只需要考虑echo进行的输出,需要将反引号内指令执行的所有标准输出替换为echo的参数。

  • 换句话说,执行一下反引号里边的内容

​ 首先增加反引号token

int _gettoken(char *s, char **p1, char **p2) {
	//...
	if(*s == '`') { // 识别反引号
    	*s = 0;
    	s++;
    	*p1 = s;
   		while(*s && (*s != '`')){
        	s++;
    	}
    	*s++ = 0;
    	*p2 = s;
    	return 'f'; 
	}
    //...
}

​ 在parsecmd中增加对应逻辑,这时反引号中间的指令已经被保存在字符串指针t中。

int parsecmd(char **argv, int *rightpipe, int mark) {
	int argc = 0;
	while (1) {
        //...
		case 'f':; // 反引号 
			runcmd(t);
			break;
		}
    	//...
	}
}

五.实现注释功能

​ 使用#实现注释功能也就是对#后面的内容进行忽略,可以在进行执行指令时对字符串#后面的部分进行截断。

//sh.c
void runcmd(char *s) {
	/*Shell Challenge: #*/
	char *p = s;
	while (*p && *p != '#') {
		p++;
	}
	*p = 0;
	gettoken(s, 0);
    //...
}

六.实现历史指令

  • 启动shell时创建.mosh_history文件
  • 运行命令时将命令保存到.mosh_history
  • history命令输出文件内容
void create_history() {
	int fd;
	if ((fd = open(".mosh_history", O_CREAT)) < 0) {
		printf("create history file failed\n");
	}
	return;
}

void save_history_cmd(char * cmd) {
	int fd;
	if ((fd = open(".mosh_history", O_WRONLY)) < 0) {
		printf("open history file failed\n");
	}
	struct Stat st;
	stat(".mosh_history", &st);
	seek(fd, st.st_size);
	write(fd, cmd, strlen(cmd));
	write(fd, "\n", 1);
	close(fd);
	return;
}

void print_history() {
	int fd;
	if ((fd = open(".mosh_history", O_RDONLY)) < 0) {
		printf("open history file failed\n");
	}
	char ch;
	while (read(fd, &ch, 1)) {
		printf("%c", ch);
	}
	close(fd);
	return;
}

七.实现一行多指令

​ 实现使用;将多条指令隔开从而从左至右依顺序执行每条指令的功能,通过修改parsecmd()函数使其增添对;的处理能力。在parsecmd的字符串解析过程中,若读到一个;,则进行一次fork,产生一个子shell,让子shell执行左边的命令,父shell等待子shell执行完成后执行右边的命令。

		case ';':;
			int child = fork();
			if (child == 0) { // child shell
				return argc;
			} else { // parent shell
				wait(child); 
				return parsecmd(argv, rightpipe);
			}
			break;

八.实现追加重定向

  • 实现>>的追加重定向的功能

​ 首先增加一种token>>,在_gettoken中实现,标记为z

int _gettoken(char *s, char **p1, char **p2) {
	//...
	if (strchr(SYMBOLS, *s)) { // 修改此处逻辑 识别 && || ;
		int t = *s;
		*p1 = s;
		/*Shell Challenge*/
		char *s1 = s + 1;
		if (*s1 == *s && *s1 == '&') {
			s++;
			t = 'a';
		} else if (*s1 == *s && *s1 == '|') {
			s++;
			t = 'o';
		} else if (*s1 == *s && *s1 == '>') {
			s++;
			t = 'z';
		}
		*s++ = 0;
		*p2 = s;
		return t;
	}
	//...
}

​ 在parsecmd中进行修改,为文件设置好偏移量后进行dup

  • stat获取文件信息
  • seek设置文件偏移量
int parsecmd(char **argv, int *rightpipe, int mark) {
	int argc = 0;
	while (1) {
		//...
		case 'z':; // 实现追加重定向
			if (gettoken(0, &t) != 'w') {
				debugf("syntax error: > not followed by word\n");
				exit();
			}
			if ((fd = open(t, O_RDONLY | O_WRONLY | O_CREAT)) < 0) {
				debugf("failed to open %s\n", t);
				exit();
			}
			struct Stat st;
			if (fstat(fd, &st) < 0) {
				debugf("failed to fstat %s\n", t);
				exit();
			}
			seek(fd, st.st_size);
			if ((r = dup(fd, 1)) < 0) {
				debugf("failed to duplicate file to <stdout>\n");
				exit();
			}
			close(fd);
			break;	
		}
	}

	return argc;
}

九.实现引号支持

shell在解析时需要将双引号内的内容看作是单个字符串,在解析字符串时加了一种token为字符串,修改_gettoken中的逻辑。

	/*Shell Challenge : "content"*/
	if (*s == '"') { // read until the next '"'
		*s++;
		*p1 = s;
		while (*s && *s != '"') {
			s++;
		}
		*(s++) = 0; // *s = '"'
		*p2 = s;
		return 'w';
	}

十.实现前后台任务管理

10.1 实现后台任务并行

​ 此功能的实现与实现一行多指令类似,后台运行任务即在fork之后父进程不会等待子进程运行完成。

		case '&':;
			job_flag = 1;
			int child = fork();
			if (child == 0) { // child shell
				return argc;
			} else { // parent shell
				return parsecmd(argv, rightpipe);
			}		
			break;	

10.2 实现jobs指令

关于后台任务的三个指令我们需要实现为内置指令,shell中分为内置指令和外部指令,外部指令需要fork一个子进程进行执行,而内置指令不会,效率更高。

  • 基本的实现思路为我们在runcmd时检查指令,如果为内置指令不再进行spawn而是直接调用sh.c中写好的函数执行。
  • 为了实现进程间的共享,我选择在内核态对后台进程进行管理,通过系统调用syscall_*进行更新
  • 加系统调用的流程上机考试早已熟悉

首先:如何实现内置指令,区别于外部指令,内部指令执行时不会进行fork,效率更高,当我们在runcmd时应当特判遇到的内置指令,调用sh.c中写好的处理函数执行。

	if (strcmp(argv[0], "jobs") == 0) {
		execute_jobs();
		exit();
	} else if (strcmp(argv[0], "fg") == 0) {
		int jobId = parseJobId(argv[1]);
		execute_fg(jobId);
		exit();
	} else if (strcmp(argv[0], "kill") == 0) {
		int jobId = parseJobId(argv[1]);
		execute_kill(jobId);
		exit();
	}

​ 为了更好的管理工作的信息,我选择新建一个Job结构体,定义在env.h

struct Job {
	int job_id;
	int job_status; // 0 for Done; 1 for Running 
	int envid;
	char cmd[1024];
};

​ 在内核态对struct Job数组进行管理,将数组定义在env.c中便于操作。

struct Job jobs[32];
int jobCnt = 0;

​ **重点在于何时将进程放入后台:**我们设置一个全局变量job_flag读到&时置为1,在runcmd时,对这个标记进行特判,从而将进程放到后台,这里需要注意的是,recv应当在syscall_create_job之后,否则会一直阻塞,不能加入到后台,直到进程运行完毕才能加入到后台。

	if (child >= 0) {
		if (flag == 1) {
			syscall_recv_return_value();
			syscall_send_return_value(env->env_parent_id, env->return_value, 0, 0);
		} else if (job_flag == 1) { // 需要创建后台任务 
			syscall_create_job(child, cmd);
			syscall_recv_return_value();
		} else {
			syscall_recv_return_value();
		}
	} 

​ 这里创建后台任务也是通过系统调用实现,实际上是对内核态jobs数组赋值的操作,这样就实现了加入后台任务。

//env.c
void env_create_job(u_int envid, char * cmd) {
	jobs[jobCnt].envid = envid;
	jobs[jobCnt].job_status = 1; // Running
	strcpy(jobs[jobCnt].cmd, cmd);
	jobs[jobCnt].job_id = jobCnt + 1;
	jobCnt++;
}

​ 加入后台时,将运行状态status设置为1(Running),0(Done).

jobs操作实际上是一个“输出”操作,我们通过系统调用实现。

//sh.c
void execute_jobs() { // 约等于无用封装
	syscall_print_jobs();
}

//env.c
void env_print_jobs() {
	for (int i = 0; i < jobCnt; i++) {
		if (jobs[i].job_status == 1) {
			printk("[%d] %-10s 0x%08x %s\n\r", jobs[i].job_id, "Running", jobs[i].envid, jobs[i].cmd);
		} else {
			printk("[%d] %-10s 0x%08x %s\n\r", jobs[i].job_id, "Done", jobs[i].envid, jobs[i].cmd);
		}
	}
}

这里还有很重要的一点是:后台进程状态的变化,当后台进程结束时,应当设置状态为Done,而我们知道后台进程结束点是在libmain,在libmain中通过系统调用更改job_status = 0(Done)

// libos.c
void libmain(int argc, char **argv) {
	//...
	syscall_set_job_done(env->env_id);
	// exit gracefully
	exit();
}

//env.c
void env_set_job_done(u_int envid) {
	for (int i = 0; i < jobCnt; i++) {
		if (jobs[i].envid == envid) {
			jobs[i].job_status = 0;
			break;
		}
	}
}

10.3 实现fg指令

  • 首先构建一个简单的辅助函数,用来将argv[1]转换为<jobid>

    int parseJobId(char *s) {
    	int jobId = 0;
    	while (*s) {
    		jobId = jobId * 10 + (*s - '0');
    		s++; 
    	}
    	return jobId;
    }
  • fg命令同样通过系统调用实现,并不需要从后台jobs中删除jobid的目录项,只需到前台运行该进程,或者说当前运行进程wait这个后台进程。

    • 说个题外话,我是没想到不用从jobs列表中删除该项的,因为linux中的行为是进行删除,而课程组的建议是仿照linux行为实现,并且挑战性任务说明书一坨,MOS居然有了这么离谱的举动,居然也不在挑战性任务中说明或给出样例,等大家猜吗...
    //sh.c
    void execute_fg(int jobId) {
    	int envid = syscall_fg_job(jobId);
    	if (envid == -1) {
    		printf("fg: job (%d) do not exist\n", jobId);
    	}  else if (envs[ENVX(envid)].env_status != ENV_RUNNABLE) {
    		printf("fg: (0x%08x) not running\n", envid);
    	} else {
    		wait(envid);
    	}
    }
    
    //env.c
    int env_fg_job(int jobId) {
    	for (int i = 0; i < jobCnt; i++) {
    		if (jobs[i].job_id == jobId) {
    			return jobs[i].envid;
    		}
    	}
    	return -1;
    }

10.4 实现kill指令

  • kill命令同样通过系统调用实现,杀死后台进程

  • 同样不需要从Jobs列表中删除...

    // sh.c
    void execute_kill(int jobId) {
    	syscall_kill_job(jobId);
    }
    
    // env.c
    void env_kill_job(int jobId) {
    	for (int i = 0; i < jobCnt; i++) {
    		if (jobs[i].job_id == jobId) {
    			if (envs[ENVX(jobs[i].envid)].env_status != ENV_RUNNABLE) {
    				printk("fg: (0x%08x) not running\n", jobs[i].envid);
    				return;
    			} else {
    				env_destroy(&envs[ENVX(jobs[i].envid)]);
    				jobs[i].job_status = 0;
    				return;
    			}
    		}
    	}
    	printk("fg: job (%d) do not exist\n", jobId);
    	return;
    }

os_challenge_sudo_shell's People

Contributors

sudo-dqr avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.