1/* $OpenBSD$ */
2
3/*
4 * Copyright (c) 2007 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 <errno.h>
22#include <fcntl.h>
23#include <stdlib.h>
24#include <string.h>
25#include <termios.h>
26#include <unistd.h>
27
28#include "tmux.h"
29
30/*
31 * Create a new session and attach to the current terminal unless -d is given.
32 */
33
34#define NEW_SESSION_TEMPLATE "#{session_name}:"
35
36static enum cmd_retval	cmd_new_session_exec(struct cmd *, struct cmdq_item *);
37
38const struct cmd_entry cmd_new_session_entry = {
39	.name = "new-session",
40	.alias = "new",
41
42	.args = { "Ac:dDe:EF:f:n:Ps:t:x:Xy:", 0, -1, NULL },
43	.usage = "[-AdDEPX] [-c start-directory] [-e environment] [-F format] "
44		 "[-f flags] [-n window-name] [-s session-name] "
45		 CMD_TARGET_SESSION_USAGE " [-x width] [-y height] "
46		 "[shell-command]",
47
48	.target = { 't', CMD_FIND_SESSION, CMD_FIND_CANFAIL },
49
50	.flags = CMD_STARTSERVER,
51	.exec = cmd_new_session_exec
52};
53
54const struct cmd_entry cmd_has_session_entry = {
55	.name = "has-session",
56	.alias = "has",
57
58	.args = { "t:", 0, 0, NULL },
59	.usage = CMD_TARGET_SESSION_USAGE,
60
61	.target = { 't', CMD_FIND_SESSION, 0 },
62
63	.flags = 0,
64	.exec = cmd_new_session_exec
65};
66
67static enum cmd_retval
68cmd_new_session_exec(struct cmd *self, struct cmdq_item *item)
69{
70	struct args		*args = cmd_get_args(self);
71	struct cmd_find_state	*current = cmdq_get_current(item);
72	struct cmd_find_state	*target = cmdq_get_target(item);
73	struct client		*c = cmdq_get_client(item);
74	struct session		*s, *as, *groupwith = NULL;
75	struct environ		*env;
76	struct options		*oo;
77	struct termios		 tio, *tiop;
78	struct session_group	*sg = NULL;
79	const char		*errstr, *template, *group, *tmp;
80	char			*cause, *cwd = NULL, *cp, *newname = NULL;
81	char			*name, *prefix = NULL;
82	int			 detached, already_attached, is_control = 0;
83	u_int			 sx, sy, dsx = 80, dsy = 24, count = args_count(args);
84	struct spawn_context	 sc = { 0 };
85	enum cmd_retval		 retval;
86	struct cmd_find_state    fs;
87	struct args_value	*av;
88
89	if (cmd_get_entry(self) == &cmd_has_session_entry) {
90		/*
91		 * cmd_find_target() will fail if the session cannot be found,
92		 * so always return success here.
93		 */
94		return (CMD_RETURN_NORMAL);
95	}
96
97	if (args_has(args, 't') && (count != 0 || args_has(args, 'n'))) {
98		cmdq_error(item, "command or window name given with target");
99		return (CMD_RETURN_ERROR);
100	}
101
102	tmp = args_get(args, 's');
103	if (tmp != NULL) {
104		name = format_single(item, tmp, c, NULL, NULL, NULL);
105		newname = session_check_name(name);
106		if (newname == NULL) {
107			cmdq_error(item, "invalid session: %s", name);
108			free(name);
109			return (CMD_RETURN_ERROR);
110		}
111		free(name);
112	}
113	if (args_has(args, 'A')) {
114		if (newname != NULL)
115			as = session_find(newname);
116		else
117			as = target->s;
118		if (as != NULL) {
119			retval = cmd_attach_session(item, as->name,
120			    args_has(args, 'D'), args_has(args, 'X'), 0, NULL,
121			    args_has(args, 'E'), args_get(args, 'f'));
122			free(newname);
123			return (retval);
124		}
125	}
126	if (newname != NULL && session_find(newname) != NULL) {
127		cmdq_error(item, "duplicate session: %s", newname);
128		goto fail;
129	}
130
131	/* Is this going to be part of a session group? */
132	group = args_get(args, 't');
133	if (group != NULL) {
134		groupwith = target->s;
135		if (groupwith == NULL)
136			sg = session_group_find(group);
137		else
138			sg = session_group_contains(groupwith);
139		if (sg != NULL)
140			prefix = xstrdup(sg->name);
141		else if (groupwith != NULL)
142			prefix = xstrdup(groupwith->name);
143		else {
144			prefix = session_check_name(group);
145			if (prefix == NULL) {
146				cmdq_error(item, "invalid session group: %s",
147				    group);
148				goto fail;
149			}
150		}
151	}
152
153	/* Set -d if no client. */
154	detached = args_has(args, 'd');
155	if (c == NULL)
156		detached = 1;
157	else if (c->flags & CLIENT_CONTROL)
158		is_control = 1;
159
160	/* Is this client already attached? */
161	already_attached = 0;
162	if (c != NULL && c->session != NULL)
163		already_attached = 1;
164
165	/* Get the new session working directory. */
166	if ((tmp = args_get(args, 'c')) != NULL)
167		cwd = format_single(item, tmp, c, NULL, NULL, NULL);
168	else
169		cwd = xstrdup(server_client_get_cwd(c, NULL));
170
171	/*
172	 * If this is a new client, check for nesting and save the termios
173	 * settings (part of which is used for new windows in this session).
174	 *
175	 * tcgetattr() is used rather than using tty.tio since if the client is
176	 * detached, tty_open won't be called. It must be done before opening
177	 * the terminal as that calls tcsetattr() to prepare for tmux taking
178	 * over.
179	 */
180	if (!detached &&
181	    !already_attached &&
182	    c->fd != -1 &&
183	    (~c->flags & CLIENT_CONTROL)) {
184		if (server_client_check_nested(cmdq_get_client(item))) {
185			cmdq_error(item, "sessions should be nested with care, "
186			    "unset $TMUX to force");
187			goto fail;
188		}
189		if (tcgetattr(c->fd, &tio) != 0)
190			fatal("tcgetattr failed");
191		tiop = &tio;
192	} else
193		tiop = NULL;
194
195	/* Open the terminal if necessary. */
196	if (!detached && !already_attached) {
197		if (server_client_open(c, &cause) != 0) {
198			cmdq_error(item, "open terminal failed: %s", cause);
199			free(cause);
200			goto fail;
201		}
202	}
203
204	/* Get default session size. */
205	if (args_has(args, 'x')) {
206		tmp = args_get(args, 'x');
207		if (strcmp(tmp, "-") == 0) {
208			if (c != NULL)
209				dsx = c->tty.sx;
210			else
211				dsx = 80;
212		} else {
213			dsx = strtonum(tmp, 1, USHRT_MAX, &errstr);
214			if (errstr != NULL) {
215				cmdq_error(item, "width %s", errstr);
216				goto fail;
217			}
218		}
219	} else
220		dsx = 80;
221	if (args_has(args, 'y')) {
222		tmp = args_get(args, 'y');
223		if (strcmp(tmp, "-") == 0) {
224			if (c != NULL)
225				dsy = c->tty.sy;
226			else
227				dsy = 24;
228		} else {
229			dsy = strtonum(tmp, 1, USHRT_MAX, &errstr);
230			if (errstr != NULL) {
231				cmdq_error(item, "height %s", errstr);
232				goto fail;
233			}
234		}
235	} else
236		dsy = 24;
237
238	/* Find new session size. */
239	if (!detached && !is_control) {
240		sx = c->tty.sx;
241		sy = c->tty.sy;
242		if (sy > 0 && options_get_number(global_s_options, "status"))
243			sy--;
244	} else {
245		tmp = options_get_string(global_s_options, "default-size");
246		if (sscanf(tmp, "%ux%u", &sx, &sy) != 2) {
247			sx = dsx;
248			sy = dsy;
249		} else {
250			if (args_has(args, 'x'))
251				sx = dsx;
252			if (args_has(args, 'y'))
253				sy = dsy;
254		}
255	}
256	if (sx == 0)
257		sx = 1;
258	if (sy == 0)
259		sy = 1;
260
261	/* Create the new session. */
262	oo = options_create(global_s_options);
263	if (args_has(args, 'x') || args_has(args, 'y')) {
264		if (!args_has(args, 'x'))
265			dsx = sx;
266		if (!args_has(args, 'y'))
267			dsy = sy;
268		options_set_string(oo, "default-size", 0, "%ux%u", dsx, dsy);
269	}
270	env = environ_create();
271	if (c != NULL && !args_has(args, 'E'))
272		environ_update(global_s_options, c->environ, env);
273	av = args_first_value(args, 'e');
274	while (av != NULL) {
275		environ_put(env, av->string, 0);
276		av = args_next_value(av);
277	}
278	s = session_create(prefix, newname, cwd, env, oo, tiop);
279
280	/* Spawn the initial window. */
281	sc.item = item;
282	sc.s = s;
283	if (!detached)
284		sc.tc = c;
285
286	sc.name = args_get(args, 'n');
287	args_to_vector(args, &sc.argc, &sc.argv);
288
289	sc.idx = -1;
290	sc.cwd = args_get(args, 'c');
291
292	sc.flags = 0;
293
294	if (spawn_window(&sc, &cause) == NULL) {
295		session_destroy(s, 0, __func__);
296		cmdq_error(item, "create window failed: %s", cause);
297		free(cause);
298		goto fail;
299	}
300
301	/*
302	 * If a target session is given, this is to be part of a session group,
303	 * so add it to the group and synchronize.
304	 */
305	if (group != NULL) {
306		if (sg == NULL) {
307			if (groupwith != NULL) {
308				sg = session_group_new(groupwith->name);
309				session_group_add(sg, groupwith);
310			} else
311				sg = session_group_new(group);
312		}
313		session_group_add(sg, s);
314		session_group_synchronize_to(s);
315		session_select(s, RB_MIN(winlinks, &s->windows)->idx);
316	}
317	notify_session("session-created", s);
318
319	/*
320	 * Set the client to the new session. If a command client exists, it is
321	 * taking this session and needs to get MSG_READY and stay around.
322	 */
323	if (!detached) {
324		if (args_has(args, 'f'))
325			server_client_set_flags(c, args_get(args, 'f'));
326		if (!already_attached) {
327			if (~c->flags & CLIENT_CONTROL)
328				proc_send(c->peer, MSG_READY, -1, NULL, 0);
329		} else if (c->session != NULL)
330			c->last_session = c->session;
331		server_client_set_session(c, s);
332		if (~cmdq_get_flags(item) & CMDQ_STATE_REPEAT)
333			server_client_set_key_table(c, NULL);
334	}
335
336	/*
337	 * If there are still configuration file errors to display, put the new
338	 * session's current window into more mode and display them now.
339	 */
340	if (cfg_finished)
341		cfg_show_causes(s);
342
343	/* Print if requested. */
344	if (args_has(args, 'P')) {
345		if ((template = args_get(args, 'F')) == NULL)
346			template = NEW_SESSION_TEMPLATE;
347		cp = format_single(item, template, c, s, s->curw, NULL);
348		cmdq_print(item, "%s", cp);
349		free(cp);
350	}
351
352	if (!detached)
353		c->flags |= CLIENT_ATTACHED;
354	if (!args_has(args, 'd'))
355		cmd_find_from_session(current, s, 0);
356
357	cmd_find_from_session(&fs, s, 0);
358	cmdq_insert_hook(s, item, &fs, "after-new-session");
359
360	if (sc.argv != NULL)
361		cmd_free_argv(sc.argc, sc.argv);
362	free(cwd);
363	free(newname);
364	free(prefix);
365	return (CMD_RETURN_NORMAL);
366
367fail:
368	if (sc.argv != NULL)
369		cmd_free_argv(sc.argc, sc.argv);
370	free(cwd);
371	free(newname);
372	free(prefix);
373	return (CMD_RETURN_ERROR);
374}
375