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
|
#include <init/shell.h>
#include <init/stdlib.h>
#include <init/tests/main.h>
#include <shared/syscalls.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(const char *args) {
int fd = _syscall_open(args, strlen(args));
static char buf[256];
int len = 256;
if (fd < 0) {
printf("couldn't open.\n");
return;
}
len = _syscall_read(fd, buf, len, 0);
_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(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");
}
}
}
|
