4 * Copyright (c) 2007 Nicholas Marriott <nicm@users.sourceforge.net>
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.
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.
19 #include <sys/types.h>
36 extern char *malloc_options
;
40 struct options global_options
; /* server options */
41 struct options global_s_options
; /* session options */
42 struct options global_w_options
; /* window options */
43 struct environ global_environ
;
45 struct event_base
*ev_base
;
58 __dead
void usage(void);
59 void parse_env(struct env_data
*);
60 char *makesockpath(const char *);
61 __dead
void shell_exec(const char *, const char *);
63 struct imsgbuf
*main_ibuf
;
66 void main_signal(int, short, unused
void *);
67 void main_callback(int, short, void *);
68 void main_dispatch(const char *);
74 "usage: %s [-28lquv] [-c shell-command] [-f file] [-L socket-name]\n"
75 " [-S socket-path] [command [flags]]\n",
81 logfile(const char *name
)
86 if (debug_level
> 0) {
87 xasprintf(&path
, "tmux-%s-%ld.log", name
, (long) getpid());
88 log_open_file(debug_level
, path
);
99 shell
= getenv("SHELL");
100 if (checkshell(shell
))
103 pw
= getpwuid(getuid());
104 if (pw
!= NULL
&& checkshell(pw
->pw_shell
))
105 return (pw
->pw_shell
);
107 return (_PATH_BSHELL
);
111 checkshell(const char *shell
)
113 if (shell
== NULL
|| *shell
== '\0' || areshell(shell
))
115 if (access(shell
, X_OK
) != 0)
121 areshell(const char *shell
)
123 const char *progname
, *ptr
;
125 if ((ptr
= strrchr(shell
, '/')) != NULL
)
129 progname
= __progname
;
130 if (*progname
== '-')
132 if (strcmp(ptr
, progname
) == 0)
138 parse_env(struct env_data
*data
)
140 char *env
, *path_pid
, *pid_idx
, buf
[256];
146 if ((env
= getenv("TMUX")) == NULL
)
149 if ((path_pid
= strchr(env
, ',')) == NULL
|| path_pid
== env
)
151 if ((pid_idx
= strchr(path_pid
+ 1, ',')) == NULL
)
153 if ((pid_idx
== path_pid
+ 1 || pid_idx
[1] == '\0'))
157 len
= path_pid
- env
;
158 data
->path
= xmalloc (len
+ 1);
159 memcpy(data
->path
, env
, len
);
160 data
->path
[len
] = '\0';
163 len
= pid_idx
- path_pid
- 1;
164 if (len
> (sizeof buf
) - 1)
166 memcpy(buf
, path_pid
+ 1, len
);
169 ll
= strtonum(buf
, 0, LONG_MAX
, &errstr
);
175 ll
= strtonum(pid_idx
+1, 0, UINT_MAX
, &errstr
);
182 makesockpath(const char *label
)
184 char base
[MAXPATHLEN
], *path
;
189 xsnprintf(base
, MAXPATHLEN
, "%s/tmux-%d", _PATH_TMP
, uid
);
191 if (mkdir(base
, S_IRWXU
) != 0 && errno
!= EEXIST
)
194 if (lstat(base
, &sb
) != 0)
196 if (!S_ISDIR(sb
.st_mode
)) {
200 if (sb
.st_uid
!= uid
|| (sb
.st_mode
& (S_IRWXG
|S_IRWXO
)) != 0) {
205 xasprintf(&path
, "%s/%s", base
, label
);
210 shell_exec(const char *shell
, const char *shellcmd
)
212 const char *shellname
, *ptr
;
215 ptr
= strrchr(shell
, '/');
216 if (ptr
!= NULL
&& *(ptr
+ 1) != '\0')
221 xasprintf(&argv0
, "-%s", shellname
);
223 xasprintf(&argv0
, "%s", shellname
);
224 setenv("SHELL", shell
, 1);
226 execl(shell
, argv0
, "-c", shellcmd
, (char *) NULL
);
227 fatal("execl failed");
231 main(int argc
, char **argv
)
233 struct cmd_list
*cmdlist
;
237 struct options
*oo
, *so
, *wo
;
238 struct keylist
*keylist
;
239 struct env_data envdata
;
240 struct msg_command_data cmddata
;
241 char *s
, *shellcmd
, *path
, *label
, *home
, *cause
;
242 char cwd
[MAXPATHLEN
], **var
;
245 int opt
, flags
, quiet
= 0, cmdflags
= 0;
249 malloc_options
= (char *) "AFGJPX";
253 shellcmd
= label
= path
= NULL
;
255 login_shell
= (**argv
== '-');
256 while ((opt
= getopt(argc
, argv
, "28c:df:lL:qS:uUv")) != -1) {
259 flags
|= IDENTIFY_256COLOURS
;
260 flags
&= ~IDENTIFY_88COLOURS
;
263 flags
|= IDENTIFY_88COLOURS
;
264 flags
&= ~IDENTIFY_256COLOURS
;
267 if (shellcmd
!= NULL
)
269 shellcmd
= xstrdup(optarg
);
272 if (cfg_file
!= NULL
)
274 cfg_file
= xstrdup(optarg
);
282 label
= xstrdup(optarg
);
290 path
= xstrdup(optarg
);
293 flags
|= IDENTIFY_UTF8
;
305 if (shellcmd
!= NULL
&& argc
!= 0)
308 log_open_tty(debug_level
);
310 if (!(flags
& IDENTIFY_UTF8
)) {
312 * If the user has set whichever of LC_ALL, LC_CTYPE or LANG
313 * exist (in that order) to contain UTF-8, it is a safe
314 * assumption that either they are using a UTF-8 terminal, or
315 * if not they know that output from UTF-8-capable programs may
318 if ((s
= getenv("LC_ALL")) == NULL
) {
319 if ((s
= getenv("LC_CTYPE")) == NULL
)
322 if (s
!= NULL
&& (strcasestr(s
, "UTF-8") != NULL
||
323 strcasestr(s
, "UTF8") != NULL
))
324 flags
|= IDENTIFY_UTF8
;
327 environ_init(&global_environ
);
328 for (var
= environ
; *var
!= NULL
; var
++)
329 environ_put(&global_environ
, *var
);
331 options_init(&global_options
, NULL
);
332 oo
= &global_options
;
333 options_set_number(oo
, "quiet", quiet
);
334 options_set_number(oo
, "escape-time", 500);
336 options_init(&global_s_options
, NULL
);
337 so
= &global_s_options
;
338 options_set_number(so
, "base-index", 0);
339 options_set_number(so
, "bell-action", BELL_ANY
);
340 options_set_number(so
, "buffer-limit", 9);
341 options_set_string(so
, "default-command", "%s", "");
342 options_set_string(so
, "default-shell", "%s", getshell());
343 options_set_string(so
, "default-terminal", "screen");
344 options_set_number(so
, "display-panes-colour", 4);
345 options_set_number(so
, "display-panes-active-colour", 1);
346 options_set_number(so
, "display-panes-time", 1000);
347 options_set_number(so
, "display-time", 750);
348 options_set_number(so
, "history-limit", 2000);
349 options_set_number(so
, "lock-after-time", 0);
350 options_set_string(so
, "lock-command", "lock -np");
351 options_set_number(so
, "lock-server", 1);
352 options_set_number(so
, "message-attr", 0);
353 options_set_number(so
, "message-bg", 3);
354 options_set_number(so
, "message-fg", 0);
355 options_set_number(so
, "message-limit", 20);
356 options_set_number(so
, "mouse-select-pane", 0);
357 options_set_number(so
, "pane-active-border-bg", 8);
358 options_set_number(so
, "pane-active-border-fg", 2);
359 options_set_number(so
, "pane-border-bg", 8);
360 options_set_number(so
, "pane-border-fg", 8);
361 options_set_number(so
, "repeat-time", 500);
362 options_set_number(so
, "set-remain-on-exit", 0);
363 options_set_number(so
, "set-titles", 0);
364 options_set_string(so
, "set-titles-string", "#S:#I:#W - \"#T\"");
365 options_set_number(so
, "status", 1);
366 options_set_number(so
, "status-attr", 0);
367 options_set_number(so
, "status-bg", 2);
368 options_set_number(so
, "status-fg", 0);
369 options_set_number(so
, "status-interval", 15);
370 options_set_number(so
, "status-justify", 0);
371 options_set_number(so
, "status-keys", MODEKEY_EMACS
);
372 options_set_string(so
, "status-left", "[#S]");
373 options_set_number(so
, "status-left-attr", 0);
374 options_set_number(so
, "status-left-bg", 8);
375 options_set_number(so
, "status-left-fg", 8);
376 options_set_number(so
, "status-left-length", 10);
377 options_set_string(so
, "status-right", "\"#22T\" %%H:%%M %%d-%%b-%%y");
378 options_set_number(so
, "status-right-attr", 0);
379 options_set_number(so
, "status-right-bg", 8);
380 options_set_number(so
, "status-right-fg", 8);
381 options_set_number(so
, "status-right-length", 40);
382 options_set_string(so
, "terminal-overrides",
383 "*88col*:colors=88,*256col*:colors=256");
384 options_set_string(so
, "update-environment", "DISPLAY "
385 "WINDOWID SSH_ASKPASS SSH_AUTH_SOCK SSH_AGENT_PID SSH_CONNECTION");
386 options_set_number(so
, "visual-activity", 0);
387 options_set_number(so
, "visual-bell", 0);
388 options_set_number(so
, "visual-content", 0);
390 keylist
= xmalloc(sizeof *keylist
);
392 ARRAY_ADD(keylist
, '\002');
393 options_set_data(so
, "prefix", keylist
, xfree
);
395 options_init(&global_w_options
, NULL
);
396 wo
= &global_w_options
;
397 options_set_number(wo
, "aggressive-resize", 0);
398 options_set_number(wo
, "alternate-screen", 1);
399 options_set_number(wo
, "automatic-rename", 1);
400 options_set_number(wo
, "clock-mode-colour", 4);
401 options_set_number(wo
, "clock-mode-style", 1);
402 options_set_number(wo
, "force-height", 0);
403 options_set_number(wo
, "force-width", 0);
404 options_set_number(wo
, "main-pane-height", 24);
405 options_set_number(wo
, "main-pane-width", 81);
406 options_set_number(wo
, "mode-attr", 0);
407 options_set_number(wo
, "mode-bg", 3);
408 options_set_number(wo
, "mode-fg", 0);
409 options_set_number(wo
, "mode-keys", MODEKEY_EMACS
);
410 options_set_number(wo
, "mode-mouse", 0);
411 options_set_number(wo
, "monitor-activity", 0);
412 options_set_string(wo
, "monitor-content", "%s", "");
413 options_set_number(wo
, "window-status-attr", 0);
414 options_set_number(wo
, "window-status-bg", 8);
415 options_set_number(wo
, "window-status-current-attr", 0);
416 options_set_number(wo
, "window-status-current-bg", 8);
417 options_set_number(wo
, "window-status-current-fg", 8);
418 options_set_number(wo
, "window-status-fg", 8);
419 options_set_string(wo
, "window-status-format", "#I:#W#F");
420 options_set_string(wo
, "window-status-current-format", "#I:#W#F");
421 options_set_string(wo
, "word-separators", " -_@");
422 options_set_number(wo
, "xterm-keys", 0);
423 options_set_number(wo
, "remain-on-exit", 0);
424 options_set_number(wo
, "synchronize-panes", 0);
426 if (flags
& IDENTIFY_UTF8
) {
427 options_set_number(so
, "status-utf8", 1);
428 options_set_number(wo
, "utf8", 1);
430 options_set_number(so
, "status-utf8", 0);
431 options_set_number(wo
, "utf8", 0);
434 if (getcwd(cwd
, sizeof cwd
) == NULL
) {
435 pw
= getpwuid(getuid());
436 if (pw
->pw_dir
!= NULL
&& *pw
->pw_dir
!= '\0')
437 strlcpy(cwd
, pw
->pw_dir
, sizeof cwd
);
439 strlcpy(cwd
, "/", sizeof cwd
);
441 options_set_string(so
, "default-path", "%s", cwd
);
443 if (cfg_file
== NULL
) {
444 home
= getenv("HOME");
445 if (home
== NULL
|| *home
== '\0') {
446 pw
= getpwuid(getuid());
450 xasprintf(&cfg_file
, "%s/%s", home
, DEFAULT_CFG
);
451 if (access(cfg_file
, R_OK
) != 0 && errno
== ENOENT
) {
458 * Figure out the socket path. If specified on the command-line with
459 * -S or -L, use it, otherwise try $TMUX or assume -L default.
463 /* No -L. Try $TMUX, or default. */
467 label
= xstrdup("default");
470 /* -L or default set. */
472 if ((path
= makesockpath(label
)) == NULL
) {
473 log_warn("can't create socket");
481 if (shellcmd
!= NULL
) {
486 cmddata
.pid
= envdata
.pid
;
487 cmddata
.idx
= envdata
.idx
;
489 /* Prepare command for server. */
492 argc
, argv
, cmddata
.argv
, sizeof cmddata
.argv
) != 0) {
493 log_warnx("command too long");
499 len
= sizeof cmddata
;
502 if (shellcmd
!= NULL
)
503 cmdflags
|= CMD_STARTSERVER
;
504 else if (argc
== 0) /* new-session is the default */
505 cmdflags
|= CMD_STARTSERVER
|CMD_SENDENVIRON
|CMD_CANTNEST
;
508 * It sucks parsing the command string twice (in client and
509 * later in server) but it is necessary to get the start server
512 if ((cmdlist
= cmd_list_parse(argc
, argv
, &cause
)) == NULL
) {
513 log_warnx("%s", cause
);
516 cmdflags
&= ~CMD_STARTSERVER
;
517 TAILQ_FOREACH(cmd
, cmdlist
, qentry
) {
518 if (cmd
->entry
->flags
& CMD_STARTSERVER
)
519 cmdflags
|= CMD_STARTSERVER
;
520 if (cmd
->entry
->flags
& CMD_SENDENVIRON
)
521 cmdflags
|= CMD_SENDENVIRON
;
522 if (cmd
->entry
->flags
& CMD_CANTNEST
)
523 cmdflags
|= CMD_CANTNEST
;
525 cmd_list_free(cmdlist
);
529 * Check if this could be a nested session, if the command can't nest:
530 * if the socket path matches $TMUX, this is probably the same server.
532 if (shellcmd
== NULL
&& envdata
.path
!= NULL
&&
533 cmdflags
& CMD_CANTNEST
&&
534 (path
== envdata
.path
|| strcmp(path
, envdata
.path
) == 0)) {
535 log_warnx("sessions should be nested with care. "
536 "unset $TMUX to force.");
540 ev_base
= event_init();
541 set_signals(main_signal
);
543 /* Initialise the client socket/start the server. */
544 if ((main_ibuf
= client_init(path
, cmdflags
, flags
)) == NULL
)
548 imsg_compose(main_ibuf
, msg
, PROTOCOL_VERSION
, -1, -1, buf
, len
);
551 if (main_ibuf
->w
.queued
> 0)
553 event_once(main_ibuf
->fd
, events
, main_callback
, shellcmd
, NULL
);
560 client_main(); /* doesn't return */
565 main_signal(int sig
, unused
short events
, unused
void *data
)
573 waitpid(WAIT_ANY
, &status
, WNOHANG
);
580 main_callback(unused
int fd
, short events
, void *data
)
582 char *shellcmd
= data
;
584 if (events
& EV_READ
)
585 main_dispatch(shellcmd
);
587 if (events
& EV_WRITE
) {
588 if (msgbuf_write(&main_ibuf
->w
) < 0)
589 fatalx("msgbuf_write failed");
593 if (main_ibuf
->w
.queued
> 0)
595 event_once(main_ibuf
->fd
, events
, main_callback
, shellcmd
, NULL
);
599 main_dispatch(const char *shellcmd
)
603 struct msg_print_data printdata
;
604 struct msg_shell_data shelldata
;
606 if ((n
= imsg_read(main_ibuf
)) == -1 || n
== 0)
607 fatalx("imsg_read failed");
610 if ((n
= imsg_get(main_ibuf
, &imsg
)) == -1)
611 fatalx("imsg_get failed");
614 datalen
= imsg
.hdr
.len
- IMSG_HEADER_SIZE
;
616 switch (imsg
.hdr
.type
) {
620 fatalx("bad MSG_EXIT size");
625 if (datalen
!= sizeof printdata
)
626 fatalx("bad MSG_PRINT size");
627 memcpy(&printdata
, imsg
.data
, sizeof printdata
);
628 printdata
.msg
[(sizeof printdata
.msg
) - 1] = '\0';
630 log_info("%s", printdata
.msg
);
631 if (imsg
.hdr
.type
== MSG_ERROR
)
636 fatalx("bad MSG_READY size");
638 event_loopexit(NULL
); /* move to client_main() */
642 fatalx("bad MSG_VERSION size");
644 log_warnx("protocol version mismatch (client %u, "
645 "server %u)", PROTOCOL_VERSION
, imsg
.hdr
.peerid
);
648 if (datalen
!= sizeof shelldata
)
649 fatalx("bad MSG_SHELL size");
650 memcpy(&shelldata
, imsg
.data
, sizeof shelldata
);
651 shelldata
.shell
[(sizeof shelldata
.shell
) - 1] = '\0';
655 shell_exec(shelldata
.shell
, shellcmd
);
657 fatalx("unexpected message");