cmd-new-session.c

   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
  36 static enum cmd_retval  cmd_new_session_exec(struct cmd *, struct cmdq_item *);
  37
  38 const 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
  54 const 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
  67 static enum cmd_retval
  68 cmd_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, dsy, 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
 367 fail:
 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 }