summaryrefslogtreecommitdiff
path: root/src/user/app/init/driver/termcook.c
blob: 818f8f98eddee1ad3e8e5b9bac06910ff6e1a58b (plain)
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
#include "driver.h"
#include <camellia/syscalls.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

static void w_output(handle_t output, const char *buf, size_t len) {
	size_t pos = 0;
	while (pos < len) {
		int ret = _syscall_write(output, buf + pos, len - pos, pos, 0);
		if (ret < 0) break;
		pos += ret;
	}
}

static void line_editor(handle_t input, handle_t output) {
	char readbuf[16], linebuf[256];
	size_t linepos = 0;
	for (;;) {
		int readlen = _syscall_read(input, readbuf, sizeof readbuf, -1);
		if (readlen < 0) return;
		for (int i = 0; i < readlen; i++) {
			char c = readbuf[i];
			switch (c) {
				case '\b':
				case 0x7f:
					if (linepos != 0) {
						printf("\b \b");
						linepos--;
					}
					break;
				case 3: /* C-c */
					_syscall_exit(1);
				case 4: /* EOT, C-d */
					if (linepos > 0) {
						w_output(output, linebuf, linepos);
						linepos = 0;
					} else {
						_syscall_write(output, NULL, 0, 0, 0); // eof
					}
					break;
				case '\n':
				case '\r':
					printf("\n");
					if (linepos < sizeof linebuf)
						linebuf[linepos++] = '\n';
					w_output(output, linebuf, linepos);
					linepos = 0;
					break;
				default:
					if (linepos < sizeof linebuf) {
						linebuf[linepos++] = c;
						printf("%c", c);
					}
					break;
			}
		}
	}
}

void termcook(void) {
	handle_t stdin_pipe[2] = {-1, -1};
	if (_syscall_pipe(stdin_pipe, 0) < 0)
		return;

	if (!fork()) {
		/* the caller continues in a child process,
		 * so it can be killed when the line editor quits */
		_syscall_dup(stdin_pipe[0], 0, 0);
		close(stdin_pipe[0]);
		close(stdin_pipe[1]);
		return;
	}
	if (!fork()) {
		close(stdin_pipe[0]);
		line_editor(0, stdin_pipe[1]);
		exit(0);
	}
	exit(_syscall_await());
}