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
|
#include <init/shell.h>
#include <init/stdlib.h>
#include <init/tests/main.h>
#include <shared/syscalls.h>
#include <stdbool.h>
static char *split(char *base) {
while (*base) {
if (*base == ' ' || *base == '\t') {
*base++ = '\0';
return base;
}
base++;
}
return NULL;
}
static int readline(char *buf, size_t max) {
char c;
size_t pos = 0;
while (_syscall_read(__tty_fd, &c, 1, 0)) {
switch (c) {
case '\b':
case 0x7f:
/* for some reason backspace outputs 0x7f (DEL) */
if (pos != 0) {
printf("\b \b");
pos--;
}
break;
case '\n':
case '\r':
printf("\n");
buf[pos++] = '\0';
return pos;
default:
if (pos < max) {
_syscall_write(__tty_fd, &c, 1, 0);
buf[pos] = c;
pos++;
}
}
}
return -1; // error
}
static void cmd_cat_ls(const char *args, bool ls) {
int fd;
static char buf[256];
int len = 256;
if (!args) return; // no argument
fd = _syscall_open(args, strlen(args));
if (fd < 0) {
printf("couldn't open.\n");
return;
}
len = _syscall_read(fd, buf, len, 0);
if (ls)
for (int i = 0; i < len; i++)
if (buf[i] == '\0') buf[i] = '\n';
_syscall_write(__tty_fd, buf, len, 0);
_syscall_close(fd);
}
void shell_loop(void) {
static char cmd[256];
int level = 0;
char *args;
for (;;) {
printf("%x$ ", level);
readline(cmd, 256);
args = split(cmd);
if (!strcmp(cmd, "echo")) {
printf("%s\n", args);
} else if (!strcmp(cmd, "cat")) {
cmd_cat_ls(args, false);
} else if (!strcmp(cmd, "ls")) {
cmd_cat_ls(args, true);
} else if (!strcmp(cmd, "catall")) {
const char *files[] = {
"/init/fake.txt",
"/init/1.txt", "/init/2.txt",
"/init/dir/3.txt", NULL};
for (int i = 0; files[i]; i++) {
printf("%s:\n", files[i]);
cmd_cat_ls(files[i], false);
printf("\n");
}
} else if (!strcmp(cmd, "shadow")) {
_syscall_mount(-1, args, strlen(args));
} else if (!strcmp(cmd, "exit")) {
_syscall_exit(0);
} else if (!strcmp(cmd, "fork")) {
if (_syscall_fork())
_syscall_await();
else level++;
} else if (!strcmp(cmd, "run_tests")) {
test_all();
} else {
printf("unknown command :(\n");
}
}
}
|
