xref: /kvmtool/term.c (revision f2a0b6a7843119094cf5cf7a86ac613818de47ef)
1b4405289SAsias He #include <poll.h>
2b4405289SAsias He #include <stdbool.h>
3b4405289SAsias He #include <termios.h>
4b4405289SAsias He #include <stdio.h>
5b4405289SAsias He #include <unistd.h>
6b4405289SAsias He #include <sys/uio.h>
7*f2a0b6a7SAnton Vorontsov #include <signal.h>
8b4405289SAsias He 
9b4405289SAsias He #include "kvm/read-write.h"
10b4405289SAsias He #include "kvm/term.h"
11b4405289SAsias He #include "kvm/util.h"
12b4405289SAsias He 
13b4405289SAsias He static struct termios	orig_term;
14b4405289SAsias He 
1516588013SAsias He int term_escape_char	= 0x01; /* ctrl-a is used for escape */
1616588013SAsias He bool term_got_escape	= false;
1716588013SAsias He 
189aa4a0ebSAsias He int active_console;
19b4405289SAsias He 
20b4405289SAsias He int term_getc(int who)
21b4405289SAsias He {
22b4405289SAsias He 	int c;
23b4405289SAsias He 
24b4405289SAsias He 	if (who != active_console)
25b4405289SAsias He 		return -1;
26b4405289SAsias He 
27b4405289SAsias He 	if (read_in_full(STDIN_FILENO, &c, 1) < 0)
28b4405289SAsias He 		return -1;
2916588013SAsias He 
3016588013SAsias He 	c &= 0xff;
3116588013SAsias He 
3216588013SAsias He 	if (term_got_escape) {
3316588013SAsias He 		term_got_escape = false;
3416588013SAsias He 		if (c == 'x') {
3516588013SAsias He 			printf("\n  # KVM session terminated.\n");
3616588013SAsias He 			exit(1);
3716588013SAsias He 		}
3816588013SAsias He 		if (c == term_escape_char)
3916588013SAsias He 			return c;
4016588013SAsias He 	}
4116588013SAsias He 
4216588013SAsias He 	if (c == term_escape_char) {
4316588013SAsias He 		term_got_escape = true;
4416588013SAsias He 		return -1;
4516588013SAsias He 	}
4616588013SAsias He 
47b4405289SAsias He 	return c;
48b4405289SAsias He }
49b4405289SAsias He 
50b4405289SAsias He int term_putc(int who, char *addr, int cnt)
51b4405289SAsias He {
52b4405289SAsias He 	if (who != active_console)
53b4405289SAsias He 		return -1;
54b4405289SAsias He 
550ea58e5bSPekka Enberg 	while (cnt--)
56b4405289SAsias He 		fprintf(stdout, "%c", *addr++);
57b4405289SAsias He 
58b4405289SAsias He 	fflush(stdout);
59b4405289SAsias He 	return cnt;
60b4405289SAsias He }
61b4405289SAsias He 
62b4405289SAsias He int term_getc_iov(int who, struct iovec *iov, int iovcnt)
63b4405289SAsias He {
6416588013SAsias He 	int c;
65b4405289SAsias He 
6616588013SAsias He 	if (who != active_console)
6716588013SAsias He 		return 0;
6816588013SAsias He 
6916588013SAsias He 	c = term_getc(who);
7016588013SAsias He 
7116588013SAsias He 	if (c < 0)
7216588013SAsias He 		return 0;
7316588013SAsias He 
7416588013SAsias He 	*((int *)iov[0].iov_base)	= c;
7516588013SAsias He 
76fbf15b30SAsias He 	return sizeof(char);
77b4405289SAsias He }
78b4405289SAsias He 
79b4405289SAsias He int term_putc_iov(int who, struct iovec *iov, int iovcnt)
80b4405289SAsias He {
81b4405289SAsias He 	if (who != active_console)
8216588013SAsias He 		return 0;
83b4405289SAsias He 
84b4405289SAsias He 	return writev(STDOUT_FILENO, iov, iovcnt);
85b4405289SAsias He }
86b4405289SAsias He 
87b4405289SAsias He bool term_readable(int who)
88b4405289SAsias He {
89b4405289SAsias He 	struct pollfd pollfd = (struct pollfd) {
90b4405289SAsias He 		.fd	= STDIN_FILENO,
91b4405289SAsias He 		.events	= POLLIN,
92b4405289SAsias He 		.revents = 0,
93b4405289SAsias He 	};
94b4405289SAsias He 
95b4405289SAsias He 	if (who != active_console)
96b4405289SAsias He 		return false;
97b4405289SAsias He 
98b4405289SAsias He 	return poll(&pollfd, 1, 0) > 0;
99b4405289SAsias He }
100b4405289SAsias He 
101b4405289SAsias He static void term_cleanup(void)
102b4405289SAsias He {
103b4405289SAsias He 	tcsetattr(STDIN_FILENO, TCSANOW, &orig_term);
104b4405289SAsias He }
105b4405289SAsias He 
106*f2a0b6a7SAnton Vorontsov static void term_sig_cleanup(int sig)
107*f2a0b6a7SAnton Vorontsov {
108*f2a0b6a7SAnton Vorontsov 	term_cleanup();
109*f2a0b6a7SAnton Vorontsov 	signal(sig, SIG_DFL);
110*f2a0b6a7SAnton Vorontsov 	raise(sig);
111*f2a0b6a7SAnton Vorontsov }
112*f2a0b6a7SAnton Vorontsov 
113b4405289SAsias He void term_init(void)
114b4405289SAsias He {
115b4405289SAsias He 	struct termios term;
116b4405289SAsias He 
117b4405289SAsias He 	if (tcgetattr(STDIN_FILENO, &orig_term) < 0)
118b4405289SAsias He 		die("unable to save initial standard input settings");
119b4405289SAsias He 
120b4405289SAsias He 	term = orig_term;
121b4405289SAsias He 	term.c_lflag &= ~(ICANON | ECHO | ISIG);
122b4405289SAsias He 	tcsetattr(STDIN_FILENO, TCSANOW, &term);
123b4405289SAsias He 
124*f2a0b6a7SAnton Vorontsov 	signal(SIGTERM, term_sig_cleanup);
125b4405289SAsias He 	atexit(term_cleanup);
126b4405289SAsias He }
127