1: /* $OpenBSD$ */
2:
3: /*
4: * Copyright (c) 2011 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: #include <sys/param.h>
21: #include <sys/wait.h>
22:
23: #include <ctype.h>
24: #include <errno.h>
25: #include <libgen.h>
26: #include <netdb.h>
27: #include <stdarg.h>
28: #include <stdlib.h>
29: #include <string.h>
30: #include <time.h>
31: #include <unistd.h>
32:
33: #include "tmux.h"
34:
35: /*
36: * Build a list of key-value pairs and use them to expand #{key} entries in a
37: * string.
38: */
39:
40: struct format_entry;
41: typedef void (*format_cb)(struct format_tree *, struct format_entry *);
42:
43: static void format_job_callback(struct job *);
44: static char *format_job_get(struct format_tree *, const char *);
45: static void format_job_timer(int, short, void *);
46:
47: static void format_cb_host(struct format_tree *, struct format_entry *);
48: static void format_cb_host_short(struct format_tree *,
49: struct format_entry *);
50: static void format_cb_pid(struct format_tree *, struct format_entry *);
51: static void format_cb_session_alerts(struct format_tree *,
52: struct format_entry *);
53: static void format_cb_window_layout(struct format_tree *,
54: struct format_entry *);
55: static void format_cb_window_visible_layout(struct format_tree *,
56: struct format_entry *);
57: static void format_cb_start_command(struct format_tree *,
58: struct format_entry *);
59: static void format_cb_current_command(struct format_tree *,
60: struct format_entry *);
61: static void format_cb_history_bytes(struct format_tree *,
62: struct format_entry *);
63: static void format_cb_pane_tabs(struct format_tree *,
64: struct format_entry *);
65:
66: static void format_cb_current_path(struct format_tree *,
67: struct format_entry *);
68:
69: static char *format_find(struct format_tree *, const char *, int);
70: static void format_add_cb(struct format_tree *, const char *, format_cb);
71: static void format_add_tv(struct format_tree *, const char *,
72: struct timeval *);
73: static int format_replace(struct format_tree *, const char *, size_t,
74: char **, size_t *, size_t *);
75:
76: static void format_defaults_session(struct format_tree *,
77: struct session *);
78: static void format_defaults_client(struct format_tree *, struct client *);
79: static void format_defaults_winlink(struct format_tree *, struct session *,
80: struct winlink *);
81:
82: /* Entry in format job tree. */
83: struct format_job {
84: u_int tag;
85: const char *cmd;
86: const char *expanded;
87:
88: time_t last;
89: char *out;
90:
91: struct job *job;
92: int status;
93:
94: RB_ENTRY(format_job) entry;
95: };
96:
97: /* Format job tree. */
98: static struct event format_job_event;
99: static int format_job_cmp(struct format_job *, struct format_job *);
100: static RB_HEAD(format_job_tree, format_job) format_jobs = RB_INITIALIZER();
101: RB_GENERATE_STATIC(format_job_tree, format_job, entry, format_job_cmp);
102:
103: /* Format job tree comparison function. */
104: static int
105: format_job_cmp(struct format_job *fj1, struct format_job *fj2)
106: {
107: if (fj1->tag < fj2->tag)
108: return (-1);
109: if (fj1->tag > fj2->tag)
110: return (1);
111: return (strcmp(fj1->cmd, fj2->cmd));
112: }
113:
114: /* Format modifiers. */
115: #define FORMAT_TIMESTRING 0x1
116: #define FORMAT_BASENAME 0x2
117: #define FORMAT_DIRNAME 0x4
118: #define FORMAT_SUBSTITUTE 0x8
119:
120: /* Entry in format tree. */
121: struct format_entry {
122: char *key;
123: char *value;
124: time_t t;
125: format_cb cb;
126: RB_ENTRY(format_entry) entry;
127: };
128:
129: /* Format entry tree. */
130: struct format_tree {
131: struct window *w;
132: struct session *s;
133: struct window_pane *wp;
134:
135: u_int tag;
136: int flags;
137:
138: RB_HEAD(format_entry_tree, format_entry) tree;
139: };
140: static int format_entry_cmp(struct format_entry *, struct format_entry *);
141: RB_GENERATE_STATIC(format_entry_tree, format_entry, entry, format_entry_cmp);
142:
143: /* Format entry tree comparison function. */
144: static int
145: format_entry_cmp(struct format_entry *fe1, struct format_entry *fe2)
146: {
147: return (strcmp(fe1->key, fe2->key));
148: }
149:
150: /* Single-character uppercase aliases. */
151: static const char *format_upper[] = {
152: NULL, /* A */
153: NULL, /* B */
154: NULL, /* C */
155: "pane_id", /* D */
156: NULL, /* E */
157: "window_flags", /* F */
158: NULL, /* G */
159: "host", /* H */
160: "window_index", /* I */
161: NULL, /* J */
162: NULL, /* K */
163: NULL, /* L */
164: NULL, /* M */
165: NULL, /* N */
166: NULL, /* O */
167: "pane_index", /* P */
168: NULL, /* Q */
169: NULL, /* R */
170: "session_name", /* S */
171: "pane_title", /* T */
172: NULL, /* U */
173: NULL, /* V */
174: "window_name", /* W */
175: NULL, /* X */
176: NULL, /* Y */
177: NULL /* Z */
178: };
179:
180: /* Single-character lowercase aliases. */
181: static const char *format_lower[] = {
182: NULL, /* a */
183: NULL, /* b */
184: NULL, /* c */
185: NULL, /* d */
186: NULL, /* e */
187: NULL, /* f */
188: NULL, /* g */
189: "host_short", /* h */
190: NULL, /* i */
191: NULL, /* j */
192: NULL, /* k */
193: NULL, /* l */
194: NULL, /* m */
195: NULL, /* n */
196: NULL, /* o */
197: NULL, /* p */
198: NULL, /* q */
199: NULL, /* r */
200: NULL, /* s */
201: NULL, /* t */
202: NULL, /* u */
203: NULL, /* v */
204: NULL, /* w */
205: NULL, /* x */
206: NULL, /* y */
207: NULL /* z */
208: };
209:
210: /* Format job callback. */
211: static void
212: format_job_callback(struct job *job)
213: {
214: struct format_job *fj = job->data;
215: char *line, *buf;
216: size_t len;
217: struct client *c;
218:
219: fj->job = NULL;
220: free(fj->out);
221:
222: buf = NULL;
223: if ((line = evbuffer_readline(job->event->input)) == NULL) {
224: len = EVBUFFER_LENGTH(job->event->input);
225: buf = xmalloc(len + 1);
226: if (len != 0)
227: memcpy(buf, EVBUFFER_DATA(job->event->input), len);
228: buf[len] = '\0';
229: } else
230: buf = line;
231: fj->out = buf;
232:
233: if (fj->status) {
234: TAILQ_FOREACH(c, &clients, entry)
235: server_status_client(c);
236: fj->status = 0;
237: }
238:
239: log_debug("%s: %s: %s", __func__, fj->cmd, fj->out);
240: }
241:
242: /* Find a job. */
243: static char *
244: format_job_get(struct format_tree *ft, const char *cmd)
245: {
246: struct format_job fj0, *fj;
247: time_t t;
248: char *expanded;
249: int force;
250:
251: fj0.tag = ft->tag;
252: fj0.cmd = cmd;
253: if ((fj = RB_FIND(format_job_tree, &format_jobs, &fj0)) == NULL) {
254: fj = xcalloc(1, sizeof *fj);
255: fj->tag = ft->tag;
256: fj->cmd = xstrdup(cmd);
257: fj->expanded = NULL;
258:
259: xasprintf(&fj->out, "<'%s' not ready>", fj->cmd);
260:
261: RB_INSERT(format_job_tree, &format_jobs, fj);
262: }
263:
264: expanded = format_expand(ft, cmd);
265: if (fj->expanded == NULL || strcmp(expanded, fj->expanded) != 0) {
266: free((void *)fj->expanded);
267: fj->expanded = xstrdup(expanded);
268: force = 1;
269: } else
270: force = (ft->flags & FORMAT_FORCE);
271:
272: t = time(NULL);
273: if (fj->job == NULL && (force || fj->last != t)) {
274: fj->job = job_run(expanded, NULL, NULL, format_job_callback,
275: NULL, fj);
276: if (fj->job == NULL) {
277: free(fj->out);
278: xasprintf(&fj->out, "<'%s' didn't start>", fj->cmd);
279: }
280: fj->last = t;
281: }
282:
283: if (ft->flags & FORMAT_STATUS)
284: fj->status = 1;
285:
286: free(expanded);
287: return (format_expand(ft, fj->out));
288: }
289:
290: /* Remove old jobs. */
291: static void
292: format_job_timer(__unused int fd, __unused short events, __unused void *arg)
293: {
294: struct format_job *fj, *fj1;
295: time_t now;
296: struct timeval tv = { .tv_sec = 60 };
297:
298: now = time(NULL);
299: RB_FOREACH_SAFE(fj, format_job_tree, &format_jobs, fj1) {
300: if (fj->last > now || now - fj->last < 3600)
301: continue;
302: RB_REMOVE(format_job_tree, &format_jobs, fj);
303:
304: log_debug("%s: %s", __func__, fj->cmd);
305:
306: if (fj->job != NULL)
307: job_free(fj->job);
308:
309: free((void *)fj->expanded);
310: free((void *)fj->cmd);
311: free(fj->out);
312:
313: free(fj);
314: }
315:
316: evtimer_del(&format_job_event);
317: evtimer_add(&format_job_event, &tv);
318: }
319:
320: /* Callback for host. */
321: static void
322: format_cb_host(__unused struct format_tree *ft, struct format_entry *fe)
323: {
324: char host[HOST_NAME_MAX + 1];
325:
326: if (gethostname(host, sizeof host) != 0)
327: fe->value = xstrdup("");
328: else
329: fe->value = xstrdup(host);
330: }
331:
332: /* Callback for host_short. */
333: static void
334: format_cb_host_short(__unused struct format_tree *ft, struct format_entry *fe)
335: {
336: char host[HOST_NAME_MAX + 1], *cp;
337:
338: if (gethostname(host, sizeof host) != 0)
339: fe->value = xstrdup("");
340: else {
341: if ((cp = strchr(host, '.')) != NULL)
342: *cp = '\0';
343: fe->value = xstrdup(host);
344: }
345: }
346:
347: /* Callback for pid. */
348: static void
349: format_cb_pid(__unused struct format_tree *ft, struct format_entry *fe)
350: {
351: xasprintf(&fe->value, "%ld", (long)getpid());
352: }
353:
354: /* Callback for session_alerts. */
355: static void
356: format_cb_session_alerts(struct format_tree *ft, struct format_entry *fe)
357: {
358: struct session *s = ft->s;
359: struct winlink *wl;
360: char alerts[256], tmp[16];
361:
362: if (s == NULL)
363: return;
364:
365: *alerts = '\0';
366: RB_FOREACH(wl, winlinks, &s->windows) {
367: if ((wl->flags & WINLINK_ALERTFLAGS) == 0)
368: continue;
369: xsnprintf(tmp, sizeof tmp, "%u", wl->idx);
370:
371: if (*alerts != '\0')
372: strlcat(alerts, ",", sizeof alerts);
373: strlcat(alerts, tmp, sizeof alerts);
374: if (wl->flags & WINLINK_ACTIVITY)
375: strlcat(alerts, "#", sizeof alerts);
376: if (wl->flags & WINLINK_BELL)
377: strlcat(alerts, "!", sizeof alerts);
378: if (wl->flags & WINLINK_SILENCE)
379: strlcat(alerts, "~", sizeof alerts);
380: }
381: fe->value = xstrdup(alerts);
382: }
383:
384: /* Callback for window_layout. */
385: static void
386: format_cb_window_layout(struct format_tree *ft, struct format_entry *fe)
387: {
388: struct window *w = ft->w;
389:
390: if (w == NULL)
391: return;
392:
393: if (w->saved_layout_root != NULL)
394: fe->value = layout_dump(w->saved_layout_root);
395: else
396: fe->value = layout_dump(w->layout_root);
397: }
398:
399: /* Callback for window_visible_layout. */
400: static void
401: format_cb_window_visible_layout(struct format_tree *ft, struct format_entry *fe)
402: {
403: struct window *w = ft->w;
404:
405: if (w == NULL)
406: return;
407:
408: fe->value = layout_dump(w->layout_root);
409: }
410:
411: /* Callback for pane_start_command. */
412: static void
413: format_cb_start_command(struct format_tree *ft, struct format_entry *fe)
414: {
415: struct window_pane *wp = ft->wp;
416:
417: if (wp == NULL)
418: return;
419:
420: fe->value = cmd_stringify_argv(wp->argc, wp->argv);
421: }
422:
423: /* Callback for pane_current_command. */
424: static void
425: format_cb_current_command(struct format_tree *ft, struct format_entry *fe)
426: {
427: struct window_pane *wp = ft->wp;
428: char *cmd;
429:
430: if (wp == NULL)
431: return;
432:
433: cmd = osdep_get_name(wp->fd, wp->tty);
434: if (cmd == NULL || *cmd == '\0') {
435: free(cmd);
436: cmd = cmd_stringify_argv(wp->argc, wp->argv);
437: if (cmd == NULL || *cmd == '\0') {
438: free(cmd);
439: cmd = xstrdup(wp->shell);
440: }
441: }
442: fe->value = parse_window_name(cmd);
443: free(cmd);
444: }
445:
446: /* Callback for pane_current_path. */
447: void
448: format_cb_current_path(struct format_tree *ft, struct format_entry *fe)
449: {
450: struct window_pane *wp = ft->wp;
451: char *cwd;
452:
453: if (wp == NULL)
454: return;
455:
456: cwd = osdep_get_cwd(wp->fd);
457: if (cwd != NULL)
458: fe->value = xstrdup(cwd);
459: }
460:
461: /* Callback for history_bytes. */
462: static void
463: format_cb_history_bytes(struct format_tree *ft, struct format_entry *fe)
464: {
465: struct window_pane *wp = ft->wp;
466: struct grid *gd;
467: struct grid_line *gl;
468: unsigned long long size;
469: u_int i;
470:
471: if (wp == NULL)
472: return;
473: gd = wp->base.grid;
474:
475: size = 0;
476: for (i = 0; i < gd->hsize; i++) {
477: gl = &gd->linedata[i];
478: size += gl->cellsize * sizeof *gl->celldata;
479: size += gl->extdsize * sizeof *gl->extddata;
480: }
481: size += gd->hsize * sizeof *gd->linedata;
482:
483: xasprintf(&fe->value, "%llu", size);
484: }
485:
486: /* Callback for pane_tabs. */
487: static void
488: format_cb_pane_tabs(struct format_tree *ft, struct format_entry *fe)
489: {
490: struct window_pane *wp = ft->wp;
491: struct evbuffer *buffer;
492: u_int i;
493: int size;
494:
495: if (wp == NULL)
496: return;
497:
498: buffer = evbuffer_new();
499: for (i = 0; i < wp->base.grid->sx; i++) {
500: if (!bit_test(wp->base.tabs, i))
501: continue;
502:
503: if (EVBUFFER_LENGTH(buffer) > 0)
504: evbuffer_add(buffer, ",", 1);
505: evbuffer_add_printf(buffer, "%u", i);
506: }
507: size = EVBUFFER_LENGTH(buffer);
508: xasprintf(&fe->value, "%.*s", size, EVBUFFER_DATA(buffer));
509: evbuffer_free(buffer);
510: }
511:
512: /* Merge a format tree. */
513: static void
514: format_merge(struct format_tree *ft, struct format_tree *from)
515: {
516: struct format_entry *fe;
517:
518: RB_FOREACH(fe, format_entry_tree, &from->tree) {
519: if (fe->value != NULL)
520: format_add(ft, fe->key, "%s", fe->value);
521: }
522: }
523:
524: /* Create a new tree. */
525: struct format_tree *
526: format_create(struct cmdq_item *item, int tag, int flags)
527: {
528: struct format_tree *ft;
529:
530: if (!event_initialized(&format_job_event)) {
531: evtimer_set(&format_job_event, format_job_timer, NULL);
532: format_job_timer(-1, 0, NULL);
533: }
534:
535: ft = xcalloc(1, sizeof *ft);
536: RB_INIT(&ft->tree);
537:
538: ft->tag = tag;
539: ft->flags = flags;
540:
541: format_add(ft, "version", "%s", VERSION);
542: format_add_cb(ft, "host", format_cb_host);
543: format_add_cb(ft, "host_short", format_cb_host_short);
544: format_add_cb(ft, "pid", format_cb_pid);
545: format_add(ft, "socket_path", "%s", socket_path);
546: format_add_tv(ft, "start_time", &start_time);
547:
548: if (item != NULL && item->cmd != NULL)
549: format_add(ft, "command", "%s", item->cmd->entry->name);
550: if (item != NULL && item->formats != NULL)
551: format_merge(ft, item->formats);
552:
553: return (ft);
554: }
555:
556: /* Free a tree. */
557: void
558: format_free(struct format_tree *ft)
559: {
560: struct format_entry *fe, *fe1;
561:
562: RB_FOREACH_SAFE(fe, format_entry_tree, &ft->tree, fe1) {
563: RB_REMOVE(format_entry_tree, &ft->tree, fe);
564: free(fe->value);
565: free(fe->key);
566: free(fe);
567: }
568:
569: free(ft);
570: }
571:
572: /* Add a key-value pair. */
573: void
574: format_add(struct format_tree *ft, const char *key, const char *fmt, ...)
575: {
576: struct format_entry *fe;
577: struct format_entry *fe_now;
578: va_list ap;
579:
580: fe = xmalloc(sizeof *fe);
581: fe->key = xstrdup(key);
582:
583: fe_now = RB_INSERT(format_entry_tree, &ft->tree, fe);
584: if (fe_now != NULL) {
585: free(fe->key);
586: free(fe);
587: free(fe_now->value);
588: fe = fe_now;
589: }
590:
591: fe->cb = NULL;
592: fe->t = 0;
593:
594: va_start(ap, fmt);
595: xvasprintf(&fe->value, fmt, ap);
596: va_end(ap);
597: }
598:
599: /* Add a key and time. */
600: static void
601: format_add_tv(struct format_tree *ft, const char *key, struct timeval *tv)
602: {
603: struct format_entry *fe;
604: struct format_entry *fe_now;
605:
606: fe = xmalloc(sizeof *fe);
607: fe->key = xstrdup(key);
608:
609: fe_now = RB_INSERT(format_entry_tree, &ft->tree, fe);
610: if (fe_now != NULL) {
611: free(fe->key);
612: free(fe);
613: free(fe_now->value);
614: fe = fe_now;
615: }
616:
617: fe->cb = NULL;
618: fe->t = tv->tv_sec;
619:
620: fe->value = NULL;
621: }
622:
623: /* Add a key and function. */
624: static void
625: format_add_cb(struct format_tree *ft, const char *key, format_cb cb)
626: {
627: struct format_entry *fe;
628: struct format_entry *fe_now;
629:
630: fe = xmalloc(sizeof *fe);
631: fe->key = xstrdup(key);
632:
633: fe_now = RB_INSERT(format_entry_tree, &ft->tree, fe);
634: if (fe_now != NULL) {
635: free(fe->key);
636: free(fe);
637: free(fe_now->value);
638: fe = fe_now;
639: }
640:
641: fe->cb = cb;
642: fe->t = 0;
643:
644: fe->value = NULL;
645: }
646:
647: /* Find a format entry. */
648: static char *
649: format_find(struct format_tree *ft, const char *key, int modifiers)
650: {
651: struct format_entry *fe, fe_find;
652: struct environ_entry *envent;
653: static char s[64];
654: struct options_entry *o;
655: const char *found;
656: int idx;
657: char *copy, *saved;
658:
659: if (~modifiers & FORMAT_TIMESTRING) {
660: o = options_parse_get(global_options, key, &idx, 0);
661: if (o == NULL && ft->w != NULL)
662: o = options_parse_get(ft->w->options, key, &idx, 0);
663: if (o == NULL)
664: o = options_parse_get(global_w_options, key, &idx, 0);
665: if (o == NULL && ft->s != NULL)
666: o = options_parse_get(ft->s->options, key, &idx, 0);
667: if (o == NULL)
668: o = options_parse_get(global_s_options, key, &idx, 0);
669: if (o != NULL) {
670: found = options_tostring(o, idx, 1);
671: goto found;
672: }
673: }
674: found = NULL;
675:
676: fe_find.key = (char *) key;
677: fe = RB_FIND(format_entry_tree, &ft->tree, &fe_find);
678: if (fe != NULL) {
679: if (modifiers & FORMAT_TIMESTRING) {
680: if (fe->t == 0)
681: return (NULL);
682: ctime_r(&fe->t, s);
683: s[strcspn(s, "\n")] = '\0';
684: found = s;
685: goto found;
686: }
687: if (fe->t != 0) {
688: xsnprintf(s, sizeof s, "%lld", (long long)fe->t);
689: found = s;
690: goto found;
691: }
692: if (fe->value == NULL && fe->cb != NULL)
693: fe->cb(ft, fe);
694: found = fe->value;
695: goto found;
696: }
697:
698: if (~modifiers & FORMAT_TIMESTRING) {
699: envent = NULL;
700: if (ft->s != NULL)
701: envent = environ_find(ft->s->environ, key);
702: if (envent == NULL)
703: envent = environ_find(global_environ, key);
704: if (envent != NULL) {
705: found = envent->value;
706: goto found;
707: }
708: }
709:
710: return (NULL);
711:
712: found:
713: if (found == NULL)
714: return (NULL);
715: copy = xstrdup(found);
716: if (modifiers & FORMAT_BASENAME) {
717: saved = copy;
718: copy = xstrdup(basename(saved));
719: free(saved);
720: }
721: if (modifiers & FORMAT_DIRNAME) {
722: saved = copy;
723: copy = xstrdup(dirname(saved));
724: free(saved);
725: }
726: return (copy);
727: }
728:
729: /* Skip until comma. */
730: static char *
731: format_skip(char *s)
732: {
733: int brackets = 0;
734:
735: for (; *s != '\0'; s++) {
736: if (*s == '{')
737: brackets++;
738: if (*s == '}')
739: brackets--;
740: if (*s == ',' && brackets == 0)
741: break;
742: }
743: if (*s == '\0')
744: return (NULL);
745: return (s);
746: }
747:
748: /* Return left and right alternatives separated by commas. */
749: static int
750: format_choose(char *s, char **left, char **right)
751: {
752: char *cp;
753:
754: cp = format_skip(s);
755: if (cp == NULL)
756: return (-1);
757: *cp = '\0';
758:
759: *left = s;
760: *right = cp + 1;
761: return (0);
762: }
763:
764: /* Is this true? */
765: static int
766: format_true(const char *s)
767: {
768: if (s != NULL && *s != '\0' && (s[0] != '0' || s[1] != '\0'))
769: return (1);
770: return (0);
771: }
772:
773: /*
774: * Replace a key/value pair in buffer. #{blah} is expanded directly,
775: * #{?blah,a,b} is replace with a if blah exists and is nonzero else b.
776: */
777: static int
778: format_replace(struct format_tree *ft, const char *key, size_t keylen,
779: char **buf, size_t *len, size_t *off)
780: {
781: char *copy, *copy0, *endptr, *ptr, *found, *new, *value;
782: char *from = NULL, *to = NULL, *left, *right;
783: size_t valuelen, newlen, fromlen, tolen, used;
784: long limit = 0;
785: int modifiers = 0, compare = 0;
786:
787: /* Make a copy of the key. */
788: copy0 = copy = xmalloc(keylen + 1);
789: memcpy(copy, key, keylen);
790: copy[keylen] = '\0';
791:
792: /* Is there a length limit or whatnot? */
793: switch (copy[0]) {
794: case '!':
795: if (copy[1] == '=' && copy[2] == ':') {
796: compare = -1;
797: copy += 3;
798: break;
799: }
800: break;
801: case '=':
802: if (copy[1] == '=' && copy[2] == ':') {
803: compare = 1;
804: copy += 3;
805: break;
806: }
807: errno = 0;
808: limit = strtol(copy + 1, &endptr, 10);
809: if (errno == ERANGE && (limit == LONG_MIN || limit == LONG_MAX))
810: break;
811: if (*endptr != ':')
812: break;
813: copy = endptr + 1;
814: break;
815: case 'b':
816: if (copy[1] != ':')
817: break;
818: modifiers |= FORMAT_BASENAME;
819: copy += 2;
820: break;
821: case 'd':
822: if (copy[1] != ':')
823: break;
824: modifiers |= FORMAT_DIRNAME;
825: copy += 2;
826: break;
827: case 't':
828: if (copy[1] != ':')
829: break;
830: modifiers |= FORMAT_TIMESTRING;
831: copy += 2;
832: break;
833: case 's':
834: if (copy[1] != '/')
835: break;
836: from = copy + 2;
837: for (copy = from; *copy != '\0' && *copy != '/'; copy++)
838: /* nothing */;
839: if (copy[0] != '/' || copy == from) {
840: copy = copy0;
841: break;
842: }
843: copy[0] = '\0';
844: to = copy + 1;
845: for (copy = to; *copy != '\0' && *copy != '/'; copy++)
846: /* nothing */;
847: if (copy[0] != '/' || copy[1] != ':') {
848: copy = copy0;
849: break;
850: }
851: copy[0] = '\0';
852:
853: modifiers |= FORMAT_SUBSTITUTE;
854: copy += 2;
855: break;
856: }
857:
858: /* Is this a comparison or a conditional? */
859: if (compare != 0) {
860: /* Comparison: compare comma-separated left and right. */
861: if (format_choose(copy, &left, &right) != 0)
862: goto fail;
863: left = format_expand(ft, left);
864: right = format_expand(ft, right);
865: if (compare == 1 && strcmp(left, right) == 0)
866: value = xstrdup("1");
867: else if (compare == -1 && strcmp(left, right) != 0)
868: value = xstrdup("1");
869: else
870: value = xstrdup("0");
871: free(right);
872: free(left);
873: } else if (*copy == '?') {
874: /* Conditional: check first and choose second or third. */
875: ptr = format_skip(copy);
876: if (ptr == NULL)
877: goto fail;
878: *ptr = '\0';
879:
880: found = format_find(ft, copy + 1, modifiers);
881: if (found == NULL)
882: found = format_expand(ft, copy + 1);
883: if (format_choose(ptr + 1, &left, &right) != 0)
884: goto fail;
885:
886: if (format_true(found))
887: value = format_expand(ft, left);
888: else
889: value = format_expand(ft, right);
890: free(found);
891: } else {
892: /* Neither: look up directly. */
893: value = format_find(ft, copy, modifiers);
894: if (value == NULL)
895: value = xstrdup("");
896: }
897:
898: /* Perform substitution if any. */
899: if (modifiers & FORMAT_SUBSTITUTE) {
900: fromlen = strlen(from);
901: tolen = strlen(to);
902:
903: newlen = strlen(value) + 1;
904: copy = new = xmalloc(newlen);
905: for (ptr = value; *ptr != '\0'; /* nothing */) {
906: if (strncmp(ptr, from, fromlen) != 0) {
907: *new++ = *ptr++;
908: continue;
909: }
910: used = new - copy;
911:
912: newlen += tolen;
913: copy = xrealloc(copy, newlen);
914:
915: new = copy + used;
916: memcpy(new, to, tolen);
917:
918: new += tolen;
919: ptr += fromlen;
920: }
921: *new = '\0';
922: free(value);
923: value = copy;
924: }
925:
926: /* Truncate the value if needed. */
927: if (limit > 0) {
928: new = utf8_trimcstr(value, limit);
929: free(value);
930: value = new;
931: } else if (limit < 0) {
932: new = utf8_rtrimcstr(value, -limit);
933: free(value);
934: value = new;
935: }
936:
937: /* Expand the buffer and copy in the value. */
938: valuelen = strlen(value);
939: while (*len - *off < valuelen + 1) {
940: *buf = xreallocarray(*buf, 2, *len);
941: *len *= 2;
942: }
943: memcpy(*buf + *off, value, valuelen);
944: *off += valuelen;
945:
946: free(value);
947: free(copy0);
948: return (0);
949:
950: fail:
951: free(copy0);
952: return (-1);
953: }
954:
955: /* Expand keys in a template, passing through strftime first. */
956: char *
957: format_expand_time(struct format_tree *ft, const char *fmt, time_t t)
958: {
959: struct tm *tm;
960: char s[2048];
961:
962: if (fmt == NULL || *fmt == '\0')
963: return (xstrdup(""));
964:
965: tm = localtime(&t);
966:
967: if (strftime(s, sizeof s, fmt, tm) == 0)
968: return (xstrdup(""));
969:
970: return (format_expand(ft, s));
971: }
972:
973: /* Expand keys in a template. */
974: char *
975: format_expand(struct format_tree *ft, const char *fmt)
976: {
977: char *buf, *out;
978: const char *ptr, *s, *saved = fmt;
979: size_t off, len, n, outlen;
980: int ch, brackets;
981:
982: if (fmt == NULL)
983: return (xstrdup(""));
984:
985: len = 64;
986: buf = xmalloc(len);
987: off = 0;
988:
989: while (*fmt != '\0') {
990: if (*fmt != '#') {
991: while (len - off < 2) {
992: buf = xreallocarray(buf, 2, len);
993: len *= 2;
994: }
995: buf[off++] = *fmt++;
996: continue;
997: }
998: fmt++;
999:
1000: ch = (u_char) *fmt++;
1001: switch (ch) {
1002: case '(':
1003: brackets = 1;
1004: for (ptr = fmt; *ptr != '\0'; ptr++) {
1005: if (*ptr == '(')
1006: brackets++;
1007: if (*ptr == ')' && --brackets == 0)
1008: break;
1009: }
1010: if (*ptr != ')' || brackets != 0)
1011: break;
1012: n = ptr - fmt;
1013:
1014: if (ft->flags & FORMAT_NOJOBS)
1015: out = xstrdup("");
1016: else
1017: out = format_job_get(ft, xstrndup(fmt, n));
1018: outlen = strlen(out);
1019:
1020: while (len - off < outlen + 1) {
1021: buf = xreallocarray(buf, 2, len);
1022: len *= 2;
1023: }
1024: memcpy(buf + off, out, outlen);
1025: off += outlen;
1026:
1027: free(out);
1028:
1029: fmt += n + 1;
1030: continue;
1031: case '{':
1032: brackets = 1;
1033: for (ptr = fmt; *ptr != '\0'; ptr++) {
1034: if (*ptr == '{')
1035: brackets++;
1036: if (*ptr == '}' && --brackets == 0)
1037: break;
1038: }
1039: if (*ptr != '}' || brackets != 0)
1040: break;
1041: n = ptr - fmt;
1042:
1043: if (format_replace(ft, fmt, n, &buf, &len, &off) != 0)
1044: break;
1045: fmt += n + 1;
1046: continue;
1047: case '#':
1048: while (len - off < 2) {
1049: buf = xreallocarray(buf, 2, len);
1050: len *= 2;
1051: }
1052: buf[off++] = '#';
1053: continue;
1054: default:
1055: s = NULL;
1056: if (ch >= 'A' && ch <= 'Z')
1057: s = format_upper[ch - 'A'];
1058: else if (ch >= 'a' && ch <= 'z')
1059: s = format_lower[ch - 'a'];
1060: if (s == NULL) {
1061: while (len - off < 3) {
1062: buf = xreallocarray(buf, 2, len);
1063: len *= 2;
1064: }
1065: buf[off++] = '#';
1066: buf[off++] = ch;
1067: continue;
1068: }
1069: n = strlen(s);
1070: if (format_replace(ft, s, n, &buf, &len, &off) != 0)
1071: break;
1072: continue;
1073: }
1074:
1075: break;
1076: }
1077: buf[off] = '\0';
1078:
1079: log_debug("format '%s' -> '%s'", saved, buf);
1080: return (buf);
1081: }
1082:
1083: /* Expand a single string. */
1084: char *
1085: format_single(struct cmdq_item *item, const char *fmt, struct client *c,
1086: struct session *s, struct winlink *wl, struct window_pane *wp)
1087: {
1088: struct format_tree *ft;
1089: char *expanded;
1090:
1091: ft = format_create(item, FORMAT_NONE, 0);
1092: format_defaults(ft, c, s, wl, wp);
1093:
1094: expanded = format_expand(ft, fmt);
1095: format_free(ft);
1096: return (expanded);
1097: }
1098:
1099: /* Set defaults for any of arguments that are not NULL. */
1100: void
1101: format_defaults(struct format_tree *ft, struct client *c, struct session *s,
1102: struct winlink *wl, struct window_pane *wp)
1103: {
1104: if (s == NULL && c != NULL)
1105: s = c->session;
1106: if (wl == NULL && s != NULL)
1107: wl = s->curw;
1108: if (wp == NULL && wl != NULL)
1109: wp = wl->window->active;
1110:
1111: if (c != NULL)
1112: format_defaults_client(ft, c);
1113: if (s != NULL)
1114: format_defaults_session(ft, s);
1115: if (s != NULL && wl != NULL)
1116: format_defaults_winlink(ft, s, wl);
1117: if (wp != NULL)
1118: format_defaults_pane(ft, wp);
1119: }
1120:
1121: /* Set default format keys for a session. */
1122: static void
1123: format_defaults_session(struct format_tree *ft, struct session *s)
1124: {
1125: struct session_group *sg;
1126:
1127: ft->s = s;
1128:
1129: format_add(ft, "session_name", "%s", s->name);
1130: format_add(ft, "session_windows", "%u", winlink_count(&s->windows));
1131: format_add(ft, "session_width", "%u", s->sx);
1132: format_add(ft, "session_height", "%u", s->sy);
1133: format_add(ft, "session_id", "$%u", s->id);
1134:
1135: sg = session_group_contains(s);
1136: format_add(ft, "session_grouped", "%d", sg != NULL);
1137: if (sg != NULL)
1138: format_add(ft, "session_group", "%s", sg->name);
1139:
1140: format_add_tv(ft, "session_created", &s->creation_time);
1141: format_add_tv(ft, "session_last_attached", &s->last_attached_time);
1142: format_add_tv(ft, "session_activity", &s->activity_time);
1143:
1144: format_add(ft, "session_attached", "%u", s->attached);
1145: format_add(ft, "session_many_attached", "%d", s->attached > 1);
1146:
1147: format_add_cb(ft, "session_alerts", format_cb_session_alerts);
1148: }
1149:
1150: /* Set default format keys for a client. */
1151: static void
1152: format_defaults_client(struct format_tree *ft, struct client *c)
1153: {
1154: struct session *s;
1155: const char *name;
1156: struct tty *tty = &c->tty;
1157: const char *types[] = TTY_TYPES;
1158:
1159: if (ft->s == NULL)
1160: ft->s = c->session;
1161:
1162: format_add(ft, "client_name", "%s", c->name);
1163: format_add(ft, "client_pid", "%ld", (long) c->pid);
1164: format_add(ft, "client_height", "%u", tty->sy);
1165: format_add(ft, "client_width", "%u", tty->sx);
1166: format_add(ft, "client_tty", "%s", c->ttyname);
1167: format_add(ft, "client_control_mode", "%d",
1168: !!(c->flags & CLIENT_CONTROL));
1169:
1170: if (tty->term_name != NULL)
1171: format_add(ft, "client_termname", "%s", tty->term_name);
1172: if (tty->term_name != NULL)
1173: format_add(ft, "client_termtype", "%s", types[tty->term_type]);
1174:
1175: format_add_tv(ft, "client_created", &c->creation_time);
1176: format_add_tv(ft, "client_activity", &c->activity_time);
1177: format_add(ft, "client_written", "%zu", tty->written);
1178:
1179: name = server_client_get_key_table(c);
1180: if (strcmp(c->keytable->name, name) == 0)
1181: format_add(ft, "client_prefix", "%d", 0);
1182: else
1183: format_add(ft, "client_prefix", "%d", 1);
1184: format_add(ft, "client_key_table", "%s", c->keytable->name);
1185:
1186: if (tty->flags & TTY_UTF8)
1187: format_add(ft, "client_utf8", "%d", 1);
1188: else
1189: format_add(ft, "client_utf8", "%d", 0);
1190:
1191: if (c->flags & CLIENT_READONLY)
1192: format_add(ft, "client_readonly", "%d", 1);
1193: else
1194: format_add(ft, "client_readonly", "%d", 0);
1195:
1196: s = c->session;
1197: if (s != NULL)
1198: format_add(ft, "client_session", "%s", s->name);
1199: s = c->last_session;
1200: if (s != NULL && session_alive(s))
1201: format_add(ft, "client_last_session", "%s", s->name);
1202: }
1203:
1204: /* Set default format keys for a window. */
1205: void
1206: format_defaults_window(struct format_tree *ft, struct window *w)
1207: {
1208: ft->w = w;
1209:
1210: format_add_tv(ft, "window_activity", &w->activity_time);
1211: format_add(ft, "window_id", "@%u", w->id);
1212: format_add(ft, "window_name", "%s", w->name);
1213: format_add(ft, "window_width", "%u", w->sx);
1214: format_add(ft, "window_height", "%u", w->sy);
1215: format_add_cb(ft, "window_layout", format_cb_window_layout);
1216: format_add_cb(ft, "window_visible_layout",
1217: format_cb_window_visible_layout);
1218: format_add(ft, "window_panes", "%u", window_count_panes(w));
1219: format_add(ft, "window_zoomed_flag", "%d",
1220: !!(w->flags & WINDOW_ZOOMED));
1221: }
1222:
1223: /* Set default format keys for a winlink. */
1224: static void
1225: format_defaults_winlink(struct format_tree *ft, struct session *s,
1226: struct winlink *wl)
1227: {
1228: struct window *w = wl->window;
1229: char *flags;
1230:
1231: if (ft->w == NULL)
1232: ft->w = wl->window;
1233:
1234: flags = window_printable_flags(s, wl);
1235:
1236: format_defaults_window(ft, w);
1237:
1238: format_add(ft, "window_index", "%d", wl->idx);
1239: format_add(ft, "window_flags", "%s", flags);
1240: format_add(ft, "window_active", "%d", wl == s->curw);
1241:
1242: format_add(ft, "window_bell_flag", "%d",
1243: !!(wl->flags & WINLINK_BELL));
1244: format_add(ft, "window_activity_flag", "%d",
1245: !!(wl->flags & WINLINK_ACTIVITY));
1246: format_add(ft, "window_silence_flag", "%d",
1247: !!(wl->flags & WINLINK_SILENCE));
1248: format_add(ft, "window_last_flag", "%d",
1249: !!(wl == TAILQ_FIRST(&s->lastw)));
1250: format_add(ft, "window_linked", "%d", session_is_linked(s, wl->window));
1251:
1252: free(flags);
1253: }
1254:
1255: /* Set default format keys for a window pane. */
1256: void
1257: format_defaults_pane(struct format_tree *ft, struct window_pane *wp)
1258: {
1259: struct grid *gd = wp->base.grid;
1260: u_int idx;
1261: int status, scroll_position;
1262:
1263: if (ft->w == NULL)
1264: ft->w = wp->window;
1265: ft->wp = wp;
1266:
1267: format_add(ft, "history_size", "%u", gd->hsize);
1268: format_add(ft, "history_limit", "%u", gd->hlimit);
1269: format_add_cb(ft, "history_bytes", format_cb_history_bytes);
1270:
1271: if (window_pane_index(wp, &idx) != 0)
1272: fatalx("index not found");
1273: format_add(ft, "pane_index", "%u", idx);
1274:
1275: format_add(ft, "pane_width", "%u", wp->sx);
1276: format_add(ft, "pane_height", "%u", wp->sy);
1277: format_add(ft, "pane_title", "%s", wp->base.title);
1278: format_add(ft, "pane_id", "%%%u", wp->id);
1279: format_add(ft, "pane_active", "%d", wp == wp->window->active);
1280: format_add(ft, "pane_input_off", "%d", !!(wp->flags & PANE_INPUTOFF));
1281:
1282: status = wp->status;
1283: if (wp->fd == -1 && WIFEXITED(status))
1284: format_add(ft, "pane_dead_status", "%d", WEXITSTATUS(status));
1285: format_add(ft, "pane_dead", "%d", wp->fd == -1);
1286:
1287: if (window_pane_visible(wp)) {
1288: format_add(ft, "pane_left", "%u", wp->xoff);
1289: format_add(ft, "pane_top", "%u", wp->yoff);
1290: format_add(ft, "pane_right", "%u", wp->xoff + wp->sx - 1);
1291: format_add(ft, "pane_bottom", "%u", wp->yoff + wp->sy - 1);
1292: }
1293:
1294: format_add(ft, "pane_in_mode", "%d", wp->screen != &wp->base);
1295: format_add(ft, "pane_synchronized", "%d",
1296: !!options_get_number(wp->window->options, "synchronize-panes"));
1297:
1298: format_add(ft, "pane_tty", "%s", wp->tty);
1299: format_add(ft, "pane_pid", "%ld", (long) wp->pid);
1300: format_add_cb(ft, "pane_start_command", format_cb_start_command);
1301: format_add_cb(ft, "pane_current_command", format_cb_current_command);
1302: format_add_cb(ft, "pane_current_path", format_cb_current_path);
1303:
1304: format_add(ft, "cursor_x", "%u", wp->base.cx);
1305: format_add(ft, "cursor_y", "%u", wp->base.cy);
1306: format_add(ft, "scroll_region_upper", "%u", wp->base.rupper);
1307: format_add(ft, "scroll_region_lower", "%u", wp->base.rlower);
1308:
1309: scroll_position = window_copy_scroll_position(wp);
1310: if (scroll_position != -1)
1311: format_add(ft, "scroll_position", "%d", scroll_position);
1312:
1313: format_add(ft, "alternate_on", "%d", wp->saved_grid ? 1 : 0);
1314: format_add(ft, "alternate_saved_x", "%u", wp->saved_cx);
1315: format_add(ft, "alternate_saved_y", "%u", wp->saved_cy);
1316:
1317: format_add(ft, "cursor_flag", "%d",
1318: !!(wp->base.mode & MODE_CURSOR));
1319: format_add(ft, "insert_flag", "%d",
1320: !!(wp->base.mode & MODE_INSERT));
1321: format_add(ft, "keypad_cursor_flag", "%d",
1322: !!(wp->base.mode & MODE_KCURSOR));
1323: format_add(ft, "keypad_flag", "%d",
1324: !!(wp->base.mode & MODE_KKEYPAD));
1325: format_add(ft, "wrap_flag", "%d",
1326: !!(wp->base.mode & MODE_WRAP));
1327:
1328: format_add(ft, "mouse_any_flag", "%d",
1329: !!(wp->base.mode & ALL_MOUSE_MODES));
1330: format_add(ft, "mouse_standard_flag", "%d",
1331: !!(wp->base.mode & MODE_MOUSE_STANDARD));
1332: format_add(ft, "mouse_button_flag", "%d",
1333: !!(wp->base.mode & MODE_MOUSE_BUTTON));
1334: format_add(ft, "mouse_all_flag", "%d",
1335: !!(wp->base.mode & MODE_MOUSE_ALL));
1336:
1337: format_add_cb(ft, "pane_tabs", format_cb_pane_tabs);
1338: }
1339:
1340: /* Set default format keys for paste buffer. */
1341: void
1342: format_defaults_paste_buffer(struct format_tree *ft, struct paste_buffer *pb)
1343: {
1344: size_t bufsize;
1345: char *s;
1346:
1347: paste_buffer_data(pb, &bufsize);
1348: format_add(ft, "buffer_size", "%zu", bufsize);
1349: format_add(ft, "buffer_name", "%s", paste_buffer_name(pb));
1350:
1351: s = paste_make_sample(pb);
1352: format_add(ft, "buffer_sample", "%s", s);
1353: free(s);
1354: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to format.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