/* $Id: client.c,v 1.29 2008-06-07 07:27:28 nicm Exp $ */ /* * Copyright (c) 2007 Nicholas Marriott <nicm@users.sourceforge.net> * * Permission to use, copy, modify, and distribute this software for any * purpose with or without fee is hereby granted, provided that the above * copyright notice and this permission notice appear in all copies. * * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */ #include <sys/types.h> #include <sys/ioctl.h> #include <sys/socket.h> #include <sys/stat.h> #include <sys/un.h> #include <errno.h> #include <fcntl.h> #include <paths.h> #include <stdlib.h> #include <string.h> #include <syslog.h> #include <unistd.h> #include "tmux.h" void client_handle_winch(struct client_ctx *); int client_init(const char *path, struct client_ctx *cctx, int start_server) { struct sockaddr_un sa; struct stat sb; struct msg_identify_data data; struct winsize ws; size_t size; int mode; u_int retries; struct buffer *b; pid_t pid; pid = 0; retries = 0; retry: if (stat(path, &sb) != 0) { if (start_server && errno == ENOENT) { if (server_start(path) != 0) goto no_start; goto retry; } goto fail; } if (!S_ISSOCK(sb.st_mode)) { errno = ENOTSOCK; goto fail; } memset(&sa, 0, sizeof sa); sa.sun_family = AF_UNIX; size = strlcpy(sa.sun_path, path, sizeof sa.sun_path); if (size >= sizeof sa.sun_path) { errno = ENAMETOOLONG; goto fail; } if ((cctx->srv_fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) fatal("socket"); if (connect( cctx->srv_fd, (struct sockaddr *) &sa, SUN_LEN(&sa)) == -1) { if (start_server && errno == ECONNREFUSED && retries < 10) { if (unlink(path) != 0) goto fail; if (server_start(path) != 0) goto no_start; } goto fail; } if ((mode = fcntl(cctx->srv_fd, F_GETFL)) == -1) fatal("fcntl"); if (fcntl(cctx->srv_fd, F_SETFL, mode|O_NONBLOCK) == -1) fatal("fcntl"); cctx->srv_in = buffer_create(BUFSIZ); cctx->srv_out = buffer_create(BUFSIZ); if (isatty(STDIN_FILENO)) { if (ioctl(STDIN_FILENO, TIOCGWINSZ, &ws) == -1) fatal("ioctl(TIOCGWINSZ)"); data.sx = ws.ws_col; data.sy = ws.ws_row; if (ttyname_r(STDIN_FILENO, data.tty, sizeof data.tty) != 0) fatal("ttyname_r failed"); b = buffer_create(BUFSIZ); cmd_send_string(b, getenv("TERM")); client_write_server2(cctx, MSG_IDENTIFY, &data, sizeof data, BUFFER_OUT(b), BUFFER_USED(b)); buffer_destroy(b); } return (0); no_start: log_warnx("server failed to start"); return (1); fail: log_warn("server not found"); return (1); } int client_main(struct client_ctx *cctx) { struct pollfd pfd; char *error; int timeout; siginit(); logfile("client"); #ifndef NO_SETPROCTITLE setproctitle("client"); #endif error = NULL; timeout = INFTIM; while (!sigterm) { if (sigwinch) client_handle_winch(cctx); switch (client_msg_dispatch(cctx, &error)) { case -1: goto out; case 0: /* May be more in buffer, don't let poll block. */ timeout = 0; break; default: /* Out of data, poll may block. */ timeout = INFTIM; break; } pfd.fd = cctx->srv_fd; pfd.events = POLLIN; if (BUFFER_USED(cctx->srv_out) > 0) pfd.events |= POLLOUT; if (poll(&pfd, 1, timeout) == -1) { if (errno == EAGAIN || errno == EINTR) continue; fatal("poll failed"); } if (buffer_poll(&pfd, cctx->srv_in, cctx->srv_out) != 0) goto server_dead; } out: if (sigterm) { printf("[terminated]\n"); return (1); } if (cctx->flags & CCTX_EXIT) { printf("[exited]\n"); return (0); } if (cctx->flags & CCTX_DETACH) { printf("[detached]\n"); return (0); } printf("[error: %s]\n", error); return (1); server_dead: printf("[lost server]\n"); return (0); } void client_handle_winch(struct client_ctx *cctx) { struct msg_resize_data data; struct winsize ws; if (ioctl(STDIN_FILENO, TIOCGWINSZ, &ws) == -1) fatal("ioctl failed"); data.sx = ws.ws_col; data.sy = ws.ws_row; client_write_server(cctx, MSG_RESIZE, &data, sizeof data); sigwinch = 0; }