Annotation of src/usr.bin/tmux/cmd.c, Revision 1.60
1.60 ! nicm 1: /* $OpenBSD: cmd.c,v 1.59 2012/01/20 19:54:07 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.34 nicm 34: &cmd_capture_pane_entry,
1.42 nicm 35: &cmd_choose_buffer_entry,
1.15 nicm 36: &cmd_choose_client_entry,
1.1 nicm 37: &cmd_choose_session_entry,
38: &cmd_choose_window_entry,
39: &cmd_clear_history_entry,
40: &cmd_clock_mode_entry,
41: &cmd_command_prompt_entry,
42: &cmd_confirm_before_entry,
43: &cmd_copy_mode_entry,
44: &cmd_delete_buffer_entry,
45: &cmd_detach_client_entry,
1.7 nicm 46: &cmd_display_message_entry,
1.16 nicm 47: &cmd_display_panes_entry,
1.1 nicm 48: &cmd_find_window_entry,
49: &cmd_has_session_entry,
1.4 nicm 50: &cmd_if_shell_entry,
1.37 nicm 51: &cmd_join_pane_entry,
1.1 nicm 52: &cmd_kill_pane_entry,
53: &cmd_kill_server_entry,
54: &cmd_kill_session_entry,
55: &cmd_kill_window_entry,
1.45 nicm 56: &cmd_last_pane_entry,
1.1 nicm 57: &cmd_last_window_entry,
58: &cmd_link_window_entry,
59: &cmd_list_buffers_entry,
60: &cmd_list_clients_entry,
61: &cmd_list_commands_entry,
62: &cmd_list_keys_entry,
1.23 nicm 63: &cmd_list_panes_entry,
1.1 nicm 64: &cmd_list_sessions_entry,
65: &cmd_list_windows_entry,
66: &cmd_load_buffer_entry,
1.19 nicm 67: &cmd_lock_client_entry,
1.1 nicm 68: &cmd_lock_server_entry,
1.19 nicm 69: &cmd_lock_session_entry,
1.1 nicm 70: &cmd_move_window_entry,
71: &cmd_new_session_entry,
72: &cmd_new_window_entry,
73: &cmd_next_layout_entry,
74: &cmd_next_window_entry,
75: &cmd_paste_buffer_entry,
1.24 nicm 76: &cmd_pipe_pane_entry,
1.1 nicm 77: &cmd_previous_layout_entry,
78: &cmd_previous_window_entry,
79: &cmd_refresh_client_entry,
80: &cmd_rename_session_entry,
81: &cmd_rename_window_entry,
82: &cmd_resize_pane_entry,
1.56 nicm 83: &cmd_respawn_pane_entry,
1.1 nicm 84: &cmd_respawn_window_entry,
85: &cmd_rotate_window_entry,
1.17 nicm 86: &cmd_run_shell_entry,
1.1 nicm 87: &cmd_save_buffer_entry,
88: &cmd_select_layout_entry,
89: &cmd_select_pane_entry,
90: &cmd_select_window_entry,
91: &cmd_send_keys_entry,
92: &cmd_send_prefix_entry,
93: &cmd_server_info_entry,
94: &cmd_set_buffer_entry,
1.13 nicm 95: &cmd_set_environment_entry,
1.1 nicm 96: &cmd_set_option_entry,
97: &cmd_set_window_option_entry,
98: &cmd_show_buffer_entry,
1.13 nicm 99: &cmd_show_environment_entry,
1.32 nicm 100: &cmd_show_messages_entry,
1.1 nicm 101: &cmd_show_options_entry,
102: &cmd_show_window_options_entry,
103: &cmd_source_file_entry,
104: &cmd_split_window_entry,
105: &cmd_start_server_entry,
106: &cmd_suspend_client_entry,
107: &cmd_swap_pane_entry,
108: &cmd_swap_window_entry,
109: &cmd_switch_client_entry,
110: &cmd_unbind_key_entry,
111: &cmd_unlink_window_entry,
112: NULL
113: };
114:
1.47 nicm 115: struct session *cmd_choose_session_list(struct sessionslist *);
1.52 nicm 116: struct session *cmd_choose_session(int);
1.31 nicm 117: struct client *cmd_choose_client(struct clients *);
1.5 nicm 118: struct client *cmd_lookup_client(const char *);
119: struct session *cmd_lookup_session(const char *, int *);
120: struct winlink *cmd_lookup_window(struct session *, const char *, int *);
1.8 nicm 121: int cmd_lookup_index(struct session *, const char *, int *);
1.51 nicm 122: struct window_pane *cmd_lookup_paneid(const char *);
1.60 ! nicm 123: struct winlink *cmd_lookup_winlink_windowid(struct session *, const char *);
! 124: struct window *cmd_lookup_windowid(const char *);
! 125: struct session *cmd_window_session(struct cmd_ctx *,
! 126: struct window *, struct winlink **);
1.43 nicm 127: struct winlink *cmd_find_window_offset(const char *, struct session *, int *);
128: int cmd_find_index_offset(const char *, struct session *, int *);
1.51 nicm 129: struct window_pane *cmd_find_pane_offset(const char *, struct winlink *);
1.5 nicm 130:
1.10 nicm 131: int
132: cmd_pack_argv(int argc, char **argv, char *buf, size_t len)
133: {
134: size_t arglen;
135: int i;
136:
137: *buf = '\0';
138: for (i = 0; i < argc; i++) {
139: if (strlcpy(buf, argv[i], len) >= len)
140: return (-1);
141: arglen = strlen(argv[i]) + 1;
142: buf += arglen;
143: len -= arglen;
144: }
145:
146: return (0);
147: }
148:
149: int
150: cmd_unpack_argv(char *buf, size_t len, int argc, char ***argv)
151: {
152: int i;
153: size_t arglen;
154:
155: if (argc == 0)
156: return (0);
157: *argv = xcalloc(argc, sizeof **argv);
158:
159: buf[len - 1] = '\0';
160: for (i = 0; i < argc; i++) {
161: if (len == 0) {
162: cmd_free_argv(argc, *argv);
163: return (-1);
164: }
165:
166: arglen = strlen(buf) + 1;
167: (*argv)[i] = xstrdup(buf);
168: buf += arglen;
169: len -= arglen;
170: }
171:
172: return (0);
1.46 nicm 173: }
174:
175: char **
1.49 nicm 176: cmd_copy_argv(int argc, char *const *argv)
1.46 nicm 177: {
178: char **new_argv;
179: int i;
180:
181: if (argc == 0)
182: return (NULL);
183: new_argv = xcalloc(argc, sizeof *new_argv);
184: for (i = 0; i < argc; i++) {
185: if (argv[i] != NULL)
186: new_argv[i] = xstrdup(argv[i]);
187: }
188: return (new_argv);
1.10 nicm 189: }
190:
191: void
192: cmd_free_argv(int argc, char **argv)
193: {
194: int i;
195:
196: if (argc == 0)
1.35 nicm 197: return;
1.10 nicm 198: for (i = 0; i < argc; i++) {
199: if (argv[i] != NULL)
200: xfree(argv[i]);
201: }
202: xfree(argv);
203: }
204:
1.1 nicm 205: struct cmd *
206: cmd_parse(int argc, char **argv, char **cause)
207: {
208: const struct cmd_entry **entryp, *entry;
1.27 deraadt 209: struct cmd *cmd;
1.49 nicm 210: struct args *args;
1.1 nicm 211: char s[BUFSIZ];
1.49 nicm 212: int ambiguous = 0;
1.1 nicm 213:
214: *cause = NULL;
1.2 nicm 215: if (argc == 0) {
216: xasprintf(cause, "no command");
1.1 nicm 217: return (NULL);
1.2 nicm 218: }
1.1 nicm 219:
220: entry = NULL;
221: for (entryp = cmd_table; *entryp != NULL; entryp++) {
222: if ((*entryp)->alias != NULL &&
223: strcmp((*entryp)->alias, argv[0]) == 0) {
1.3 nicm 224: ambiguous = 0;
1.1 nicm 225: entry = *entryp;
226: break;
227: }
228:
229: if (strncmp((*entryp)->name, argv[0], strlen(argv[0])) != 0)
230: continue;
231: if (entry != NULL)
1.3 nicm 232: ambiguous = 1;
1.1 nicm 233: entry = *entryp;
234:
235: /* Bail now if an exact match. */
236: if (strcmp(entry->name, argv[0]) == 0)
237: break;
238: }
1.3 nicm 239: if (ambiguous)
240: goto ambiguous;
1.1 nicm 241: if (entry == NULL) {
242: xasprintf(cause, "unknown command: %s", argv[0]);
243: return (NULL);
244: }
245:
1.49 nicm 246: args = args_parse(entry->args_template, argc, argv);
247: if (args == NULL)
248: goto usage;
249: if (entry->args_lower != -1 && args->argc < entry->args_lower)
250: goto usage;
251: if (entry->args_upper != -1 && args->argc > entry->args_upper)
252: goto usage;
253: if (entry->check != NULL && entry->check(args) != 0)
254: goto usage;
1.1 nicm 255:
256: cmd = xmalloc(sizeof *cmd);
257: cmd->entry = entry;
1.49 nicm 258: cmd->args = args;
1.1 nicm 259: return (cmd);
260:
261: ambiguous:
262: *s = '\0';
263: for (entryp = cmd_table; *entryp != NULL; entryp++) {
264: if (strncmp((*entryp)->name, argv[0], strlen(argv[0])) != 0)
265: continue;
266: if (strlcat(s, (*entryp)->name, sizeof s) >= sizeof s)
267: break;
268: if (strlcat(s, ", ", sizeof s) >= sizeof s)
269: break;
270: }
271: s[strlen(s) - 2] = '\0';
272: xasprintf(cause, "ambiguous command: %s, could be: %s", argv[0], s);
273: return (NULL);
274:
275: usage:
1.49 nicm 276: if (args != NULL)
277: args_free(args);
1.1 nicm 278: xasprintf(cause, "usage: %s %s", entry->name, entry->usage);
279: return (NULL);
280: }
281:
282: int
283: cmd_exec(struct cmd *cmd, struct cmd_ctx *ctx)
284: {
285: return (cmd->entry->exec(cmd, ctx));
286: }
287:
288: void
289: cmd_free(struct cmd *cmd)
290: {
1.49 nicm 291: if (cmd->args != NULL)
292: args_free(cmd->args);
1.1 nicm 293: xfree(cmd);
294: }
295:
296: size_t
297: cmd_print(struct cmd *cmd, char *buf, size_t len)
298: {
1.49 nicm 299: size_t off, used;
300:
301: off = xsnprintf(buf, len, "%s ", cmd->entry->name);
302: if (off < len) {
303: used = args_print(cmd->args, buf + off, len - off);
304: if (used == 0)
305: buf[off - 1] = '\0';
306: else {
307: off += used;
308: buf[off] = '\0';
309: }
310: }
311: return (off);
1.1 nicm 312: }
313:
1.5 nicm 314: /*
315: * Figure out the current session. Use: 1) the current session, if the command
1.31 nicm 316: * context has one; 2) the most recently used session containing the pty of the
317: * calling client, if any; 3) the session specified in the TMUX variable from
318: * the environment (as passed from the client); 4) the most recently used
319: * session from all sessions.
1.5 nicm 320: */
1.1 nicm 321: struct session *
1.52 nicm 322: cmd_current_session(struct cmd_ctx *ctx, int prefer_unattached)
1.1 nicm 323: {
324: struct msg_command_data *data = ctx->msgdata;
1.11 nicm 325: struct client *c = ctx->cmdclient;
1.5 nicm 326: struct session *s;
1.47 nicm 327: struct sessionslist ss;
1.11 nicm 328: struct winlink *wl;
329: struct window_pane *wp;
330: int found;
1.1 nicm 331:
1.14 nicm 332: if (ctx->curclient != NULL && ctx->curclient->session != NULL)
333: return (ctx->curclient->session);
1.1 nicm 334:
1.11 nicm 335: /*
336: * If the name of the calling client's pty is know, build a list of the
337: * sessions that contain it and if any choose either the first or the
338: * newest.
339: */
340: if (c != NULL && c->tty.path != NULL) {
341: ARRAY_INIT(&ss);
1.47 nicm 342: RB_FOREACH(s, sessions, &sessions) {
1.11 nicm 343: found = 0;
344: RB_FOREACH(wl, winlinks, &s->windows) {
345: TAILQ_FOREACH(wp, &wl->window->panes, entry) {
346: if (strcmp(wp->tty, c->tty.path) == 0) {
347: found = 1;
348: break;
349: }
350: }
351: if (found)
352: break;
353: }
354: if (found)
355: ARRAY_ADD(&ss, s);
356: }
357:
1.47 nicm 358: s = cmd_choose_session_list(&ss);
1.11 nicm 359: ARRAY_FREE(&ss);
360: if (s != NULL)
361: return (s);
362: }
363:
364: /* Use the session from the TMUX environment variable. */
1.50 nicm 365: if (data != NULL && data->pid == getpid() && data->idx != -1) {
1.47 nicm 366: s = session_find_by_index(data->idx);
367: if (s != NULL)
368: return (s);
369: }
370:
1.52 nicm 371: return (cmd_choose_session(prefer_unattached));
1.47 nicm 372: }
373:
1.52 nicm 374: /*
375: * Find the most recently used session, preferring unattached if the flag is
376: * set.
377: */
1.47 nicm 378: struct session *
1.52 nicm 379: cmd_choose_session(int prefer_unattached)
1.47 nicm 380: {
381: struct session *s, *sbest;
382: struct timeval *tv = NULL;
383:
384: sbest = NULL;
385: RB_FOREACH(s, sessions, &sessions) {
1.52 nicm 386: if (tv == NULL || timercmp(&s->activity_time, tv, >) ||
387: (prefer_unattached &&
388: !(sbest->flags & SESSION_UNATTACHED) &&
389: (s->flags & SESSION_UNATTACHED))) {
1.47 nicm 390: sbest = s;
391: tv = &s->activity_time;
392: }
393: }
1.1 nicm 394:
1.47 nicm 395: return (sbest);
1.5 nicm 396: }
397:
1.31 nicm 398: /* Find the most recently used session from a list. */
1.5 nicm 399: struct session *
1.47 nicm 400: cmd_choose_session_list(struct sessionslist *ss)
1.5 nicm 401: {
1.31 nicm 402: struct session *s, *sbest;
1.5 nicm 403: struct timeval *tv = NULL;
404: u_int i;
405:
1.31 nicm 406: sbest = NULL;
1.11 nicm 407: for (i = 0; i < ARRAY_LENGTH(ss); i++) {
408: if ((s = ARRAY_ITEM(ss, i)) == NULL)
1.5 nicm 409: continue;
410:
1.31 nicm 411: if (tv == NULL || timercmp(&s->activity_time, tv, >)) {
412: sbest = s;
413: tv = &s->activity_time;
1.1 nicm 414: }
415: }
1.5 nicm 416:
1.31 nicm 417: return (sbest);
1.1 nicm 418: }
419:
1.30 nicm 420: /*
421: * Find the current client. First try the current client if set, then pick the
1.31 nicm 422: * most recently used of the clients attached to the current session if any,
423: * then of all clients.
1.30 nicm 424: */
425: struct client *
426: cmd_current_client(struct cmd_ctx *ctx)
427: {
428: struct session *s;
429: struct client *c;
430: struct clients cc;
431: u_int i;
432:
433: if (ctx->curclient != NULL)
434: return (ctx->curclient);
435:
436: /*
437: * No current client set. Find the current session and return the
438: * newest of its clients.
439: */
1.52 nicm 440: s = cmd_current_session(ctx, 0);
1.30 nicm 441: if (s != NULL && !(s->flags & SESSION_UNATTACHED)) {
442: ARRAY_INIT(&cc);
443: for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
444: if ((c = ARRAY_ITEM(&clients, i)) == NULL)
445: continue;
446: if (s == c->session)
447: ARRAY_ADD(&cc, c);
448: }
449:
1.31 nicm 450: c = cmd_choose_client(&cc);
1.30 nicm 451: ARRAY_FREE(&cc);
452: if (c != NULL)
453: return (c);
454: }
455:
1.31 nicm 456: return (cmd_choose_client(&clients));
1.30 nicm 457: }
458:
1.31 nicm 459: /* Choose the most recently used client from a list. */
1.20 nicm 460: struct client *
1.31 nicm 461: cmd_choose_client(struct clients *cc)
1.20 nicm 462: {
1.31 nicm 463: struct client *c, *cbest;
1.20 nicm 464: struct timeval *tv = NULL;
465: u_int i;
466:
1.31 nicm 467: cbest = NULL;
1.30 nicm 468: for (i = 0; i < ARRAY_LENGTH(cc); i++) {
469: if ((c = ARRAY_ITEM(cc, i)) == NULL)
1.20 nicm 470: continue;
471: if (c->session == NULL)
472: continue;
473:
1.31 nicm 474: if (tv == NULL || timercmp(&c->activity_time, tv, >)) {
475: cbest = c;
476: tv = &c->activity_time;
1.20 nicm 477: }
478: }
479:
1.31 nicm 480: return (cbest);
1.20 nicm 481: }
482:
1.5 nicm 483: /* Find the target client or report an error and return NULL. */
1.1 nicm 484: struct client *
485: cmd_find_client(struct cmd_ctx *ctx, const char *arg)
486: {
1.30 nicm 487: struct client *c;
1.5 nicm 488: char *tmparg;
489: size_t arglen;
1.1 nicm 490:
1.5 nicm 491: /* A NULL argument means the current client. */
1.30 nicm 492: if (arg == NULL)
493: return (cmd_current_client(ctx));
1.5 nicm 494: tmparg = xstrdup(arg);
495:
496: /* Trim a single trailing colon if any. */
497: arglen = strlen(tmparg);
498: if (arglen != 0 && tmparg[arglen - 1] == ':')
499: tmparg[arglen - 1] = '\0';
500:
501: /* Find the client, if any. */
502: c = cmd_lookup_client(tmparg);
503:
504: /* If no client found, report an error. */
505: if (c == NULL)
506: ctx->error(ctx, "client not found: %s", tmparg);
507:
508: xfree(tmparg);
509: return (c);
510: }
511:
512: /*
513: * Lookup a client by device path. Either of a full match and a match without a
514: * leading _PATH_DEV ("/dev/") is accepted.
515: */
516: struct client *
517: cmd_lookup_client(const char *name)
518: {
519: struct client *c;
520: const char *path;
521: u_int i;
522:
523: for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
1.25 nicm 524: c = ARRAY_ITEM(&clients, i);
525: if (c == NULL || c->session == NULL)
1.5 nicm 526: continue;
527: path = c->tty.path;
528:
529: /* Check for exact matches. */
530: if (strcmp(name, path) == 0)
531: return (c);
532:
533: /* Check without leading /dev if present. */
534: if (strncmp(path, _PATH_DEV, (sizeof _PATH_DEV) - 1) != 0)
535: continue;
536: if (strcmp(name, path + (sizeof _PATH_DEV) - 1) == 0)
537: return (c);
538: }
539:
540: return (NULL);
541: }
542:
543: /* Lookup a session by name. If no session is found, NULL is returned. */
544: struct session *
545: cmd_lookup_session(const char *name, int *ambiguous)
546: {
547: struct session *s, *sfound;
548:
549: *ambiguous = 0;
550:
551: /*
1.28 nicm 552: * Look for matches. First look for exact matches - session names must
553: * be unique so an exact match can't be ambigious and can just be
554: * returned.
1.5 nicm 555: */
1.47 nicm 556: if ((s = session_find(name)) != NULL)
557: return (s);
1.28 nicm 558:
559: /*
560: * Otherwise look for partial matches, returning early if it is found to
561: * be ambiguous.
562: */
563: sfound = NULL;
1.47 nicm 564: RB_FOREACH(s, sessions, &sessions) {
1.5 nicm 565: if (strncmp(name, s->name, strlen(name)) == 0 ||
566: fnmatch(name, s->name, 0) == 0) {
567: if (sfound != NULL) {
568: *ambiguous = 1;
569: return (NULL);
570: }
571: sfound = s;
572: }
573: }
1.35 nicm 574: return (sfound);
1.5 nicm 575: }
576:
577: /*
1.8 nicm 578: * Lookup a window or return -1 if not found or ambigious. First try as an
579: * index and if invalid, use fnmatch or leading prefix. Return NULL but fill in
1.41 nicm 580: * idx if the window index is a valid number but there is no window with that
1.8 nicm 581: * index.
1.5 nicm 582: */
583: struct winlink *
584: cmd_lookup_window(struct session *s, const char *name, int *ambiguous)
585: {
586: struct winlink *wl, *wlfound;
587: const char *errstr;
588: u_int idx;
589:
590: *ambiguous = 0;
591:
1.60 ! nicm 592: /* Try as a window id. */
! 593: if ((wl = cmd_lookup_winlink_windowid(s, name)) != NULL)
! 594: return (wl);
! 595:
1.5 nicm 596: /* First see if this is a valid window index in this session. */
597: idx = strtonum(name, 0, INT_MAX, &errstr);
598: if (errstr == NULL) {
599: if ((wl = winlink_find_by_index(&s->windows, idx)) != NULL)
600: return (wl);
601: }
1.35 nicm 602:
1.5 nicm 603: /* Look for exact matches, error if more than one. */
604: wlfound = NULL;
605: RB_FOREACH(wl, winlinks, &s->windows) {
1.8 nicm 606: if (strcmp(name, wl->window->name) == 0) {
1.5 nicm 607: if (wlfound != NULL) {
608: *ambiguous = 1;
609: return (NULL);
610: }
611: wlfound = wl;
1.1 nicm 612: }
613: }
1.5 nicm 614: if (wlfound != NULL)
615: return (wlfound);
616:
617: /* Now look for pattern matches, again error if multiple. */
618: wlfound = NULL;
619: RB_FOREACH(wl, winlinks, &s->windows) {
1.8 nicm 620: if (strncmp(name, wl->window->name, strlen(name)) == 0 ||
621: fnmatch(name, wl->window->name, 0) == 0) {
1.5 nicm 622: if (wlfound != NULL) {
623: *ambiguous = 1;
624: return (NULL);
625: }
626: wlfound = wl;
627: }
1.35 nicm 628: }
1.5 nicm 629: if (wlfound != NULL)
630: return (wlfound);
631:
632: return (NULL);
1.1 nicm 633: }
634:
1.8 nicm 635: /*
636: * Find a window index - if the window doesn't exist, check if it is a
637: * potential index and return it anyway.
638: */
639: int
640: cmd_lookup_index(struct session *s, const char *name, int *ambiguous)
641: {
642: struct winlink *wl;
643: const char *errstr;
644: u_int idx;
645:
646: if ((wl = cmd_lookup_window(s, name, ambiguous)) != NULL)
647: return (wl->idx);
648: if (*ambiguous)
649: return (-1);
650:
651: idx = strtonum(name, 0, INT_MAX, &errstr);
652: if (errstr == NULL)
653: return (idx);
654:
655: return (-1);
656: }
657:
1.60 ! nicm 658: /* Lookup pane id. An initial % means a pane id. */
1.51 nicm 659: struct window_pane *
660: cmd_lookup_paneid(const char *arg)
661: {
662: const char *errstr;
663: u_int paneid;
664:
665: if (*arg != '%')
666: return (NULL);
667:
668: paneid = strtonum(arg + 1, 0, UINT_MAX, &errstr);
669: if (errstr != NULL)
670: return (NULL);
671: return (window_pane_find_by_id(paneid));
672: }
673:
1.60 ! nicm 674: /* Lookup window id in a session. An initial @ means a window id. */
! 675: struct winlink *
! 676: cmd_lookup_winlink_windowid(struct session *s, const char *arg)
! 677: {
! 678: const char *errstr;
! 679: u_int windowid;
! 680:
! 681: if (*arg != '@')
! 682: return (NULL);
! 683:
! 684: windowid = strtonum(arg + 1, 0, UINT_MAX, &errstr);
! 685: if (errstr != NULL)
! 686: return (NULL);
! 687: return (winlink_find_by_window_id(&s->windows, windowid));
! 688: }
! 689:
! 690: /* Lookup window id. An initial @ means a window id. */
! 691: struct window *
! 692: cmd_lookup_windowid(const char *arg)
! 693: {
! 694: const char *errstr;
! 695: u_int windowid;
! 696:
! 697: if (*arg != '@')
! 698: return (NULL);
! 699:
! 700: windowid = strtonum(arg + 1, 0, UINT_MAX, &errstr);
! 701: if (errstr != NULL)
! 702: return (NULL);
! 703: return (window_find_by_id(windowid));
! 704: }
! 705:
! 706: /* Find session and winlink for window. */
1.51 nicm 707: struct session *
1.60 ! nicm 708: cmd_window_session(struct cmd_ctx *ctx, struct window *w, struct winlink **wlp)
1.51 nicm 709: {
710: struct session *s;
711: struct sessionslist ss;
712: struct winlink *wl;
713:
1.60 ! nicm 714: /* If this window is in the current session, return that winlink. */
1.52 nicm 715: s = cmd_current_session(ctx, 0);
1.51 nicm 716: if (s != NULL) {
1.60 ! nicm 717: wl = winlink_find_by_window(&s->windows, w);
1.51 nicm 718: if (wl != NULL) {
719: if (wlp != NULL)
720: *wlp = wl;
721: return (s);
722: }
723: }
724:
1.60 ! nicm 725: /* Otherwise choose from all sessions with this window. */
1.51 nicm 726: ARRAY_INIT(&ss);
727: RB_FOREACH(s, sessions, &sessions) {
1.60 ! nicm 728: if (winlink_find_by_window(&s->windows, w) != NULL)
1.51 nicm 729: ARRAY_ADD(&ss, s);
730: }
731: s = cmd_choose_session_list(&ss);
732: ARRAY_FREE(&ss);
733: if (wlp != NULL)
1.60 ! nicm 734: *wlp = winlink_find_by_window(&s->windows, w);
1.51 nicm 735: return (s);
736: }
737:
1.5 nicm 738: /* Find the target session or report an error and return NULL. */
1.1 nicm 739: struct session *
1.52 nicm 740: cmd_find_session(struct cmd_ctx *ctx, const char *arg, int prefer_unattached)
1.1 nicm 741: {
1.51 nicm 742: struct session *s;
743: struct window_pane *wp;
1.60 ! nicm 744: struct window *w;
1.51 nicm 745: struct client *c;
746: char *tmparg;
747: size_t arglen;
748: int ambiguous;
1.1 nicm 749:
1.5 nicm 750: /* A NULL argument means the current session. */
1.1 nicm 751: if (arg == NULL)
1.52 nicm 752: return (cmd_current_session(ctx, prefer_unattached));
1.5 nicm 753:
1.60 ! nicm 754: /* Lookup as pane id or window id. */
1.51 nicm 755: if ((wp = cmd_lookup_paneid(arg)) != NULL)
1.60 ! nicm 756: return (cmd_window_session(ctx, wp->window, NULL));
! 757: if ((w = cmd_lookup_windowid(arg)) != NULL)
! 758: return (cmd_window_session(ctx, w, NULL));
1.51 nicm 759:
1.5 nicm 760: /* Trim a single trailing colon if any. */
1.54 nicm 761: tmparg = xstrdup(arg);
1.5 nicm 762: arglen = strlen(tmparg);
763: if (arglen != 0 && tmparg[arglen - 1] == ':')
764: tmparg[arglen - 1] = '\0';
765:
1.53 nicm 766: /* An empty session name is the current session. */
767: if (*tmparg == '\0') {
1.54 nicm 768: xfree(tmparg);
1.53 nicm 769: return (cmd_current_session(ctx, prefer_unattached));
770: }
771:
1.5 nicm 772: /* Find the session, if any. */
773: s = cmd_lookup_session(tmparg, &ambiguous);
774:
775: /* If it doesn't, try to match it as a client. */
776: if (s == NULL && (c = cmd_lookup_client(tmparg)) != NULL)
777: s = c->session;
778:
779: /* If no session found, report an error. */
780: if (s == NULL) {
781: if (ambiguous)
782: ctx->error(ctx, "more than one session: %s", tmparg);
783: else
784: ctx->error(ctx, "session not found: %s", tmparg);
1.1 nicm 785: }
1.5 nicm 786:
787: xfree(tmparg);
1.1 nicm 788: return (s);
789: }
790:
1.5 nicm 791: /* Find the target session and window or report an error and return NULL. */
1.1 nicm 792: struct winlink *
793: cmd_find_window(struct cmd_ctx *ctx, const char *arg, struct session **sp)
794: {
1.51 nicm 795: struct session *s;
796: struct winlink *wl;
797: struct window_pane *wp;
798: const char *winptr;
799: char *sessptr = NULL;
800: int ambiguous = 0;
1.5 nicm 801:
802: /*
803: * Find the current session. There must always be a current session, if
804: * it can't be found, report an error.
805: */
1.52 nicm 806: if ((s = cmd_current_session(ctx, 0)) == NULL) {
1.5 nicm 807: ctx->error(ctx, "can't establish current session");
808: return (NULL);
809: }
810:
811: /* A NULL argument means the current session and window. */
812: if (arg == NULL) {
813: if (sp != NULL)
814: *sp = s;
815: return (s->curw);
816: }
817:
1.51 nicm 818: /* Lookup as pane id. */
819: if ((wp = cmd_lookup_paneid(arg)) != NULL) {
1.60 ! nicm 820: s = cmd_window_session(ctx, wp->window, &wl);
1.51 nicm 821: if (sp != NULL)
822: *sp = s;
823: return (wl);
824: }
825:
1.5 nicm 826: /* Time to look at the argument. If it is empty, that is an error. */
827: if (*arg == '\0')
828: goto not_found;
829:
1.8 nicm 830: /* Find the separating colon and split into window and session. */
1.5 nicm 831: winptr = strchr(arg, ':');
832: if (winptr == NULL)
1.8 nicm 833: goto no_colon;
834: winptr++; /* skip : */
835: sessptr = xstrdup(arg);
836: *strchr(sessptr, ':') = '\0';
1.5 nicm 837:
838: /* Try to lookup the session if present. */
1.8 nicm 839: if (*sessptr != '\0') {
840: if ((s = cmd_lookup_session(sessptr, &ambiguous)) == NULL)
1.5 nicm 841: goto no_session;
842: }
843: if (sp != NULL)
844: *sp = s;
845:
846: /*
847: * Then work out the window. An empty string is the current window,
1.38 nicm 848: * otherwise try special cases then to look it up in the session.
1.5 nicm 849: */
1.8 nicm 850: if (*winptr == '\0')
1.5 nicm 851: wl = s->curw;
1.38 nicm 852: else if (winptr[0] == '!' && winptr[1] == '\0')
853: wl = TAILQ_FIRST(&s->lastw);
1.43 nicm 854: else if (winptr[0] == '+' || winptr[0] == '-')
855: wl = cmd_find_window_offset(winptr, s, &ambiguous);
856: else
1.38 nicm 857: wl = cmd_lookup_window(s, winptr, &ambiguous);
858: if (wl == NULL)
1.5 nicm 859: goto not_found;
1.35 nicm 860:
1.5 nicm 861: if (sessptr != NULL)
862: xfree(sessptr);
863: return (wl);
864:
1.8 nicm 865: no_colon:
1.38 nicm 866: /*
867: * No colon in the string, first try special cases, then as a window
868: * and lastly as a session.
869: */
870: if (arg[0] == '!' && arg[1] == '\0') {
871: if ((wl = TAILQ_FIRST(&s->lastw)) == NULL)
872: goto not_found;
1.41 nicm 873: } else if (arg[0] == '+' || arg[0] == '-') {
1.43 nicm 874: if ((wl = cmd_find_window_offset(arg, s, &ambiguous)) == NULL)
1.41 nicm 875: goto lookup_session;
876: } else if ((wl = cmd_lookup_window(s, arg, &ambiguous)) == NULL)
877: goto lookup_session;
1.8 nicm 878:
879: if (sp != NULL)
880: *sp = s;
881:
882: return (wl);
883:
1.41 nicm 884: lookup_session:
885: if (ambiguous)
886: goto not_found;
1.53 nicm 887: if (*arg != '\0' && (s = cmd_lookup_session(arg, &ambiguous)) == NULL)
1.41 nicm 888: goto no_session;
889:
890: if (sp != NULL)
891: *sp = s;
892:
893: return (s->curw);
894:
1.5 nicm 895: no_session:
896: if (ambiguous)
1.8 nicm 897: ctx->error(ctx, "multiple sessions: %s", arg);
1.5 nicm 898: else
1.8 nicm 899: ctx->error(ctx, "session not found: %s", arg);
1.5 nicm 900: if (sessptr != NULL)
901: xfree(sessptr);
902: return (NULL);
903:
904: not_found:
905: if (ambiguous)
906: ctx->error(ctx, "multiple windows: %s", arg);
907: else
908: ctx->error(ctx, "window not found: %s", arg);
909: if (sessptr != NULL)
910: xfree(sessptr);
911: return (NULL);
912: }
1.1 nicm 913:
1.43 nicm 914: struct winlink *
915: cmd_find_window_offset(const char *winptr, struct session *s, int *ambiguous)
916: {
917: struct winlink *wl;
918: int offset = 1;
919:
920: if (winptr[1] != '\0')
921: offset = strtonum(winptr + 1, 1, INT_MAX, NULL);
922: if (offset == 0)
923: wl = cmd_lookup_window(s, winptr, ambiguous);
924: else {
925: if (winptr[0] == '+')
926: wl = winlink_next_by_number(s->curw, s, offset);
927: else
928: wl = winlink_previous_by_number(s->curw, s, offset);
929: }
930:
931: return (wl);
932: }
933:
1.5 nicm 934: /*
935: * Find the target session and window index, whether or not it exists in the
936: * session. Return -2 on error or -1 if no window index is specified. This is
1.8 nicm 937: * used when parsing an argument for a window target that may not exist (for
938: * example if it is going to be created).
1.5 nicm 939: */
940: int
941: cmd_find_index(struct cmd_ctx *ctx, const char *arg, struct session **sp)
942: {
943: struct session *s;
1.38 nicm 944: struct winlink *wl;
1.8 nicm 945: const char *winptr;
1.5 nicm 946: char *sessptr = NULL;
947: int idx, ambiguous = 0;
948:
949: /*
950: * Find the current session. There must always be a current session, if
951: * it can't be found, report an error.
952: */
1.52 nicm 953: if ((s = cmd_current_session(ctx, 0)) == NULL) {
1.5 nicm 954: ctx->error(ctx, "can't establish current session");
1.9 nicm 955: return (-2);
1.1 nicm 956: }
1.5 nicm 957:
958: /* A NULL argument means the current session and "no window" (-1). */
959: if (arg == NULL) {
960: if (sp != NULL)
961: *sp = s;
962: return (-1);
963: }
964:
965: /* Time to look at the argument. If it is empty, that is an error. */
966: if (*arg == '\0')
967: goto not_found;
968:
969: /* Find the separating colon. If none, assume the current session. */
970: winptr = strchr(arg, ':');
971: if (winptr == NULL)
1.8 nicm 972: goto no_colon;
973: winptr++; /* skip : */
974: sessptr = xstrdup(arg);
975: *strchr(sessptr, ':') = '\0';
1.5 nicm 976:
977: /* Try to lookup the session if present. */
978: if (sessptr != NULL && *sessptr != '\0') {
1.8 nicm 979: if ((s = cmd_lookup_session(sessptr, &ambiguous)) == NULL)
1.5 nicm 980: goto no_session;
981: }
1.1 nicm 982: if (sp != NULL)
983: *sp = s;
984:
1.5 nicm 985: /*
1.8 nicm 986: * Then work out the window. An empty string is a new window otherwise
987: * try to look it up in the session.
1.5 nicm 988: */
1.8 nicm 989: if (*winptr == '\0')
1.38 nicm 990: idx = -1;
991: else if (winptr[0] == '!' && winptr[1] == '\0') {
992: if ((wl = TAILQ_FIRST(&s->lastw)) == NULL)
993: goto not_found;
994: idx = wl->idx;
1.41 nicm 995: } else if (winptr[0] == '+' || winptr[0] == '-') {
1.43 nicm 996: if ((idx = cmd_find_index_offset(winptr, s, &ambiguous)) < 0)
1.41 nicm 997: goto invalid_index;
998: } else if ((idx = cmd_lookup_index(s, winptr, &ambiguous)) == -1)
999: goto invalid_index;
1.35 nicm 1000:
1.5 nicm 1001: if (sessptr != NULL)
1002: xfree(sessptr);
1003: return (idx);
1004:
1.8 nicm 1005: no_colon:
1.38 nicm 1006: /*
1007: * No colon in the string, first try special cases, then as a window
1008: * and lastly as a session.
1009: */
1010: if (arg[0] == '!' && arg[1] == '\0') {
1011: if ((wl = TAILQ_FIRST(&s->lastw)) == NULL)
1012: goto not_found;
1013: idx = wl->idx;
1.41 nicm 1014: } else if (arg[0] == '+' || arg[0] == '-') {
1.43 nicm 1015: if ((idx = cmd_find_index_offset(arg, s, &ambiguous)) < 0)
1.41 nicm 1016: goto lookup_session;
1017: } else if ((idx = cmd_lookup_index(s, arg, &ambiguous)) == -1)
1018: goto lookup_session;
1.8 nicm 1019:
1020: if (sp != NULL)
1021: *sp = s;
1022:
1023: return (idx);
1024:
1.41 nicm 1025: lookup_session:
1026: if (ambiguous)
1027: goto not_found;
1.53 nicm 1028: if (*arg != '\0' && (s = cmd_lookup_session(arg, &ambiguous)) == NULL)
1.41 nicm 1029: goto no_session;
1030:
1031: if (sp != NULL)
1032: *sp = s;
1033:
1034: return (-1);
1035:
1.5 nicm 1036: no_session:
1037: if (ambiguous)
1.35 nicm 1038: ctx->error(ctx, "multiple sessions: %s", arg);
1.5 nicm 1039: else
1.8 nicm 1040: ctx->error(ctx, "session not found: %s", arg);
1.5 nicm 1041: if (sessptr != NULL)
1042: xfree(sessptr);
1043: return (-2);
1044:
1.41 nicm 1045: invalid_index:
1046: if (ambiguous)
1047: goto not_found;
1048: ctx->error(ctx, "invalid index: %s", arg);
1049:
1050: if (sessptr != NULL)
1051: xfree(sessptr);
1052: return (-2);
1053:
1.5 nicm 1054: not_found:
1055: if (ambiguous)
1056: ctx->error(ctx, "multiple windows: %s", arg);
1.1 nicm 1057: else
1.5 nicm 1058: ctx->error(ctx, "window not found: %s", arg);
1059: if (sessptr != NULL)
1060: xfree(sessptr);
1061: return (-2);
1.12 nicm 1062: }
1063:
1.43 nicm 1064: int
1065: cmd_find_index_offset(const char *winptr, struct session *s, int *ambiguous)
1066: {
1067: int idx, offset = 1;
1068:
1069: if (winptr[1] != '\0')
1070: offset = strtonum(winptr + 1, 1, INT_MAX, NULL);
1071: if (offset == 0)
1072: idx = cmd_lookup_index(s, winptr, ambiguous);
1073: else {
1074: if (winptr[0] == '+') {
1075: if (s->curw->idx == INT_MAX)
1076: idx = cmd_lookup_index(s, winptr, ambiguous);
1077: else
1078: idx = s->curw->idx + offset;
1079: } else {
1080: if (s->curw->idx == 0)
1081: idx = cmd_lookup_index(s, winptr, ambiguous);
1082: else
1083: idx = s->curw->idx - offset;
1084: }
1085: }
1086:
1087: return (idx);
1088: }
1089:
1.12 nicm 1090: /*
1091: * Find the target session, window and pane number or report an error and
1092: * return NULL. The pane number is separated from the session:window by a .,
1093: * such as mysession:mywindow.0.
1094: */
1095: struct winlink *
1096: cmd_find_pane(struct cmd_ctx *ctx,
1097: const char *arg, struct session **sp, struct window_pane **wpp)
1098: {
1.55 nicm 1099: struct session *s;
1100: struct winlink *wl;
1101: const char *period, *errstr;
1102: char *winptr, *paneptr;
1103: u_int idx;
1.12 nicm 1104:
1105: /* Get the current session. */
1.52 nicm 1106: if ((s = cmd_current_session(ctx, 0)) == NULL) {
1.27 deraadt 1107: ctx->error(ctx, "can't establish current session");
1.12 nicm 1108: return (NULL);
1109: }
1110: if (sp != NULL)
1111: *sp = s;
1112:
1113: /* A NULL argument means the current session, window and pane. */
1114: if (arg == NULL) {
1115: *wpp = s->curw->window->active;
1116: return (s->curw);
1.51 nicm 1117: }
1118:
1119: /* Lookup as pane id. */
1120: if ((*wpp = cmd_lookup_paneid(arg)) != NULL) {
1.60 ! nicm 1121: s = cmd_window_session(ctx, (*wpp)->window, &wl);
1.51 nicm 1122: if (sp != NULL)
1123: *sp = s;
1124: return (wl);
1.12 nicm 1125: }
1126:
1127: /* Look for a separating period. */
1128: if ((period = strrchr(arg, '.')) == NULL)
1129: goto no_period;
1130:
1131: /* Pull out the window part and parse it. */
1132: winptr = xstrdup(arg);
1133: winptr[period - arg] = '\0';
1134: if (*winptr == '\0')
1135: wl = s->curw;
1136: else if ((wl = cmd_find_window(ctx, winptr, sp)) == NULL)
1137: goto error;
1138:
1139: /* Find the pane section and look it up. */
1140: paneptr = winptr + (period - arg) + 1;
1141: if (*paneptr == '\0')
1142: *wpp = wl->window->active;
1.43 nicm 1143: else if (paneptr[0] == '+' || paneptr[0] == '-')
1144: *wpp = cmd_find_pane_offset(paneptr, wl);
1145: else {
1.12 nicm 1146: idx = strtonum(paneptr, 0, INT_MAX, &errstr);
1.36 nicm 1147: if (errstr != NULL)
1148: goto lookup_string;
1.12 nicm 1149: *wpp = window_pane_at_index(wl->window, idx);
1.36 nicm 1150: if (*wpp == NULL)
1151: goto lookup_string;
1.12 nicm 1152: }
1153:
1154: xfree(winptr);
1155: return (wl);
1156:
1.36 nicm 1157: lookup_string:
1158: /* Try pane string description. */
1.55 nicm 1159: if ((*wpp = window_find_string(wl->window, paneptr)) == NULL) {
1.36 nicm 1160: ctx->error(ctx, "can't find pane: %s", paneptr);
1161: goto error;
1162: }
1163:
1164: xfree(winptr);
1.39 nicm 1165: return (wl);
1.36 nicm 1166:
1.12 nicm 1167: no_period:
1168: /* Try as a pane number alone. */
1169: idx = strtonum(arg, 0, INT_MAX, &errstr);
1170: if (errstr != NULL)
1171: goto lookup_window;
1172:
1173: /* Try index in the current session and window. */
1174: if ((*wpp = window_pane_at_index(s->curw->window, idx)) == NULL)
1175: goto lookup_window;
1.35 nicm 1176:
1.12 nicm 1177: return (s->curw);
1178:
1179: lookup_window:
1.36 nicm 1180: /* Try pane string description. */
1.55 nicm 1181: if ((*wpp = window_find_string(s->curw->window, arg)) != NULL)
1.36 nicm 1182: return (s->curw);
1183:
1.12 nicm 1184: /* Try as a window and use the active pane. */
1185: if ((wl = cmd_find_window(ctx, arg, sp)) != NULL)
1186: *wpp = wl->window->active;
1187: return (wl);
1.35 nicm 1188:
1.12 nicm 1189: error:
1190: xfree(winptr);
1191: return (NULL);
1.43 nicm 1192: }
1193:
1194: struct window_pane *
1195: cmd_find_pane_offset(const char *paneptr, struct winlink *wl)
1196: {
1197: struct window *w = wl->window;
1198: struct window_pane *wp = w->active;
1199: u_int offset = 1;
1200:
1201: if (paneptr[1] != '\0')
1202: offset = strtonum(paneptr + 1, 1, INT_MAX, NULL);
1203: if (offset > 0) {
1204: if (paneptr[0] == '+')
1205: wp = window_pane_next_by_number(w, wp, offset);
1206: else
1207: wp = window_pane_previous_by_number(w, wp, offset);
1208: }
1209:
1210: return (wp);
1.15 nicm 1211: }
1212:
1213: /* Replace the first %% or %idx in template by s. */
1214: char *
1215: cmd_template_replace(char *template, const char *s, int idx)
1216: {
1217: char ch;
1218: char *buf, *ptr;
1219: int replaced;
1220: size_t len;
1221:
1222: if (strstr(template, "%") == NULL)
1223: return (xstrdup(template));
1224:
1225: buf = xmalloc(1);
1226: *buf = '\0';
1227: len = 0;
1228: replaced = 0;
1229:
1230: ptr = template;
1231: while (*ptr != '\0') {
1232: switch (ch = *ptr++) {
1233: case '%':
1234: if (*ptr < '1' || *ptr > '9' || *ptr - '0' != idx) {
1235: if (*ptr != '%' || replaced)
1236: break;
1237: replaced = 1;
1238: }
1239: ptr++;
1240:
1241: len += strlen(s);
1242: buf = xrealloc(buf, 1, len + 1);
1243: strlcat(buf, s, len + 1);
1244: continue;
1245: }
1246: buf = xrealloc(buf, 1, len + 2);
1247: buf[len++] = ch;
1248: buf[len] = '\0';
1249: }
1250:
1251: return (buf);
1.57 nicm 1252: }
1253:
1254: /* Return the default path for a new pane. */
1.59 nicm 1255: const char *
1.57 nicm 1256: cmd_get_default_path(struct cmd_ctx *ctx)
1257: {
1.59 nicm 1258: const char *cwd;
1.57 nicm 1259: struct session *s;
1260: struct window_pane *wp;
1.58 stsp 1261: struct environ_entry *envent;
1.57 nicm 1262:
1263: if ((s = cmd_current_session(ctx, 0)) == NULL)
1264: return (NULL);
1265:
1266: cwd = options_get_string(&s->options, "default-path");
1.58 stsp 1267: if ((cwd[0] == '~' && cwd[1] == '\0') || !strcmp(cwd, "$HOME")) {
1268: envent = environ_find(&global_environ, "HOME");
1269: if (envent != NULL && *envent->value != '\0')
1270: return envent->value;
1271: cwd = "";
1272: }
1.57 nicm 1273: if (*cwd == '\0') {
1274: if (ctx->cmdclient != NULL && ctx->cmdclient->cwd != NULL)
1275: return (ctx->cmdclient->cwd);
1276: if (ctx->curclient != NULL) {
1277: wp = s->curw->window->active;
1278: if ((cwd = get_proc_cwd(wp->pid)) != NULL)
1279: return (cwd);
1280: }
1281: return (s->cwd);
1282: }
1283: return (cwd);
1.1 nicm 1284: }