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: }