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