Terminate strftime buffer properly even if a really long format string
[tmux-openbsd.git] / tmux.c
blob35d4390981ea8984d5cbb48fb48b4b7b3707dfae
1 /* $OpenBSD$ */
3 /*
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>
20 #include <sys/stat.h>
22 #include <errno.h>
23 #include <event.h>
24 #include <fcntl.h>
25 #include <paths.h>
26 #include <pwd.h>
27 #include <stdlib.h>
28 #include <string.h>
29 #include <unistd.h>
31 #include "tmux.h"
33 #ifdef DEBUG
34 extern char *malloc_options;
35 #endif
37 struct options global_options; /* server options */
38 struct options global_s_options; /* session options */
39 struct options global_w_options; /* window options */
40 struct environ global_environ;
42 struct event_base *ev_base;
44 char *cfg_file;
45 char *shell_cmd;
46 int debug_level;
47 time_t start_time;
48 char socket_path[MAXPATHLEN];
49 int login_shell;
50 char *environ_path;
51 pid_t environ_pid = -1;
52 int environ_idx = -1;
54 __dead void usage(void);
55 void parseenvironment(void);
56 char *makesocketpath(const char *);
58 __dead void
59 usage(void)
61 fprintf(stderr,
62 "usage: %s [-28lquv] [-c shell-command] [-f file] [-L socket-name]\n"
63 " [-S socket-path] [command [flags]]\n",
64 __progname);
65 exit(1);
68 void
69 logfile(const char *name)
71 char *path;
73 log_close();
74 if (debug_level > 0) {
75 xasprintf(&path, "tmux-%s-%ld.log", name, (long) getpid());
76 log_open_file(debug_level, path);
77 xfree(path);
81 const char *
82 getshell(void)
84 struct passwd *pw;
85 const char *shell;
87 shell = getenv("SHELL");
88 if (checkshell(shell))
89 return (shell);
91 pw = getpwuid(getuid());
92 if (pw != NULL && checkshell(pw->pw_shell))
93 return (pw->pw_shell);
95 return (_PATH_BSHELL);
98 int
99 checkshell(const char *shell)
101 if (shell == NULL || *shell == '\0' || *shell != '/')
102 return (0);
103 if (areshell(shell))
104 return (0);
105 if (access(shell, X_OK) != 0)
106 return (0);
107 return (1);
111 areshell(const char *shell)
113 const char *progname, *ptr;
115 if ((ptr = strrchr(shell, '/')) != NULL)
116 ptr++;
117 else
118 ptr = shell;
119 progname = __progname;
120 if (*progname == '-')
121 progname++;
122 if (strcmp(ptr, progname) == 0)
123 return (1);
124 return (0);
127 const char*
128 get_full_path(const char *wd, const char *path)
130 static char newpath[MAXPATHLEN];
131 char oldpath[MAXPATHLEN];
133 if (getcwd(oldpath, sizeof oldpath) == NULL)
134 return (NULL);
135 if (chdir(wd) != 0)
136 return (NULL);
137 if (realpath(path, newpath) != 0)
138 return (NULL);
139 chdir(oldpath);
140 return (newpath);
143 void
144 parseenvironment(void)
146 char *env, path[256];
147 long pid;
148 int idx;
150 if ((env = getenv("TMUX")) == NULL)
151 return;
153 if (sscanf(env, "%255[^,],%ld,%d", path, &pid, &idx) != 3)
154 return;
155 environ_path = xstrdup(path);
156 environ_pid = pid;
157 environ_idx = idx;
160 char *
161 makesocketpath(const char *label)
163 char base[MAXPATHLEN], *path, *s;
164 struct stat sb;
165 u_int uid;
167 uid = getuid();
168 if ((s = getenv("TMPDIR")) == NULL || *s == '\0')
169 xsnprintf(base, sizeof base, "%s/tmux-%u", _PATH_TMP, uid);
170 else
171 xsnprintf(base, sizeof base, "%s/tmux-%u", s, uid);
173 if (mkdir(base, S_IRWXU) != 0 && errno != EEXIST)
174 return (NULL);
176 if (lstat(base, &sb) != 0)
177 return (NULL);
178 if (!S_ISDIR(sb.st_mode)) {
179 errno = ENOTDIR;
180 return (NULL);
182 if (sb.st_uid != uid || (sb.st_mode & (S_IRWXG|S_IRWXO)) != 0) {
183 errno = EACCES;
184 return (NULL);
187 xasprintf(&path, "%s/%s", base, label);
188 return (path);
191 void
192 setblocking(int fd, int state)
194 int mode;
196 if ((mode = fcntl(fd, F_GETFL)) != -1) {
197 if (!state)
198 mode |= O_NONBLOCK;
199 else
200 mode &= ~O_NONBLOCK;
201 fcntl(fd, F_SETFL, mode);
205 __dead void
206 shell_exec(const char *shell, const char *shellcmd)
208 const char *shellname, *ptr;
209 char *argv0;
211 ptr = strrchr(shell, '/');
212 if (ptr != NULL && *(ptr + 1) != '\0')
213 shellname = ptr + 1;
214 else
215 shellname = shell;
216 if (login_shell)
217 xasprintf(&argv0, "-%s", shellname);
218 else
219 xasprintf(&argv0, "%s", shellname);
220 setenv("SHELL", shell, 1);
222 setblocking(STDIN_FILENO, 1);
223 setblocking(STDOUT_FILENO, 1);
224 setblocking(STDERR_FILENO, 1);
225 closefrom(STDERR_FILENO + 1);
227 execl(shell, argv0, "-c", shellcmd, (char *) NULL);
228 fatal("execl failed");
232 main(int argc, char **argv)
234 struct passwd *pw;
235 char *s, *path, *label, *home, **var;
236 int opt, flags, quiet, keys;
238 #ifdef DEBUG
239 malloc_options = (char *) "AFGJPX";
240 #endif
242 quiet = flags = 0;
243 label = path = NULL;
244 login_shell = (**argv == '-');
245 while ((opt = getopt(argc, argv, "28c:df:lL:qS:uUv")) != -1) {
246 switch (opt) {
247 case '2':
248 flags |= IDENTIFY_256COLOURS;
249 flags &= ~IDENTIFY_88COLOURS;
250 break;
251 case '8':
252 flags |= IDENTIFY_88COLOURS;
253 flags &= ~IDENTIFY_256COLOURS;
254 break;
255 case 'c':
256 if (shell_cmd != NULL)
257 xfree(shell_cmd);
258 shell_cmd = xstrdup(optarg);
259 break;
260 case 'f':
261 if (cfg_file != NULL)
262 xfree(cfg_file);
263 cfg_file = xstrdup(optarg);
264 break;
265 case 'l':
266 login_shell = 1;
267 break;
268 case 'L':
269 if (label != NULL)
270 xfree(label);
271 label = xstrdup(optarg);
272 break;
273 case 'q':
274 quiet = 1;
275 break;
276 case 'S':
277 if (path != NULL)
278 xfree(path);
279 path = xstrdup(optarg);
280 break;
281 case 'u':
282 flags |= IDENTIFY_UTF8;
283 break;
284 case 'v':
285 debug_level++;
286 break;
287 default:
288 usage();
291 argc -= optind;
292 argv += optind;
294 if (shell_cmd != NULL && argc != 0)
295 usage();
297 log_open_tty(debug_level);
299 if (!(flags & IDENTIFY_UTF8)) {
301 * If the user has set whichever of LC_ALL, LC_CTYPE or LANG
302 * exist (in that order) to contain UTF-8, it is a safe
303 * assumption that either they are using a UTF-8 terminal, or
304 * if not they know that output from UTF-8-capable programs may
305 * be wrong.
307 if ((s = getenv("LC_ALL")) == NULL || *s == '\0') {
308 if ((s = getenv("LC_CTYPE")) == NULL || *s == '\0')
309 s = getenv("LANG");
311 if (s != NULL && (strcasestr(s, "UTF-8") != NULL ||
312 strcasestr(s, "UTF8") != NULL))
313 flags |= IDENTIFY_UTF8;
316 environ_init(&global_environ);
317 for (var = environ; *var != NULL; var++)
318 environ_put(&global_environ, *var);
320 options_init(&global_options, NULL);
321 options_table_populate_tree(server_options_table, &global_options);
322 options_set_number(&global_options, "quiet", quiet);
324 options_init(&global_s_options, NULL);
325 options_table_populate_tree(session_options_table, &global_s_options);
326 options_set_string(&global_s_options, "default-shell", "%s", getshell());
328 options_init(&global_w_options, NULL);
329 options_table_populate_tree(window_options_table, &global_w_options);
331 /* Enable UTF-8 if the first client is on UTF-8 terminal. */
332 if (flags & IDENTIFY_UTF8) {
333 options_set_number(&global_s_options, "status-utf8", 1);
334 options_set_number(&global_s_options, "mouse-utf8", 1);
335 options_set_number(&global_w_options, "utf8", 1);
338 /* Override keys to vi if VISUAL or EDITOR are set. */
339 if ((s = getenv("VISUAL")) != NULL || (s = getenv("EDITOR")) != NULL) {
340 if (strrchr(s, '/') != NULL)
341 s = strrchr(s, '/') + 1;
342 if (strstr(s, "vi") != NULL)
343 keys = MODEKEY_VI;
344 else
345 keys = MODEKEY_EMACS;
346 options_set_number(&global_s_options, "status-keys", keys);
347 options_set_number(&global_w_options, "mode-keys", keys);
350 /* Locate the configuration file. */
351 if (cfg_file == NULL) {
352 home = getenv("HOME");
353 if (home == NULL || *home == '\0') {
354 pw = getpwuid(getuid());
355 if (pw != NULL)
356 home = pw->pw_dir;
358 xasprintf(&cfg_file, "%s/%s", home, DEFAULT_CFG);
359 if (access(cfg_file, R_OK) != 0 && errno == ENOENT) {
360 xfree(cfg_file);
361 cfg_file = NULL;
366 * Figure out the socket path. If specified on the command-line with -S
367 * or -L, use it, otherwise try $TMUX or assume -L default.
369 parseenvironment();
370 if (path == NULL) {
371 /* If no -L, use the environment. */
372 if (label == NULL) {
373 if (environ_path != NULL)
374 path = xstrdup(environ_path);
375 else
376 label = xstrdup("default");
379 /* -L or default set. */
380 if (label != NULL) {
381 if ((path = makesocketpath(label)) == NULL) {
382 log_warn("can't create socket");
383 exit(1);
387 if (label != NULL)
388 xfree(label);
389 if (realpath(path, socket_path) == NULL)
390 strlcpy(socket_path, path, sizeof socket_path);
391 xfree(path);
393 /* Set process title. */
394 setproctitle("%s (%s)", __progname, socket_path);
396 /* Pass control to the client. */
397 ev_base = event_init();
398 exit(client_main(argc, argv, flags));