static void client_sigwinch_handler(int sig) { client.need_resize = true; } static bool client_send_packet(Packet *pkt) { print_packet("client-send:", pkt); if (send_packet(server.socket, pkt)) return true; debug("FAILED\n"); server.running = false; return false; } static bool client_recv_packet(Packet *pkt) { if (recv_packet(server.socket, pkt)) { print_packet("client-recv:", pkt); return true; } debug("client-recv: FAILED\n"); server.running = false; return false; } static void client_restore_terminal(void) { if (!has_term) return; tcsetattr(STDIN_FILENO, TCSAFLUSH, &orig_term); if (alternate_buffer) { printf("\033[?25h\033[?1049l"); fflush(stdout); alternate_buffer = false; } } static void client_setup_terminal(void) { if (!has_term) return; atexit(client_restore_terminal); cur_term = orig_term; cur_term.c_iflag &= ~(IGNBRK|BRKINT|PARMRK|ISTRIP|INLCR|IGNCR|ICRNL|IXON|IXOFF); cur_term.c_oflag &= ~(OPOST); cur_term.c_lflag &= ~(ECHO|ECHONL|ICANON|ISIG|IEXTEN); cur_term.c_cflag &= ~(CSIZE|PARENB); cur_term.c_cflag |= CS8; cur_term.c_cc[VLNEXT] = _POSIX_VDISABLE; cur_term.c_cc[VMIN] = 1; cur_term.c_cc[VTIME] = 0; tcsetattr(STDIN_FILENO, TCSANOW, &cur_term); if (!alternate_buffer) { printf("\033[?1049h\033[H"); fflush(stdout); alternate_buffer = true; } } static int client_mainloop(void) { sigset_t emptyset, blockset; sigemptyset(&emptyset); sigemptyset(&blockset); sigaddset(&blockset, SIGWINCH); sigprocmask(SIG_BLOCK, &blockset, NULL); client.need_resize = true; client.await_command = false; Packet pkt = { .type = MSG_ATTACH, .u.i = client.flags, .len = sizeof(pkt.u.i), }; client_send_packet(&pkt); while (server.running) { fd_set fds; FD_ZERO(&fds); FD_SET(STDIN_FILENO, &fds); FD_SET(server.socket, &fds); if (client.need_resize) { struct winsize ws; if (ioctl(STDIN_FILENO, TIOCGWINSZ, &ws) != -1) { Packet pkt = { .type = MSG_RESIZE, .u = { .ws = { .rows = ws.ws_row, .cols = ws.ws_col } }, .len = sizeof(pkt.u.ws), }; if (client_send_packet(&pkt)) client.need_resize = false; } } if (pselect(server.socket+1, &fds, NULL, NULL, NULL, &emptyset) == -1) { if (errno == EINTR) continue; die("client-mainloop"); } if (FD_ISSET(server.socket, &fds)) { Packet pkt; if (client_recv_packet(&pkt)) { switch (pkt.type) { case MSG_CONTENT: write_all(STDOUT_FILENO, pkt.u.msg, pkt.len); break; case MSG_RESIZE: client.need_resize = true; break; case MSG_EXIT: client_send_packet(&pkt); close(server.socket); return pkt.u.i; } } } if (FD_ISSET(STDIN_FILENO, &fds)) { Packet pkt = { .type = MSG_CONTENT }; ssize_t len = read(STDIN_FILENO, pkt.u.msg, sizeof(pkt.u.msg)); if (len == -1 && errno != EAGAIN && errno != EINTR) die("client-stdin"); if (len > 0) { debug("client-stdin: %c\n", pkt.u.msg[0]); if (client.await_command) { client.await_command = false; int rv = 0; switch (pkt.u.msg[0]) { case 'd': rv = RV_DETACH; break; case '0': case '1': case '2': case '3': case '4': case '5': case '6': case '7': case '8': case '9': rv = RV_DETACH-(int)(pkt.u.msg[0]-'0')-1; break; } if (rv != 0) { pkt.type = MSG_DETACH; pkt.len = 0; client_send_packet(&pkt); close(server.socket); return rv; } } pkt.len = len; if (KEY_REDRAW && pkt.u.msg[0] == KEY_REDRAW) { client.need_resize = true; } else if (pkt.u.msg[0] == KEY_DETACH) { client.await_command = true; } else if (!(client.flags & CLIENT_READONLY)) { client_send_packet(&pkt); } } else if (len == 0) { debug("client-stdin: EOF\n"); return RV_DETACH; } } } return -EIO; }