OS-Shell Challenge

Foolish-Han Lv2

Shell Challenge 任务文档

实现不带 .b 后缀指令

你需要实现不带 .b 后缀的指令,但仍需兼容带有 .b 后缀的指令,如 lsls.b 都应能够正确列出当前目录下的文件。

/home/git/22371236/user/lib/spawn.cspawn 函数中,对程序路径进行特判,若其不含有 .b 后缀,则进行添加,实现兼容。同时,为避免修改到后面的参数,将 prog 复制给 temp 用于打开处理文件。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
   //! challenge-shell start
/*
* add suffix ".b"
*/
char temp[1024];
int l = strlen(prog);
for (int i = 0; i < l; i++) {
temp[i] = prog[i];
}
if (l <= 2 || temp[l - 1] != 'b' || temp[l - 2] != '.') {
temp[l] = '.';
temp[l + 1] = 'b';
temp[l + 2] = '\0';
}

if ((fd = open(temp, O_RDONLY)) < 0) {
return fd;
}
//! challenge-shell end

实现指令条件执行

你需要实现 Linux shell 中的 &&||。 对于 command1 && command2command2 被执行当且仅当 command1 返回 0;对于 command1 || command2command2 被执行当且仅当 command1 返回非 0 值。

注: 评测中保证不出现括号。并且需要注意的是,在 bash 中 &&|| 的优先级相同,按照从左到右的顺序求值。
例如 cmd1 || cmd2 && cmd3,若 cmd1 返回 0,则 cmd1 执行后 cmd2 不会被执行,cmd3 会被执行;若 cmd1 返回非 0 且 cmd2 返回非 0,则 cmd3 将不会被执行。
提示:你可能需要修改 MOS 中对用户进程 exit 的实现,使其能够返回值。

获取返回值

我们知道,spawn 出的程序装入内存后,在 user/lib/libos.c 中开始执行,所以我们可以在这里获取程序执行的返回值。在这里,我是给进程控制块中增加了 u_int env_ret; 属性来记录进程的返回值,同时增加相应的系统调用来设置进程返回值。

1
2
3
4
5
6
7
8
void libmain(int argc, char** argv) {
env = &envs[ENVX(syscall_getenvid())];
//! challenge-shell start
int r = main(argc, argv);
syscall_set_ret(r);
//! challenge-shell end
exit();
}

识别标识符

user/sh.cgettoken 函数中,c 代表上一次识别的 tokennc 代表这次识别到的 token,假如这两次都识别 &| 标识符,即识别到对应符号。同时,我们再次调用 _gettoker ,使得下一次执行时从标识符后面开始扫描。最后,我们返回相应的宏,使得 parsecmd 可以识别到。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
//! challenge-shell start
/*
* find token "&&", "||" and ">>"
*/
if (c == '&' && nc == '&') {
*p1 = np1;
nc = _gettoken(np2, &np1, &np2);
return AND;
} else if (c == '|' && nc == '|') {
*p1 = np1;
nc = _gettoken(np2, &np1, &np2);
return OR;
} else if (c == '>' && nc == '>') {
*p1 = np1;
nc = _gettoken(np2, &np1, &np2);
return APPEND;
}
//! challenge-shell start

条件执行

parsecmd 遇到对应的标识符时,fork 出一个子进程,父进程执行标识符左边的指令,子进程执行右边的指令。左边的指令执行完毕后,父进程通过 ipc 把执行结果的返回值发送给子进程,然后等待子进程执行完毕;子进程解析完右边的指令后,通过 ipc 接收父进程发送的返回值,据此根据标识符的结果决定是否执行,同时设置子进程的返回值。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
// parsecmd
case AND:
r = fork();
if (r == 0) {
r = opencons();
dup(r, 0);
dup(0, 1);
*waitType = AND;
*nextCmd = 0;
return parsecmd(argv, rightpipe, waitType, nextCmd, append);
}
*nextCmd = r;
return argc;
break;

这里给 parsecmd 新增了 waitTypenextCmd 参数,分别记录标识符的类型和执行后面指令的进程 id。对于标识符右边的指令,需要记录标识符的类型,同时我们暂时认为它后面没有要执行的指令;对于标识符左边的指令,需要记录执行后面指令的进程 id。对于子进程,这里重新打开了控制台,并重置标准输入输出,是为了防止标识符前面的指令对输入输出进行重定向影响到子进程的标准输入输出。

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
// runcmd
// parsecmd 结束...
//! challenge-shell start
if (waitType == AND || waitType == OR) {
tmp = ipc_recv(0, 0, 0);
}
if (waitType == AND && tmp == 1) {
ret = 1;
} else if (waitType == OR && tmp == 0) {
ret = 0;
} else {
int child;
if (strcmp(argv[0], "history") == 0) {
child = 1;
historyOut();
} else {
child = spawn(argv[0], argv);
}

close_all();
if (child >= 0) {
wait(child);
ret = envs[ENVX(child)].env_ret;
} else {
debugf("spawn %s: %d\n", argv[0], child);
ret = 1;
}
}
//! challenge-shell end

runcmd 中,我们解析完指令后,首先根据标识符类型,通过 ipc 获取并记录返回值。接着,根据标识符的类型,决定是否执行右边的指令。若指令不执行,则根据左边指令执行的结果和标识符类型,设置自己的返回值(应对连续几个标识符的情况);若指令执行,则根据 spawn 的结果设置返回值。

1
2
3
4
5
6
//! challenge-shell start
if (nextCmd) {
ipc_send(nextCmd, ret, 0, 0);
wait(nextCmd);
}
//! challenge-shell end

在指令的返回值确定后,我们根据 nextCmd 参数确定其右边是否存在指令。若存在,则向其发送自己的返回值,然后等待进程结束。

进程关系梳理

在指令条件执行这一节,进程比较多,调用关系复杂,在此以 cmd1 || cmd2 && cmd3 为例进行梳理,假定 cmd1cmd3 为返回 0,cmd2 返回非 0 值:

进程创建过程:

  • 进程 1:读取我们输入的指令(输出 $ 的那个)
    • 进程 2: 由进程 1 fork 出来,用于 解析并执行 指令 cmd1 || cmd2 && cmd3
      • 进程 3:由进程 2 spawn 出来,用于 执行 进程 2 经过 parsecmd 解析后的指令 cmd1
      • 进程 4:由进程 2 fork 出来,用于 解析并执行 指令|| cmd2 && cmd3
        • (进程 5):由于 cmd1 为真,实际上不存在(由进程 4 spawn 出来,用于 执行 进程 2 经过 parsecmd 解析后的指令 || cmd2
        • 进程 6:由进程 4 fork 出来,用于 解析并执行 指令&& cmd3
          • 进程 7:由进程 6 spawn 出来,用于 执行 进程 2 经过 parsecmd 解析后的指令 cmd3

进程退出过程:

  • 进程 1:等待 进程 2 退出,然后读取下一条输入
    • 进程 2:等待 进程 3 退出,并据此设置 cmd1 的返回值为 0,然后 发送 给进程 4,等待 进程 4 结束后退出
      • 进程 3:按照对应的 .c 文件主函数正常执行,并在 libos.c 对应位置退出
      • 进程 4:等待 cmd1 的返回值,然后据此决定 cmd2 不用执行,从而设置 cmd1 || cmd2 的返回值为 0,接着 发送 给进程 6,等待 进程 6 结束后退出
        • 进程 5:由于 cmd1 为真,没有被创建
        • 进程 6:等待 cmd1 || cmd2 的返回值,然后据此决定 cmd3 需要执行,等待 进程 7 结束后退出
          • 进程 7:按照对应的 .c 文件主函数正常执行,并在 libos.c 对应位置退出

可以看到,上述的进程具有类似 递归 的调用关系,同时存在着父进程等待子进程退出,子进程等待父进程发送返回值的情况,稍有不慎就会产生 死锁,这实际上是个糟糕的设计!应该反过来,由父进程执行标识符后面的指令,子进程执行标识符前面的指令,这样每个进程执行完了就可以退出,而不需等待。

实现更多指令

你需要实现 touchmkdirrm 指令,只需要考虑如下情形:

  • touch:

    touch <file>:创建空文件 file,若文件存在则放弃创建,正常退出无输出。 若创建文件的父目录不存在则输出 touch: cannot touch '<file>': No such file or directory。 例如 touch nonexistent/dir/a.txt 时应输出 touch: cannot touch 'nonexistent/dir/a.txt': No such file or directory

  • mkdir:

    • mkdir <dir>:若目录已存在则输出 mkdir: cannot create directory '<dir>': File exists,若创建目录的父目录不存在则输出 mkdir: cannot create directory '<dir>': No such file or directory,否则正常创建目录。
    • mkdir -p <dir>:当使用 -p 选项时忽略错误,若目录已存在则直接退出,若创建目录的父目录不存在则递归创建目录。
  • rm:

    • rm <file>:若文件存在则删除 <file>,否则输出 rm: cannot remove '<file>': No such file or directory
    • rm <dir>:命令行输出: rm: cannot remove '<dir>': Is a directory
    • rm -r <dir>|<file>:若文件或文件夹存在则删除,否则输出 rm: cannot remove '<dir>|<file>': No such file or directory
    • rm -rf <dir>|<file>:如果对应文件或文件夹存在则删除,否则直接退出。

这三个指令的实现比较简单,主要集中在 touch.cmkdirrm 三个文件中。

  • touch
    1. 分割目录与文件名(如果有上级目录)
    2. 若目录不存在(open 目录返回值为负),输出错误信息
    3. 若目录存在,创建文件(open 完整路径)
  • mkdir:
    1. 若目录存在(open 目录返回值非负),且没有 -p 选项,则输出错误信息
    2. 若目录不存在,分割出父级目录
      1. 若没有 -p 选项
        • 若父级目录(open 父级目录返回值为负)不存在,输出错误信息
        • 若父级目录存在,创建目录(open 完整路径)
      2. 若有 -p 选项
        • 从顶级目录开始,依次尝试打开并创建目录文件(open 对应级别的完整路径)
  • rm:
    1. 若没有选项,尝试打开文件(open 路径):
      • 若返回负值,则输出文件不存在的错误信息
      • 若返回非负值,则判断文件类型
        • 若为普通文件,则删除(remove 路径)
        • 若为目录文件,则输出错误信息
    2. 若有选项,尝试打开目录(open 路径):
      • 若返回负值,且没有 f 选项,则输出报错信息
      • 若返回非负值,则直接删除目录

在这里,我们创建文件和目录统一使用了 open ,给打开模式中加入 O_CREAT 即可;对于目录,再额外加上 O_MKDIR。由于 fs/serv.cserve_open 中没有加入创建目录的功能,所以我们需要在其中加入相关内容。

1
2
3
4
5
//! challenge-shell start
if (rq->req_omode & O_MKDIR) {
f->f_type = FTYPE_DIR;
}
//! challenge-shell end

实现反引号

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

echo ls | cat | cat | cat

识别反引号

user/sh.c_gettoken 函数中,我们需要读取出反引号之间的内容,同时修改指针,使得前面的字符串从左边的反引号后面的第一个字符开始,后面的字符串从右边的反引号后面的第一个字符开始,使得 parsecmd 可以识别到反引号,并获取中间的字符串。

1
2
3
4
5
6
7
8
9
10
11
12
//! challenge-shell start
if (*s == '`') {
s++;
*p1 = s;
while (*s && *s != '`') {
s++;
}
*s = '\0';
*p2 = s + 1;
return '`';
}
//! challenge-shell end

指令替换

这里我们使用 fork 出的子进程去执行反引号之间的指令,然后将结果传递给父进程,父进程将结果解析为自身的参数,然后接着解析剩余的参数并执行。在结果传递的过程中,我使用了文件为媒介进行传递。子进程写文件,父进程读文件,实际上也可以使用管道。

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
// sh.c 的全局变量
char temp[10][4096];
int num;
// ...
case '`':;
r = fork();
if (r == 0) {
*append = 1;
strcpy(temp[num], t);
fd = open(".TEMP", O_WRONLY | O_CREAT | O_TRUNC);
if (fd < 0) {
debugf("open for writing %s error\n", ".TEMP");
exit();
}
dup(fd, 1);
close(fd);
runcmd(temp[num], append);
exit();
} else {
wait(r);
r = open(".TEMP", O_RDONLY);
read(r, temp[num], 4096);
struct Filefd* filefd = num2fd(r);
int l = filefd->f_file.f_size;
argv[argc++] = temp[num];
for (int i = 0; i < l; i++) {
if (strchr(WHITESPACE, temp[num][i])) {
while (i < l && strchr(WHITESPACE, temp[num][i])) {
temp[num][i] = 0;
i++;
}
if (i < l) {
argv[argc++] = temp[num] + i;
}
}
}
close(r);
remove(".TEMP");
num++;
}
break;

在子进程中,我们创建 “.TEMP” 文件,将子进程的标准输出重定向到文件中,然后使用 runcmd 执行指令;在父进程中,我们等待子进程执行完毕,然后打开文件并读取其内容,接着过滤掉空白符,将字符串写入到参数中。

注意到这里使用了 temp 这个二维数组全局变量来存储反引号内指令执行的结果,这是因为在 parsecmdruncmd 中, argv 存储的都是字符串的指针,这首先意味着我们需要使用全局变量(最早的 argv 实际上是由 main 函数中 fork 出的子进程带入)。因为这些都是指针,所以我们要保证解析完毕后其指向的空间不会再被写入,因此我们采用了二维数组,以此来避免一行指令中具有多个反引号重复写 temp 的情况。

此外,我们这里的子进程使用 runcmd 而非 parsecmd 来执行指令,这是因为 gettoken 函数中的 np1np2 都是静态变量,存储的仍然是原来的命令的指针,因此我们需要调用 runcmd 来重置其 gettoken 函数并进行执行指令的操作。

实现注释功能

你需要使用 # 实现注释功能,例如 ls | cat # this is a comment meowls | cat 会被正确执行,而后面的注释则会被抛弃。

1
2
3
4
5
6
7
8
9
//! challenge-shell start
if (*s == '#') {
return '#';
}
//! challenge-shell end
//! challenge-shell start
case '#':
return argc;
//! challenge-shell end

这里只需要在 _gettoker 中识别 # ,并在 parsecmd 中增加新的分支,遇到 # 时停止解析即可。

实现历史指令

你需要实现 shell 中保存历史指令的功能,可以通过 ↑ UpDown ↓ 选择所保存的指令并执行。你需要将历史指令保存到根目录的 .mosh_history 文件中(一条指令一行),为了评测的方便,我们设定 $HISTFILESIZE=20(bash 中默认为 500),即在 .mosh_history 中至多保存最近的 20 条指令。你还需要支持通过 history 命令输出 .mosh_history 文件中的内容。

注:在 bash 中,history 为 shell built-in command,我们规定需要将 history 实现为 built-in command。

你需要将当前执行的指令先存入 .mosh_history 中,例如:

1
2
3
4
echo `ls | cat`
echo meow # comment
history
history | cat

当历史指令为空时,依次执行上述四条指令后,后两条指令会分别输出

1
2
3
echo `ls | cat`
echo meow # comment
history

1
2
3
4
echo `ls | cat`
echo meow # comment
history
history | cat

使用↑ Up能够切换到上一条指令(如果上一条指令存在),使用 Down ↓ 能够切换到下一条指令(如果下一条指令存在)。能够选择的指令范围为:用户当前输入的指令与 .mosh_history 文件中保存的所有指令。例如在执行了上述四条指令后,用户输入了 echo,此时 ↑ Up 应该将指令切换至 history | cat,再次进行三次 ↑ Up 后切换至 echo ` ls | cat`,此时再次 ↑ Up 应保留在该指令(因为已经不存在上一条指令);再进行四次 Down ↓ 后将切换回 echo,此时再次 Down ↓ 应保留在该指令(因为不存在下一条指令)。

历史文件读写

由于一行指令的长度并不统一,所以为了便于找到一行指令在 .mosh_history 中的准确位置,我规定一行指令在文件中占据固定的大小,便于我们快速根据行号读取文件中一行的内容。

1
2
3
4
5
6
7
8
9
10
11
int readhis(char* buf, int lineNum) {
struct Fd* fd = num2fd(history);
fd->fd_offset = ((lineNum - 1) % MAXHISTORYNUM) * MAXLINELENGTH;
return read(history, buf, MAXLINELENGTH);
}

int writehis(char* buf, int lineNum) {
struct Fd* fd = num2fd(history);
fd->fd_offset = ((lineNum - 1) % MAXHISTORYNUM) * MAXLINELENGTH;
return write(history, buf, MAXLINELENGTH);
}

历史的记录与查询

一旦 shell 打开,我们便以读写的权限打开创建 .mosh_history 文件,在每次输入一行指令时进行读写。在这里,我根据需要,定义了三个行号指针,便于进行记录与查询操作。

因为要求记录最近的20条指令,所以历史文件也是循环写入,我们需要记录 逻辑上 的行号在 文件 中对应的行号。在这里,我的 hisStarthisEndhisNow 分别对应起始行(最上面的行)、结束行(最下面的行)、当前观察行(终端显示的行)。注意到,在上面的 readhiswritehis 函数中,我们均对行号进行了取模操作,所以这里的三个变量不用保持在 [1, 20],无限递增即可。这样做是为了便于后续处理,不用再区别考虑起始行在结束行前面或后面的情况。没错,管道中也是这么实现的,这里算是学习到了一点源代码的精髓

1
2
3
int hisStart = 1;
int hisEnd = 1;
int hisNow = 1;

历史的记录与查询主体在原有的 readline 函数中进行。

当我们识别当上箭头(连续的三个字符 "\e[A")后:

  1. 若当前行等于结束行,我们需要把目前输入的内容保存,便于输入下箭头时可以回来重新编辑
  2. 输出下箭头,使光标保持不动;回到行首并输出空格,“清空”屏幕上显示的原有内容
  3. 当前行自减,前往上一行。若小于起始行,则重置为起始行
  4. 从历史中读取当前行,并进行输出,同时保持原有的终端符号

当我们识别当上箭头(连续的三个字符 "\e[A")后:

  1. 若当前行等于结束行,我们需要把结束行保存(确保下箭头按到底时,可以照常从文件中输出)
  2. 回到行首并输出空格,“清空”屏幕上显示的原有内容(这里没有输出上箭头,是因为输入下箭头时,光标并不会移动)
  3. 当前行自增,前往下一行。若大于结束,则重置为结束行
  4. 从历史中读取当前行,并进行输出,同时保持原有的终端符号
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
if (buf[i] == '\e') {
i++;
read(0, buf + i, 1);
if (buf[i] == '[') {
i++;
read(0, buf + i, 1);
if (buf[i] == 'A') {
i -= 2;
buf[i] = '\0';
if (hisNow == hisEnd) {
writehis(buf, hisEnd);
}
debugf("%s", DOWN);
debugf("\r%s", WHITELINE);
hisNow--;
if (hisNow < hisStart) {
hisNow = hisStart;
}
readhis(buf, hisNow);
debugf("\r$ %s", buf);
i = strlen(buf) - 1;
continue;

} else if (buf[i] == 'B') {
i -= 2;
buf[i] = '\0';
if (hisNow == hisEnd) {
writehis(buf, hisEnd);
}
debugf("\r%s", WHITELINE);
hisNow++;
if (hisNow > hisEnd) {
hisNow = hisEnd;
}
readhis(buf, hisNow);
debugf("\r$ %s", buf);
i = strlen(buf) - 1;
continue;
}
}
}

注意到,上面读写历史指令时,都把它放在了会读取输入的 buf 中,同时重置了 i。这样的话,buf 中始终保存了当前屏幕上显示的那条指令,便于我们在移动到某条历史指令的时候,可以直接对其进行修改然后执行。正因如此,当我们接收到回车键时,只需把 buf 内容写入文件中,修改起始行、结束行和当前行指针,然后退出即可。

1
2
3
4
5
6
7
8
9
10
if (buf[i] == '\r' || buf[i] == '\n') {
buf[i] = 0;
writehis(buf, hisEnd);
if (hisEnd - hisStart >= MAXHISTORYNUM) {
hisStart++;
}
hisEnd++;
hisNow = hisEnd;
return;
}

历史输出

history 作为一条内置指令,并不存在 history.c 文件实现其功能,也不会采用 spawn 来装载程序。为了保证行为的一致性,在这里,我 fork 出一个子进程来专门执行指令。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
int child;
if (strcmp(argv[0], "history") == 0) {
child = fork();
if (child > 0) {
close_all();
wait(child);
ret = 0;
} else {
historyOut();
exit();
}
} else {
child = spawn(argv[0], argv);
close_all();
if (child >= 0) {
wait(child);
ret = envs[ENVX(child)].env_ret;
} else {
debugf("spawn %s: %d\n", argv[0], child);
ret = 1;
}
}

实现一行多指令

你需要实现使用 ; 将多条指令隔开从而从左至右依顺序执行每条指令的功能。例如:

1
ls;ls | cat; echo nihao,mosh; echo `ls; echo meow`

由于 SYMBOLS 中已经含有 ;gettoken 函数已经能够解析到,所以我们只需在 parsecmd 函数中增加相应分支进行处理即可。

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
case ';':
r = fork();
if (r == 0) {
return argc;
} else {
wait(r);
if (*append == 1) {
fd = open(".TEMP", O_WRONLY | O_CREAT);
if (fd < 0) {
debugf("open for writing %s error\n", ".TEMP");
exit();
}
struct Filefd* filefd = num2fd(fd);
int size = filefd->f_file.f_size;
dup(fd, 1);
struct Fd* fd1 = num2fd(1);
fd1->fd_offset = size;
close(fd);
} else {
r = opencons();
dup(r, 0);
dup(0, 1);
}
argc = 0;
}
break;

吸取了条件执行复杂度飙升的教训之后,这次我们让 fork 出来的子进程执行 ; 左边的指令,父进程执行 ; 右边的指令。由于 ; 前面可能通过重定向更改了标准输入输出,所以我们需要重置父进程的标准输入输出。注意到,我们之前在处理反引号时,有个 *append = 1; ,它实际上就是告诉我们这里的标准输入输出应该重置为 .TEMP;不然,则重置为终端。

实现追加重定向

你需要实现 shell 中 >> 追加重定向的功能,例如:

ls >> file1; ls >> file1

最后文件 file1 中将会有两次 ls 指令的输出。

&&|| 的识别类似,识别标识符时,我们只需在 gettoken 函数中判断前后两次解析到的元素是否均为 > ;若符合条件,返回相应的宏即可。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
case APPEND:
if (gettoken(0, &t) != 'w') {
debugf("syntax error: > not followed by word\n");
exit();
}
fd = open(t, O_WRONLY | O_CREAT);
if (fd < 0) {
debugf("open for writing %s error\n", t);
exit();
}
struct Filefd* filefd = num2fd(fd);
int size = filefd->f_file.f_size;
dup(fd, 1);
struct Fd* fd1 = num2fd(1);
fd1->fd_offset = size;
close(fd);
break;

与普通输出重定向类似,我们都需要把标准输出重新设置为对应的文件。所不同的是,我们需要修改标准输出文件描述符的偏移量,移动到文件的末尾,从而实现追加写的功能。

实现引号支持

你需要实现引号支持,比如 echo "ls >",shell 在解析时需要将双引号内的内容看作是单个字符串。

所谓引号支持,就是把引号的字符串看做一个完整的参数,而不进行额外的解析。与 ` 的处理类似,我们只需在 _gettoker 中识别到其中的字符串,返回普通的参数标识即可。

1
2
3
4
5
6
7
8
9
10
11
12
//! challenge-shell start
if (*s == '\"') {
s++;
*p1 = s;
while (*s && *s != '\"') {
s++;
}
*s = '\0';
*p2 = s + 1;
return 'w';
}
//! challenge-shell end

实现前后台任务管理

  • 你需要支持 mosh 运行后台进程,当命令的末尾添加上 & 符号时,该命令应该在后台执行。
  • 实现 jobs 指令列出当前 shell 中所有后台任务的状态。你需要为任务创建 ID(每次启动 mosh 时,任务从 1 开始编号,每个新增任务编号应加 1),并且通过 jobs 指令输出包括:任务 ID(job_id)、任务的运行状态(status:可能的取值为 RunningDone)、任务的进程 ID(env_id)与运行任务时输入的指令(cmd)。请以 printf("[%d] %-10s 0x%08x %s", job_id, status, env_id, cmd) 的格式进行输出。
  • 实现 fg 将后台任务带回前台继续运行,用户通过 fg <job_id> 的方式将对应任务带回前台。
  • 实现 kill 指令,用户通过 kill <job_id> 来实现结束后台任务。

fgkill 指令中,若 job_id 对应的后台任务不存在则输出 printf("fg: job (%d) do not exist\n", job_id),若 job_id 对应的 ID 为 envid 的进程状态不为 Running 则输出 printf("fg: (0x%08x) not running\n", envid)

例如:

1
2
3
4
5
sleep 10&
sleep 60 &
jobs
# wait for about 10 seconds...
jobs

依次执行上述指令,则第一个 jobs 应输出(其中进程 ID 的值可能与你本地运行的输出结果不同):

1
2
[1] Running    0x00003805 sleep 10&
[2] Running 0x00005006 sleep 60 &

第二个 jobs 应输出:

1
2
[1] Done       0x00003805 sleep 10&
[2] Running 0x00005006 sleep 60 &

我对这个任务采取了一种简化的实现方法——即认为 & 只会出现在一行指令的结尾,不会出现一行中多条命令分别 & 的情况;同时 fgkilljobs 也作为单独的指令来出现,不会出现和其他指令嵌套的情况。

对于以 & 结尾的指令,我在 readline 结束后即进行扫描,检验这行指令是否需要后台执行。若需要,则子 shell 不需等待其执行完成就可输入下一条指令;不然,则需在终端阻塞等待执行指令的子 shell,显示地展示指令的执行过程。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
if ((r = fork()) < 0) {
user_panic("fork: %d", r);
}
if (r == 0) {
if (flag) {
buf[l - 1] = 0;
}
int append = 0;
runcmd(buf, &append);
debugf("runcmd has exited! can't reach here!\n");
exit();
} else {
if (flag) {
add_job(r, buf);
} else {
wait(r);
}
}

对于 fdkilljobs 这三条指令,我也是在 readline 结束后进行检验执行,不进入 parsecmd 中进行处理。从这个角度来说,我实现的 shell 是具有明显缺陷的,这应当是日后改进的方向。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
if (strncmp(buf, "fg", 2) == 0) {
int id = 0;
for (int j = 3; j < strlen(buf); j++) {
if (isdigit(buf[j])) {
id = id * 10 + buf[j] - '0';
}
}
fg_job(id);
continue;
} else if (strncmp(buf, "kill", 4) == 0) {
int id = 0;
for (int j = 5; j < strlen(buf); j++) {
if (isdigit(buf[j])) {
id = id * 10 + buf[j] - '0';
}
}
kill_job(id);
continue;
} else if (strcmp(buf, "jobs") == 0) {
cmd_jobs();
continue;
}

相关具体处理函数如下:

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
struct Job {
/* data */
char cmd[128];
int job_id;
int env_id;
} jobs[MAXJOBNUM];
int jobIds[MAXJOBNUM + 1];

void cmd_jobs() {
for (int i = 1; i < MAXJOBNUM; i++) {
if (jobIds[i]) {
char status[32];
if (syscall_env_status(jobs[i].env_id) == 0) {
strcpy(status, "Done");
} else {
strcpy(status, "Running");
}
printf("[%d] %-10s 0x%08x %s\n", jobs[i].job_id, status,
jobs[i].env_id, jobs[i].cmd);
}
}
}

void add_job(int env_id, char* cmd) {
for (int i = 1; i < MAXJOBNUM; i++) {
if (jobIds[i] == 0) {
jobIds[i] = 1;
jobs[i].job_id = i;
jobs[i].env_id = env_id;
strcpy(jobs[i].cmd, cmd);
break;
}
}
}

void fg_job(int job_id) {
if (jobIds[job_id] == 0) {
printf("fg: job (%d) do not exist\n", job_id);
} else if (syscall_env_status(jobs[job_id].env_id) == 0) {
printf("fg: (0x%08x) not running\n", jobs[job_id].env_id);
} else {
printf("\n$ %s\n", jobs[job_id].cmd);
wait(jobs[job_id].env_id);
}
}

void kill_job(int job_id) {
if (jobIds[job_id] == 0) {
printf("fg: job (%d) do not exist\n", job_id);
} else if (syscall_env_status(jobs[job_id].env_id) == 0) {
printf("fg: (0x%08x) not running\n", jobs[job_id].env_id);
jobIds[job_id] = 0;
} else {
syscall_env_destroy(jobs[job_id].env_id);
}
}