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