xref: /kvmtool/term.c (revision 492aa8f37a1364cf5b53de096782d967ba13d663)
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>
7f2a0b6a7SAnton Vorontsov #include <signal.h>
81add4b76SSasha Levin #include <pty.h>
91add4b76SSasha Levin #include <utmp.h>
10b4405289SAsias He 
11b4405289SAsias He #include "kvm/read-write.h"
12b4405289SAsias He #include "kvm/term.h"
13b4405289SAsias He #include "kvm/util.h"
145358b0e6SSasha Levin #include "kvm/kvm.h"
15629ad9d3SSasha Levin #include "kvm/kvm-cpu.h"
16b4405289SAsias He 
171add4b76SSasha Levin #define TERM_FD_IN      0
181add4b76SSasha Levin #define TERM_FD_OUT     1
191add4b76SSasha Levin 
20b4405289SAsias He static struct termios	orig_term;
21b4405289SAsias He 
2216588013SAsias He int term_escape_char	= 0x01; /* ctrl-a is used for escape */
2316588013SAsias He bool term_got_escape	= false;
2416588013SAsias He 
257787f0ffSJonathan Austin int term_fds[TERM_MAX_DEVS][2];
261add4b76SSasha Levin 
2712c406a8SJonathan Austin static pthread_t term_poll_thread;
2812c406a8SJonathan Austin 
294346fd8fSSasha Levin int term_getc(struct kvm *kvm, int term)
30b4405289SAsias He {
31e382487aSMatt Evans 	unsigned char c;
32b4405289SAsias He 
331add4b76SSasha Levin 	if (read_in_full(term_fds[term][TERM_FD_IN], &c, 1) < 0)
34b4405289SAsias He 		return -1;
3516588013SAsias He 
3616588013SAsias He 	if (term_got_escape) {
3716588013SAsias He 		term_got_escape = false;
38c23d9748SSasha Levin 		if (c == 'x')
39df4239fbSSasha Levin 			kvm_cpu__reboot(kvm);
4016588013SAsias He 		if (c == term_escape_char)
4116588013SAsias He 			return c;
4216588013SAsias He 	}
4316588013SAsias He 
4416588013SAsias He 	if (c == term_escape_char) {
4516588013SAsias He 		term_got_escape = true;
4616588013SAsias He 		return -1;
4716588013SAsias He 	}
4816588013SAsias He 
49b4405289SAsias He 	return c;
50b4405289SAsias He }
51b4405289SAsias He 
522651ea58SSasha Levin int term_putc(char *addr, int cnt, int term)
53b4405289SAsias He {
541add4b76SSasha Levin 	int ret;
55a5f4f263SDavid Daney 	int num_remaining = cnt;
561add4b76SSasha Levin 
57a5f4f263SDavid Daney 	while (num_remaining) {
58a5f4f263SDavid Daney 		ret = write(term_fds[term][TERM_FD_OUT], addr, num_remaining);
591add4b76SSasha Levin 		if (ret < 0)
60*492aa8f3SAndreas Herrmann 			return cnt - num_remaining;
61a5f4f263SDavid Daney 		num_remaining -= ret;
62a5f4f263SDavid Daney 		addr += ret;
631add4b76SSasha Levin 	}
64b4405289SAsias He 
65b4405289SAsias He 	return cnt;
66b4405289SAsias He }
67b4405289SAsias He 
684346fd8fSSasha Levin int term_getc_iov(struct kvm *kvm, struct iovec *iov, int iovcnt, int term)
69b4405289SAsias He {
7016588013SAsias He 	int c;
71b4405289SAsias He 
724346fd8fSSasha Levin 	c = term_getc(kvm, term);
7316588013SAsias He 
7416588013SAsias He 	if (c < 0)
7516588013SAsias He 		return 0;
7616588013SAsias He 
77e382487aSMatt Evans 	*((char *)iov[TERM_FD_IN].iov_base)	= (char)c;
7816588013SAsias He 
79fbf15b30SAsias He 	return sizeof(char);
80b4405289SAsias He }
81b4405289SAsias He 
822651ea58SSasha Levin int term_putc_iov(struct iovec *iov, int iovcnt, int term)
83b4405289SAsias He {
841add4b76SSasha Levin 	return writev(term_fds[term][TERM_FD_OUT], iov, iovcnt);
85b4405289SAsias He }
86b4405289SAsias He 
872651ea58SSasha Levin bool term_readable(int term)
88b4405289SAsias He {
89b4405289SAsias He 	struct pollfd pollfd = (struct pollfd) {
901add4b76SSasha Levin 		.fd	= term_fds[term][TERM_FD_IN],
91b4405289SAsias He 		.events	= POLLIN,
92b4405289SAsias He 		.revents = 0,
93b4405289SAsias He 	};
943ad9fad4SMarc Zyngier 	int err;
95b4405289SAsias He 
963ad9fad4SMarc Zyngier 	err = poll(&pollfd, 1, 0);
973ad9fad4SMarc Zyngier 	return (err > 0 && (pollfd.revents & POLLIN));
98b4405289SAsias He }
99b4405289SAsias He 
10012c406a8SJonathan Austin static void *term_poll_thread_loop(void *param)
10112c406a8SJonathan Austin {
10212c406a8SJonathan Austin 	struct pollfd fds[TERM_MAX_DEVS];
10312c406a8SJonathan Austin 	struct kvm *kvm = (struct kvm *) param;
10412c406a8SJonathan Austin 
10512c406a8SJonathan Austin 	int i;
10612c406a8SJonathan Austin 
10712c406a8SJonathan Austin 	for (i = 0; i < TERM_MAX_DEVS; i++) {
10812c406a8SJonathan Austin 		fds[i].fd = term_fds[i][TERM_FD_IN];
10912c406a8SJonathan Austin 		fds[i].events = POLLIN;
11012c406a8SJonathan Austin 		fds[i].revents = 0;
11112c406a8SJonathan Austin 	}
11212c406a8SJonathan Austin 
11312c406a8SJonathan Austin 	while (1) {
11412c406a8SJonathan Austin 		/* Poll with infinite timeout */
11512c406a8SJonathan Austin 		if(poll(fds, TERM_MAX_DEVS, -1) < 1)
11612c406a8SJonathan Austin 			break;
11712c406a8SJonathan Austin 		kvm__arch_read_term(kvm);
11812c406a8SJonathan Austin 	}
11912c406a8SJonathan Austin 
12012c406a8SJonathan Austin 	die("term_poll_thread_loop: error polling device fds %d\n", errno);
12112c406a8SJonathan Austin 	return NULL;
12212c406a8SJonathan Austin }
12312c406a8SJonathan Austin 
124b4405289SAsias He static void term_cleanup(void)
125b4405289SAsias He {
1261add4b76SSasha Levin 	int i;
1271add4b76SSasha Levin 
1287787f0ffSJonathan Austin 	for (i = 0; i < TERM_MAX_DEVS; i++)
1291add4b76SSasha Levin 		tcsetattr(term_fds[i][TERM_FD_IN], TCSANOW, &orig_term);
130b4405289SAsias He }
131b4405289SAsias He 
132f2a0b6a7SAnton Vorontsov static void term_sig_cleanup(int sig)
133f2a0b6a7SAnton Vorontsov {
134f2a0b6a7SAnton Vorontsov 	term_cleanup();
135f2a0b6a7SAnton Vorontsov 	signal(sig, SIG_DFL);
136f2a0b6a7SAnton Vorontsov 	raise(sig);
137f2a0b6a7SAnton Vorontsov }
138f2a0b6a7SAnton Vorontsov 
1391add4b76SSasha Levin void term_set_tty(int term)
1401add4b76SSasha Levin {
1411add4b76SSasha Levin 	struct termios orig_term;
1421add4b76SSasha Levin 	int master, slave;
1431add4b76SSasha Levin 	char new_pty[PATH_MAX];
1441add4b76SSasha Levin 
1451add4b76SSasha Levin 	if (tcgetattr(STDIN_FILENO, &orig_term) < 0)
1461add4b76SSasha Levin 		die("unable to save initial standard input settings");
1471add4b76SSasha Levin 
1481add4b76SSasha Levin 	orig_term.c_lflag &= ~(ICANON | ECHO | ISIG);
1491add4b76SSasha Levin 
1501add4b76SSasha Levin 	if (openpty(&master, &slave, new_pty, &orig_term, NULL) < 0)
1511add4b76SSasha Levin 		return;
1521add4b76SSasha Levin 
1531add4b76SSasha Levin 	close(slave);
1541add4b76SSasha Levin 
1551add4b76SSasha Levin 	pr_info("Assigned terminal %d to pty %s\n", term, new_pty);
1561add4b76SSasha Levin 
1571add4b76SSasha Levin 	term_fds[term][TERM_FD_IN] = term_fds[term][TERM_FD_OUT] = master;
1581add4b76SSasha Levin }
1591add4b76SSasha Levin 
160dca745e4SSasha Levin int tty_parser(const struct option *opt, const char *arg, int unset)
161dca745e4SSasha Levin {
162dca745e4SSasha Levin 	int tty = atoi(arg);
163dca745e4SSasha Levin 
164dca745e4SSasha Levin 	term_set_tty(tty);
165dca745e4SSasha Levin 
166dca745e4SSasha Levin 	return 0;
167dca745e4SSasha Levin }
168dca745e4SSasha Levin 
169dca745e4SSasha Levin int term_init(struct kvm *kvm)
170b4405289SAsias He {
171b4405289SAsias He 	struct termios term;
172dca745e4SSasha Levin 	int i, r;
173b4405289SAsias He 
1747787f0ffSJonathan Austin 	for (i = 0; i < TERM_MAX_DEVS; i++)
17536ad4dcfSMichael Ellerman 		if (term_fds[i][TERM_FD_IN] == 0) {
17636ad4dcfSMichael Ellerman 			term_fds[i][TERM_FD_IN] = STDIN_FILENO;
17736ad4dcfSMichael Ellerman 			term_fds[i][TERM_FD_OUT] = STDOUT_FILENO;
17836ad4dcfSMichael Ellerman 		}
17936ad4dcfSMichael Ellerman 
18036ad4dcfSMichael Ellerman 	if (!isatty(STDIN_FILENO) || !isatty(STDOUT_FILENO))
18136ad4dcfSMichael Ellerman 		return 0;
18236ad4dcfSMichael Ellerman 
183dca745e4SSasha Levin 	r = tcgetattr(STDIN_FILENO, &orig_term);
184dca745e4SSasha Levin 	if (r < 0) {
185dca745e4SSasha Levin 		pr_warning("unable to save initial standard input settings");
186dca745e4SSasha Levin 		return r;
187dca745e4SSasha Levin 	}
188dca745e4SSasha Levin 
189b4405289SAsias He 
190b4405289SAsias He 	term = orig_term;
191b4405289SAsias He 	term.c_lflag &= ~(ICANON | ECHO | ISIG);
192b4405289SAsias He 	tcsetattr(STDIN_FILENO, TCSANOW, &term);
193b4405289SAsias He 
19412c406a8SJonathan Austin 
19512c406a8SJonathan Austin 	/* Use our own blocking thread to read stdin, don't require a tick */
19612c406a8SJonathan Austin 	if(pthread_create(&term_poll_thread, NULL, term_poll_thread_loop,kvm))
19712c406a8SJonathan Austin 		die("Unable to create console input poll thread\n");
19812c406a8SJonathan Austin 
199f2a0b6a7SAnton Vorontsov 	signal(SIGTERM, term_sig_cleanup);
200b4405289SAsias He 	atexit(term_cleanup);
201dca745e4SSasha Levin 
202dca745e4SSasha Levin 	return 0;
203dca745e4SSasha Levin }
20449a8afd1SSasha Levin dev_init(term_init);
205dca745e4SSasha Levin 
206dca745e4SSasha Levin int term_exit(struct kvm *kvm)
207dca745e4SSasha Levin {
208dca745e4SSasha Levin 	return 0;
209b4405289SAsias He }
21049a8afd1SSasha Levin dev_exit(term_exit);
211