[BACK]Return to server-client.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / tmux

Annotation of src/usr.bin/tmux/server-client.c, Revision 1.161

1.161   ! nicm        1: /* $OpenBSD: server-client.c,v 1.160 2015/10/26 17:17:06 nicm Exp $ */
1.1       nicm        2:
                      3: /*
                      4:  * Copyright (c) 2009 Nicholas Marriott <nicm@users.sourceforge.net>
                      5:  *
                      6:  * Permission to use, copy, modify, and distribute this software for any
                      7:  * purpose with or without fee is hereby granted, provided that the above
                      8:  * copyright notice and this permission notice appear in all copies.
                      9:  *
                     10:  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
                     11:  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
                     12:  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
                     13:  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
                     14:  * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
                     15:  * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
                     16:  * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
                     17:  */
                     18:
                     19: #include <sys/types.h>
1.92      nicm       20: #include <sys/ioctl.h>
1.1       nicm       21:
1.114     benno      22: #include <errno.h>
1.12      nicm       23: #include <event.h>
1.1       nicm       24: #include <fcntl.h>
1.98      nicm       25: #include <paths.h>
                     26: #include <stdlib.h>
1.1       nicm       27: #include <string.h>
1.4       nicm       28: #include <time.h>
1.1       nicm       29: #include <unistd.h>
                     30:
                     31: #include "tmux.h"
                     32:
1.132     nicm       33: void   server_client_key_table(struct client *, const char *);
1.141     nicm       34: void   server_client_free(int, short, void *);
1.91      nicm       35: void   server_client_check_focus(struct window_pane *);
1.92      nicm       36: void   server_client_check_resize(struct window_pane *);
1.131     nicm       37: int    server_client_check_mouse(struct client *);
1.17      nicm       38: void   server_client_repeat_timer(int, short, void *);
1.36      nicm       39: void   server_client_check_exit(struct client *);
1.1       nicm       40: void   server_client_check_redraw(struct client *);
                     41: void   server_client_set_title(struct client *);
1.18      nicm       42: void   server_client_reset_state(struct client *);
1.82      nicm       43: int    server_client_assume_paste(struct session *);
1.1       nicm       44:
                     45: int    server_client_msg_dispatch(struct client *);
1.108     nicm       46: void   server_client_msg_command(struct client *, struct imsg *);
1.109     nicm       47: void   server_client_msg_identify(struct client *, struct imsg *);
1.1       nicm       48: void   server_client_msg_shell(struct client *);
1.140     nicm       49:
                     50: /* Check if this client is inside this server. */
                     51: int
                     52: server_client_check_nested(struct client *c)
                     53: {
                     54:        struct environ_entry    *envent;
                     55:        struct window_pane      *wp;
                     56:
                     57:        if (c->tty.path == NULL)
                     58:                return (0);
                     59:
                     60:        envent = environ_find(&c->environ, "TMUX");
                     61:        if (envent == NULL || *envent->value == '\0')
                     62:                return (0);
                     63:
                     64:        RB_FOREACH(wp, window_pane_tree, &all_window_panes) {
                     65:                if (strcmp(wp->tty, c->tty.path) == 0)
                     66:                        return (1);
                     67:        }
                     68:        return (0);
                     69: }
1.1       nicm       70:
1.132     nicm       71: /* Set client key table. */
                     72: void
                     73: server_client_key_table(struct client *c, const char *name)
                     74: {
                     75:        key_bindings_unref_table(c->keytable);
                     76:        c->keytable = key_bindings_get_table(name, 1);
                     77:        c->keytable->references++;
                     78: }
                     79:
1.1       nicm       80: /* Create a new client. */
                     81: void
                     82: server_client_create(int fd)
                     83: {
                     84:        struct client   *c;
                     85:
1.49      nicm       86:        setblocking(fd, 0);
1.1       nicm       87:
                     88:        c = xcalloc(1, sizeof *c);
1.141     nicm       89:        c->references = 1;
1.1       nicm       90:        imsg_init(&c->ibuf, fd);
1.14      nicm       91:        server_update_event(c);
1.25      nicm       92:
1.10      nicm       93:        if (gettimeofday(&c->creation_time, NULL) != 0)
1.1       nicm       94:                fatal("gettimeofday failed");
1.11      nicm       95:        memcpy(&c->activity_time, &c->creation_time, sizeof c->activity_time);
1.111     nicm       96:
                     97:        environ_init(&c->environ);
1.1       nicm       98:
1.146     nicm       99:        c->fd = -1;
1.145     nicm      100:        c->cwd = -1;
                    101:
1.94      nicm      102:        c->cmdq = cmdq_new(c);
                    103:        c->cmdq->client_exit = 1;
                    104:
1.116     nicm      105:        c->stdin_data = evbuffer_new();
                    106:        c->stdout_data = evbuffer_new();
                    107:        c->stderr_data = evbuffer_new();
1.33      nicm      108:
1.1       nicm      109:        c->tty.fd = -1;
                    110:        c->title = NULL;
                    111:
                    112:        c->session = NULL;
1.45      nicm      113:        c->last_session = NULL;
1.1       nicm      114:        c->tty.sx = 80;
                    115:        c->tty.sy = 24;
                    116:
                    117:        screen_init(&c->status, c->tty.sx, 1, 0);
                    118:
                    119:        c->message_string = NULL;
1.136     nicm      120:        TAILQ_INIT(&c->message_log);
1.1       nicm      121:
                    122:        c->prompt_string = NULL;
                    123:        c->prompt_buffer = NULL;
                    124:        c->prompt_index = 0;
                    125:
1.93      nicm      126:        c->flags |= CLIENT_FOCUSED;
                    127:
1.132     nicm      128:        c->keytable = key_bindings_get_table("root", 1);
                    129:        c->keytable->references++;
                    130:
1.17      nicm      131:        evtimer_set(&c->repeat_timer, server_client_repeat_timer, c);
                    132:
1.135     nicm      133:        TAILQ_INSERT_TAIL(&clients, c, entry);
1.157     nicm      134:        log_debug("new client %p", c);
1.72      nicm      135: }
                    136:
                    137: /* Open client terminal if needed. */
                    138: int
1.119     nicm      139: server_client_open(struct client *c, char **cause)
1.72      nicm      140: {
1.75      nicm      141:        if (c->flags & CLIENT_CONTROL)
                    142:                return (0);
1.120     nicm      143:
                    144:        if (strcmp(c->ttyname, "/dev/tty") == 0) {
                    145:                *cause = xstrdup("can't use /dev/tty");
                    146:                return (-1);
                    147:        }
1.75      nicm      148:
1.72      nicm      149:        if (!(c->flags & CLIENT_TERMINAL)) {
1.116     nicm      150:                *cause = xstrdup("not a terminal");
1.72      nicm      151:                return (-1);
                    152:        }
                    153:
1.119     nicm      154:        if (tty_open(&c->tty, cause) != 0)
1.72      nicm      155:                return (-1);
                    156:
                    157:        return (0);
1.1       nicm      158: }
                    159:
                    160: /* Lost a client. */
                    161: void
                    162: server_client_lost(struct client *c)
                    163: {
1.136     nicm      164:        struct message_entry    *msg, *msg1;
1.1       nicm      165:
1.141     nicm      166:        c->flags |= CLIENT_DEAD;
                    167:
                    168:        status_prompt_clear(c);
                    169:        status_message_clear(c);
                    170:
                    171:        if (c->stdin_callback != NULL)
                    172:                c->stdin_callback(c, 1, c->stdin_callback_data);
                    173:
1.135     nicm      174:        TAILQ_REMOVE(&clients, c, entry);
1.157     nicm      175:        log_debug("lost client %p", c);
1.1       nicm      176:
                    177:        /*
                    178:         * If CLIENT_TERMINAL hasn't been set, then tty_init hasn't been called
                    179:         * and tty_free might close an unrelated fd.
                    180:         */
                    181:        if (c->flags & CLIENT_TERMINAL)
                    182:                tty_free(&c->tty);
1.109     nicm      183:        free(c->ttyname);
                    184:        free(c->term);
1.1       nicm      185:
1.113     nicm      186:        evbuffer_free(c->stdin_data);
                    187:        evbuffer_free(c->stdout_data);
1.97      nicm      188:        if (c->stderr_data != c->stdout_data)
1.126     nicm      189:                evbuffer_free(c->stderr_data);
1.33      nicm      190:
1.148     nicm      191:        if (event_initialized(&c->status_timer))
                    192:                evtimer_del(&c->status_timer);
1.1       nicm      193:        screen_free(&c->status);
                    194:
1.77      nicm      195:        free(c->title);
1.109     nicm      196:        close(c->cwd);
1.1       nicm      197:
1.17      nicm      198:        evtimer_del(&c->repeat_timer);
                    199:
1.132     nicm      200:        key_bindings_unref_table(c->keytable);
                    201:
1.69      nicm      202:        if (event_initialized(&c->identify_timer))
                    203:                evtimer_del(&c->identify_timer);
1.15      nicm      204:
1.77      nicm      205:        free(c->message_string);
1.116     nicm      206:        if (event_initialized(&c->message_timer))
1.69      nicm      207:                evtimer_del(&c->message_timer);
1.136     nicm      208:        TAILQ_FOREACH_SAFE(msg, &c->message_log, entry, msg1) {
1.77      nicm      209:                free(msg->msg);
1.136     nicm      210:                TAILQ_REMOVE(&c->message_log, msg, entry);
                    211:                free(msg);
1.21      nicm      212:        }
1.1       nicm      213:
1.77      nicm      214:        free(c->prompt_string);
                    215:        free(c->prompt_buffer);
1.61      nicm      216:
1.154     nicm      217:        c->cmdq->flags |= CMD_Q_DEAD;
1.94      nicm      218:        cmdq_free(c->cmdq);
                    219:        c->cmdq = NULL;
                    220:
1.61      nicm      221:        environ_free(&c->environ);
1.1       nicm      222:
                    223:        close(c->ibuf.fd);
                    224:        imsg_clear(&c->ibuf);
1.69      nicm      225:        if (event_initialized(&c->event))
                    226:                event_del(&c->event);
1.13      nicm      227:
1.142     nicm      228:        server_client_unref(c);
1.71      nicm      229:
                    230:        server_add_accept(0); /* may be more file descriptors now */
1.1       nicm      231:
                    232:        recalculate_sizes();
1.41      nicm      233:        server_check_unattached();
1.19      nicm      234:        server_update_socket();
1.141     nicm      235: }
                    236:
                    237: /* Remove reference from a client. */
                    238: void
1.142     nicm      239: server_client_unref(struct client *c)
1.141     nicm      240: {
1.157     nicm      241:        log_debug("unref client %p (%d references)", c, c->references);
1.141     nicm      242:
                    243:        c->references--;
                    244:        if (c->references == 0)
                    245:                event_once(-1, EV_TIMEOUT, server_client_free, c, NULL);
                    246: }
                    247:
                    248: /* Free dead client. */
                    249: void
                    250: server_client_free(unused int fd, unused short events, void *arg)
                    251: {
                    252:        struct client   *c = arg;
                    253:
1.157     nicm      254:        log_debug("free client %p (%d references)", c, c->references);
1.141     nicm      255:
                    256:        if (c->references == 0)
                    257:                free(c);
1.9       nicm      258: }
                    259:
1.1       nicm      260: /* Process a single client event. */
                    261: void
1.12      nicm      262: server_client_callback(int fd, short events, void *data)
1.1       nicm      263: {
                    264:        struct client   *c = data;
1.7       nicm      265:
                    266:        if (c->flags & CLIENT_DEAD)
                    267:                return;
1.1       nicm      268:
                    269:        if (fd == c->ibuf.fd) {
1.122     krw       270:                if (events & EV_WRITE && msgbuf_write(&c->ibuf.w) <= 0 &&
1.114     benno     271:                    errno != EAGAIN)
1.1       nicm      272:                        goto client_lost;
                    273:
                    274:                if (c->flags & CLIENT_BAD) {
                    275:                        if (c->ibuf.w.queued == 0)
                    276:                                goto client_lost;
                    277:                        return;
                    278:                }
                    279:
1.12      nicm      280:                if (events & EV_READ && server_client_msg_dispatch(c) != 0)
1.1       nicm      281:                        goto client_lost;
                    282:        }
1.14      nicm      283:
1.73      nicm      284:        server_push_stdout(c);
                    285:        server_push_stderr(c);
                    286:
1.25      nicm      287:        server_update_event(c);
1.1       nicm      288:        return;
                    289:
                    290: client_lost:
                    291:        server_client_lost(c);
1.16      nicm      292: }
                    293:
1.66      nicm      294: /* Check for mouse keys. */
1.131     nicm      295: int
                    296: server_client_check_mouse(struct client *c)
1.66      nicm      297: {
1.131     nicm      298:        struct session                          *s = c->session;
                    299:        struct mouse_event                      *m = &c->tty.mouse;
                    300:        struct window                           *w;
                    301:        struct window_pane                      *wp;
                    302:        enum { NOTYPE, DOWN, UP, DRAG, WHEEL }   type = NOTYPE;
                    303:        enum { NOWHERE, PANE, STATUS, BORDER }   where = NOWHERE;
                    304:        u_int                                    x, y, b;
                    305:        int                                      key;
                    306:
                    307:        log_debug("mouse %02x at %u,%u (last %u,%u) (%d)", m->b, m->x, m->y,
                    308:            m->lx, m->ly, c->tty.mouse_drag_flag);
                    309:
                    310:        /* What type of event is this? */
                    311:        if (MOUSE_DRAG(m->b)) {
                    312:                type = DRAG;
                    313:                if (c->tty.mouse_drag_flag) {
                    314:                        x = m->x, y = m->y, b = m->b;
                    315:                        log_debug("drag update at %u,%u", x, y);
                    316:                } else {
                    317:                        x = m->lx, y = m->ly, b = m->lb;
                    318:                        log_debug("drag start at %u,%u", x, y);
                    319:                }
                    320:        } else if (MOUSE_WHEEL(m->b)) {
                    321:                type = WHEEL;
                    322:                x = m->x, y = m->y, b = m->b;
                    323:                log_debug("wheel at %u,%u", x, y);
                    324:        } else if (MOUSE_BUTTONS(m->b) == 3) {
                    325:                type = UP;
                    326:                x = m->x, y = m->y, b = m->lb;
                    327:                log_debug("up at %u,%u", x, y);
                    328:        } else {
                    329:                type = DOWN;
                    330:                x = m->x, y = m->y, b = m->b;
                    331:                log_debug("down at %u,%u", x, y);
                    332:        }
                    333:        if (type == NOTYPE)
                    334:                return (KEYC_NONE);
                    335:
                    336:        /* Always save the session. */
                    337:        m->s = s->id;
                    338:
                    339:        /* Is this on the status line? */
                    340:        m->statusat = status_at_line(c);
                    341:        if (m->statusat != -1 && y == (u_int)m->statusat) {
                    342:                w = status_get_window_at(c, x);
                    343:                if (w == NULL)
                    344:                        return (KEYC_NONE);
                    345:                m->w = w->id;
                    346:                where = STATUS;
                    347:        } else
                    348:                m->w = -1;
                    349:
                    350:        /* Not on status line. Adjust position and check for border or pane. */
                    351:        if (where == NOWHERE) {
                    352:                if (m->statusat == 0 && y > 0)
                    353:                        y--;
                    354:                else if (m->statusat > 0 && y >= (u_int)m->statusat)
                    355:                        y = m->statusat - 1;
                    356:
                    357:                TAILQ_FOREACH(wp, &s->curw->window->panes, entry) {
                    358:                        if ((wp->xoff + wp->sx == x &&
                    359:                            wp->yoff <= 1 + y &&
                    360:                            wp->yoff + wp->sy >= y) ||
                    361:                            (wp->yoff + wp->sy == y &&
                    362:                            wp->xoff <= 1 + x &&
                    363:                            wp->xoff + wp->sx >= x))
                    364:                                break;
                    365:                }
                    366:                if (wp != NULL)
                    367:                        where = BORDER;
                    368:                else {
                    369:                        wp = window_get_active_at(s->curw->window, x, y);
                    370:                        if (wp != NULL)
                    371:                                where = PANE;
1.160     nicm      372:                        log_debug("mouse at %u,%u is on pane %%%u", x, y,
                    373:                            wp->id);
1.131     nicm      374:                }
                    375:                if (where == NOWHERE)
                    376:                        return (KEYC_NONE);
                    377:                m->wp = wp->id;
                    378:                m->w = wp->window->id;
                    379:        } else
                    380:                m->wp = -1;
                    381:
                    382:        /* Stop dragging if needed. */
                    383:        if (type != DRAG && c->tty.mouse_drag_flag) {
                    384:                if (c->tty.mouse_drag_release != NULL)
                    385:                        c->tty.mouse_drag_release(c, m);
                    386:
                    387:                c->tty.mouse_drag_update = NULL;
                    388:                c->tty.mouse_drag_release = NULL;
                    389:
                    390:                c->tty.mouse_drag_flag = 0;
1.133     nicm      391:                return (KEYC_MOUSE); /* not a key, but still may want to pass */
1.131     nicm      392:        }
                    393:
                    394:        /* Convert to a key binding. */
                    395:        key = KEYC_NONE;
                    396:        switch (type) {
                    397:        case NOTYPE:
                    398:                break;
                    399:        case DRAG:
                    400:                if (c->tty.mouse_drag_update != NULL)
                    401:                        c->tty.mouse_drag_update(c, m);
                    402:                else {
                    403:                        switch (MOUSE_BUTTONS(b)) {
                    404:                        case 0:
                    405:                                if (where == PANE)
                    406:                                        key = KEYC_MOUSEDRAG1_PANE;
                    407:                                if (where == STATUS)
                    408:                                        key = KEYC_MOUSEDRAG1_STATUS;
                    409:                                if (where == BORDER)
                    410:                                        key = KEYC_MOUSEDRAG1_BORDER;
                    411:                                break;
                    412:                        case 1:
                    413:                                if (where == PANE)
                    414:                                        key = KEYC_MOUSEDRAG2_PANE;
                    415:                                if (where == STATUS)
                    416:                                        key = KEYC_MOUSEDRAG2_STATUS;
                    417:                                if (where == BORDER)
                    418:                                        key = KEYC_MOUSEDRAG2_BORDER;
                    419:                                break;
                    420:                        case 2:
                    421:                                if (where == PANE)
                    422:                                        key = KEYC_MOUSEDRAG3_PANE;
                    423:                                if (where == STATUS)
                    424:                                        key = KEYC_MOUSEDRAG3_STATUS;
                    425:                                if (where == BORDER)
                    426:                                        key = KEYC_MOUSEDRAG3_BORDER;
                    427:                                break;
                    428:                        }
                    429:                }
1.66      nicm      430:
1.131     nicm      431:                c->tty.mouse_drag_flag = 1;
                    432:                break;
                    433:        case WHEEL:
                    434:                if (MOUSE_BUTTONS(b) == MOUSE_WHEEL_UP) {
                    435:                        if (where == PANE)
                    436:                                key = KEYC_WHEELUP_PANE;
                    437:                        if (where == STATUS)
                    438:                                key = KEYC_WHEELUP_STATUS;
                    439:                        if (where == BORDER)
                    440:                                key = KEYC_WHEELUP_BORDER;
                    441:                } else {
                    442:                        if (where == PANE)
                    443:                                key = KEYC_WHEELDOWN_PANE;
                    444:                        if (where == STATUS)
                    445:                                key = KEYC_WHEELDOWN_STATUS;
                    446:                        if (where == BORDER)
                    447:                                key = KEYC_WHEELDOWN_BORDER;
                    448:                }
                    449:                break;
                    450:        case UP:
                    451:                switch (MOUSE_BUTTONS(b)) {
                    452:                case 0:
                    453:                        if (where == PANE)
                    454:                                key = KEYC_MOUSEUP1_PANE;
                    455:                        if (where == STATUS)
                    456:                                key = KEYC_MOUSEUP1_STATUS;
                    457:                        if (where == BORDER)
                    458:                                key = KEYC_MOUSEUP1_BORDER;
                    459:                        break;
                    460:                case 1:
                    461:                        if (where == PANE)
                    462:                                key = KEYC_MOUSEUP2_PANE;
                    463:                        if (where == STATUS)
                    464:                                key = KEYC_MOUSEUP2_STATUS;
                    465:                        if (where == BORDER)
                    466:                                key = KEYC_MOUSEUP2_BORDER;
                    467:                        break;
                    468:                case 2:
                    469:                        if (where == PANE)
                    470:                                key = KEYC_MOUSEUP3_PANE;
                    471:                        if (where == STATUS)
                    472:                                key = KEYC_MOUSEUP3_STATUS;
                    473:                        if (where == BORDER)
                    474:                                key = KEYC_MOUSEUP3_BORDER;
                    475:                        break;
                    476:                }
                    477:                break;
                    478:        case DOWN:
                    479:                switch (MOUSE_BUTTONS(b)) {
                    480:                case 0:
                    481:                        if (where == PANE)
                    482:                                key = KEYC_MOUSEDOWN1_PANE;
                    483:                        if (where == STATUS)
                    484:                                key = KEYC_MOUSEDOWN1_STATUS;
                    485:                        if (where == BORDER)
                    486:                                key = KEYC_MOUSEDOWN1_BORDER;
                    487:                        break;
                    488:                case 1:
                    489:                        if (where == PANE)
                    490:                                key = KEYC_MOUSEDOWN2_PANE;
                    491:                        if (where == STATUS)
                    492:                                key = KEYC_MOUSEDOWN2_STATUS;
                    493:                        if (where == BORDER)
                    494:                                key = KEYC_MOUSEDOWN2_BORDER;
                    495:                        break;
                    496:                case 2:
                    497:                        if (where == PANE)
                    498:                                key = KEYC_MOUSEDOWN3_PANE;
                    499:                        if (where == STATUS)
                    500:                                key = KEYC_MOUSEDOWN3_STATUS;
                    501:                        if (where == BORDER)
                    502:                                key = KEYC_MOUSEDOWN3_BORDER;
                    503:                        break;
1.66      nicm      504:                }
1.131     nicm      505:                break;
1.66      nicm      506:        }
1.131     nicm      507:        if (key == KEYC_NONE)
                    508:                return (KEYC_NONE);
1.66      nicm      509:
1.131     nicm      510:        /* Apply modifiers if any. */
                    511:        if (b & MOUSE_MASK_META)
                    512:                key |= KEYC_ESCAPE;
                    513:        if (b & MOUSE_MASK_CTRL)
                    514:                key |= KEYC_CTRL;
                    515:        if (b & MOUSE_MASK_SHIFT)
                    516:                key |= KEYC_SHIFT;
1.66      nicm      517:
1.131     nicm      518:        return (key);
1.66      nicm      519: }
                    520:
1.82      nicm      521: /* Is this fast enough to probably be a paste? */
                    522: int
                    523: server_client_assume_paste(struct session *s)
                    524: {
                    525:        struct timeval  tv;
1.84      nicm      526:        int             t;
1.82      nicm      527:
                    528:        if ((t = options_get_number(&s->options, "assume-paste-time")) == 0)
1.83      nicm      529:                return (0);
1.82      nicm      530:
                    531:        timersub(&s->activity_time, &s->last_activity_time, &tv);
                    532:        if (tv.tv_sec == 0 && tv.tv_usec < t * 1000)
1.83      nicm      533:                return (1);
                    534:        return (0);
1.82      nicm      535: }
                    536:
1.18      nicm      537: /* Handle data key input from client. */
                    538: void
1.74      nicm      539: server_client_handle_key(struct client *c, int key)
1.18      nicm      540: {
1.131     nicm      541:        struct mouse_event      *m = &c->tty.mouse;
1.132     nicm      542:        struct session          *s = c->session;
1.18      nicm      543:        struct window           *w;
                    544:        struct window_pane      *wp;
                    545:        struct timeval           tv;
1.156     nicm      546:        struct key_table        *table;
1.132     nicm      547:        struct key_binding       bd_find, *bd;
                    548:        int                      xtimeout;
1.18      nicm      549:
                    550:        /* Check the client is good to accept input. */
1.132     nicm      551:        if (s == NULL || (c->flags & (CLIENT_DEAD|CLIENT_SUSPENDED)) != 0)
1.18      nicm      552:                return;
1.132     nicm      553:        w = s->curw->window;
1.18      nicm      554:
                    555:        /* Update the activity timer. */
                    556:        if (gettimeofday(&c->activity_time, NULL) != 0)
                    557:                fatal("gettimeofday failed");
1.149     nicm      558:        session_update_activity(s, &c->activity_time);
1.18      nicm      559:
1.132     nicm      560:        /* Number keys jump to pane in identify mode. */
1.25      nicm      561:        if (c->flags & CLIENT_IDENTIFY && key >= '0' && key <= '9') {
1.30      nicm      562:                if (c->flags & CLIENT_READONLY)
                    563:                        return;
1.95      nicm      564:                window_unzoom(w);
1.18      nicm      565:                wp = window_pane_at_index(w, key - '0');
                    566:                if (wp != NULL && window_pane_visible(wp))
                    567:                        window_set_active_pane(w, wp);
                    568:                server_clear_identify(c);
                    569:                return;
                    570:        }
                    571:
                    572:        /* Handle status line. */
1.30      nicm      573:        if (!(c->flags & CLIENT_READONLY)) {
                    574:                status_message_clear(c);
                    575:                server_clear_identify(c);
                    576:        }
1.18      nicm      577:        if (c->prompt_string != NULL) {
1.30      nicm      578:                if (!(c->flags & CLIENT_READONLY))
                    579:                        status_prompt_key(c, key);
1.18      nicm      580:                return;
                    581:        }
                    582:
                    583:        /* Check for mouse keys. */
                    584:        if (key == KEYC_MOUSE) {
1.30      nicm      585:                if (c->flags & CLIENT_READONLY)
                    586:                        return;
1.131     nicm      587:                key = server_client_check_mouse(c);
                    588:                if (key == KEYC_NONE)
                    589:                        return;
                    590:
                    591:                m->valid = 1;
                    592:                m->key = key;
                    593:
1.161   ! nicm      594:                if (!options_get_number(&s->options, "mouse"))
        !           595:                        goto forward;
1.131     nicm      596:        } else
                    597:                m->valid = 0;
1.18      nicm      598:
1.132     nicm      599:        /* Treat everything as a regular key when pasting is detected. */
1.161   ! nicm      600:        if (!KEYC_IS_MOUSE(key) && server_client_assume_paste(s))
        !           601:                goto forward;
1.18      nicm      602:
1.132     nicm      603: retry:
                    604:        /* Try to see if there is a key binding in the current table. */
                    605:        bd_find.key = key;
1.156     nicm      606:        bd = RB_FIND(key_bindings, &c->keytable->key_bindings, &bd_find);
1.132     nicm      607:        if (bd != NULL) {
                    608:                /*
                    609:                 * Key was matched in this table. If currently repeating but a
                    610:                 * non-repeating binding was found, stop repeating and try
                    611:                 * again in the root table.
                    612:                 */
                    613:                if ((c->flags & CLIENT_REPEAT) && !bd->can_repeat) {
                    614:                        server_client_key_table(c, "root");
                    615:                        c->flags &= ~CLIENT_REPEAT;
1.85      nicm      616:                        server_status_client(c);
1.132     nicm      617:                        goto retry;
1.18      nicm      618:                }
1.82      nicm      619:
1.132     nicm      620:                /*
                    621:                 * Take a reference to this table to make sure the key binding
                    622:                 * doesn't disappear.
                    623:                 */
1.156     nicm      624:                table = c->keytable;
1.132     nicm      625:                table->references++;
                    626:
                    627:                /*
                    628:                 * If this is a repeating key, start the timer. Otherwise reset
                    629:                 * the client back to the root table.
                    630:                 */
                    631:                xtimeout = options_get_number(&s->options, "repeat-time");
                    632:                if (xtimeout != 0 && bd->can_repeat) {
                    633:                        c->flags |= CLIENT_REPEAT;
                    634:
                    635:                        tv.tv_sec = xtimeout / 1000;
                    636:                        tv.tv_usec = (xtimeout % 1000) * 1000L;
                    637:                        evtimer_del(&c->repeat_timer);
                    638:                        evtimer_add(&c->repeat_timer, &tv);
                    639:                } else {
1.18      nicm      640:                        c->flags &= ~CLIENT_REPEAT;
1.132     nicm      641:                        server_client_key_table(c, "root");
1.18      nicm      642:                }
1.132     nicm      643:                server_status_client(c);
                    644:
                    645:                /* Dispatch the key binding. */
                    646:                key_bindings_dispatch(bd, c, m);
                    647:                key_bindings_unref_table(table);
1.18      nicm      648:                return;
                    649:        }
                    650:
1.132     nicm      651:        /*
                    652:         * No match in this table. If repeating, switch the client back to the
                    653:         * root table and try again.
                    654:         */
                    655:        if (c->flags & CLIENT_REPEAT) {
                    656:                server_client_key_table(c, "root");
1.18      nicm      657:                c->flags &= ~CLIENT_REPEAT;
1.132     nicm      658:                server_status_client(c);
                    659:                goto retry;
1.18      nicm      660:        }
                    661:
1.132     nicm      662:        /* If no match and we're not in the root table, that's it. */
                    663:        if (strcmp(c->keytable->name, "root") != 0) {
                    664:                server_client_key_table(c, "root");
                    665:                server_status_client(c);
                    666:                return;
1.18      nicm      667:        }
                    668:
1.132     nicm      669:        /*
                    670:         * No match, but in the root table. Prefix switches to the prefix table
                    671:         * and everything else is passed through.
                    672:         */
                    673:        if (key == options_get_number(&s->options, "prefix") ||
                    674:            key == options_get_number(&s->options, "prefix2")) {
                    675:                server_client_key_table(c, "prefix");
                    676:                server_status_client(c);
1.161   ! nicm      677:                return;
        !           678:        }
        !           679:
        !           680: forward:
        !           681:        if (c->flags & CLIENT_READONLY)
        !           682:                return;
        !           683:        if (KEYC_IS_MOUSE(key))
        !           684:                wp = cmd_mouse_pane(m, NULL, NULL);
        !           685:        else
        !           686:                wp = w->active;
        !           687:        if (wp != NULL)
1.132     nicm      688:                window_pane_key(wp, c, s, key, m);
1.18      nicm      689: }
                    690:
1.2       nicm      691: /* Client functions that need to happen every loop. */
                    692: void
                    693: server_client_loop(void)
                    694: {
                    695:        struct client           *c;
                    696:        struct window           *w;
                    697:        struct window_pane      *wp;
                    698:
1.135     nicm      699:        TAILQ_FOREACH(c, &clients, entry) {
1.36      nicm      700:                server_client_check_exit(c);
                    701:                if (c->session != NULL) {
                    702:                        server_client_check_redraw(c);
                    703:                        server_client_reset_state(c);
                    704:                }
1.2       nicm      705:        }
                    706:
                    707:        /*
                    708:         * Any windows will have been redrawn as part of clients, so clear
1.92      nicm      709:         * their flags now. Also check pane focus and resize.
1.2       nicm      710:         */
1.134     nicm      711:        RB_FOREACH(w, windows, &windows) {
1.2       nicm      712:                w->flags &= ~WINDOW_REDRAW;
1.91      nicm      713:                TAILQ_FOREACH(wp, &w->panes, entry) {
1.101     nicm      714:                        if (wp->fd != -1) {
                    715:                                server_client_check_focus(wp);
                    716:                                server_client_check_resize(wp);
                    717:                        }
1.2       nicm      718:                        wp->flags &= ~PANE_REDRAW;
1.91      nicm      719:                }
1.150     nicm      720:                check_window_name(w);
1.2       nicm      721:        }
1.92      nicm      722: }
                    723:
                    724: /* Check if pane should be resized. */
                    725: void
                    726: server_client_check_resize(struct window_pane *wp)
                    727: {
                    728:        struct winsize  ws;
                    729:
1.101     nicm      730:        if (!(wp->flags & PANE_RESIZE))
1.92      nicm      731:                return;
                    732:
                    733:        memset(&ws, 0, sizeof ws);
                    734:        ws.ws_col = wp->sx;
                    735:        ws.ws_row = wp->sy;
                    736:
1.100     nicm      737:        if (ioctl(wp->fd, TIOCSWINSZ, &ws) == -1)
1.92      nicm      738:                fatal("ioctl failed");
                    739:
                    740:        wp->flags &= ~PANE_RESIZE;
1.91      nicm      741: }
                    742:
                    743: /* Check whether pane should be focused. */
                    744: void
                    745: server_client_check_focus(struct window_pane *wp)
                    746: {
1.93      nicm      747:        struct client   *c;
1.102     nicm      748:        int              push;
1.103     nicm      749:
                    750:        /* Are focus events off? */
                    751:        if (!options_get_number(&global_options, "focus-events"))
                    752:                return;
1.102     nicm      753:
                    754:        /* Do we need to push the focus state? */
                    755:        push = wp->flags & PANE_FOCUSPUSH;
                    756:        wp->flags &= ~PANE_FOCUSPUSH;
1.91      nicm      757:
                    758:        /* If we don't care about focus, forget it. */
                    759:        if (!(wp->base.mode & MODE_FOCUSON))
                    760:                return;
                    761:
                    762:        /* If we're not the active pane in our window, we're not focused. */
                    763:        if (wp->window->active != wp)
                    764:                goto not_focused;
                    765:
                    766:        /* If we're in a mode, we're not focused. */
                    767:        if (wp->screen != &wp->base)
                    768:                goto not_focused;
                    769:
                    770:        /*
1.93      nicm      771:         * If our window is the current window in any focused clients with an
                    772:         * attached session, we're focused.
1.91      nicm      773:         */
1.135     nicm      774:        TAILQ_FOREACH(c, &clients, entry) {
                    775:                if (c->session == NULL || !(c->flags & CLIENT_FOCUSED))
1.91      nicm      776:                        continue;
1.93      nicm      777:                if (c->session->flags & SESSION_UNATTACHED)
                    778:                        continue;
                    779:
                    780:                if (c->session->curw->window == wp->window)
1.91      nicm      781:                        goto focused;
                    782:        }
                    783:
                    784: not_focused:
1.102     nicm      785:        if (push || (wp->flags & PANE_FOCUSED))
1.91      nicm      786:                bufferevent_write(wp->event, "\033[O", 3);
                    787:        wp->flags &= ~PANE_FOCUSED;
                    788:        return;
                    789:
                    790: focused:
1.102     nicm      791:        if (push || !(wp->flags & PANE_FOCUSED))
1.91      nicm      792:                bufferevent_write(wp->event, "\033[I", 3);
                    793:        wp->flags |= PANE_FOCUSED;
1.2       nicm      794: }
                    795:
1.18      nicm      796: /*
                    797:  * Update cursor position and mode settings. The scroll region and attributes
                    798:  * are cleared when idle (waiting for an event) as this is the most likely time
                    799:  * a user may interrupt tmux, for example with ~^Z in ssh(1). This is a
                    800:  * compromise between excessive resets and likelihood of an interrupt.
                    801:  *
                    802:  * tty_region/tty_reset/tty_update_mode already take care of not resetting
                    803:  * things that are already in their default state.
                    804:  */
1.1       nicm      805: void
1.18      nicm      806: server_client_reset_state(struct client *c)
1.1       nicm      807: {
1.18      nicm      808:        struct window           *w = c->session->curw->window;
                    809:        struct window_pane      *wp = w->active;
                    810:        struct screen           *s = wp->screen;
                    811:        struct options          *oo = &c->session->options;
1.66      nicm      812:        int                      status, mode, o;
1.60      nicm      813:
                    814:        if (c->flags & CLIENT_SUSPENDED)
                    815:                return;
1.1       nicm      816:
1.86      nicm      817:        if (c->flags & CLIENT_CONTROL)
                    818:                return;
                    819:
1.1       nicm      820:        tty_region(&c->tty, 0, c->tty.sy - 1);
                    821:
                    822:        status = options_get_number(oo, "status");
                    823:        if (!window_pane_visible(wp) || wp->yoff + s->cy >= c->tty.sy - status)
                    824:                tty_cursor(&c->tty, 0, 0);
1.66      nicm      825:        else {
1.126     nicm      826:                o = status && options_get_number(oo, "status-position") == 0;
1.66      nicm      827:                tty_cursor(&c->tty, wp->xoff + s->cx, o + wp->yoff + s->cy);
                    828:        }
1.1       nicm      829:
1.50      nicm      830:        /*
1.131     nicm      831:         * Set mouse mode if requested. To support dragging, always use button
                    832:         * mode.
1.57      nicm      833:         */
                    834:        mode = s->mode;
1.131     nicm      835:        if (options_get_number(oo, "mouse"))
                    836:                mode = (mode & ~ALL_MOUSE_MODES) | MODE_MOUSE_BUTTON;
1.48      nicm      837:
                    838:        /*
                    839:         * Set UTF-8 mouse input if required. If the terminal is UTF-8, the
                    840:         * user has set mouse-utf8 and any mouse mode is in effect, turn on
                    841:         * UTF-8 mouse input. If the receiving terminal hasn't requested it
                    842:         * (that is, it isn't in s->mode), then it'll be converted in
                    843:         * input_mouse.
                    844:         */
                    845:        if ((c->tty.flags & TTY_UTF8) &&
                    846:            (mode & ALL_MOUSE_MODES) && options_get_number(oo, "mouse-utf8"))
                    847:                mode |= MODE_MOUSE_UTF8;
                    848:        else
                    849:                mode &= ~MODE_MOUSE_UTF8;
                    850:
                    851:        /* Set the terminal mode and reset attributes. */
1.59      nicm      852:        tty_update_mode(&c->tty, mode, s);
1.1       nicm      853:        tty_reset(&c->tty);
1.17      nicm      854: }
                    855:
                    856: /* Repeat time callback. */
                    857: void
                    858: server_client_repeat_timer(unused int fd, unused short events, void *data)
                    859: {
                    860:        struct client   *c = data;
                    861:
1.85      nicm      862:        if (c->flags & CLIENT_REPEAT) {
1.132     nicm      863:                server_client_key_table(c, "root");
                    864:                c->flags &= ~CLIENT_REPEAT;
                    865:                server_status_client(c);
1.85      nicm      866:        }
1.1       nicm      867: }
                    868:
1.36      nicm      869: /* Check if client should be exited. */
                    870: void
                    871: server_client_check_exit(struct client *c)
                    872: {
                    873:        if (!(c->flags & CLIENT_EXIT))
                    874:                return;
                    875:
1.73      nicm      876:        if (EVBUFFER_LENGTH(c->stdin_data) != 0)
                    877:                return;
                    878:        if (EVBUFFER_LENGTH(c->stdout_data) != 0)
1.36      nicm      879:                return;
1.73      nicm      880:        if (EVBUFFER_LENGTH(c->stderr_data) != 0)
1.36      nicm      881:                return;
                    882:
1.107     nicm      883:        server_write_client(c, MSG_EXIT, &c->retval, sizeof c->retval);
1.36      nicm      884:        c->flags &= ~CLIENT_EXIT;
1.38      nicm      885: }
                    886:
1.1       nicm      887: /* Check for client redraws. */
                    888: void
                    889: server_client_check_redraw(struct client *c)
                    890: {
                    891:        struct session          *s = c->session;
1.137     nicm      892:        struct tty              *tty = &c->tty;
1.1       nicm      893:        struct window_pane      *wp;
                    894:        int                      flags, redraw;
                    895:
1.86      nicm      896:        if (c->flags & (CLIENT_CONTROL|CLIENT_SUSPENDED))
1.79      nicm      897:                return;
                    898:
1.1       nicm      899:        if (c->flags & (CLIENT_REDRAW|CLIENT_STATUS)) {
                    900:                if (options_get_number(&s->options, "set-titles"))
                    901:                        server_client_set_title(c);
1.12      nicm      902:
1.1       nicm      903:                if (c->message_string != NULL)
                    904:                        redraw = status_message_redraw(c);
                    905:                else if (c->prompt_string != NULL)
                    906:                        redraw = status_prompt_redraw(c);
                    907:                else
                    908:                        redraw = status_redraw(c);
                    909:                if (!redraw)
                    910:                        c->flags &= ~CLIENT_STATUS;
                    911:        }
                    912:
1.137     nicm      913:        flags = tty->flags & (TTY_FREEZE|TTY_NOCURSOR);
                    914:        tty->flags = (tty->flags & ~TTY_FREEZE) | TTY_NOCURSOR;
                    915:
1.1       nicm      916:        if (c->flags & CLIENT_REDRAW) {
1.137     nicm      917:                tty_update_mode(tty, tty->mode, NULL);
1.115     nicm      918:                screen_redraw_screen(c, 1, 1, 1);
1.27      nicm      919:                c->flags &= ~(CLIENT_STATUS|CLIENT_BORDERS);
1.38      nicm      920:        } else if (c->flags & CLIENT_REDRAWWINDOW) {
1.137     nicm      921:                tty_update_mode(tty, tty->mode, NULL);
1.38      nicm      922:                TAILQ_FOREACH(wp, &c->session->curw->window->panes, entry)
                    923:                        screen_redraw_pane(c, wp);
                    924:                c->flags &= ~CLIENT_REDRAWWINDOW;
1.1       nicm      925:        } else {
                    926:                TAILQ_FOREACH(wp, &c->session->curw->window->panes, entry) {
1.137     nicm      927:                        if (wp->flags & PANE_REDRAW) {
                    928:                                tty_update_mode(tty, tty->mode, NULL);
1.1       nicm      929:                                screen_redraw_pane(c, wp);
1.137     nicm      930:                        }
1.1       nicm      931:                }
                    932:        }
                    933:
1.137     nicm      934:        if (c->flags & CLIENT_BORDERS) {
                    935:                tty_update_mode(tty, tty->mode, NULL);
1.115     nicm      936:                screen_redraw_screen(c, 0, 0, 1);
1.137     nicm      937:        }
1.27      nicm      938:
1.137     nicm      939:        if (c->flags & CLIENT_STATUS) {
                    940:                tty_update_mode(tty, tty->mode, NULL);
1.115     nicm      941:                screen_redraw_screen(c, 0, 1, 0);
1.137     nicm      942:        }
1.1       nicm      943:
1.137     nicm      944:        tty->flags = (tty->flags & ~(TTY_FREEZE|TTY_NOCURSOR)) | flags;
                    945:        tty_update_mode(tty, tty->mode, NULL);
1.1       nicm      946:
1.153     nicm      947:        c->flags &= ~(CLIENT_REDRAW|CLIENT_BORDERS|CLIENT_STATUS|
                    948:            CLIENT_STATUSFORCE);
1.1       nicm      949: }
                    950:
                    951: /* Set client title. */
                    952: void
                    953: server_client_set_title(struct client *c)
                    954: {
1.128     nicm      955:        struct session          *s = c->session;
                    956:        const char              *template;
                    957:        char                    *title;
                    958:        struct format_tree      *ft;
1.1       nicm      959:
                    960:        template = options_get_string(&s->options, "set-titles-string");
1.25      nicm      961:
1.128     nicm      962:        ft = format_create();
                    963:        format_defaults(ft, c, NULL, NULL, NULL);
                    964:
                    965:        title = format_expand_time(ft, template, time(NULL));
1.1       nicm      966:        if (c->title == NULL || strcmp(title, c->title) != 0) {
1.77      nicm      967:                free(c->title);
1.1       nicm      968:                c->title = xstrdup(title);
                    969:                tty_set_title(&c->tty, c->title);
                    970:        }
1.77      nicm      971:        free(title);
1.128     nicm      972:
                    973:        format_free(ft);
1.1       nicm      974: }
                    975:
                    976: /* Dispatch message from client. */
                    977: int
                    978: server_client_msg_dispatch(struct client *c)
                    979: {
                    980:        struct imsg              imsg;
1.73      nicm      981:        struct msg_stdin_data    stdindata;
1.108     nicm      982:        const char              *data;
1.1       nicm      983:        ssize_t                  n, datalen;
1.149     nicm      984:        struct session          *s;
1.1       nicm      985:
1.8       deraadt   986:        if ((n = imsg_read(&c->ibuf)) == -1 || n == 0)
                    987:                return (-1);
1.1       nicm      988:
                    989:        for (;;) {
                    990:                if ((n = imsg_get(&c->ibuf, &imsg)) == -1)
                    991:                        return (-1);
                    992:                if (n == 0)
                    993:                        return (0);
1.108     nicm      994:
                    995:                data = imsg.data;
1.1       nicm      996:                datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
                    997:
                    998:                if (imsg.hdr.peerid != PROTOCOL_VERSION) {
                    999:                        server_write_client(c, MSG_VERSION, NULL, 0);
                   1000:                        c->flags |= CLIENT_BAD;
1.112     nicm     1001:                        if (imsg.fd != -1)
                   1002:                                close(imsg.fd);
1.1       nicm     1003:                        imsg_free(&imsg);
                   1004:                        continue;
                   1005:                }
                   1006:
1.157     nicm     1007:                log_debug("got %u from client %p", imsg.hdr.type, c);
1.1       nicm     1008:                switch (imsg.hdr.type) {
1.109     nicm     1009:                case MSG_IDENTIFY_FLAGS:
                   1010:                case MSG_IDENTIFY_TERM:
                   1011:                case MSG_IDENTIFY_TTYNAME:
                   1012:                case MSG_IDENTIFY_CWD:
                   1013:                case MSG_IDENTIFY_STDIN:
                   1014:                case MSG_IDENTIFY_ENVIRON:
1.143     nicm     1015:                case MSG_IDENTIFY_CLIENTPID:
1.109     nicm     1016:                case MSG_IDENTIFY_DONE:
                   1017:                        server_client_msg_identify(c, &imsg);
1.1       nicm     1018:                        break;
1.108     nicm     1019:                case MSG_COMMAND:
                   1020:                        server_client_msg_command(c, &imsg);
                   1021:                        break;
1.73      nicm     1022:                case MSG_STDIN:
                   1023:                        if (datalen != sizeof stdindata)
                   1024:                                fatalx("bad MSG_STDIN size");
1.108     nicm     1025:                        memcpy(&stdindata, data, sizeof stdindata);
1.36      nicm     1026:
1.73      nicm     1027:                        if (c->stdin_callback == NULL)
                   1028:                                break;
                   1029:                        if (stdindata.size <= 0)
                   1030:                                c->stdin_closed = 1;
                   1031:                        else {
                   1032:                                evbuffer_add(c->stdin_data, stdindata.data,
                   1033:                                    stdindata.size);
                   1034:                        }
                   1035:                        c->stdin_callback(c, c->stdin_closed,
                   1036:                            c->stdin_callback_data);
1.33      nicm     1037:                        break;
1.1       nicm     1038:                case MSG_RESIZE:
                   1039:                        if (datalen != 0)
                   1040:                                fatalx("bad MSG_RESIZE size");
                   1041:
1.86      nicm     1042:                        if (c->flags & CLIENT_CONTROL)
                   1043:                                break;
1.32      nicm     1044:                        if (tty_resize(&c->tty)) {
                   1045:                                recalculate_sizes();
                   1046:                                server_redraw_client(c);
                   1047:                        }
1.1       nicm     1048:                        break;
                   1049:                case MSG_EXITING:
                   1050:                        if (datalen != 0)
                   1051:                                fatalx("bad MSG_EXITING size");
                   1052:
                   1053:                        c->session = NULL;
                   1054:                        tty_close(&c->tty);
                   1055:                        server_write_client(c, MSG_EXITED, NULL, 0);
                   1056:                        break;
                   1057:                case MSG_WAKEUP:
                   1058:                case MSG_UNLOCK:
                   1059:                        if (datalen != 0)
                   1060:                                fatalx("bad MSG_WAKEUP size");
                   1061:
                   1062:                        if (!(c->flags & CLIENT_SUSPENDED))
                   1063:                                break;
                   1064:                        c->flags &= ~CLIENT_SUSPENDED;
1.121     nicm     1065:
                   1066:                        if (c->tty.fd == -1) /* exited in the meantime */
                   1067:                                break;
1.149     nicm     1068:                        s = c->session;
1.10      nicm     1069:
1.11      nicm     1070:                        if (gettimeofday(&c->activity_time, NULL) != 0)
1.152     nicm     1071:                                fatal("gettimeofday failed");
1.149     nicm     1072:                        if (s != NULL)
                   1073:                                session_update_activity(s, &c->activity_time);
1.10      nicm     1074:
1.1       nicm     1075:                        tty_start_tty(&c->tty);
                   1076:                        server_redraw_client(c);
                   1077:                        recalculate_sizes();
                   1078:                        break;
                   1079:                case MSG_SHELL:
                   1080:                        if (datalen != 0)
                   1081:                                fatalx("bad MSG_SHELL size");
                   1082:
                   1083:                        server_client_msg_shell(c);
                   1084:                        break;
                   1085:                }
                   1086:
                   1087:                imsg_free(&imsg);
                   1088:        }
                   1089: }
                   1090:
                   1091: /* Handle command message. */
                   1092: void
1.108     nicm     1093: server_client_msg_command(struct client *c, struct imsg *imsg)
1.1       nicm     1094: {
1.108     nicm     1095:        struct msg_command_data   data;
                   1096:        char                     *buf;
                   1097:        size_t                    len;
                   1098:        struct cmd_list          *cmdlist = NULL;
                   1099:        int                       argc;
                   1100:        char                    **argv, *cause;
                   1101:
                   1102:        if (imsg->hdr.len - IMSG_HEADER_SIZE < sizeof data)
                   1103:                fatalx("bad MSG_COMMAND size");
                   1104:        memcpy(&data, imsg->data, sizeof data);
                   1105:
1.124     nicm     1106:        buf = (char *)imsg->data + sizeof data;
1.108     nicm     1107:        len = imsg->hdr.len  - IMSG_HEADER_SIZE - sizeof data;
                   1108:        if (len > 0 && buf[len - 1] != '\0')
                   1109:                fatalx("bad MSG_COMMAND string");
                   1110:
                   1111:        argc = data.argc;
                   1112:        if (cmd_unpack_argv(buf, len, argc, &argv) != 0) {
1.94      nicm     1113:                cmdq_error(c->cmdq, "command too long");
1.1       nicm     1114:                goto error;
                   1115:        }
                   1116:
                   1117:        if (argc == 0) {
                   1118:                argc = 1;
                   1119:                argv = xcalloc(1, sizeof *argv);
                   1120:                *argv = xstrdup("new-session");
                   1121:        }
                   1122:
1.94      nicm     1123:        if ((cmdlist = cmd_list_parse(argc, argv, NULL, 0, &cause)) == NULL) {
                   1124:                cmdq_error(c->cmdq, "%s", cause);
1.1       nicm     1125:                cmd_free_argv(argc, argv);
                   1126:                goto error;
                   1127:        }
                   1128:        cmd_free_argv(argc, argv);
                   1129:
1.113     nicm     1130:        if (c != cfg_client || cfg_finished)
1.131     nicm     1131:                cmdq_run(c->cmdq, cmdlist, NULL);
1.113     nicm     1132:        else
1.131     nicm     1133:                cmdq_append(c->cmdq, cmdlist, NULL);
1.1       nicm     1134:        cmd_list_free(cmdlist);
                   1135:        return;
                   1136:
                   1137: error:
                   1138:        if (cmdlist != NULL)
                   1139:                cmd_list_free(cmdlist);
1.88      nicm     1140:
1.36      nicm     1141:        c->flags |= CLIENT_EXIT;
1.1       nicm     1142: }
                   1143:
                   1144: /* Handle identify message. */
                   1145: void
1.109     nicm     1146: server_client_msg_identify(struct client *c, struct imsg *imsg)
1.1       nicm     1147: {
1.109     nicm     1148:        const char      *data;
                   1149:        size_t           datalen;
                   1150:        int              flags;
                   1151:
                   1152:        if (c->flags & CLIENT_IDENTIFIED)
                   1153:                fatalx("out-of-order identify message");
                   1154:
                   1155:        data = imsg->data;
                   1156:        datalen = imsg->hdr.len - IMSG_HEADER_SIZE;
                   1157:
                   1158:        switch (imsg->hdr.type) {
                   1159:        case MSG_IDENTIFY_FLAGS:
                   1160:                if (datalen != sizeof flags)
                   1161:                        fatalx("bad MSG_IDENTIFY_FLAGS size");
                   1162:                memcpy(&flags, data, sizeof flags);
                   1163:                c->flags |= flags;
1.158     nicm     1164:                log_debug("client %p IDENTIFY_FLAGS %#x", c, flags);
1.109     nicm     1165:                break;
                   1166:        case MSG_IDENTIFY_TERM:
1.110     nicm     1167:                if (datalen == 0 || data[datalen - 1] != '\0')
1.109     nicm     1168:                        fatalx("bad MSG_IDENTIFY_TERM string");
                   1169:                c->term = xstrdup(data);
1.158     nicm     1170:                log_debug("client %p IDENTIFY_TERM %s", c, data);
1.109     nicm     1171:                break;
                   1172:        case MSG_IDENTIFY_TTYNAME:
1.110     nicm     1173:                if (datalen == 0 || data[datalen - 1] != '\0')
1.109     nicm     1174:                        fatalx("bad MSG_IDENTIFY_TTYNAME string");
                   1175:                c->ttyname = xstrdup(data);
1.158     nicm     1176:                log_debug("client %p IDENTIFY_TTYNAME %s", c, data);
1.109     nicm     1177:                break;
                   1178:        case MSG_IDENTIFY_CWD:
1.155     nicm     1179:                if (datalen == 0 || data[datalen - 1] != '\0')
                   1180:                        fatalx("bad MSG_IDENTIFY_CWD string");
                   1181:                if ((c->cwd = open(data, O_RDONLY)) == -1)
                   1182:                        c->cwd = open("/", O_RDONLY);
1.158     nicm     1183:                log_debug("client %p IDENTIFY_CWD %s", c, data);
1.109     nicm     1184:                break;
                   1185:        case MSG_IDENTIFY_STDIN:
                   1186:                if (datalen != 0)
                   1187:                        fatalx("bad MSG_IDENTIFY_STDIN size");
                   1188:                c->fd = imsg->fd;
1.158     nicm     1189:                log_debug("client %p IDENTIFY_STDIN %d", c, imsg->fd);
1.109     nicm     1190:                break;
                   1191:        case MSG_IDENTIFY_ENVIRON:
1.110     nicm     1192:                if (datalen == 0 || data[datalen - 1] != '\0')
1.109     nicm     1193:                        fatalx("bad MSG_IDENTIFY_ENVIRON string");
                   1194:                if (strchr(data, '=') != NULL)
                   1195:                        environ_put(&c->environ, data);
1.158     nicm     1196:                log_debug("client %p IDENTIFY_ENVIRON %s", c, data);
1.143     nicm     1197:                break;
                   1198:        case MSG_IDENTIFY_CLIENTPID:
                   1199:                if (datalen != sizeof c->pid)
                   1200:                        fatalx("bad MSG_IDENTIFY_CLIENTPID size");
                   1201:                memcpy(&c->pid, data, sizeof c->pid);
1.158     nicm     1202:                log_debug("client %p IDENTIFY_CLIENTPID %ld", c, (long)c->pid);
1.109     nicm     1203:                break;
                   1204:        default:
                   1205:                break;
                   1206:        }
                   1207:
                   1208:        if (imsg->hdr.type != MSG_IDENTIFY_DONE)
                   1209:                return;
                   1210:        c->flags |= CLIENT_IDENTIFIED;
1.75      nicm     1211:
1.109     nicm     1212:        if (c->flags & CLIENT_CONTROL) {
1.75      nicm     1213:                c->stdin_callback = control_callback;
1.109     nicm     1214:
1.97      nicm     1215:                evbuffer_free(c->stderr_data);
                   1216:                c->stderr_data = c->stdout_data;
1.109     nicm     1217:
                   1218:                if (c->flags & CLIENT_CONTROLCONTROL)
1.96      nicm     1219:                        evbuffer_add_printf(c->stdout_data, "\033P1000p");
1.79      nicm     1220:                server_write_client(c, MSG_STDIN, NULL, 0);
1.75      nicm     1221:
                   1222:                c->tty.fd = -1;
                   1223:                c->tty.log_fd = -1;
                   1224:
1.109     nicm     1225:                close(c->fd);
                   1226:                c->fd = -1;
                   1227:
1.75      nicm     1228:                return;
                   1229:        }
1.1       nicm     1230:
1.109     nicm     1231:        if (c->fd == -1)
1.104     nicm     1232:                return;
1.145     nicm     1233:        if (tty_init(&c->tty, c, c->fd, c->term) != 0) {
1.109     nicm     1234:                close(c->fd);
                   1235:                c->fd = -1;
1.80      nicm     1236:                return;
                   1237:        }
1.109     nicm     1238:        if (c->flags & CLIENT_UTF8)
1.1       nicm     1239:                c->tty.flags |= TTY_UTF8;
1.109     nicm     1240:        if (c->flags & CLIENT_256COLOURS)
1.1       nicm     1241:                c->tty.term_flags |= TERM_256COLOURS;
                   1242:
                   1243:        tty_resize(&c->tty);
                   1244:
1.109     nicm     1245:        if (!(c->flags & CLIENT_CONTROL))
1.86      nicm     1246:                c->flags |= CLIENT_TERMINAL;
1.1       nicm     1247: }
                   1248:
                   1249: /* Handle shell message. */
                   1250: void
                   1251: server_client_msg_shell(struct client *c)
                   1252: {
1.107     nicm     1253:        const char      *shell;
1.25      nicm     1254:
1.1       nicm     1255:        shell = options_get_string(&global_s_options, "default-shell");
                   1256:        if (*shell == '\0' || areshell(shell))
                   1257:                shell = _PATH_BSHELL;
1.107     nicm     1258:        server_write_client(c, MSG_SHELL, shell, strlen(shell) + 1);
1.25      nicm     1259:
1.1       nicm     1260:        c->flags |= CLIENT_BAD; /* it will die after exec */
                   1261: }