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

Annotation of src/usr.bin/tmux/cmd.c, Revision 1.30

1.30    ! nicm        1: /* $OpenBSD: cmd.c,v 1.29 2009/11/03 20:29:47 nicm Exp $ */
1.1       nicm        2:
                      3: /*
                      4:  * Copyright (c) 2007 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>
                     20: #include <sys/time.h>
                     21:
1.5       nicm       22: #include <fnmatch.h>
                     23: #include <paths.h>
1.1       nicm       24: #include <stdlib.h>
                     25: #include <string.h>
                     26: #include <unistd.h>
                     27:
                     28: #include "tmux.h"
                     29:
                     30: const struct cmd_entry *cmd_table[] = {
                     31:        &cmd_attach_session_entry,
                     32:        &cmd_bind_key_entry,
                     33:        &cmd_break_pane_entry,
1.15      nicm       34:        &cmd_choose_client_entry,
1.1       nicm       35:        &cmd_choose_session_entry,
                     36:        &cmd_choose_window_entry,
                     37:        &cmd_clear_history_entry,
                     38:        &cmd_clock_mode_entry,
                     39:        &cmd_command_prompt_entry,
                     40:        &cmd_confirm_before_entry,
                     41:        &cmd_copy_buffer_entry,
                     42:        &cmd_copy_mode_entry,
                     43:        &cmd_delete_buffer_entry,
                     44:        &cmd_detach_client_entry,
1.7       nicm       45:        &cmd_display_message_entry,
1.16      nicm       46:        &cmd_display_panes_entry,
1.1       nicm       47:        &cmd_down_pane_entry,
                     48:        &cmd_find_window_entry,
                     49:        &cmd_has_session_entry,
1.4       nicm       50:        &cmd_if_shell_entry,
1.1       nicm       51:        &cmd_kill_pane_entry,
                     52:        &cmd_kill_server_entry,
                     53:        &cmd_kill_session_entry,
                     54:        &cmd_kill_window_entry,
                     55:        &cmd_last_window_entry,
                     56:        &cmd_link_window_entry,
                     57:        &cmd_list_buffers_entry,
                     58:        &cmd_list_clients_entry,
                     59:        &cmd_list_commands_entry,
                     60:        &cmd_list_keys_entry,
1.23      nicm       61:        &cmd_list_panes_entry,
1.1       nicm       62:        &cmd_list_sessions_entry,
                     63:        &cmd_list_windows_entry,
                     64:        &cmd_load_buffer_entry,
1.19      nicm       65:        &cmd_lock_client_entry,
1.1       nicm       66:        &cmd_lock_server_entry,
1.19      nicm       67:        &cmd_lock_session_entry,
1.1       nicm       68:        &cmd_move_window_entry,
                     69:        &cmd_new_session_entry,
                     70:        &cmd_new_window_entry,
                     71:        &cmd_next_layout_entry,
                     72:        &cmd_next_window_entry,
                     73:        &cmd_paste_buffer_entry,
1.24      nicm       74:        &cmd_pipe_pane_entry,
1.1       nicm       75:        &cmd_previous_layout_entry,
                     76:        &cmd_previous_window_entry,
                     77:        &cmd_refresh_client_entry,
                     78:        &cmd_rename_session_entry,
                     79:        &cmd_rename_window_entry,
                     80:        &cmd_resize_pane_entry,
                     81:        &cmd_respawn_window_entry,
                     82:        &cmd_rotate_window_entry,
1.17      nicm       83:        &cmd_run_shell_entry,
1.1       nicm       84:        &cmd_save_buffer_entry,
                     85:        &cmd_select_layout_entry,
                     86:        &cmd_select_pane_entry,
                     87:        &cmd_select_prompt_entry,
                     88:        &cmd_select_window_entry,
                     89:        &cmd_send_keys_entry,
                     90:        &cmd_send_prefix_entry,
                     91:        &cmd_server_info_entry,
                     92:        &cmd_set_buffer_entry,
1.13      nicm       93:        &cmd_set_environment_entry,
1.1       nicm       94:        &cmd_set_option_entry,
                     95:        &cmd_set_window_option_entry,
                     96:        &cmd_show_buffer_entry,
1.13      nicm       97:        &cmd_show_environment_entry,
1.1       nicm       98:        &cmd_show_options_entry,
                     99:        &cmd_show_window_options_entry,
                    100:        &cmd_source_file_entry,
                    101:        &cmd_split_window_entry,
                    102:        &cmd_start_server_entry,
                    103:        &cmd_suspend_client_entry,
                    104:        &cmd_swap_pane_entry,
                    105:        &cmd_swap_window_entry,
                    106:        &cmd_switch_client_entry,
                    107:        &cmd_unbind_key_entry,
                    108:        &cmd_unlink_window_entry,
                    109:        &cmd_up_pane_entry,
                    110:        NULL
                    111: };
                    112:
1.11      nicm      113: struct session *cmd_newest_session(struct sessions *);
1.30    ! nicm      114: struct client  *cmd_newest_client(struct clients *);
1.5       nicm      115: struct client  *cmd_lookup_client(const char *);
                    116: struct session *cmd_lookup_session(const char *, int *);
                    117: struct winlink *cmd_lookup_window(struct session *, const char *, int *);
1.8       nicm      118: int             cmd_lookup_index(struct session *, const char *, int *);
1.5       nicm      119:
1.10      nicm      120: int
                    121: cmd_pack_argv(int argc, char **argv, char *buf, size_t len)
                    122: {
                    123:        size_t  arglen;
                    124:        int     i;
                    125:
                    126:        *buf = '\0';
                    127:        for (i = 0; i < argc; i++) {
                    128:                if (strlcpy(buf, argv[i], len) >= len)
                    129:                        return (-1);
                    130:                arglen = strlen(argv[i]) + 1;
                    131:                buf += arglen;
                    132:                len -= arglen;
                    133:        }
                    134:
                    135:        return (0);
                    136: }
                    137:
                    138: int
                    139: cmd_unpack_argv(char *buf, size_t len, int argc, char ***argv)
                    140: {
                    141:        int     i;
                    142:        size_t  arglen;
                    143:
                    144:        if (argc == 0)
                    145:                return (0);
                    146:        *argv = xcalloc(argc, sizeof **argv);
                    147:
                    148:        buf[len - 1] = '\0';
                    149:        for (i = 0; i < argc; i++) {
                    150:                if (len == 0) {
                    151:                        cmd_free_argv(argc, *argv);
                    152:                        return (-1);
                    153:                }
                    154:
                    155:                arglen = strlen(buf) + 1;
                    156:                (*argv)[i] = xstrdup(buf);
                    157:                buf += arglen;
                    158:                len -= arglen;
                    159:        }
                    160:
                    161:        return (0);
                    162: }
                    163:
                    164: void
                    165: cmd_free_argv(int argc, char **argv)
                    166: {
                    167:        int     i;
                    168:
                    169:        if (argc == 0)
                    170:                return;
                    171:        for (i = 0; i < argc; i++) {
                    172:                if (argv[i] != NULL)
                    173:                        xfree(argv[i]);
                    174:        }
                    175:        xfree(argv);
                    176: }
                    177:
1.1       nicm      178: struct cmd *
                    179: cmd_parse(int argc, char **argv, char **cause)
                    180: {
                    181:        const struct cmd_entry **entryp, *entry;
1.27      deraadt   182:        struct cmd              *cmd;
1.1       nicm      183:        char                     s[BUFSIZ];
1.3       nicm      184:        int                      opt, ambiguous = 0;
1.1       nicm      185:
                    186:        *cause = NULL;
1.2       nicm      187:        if (argc == 0) {
                    188:                xasprintf(cause, "no command");
1.1       nicm      189:                return (NULL);
1.2       nicm      190:        }
1.1       nicm      191:
                    192:        entry = NULL;
                    193:        for (entryp = cmd_table; *entryp != NULL; entryp++) {
                    194:                if ((*entryp)->alias != NULL &&
                    195:                    strcmp((*entryp)->alias, argv[0]) == 0) {
1.3       nicm      196:                        ambiguous = 0;
1.1       nicm      197:                        entry = *entryp;
                    198:                        break;
                    199:                }
                    200:
                    201:                if (strncmp((*entryp)->name, argv[0], strlen(argv[0])) != 0)
                    202:                        continue;
                    203:                if (entry != NULL)
1.3       nicm      204:                        ambiguous = 1;
1.1       nicm      205:                entry = *entryp;
                    206:
                    207:                /* Bail now if an exact match. */
                    208:                if (strcmp(entry->name, argv[0]) == 0)
                    209:                        break;
                    210:        }
1.3       nicm      211:        if (ambiguous)
                    212:                goto ambiguous;
1.1       nicm      213:        if (entry == NULL) {
                    214:                xasprintf(cause, "unknown command: %s", argv[0]);
                    215:                return (NULL);
                    216:        }
                    217:
                    218:        optreset = 1;
                    219:        optind = 1;
                    220:        if (entry->parse == NULL) {
                    221:                while ((opt = getopt(argc, argv, "")) != -1) {
                    222:                        switch (opt) {
                    223:                        default:
                    224:                                goto usage;
                    225:                        }
                    226:                }
                    227:                argc -= optind;
                    228:                argv += optind;
                    229:                if (argc != 0)
                    230:                        goto usage;
                    231:        }
                    232:
                    233:        cmd = xmalloc(sizeof *cmd);
                    234:        cmd->entry = entry;
                    235:        cmd->data = NULL;
                    236:        if (entry->parse != NULL) {
                    237:                if (entry->parse(cmd, argc, argv, cause) != 0) {
                    238:                        xfree(cmd);
                    239:                        return (NULL);
                    240:                }
                    241:        }
                    242:        return (cmd);
                    243:
                    244: ambiguous:
                    245:        *s = '\0';
                    246:        for (entryp = cmd_table; *entryp != NULL; entryp++) {
                    247:                if (strncmp((*entryp)->name, argv[0], strlen(argv[0])) != 0)
                    248:                        continue;
                    249:                if (strlcat(s, (*entryp)->name, sizeof s) >= sizeof s)
                    250:                        break;
                    251:                if (strlcat(s, ", ", sizeof s) >= sizeof s)
                    252:                        break;
                    253:        }
                    254:        s[strlen(s) - 2] = '\0';
                    255:        xasprintf(cause, "ambiguous command: %s, could be: %s", argv[0], s);
                    256:        return (NULL);
                    257:
                    258: usage:
                    259:        xasprintf(cause, "usage: %s %s", entry->name, entry->usage);
                    260:        return (NULL);
                    261: }
                    262:
                    263: int
                    264: cmd_exec(struct cmd *cmd, struct cmd_ctx *ctx)
                    265: {
                    266:        return (cmd->entry->exec(cmd, ctx));
                    267: }
                    268:
                    269: void
                    270: cmd_free(struct cmd *cmd)
                    271: {
                    272:        if (cmd->data != NULL && cmd->entry->free != NULL)
                    273:                cmd->entry->free(cmd);
                    274:        xfree(cmd);
                    275: }
                    276:
                    277: size_t
                    278: cmd_print(struct cmd *cmd, char *buf, size_t len)
                    279: {
                    280:        if (cmd->entry->print == NULL) {
                    281:                return (xsnprintf(buf, len, "%s", cmd->entry->name));
                    282:        }
                    283:        return (cmd->entry->print(cmd, buf, len));
                    284: }
                    285:
1.5       nicm      286: /*
                    287:  * Figure out the current session. Use: 1) the current session, if the command
1.11      nicm      288:  * context has one; 2) the session containing the pty of the calling client, if
                    289:  * any 3) the session specified in the TMUX variable from the environment (as
                    290:  * passed from the client); 3) the newest session.
1.5       nicm      291:  */
1.1       nicm      292: struct session *
                    293: cmd_current_session(struct cmd_ctx *ctx)
                    294: {
                    295:        struct msg_command_data *data = ctx->msgdata;
1.11      nicm      296:        struct client           *c = ctx->cmdclient;
1.5       nicm      297:        struct session          *s;
1.11      nicm      298:        struct sessions          ss;
                    299:        struct winlink          *wl;
                    300:        struct window_pane      *wp;
                    301:        u_int                    i;
                    302:        int                      found;
1.1       nicm      303:
1.14      nicm      304:        if (ctx->curclient != NULL && ctx->curclient->session != NULL)
                    305:                return (ctx->curclient->session);
1.1       nicm      306:
1.11      nicm      307:        /*
                    308:         * If the name of the calling client's pty is know, build a list of the
                    309:         * sessions that contain it and if any choose either the first or the
                    310:         * newest.
                    311:         */
                    312:        if (c != NULL && c->tty.path != NULL) {
                    313:                ARRAY_INIT(&ss);
                    314:                for (i = 0; i < ARRAY_LENGTH(&sessions); i++) {
                    315:                        if ((s = ARRAY_ITEM(&sessions, i)) == NULL)
                    316:                                continue;
                    317:                        found = 0;
                    318:                        RB_FOREACH(wl, winlinks, &s->windows) {
                    319:                                TAILQ_FOREACH(wp, &wl->window->panes, entry) {
                    320:                                        if (strcmp(wp->tty, c->tty.path) == 0) {
                    321:                                                found = 1;
                    322:                                                break;
                    323:                                        }
                    324:                                }
                    325:                                if (found)
                    326:                                        break;
                    327:                        }
                    328:                        if (found)
                    329:                                ARRAY_ADD(&ss, s);
                    330:                }
                    331:
                    332:                s = cmd_newest_session(&ss);
                    333:                ARRAY_FREE(&ss);
                    334:                if (s != NULL)
                    335:                        return (s);
                    336:        }
                    337:
                    338:        /* Use the session from the TMUX environment variable. */
1.1       nicm      339:        if (data != NULL && data->pid != -1) {
1.5       nicm      340:                if (data->pid != getpid())
1.1       nicm      341:                        return (NULL);
1.5       nicm      342:                if (data->idx > ARRAY_LENGTH(&sessions))
1.1       nicm      343:                        return (NULL);
1.5       nicm      344:                if ((s = ARRAY_ITEM(&sessions, data->idx)) == NULL)
1.1       nicm      345:                        return (NULL);
                    346:                return (s);
                    347:        }
                    348:
1.11      nicm      349:        return (cmd_newest_session(&sessions));
1.5       nicm      350: }
                    351:
                    352: /* Find the newest session. */
                    353: struct session *
1.11      nicm      354: cmd_newest_session(struct sessions *ss)
1.5       nicm      355: {
                    356:        struct session  *s, *snewest;
                    357:        struct timeval  *tv = NULL;
                    358:        u_int            i;
                    359:
                    360:        snewest = NULL;
1.11      nicm      361:        for (i = 0; i < ARRAY_LENGTH(ss); i++) {
                    362:                if ((s = ARRAY_ITEM(ss, i)) == NULL)
1.5       nicm      363:                        continue;
                    364:
1.29      nicm      365:                if (tv == NULL || timercmp(&s->creation_time, tv, >)) {
1.5       nicm      366:                        snewest = s;
1.29      nicm      367:                        tv = &s->creation_time;
1.1       nicm      368:                }
                    369:        }
1.5       nicm      370:
                    371:        return (snewest);
1.1       nicm      372: }
                    373:
1.30    ! nicm      374: /*
        !           375:  * Find the current client. First try the current client if set, then pick the
        !           376:  * newest of the clients attached to the current session if any, then the
        !           377:  * newest client.
        !           378:  */
        !           379: struct client *
        !           380: cmd_current_client(struct cmd_ctx *ctx)
        !           381: {
        !           382:        struct session          *s;
        !           383:        struct client           *c;
        !           384:        struct clients           cc;
        !           385:        u_int                    i;
        !           386:
        !           387:        if (ctx->curclient != NULL)
        !           388:                return (ctx->curclient);
        !           389:
        !           390:        /*
        !           391:         * No current client set. Find the current session and return the
        !           392:         * newest of its clients.
        !           393:         */
        !           394:        s = cmd_current_session(ctx);
        !           395:        if (s != NULL && !(s->flags & SESSION_UNATTACHED)) {
        !           396:                ARRAY_INIT(&cc);
        !           397:                for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
        !           398:                        if ((c = ARRAY_ITEM(&clients, i)) == NULL)
        !           399:                                continue;
        !           400:                        if (s == c->session)
        !           401:                                ARRAY_ADD(&cc, c);
        !           402:                }
        !           403:
        !           404:                c = cmd_newest_client(&cc);
        !           405:                ARRAY_FREE(&cc);
        !           406:                if (c != NULL)
        !           407:                        return (c);
        !           408:        }
        !           409:
        !           410:        return (cmd_newest_client(&clients));
        !           411: }
        !           412:
1.20      nicm      413: /* Find the newest client. */
                    414: struct client *
1.30    ! nicm      415: cmd_newest_client(struct clients *cc)
1.20      nicm      416: {
                    417:        struct client   *c, *cnewest;
                    418:        struct timeval  *tv = NULL;
                    419:        u_int            i;
                    420:
                    421:        cnewest = NULL;
1.30    ! nicm      422:        for (i = 0; i < ARRAY_LENGTH(cc); i++) {
        !           423:                if ((c = ARRAY_ITEM(cc, i)) == NULL)
1.20      nicm      424:                        continue;
                    425:                if (c->session == NULL)
                    426:                        continue;
                    427:
1.29      nicm      428:                if (tv == NULL || timercmp(&c->creation_time, tv, >)) {
1.20      nicm      429:                        cnewest = c;
1.29      nicm      430:                        tv = &c->creation_time;
1.20      nicm      431:                }
                    432:        }
                    433:
                    434:        return (cnewest);
                    435: }
                    436:
1.5       nicm      437: /* Find the target client or report an error and return NULL. */
1.1       nicm      438: struct client *
                    439: cmd_find_client(struct cmd_ctx *ctx, const char *arg)
                    440: {
1.30    ! nicm      441:        struct client   *c;
1.5       nicm      442:        char            *tmparg;
                    443:        size_t           arglen;
1.1       nicm      444:
1.5       nicm      445:        /* A NULL argument means the current client. */
1.30    ! nicm      446:        if (arg == NULL)
        !           447:                return (cmd_current_client(ctx));
1.5       nicm      448:        tmparg = xstrdup(arg);
                    449:
                    450:        /* Trim a single trailing colon if any. */
                    451:        arglen = strlen(tmparg);
                    452:        if (arglen != 0 && tmparg[arglen - 1] == ':')
                    453:                tmparg[arglen - 1] = '\0';
                    454:
                    455:        /* Find the client, if any. */
                    456:        c = cmd_lookup_client(tmparg);
                    457:
                    458:        /* If no client found, report an error. */
                    459:        if (c == NULL)
                    460:                ctx->error(ctx, "client not found: %s", tmparg);
                    461:
                    462:        xfree(tmparg);
                    463:        return (c);
                    464: }
                    465:
                    466: /*
                    467:  * Lookup a client by device path. Either of a full match and a match without a
                    468:  * leading _PATH_DEV ("/dev/") is accepted.
                    469:  */
                    470: struct client *
                    471: cmd_lookup_client(const char *name)
                    472: {
                    473:        struct client   *c;
                    474:        const char      *path;
                    475:        u_int            i;
                    476:
                    477:        for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
1.25      nicm      478:                c = ARRAY_ITEM(&clients, i);
                    479:                if (c == NULL || c->session == NULL)
1.5       nicm      480:                        continue;
                    481:                path = c->tty.path;
                    482:
                    483:                /* Check for exact matches. */
                    484:                if (strcmp(name, path) == 0)
                    485:                        return (c);
                    486:
                    487:                /* Check without leading /dev if present. */
                    488:                if (strncmp(path, _PATH_DEV, (sizeof _PATH_DEV) - 1) != 0)
                    489:                        continue;
                    490:                if (strcmp(name, path + (sizeof _PATH_DEV) - 1) == 0)
                    491:                        return (c);
                    492:        }
                    493:
                    494:        return (NULL);
                    495: }
                    496:
                    497: /* Lookup a session by name. If no session is found, NULL is returned. */
                    498: struct session *
                    499: cmd_lookup_session(const char *name, int *ambiguous)
                    500: {
                    501:        struct session  *s, *sfound;
                    502:        u_int            i;
                    503:
                    504:        *ambiguous = 0;
                    505:
                    506:        /*
1.28      nicm      507:         * Look for matches. First look for exact matches - session names must
                    508:         * be unique so an exact match can't be ambigious and can just be
                    509:         * returned.
1.5       nicm      510:         */
                    511:        for (i = 0; i < ARRAY_LENGTH(&sessions); i++) {
                    512:                if ((s = ARRAY_ITEM(&sessions, i)) == NULL)
                    513:                        continue;
                    514:                if (strcmp(name, s->name) == 0)
                    515:                        return (s);
1.28      nicm      516:        }
                    517:
                    518:        /*
                    519:         * Otherwise look for partial matches, returning early if it is found to
                    520:         * be ambiguous.
                    521:         */
                    522:        sfound = NULL;
                    523:        for (i = 0; i < ARRAY_LENGTH(&sessions); i++) {
                    524:                if ((s = ARRAY_ITEM(&sessions, i)) == NULL)
                    525:                        continue;
1.5       nicm      526:                if (strncmp(name, s->name, strlen(name)) == 0 ||
                    527:                    fnmatch(name, s->name, 0) == 0) {
                    528:                        if (sfound != NULL) {
                    529:                                *ambiguous = 1;
                    530:                                return (NULL);
                    531:                        }
                    532:                        sfound = s;
                    533:                }
                    534:        }
                    535:        return (sfound);
                    536: }
                    537:
                    538: /*
1.8       nicm      539:  * Lookup a window or return -1 if not found or ambigious. First try as an
                    540:  * index and if invalid, use fnmatch or leading prefix. Return NULL but fill in
                    541:  * idx if the window index is a valid number but there is now window with that
                    542:  * index.
1.5       nicm      543:  */
                    544: struct winlink *
                    545: cmd_lookup_window(struct session *s, const char *name, int *ambiguous)
                    546: {
                    547:        struct winlink  *wl, *wlfound;
                    548:        const char      *errstr;
                    549:        u_int            idx;
                    550:
                    551:        *ambiguous = 0;
                    552:
                    553:        /* First see if this is a valid window index in this session. */
                    554:        idx = strtonum(name, 0, INT_MAX, &errstr);
                    555:        if (errstr == NULL) {
                    556:                if ((wl = winlink_find_by_index(&s->windows, idx)) != NULL)
                    557:                        return (wl);
                    558:        }
                    559:
                    560:        /* Look for exact matches, error if more than one. */
                    561:        wlfound = NULL;
                    562:        RB_FOREACH(wl, winlinks, &s->windows) {
1.8       nicm      563:                if (strcmp(name, wl->window->name) == 0) {
1.5       nicm      564:                        if (wlfound != NULL) {
                    565:                                *ambiguous = 1;
                    566:                                return (NULL);
                    567:                        }
                    568:                        wlfound = wl;
1.1       nicm      569:                }
                    570:        }
1.5       nicm      571:        if (wlfound != NULL)
                    572:                return (wlfound);
                    573:
                    574:        /* Now look for pattern matches, again error if multiple. */
                    575:        wlfound = NULL;
                    576:        RB_FOREACH(wl, winlinks, &s->windows) {
1.8       nicm      577:                if (strncmp(name, wl->window->name, strlen(name)) == 0 ||
                    578:                    fnmatch(name, wl->window->name, 0) == 0) {
1.5       nicm      579:                        if (wlfound != NULL) {
                    580:                                *ambiguous = 1;
                    581:                                return (NULL);
                    582:                        }
                    583:                        wlfound = wl;
                    584:                }
                    585:        }
                    586:        if (wlfound != NULL)
                    587:                return (wlfound);
                    588:
                    589:        return (NULL);
1.1       nicm      590: }
                    591:
1.8       nicm      592: /*
                    593:  * Find a window index - if the window doesn't exist, check if it is a
                    594:  * potential index and return it anyway.
                    595:  */
                    596: int
                    597: cmd_lookup_index(struct session *s, const char *name, int *ambiguous)
                    598: {
                    599:        struct winlink  *wl;
                    600:        const char      *errstr;
                    601:        u_int            idx;
                    602:
                    603:        if ((wl = cmd_lookup_window(s, name, ambiguous)) != NULL)
                    604:                return (wl->idx);
                    605:        if (*ambiguous)
                    606:                return (-1);
                    607:
                    608:        idx = strtonum(name, 0, INT_MAX, &errstr);
                    609:        if (errstr == NULL)
                    610:                return (idx);
                    611:
                    612:        return (-1);
                    613: }
                    614:
1.5       nicm      615: /* Find the target session or report an error and return NULL. */
1.1       nicm      616: struct session *
                    617: cmd_find_session(struct cmd_ctx *ctx, const char *arg)
                    618: {
                    619:        struct session  *s;
1.5       nicm      620:        struct client   *c;
                    621:        char            *tmparg;
                    622:        size_t           arglen;
                    623:        int              ambiguous;
1.1       nicm      624:
1.5       nicm      625:        /* A NULL argument means the current session. */
1.1       nicm      626:        if (arg == NULL)
1.5       nicm      627:                return (cmd_current_session(ctx));
                    628:        tmparg = xstrdup(arg);
                    629:
                    630:        /* Trim a single trailing colon if any. */
                    631:        arglen = strlen(tmparg);
                    632:        if (arglen != 0 && tmparg[arglen - 1] == ':')
                    633:                tmparg[arglen - 1] = '\0';
                    634:
                    635:        /* Find the session, if any. */
                    636:        s = cmd_lookup_session(tmparg, &ambiguous);
                    637:
                    638:        /* If it doesn't, try to match it as a client. */
                    639:        if (s == NULL && (c = cmd_lookup_client(tmparg)) != NULL)
                    640:                s = c->session;
                    641:
                    642:        /* If no session found, report an error. */
                    643:        if (s == NULL) {
                    644:                if (ambiguous)
                    645:                        ctx->error(ctx, "more than one session: %s", tmparg);
                    646:                else
                    647:                        ctx->error(ctx, "session not found: %s", tmparg);
1.1       nicm      648:        }
1.5       nicm      649:
                    650:        xfree(tmparg);
1.1       nicm      651:        return (s);
                    652: }
                    653:
1.5       nicm      654: /* Find the target session and window or report an error and return NULL. */
1.1       nicm      655: struct winlink *
                    656: cmd_find_window(struct cmd_ctx *ctx, const char *arg, struct session **sp)
                    657: {
                    658:        struct session  *s;
                    659:        struct winlink  *wl;
1.5       nicm      660:        const char      *winptr;
                    661:        char            *sessptr = NULL;
                    662:        int              ambiguous = 0;
                    663:
                    664:        /*
                    665:         * Find the current session. There must always be a current session, if
                    666:         * it can't be found, report an error.
                    667:         */
                    668:        if ((s = cmd_current_session(ctx)) == NULL) {
                    669:                ctx->error(ctx, "can't establish current session");
                    670:                return (NULL);
                    671:        }
                    672:
                    673:        /* A NULL argument means the current session and window. */
                    674:        if (arg == NULL) {
                    675:                if (sp != NULL)
                    676:                        *sp = s;
                    677:                return (s->curw);
                    678:        }
                    679:
                    680:        /* Time to look at the argument. If it is empty, that is an error. */
                    681:        if (*arg == '\0')
                    682:                goto not_found;
                    683:
1.8       nicm      684:        /* Find the separating colon and split into window and session. */
1.5       nicm      685:        winptr = strchr(arg, ':');
                    686:        if (winptr == NULL)
1.8       nicm      687:                goto no_colon;
                    688:        winptr++;       /* skip : */
                    689:        sessptr = xstrdup(arg);
                    690:        *strchr(sessptr, ':') = '\0';
1.5       nicm      691:
                    692:        /* Try to lookup the session if present. */
1.8       nicm      693:        if (*sessptr != '\0') {
                    694:                if ((s = cmd_lookup_session(sessptr, &ambiguous)) == NULL)
1.5       nicm      695:                        goto no_session;
                    696:        }
                    697:        if (sp != NULL)
                    698:                *sp = s;
                    699:
                    700:        /*
                    701:         * Then work out the window. An empty string is the current window,
                    702:         * otherwise try to look it up in the session.
                    703:         */
1.8       nicm      704:        if (*winptr == '\0')
1.5       nicm      705:                wl = s->curw;
                    706:        else if ((wl = cmd_lookup_window(s, winptr, &ambiguous)) == NULL)
                    707:                goto not_found;
                    708:
                    709:        if (sessptr != NULL)
                    710:                xfree(sessptr);
                    711:        return (wl);
                    712:
1.8       nicm      713: no_colon:
                    714:        /* No colon in the string, first try as a window then as a session. */
                    715:        if ((wl = cmd_lookup_window(s, arg, &ambiguous)) == NULL) {
                    716:                if (ambiguous)
                    717:                        goto not_found;
                    718:                if ((s = cmd_lookup_session(arg, &ambiguous)) == NULL)
                    719:                        goto no_session;
                    720:                wl = s->curw;
                    721:        }
                    722:
                    723:        if (sp != NULL)
                    724:                *sp = s;
                    725:
                    726:        return (wl);
                    727:
1.5       nicm      728: no_session:
                    729:        if (ambiguous)
1.8       nicm      730:                ctx->error(ctx, "multiple sessions: %s", arg);
1.5       nicm      731:        else
1.8       nicm      732:                ctx->error(ctx, "session not found: %s", arg);
1.5       nicm      733:        if (sessptr != NULL)
                    734:                xfree(sessptr);
                    735:        return (NULL);
                    736:
                    737: not_found:
                    738:        if (ambiguous)
                    739:                ctx->error(ctx, "multiple windows: %s", arg);
                    740:        else
                    741:                ctx->error(ctx, "window not found: %s", arg);
                    742:        if (sessptr != NULL)
                    743:                xfree(sessptr);
                    744:        return (NULL);
                    745: }
1.1       nicm      746:
1.5       nicm      747: /*
                    748:  * Find the target session and window index, whether or not it exists in the
                    749:  * session. Return -2 on error or -1 if no window index is specified. This is
1.8       nicm      750:  * used when parsing an argument for a window target that may not exist (for
                    751:  * example if it is going to be created).
1.5       nicm      752:  */
                    753: int
                    754: cmd_find_index(struct cmd_ctx *ctx, const char *arg, struct session **sp)
                    755: {
                    756:        struct session  *s;
1.8       nicm      757:        const char      *winptr;
1.5       nicm      758:        char            *sessptr = NULL;
                    759:        int              idx, ambiguous = 0;
                    760:
                    761:        /*
                    762:         * Find the current session. There must always be a current session, if
                    763:         * it can't be found, report an error.
                    764:         */
                    765:        if ((s = cmd_current_session(ctx)) == NULL) {
                    766:                ctx->error(ctx, "can't establish current session");
1.9       nicm      767:                return (-2);
1.1       nicm      768:        }
1.5       nicm      769:
                    770:        /* A NULL argument means the current session and "no window" (-1). */
                    771:        if (arg == NULL) {
                    772:                if (sp != NULL)
                    773:                        *sp = s;
                    774:                return (-1);
                    775:        }
                    776:
                    777:        /* Time to look at the argument. If it is empty, that is an error. */
                    778:        if (*arg == '\0')
                    779:                goto not_found;
                    780:
                    781:        /* Find the separating colon. If none, assume the current session. */
                    782:        winptr = strchr(arg, ':');
                    783:        if (winptr == NULL)
1.8       nicm      784:                goto no_colon;
                    785:        winptr++;       /* skip : */
                    786:        sessptr = xstrdup(arg);
                    787:        *strchr(sessptr, ':') = '\0';
1.5       nicm      788:
                    789:        /* Try to lookup the session if present. */
                    790:        if (sessptr != NULL && *sessptr != '\0') {
1.8       nicm      791:                if ((s = cmd_lookup_session(sessptr, &ambiguous)) == NULL)
1.5       nicm      792:                        goto no_session;
                    793:        }
1.1       nicm      794:        if (sp != NULL)
                    795:                *sp = s;
                    796:
1.5       nicm      797:        /*
1.8       nicm      798:         * Then work out the window. An empty string is a new window otherwise
                    799:         * try to look it up in the session.
1.5       nicm      800:         */
1.8       nicm      801:        if (*winptr == '\0')
                    802:                 idx = -1;
                    803:        else if ((idx = cmd_lookup_index(s, winptr, &ambiguous)) == -1) {
1.5       nicm      804:                if (ambiguous)
                    805:                        goto not_found;
1.8       nicm      806:                ctx->error(ctx, "invalid index: %s", arg);
                    807:                idx = -2;
                    808:        }
1.5       nicm      809:
                    810:        if (sessptr != NULL)
                    811:                xfree(sessptr);
                    812:        return (idx);
                    813:
1.8       nicm      814: no_colon:
                    815:        /* No colon in the string, first try as a window then as a session. */
                    816:        if ((idx = cmd_lookup_index(s, arg, &ambiguous)) == -1) {
                    817:                if (ambiguous)
                    818:                        goto not_found;
                    819:                if ((s = cmd_lookup_session(arg, &ambiguous)) == NULL)
                    820:                        goto no_session;
                    821:                idx = -1;
                    822:        }
                    823:
                    824:        if (sp != NULL)
                    825:                *sp = s;
                    826:
                    827:        return (idx);
                    828:
1.5       nicm      829: no_session:
                    830:        if (ambiguous)
1.8       nicm      831:                ctx->error(ctx, "multiple sessions: %s", arg);
1.5       nicm      832:        else
1.8       nicm      833:                ctx->error(ctx, "session not found: %s", arg);
1.5       nicm      834:        if (sessptr != NULL)
                    835:                xfree(sessptr);
                    836:        return (-2);
                    837:
                    838: not_found:
                    839:        if (ambiguous)
                    840:                ctx->error(ctx, "multiple windows: %s", arg);
1.1       nicm      841:        else
1.5       nicm      842:                ctx->error(ctx, "window not found: %s", arg);
                    843:        if (sessptr != NULL)
                    844:                xfree(sessptr);
                    845:        return (-2);
1.12      nicm      846: }
                    847:
                    848: /*
                    849:  * Find the target session, window and pane number or report an error and
                    850:  * return NULL. The pane number is separated from the session:window by a .,
                    851:  * such as mysession:mywindow.0.
                    852:  */
                    853: struct winlink *
                    854: cmd_find_pane(struct cmd_ctx *ctx,
                    855:     const char *arg, struct session **sp, struct window_pane **wpp)
                    856: {
                    857:        struct session  *s;
                    858:        struct winlink  *wl;
                    859:        const char      *period;
                    860:        char            *winptr, *paneptr;
                    861:        const char      *errstr;
                    862:        u_int            idx;
                    863:
                    864:        /* Get the current session. */
                    865:        if ((s = cmd_current_session(ctx)) == NULL) {
1.27      deraadt   866:                ctx->error(ctx, "can't establish current session");
1.12      nicm      867:                return (NULL);
                    868:        }
                    869:        if (sp != NULL)
                    870:                *sp = s;
                    871:
                    872:        /* A NULL argument means the current session, window and pane. */
                    873:        if (arg == NULL) {
                    874:                *wpp = s->curw->window->active;
                    875:                return (s->curw);
                    876:        }
                    877:
                    878:        /* Look for a separating period. */
                    879:        if ((period = strrchr(arg, '.')) == NULL)
                    880:                goto no_period;
                    881:
                    882:        /* Pull out the window part and parse it. */
                    883:        winptr = xstrdup(arg);
                    884:        winptr[period - arg] = '\0';
                    885:        if (*winptr == '\0')
                    886:                wl = s->curw;
                    887:        else if ((wl = cmd_find_window(ctx, winptr, sp)) == NULL)
                    888:                goto error;
                    889:
                    890:        /* Find the pane section and look it up. */
                    891:        paneptr = winptr + (period - arg) + 1;
                    892:        if (*paneptr == '\0')
                    893:                *wpp = wl->window->active;
                    894:        else {
                    895:                idx = strtonum(paneptr, 0, INT_MAX, &errstr);
                    896:                if (errstr != NULL) {
                    897:                        ctx->error(ctx, "pane %s: %s", errstr, paneptr);
                    898:                        goto error;
                    899:                }
                    900:                *wpp = window_pane_at_index(wl->window, idx);
                    901:                if (*wpp == NULL) {
                    902:                        ctx->error(ctx, "no such pane: %u", idx);
                    903:                        goto error;
                    904:                }
                    905:        }
                    906:
                    907:        xfree(winptr);
                    908:        return (wl);
                    909:
                    910: no_period:
                    911:        /* Try as a pane number alone. */
                    912:        idx = strtonum(arg, 0, INT_MAX, &errstr);
                    913:        if (errstr != NULL)
                    914:                goto lookup_window;
                    915:
                    916:        /* Try index in the current session and window. */
                    917:        if ((*wpp = window_pane_at_index(s->curw->window, idx)) == NULL)
                    918:                goto lookup_window;
                    919:
                    920:        return (s->curw);
                    921:
                    922: lookup_window:
                    923:        /* Try as a window and use the active pane. */
                    924:        if ((wl = cmd_find_window(ctx, arg, sp)) != NULL)
                    925:                *wpp = wl->window->active;
                    926:        return (wl);
                    927:
                    928: error:
                    929:        xfree(winptr);
                    930:        return (NULL);
1.15      nicm      931: }
                    932:
                    933: /* Replace the first %% or %idx in template by s. */
                    934: char *
                    935: cmd_template_replace(char *template, const char *s, int idx)
                    936: {
                    937:        char     ch;
                    938:        char    *buf, *ptr;
                    939:        int      replaced;
                    940:        size_t   len;
                    941:
                    942:        if (strstr(template, "%") == NULL)
                    943:                return (xstrdup(template));
                    944:
                    945:        buf = xmalloc(1);
                    946:        *buf = '\0';
                    947:        len = 0;
                    948:        replaced = 0;
                    949:
                    950:        ptr = template;
                    951:        while (*ptr != '\0') {
                    952:                switch (ch = *ptr++) {
                    953:                case '%':
                    954:                        if (*ptr < '1' || *ptr > '9' || *ptr - '0' != idx) {
                    955:                                if (*ptr != '%' || replaced)
                    956:                                        break;
                    957:                                replaced = 1;
                    958:                        }
                    959:                        ptr++;
                    960:
                    961:                        len += strlen(s);
                    962:                        buf = xrealloc(buf, 1, len + 1);
                    963:                        strlcat(buf, s, len + 1);
                    964:                        continue;
                    965:                }
                    966:                buf = xrealloc(buf, 1, len + 2);
                    967:                buf[len++] = ch;
                    968:                buf[len] = '\0';
                    969:        }
                    970:
                    971:        return (buf);
1.1       nicm      972: }