1: /* $OpenBSD$ */
2:
3: /*
4: * Copyright (c) 2015 Nicholas Marriott <nicholas.marriott@gmail.com>
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:
21: #include <fnmatch.h>
22: #include <limits.h>
23: #include <stdlib.h>
24: #include <string.h>
25: #include <unistd.h>
26:
27: #include "tmux.h"
28:
29: static struct session *cmd_find_try_TMUX(struct client *, struct window *);
30: static int cmd_find_client_better(struct client *, struct client *);
31: static struct client *cmd_find_best_client(struct client **, u_int);
32: static int cmd_find_session_better(struct session *, struct session *,
33: int);
34: static struct session *cmd_find_best_session(struct session **, u_int, int);
35: static int cmd_find_best_session_with_window(struct cmd_find_state *);
36: static int cmd_find_best_winlink_with_window(struct cmd_find_state *);
37:
38: static int cmd_find_current_session_with_client(struct cmd_find_state *);
39: static int cmd_find_current_session(struct cmd_find_state *);
40: static struct client *cmd_find_current_client(struct cmdq_item *);
41:
42: static const char *cmd_find_map_table(const char *[][2], const char *);
43:
44: static int cmd_find_get_session(struct cmd_find_state *, const char *);
45: static int cmd_find_get_window(struct cmd_find_state *, const char *, int);
46: static int cmd_find_get_window_with_session(struct cmd_find_state *,
47: const char *);
48: static int cmd_find_get_pane(struct cmd_find_state *, const char *, int);
49: static int cmd_find_get_pane_with_session(struct cmd_find_state *,
50: const char *);
51: static int cmd_find_get_pane_with_window(struct cmd_find_state *,
52: const char *);
53:
54: static const char *cmd_find_session_table[][2] = {
55: { NULL, NULL }
56: };
57: static const char *cmd_find_window_table[][2] = {
58: { "{start}", "^" },
59: { "{last}", "!" },
60: { "{end}", "$" },
61: { "{next}", "+" },
62: { "{previous}", "-" },
63: { NULL, NULL }
64: };
65: static const char *cmd_find_pane_table[][2] = {
66: { "{last}", "!" },
67: { "{next}", "+" },
68: { "{previous}", "-" },
69: { "{top}", "top" },
70: { "{bottom}", "bottom" },
71: { "{left}", "left" },
72: { "{right}", "right" },
73: { "{top-left}", "top-left" },
74: { "{top-right}", "top-right" },
75: { "{bottom-left}", "bottom-left" },
76: { "{bottom-right}", "bottom-right" },
77: { "{up-of}", "{up-of}" },
78: { "{down-of}", "{down-of}" },
79: { "{left-of}", "{left-of}" },
80: { "{right-of}", "{right-of}" },
81: { NULL, NULL }
82: };
83:
84: /* Get session from TMUX if present. */
85: static struct session *
86: cmd_find_try_TMUX(struct client *c, struct window *w)
87: {
88: struct environ_entry *envent;
89: char tmp[256];
90: long long pid;
91: u_int session;
92: struct session *s;
93:
94: envent = environ_find(c->environ, "TMUX");
95: if (envent == NULL)
96: return (NULL);
97:
98: if (sscanf(envent->value, "%255[^,],%lld,%d", tmp, &pid, &session) != 3)
99: return (NULL);
100: if (pid != getpid())
101: return (NULL);
102: log_debug("client %p TMUX is %s (session @%u)", c, envent->value,
103: session);
104:
105: s = session_find_by_id(session);
106: if (s == NULL || (w != NULL && !session_has(s, w)))
107: return (NULL);
108: return (s);
109: }
110:
111: /* Is this client better? */
112: static int
113: cmd_find_client_better(struct client *c, struct client *than)
114: {
115: if (than == NULL)
116: return (1);
117: return (timercmp(&c->activity_time, &than->activity_time, >));
118: }
119:
120: /* Find best client from a list, or all if list is NULL. */
121: static struct client *
122: cmd_find_best_client(struct client **clist, u_int csize)
123: {
124: struct client *c_loop, *c;
125: u_int i;
126:
127: c = NULL;
128: if (clist != NULL) {
129: for (i = 0; i < csize; i++) {
130: if (clist[i]->session == NULL)
131: continue;
132: if (cmd_find_client_better(clist[i], c))
133: c = clist[i];
134: }
135: } else {
136: TAILQ_FOREACH(c_loop, &clients, entry) {
137: if (c_loop->session == NULL)
138: continue;
139: if (cmd_find_client_better(c_loop, c))
140: c = c_loop;
141: }
142: }
143: return (c);
144: }
145:
146: /* Is this session better? */
147: static int
148: cmd_find_session_better(struct session *s, struct session *than, int flags)
149: {
150: int attached;
151:
152: if (than == NULL)
153: return (1);
154: if (flags & CMD_FIND_PREFER_UNATTACHED) {
155: attached = (~than->flags & SESSION_UNATTACHED);
156: if (attached && (s->flags & SESSION_UNATTACHED))
157: return (1);
158: else if (!attached && (~s->flags & SESSION_UNATTACHED))
159: return (0);
160: }
161: return (timercmp(&s->activity_time, &than->activity_time, >));
162: }
163:
164: /* Find best session from a list, or all if list is NULL. */
165: static struct session *
166: cmd_find_best_session(struct session **slist, u_int ssize, int flags)
167: {
168: struct session *s_loop, *s;
169: u_int i;
170:
171: s = NULL;
172: if (slist != NULL) {
173: for (i = 0; i < ssize; i++) {
174: if (cmd_find_session_better(slist[i], s, flags))
175: s = slist[i];
176: }
177: } else {
178: RB_FOREACH(s_loop, sessions, &sessions) {
179: if (cmd_find_session_better(s_loop, s, flags))
180: s = s_loop;
181: }
182: }
183: return (s);
184: }
185:
186: /* Find best session and winlink for window. */
187: static int
188: cmd_find_best_session_with_window(struct cmd_find_state *fs)
189: {
190: struct session **slist = NULL;
191: u_int ssize;
192: struct session *s;
193:
194: if (fs->item != NULL && fs->item->client != NULL) {
195: fs->s = cmd_find_try_TMUX(fs->item->client, fs->w);
196: if (fs->s != NULL)
197: return (cmd_find_best_winlink_with_window(fs));
198: }
199:
200: ssize = 0;
201: RB_FOREACH(s, sessions, &sessions) {
202: if (!session_has(s, fs->w))
203: continue;
204: slist = xreallocarray(slist, ssize + 1, sizeof *slist);
205: slist[ssize++] = s;
206: }
207: if (ssize == 0)
208: goto fail;
209: fs->s = cmd_find_best_session(slist, ssize, fs->flags);
210: if (fs->s == NULL)
211: goto fail;
212: free(slist);
213: return (cmd_find_best_winlink_with_window(fs));
214:
215: fail:
216: free(slist);
217: return (-1);
218: }
219:
220: /*
221: * Find the best winlink for a window (the current if it contains the pane,
222: * otherwise the first).
223: */
224: static int
225: cmd_find_best_winlink_with_window(struct cmd_find_state *fs)
226: {
227: struct winlink *wl, *wl_loop;
228:
229: wl = NULL;
230: if (fs->s->curw != NULL && fs->s->curw->window == fs->w)
231: wl = fs->s->curw;
232: else {
233: RB_FOREACH(wl_loop, winlinks, &fs->s->windows) {
234: if (wl_loop->window == fs->w) {
235: wl = wl_loop;
236: break;
237: }
238: }
239: }
240: if (wl == NULL)
241: return (-1);
242: fs->wl = wl;
243: fs->idx = fs->wl->idx;
244: return (0);
245: }
246:
247: /* Find current session when we have an unattached client. */
248: static int
249: cmd_find_current_session_with_client(struct cmd_find_state *fs)
250: {
251: struct window_pane *wp;
252:
253: /*
254: * If this is running in a pane, we can use that to limit the list of
255: * sessions to those containing that pane (we still use the current
256: * window in the best session).
257: */
258: if (fs->item != NULL) {
259: RB_FOREACH(wp, window_pane_tree, &all_window_panes) {
260: if (strcmp(wp->tty, fs->item->client->ttyname) == 0)
261: break;
262: }
263: } else
264: wp = NULL;
265:
266: /* Not running in a pane. We know nothing. Find the best session. */
267: if (wp == NULL)
268: goto unknown_pane;
269:
270: /* Find the best session and winlink containing this pane. */
271: fs->w = wp->window;
272: if (cmd_find_best_session_with_window(fs) != 0) {
273: if (wp != NULL) {
274: /*
275: * The window may have been destroyed but the pane
276: * still on all_window_panes due to something else
277: * holding a reference.
278: */
279: goto unknown_pane;
280: }
281: return (-1);
282: }
283:
284: /* Use the current window and pane from this session. */
285: fs->wl = fs->s->curw;
286: fs->idx = fs->wl->idx;
287: fs->w = fs->wl->window;
288: fs->wp = fs->w->active;
289:
290: return (0);
291:
292: unknown_pane:
293: fs->s = NULL;
294: if (fs->item != NULL)
295: fs->s = cmd_find_try_TMUX(fs->item->client, NULL);
296: if (fs->s == NULL)
297: fs->s = cmd_find_best_session(NULL, 0, fs->flags);
298: if (fs->s == NULL)
299: return (-1);
300: fs->wl = fs->s->curw;
301: fs->idx = fs->wl->idx;
302: fs->w = fs->wl->window;
303: fs->wp = fs->w->active;
304:
305: return (0);
306: }
307:
308: /*
309: * Work out the best current state. If this function succeeds, the state is
310: * guaranteed to be completely filled in.
311: */
312: static int
313: cmd_find_current_session(struct cmd_find_state *fs)
314: {
315: /* If we know the current client, use it. */
316: if (fs->item != NULL && fs->item->client != NULL) {
317: log_debug("%s: have client %p%s", __func__, fs->item->client,
318: fs->item->client->session == NULL ? "" : " (with session)");
319: if (fs->item->client->session == NULL)
320: return (cmd_find_current_session_with_client(fs));
321: fs->s = fs->item->client->session;
322: fs->wl = fs->s->curw;
323: fs->idx = fs->wl->idx;
324: fs->w = fs->wl->window;
325: fs->wp = fs->w->active;
326: return (0);
327: }
328:
329: /* We know nothing, find the best session and client. */
330: fs->s = cmd_find_best_session(NULL, 0, fs->flags);
331: if (fs->s == NULL)
332: return (-1);
333: fs->wl = fs->s->curw;
334: fs->idx = fs->wl->idx;
335: fs->w = fs->wl->window;
336: fs->wp = fs->w->active;
337:
338: return (0);
339: }
340:
341: /* Work out the best current client. */
342: static struct client *
343: cmd_find_current_client(struct cmdq_item *item)
344: {
345: struct cmd_find_state current;
346: struct session *s;
347: struct client *c, **clist = NULL;
348: u_int csize;
349:
350: /* If the queue client has a session, use it. */
351: if (item->client != NULL && item->client->session != NULL) {
352: log_debug("%s: using item %p client %p", __func__, item,
353: item->client);
354: return (item->client);
355: }
356:
357: /* Otherwise find the current session. */
358: cmd_find_clear_state(¤t, item, 0);
359: if (cmd_find_current_session(¤t) != 0)
360: return (NULL);
361:
362: /* If it is attached, find the best of it's clients. */
363: s = current.s;
364: log_debug("%s: current session $%u %s", __func__, s->id, s->name);
365: if (~s->flags & SESSION_UNATTACHED) {
366: csize = 0;
367: TAILQ_FOREACH(c, &clients, entry) {
368: if (c->session != s)
369: continue;
370: clist = xreallocarray(clist, csize + 1, sizeof *clist);
371: clist[csize++] = c;
372: }
373: if (csize != 0) {
374: c = cmd_find_best_client(clist, csize);
375: if (c != NULL) {
376: free(clist);
377: return (c);
378: }
379: }
380: free(clist);
381: }
382:
383: /* Otherwise pick best of all clients. */
384: return (cmd_find_best_client(NULL, 0));
385: }
386:
387: /* Maps string in table. */
388: static const char *
389: cmd_find_map_table(const char *table[][2], const char *s)
390: {
391: u_int i;
392:
393: for (i = 0; table[i][0] != NULL; i++) {
394: if (strcmp(s, table[i][0]) == 0)
395: return (table[i][1]);
396: }
397: return (s);
398: }
399:
400: /* Find session from string. Fills in s. */
401: static int
402: cmd_find_get_session(struct cmd_find_state *fs, const char *session)
403: {
404: struct session *s, *s_loop;
405: struct client *c;
406:
407: log_debug("%s: %s", __func__, session);
408:
409: /* Check for session ids starting with $. */
410: if (*session == '$') {
411: fs->s = session_find_by_id_str(session);
412: if (fs->s == NULL)
413: return (-1);
414: return (0);
415: }
416:
417: /* Look for exactly this session. */
418: fs->s = session_find(session);
419: if (fs->s != NULL)
420: return (0);
421:
422: /* Look for as a client. */
423: c = cmd_find_client(NULL, session, 1);
424: if (c != NULL && c->session != NULL) {
425: fs->s = c->session;
426: return (0);
427: }
428:
429: /* Stop now if exact only. */
430: if (fs->flags & CMD_FIND_EXACT_SESSION)
431: return (-1);
432:
433: /* Otherwise look for prefix. */
434: s = NULL;
435: RB_FOREACH(s_loop, sessions, &sessions) {
436: if (strncmp(session, s_loop->name, strlen(session)) == 0) {
437: if (s != NULL)
438: return (-1);
439: s = s_loop;
440: }
441: }
442: if (s != NULL) {
443: fs->s = s;
444: return (0);
445: }
446:
447: /* Then as a pattern. */
448: s = NULL;
449: RB_FOREACH(s_loop, sessions, &sessions) {
450: if (fnmatch(session, s_loop->name, 0) == 0) {
451: if (s != NULL)
452: return (-1);
453: s = s_loop;
454: }
455: }
456: if (s != NULL) {
457: fs->s = s;
458: return (0);
459: }
460:
461: return (-1);
462: }
463:
464: /* Find window from string. Fills in s, wl, w. */
465: static int
466: cmd_find_get_window(struct cmd_find_state *fs, const char *window, int only)
467: {
468: log_debug("%s: %s", __func__, window);
469:
470: /* Check for window ids starting with @. */
471: if (*window == '@') {
472: fs->w = window_find_by_id_str(window);
473: if (fs->w == NULL)
474: return (-1);
475: return (cmd_find_best_session_with_window(fs));
476: }
477:
478: /* Not a window id, so use the current session. */
479: fs->s = fs->current->s;
480:
481: /* We now only need to find the winlink in this session. */
482: if (cmd_find_get_window_with_session(fs, window) == 0)
483: return (0);
484:
485: /* Otherwise try as a session itself. */
486: if (!only && cmd_find_get_session(fs, window) == 0) {
487: fs->wl = fs->s->curw;
488: fs->w = fs->wl->window;
489: if (~fs->flags & CMD_FIND_WINDOW_INDEX)
490: fs->idx = fs->wl->idx;
491: return (0);
492: }
493:
494: return (-1);
495: }
496:
497: /*
498: * Find window from string, assuming it is in given session. Needs s, fills in
499: * wl and w.
500: */
501: static int
502: cmd_find_get_window_with_session(struct cmd_find_state *fs, const char *window)
503: {
504: struct winlink *wl;
505: const char *errstr;
506: int idx, n, exact;
507: struct session *s;
508:
509: log_debug("%s: %s", __func__, window);
510: exact = (fs->flags & CMD_FIND_EXACT_WINDOW);
511:
512: /*
513: * Start with the current window as the default. So if only an index is
514: * found, the window will be the current.
515: */
516: fs->wl = fs->s->curw;
517: fs->w = fs->wl->window;
518:
519: /* Check for window ids starting with @. */
520: if (*window == '@') {
521: fs->w = window_find_by_id_str(window);
522: if (fs->w == NULL || !session_has(fs->s, fs->w))
523: return (-1);
524: return (cmd_find_best_winlink_with_window(fs));
525: }
526:
527: /* Try as an offset. */
528: if (!exact && (window[0] == '+' || window[0] == '-')) {
529: if (window[1] != '\0')
530: n = strtonum(window + 1, 1, INT_MAX, NULL);
531: else
532: n = 1;
533: s = fs->s;
534: if (fs->flags & CMD_FIND_WINDOW_INDEX) {
535: if (window[0] == '+') {
536: if (INT_MAX - s->curw->idx < n)
537: return (-1);
538: fs->idx = s->curw->idx + n;
539: } else {
540: if (n < s->curw->idx)
541: return (-1);
542: fs->idx = s->curw->idx - n;
543: }
544: return (0);
545: }
546: if (window[0] == '+')
547: fs->wl = winlink_next_by_number(s->curw, s, n);
548: else
549: fs->wl = winlink_previous_by_number(s->curw, s, n);
550: if (fs->wl != NULL) {
551: fs->idx = fs->wl->idx;
552: fs->w = fs->wl->window;
553: return (0);
554: }
555: }
556:
557: /* Try special characters. */
558: if (!exact) {
559: if (strcmp(window, "!") == 0) {
560: fs->wl = TAILQ_FIRST(&fs->s->lastw);
561: if (fs->wl == NULL)
562: return (-1);
563: fs->idx = fs->wl->idx;
564: fs->w = fs->wl->window;
565: return (0);
566: } else if (strcmp(window, "^") == 0) {
567: fs->wl = RB_MIN(winlinks, &fs->s->windows);
568: if (fs->wl == NULL)
569: return (-1);
570: fs->idx = fs->wl->idx;
571: fs->w = fs->wl->window;
572: return (0);
573: } else if (strcmp(window, "$") == 0) {
574: fs->wl = RB_MAX(winlinks, &fs->s->windows);
575: if (fs->wl == NULL)
576: return (-1);
577: fs->idx = fs->wl->idx;
578: fs->w = fs->wl->window;
579: return (0);
580: }
581: }
582:
583: /* First see if this is a valid window index in this session. */
584: if (window[0] != '+' && window[0] != '-') {
585: idx = strtonum(window, 0, INT_MAX, &errstr);
586: if (errstr == NULL) {
587: if (fs->flags & CMD_FIND_WINDOW_INDEX) {
588: fs->idx = idx;
589: return (0);
590: }
591: fs->wl = winlink_find_by_index(&fs->s->windows, idx);
592: if (fs->wl != NULL) {
593: fs->w = fs->wl->window;
594: return (0);
595: }
596: }
597: }
598:
599: /* Look for exact matches, error if more than one. */
600: fs->wl = NULL;
601: RB_FOREACH(wl, winlinks, &fs->s->windows) {
602: if (strcmp(window, wl->window->name) == 0) {
603: if (fs->wl != NULL)
604: return (-1);
605: fs->wl = wl;
606: }
607: }
608: if (fs->wl != NULL) {
609: fs->idx = fs->wl->idx;
610: fs->w = fs->wl->window;
611: return (0);
612: }
613:
614: /* Stop now if exact only. */
615: if (exact)
616: return (-1);
617:
618: /* Try as the start of a window name, error if multiple. */
619: fs->wl = NULL;
620: RB_FOREACH(wl, winlinks, &fs->s->windows) {
621: if (strncmp(window, wl->window->name, strlen(window)) == 0) {
622: if (fs->wl != NULL)
623: return (-1);
624: fs->wl = wl;
625: }
626: }
627: if (fs->wl != NULL) {
628: fs->idx = fs->wl->idx;
629: fs->w = fs->wl->window;
630: return (0);
631: }
632:
633: /* Now look for pattern matches, again error if multiple. */
634: fs->wl = NULL;
635: RB_FOREACH(wl, winlinks, &fs->s->windows) {
636: if (fnmatch(window, wl->window->name, 0) == 0) {
637: if (fs->wl != NULL)
638: return (-1);
639: fs->wl = wl;
640: }
641: }
642: if (fs->wl != NULL) {
643: fs->idx = fs->wl->idx;
644: fs->w = fs->wl->window;
645: return (0);
646: }
647:
648: return (-1);
649: }
650:
651: /* Find pane from string. Fills in s, wl, w, wp. */
652: static int
653: cmd_find_get_pane(struct cmd_find_state *fs, const char *pane, int only)
654: {
655: log_debug("%s: %s", __func__, pane);
656:
657: /* Check for pane ids starting with %. */
658: if (*pane == '%') {
659: fs->wp = window_pane_find_by_id_str(pane);
660: if (fs->wp == NULL || window_pane_outside(fs->wp))
661: return (-1);
662: fs->w = fs->wp->window;
663: return (cmd_find_best_session_with_window(fs));
664: }
665:
666: /* Not a pane id, so try the current session and window. */
667: fs->s = fs->current->s;
668: fs->wl = fs->current->wl;
669: fs->idx = fs->current->idx;
670: fs->w = fs->current->w;
671:
672: /* We now only need to find the pane in this window. */
673: if (cmd_find_get_pane_with_window(fs, pane) == 0)
674: return (0);
675:
676: /* Otherwise try as a window itself (this will also try as session). */
677: if (!only && cmd_find_get_window(fs, pane, 0) == 0) {
678: fs->wp = fs->w->active;
679: return (0);
680: }
681:
682: return (-1);
683: }
684:
685: /*
686: * Find pane from string, assuming it is in given session. Needs s, fills in wl
687: * and w and wp.
688: */
689: static int
690: cmd_find_get_pane_with_session(struct cmd_find_state *fs, const char *pane)
691: {
692: log_debug("%s: %s", __func__, pane);
693:
694: /* Check for pane ids starting with %. */
695: if (*pane == '%') {
696: fs->wp = window_pane_find_by_id_str(pane);
697: if (fs->wp == NULL || window_pane_outside(fs->wp))
698: return (-1);
699: fs->w = fs->wp->window;
700: return (cmd_find_best_winlink_with_window(fs));
701: }
702:
703: /* Otherwise use the current window. */
704: fs->wl = fs->s->curw;
705: fs->idx = fs->wl->idx;
706: fs->w = fs->wl->window;
707:
708: /* Now we just need to look up the pane. */
709: return (cmd_find_get_pane_with_window(fs, pane));
710: }
711:
712: /*
713: * Find pane from string, assuming it is in the given window. Needs w, fills in
714: * wp.
715: */
716: static int
717: cmd_find_get_pane_with_window(struct cmd_find_state *fs, const char *pane)
718: {
719: const char *errstr;
720: int idx;
721: struct window_pane *wp;
722: u_int n;
723:
724: log_debug("%s: %s", __func__, pane);
725:
726: /* Check for pane ids starting with %. */
727: if (*pane == '%') {
728: fs->wp = window_pane_find_by_id_str(pane);
729: if (fs->wp == NULL || window_pane_outside(fs->wp))
730: return (-1);
731: if (fs->wp->window != fs->w)
732: return (-1);
733: return (0);
734: }
735:
736: /* Try special characters. */
737: if (strcmp(pane, "!") == 0) {
738: if (fs->w->last == NULL)
739: return (-1);
740: fs->wp = fs->w->last;
741: if (fs->wp == NULL || window_pane_outside(fs->wp))
742: return (-1);
743: return (0);
744: } else if (strcmp(pane, "{up-of}") == 0) {
745: fs->wp = window_pane_find_up(fs->w->active);
746: if (fs->wp == NULL || window_pane_outside(fs->wp))
747: return (-1);
748: return (0);
749: } else if (strcmp(pane, "{down-of}") == 0) {
750: fs->wp = window_pane_find_down(fs->w->active);
751: if (fs->wp == NULL || window_pane_outside(fs->wp))
752: return (-1);
753: return (0);
754: } else if (strcmp(pane, "{left-of}") == 0) {
755: fs->wp = window_pane_find_left(fs->w->active);
756: if (fs->wp == NULL || window_pane_outside(fs->wp))
757: return (-1);
758: return (0);
759: } else if (strcmp(pane, "{right-of}") == 0) {
760: fs->wp = window_pane_find_right(fs->w->active);
761: if (fs->wp == NULL || window_pane_outside(fs->wp))
762: return (-1);
763: return (0);
764: }
765:
766: /* Try as an offset. */
767: if (pane[0] == '+' || pane[0] == '-') {
768: if (pane[1] != '\0')
769: n = strtonum(pane + 1, 1, INT_MAX, NULL);
770: else
771: n = 1;
772: wp = fs->w->active;
773: if (pane[0] == '+')
774: fs->wp = window_pane_next_by_number(fs->w, wp, n);
775: else
776: fs->wp = window_pane_previous_by_number(fs->w, wp, n);
777: if (fs->wp != NULL && !window_pane_outside(fs->wp))
778: return (0);
779: }
780:
781: /* Get pane by index. */
782: idx = strtonum(pane, 0, INT_MAX, &errstr);
783: if (errstr == NULL) {
784: fs->wp = window_pane_at_index(fs->w, idx);
785: if (fs->wp != NULL && !window_pane_outside(fs->wp))
786: return (0);
787: }
788:
789: /* Try as a description. */
790: fs->wp = window_find_string(fs->w, pane);
791: if (fs->wp != NULL && !window_pane_outside(fs->wp))
792: return (0);
793:
794: return (-1);
795: }
796:
797: /* Clear state. */
798: void
799: cmd_find_clear_state(struct cmd_find_state *fs, struct cmdq_item *item,
800: int flags)
801: {
802: memset(fs, 0, sizeof *fs);
803:
804: fs->item = item;
805: fs->flags = flags;
806:
807: fs->idx = -1;
808: }
809:
810: /* Check if state is empty/ */
811: int
812: cmd_find_empty_state(struct cmd_find_state *fs)
813: {
814: if (fs->s == NULL && fs->wl == NULL && fs->w == NULL && fs->wp == NULL)
815: return (1);
816: return (0);
817: }
818:
819: /* Check if a state if valid. */
820: int
821: cmd_find_valid_state(struct cmd_find_state *fs)
822: {
823: struct winlink *wl;
824:
825: if (fs->s == NULL || fs->wl == NULL || fs->w == NULL || fs->wp == NULL)
826: return (0);
827:
828: if (!session_alive(fs->s))
829: return (0);
830:
831: RB_FOREACH(wl, winlinks, &fs->s->windows) {
832: if (wl->window == fs->w && wl == fs->wl)
833: break;
834: }
835: if (wl == NULL)
836: return (0);
837:
838: if (fs->w != fs->wl->window)
839: return (0);
840:
841: if (!window_has_pane(fs->w, fs->wp))
842: return (0);
843: return (!window_pane_outside(fs->wp));
844: }
845:
846: /* Copy a state. */
847: void
848: cmd_find_copy_state(struct cmd_find_state *dst, struct cmd_find_state *src)
849: {
850: dst->s = src->s;
851: dst->wl = src->wl;
852: dst->idx = src->idx;
853: dst->w = src->w;
854: dst->wp = src->wp;
855: }
856:
857: /* Log the result. */
858: void
859: cmd_find_log_state(const char *prefix, struct cmd_find_state *fs)
860: {
861: if (fs->s != NULL)
862: log_debug("%s: s=$%u", prefix, fs->s->id);
863: else
864: log_debug("%s: s=none", prefix);
865: if (fs->wl != NULL) {
866: log_debug("%s: wl=%u %d w=@%u %s", prefix, fs->wl->idx,
867: fs->wl->window == fs->w, fs->w->id, fs->w->name);
868: } else
869: log_debug("%s: wl=none", prefix);
870: if (fs->wp != NULL)
871: log_debug("%s: wp=%%%u", prefix, fs->wp->id);
872: else
873: log_debug("%s: wp=none", prefix);
874: if (fs->idx != -1)
875: log_debug("%s: idx=%d", prefix, fs->idx);
876: else
877: log_debug("%s: idx=none", prefix);
878: }
879:
880: /* Find state from a session. */
881: int
882: cmd_find_from_session(struct cmd_find_state *fs, struct session *s)
883: {
884: cmd_find_clear_state(fs, NULL, 0);
885:
886: fs->s = s;
887: fs->wl = fs->s->curw;
888: fs->w = fs->wl->window;
889: fs->wp = fs->w->active;
890:
891: cmd_find_log_state(__func__, fs);
892: return (0);
893: }
894:
895: /* Find state from a winlink. */
896: int
897: cmd_find_from_winlink(struct cmd_find_state *fs, struct session *s,
898: struct winlink *wl)
899: {
900: cmd_find_clear_state(fs, NULL, 0);
901:
902: fs->s = s;
903: fs->wl = wl;
904: fs->w = wl->window;
905: fs->wp = wl->window->active;
906:
907: cmd_find_log_state(__func__, fs);
908: return (0);
909: }
910:
911: /* Find state from a session and window. */
912: int
913: cmd_find_from_session_window(struct cmd_find_state *fs, struct session *s,
914: struct window *w)
915: {
916: cmd_find_clear_state(fs, NULL, 0);
917:
918: fs->s = s;
919: fs->w = w;
920: if (cmd_find_best_winlink_with_window(fs) != 0)
921: return (-1);
922: fs->wp = fs->w->active;
923:
924: cmd_find_log_state(__func__, fs);
925: return (0);
926: }
927:
928: /* Find state from a window. */
929: int
930: cmd_find_from_window(struct cmd_find_state *fs, struct window *w)
931: {
932: cmd_find_clear_state(fs, NULL, 0);
933:
934: fs->w = w;
935: if (cmd_find_best_session_with_window(fs) != 0)
936: return (-1);
937: if (cmd_find_best_winlink_with_window(fs) != 0)
938: return (-1);
939: fs->wp = fs->w->active;
940:
941: cmd_find_log_state(__func__, fs);
942: return (0);
943: }
944:
945: /* Find state from a pane. */
946: int
947: cmd_find_from_pane(struct cmd_find_state *fs, struct window_pane *wp)
948: {
949: if (cmd_find_from_window(fs, wp->window) != 0)
950: return (-1);
951: if (window_pane_outside(wp))
952: return (-1);
953: fs->wp = wp;
954:
955: cmd_find_log_state(__func__, fs);
956: return (0);
957: }
958:
959: /* Find current state. */
960: int
961: cmd_find_current(struct cmd_find_state *fs, struct cmdq_item *item, int flags)
962: {
963: cmd_find_clear_state(fs, item, flags);
964: if (cmd_find_current_session(fs) != 0) {
965: if (~flags & CMD_FIND_QUIET)
966: cmdq_error(item, "no current session");
967: return (-1);
968: }
969: return (0);
970: }
971:
972: /*
973: * Split target into pieces and resolve for the given type. Fills in the given
974: * state. Returns 0 on success or -1 on error.
975: */
976: int
977: cmd_find_target(struct cmd_find_state *fs, struct cmd_find_state *current,
978: struct cmdq_item *item, const char *target, enum cmd_find_type type,
979: int flags)
980: {
981: struct mouse_event *m;
982: char *colon, *period, *copy = NULL;
983: const char *session, *window, *pane;
984: int window_only = 0, pane_only = 0;
985:
986: /* Log the arguments. */
987: if (target == NULL)
988: log_debug("%s: target none, type %d", __func__, type);
989: else
990: log_debug("%s: target %s, type %d", __func__, target, type);
991: log_debug("%s: item %p, flags %#x", __func__, item, flags);
992:
993: /* Clear new state. */
994: cmd_find_clear_state(fs, item, flags);
995:
996: /* Find current state. */
997: if (server_check_marked() && (flags & CMD_FIND_DEFAULT_MARKED)) {
998: fs->current = &marked_pane;
999: log_debug("%s: current is marked pane", __func__);
1000: } else if (cmd_find_valid_state(&item->current)) {
1001: fs->current = &item->current;
1002: log_debug("%s: current is from queue", __func__);
1003: } else {
1004: fs->current = current;
1005: log_debug("%s: current is from argument", __func__);
1006: }
1007: if (!cmd_find_empty_state(fs->current) &&
1008: !cmd_find_valid_state(fs->current))
1009: fatalx("invalid current find state");
1010:
1011: /* An empty or NULL target is the current. */
1012: if (target == NULL || *target == '\0')
1013: goto current;
1014:
1015: /* Mouse target is a plain = or {mouse}. */
1016: if (strcmp(target, "=") == 0 || strcmp(target, "{mouse}") == 0) {
1017: m = &item->mouse;
1018: switch (type) {
1019: case CMD_FIND_PANE:
1020: fs->wp = cmd_mouse_pane(m, &fs->s, &fs->wl);
1021: if (fs->wp != NULL && !window_pane_outside(fs->wp))
1022: fs->w = fs->wl->window;
1023: break;
1024: case CMD_FIND_WINDOW:
1025: case CMD_FIND_SESSION:
1026: fs->wl = cmd_mouse_window(m, &fs->s);
1027: if (fs->wl != NULL) {
1028: fs->w = fs->wl->window;
1029: fs->wp = fs->w->active;
1030: }
1031: break;
1032: }
1033: if (fs->wp == NULL) {
1034: if (~flags & CMD_FIND_QUIET)
1035: cmdq_error(item, "no mouse target");
1036: goto error;
1037: }
1038: goto found;
1039: }
1040:
1041: /* Marked target is a plain ~ or {marked}. */
1042: if (strcmp(target, "~") == 0 || strcmp(target, "{marked}") == 0) {
1043: if (!server_check_marked()) {
1044: if (~flags & CMD_FIND_QUIET)
1045: cmdq_error(item, "no marked target");
1046: goto error;
1047: }
1048: cmd_find_copy_state(fs, &marked_pane);
1049: goto found;
1050: }
1051:
1052: /* Find separators if they exist. */
1053: copy = xstrdup(target);
1054: colon = strchr(copy, ':');
1055: if (colon != NULL)
1056: *colon++ = '\0';
1057: if (colon == NULL)
1058: period = strchr(copy, '.');
1059: else
1060: period = strchr(colon, '.');
1061: if (period != NULL)
1062: *period++ = '\0';
1063:
1064: /* Set session, window and pane parts. */
1065: session = window = pane = NULL;
1066: if (colon != NULL && period != NULL) {
1067: session = copy;
1068: window = colon;
1069: window_only = 1;
1070: pane = period;
1071: pane_only = 1;
1072: } else if (colon != NULL && period == NULL) {
1073: session = copy;
1074: window = colon;
1075: window_only = 1;
1076: } else if (colon == NULL && period != NULL) {
1077: window = copy;
1078: pane = period;
1079: pane_only = 1;
1080: } else {
1081: if (*copy == '$')
1082: session = copy;
1083: else if (*copy == '@')
1084: window = copy;
1085: else if (*copy == '%')
1086: pane = copy;
1087: else {
1088: switch (type) {
1089: case CMD_FIND_SESSION:
1090: session = copy;
1091: break;
1092: case CMD_FIND_WINDOW:
1093: window = copy;
1094: break;
1095: case CMD_FIND_PANE:
1096: pane = copy;
1097: break;
1098: }
1099: }
1100: }
1101:
1102: /* Set exact match flags. */
1103: if (session != NULL && *session == '=') {
1104: session++;
1105: fs->flags |= CMD_FIND_EXACT_SESSION;
1106: }
1107: if (window != NULL && *window == '=') {
1108: window++;
1109: fs->flags |= CMD_FIND_EXACT_WINDOW;
1110: }
1111:
1112: /* Empty is the same as NULL. */
1113: if (session != NULL && *session == '\0')
1114: session = NULL;
1115: if (window != NULL && *window == '\0')
1116: window = NULL;
1117: if (pane != NULL && *pane == '\0')
1118: pane = NULL;
1119:
1120: /* Map though conversion table. */
1121: if (session != NULL)
1122: session = cmd_find_map_table(cmd_find_session_table, session);
1123: if (window != NULL)
1124: window = cmd_find_map_table(cmd_find_window_table, window);
1125: if (pane != NULL)
1126: pane = cmd_find_map_table(cmd_find_pane_table, pane);
1127:
1128: log_debug("target %s (flags %#x): session=%s, window=%s, pane=%s",
1129: target, flags, session == NULL ? "none" : session,
1130: window == NULL ? "none" : window, pane == NULL ? "none" : pane);
1131:
1132: /* No pane is allowed if want an index. */
1133: if (pane != NULL && (flags & CMD_FIND_WINDOW_INDEX)) {
1134: if (~flags & CMD_FIND_QUIET)
1135: cmdq_error(item, "can't specify pane here");
1136: goto error;
1137: }
1138:
1139: /* If the session isn't NULL, look it up. */
1140: if (session != NULL) {
1141: /* This will fill in session. */
1142: if (cmd_find_get_session(fs, session) != 0)
1143: goto no_session;
1144:
1145: /* If window and pane are NULL, use that session's current. */
1146: if (window == NULL && pane == NULL) {
1147: fs->wl = fs->s->curw;
1148: fs->idx = -1;
1149: fs->w = fs->wl->window;
1150: fs->wp = fs->w->active;
1151: goto found;
1152: }
1153:
1154: /* If window is present but pane not, find window in session. */
1155: if (window != NULL && pane == NULL) {
1156: /* This will fill in winlink and window. */
1157: if (cmd_find_get_window_with_session(fs, window) != 0)
1158: goto no_window;
1159: fs->wp = fs->wl->window->active;
1160: goto found;
1161: }
1162:
1163: /* If pane is present but window not, find pane. */
1164: if (window == NULL && pane != NULL) {
1165: /* This will fill in winlink and window and pane. */
1166: if (cmd_find_get_pane_with_session(fs, pane) != 0)
1167: goto no_pane;
1168: goto found;
1169: }
1170:
1171: /*
1172: * If window and pane are present, find both in session. This
1173: * will fill in winlink and window.
1174: */
1175: if (cmd_find_get_window_with_session(fs, window) != 0)
1176: goto no_window;
1177: /* This will fill in pane. */
1178: if (cmd_find_get_pane_with_window(fs, pane) != 0)
1179: goto no_pane;
1180: goto found;
1181: }
1182:
1183: /* No session. If window and pane, try them. */
1184: if (window != NULL && pane != NULL) {
1185: /* This will fill in session, winlink and window. */
1186: if (cmd_find_get_window(fs, window, window_only) != 0)
1187: goto no_window;
1188: /* This will fill in pane. */
1189: if (cmd_find_get_pane_with_window(fs, pane) != 0)
1190: goto no_pane;
1191: goto found;
1192: }
1193:
1194: /* If just window is present, try it. */
1195: if (window != NULL && pane == NULL) {
1196: /* This will fill in session, winlink and window. */
1197: if (cmd_find_get_window(fs, window, window_only) != 0)
1198: goto no_window;
1199: fs->wp = fs->wl->window->active;
1200: goto found;
1201: }
1202:
1203: /* If just pane is present, try it. */
1204: if (window == NULL && pane != NULL) {
1205: /* This will fill in session, winlink, window and pane. */
1206: if (cmd_find_get_pane(fs, pane, pane_only) != 0)
1207: goto no_pane;
1208: goto found;
1209: }
1210:
1211: current:
1212: /* Use the current session. */
1213: cmd_find_copy_state(fs, fs->current);
1214: if (flags & CMD_FIND_WINDOW_INDEX)
1215: fs->idx = -1;
1216: goto found;
1217:
1218: error:
1219: fs->current = NULL;
1220: log_debug("%s: error", __func__);
1221:
1222: free(copy);
1223: return (-1);
1224:
1225: found:
1226: fs->current = NULL;
1227: cmd_find_log_state(__func__, fs);
1228:
1229: free(copy);
1230: return (0);
1231:
1232: no_session:
1233: if (~flags & CMD_FIND_QUIET)
1234: cmdq_error(item, "can't find session %s", session);
1235: goto error;
1236:
1237: no_window:
1238: if (~flags & CMD_FIND_QUIET)
1239: cmdq_error(item, "can't find window %s", window);
1240: goto error;
1241:
1242: no_pane:
1243: if (~flags & CMD_FIND_QUIET)
1244: cmdq_error(item, "can't find pane %s", pane);
1245: goto error;
1246: }
1247:
1248: /* Find the target client or report an error and return NULL. */
1249: struct client *
1250: cmd_find_client(struct cmdq_item *item, const char *target, int quiet)
1251: {
1252: struct client *c;
1253: char *copy;
1254: size_t size;
1255:
1256: /* A NULL argument means the current client. */
1257: if (item != NULL && target == NULL) {
1258: c = cmd_find_current_client(item);
1259: if (c == NULL && !quiet)
1260: cmdq_error(item, "no current client");
1261: log_debug("%s: no target, return %p", __func__, c);
1262: return (c);
1263: }
1264: copy = xstrdup(target);
1265:
1266: /* Trim a single trailing colon if any. */
1267: size = strlen(copy);
1268: if (size != 0 && copy[size - 1] == ':')
1269: copy[size - 1] = '\0';
1270:
1271: /* Check name and path of each client. */
1272: TAILQ_FOREACH(c, &clients, entry) {
1273: if (c->session == NULL)
1274: continue;
1275: if (strcmp(copy, c->name) == 0)
1276: break;
1277:
1278: if (*c->ttyname == '\0')
1279: continue;
1280: if (strcmp(copy, c->ttyname) == 0)
1281: break;
1282: if (strncmp(c->ttyname, _PATH_DEV, (sizeof _PATH_DEV) - 1) != 0)
1283: continue;
1284: if (strcmp(copy, c->ttyname + (sizeof _PATH_DEV) - 1) == 0)
1285: break;
1286: }
1287:
1288: /* If no client found, report an error. */
1289: if (c == NULL && !quiet)
1290: cmdq_error(item, "can't find client %s", copy);
1291:
1292: free(copy);
1293: log_debug("%s: target %s, return %p", __func__, target, c);
1294: return (c);
1295: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to cmd-find.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / tmux