mirror of
https://github.com/tmate-io/tmate.git
synced 2024-12-25 16:19:02 +01:00
ae2ea52575
redraw it as to draw the entire screen, just skip all lines but the last. This makes horizontal split redraw properly when the status line is off.
1136 lines
25 KiB
C
1136 lines
25 KiB
C
/* $OpenBSD$ */
|
|
|
|
/*
|
|
* 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 <sys/wait.h>
|
|
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <paths.h>
|
|
#include <signal.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <syslog.h>
|
|
#include <termios.h>
|
|
#include <time.h>
|
|
#include <unistd.h>
|
|
|
|
#include "tmux.h"
|
|
|
|
/*
|
|
* Main server functions.
|
|
*/
|
|
|
|
/* Client list. */
|
|
struct clients clients;
|
|
|
|
int server_create_socket(void);
|
|
int server_main(int);
|
|
void server_shutdown(void);
|
|
void server_child_signal(void);
|
|
void server_fill_windows(struct pollfd **);
|
|
void server_handle_windows(struct pollfd **);
|
|
void server_fill_clients(struct pollfd **);
|
|
void server_handle_clients(struct pollfd **);
|
|
struct client *server_accept_client(int);
|
|
void server_handle_client(struct client *);
|
|
void server_handle_window(struct window *, struct window_pane *);
|
|
int server_check_window_bell(struct session *, struct window *,
|
|
struct window_pane *);
|
|
int server_check_window_activity(struct session *,
|
|
struct window *);
|
|
int server_check_window_content(struct session *, struct window *,
|
|
struct window_pane *);
|
|
void server_lost_client(struct client *);
|
|
void server_check_window(struct window *);
|
|
void server_check_redraw(struct client *);
|
|
void server_redraw_locked(struct client *);
|
|
void server_check_timers(struct client *);
|
|
void server_second_timers(void);
|
|
int server_update_socket(void);
|
|
|
|
/* Create a new client. */
|
|
struct client *
|
|
server_create_client(int fd)
|
|
{
|
|
struct client *c;
|
|
int mode;
|
|
u_int i;
|
|
|
|
if ((mode = fcntl(fd, F_GETFL)) == -1)
|
|
fatal("fcntl failed");
|
|
if (fcntl(fd, F_SETFL, mode|O_NONBLOCK) == -1)
|
|
fatal("fcntl failed");
|
|
if (fcntl(fd, F_SETFD, FD_CLOEXEC) == -1)
|
|
fatal("fcntl failed");
|
|
|
|
c = xcalloc(1, sizeof *c);
|
|
c->fd = fd;
|
|
c->in = buffer_create(BUFSIZ);
|
|
c->out = buffer_create(BUFSIZ);
|
|
|
|
ARRAY_INIT(&c->prompt_hdata);
|
|
|
|
c->tty.fd = -1;
|
|
c->title = NULL;
|
|
|
|
c->session = NULL;
|
|
c->tty.sx = 80;
|
|
c->tty.sy = 25;
|
|
screen_init(&c->status, c->tty.sx, 1, 0);
|
|
|
|
c->message_string = NULL;
|
|
|
|
c->prompt_string = NULL;
|
|
c->prompt_buffer = NULL;
|
|
c->prompt_index = 0;
|
|
|
|
for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
|
|
if (ARRAY_ITEM(&clients, i) == NULL) {
|
|
ARRAY_SET(&clients, i, c);
|
|
return (c);
|
|
}
|
|
}
|
|
ARRAY_ADD(&clients, c);
|
|
return (c);
|
|
}
|
|
|
|
/* Find client index. */
|
|
int
|
|
server_client_index(struct client *c)
|
|
{
|
|
u_int i;
|
|
|
|
for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
|
|
if (c == ARRAY_ITEM(&clients, i))
|
|
return (i);
|
|
}
|
|
return (-1);
|
|
}
|
|
|
|
/* Fork new server. */
|
|
int
|
|
server_start(char *path)
|
|
{
|
|
int pair[2], srv_fd, null_fd;
|
|
char *cause;
|
|
char rpathbuf[MAXPATHLEN];
|
|
|
|
/* The first client is special and gets a socketpair; create it. */
|
|
if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, pair) != 0)
|
|
fatal("socketpair failed");
|
|
|
|
switch (fork()) {
|
|
case -1:
|
|
fatal("fork failed");
|
|
case 0:
|
|
break;
|
|
default:
|
|
close(pair[1]);
|
|
return (pair[0]);
|
|
}
|
|
close(pair[0]);
|
|
|
|
/*
|
|
* Must daemonise before loading configuration as the PID changes so
|
|
* $TMUX would be wrong for sessions created in the config file.
|
|
*/
|
|
if (daemon(1, 1) != 0)
|
|
fatal("daemon failed");
|
|
|
|
ARRAY_INIT(&windows);
|
|
ARRAY_INIT(&clients);
|
|
ARRAY_INIT(&sessions);
|
|
key_bindings_init();
|
|
utf8_build();
|
|
|
|
server_locked = 0;
|
|
server_password = NULL;
|
|
server_activity = time(NULL);
|
|
|
|
start_time = time(NULL);
|
|
socket_path = path;
|
|
|
|
if (access(SYSTEM_CFG, R_OK) != 0) {
|
|
if (errno != ENOENT) {
|
|
log_warn("%s", SYSTEM_CFG);
|
|
exit(1);
|
|
}
|
|
} else {
|
|
if (load_cfg(SYSTEM_CFG, &cause) != 0) {
|
|
log_warnx("%s", cause);
|
|
exit(1);
|
|
}
|
|
}
|
|
if (cfg_file != NULL && load_cfg(cfg_file, &cause) != 0) {
|
|
log_warnx("%s", cause);
|
|
exit(1);
|
|
}
|
|
logfile("server");
|
|
|
|
/*
|
|
* Close stdin/stdout/stderr. Can't let daemon() do this as they are
|
|
* needed until now to print configuration file errors.
|
|
*/
|
|
if ((null_fd = open(_PATH_DEVNULL, O_RDWR)) != -1) {
|
|
dup2(null_fd, STDIN_FILENO);
|
|
dup2(null_fd, STDOUT_FILENO);
|
|
dup2(null_fd, STDERR_FILENO);
|
|
if (null_fd > 2)
|
|
close(null_fd);
|
|
}
|
|
|
|
log_debug("server started, pid %ld", (long) getpid());
|
|
log_debug("socket path %s", socket_path);
|
|
|
|
if (realpath(socket_path, rpathbuf) == NULL)
|
|
strlcpy(rpathbuf, socket_path, sizeof rpathbuf);
|
|
setproctitle("server (%s)", rpathbuf);
|
|
|
|
srv_fd = server_create_socket();
|
|
server_create_client(pair[1]);
|
|
|
|
exit(server_main(srv_fd));
|
|
}
|
|
|
|
/* Create server socket. */
|
|
int
|
|
server_create_socket(void)
|
|
{
|
|
struct sockaddr_un sa;
|
|
size_t size;
|
|
mode_t mask;
|
|
int fd, mode;
|
|
|
|
memset(&sa, 0, sizeof sa);
|
|
sa.sun_family = AF_UNIX;
|
|
size = strlcpy(sa.sun_path, socket_path, sizeof sa.sun_path);
|
|
if (size >= sizeof sa.sun_path) {
|
|
errno = ENAMETOOLONG;
|
|
fatal("socket failed");
|
|
}
|
|
unlink(sa.sun_path);
|
|
|
|
if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
|
|
fatal("socket failed");
|
|
|
|
mask = umask(S_IXUSR|S_IRWXG|S_IRWXO);
|
|
if (bind(fd, (struct sockaddr *) &sa, SUN_LEN(&sa)) == -1)
|
|
fatal("bind failed");
|
|
umask(mask);
|
|
|
|
if (listen(fd, 16) == -1)
|
|
fatal("listen failed");
|
|
|
|
if ((mode = fcntl(fd, F_GETFL)) == -1)
|
|
fatal("fcntl failed");
|
|
if (fcntl(fd, F_SETFL, mode|O_NONBLOCK) == -1)
|
|
fatal("fcntl failed");
|
|
if (fcntl(fd, F_SETFD, FD_CLOEXEC) == -1)
|
|
fatal("fcntl failed");
|
|
|
|
return (fd);
|
|
}
|
|
|
|
/* Main server loop. */
|
|
int
|
|
server_main(int srv_fd)
|
|
{
|
|
struct window *w;
|
|
struct pollfd *pfds, *pfd;
|
|
int nfds, xtimeout;
|
|
u_int i, n;
|
|
time_t now, last;
|
|
|
|
siginit();
|
|
|
|
last = time(NULL);
|
|
|
|
pfds = NULL;
|
|
for (;;) {
|
|
/* If sigterm, kill all windows and clients. */
|
|
if (sigterm)
|
|
server_shutdown();
|
|
|
|
/* Handle child exit. */
|
|
if (sigchld) {
|
|
server_child_signal();
|
|
sigchld = 0;
|
|
}
|
|
|
|
/* Recreate socket on SIGUSR1. */
|
|
if (sigusr1) {
|
|
close(srv_fd);
|
|
srv_fd = server_create_socket();
|
|
sigusr1 = 0;
|
|
}
|
|
|
|
/* Initialise pollfd array. */
|
|
nfds = 1;
|
|
for (i = 0; i < ARRAY_LENGTH(&windows); i++) {
|
|
w = ARRAY_ITEM(&windows, i);
|
|
if (w != NULL)
|
|
nfds += window_count_panes(w);
|
|
}
|
|
nfds += ARRAY_LENGTH(&clients) * 2;
|
|
pfds = xrealloc(pfds, nfds, sizeof *pfds);
|
|
memset(pfds, 0, nfds * sizeof *pfds);
|
|
pfd = pfds;
|
|
|
|
/* Fill server socket. */
|
|
pfd->fd = srv_fd;
|
|
pfd->events = POLLIN;
|
|
pfd++;
|
|
|
|
/* Fill window and client sockets. */
|
|
server_fill_windows(&pfd);
|
|
server_fill_clients(&pfd);
|
|
|
|
/* Update socket permissions. */
|
|
xtimeout = INFTIM;
|
|
if (sigterm || server_update_socket() != 0)
|
|
xtimeout = POLL_TIMEOUT;
|
|
|
|
/* Do the poll. */
|
|
if (poll(pfds, nfds, xtimeout) == -1) {
|
|
if (errno == EAGAIN || errno == EINTR)
|
|
continue;
|
|
fatal("poll failed");
|
|
}
|
|
pfd = pfds;
|
|
|
|
/* Handle server socket. */
|
|
if (pfd->revents & (POLLERR|POLLNVAL|POLLHUP))
|
|
fatalx("lost server socket");
|
|
if (pfd->revents & POLLIN) {
|
|
server_accept_client(srv_fd);
|
|
continue;
|
|
}
|
|
pfd++;
|
|
|
|
/* Call second-based timers. */
|
|
now = time(NULL);
|
|
if (now != last) {
|
|
last = now;
|
|
server_second_timers();
|
|
}
|
|
|
|
/* Set window names. */
|
|
set_window_names();
|
|
|
|
/*
|
|
* Handle window and client sockets. Clients can create
|
|
* windows, so windows must come first to avoid messing up by
|
|
* increasing the array size.
|
|
*/
|
|
server_handle_windows(&pfd);
|
|
server_handle_clients(&pfd);
|
|
|
|
/* Collect any unset key bindings. */
|
|
key_bindings_clean();
|
|
|
|
/*
|
|
* If we have no sessions and clients left, let's get out
|
|
* of here...
|
|
*/
|
|
n = 0;
|
|
for (i = 0; i < ARRAY_LENGTH(&sessions); i++) {
|
|
if (ARRAY_ITEM(&sessions, i) != NULL)
|
|
n++;
|
|
}
|
|
for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
|
|
if (ARRAY_ITEM(&clients, i) != NULL)
|
|
n++;
|
|
}
|
|
if (n == 0)
|
|
break;
|
|
}
|
|
if (pfds != NULL)
|
|
xfree(pfds);
|
|
|
|
for (i = 0; i < ARRAY_LENGTH(&sessions); i++) {
|
|
if (ARRAY_ITEM(&sessions, i) != NULL)
|
|
session_destroy(ARRAY_ITEM(&sessions, i));
|
|
}
|
|
ARRAY_FREE(&sessions);
|
|
|
|
for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
|
|
if (ARRAY_ITEM(&clients, i) != NULL)
|
|
server_lost_client(ARRAY_ITEM(&clients, i));
|
|
}
|
|
ARRAY_FREE(&clients);
|
|
|
|
key_bindings_free();
|
|
|
|
close(srv_fd);
|
|
|
|
unlink(socket_path);
|
|
xfree(socket_path);
|
|
|
|
options_free(&global_s_options);
|
|
options_free(&global_w_options);
|
|
if (server_password != NULL)
|
|
xfree(server_password);
|
|
|
|
return (0);
|
|
}
|
|
|
|
/* Kill all clients. */
|
|
void
|
|
server_shutdown(void)
|
|
{
|
|
struct session *s;
|
|
struct client *c;
|
|
u_int i, j;
|
|
|
|
for (i = 0; i < ARRAY_LENGTH(&sessions); i++) {
|
|
s = ARRAY_ITEM(&sessions, i);
|
|
for (j = 0; j < ARRAY_LENGTH(&clients); j++) {
|
|
c = ARRAY_ITEM(&clients, j);
|
|
if (c != NULL && c->session == s) {
|
|
s = NULL;
|
|
break;
|
|
}
|
|
}
|
|
if (s != NULL)
|
|
session_destroy(s);
|
|
}
|
|
|
|
for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
|
|
c = ARRAY_ITEM(&clients, i);
|
|
if (c != NULL)
|
|
server_write_client(c, MSG_SHUTDOWN, NULL, 0);
|
|
}
|
|
}
|
|
|
|
/* Handle SIGCHLD. */
|
|
void
|
|
server_child_signal(void)
|
|
{
|
|
struct window *w;
|
|
struct window_pane *wp;
|
|
int status;
|
|
pid_t pid;
|
|
u_int i;
|
|
|
|
for (;;) {
|
|
switch (pid = waitpid(WAIT_ANY, &status, WNOHANG|WUNTRACED)) {
|
|
case -1:
|
|
if (errno == ECHILD)
|
|
return;
|
|
fatal("waitpid");
|
|
case 0:
|
|
return;
|
|
}
|
|
if (!WIFSTOPPED(status))
|
|
continue;
|
|
if (WSTOPSIG(status) == SIGTTIN || WSTOPSIG(status) == SIGTTOU)
|
|
continue;
|
|
|
|
for (i = 0; i < ARRAY_LENGTH(&windows); i++) {
|
|
w = ARRAY_ITEM(&windows, i);
|
|
if (w == NULL)
|
|
continue;
|
|
TAILQ_FOREACH(wp, &w->panes, entry) {
|
|
if (wp->pid == pid) {
|
|
if (killpg(pid, SIGCONT) != 0)
|
|
kill(pid, SIGCONT);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Fill window pollfds. */
|
|
void
|
|
server_fill_windows(struct pollfd **pfd)
|
|
{
|
|
struct window *w;
|
|
struct window_pane *wp;
|
|
u_int i;
|
|
|
|
for (i = 0; i < ARRAY_LENGTH(&windows); i++) {
|
|
w = ARRAY_ITEM(&windows, i);
|
|
if (w == NULL)
|
|
continue;
|
|
|
|
TAILQ_FOREACH(wp, &w->panes, entry) {
|
|
(*pfd)->fd = wp->fd;
|
|
if (wp->fd != -1) {
|
|
(*pfd)->events = POLLIN;
|
|
if (BUFFER_USED(wp->out) > 0)
|
|
(*pfd)->events |= POLLOUT;
|
|
}
|
|
(*pfd)++;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Handle window pollfds. */
|
|
void
|
|
server_handle_windows(struct pollfd **pfd)
|
|
{
|
|
struct window *w;
|
|
struct window_pane *wp;
|
|
u_int i;
|
|
|
|
for (i = 0; i < ARRAY_LENGTH(&windows); i++) {
|
|
w = ARRAY_ITEM(&windows, i);
|
|
if (w == NULL)
|
|
continue;
|
|
|
|
TAILQ_FOREACH(wp, &w->panes, entry) {
|
|
if (wp->fd != -1) {
|
|
if (buffer_poll(*pfd, wp->in, wp->out) != 0) {
|
|
close(wp->fd);
|
|
wp->fd = -1;
|
|
} else
|
|
server_handle_window(w, wp);
|
|
}
|
|
(*pfd)++;
|
|
}
|
|
|
|
server_check_window(w);
|
|
}
|
|
}
|
|
|
|
/* Check for general redraw on client. */
|
|
void
|
|
server_check_redraw(struct client *c)
|
|
{
|
|
struct session *s;
|
|
struct window_pane *wp;
|
|
char title[512];
|
|
int flags, redraw;
|
|
|
|
if (c == NULL || c->session == NULL)
|
|
return;
|
|
s = c->session;
|
|
|
|
flags = c->tty.flags & TTY_FREEZE;
|
|
c->tty.flags &= ~TTY_FREEZE;
|
|
|
|
if (options_get_number(&s->options, "set-titles")) {
|
|
xsnprintf(title, sizeof title, "%s:%u:%s - \"%s\"",
|
|
s->name, s->curw->idx, s->curw->window->name,
|
|
s->curw->window->active->screen->title);
|
|
if (c->title == NULL || strcmp(title, c->title) != 0) {
|
|
if (c->title != NULL)
|
|
xfree(c->title);
|
|
c->title = xstrdup(title);
|
|
tty_set_title(&c->tty, c->title);
|
|
}
|
|
}
|
|
|
|
if (c->flags & (CLIENT_REDRAW|CLIENT_STATUS)) {
|
|
if (c->message_string != NULL)
|
|
redraw = status_message_redraw(c);
|
|
else if (c->prompt_string != NULL)
|
|
redraw = status_prompt_redraw(c);
|
|
else
|
|
redraw = status_redraw(c);
|
|
if (!redraw)
|
|
c->flags &= ~CLIENT_STATUS;
|
|
}
|
|
|
|
if (c->flags & CLIENT_REDRAW) {
|
|
if (server_locked)
|
|
server_redraw_locked(c);
|
|
else
|
|
screen_redraw_screen(c, 0);
|
|
c->flags &= ~CLIENT_STATUS;
|
|
} else {
|
|
TAILQ_FOREACH(wp, &c->session->curw->window->panes, entry) {
|
|
if (wp->flags & PANE_REDRAW)
|
|
screen_redraw_pane(c, wp);
|
|
}
|
|
}
|
|
|
|
if (c->flags & CLIENT_STATUS)
|
|
screen_redraw_screen(c, 1);
|
|
|
|
c->tty.flags |= flags;
|
|
|
|
c->flags &= ~(CLIENT_REDRAW|CLIENT_STATUS);
|
|
}
|
|
|
|
/* Redraw client when locked. */
|
|
void
|
|
server_redraw_locked(struct client *c)
|
|
{
|
|
struct screen_write_ctx ctx;
|
|
struct screen screen;
|
|
u_int colour, xx, yy, i;
|
|
int style;
|
|
|
|
xx = c->tty.sx;
|
|
yy = c->tty.sy - 1;
|
|
if (xx == 0 || yy == 0)
|
|
return;
|
|
colour = options_get_number(&global_w_options, "clock-mode-colour");
|
|
style = options_get_number(&global_w_options, "clock-mode-style");
|
|
|
|
screen_init(&screen, xx, yy, 0);
|
|
|
|
screen_write_start(&ctx, NULL, &screen);
|
|
clock_draw(&ctx, colour, style);
|
|
screen_write_stop(&ctx);
|
|
|
|
for (i = 0; i < screen_size_y(&screen); i++)
|
|
tty_draw_line(&c->tty, &screen, i, 0, 0);
|
|
screen_redraw_screen(c, 1);
|
|
|
|
screen_free(&screen);
|
|
}
|
|
|
|
/* Check for timers on client. */
|
|
void
|
|
server_check_timers(struct client *c)
|
|
{
|
|
struct session *s;
|
|
struct timeval tv;
|
|
u_int interval;
|
|
|
|
if (c == NULL || c->session == NULL)
|
|
return;
|
|
s = c->session;
|
|
|
|
if (gettimeofday(&tv, NULL) != 0)
|
|
fatal("gettimeofday");
|
|
|
|
if (c->message_string != NULL && timercmp(&tv, &c->message_timer, >))
|
|
status_message_clear(c);
|
|
|
|
if (c->message_string != NULL || c->prompt_string != NULL) {
|
|
/*
|
|
* Don't need timed redraw for messages/prompts so bail now.
|
|
* The status timer isn't reset when they are redrawn anyway.
|
|
*/
|
|
return;
|
|
}
|
|
if (!options_get_number(&s->options, "status"))
|
|
return;
|
|
|
|
/* Check timer; resolution is only a second so don't be too clever. */
|
|
interval = options_get_number(&s->options, "status-interval");
|
|
if (interval == 0)
|
|
return;
|
|
if (tv.tv_sec < c->status_timer.tv_sec ||
|
|
((u_int) tv.tv_sec) - c->status_timer.tv_sec >= interval)
|
|
c->flags |= CLIENT_STATUS;
|
|
}
|
|
|
|
/* Fill client pollfds. */
|
|
void
|
|
server_fill_clients(struct pollfd **pfd)
|
|
{
|
|
struct client *c;
|
|
struct window *w;
|
|
struct window_pane *wp;
|
|
u_int i;
|
|
|
|
for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
|
|
c = ARRAY_ITEM(&clients, i);
|
|
|
|
server_check_timers(c);
|
|
server_check_redraw(c);
|
|
|
|
if (c == NULL)
|
|
(*pfd)->fd = -1;
|
|
else {
|
|
(*pfd)->fd = c->fd;
|
|
(*pfd)->events = POLLIN;
|
|
if (BUFFER_USED(c->out) > 0)
|
|
(*pfd)->events |= POLLOUT;
|
|
}
|
|
(*pfd)++;
|
|
|
|
if (c == NULL || c->flags & CLIENT_SUSPENDED ||
|
|
c->tty.fd == -1 || c->session == NULL)
|
|
(*pfd)->fd = -1;
|
|
else {
|
|
(*pfd)->fd = c->tty.fd;
|
|
(*pfd)->events = POLLIN;
|
|
if (BUFFER_USED(c->tty.out) > 0)
|
|
(*pfd)->events |= POLLOUT;
|
|
}
|
|
(*pfd)++;
|
|
}
|
|
|
|
/*
|
|
* Clear any window redraw flags (will have been redrawn as part of
|
|
* client).
|
|
*/
|
|
for (i = 0; i < ARRAY_LENGTH(&windows); i++) {
|
|
w = ARRAY_ITEM(&windows, i);
|
|
if (w == NULL)
|
|
continue;
|
|
|
|
w->flags &= ~WINDOW_REDRAW;
|
|
TAILQ_FOREACH(wp, &w->panes, entry)
|
|
wp->flags &= ~PANE_REDRAW;
|
|
}
|
|
}
|
|
|
|
/* Handle client pollfds. */
|
|
void
|
|
server_handle_clients(struct pollfd **pfd)
|
|
{
|
|
struct client *c;
|
|
u_int i;
|
|
|
|
for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
|
|
c = ARRAY_ITEM(&clients, i);
|
|
|
|
if (c != NULL) {
|
|
if (buffer_poll(*pfd, c->in, c->out) != 0) {
|
|
server_lost_client(c);
|
|
(*pfd) += 2;
|
|
continue;
|
|
} else
|
|
server_msg_dispatch(c);
|
|
}
|
|
(*pfd)++;
|
|
|
|
if (c != NULL && !(c->flags & CLIENT_SUSPENDED) &&
|
|
c->tty.fd != -1 && c->session != NULL) {
|
|
if (buffer_poll(*pfd, c->tty.in, c->tty.out) != 0)
|
|
server_lost_client(c);
|
|
else
|
|
server_handle_client(c);
|
|
}
|
|
(*pfd)++;
|
|
}
|
|
}
|
|
|
|
/* accept(2) and create new client. */
|
|
struct client *
|
|
server_accept_client(int srv_fd)
|
|
{
|
|
struct sockaddr_storage sa;
|
|
socklen_t slen = sizeof sa;
|
|
int fd;
|
|
|
|
fd = accept(srv_fd, (struct sockaddr *) &sa, &slen);
|
|
if (fd == -1) {
|
|
if (errno == EAGAIN || errno == EINTR || errno == ECONNABORTED)
|
|
return (NULL);
|
|
fatal("accept failed");
|
|
}
|
|
if (sigterm) {
|
|
close(fd);
|
|
return (NULL);
|
|
}
|
|
return (server_create_client(fd));
|
|
}
|
|
|
|
/* Input data from client. */
|
|
void
|
|
server_handle_client(struct client *c)
|
|
{
|
|
struct window_pane *wp;
|
|
struct screen *s;
|
|
struct timeval tv;
|
|
struct key_binding *bd;
|
|
int key, prefix, status, xtimeout;
|
|
int mode;
|
|
u_char mouse[3];
|
|
|
|
xtimeout = options_get_number(&c->session->options, "repeat-time");
|
|
if (xtimeout != 0 && c->flags & CLIENT_REPEAT) {
|
|
if (gettimeofday(&tv, NULL) != 0)
|
|
fatal("gettimeofday");
|
|
if (timercmp(&tv, &c->repeat_timer, >))
|
|
c->flags &= ~(CLIENT_PREFIX|CLIENT_REPEAT);
|
|
}
|
|
|
|
/* Process keys. */
|
|
prefix = options_get_number(&c->session->options, "prefix");
|
|
while (tty_keys_next(&c->tty, &key, mouse) == 0) {
|
|
server_activity = time(NULL);
|
|
|
|
if (c->session == NULL)
|
|
return;
|
|
wp = c->session->curw->window->active; /* could die */
|
|
|
|
status_message_clear(c);
|
|
if (c->prompt_string != NULL) {
|
|
status_prompt_key(c, key);
|
|
continue;
|
|
}
|
|
if (server_locked)
|
|
continue;
|
|
|
|
/* Check for mouse keys. */
|
|
if (key == KEYC_MOUSE) {
|
|
window_pane_mouse(wp, c, mouse[0], mouse[1], mouse[2]);
|
|
continue;
|
|
}
|
|
|
|
/* No previous prefix key. */
|
|
if (!(c->flags & CLIENT_PREFIX)) {
|
|
if (key == prefix)
|
|
c->flags |= CLIENT_PREFIX;
|
|
else
|
|
window_pane_key(wp, c, key);
|
|
continue;
|
|
}
|
|
|
|
/* Prefix key already pressed. Reset prefix and lookup key. */
|
|
c->flags &= ~CLIENT_PREFIX;
|
|
if ((bd = key_bindings_lookup(key)) == NULL) {
|
|
/* If repeating, treat this as a key, else ignore. */
|
|
if (c->flags & CLIENT_REPEAT) {
|
|
c->flags &= ~CLIENT_REPEAT;
|
|
if (key == prefix)
|
|
c->flags |= CLIENT_PREFIX;
|
|
else
|
|
window_pane_key(wp, c, key);
|
|
}
|
|
continue;
|
|
}
|
|
|
|
/* If already repeating, but this key can't repeat, skip it. */
|
|
if (c->flags & CLIENT_REPEAT && !bd->can_repeat) {
|
|
c->flags &= ~CLIENT_REPEAT;
|
|
if (key == prefix)
|
|
c->flags |= CLIENT_PREFIX;
|
|
else
|
|
window_pane_key(wp, c, key);
|
|
continue;
|
|
}
|
|
|
|
/* If this key can repeat, reset the repeat flags and timer. */
|
|
if (xtimeout != 0 && bd->can_repeat) {
|
|
c->flags |= CLIENT_PREFIX|CLIENT_REPEAT;
|
|
|
|
tv.tv_sec = xtimeout / 1000;
|
|
tv.tv_usec = (xtimeout % 1000) * 1000L;
|
|
if (gettimeofday(&c->repeat_timer, NULL) != 0)
|
|
fatal("gettimeofday");
|
|
timeradd(&c->repeat_timer, &tv, &c->repeat_timer);
|
|
}
|
|
|
|
/* Dispatch the command. */
|
|
key_bindings_dispatch(bd, c);
|
|
}
|
|
if (c->session == NULL)
|
|
return;
|
|
wp = c->session->curw->window->active; /* could die - do each loop */
|
|
s = wp->screen;
|
|
|
|
/* Ensure cursor position and mode settings. */
|
|
status = options_get_number(&c->session->options, "status");
|
|
if (wp->yoff + s->cy < c->tty.sy - status)
|
|
tty_cursor(&c->tty, s->cx, s->cy, wp->xoff, wp->yoff);
|
|
|
|
mode = s->mode;
|
|
if (server_locked)
|
|
mode &= ~TTY_NOCURSOR;
|
|
tty_update_mode(&c->tty, mode);
|
|
}
|
|
|
|
/* Lost a client. */
|
|
void
|
|
server_lost_client(struct client *c)
|
|
{
|
|
u_int i;
|
|
|
|
for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
|
|
if (ARRAY_ITEM(&clients, i) == c)
|
|
ARRAY_SET(&clients, i, NULL);
|
|
}
|
|
|
|
tty_free(&c->tty, c->flags & CLIENT_SUSPENDED);
|
|
|
|
screen_free(&c->status);
|
|
|
|
if (c->title != NULL)
|
|
xfree(c->title);
|
|
|
|
if (c->message_string != NULL)
|
|
xfree(c->message_string);
|
|
|
|
if (c->prompt_string != NULL)
|
|
xfree(c->prompt_string);
|
|
if (c->prompt_buffer != NULL)
|
|
xfree(c->prompt_buffer);
|
|
for (i = 0; i < ARRAY_LENGTH(&c->prompt_hdata); i++)
|
|
xfree(ARRAY_ITEM(&c->prompt_hdata, i));
|
|
ARRAY_FREE(&c->prompt_hdata);
|
|
|
|
if (c->cwd != NULL)
|
|
xfree(c->cwd);
|
|
|
|
close(c->fd);
|
|
buffer_destroy(c->in);
|
|
buffer_destroy(c->out);
|
|
xfree(c);
|
|
|
|
recalculate_sizes();
|
|
}
|
|
|
|
/* Handle window data. */
|
|
void
|
|
server_handle_window(struct window *w, struct window_pane *wp)
|
|
{
|
|
struct session *s;
|
|
u_int i;
|
|
int update;
|
|
|
|
window_pane_parse(wp);
|
|
|
|
if ((w->flags & (WINDOW_BELL|WINDOW_ACTIVITY|WINDOW_CONTENT)) == 0)
|
|
return;
|
|
|
|
update = 0;
|
|
for (i = 0; i < ARRAY_LENGTH(&sessions); i++) {
|
|
s = ARRAY_ITEM(&sessions, i);
|
|
if (s == NULL || !session_has(s, w))
|
|
continue;
|
|
|
|
update += server_check_window_bell(s, w, wp);
|
|
update += server_check_window_activity(s, w);
|
|
update += server_check_window_content(s, w, wp);
|
|
}
|
|
if (update)
|
|
server_status_window(w);
|
|
|
|
w->flags &= ~(WINDOW_BELL|WINDOW_ACTIVITY|WINDOW_CONTENT);
|
|
}
|
|
|
|
int
|
|
server_check_window_bell(
|
|
struct session *s, struct window *w, struct window_pane *wp)
|
|
{
|
|
struct client *c;
|
|
u_int i;
|
|
int action;
|
|
|
|
if (!(w->flags & WINDOW_BELL))
|
|
return (0);
|
|
if (session_alert_has_window(s, w, WINDOW_BELL))
|
|
return (0);
|
|
session_alert_add(s, w, WINDOW_BELL);
|
|
|
|
action = options_get_number(&s->options, "bell-action");
|
|
switch (action) {
|
|
case BELL_ANY:
|
|
if (s->flags & SESSION_UNATTACHED)
|
|
break;
|
|
for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
|
|
c = ARRAY_ITEM(&clients, i);
|
|
if (c != NULL && c->session == s)
|
|
tty_putcode(&c->tty, TTYC_BEL);
|
|
}
|
|
break;
|
|
case BELL_CURRENT:
|
|
if (w->active != wp)
|
|
break;
|
|
for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
|
|
c = ARRAY_ITEM(&clients, i);
|
|
if (c != NULL && c->session == s)
|
|
tty_putcode(&c->tty, TTYC_BEL);
|
|
}
|
|
break;
|
|
}
|
|
return (1);
|
|
}
|
|
|
|
int
|
|
server_check_window_activity(struct session *s, struct window *w)
|
|
{
|
|
if (!(w->flags & WINDOW_ACTIVITY))
|
|
return (0);
|
|
if (!options_get_number(&w->options, "monitor-activity"))
|
|
return (0);
|
|
if (session_alert_has_window(s, w, WINDOW_ACTIVITY))
|
|
return (0);
|
|
session_alert_add(s, w, WINDOW_ACTIVITY);
|
|
return (1);
|
|
}
|
|
|
|
int
|
|
server_check_window_content(
|
|
struct session *s, struct window *w, struct window_pane *wp)
|
|
{
|
|
char *found, *ptr;
|
|
|
|
if (!(w->flags & WINDOW_CONTENT))
|
|
return (0);
|
|
if ((ptr = options_get_string(&w->options, "monitor-content")) == NULL)
|
|
return (0);
|
|
if (*ptr == '\0')
|
|
return (0);
|
|
if (session_alert_has_window(s, w, WINDOW_CONTENT))
|
|
return (0);
|
|
if ((found = window_pane_search(wp, ptr, NULL)) == NULL)
|
|
return (0);
|
|
session_alert_add(s, w, WINDOW_CONTENT);
|
|
xfree(found);
|
|
return (1);
|
|
}
|
|
|
|
/* Check if window still exists. */
|
|
void
|
|
server_check_window(struct window *w)
|
|
{
|
|
struct window_pane *wp, *wq;
|
|
struct client *c;
|
|
struct session *s;
|
|
struct winlink *wl;
|
|
u_int i, j;
|
|
int destroyed, flag;
|
|
|
|
flag = options_get_number(&w->options, "remain-on-exit");
|
|
|
|
destroyed = 1;
|
|
|
|
wp = TAILQ_FIRST(&w->panes);
|
|
while (wp != NULL) {
|
|
wq = TAILQ_NEXT(wp, entry);
|
|
/*
|
|
* If the pane has died and the remain-on-exit flag is not set,
|
|
* remove the pane; otherwise, if the flag is set, don't allow
|
|
* the window to be destroyed (or it'll close when the last
|
|
* pane dies).
|
|
*/
|
|
if (wp->fd == -1 && !flag) {
|
|
window_remove_pane(w, wp);
|
|
server_redraw_window(w);
|
|
layout_refresh(w, 0);
|
|
} else
|
|
destroyed = 0;
|
|
wp = wq;
|
|
}
|
|
|
|
if (!destroyed)
|
|
return;
|
|
|
|
for (i = 0; i < ARRAY_LENGTH(&sessions); i++) {
|
|
s = ARRAY_ITEM(&sessions, i);
|
|
if (s == NULL)
|
|
continue;
|
|
if (!session_has(s, w))
|
|
continue;
|
|
|
|
restart:
|
|
/* Detach window and either redraw or kill clients. */
|
|
RB_FOREACH(wl, winlinks, &s->windows) {
|
|
if (wl->window != w)
|
|
continue;
|
|
destroyed = session_detach(s, wl);
|
|
for (j = 0; j < ARRAY_LENGTH(&clients); j++) {
|
|
c = ARRAY_ITEM(&clients, j);
|
|
if (c == NULL || c->session != s)
|
|
continue;
|
|
if (!destroyed) {
|
|
server_redraw_client(c);
|
|
continue;
|
|
}
|
|
c->session = NULL;
|
|
server_write_client(c, MSG_EXIT, NULL, 0);
|
|
}
|
|
/* If the session was destroyed, bail now. */
|
|
if (destroyed)
|
|
break;
|
|
goto restart;
|
|
}
|
|
}
|
|
|
|
recalculate_sizes();
|
|
}
|
|
|
|
/* Call any once-per-second timers. */
|
|
void
|
|
server_second_timers(void)
|
|
{
|
|
struct window *w;
|
|
struct window_pane *wp;
|
|
u_int i;
|
|
int xtimeout;
|
|
struct tm now, then;
|
|
static time_t last_t = 0;
|
|
time_t t;
|
|
|
|
t = time(NULL);
|
|
xtimeout = options_get_number(&global_s_options, "lock-after-time");
|
|
if (xtimeout > 0 && t > server_activity + xtimeout)
|
|
server_lock();
|
|
|
|
for (i = 0; i < ARRAY_LENGTH(&windows); i++) {
|
|
w = ARRAY_ITEM(&windows, i);
|
|
if (w == NULL)
|
|
continue;
|
|
|
|
TAILQ_FOREACH(wp, &w->panes, entry) {
|
|
if (wp->mode != NULL && wp->mode->timer != NULL)
|
|
wp->mode->timer(wp);
|
|
}
|
|
}
|
|
|
|
/* Check for a minute having passed. */
|
|
gmtime_r(&t, &now);
|
|
gmtime_r(&last_t, &then);
|
|
if (now.tm_min == then.tm_min)
|
|
return;
|
|
last_t = t;
|
|
|
|
/* If locked, redraw all clients. */
|
|
if (server_locked) {
|
|
for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
|
|
if (ARRAY_ITEM(&clients, i) != NULL)
|
|
server_redraw_client(ARRAY_ITEM(&clients, i));
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Update socket execute permissions based on whether sessions are attached. */
|
|
int
|
|
server_update_socket(void)
|
|
{
|
|
struct session *s;
|
|
u_int i;
|
|
static int last = -1;
|
|
int n;
|
|
|
|
n = 0;
|
|
for (i = 0; i < ARRAY_LENGTH(&sessions); i++) {
|
|
s = ARRAY_ITEM(&sessions, i);
|
|
if (s != NULL && !(s->flags & SESSION_UNATTACHED)) {
|
|
n++;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (n != last) {
|
|
last = n;
|
|
if (n != 0)
|
|
chmod(socket_path, S_IRWXU);
|
|
else
|
|
chmod(socket_path, S_IRUSR|S_IWUSR);
|
|
}
|
|
|
|
return (n);
|
|
}
|