Annotation of src/usr.bin/tmux/tmux.c, Revision 1.38
1.38 ! nicm 1: /* $OpenBSD: tmux.c,v 1.37 2009/09/01 09:00:54 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/stat.h>
21:
22: #include <errno.h>
23: #include <paths.h>
24: #include <pwd.h>
25: #include <signal.h>
26: #include <stdlib.h>
27: #include <string.h>
28: #include <syslog.h>
29: #include <unistd.h>
30:
31: #include "tmux.h"
32:
33: #ifdef DEBUG
34: const char *malloc_options = "AFGJPX";
35: #endif
36:
37: volatile sig_atomic_t sigwinch;
38: volatile sig_atomic_t sigterm;
39: volatile sig_atomic_t sigcont;
40: volatile sig_atomic_t sigchld;
41: volatile sig_atomic_t sigusr1;
42: volatile sig_atomic_t sigusr2;
43:
44: char *cfg_file;
1.12 nicm 45: struct options global_s_options; /* session options */
46: struct options global_w_options; /* window options */
1.29 nicm 47: struct environ global_environ;
1.1 nicm 48:
49: int server_locked;
1.19 nicm 50: u_int password_failures;
1.1 nicm 51: char *server_password;
52: time_t server_activity;
53:
54: int debug_level;
55: int be_quiet;
56: time_t start_time;
57: char *socket_path;
58:
59: __dead void usage(void);
60: char *makesockpath(const char *);
1.23 nicm 61: int prepare_unlock(enum msgtype *, void **, size_t *, int);
62: int prepare_cmd(enum msgtype *, void **, size_t *, int, char **);
1.31 nicm 63: int dispatch_imsg(struct client_ctx *, int *);
1.1 nicm 64:
65: __dead void
66: usage(void)
67: {
1.4 sobrado 68: fprintf(stderr,
69: "usage: %s [-28dqUuv] [-f file] [-L socket-name] [-S socket-path]\n"
70: " [command [flags]]\n",
1.1 nicm 71: __progname);
72: exit(1);
73: }
74:
75: void
76: logfile(const char *name)
77: {
78: char *path;
79:
80: log_close();
81: if (debug_level > 0) {
1.32 nicm 82: xasprintf(&path, "tmux-%s-%ld.log", name, (long) getpid());
1.1 nicm 83: log_open_file(debug_level, path);
84: xfree(path);
85: }
86: }
87:
88: void
89: sighandler(int sig)
90: {
91: int saved_errno;
92:
93: saved_errno = errno;
94: switch (sig) {
95: case SIGWINCH:
96: sigwinch = 1;
97: break;
98: case SIGTERM:
99: sigterm = 1;
100: break;
101: case SIGCHLD:
102: sigchld = 1;
103: break;
104: case SIGCONT:
105: sigcont = 1;
106: break;
107: case SIGUSR1:
108: sigusr1 = 1;
109: break;
110: case SIGUSR2:
111: sigusr2 = 1;
112: break;
113: }
114: errno = saved_errno;
115: }
116:
117: void
118: siginit(void)
119: {
120: struct sigaction act;
121:
122: memset(&act, 0, sizeof act);
123: sigemptyset(&act.sa_mask);
124: act.sa_flags = SA_RESTART;
125:
126: act.sa_handler = SIG_IGN;
127: if (sigaction(SIGPIPE, &act, NULL) != 0)
128: fatal("sigaction failed");
129: if (sigaction(SIGINT, &act, NULL) != 0)
130: fatal("sigaction failed");
131: if (sigaction(SIGTSTP, &act, NULL) != 0)
132: fatal("sigaction failed");
133: if (sigaction(SIGQUIT, &act, NULL) != 0)
134: fatal("sigaction failed");
135:
136: act.sa_handler = sighandler;
137: if (sigaction(SIGWINCH, &act, NULL) != 0)
138: fatal("sigaction failed");
139: if (sigaction(SIGTERM, &act, NULL) != 0)
140: fatal("sigaction failed");
141: if (sigaction(SIGCHLD, &act, NULL) != 0)
142: fatal("sigaction failed");
143: if (sigaction(SIGUSR1, &act, NULL) != 0)
144: fatal("sigaction failed");
145: if (sigaction(SIGUSR2, &act, NULL) != 0)
146: fatal("sigaction failed");
147: }
148:
149: void
150: sigreset(void)
151: {
152: struct sigaction act;
153:
154: memset(&act, 0, sizeof act);
155: sigemptyset(&act.sa_mask);
156:
157: act.sa_handler = SIG_DFL;
158: if (sigaction(SIGPIPE, &act, NULL) != 0)
159: fatal("sigaction failed");
160: if (sigaction(SIGUSR1, &act, NULL) != 0)
161: fatal("sigaction failed");
162: if (sigaction(SIGUSR2, &act, NULL) != 0)
163: fatal("sigaction failed");
164: if (sigaction(SIGINT, &act, NULL) != 0)
165: fatal("sigaction failed");
166: if (sigaction(SIGTSTP, &act, NULL) != 0)
167: fatal("sigaction failed");
168: if (sigaction(SIGQUIT, &act, NULL) != 0)
169: fatal("sigaction failed");
170: if (sigaction(SIGWINCH, &act, NULL) != 0)
171: fatal("sigaction failed");
172: if (sigaction(SIGTERM, &act, NULL) != 0)
173: fatal("sigaction failed");
174: if (sigaction(SIGCHLD, &act, NULL) != 0)
175: fatal("sigaction failed");
176: }
177:
178: char *
179: makesockpath(const char *label)
180: {
181: char base[MAXPATHLEN], *path;
182: struct stat sb;
183: u_int uid;
184:
185: uid = getuid();
1.32 nicm 186: xsnprintf(base, MAXPATHLEN, "%s/tmux-%d", _PATH_TMP, uid);
1.1 nicm 187:
188: if (mkdir(base, S_IRWXU) != 0 && errno != EEXIST)
189: return (NULL);
190:
191: if (lstat(base, &sb) != 0)
192: return (NULL);
193: if (!S_ISDIR(sb.st_mode)) {
194: errno = ENOTDIR;
195: return (NULL);
196: }
197: if (sb.st_uid != uid || (sb.st_mode & (S_IRWXG|S_IRWXO)) != 0) {
198: errno = EACCES;
199: return (NULL);
200: }
201:
202: xasprintf(&path, "%s/%s", base, label);
203: return (path);
204: }
205:
206: int
1.23 nicm 207: prepare_unlock(enum msgtype *msg, void **buf, size_t *len, int argc)
1.21 nicm 208: {
209: static struct msg_unlock_data unlockdata;
210: char *pass;
211:
212: if (argc != 0) {
213: log_warnx("can't specify a command when unlocking");
214: return (-1);
215: }
216:
1.38 ! nicm 217: if ((pass = getpass("Password:")) == NULL)
1.21 nicm 218: return (-1);
219:
220: if (strlen(pass) >= sizeof unlockdata.pass) {
221: log_warnx("password too long");
222: return (-1);
223: }
224:
225: strlcpy(unlockdata.pass, pass, sizeof unlockdata.pass);
226: memset(pass, 0, strlen(pass));
227:
228: *buf = &unlockdata;
229: *len = sizeof unlockdata;
230:
231: *msg = MSG_UNLOCK;
232: return (0);
233: }
234:
235: int
1.23 nicm 236: prepare_cmd(enum msgtype *msg, void **buf, size_t *len, int argc, char **argv)
1.21 nicm 237: {
238: static struct msg_command_data cmddata;
239:
240: client_fill_session(&cmddata);
241:
242: cmddata.argc = argc;
243: if (cmd_pack_argv(argc, argv, cmddata.argv, sizeof cmddata.argv) != 0) {
244: log_warnx("command too long");
245: return (-1);
246: }
247:
248: *buf = &cmddata;
249: *len = sizeof cmddata;
250:
251: *msg = MSG_COMMAND;
252: return (0);
253: }
254:
255: int
1.1 nicm 256: main(int argc, char **argv)
257: {
258: struct client_ctx cctx;
259: struct cmd_list *cmdlist;
260: struct cmd *cmd;
261: struct pollfd pfd;
1.23 nicm 262: enum msgtype msg;
1.1 nicm 263: struct passwd *pw;
1.29 nicm 264: char *s, *path, *label, *home, *cause, **var;
1.1 nicm 265: char cwd[MAXPATHLEN];
1.21 nicm 266: void *buf;
267: size_t len;
1.20 nicm 268: int retcode, opt, flags, unlock, cmdflags = 0;
1.31 nicm 269: int nfds;
1.1 nicm 270:
271: unlock = flags = 0;
272: label = path = NULL;
273: while ((opt = getopt(argc, argv, "28df:L:qS:uUv")) != -1) {
274: switch (opt) {
275: case '2':
276: flags |= IDENTIFY_256COLOURS;
277: flags &= ~IDENTIFY_88COLOURS;
278: break;
279: case '8':
280: flags |= IDENTIFY_88COLOURS;
281: flags &= ~IDENTIFY_256COLOURS;
282: break;
1.37 nicm 283: case 'd':
284: flags |= IDENTIFY_HASDEFAULTS;
285: break;
1.1 nicm 286: case 'f':
1.2 ray 287: if (cfg_file)
288: xfree(cfg_file);
1.1 nicm 289: cfg_file = xstrdup(optarg);
290: break;
291: case 'L':
292: if (label != NULL)
293: xfree(label);
294: label = xstrdup(optarg);
295: break;
1.37 nicm 296: case 'q':
297: be_quiet = 1;
298: break;
1.1 nicm 299: case 'S':
300: if (path != NULL)
301: xfree(path);
302: path = xstrdup(optarg);
303: break;
304: case 'u':
305: flags |= IDENTIFY_UTF8;
306: break;
307: case 'U':
308: unlock = 1;
309: break;
310: case 'v':
311: debug_level++;
312: break;
313: default:
314: usage();
315: }
316: }
317: argc -= optind;
318: argv += optind;
319:
320: log_open_tty(debug_level);
321: siginit();
322:
1.29 nicm 323: environ_init(&global_environ);
324: for (var = environ; *var != NULL; var++)
325: environ_put(&global_environ, *var);
326:
1.15 nicm 327: if (!(flags & IDENTIFY_UTF8)) {
328: /*
329: * If the user has set whichever of LC_ALL, LC_CTYPE or LANG
330: * exist (in that order) to contain UTF-8, it is a safe
331: * assumption that either they are using a UTF-8 terminal, or
332: * if not they know that output from UTF-8-capable programs may
333: * be wrong.
334: */
335: if ((s = getenv("LC_ALL")) == NULL) {
336: if ((s = getenv("LC_CTYPE")) == NULL)
337: s = getenv("LANG");
338: }
1.26 nicm 339: if (s != NULL && (strcasestr(s, "UTF-8") != NULL ||
340: strcasestr(s, "UTF8") != NULL))
1.15 nicm 341: flags |= IDENTIFY_UTF8;
342: }
343:
1.12 nicm 344: options_init(&global_s_options, NULL);
1.33 nicm 345: options_set_number(&global_s_options, "base-index", 0);
1.12 nicm 346: options_set_number(&global_s_options, "bell-action", BELL_ANY);
347: options_set_number(&global_s_options, "buffer-limit", 9);
1.13 nicm 348: options_set_string(&global_s_options, "default-command", "%s", "");
1.14 nicm 349: options_set_string(&global_s_options, "default-terminal", "screen");
1.36 nicm 350: options_set_number(&global_s_options, "display-panes-colour", 4);
351: options_set_number(&global_s_options, "display-panes-time", 1000);
1.12 nicm 352: options_set_number(&global_s_options, "display-time", 750);
353: options_set_number(&global_s_options, "history-limit", 2000);
354: options_set_number(&global_s_options, "lock-after-time", 0);
1.27 nicm 355: options_set_number(&global_s_options, "message-attr", 0);
1.12 nicm 356: options_set_number(&global_s_options, "message-bg", 3);
357: options_set_number(&global_s_options, "message-fg", 0);
358: options_set_number(&global_s_options, "prefix", '\002');
359: options_set_number(&global_s_options, "repeat-time", 500);
360: options_set_number(&global_s_options, "set-remain-on-exit", 0);
361: options_set_number(&global_s_options, "set-titles", 0);
362: options_set_number(&global_s_options, "status", 1);
1.27 nicm 363: options_set_number(&global_s_options, "status-attr", 0);
1.12 nicm 364: options_set_number(&global_s_options, "status-bg", 2);
365: options_set_number(&global_s_options, "status-fg", 0);
366: options_set_number(&global_s_options, "status-interval", 15);
367: options_set_number(&global_s_options, "status-keys", MODEKEY_EMACS);
1.18 nicm 368: options_set_number(&global_s_options, "status-justify", 0);
1.28 nicm 369: options_set_string(&global_s_options, "status-left", "[#S]");
370: options_set_number(&global_s_options, "status-left-attr", 0);
371: options_set_number(&global_s_options, "status-left-fg", 8);
372: options_set_number(&global_s_options, "status-left-bg", 8);
1.12 nicm 373: options_set_number(&global_s_options, "status-left-length", 10);
1.1 nicm 374: options_set_string(
1.22 nicm 375: &global_s_options, "status-right", "\"#22T\" %%H:%%M %%d-%%b-%%y");
1.28 nicm 376: options_set_number(&global_s_options, "status-right-attr", 0);
377: options_set_number(&global_s_options, "status-right-fg", 8);
378: options_set_number(&global_s_options, "status-right-bg", 8);
379: options_set_number(&global_s_options, "status-right-length", 40);
1.15 nicm 380: if (flags & IDENTIFY_UTF8)
381: options_set_number(&global_s_options, "status-utf8", 1);
382: else
383: options_set_number(&global_s_options, "status-utf8", 0);
1.25 nicm 384: options_set_string(&global_s_options,
385: "terminal-overrides", "*88col*:colors=88,*256col*:colors=256");
1.35 nicm 386: options_set_string(&global_s_options, "update-environment", "DISPLAY "
387: "WINDOWID SSH_ASKPASS SSH_AUTH_SOCK SSH_AGENT_PID SSH_CONNECTION");
1.16 nicm 388: options_set_number(&global_s_options, "visual-activity", 0);
389: options_set_number(&global_s_options, "visual-bell", 0);
390: options_set_number(&global_s_options, "visual-content", 0);
1.1 nicm 391:
1.12 nicm 392: options_init(&global_w_options, NULL);
393: options_set_number(&global_w_options, "aggressive-resize", 0);
394: options_set_number(&global_w_options, "automatic-rename", 1);
395: options_set_number(&global_w_options, "clock-mode-colour", 4);
396: options_set_number(&global_w_options, "clock-mode-style", 1);
397: options_set_number(&global_w_options, "force-height", 0);
398: options_set_number(&global_w_options, "force-width", 0);
399: options_set_number(&global_w_options, "main-pane-width", 81);
400: options_set_number(&global_w_options, "main-pane-height", 24);
1.27 nicm 401: options_set_number(&global_w_options, "mode-attr", 0);
1.12 nicm 402: options_set_number(&global_w_options, "mode-bg", 3);
403: options_set_number(&global_w_options, "mode-fg", 0);
404: options_set_number(&global_w_options, "mode-keys", MODEKEY_EMACS);
1.34 nicm 405: options_set_number(&global_w_options, "mode-mouse", 0);
1.12 nicm 406: options_set_number(&global_w_options, "monitor-activity", 0);
407: options_set_string(&global_w_options, "monitor-content", "%s", "");
1.15 nicm 408: if (flags & IDENTIFY_UTF8)
409: options_set_number(&global_w_options, "utf8", 1);
410: else
411: options_set_number(&global_w_options, "utf8", 0);
1.12 nicm 412: options_set_number(&global_w_options, "window-status-attr", 0);
413: options_set_number(&global_w_options, "window-status-bg", 8);
414: options_set_number(&global_w_options, "window-status-fg", 8);
1.17 nicm 415: options_set_number(&global_w_options, "window-status-current-attr", 0);
416: options_set_number(&global_w_options, "window-status-current-bg", 8);
417: options_set_number(&global_w_options, "window-status-current-fg", 8);
1.12 nicm 418: options_set_number(&global_w_options, "xterm-keys", 0);
419: options_set_number(&global_w_options, "remain-on-exit", 0);
1.1 nicm 420:
421: if (cfg_file == NULL) {
422: home = getenv("HOME");
423: if (home == NULL || *home == '\0') {
424: pw = getpwuid(getuid());
425: if (pw != NULL)
426: home = pw->pw_dir;
427: }
428: xasprintf(&cfg_file, "%s/%s", home, DEFAULT_CFG);
429: if (access(cfg_file, R_OK) != 0) {
430: xfree(cfg_file);
431: cfg_file = NULL;
432: }
433: } else {
434: if (access(cfg_file, R_OK) != 0) {
435: log_warn("%s", cfg_file);
436: exit(1);
437: }
438: }
1.21 nicm 439:
1.1 nicm 440: if (label == NULL)
441: label = xstrdup("default");
442: if (path == NULL && (path = makesockpath(label)) == NULL) {
443: log_warn("can't create socket");
444: exit(1);
445: }
446: xfree(label);
447:
448: if (getcwd(cwd, sizeof cwd) == NULL) {
1.11 nicm 449: pw = getpwuid(getuid());
450: if (pw->pw_dir != NULL && *pw->pw_dir != '\0')
451: strlcpy(cwd, pw->pw_dir, sizeof cwd);
452: else
453: strlcpy(cwd, "/", sizeof cwd);
1.1 nicm 454: }
1.12 nicm 455: options_set_string(&global_s_options, "default-path", "%s", cwd);
1.1 nicm 456:
457: if (unlock) {
1.21 nicm 458: if (prepare_unlock(&msg, &buf, &len, argc) != 0)
1.1 nicm 459: exit(1);
1.21 nicm 460: } else {
461: if (prepare_cmd(&msg, &buf, &len, argc, argv) != 0)
1.1 nicm 462: exit(1);
1.21 nicm 463: }
464:
465: if (unlock)
1.20 nicm 466: cmdflags &= ~CMD_STARTSERVER;
1.21 nicm 467: else if (argc == 0)
1.30 nicm 468: cmdflags |= CMD_STARTSERVER|CMD_SENDENVIRON;
1.21 nicm 469: else {
470: /*
471: * It sucks parsing the command string twice (in client and
472: * later in server) but it is necessary to get the start server
473: * flag.
474: */
475: if ((cmdlist = cmd_list_parse(argc, argv, &cause)) == NULL) {
476: log_warnx("%s", cause);
477: exit(1);
1.1 nicm 478: }
1.21 nicm 479: cmdflags &= ~CMD_STARTSERVER;
1.1 nicm 480: TAILQ_FOREACH(cmd, cmdlist, qentry) {
1.29 nicm 481: if (cmd->entry->flags & CMD_STARTSERVER)
1.20 nicm 482: cmdflags |= CMD_STARTSERVER;
1.29 nicm 483: if (cmd->entry->flags & CMD_SENDENVIRON)
484: cmdflags |= CMD_SENDENVIRON;
1.1 nicm 485: }
1.21 nicm 486: cmd_list_free(cmdlist);
1.1 nicm 487: }
488:
489: memset(&cctx, 0, sizeof cctx);
1.20 nicm 490: if (client_init(path, &cctx, cmdflags, flags) != 0)
1.1 nicm 491: exit(1);
492: xfree(path);
493:
1.21 nicm 494: client_write_server(&cctx, msg, buf, len);
495: memset(buf, 0, len);
1.1 nicm 496:
497: retcode = 0;
498: for (;;) {
1.31 nicm 499: pfd.fd = cctx.ibuf.fd;
1.1 nicm 500: pfd.events = POLLIN;
1.31 nicm 501: if (cctx.ibuf.w.queued != 0)
1.1 nicm 502: pfd.events |= POLLOUT;
503:
1.31 nicm 504: if ((nfds = poll(&pfd, 1, INFTIM)) == -1) {
1.1 nicm 505: if (errno == EAGAIN || errno == EINTR)
506: continue;
507: fatal("poll failed");
508: }
1.31 nicm 509: if (nfds == 0)
510: continue;
511:
512: if (pfd.revents & (POLLERR|POLLHUP|POLLNVAL))
513: fatalx("socket error");
514:
515: if (pfd.revents & POLLIN) {
516: if (dispatch_imsg(&cctx, &retcode) != 0)
517: break;
518: }
519:
520: if (pfd.revents & POLLOUT) {
521: if (msgbuf_write(&cctx.ibuf.w) < 0)
522: fatalx("msgbuf_write failed");
523: }
524: }
525:
526: options_free(&global_s_options);
527: options_free(&global_w_options);
528:
529: return (retcode);
530: }
531:
532: int
533: dispatch_imsg(struct client_ctx *cctx, int *retcode)
534: {
535: struct imsg imsg;
536: ssize_t n, datalen;
537: struct msg_print_data printdata;
1.1 nicm 538:
1.31 nicm 539: if ((n = imsg_read(&cctx->ibuf)) == -1 || n == 0)
540: fatalx("imsg_read failed");
1.1 nicm 541:
1.31 nicm 542: for (;;) {
543: if ((n = imsg_get(&cctx->ibuf, &imsg)) == -1)
544: fatalx("imsg_get failed");
545: if (n == 0)
546: return (0);
547: datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
1.1 nicm 548:
1.31 nicm 549: switch (imsg.hdr.type) {
1.1 nicm 550: case MSG_EXIT:
551: case MSG_SHUTDOWN:
1.31 nicm 552: if (datalen != 0)
553: fatalx("bad MSG_EXIT size");
554:
555: return (-1);
1.1 nicm 556: case MSG_ERROR:
1.31 nicm 557: *retcode = 1;
1.1 nicm 558: /* FALLTHROUGH */
559: case MSG_PRINT:
1.31 nicm 560: if (datalen != sizeof printdata)
1.1 nicm 561: fatalx("bad MSG_PRINT size");
1.31 nicm 562: memcpy(&printdata, imsg.data, sizeof printdata);
563: printdata.msg[(sizeof printdata.msg) - 1] = '\0';
1.21 nicm 564:
565: log_info("%s", printdata.msg);
1.31 nicm 566: break;
1.1 nicm 567: case MSG_READY:
1.31 nicm 568: if (datalen != 0)
569: fatalx("bad MSG_READY size");
570:
571: *retcode = client_main(cctx);
572: return (-1);
573: case MSG_VERSION:
574: if (datalen != 0)
575: fatalx("bad MSG_VERSION size");
576:
577: log_warnx("protocol version mismatch (client %u, "
578: "server %u)", PROTOCOL_VERSION, imsg.hdr.peerid);
579: *retcode = 1;
580: return (-1);
1.1 nicm 581: default:
1.31 nicm 582: fatalx("unexpected message");
1.1 nicm 583: }
1.31 nicm 584:
585: imsg_free(&imsg);
1.1 nicm 586: }
587: }
![[BACK]](/icons/back.gif){kind=icon}
Return to tmux.c CVS log ![[TXT]](/icons/text.gif){kind=icon}
![[DIR]](/icons/dir.gif){kind=icon}
Up to [local] / src / usr.bin / tmux