abduco.c

   1 /*
   2  * Copyright (c) 2013-2015 Marc André Tanner <mat at brain-dump.org>
   3  *
   4  * Permission to use, copy, modify, and/or distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  */
  16 #include <errno.h>
  17 #include <fcntl.h>
  18 #include <stdio.h>
  19 #include <stdarg.h>
  20 #include <stdlib.h>
  21 #include <stdbool.h>
  22 #include <stddef.h>
  23 #include <signal.h>
  24 #include <libgen.h>
  25 #include <string.h>
  26 #include <limits.h>
  27 #include <dirent.h>
  28 #include <termios.h>
  29 #include <time.h>
  30 #include <unistd.h>
  31 #include <pwd.h>
  32 #include <sys/select.h>
  33 #include <sys/stat.h>
  34 #include <sys/ioctl.h>
  35 #include <sys/types.h>
  36 #include <sys/wait.h>
  37 #include <sys/socket.h>
  38 #include <sys/un.h>
  39 #if defined(__linux__) || defined(__CYGWIN__)
  40 # include <pty.h>
  41 #elif defined(__FreeBSD__) || defined(__DragonFly__)
  42 # include <libutil.h>
  43 #elif defined(__OpenBSD__) || defined(__NetBSD__) || defined(__APPLE__)
  44 # include <util.h>
  45 #endif
  46
  47 #if defined CTRL && defined _AIX
  48   #undef CTRL
  49 #endif
  50 #ifndef CTRL
  51   #define CTRL(k)   ((k) & 0x1F)
  52 #endif
  53
  54 #include "config.h"
  55
  56 #if defined(_AIX)
  57 # include "forkpty-aix.c"
  58 #elif defined(__sun)
  59 # include "forkpty-sunos.c"
  60 #endif
  61
  62 #define countof(arr) (sizeof(arr) / sizeof((arr)[0]))
  63
  64 enum PacketType {
  65         MSG_CONTENT = 0,
  66         MSG_ATTACH  = 1,
  67         MSG_DETACH  = 2,
  68         MSG_RESIZE  = 3,
  69         MSG_REDRAW  = 4,
  70         MSG_EXIT    = 5,
  71 };
  72
  73 typedef struct {
  74         unsigned int type;
  75         size_t len;
  76         union {
  77                 char msg[BUFSIZ];
  78                 struct winsize ws;
  79                 int i;
  80                 bool b;
  81         } u;
  82 } Packet;
  83
  84 typedef struct Client Client;
  85 struct Client {
  86         int socket;
  87         enum {
  88                 STATE_CONNECTED,
  89                 STATE_ATTACHED,
  90                 STATE_DETACHED,
  91                 STATE_DISCONNECTED,
  92         } state;
  93         bool need_resize;
  94         bool readonly;
  95         Client *next;
  96 };
  97
  98 typedef struct {
  99         Client *clients;
 100         int socket;
 101         Packet pty_output;
 102         int pty;
 103         int exit_status;
 104         struct termios term;
 105         struct winsize winsize;
 106         pid_t pid;
 107         volatile sig_atomic_t running;
 108         const char *name;
 109         const char *session_name;
 110         char host[255];
 111         bool read_pty;
 112 } Server;
 113
 114 static Server server = { .running = true, .exit_status = -1, .host = "@localhost" };
 115 static Client client;
 116 static struct termios orig_term, cur_term;
 117 static bool has_term, alternate_buffer;
 118
 119 static struct sockaddr_un sockaddr = {
 120         .sun_family = AF_UNIX,
 121 };
 122
 123 static bool set_socket_name(struct sockaddr_un *sockaddr, const char *name);
 124 static void die(const char *s);
 125 static void info(const char *str, ...);
 126
 127 #include "debug.c"
 128
 129 static inline size_t packet_header_size() {
 130         return offsetof(Packet, u);
 131 }
 132
 133 static size_t packet_size(Packet *pkt) {
 134         return packet_header_size() + pkt->len;
 135 }
 136
 137 static ssize_t write_all(int fd, const char *buf, size_t len) {
 138         debug("write_all(%d)\n", len);
 139         ssize_t ret = len;
 140         while (len > 0) {
 141                 ssize_t res = write(fd, buf, len);
 142                 if (res < 0) {
 143                         if (errno == EAGAIN || errno == EWOULDBLOCK || errno == EINTR)
 144                                 continue;
 145                         return -1;
 146                 }
 147                 if (res == 0)
 148                         return ret - len;
 149                 buf += res;
 150                 len -= res;
 151         }
 152         return ret;
 153 }
 154
 155 static ssize_t read_all(int fd, char *buf, size_t len) {
 156         debug("read_all(%d)\n", len);
 157         ssize_t ret = len;
 158         while (len > 0) {
 159                 ssize_t res = read(fd, buf, len);
 160                 if (res < 0) {
 161                         if (errno == EWOULDBLOCK)
 162                                 return ret - len;
 163                         if (errno == EAGAIN || errno == EINTR)
 164                                 continue;
 165                         return -1;
 166                 }
 167                 if (res == 0)
 168                         return ret - len;
 169                 buf += res;
 170                 len -= res;
 171         }
 172         return ret;
 173 }
 174
 175 static bool send_packet(int socket, Packet *pkt) {
 176         size_t size = packet_size(pkt);
 177         if (size > sizeof(*pkt))
 178                 return false;
 179         return write_all(socket, (char *)pkt, size) == size;
 180 }
 181
 182 static bool recv_packet(int socket, Packet *pkt) {
 183         ssize_t len = read_all(socket, (char*)pkt, packet_header_size());
 184         if (len <= 0 || len != packet_header_size())
 185                 return false;
 186         if (pkt->len > sizeof(pkt->u.msg)) {
 187                 pkt->len = 0;
 188                 return false;
 189         }
 190         if (pkt->len > 0) {
 191                 len = read_all(socket, pkt->u.msg, pkt->len);
 192                 if (len <= 0 || len != pkt->len)
 193                         return false;
 194         }
 195         return true;
 196 }
 197
 198 #include "client.c"
 199 #include "server.c"
 200
 201 static void info(const char *str, ...) {
 202         va_list ap;
 203         va_start(ap, str);
 204         if (str) {
 205                 fprintf(stderr, "%s: %s: ", server.name, server.session_name);
 206                 vfprintf(stderr, str, ap);
 207                 fprintf(stderr, "\r\n");
 208         }
 209         fflush(stderr);
 210         va_end(ap);
 211 }
 212
 213 static void die(const char *s) {
 214         perror(s);
 215         exit(EXIT_FAILURE);
 216 }
 217
 218 static void usage(void) {
 219         fprintf(stderr, "usage: abduco [-a|-A|-c|-n] [-r] [-f] [-e detachkey] name command\n");
 220         exit(EXIT_FAILURE);
 221 }
 222
 223 static bool xsnprintf(char *buf, size_t size, const char *fmt, ...) {
 224         va_list ap;
 225         if (size > INT_MAX)
 226                 return false;
 227         va_start(ap, fmt);
 228         int n = vsnprintf(buf, size, fmt, ap);
 229         va_end(ap);
 230         if (n == -1)
 231                 return false;
 232         if (n >= size) {
 233                 errno = ENAMETOOLONG;
 234                 return false;
 235         }
 236         return true;
 237 }
 238
 239 static int session_connect(const char *name) {
 240         int fd;
 241         struct stat sb;
 242         if (!set_socket_name(&sockaddr, name) || (fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
 243                 return -1;
 244         socklen_t socklen = offsetof(struct sockaddr_un, sun_path) + strlen(sockaddr.sun_path) + 1;
 245         if (connect(fd, (struct sockaddr*)&sockaddr, socklen) == -1) {
 246                 if (errno == ECONNREFUSED && stat(sockaddr.sun_path, &sb) == 0 && S_ISSOCK(sb.st_mode))
 247                         unlink(sockaddr.sun_path);
 248                 close(fd);
 249                 return -1;
 250         }
 251         return fd;
 252 }
 253
 254 static bool session_exists(const char *name) {
 255         int fd = session_connect(name);
 256         if (fd != -1)
 257                 close(fd);
 258         return fd != -1;
 259 }
 260
 261 static bool session_alive(const char *name) {
 262         struct stat sb;
 263         return session_exists(name) &&
 264                stat(sockaddr.sun_path, &sb) == 0 &&
 265                S_ISSOCK(sb.st_mode) && (sb.st_mode & S_IXGRP) == 0;
 266 }
 267
 268 static bool create_socket_dir(struct sockaddr_un *sockaddr) {
 269         sockaddr->sun_path[0] = '\0';
 270         int socketfd = socket(AF_UNIX, SOCK_STREAM, 0);
 271         if (socketfd == -1)
 272                 return false;
 273
 274         size_t maxlen = sizeof(sockaddr->sun_path);
 275         uid_t uid = getuid();
 276         struct passwd *pw = getpwuid(uid);
 277         char *home = getenv("HOME");
 278         if ((!home || !home[0]) && pw)
 279                 home = pw->pw_dir;
 280
 281         struct {
 282                 char *dir;
 283                 bool personal; /* whether it is a per user directory */
 284         } dirs[] = {
 285                 { home, true },
 286                 { getenv("TMPDIR"), false },
 287                 { "/tmp", false },
 288         };
 289
 290         for (unsigned int i = 0; i < countof(dirs); i++) {
 291                 struct stat sb;
 292                 char *dir = dirs[i].dir;
 293                 bool ispersonal = dirs[i].personal;
 294                 if (!dir)
 295                         continue;
 296                 if (!xsnprintf(sockaddr->sun_path, maxlen, "%s/%s%s/", dir, dir == home ? "." : "", server.name))
 297                         continue;
 298                 mode_t mask = umask(0);
 299                 int r = mkdir(sockaddr->sun_path, ispersonal ? S_IRWXU : S_IRWXU|S_IRWXG|S_IRWXO|S_ISVTX);
 300                 umask(mask);
 301                 if (r != 0 && errno != EEXIST)
 302                         continue;
 303                 if (lstat(sockaddr->sun_path, &sb) != 0)
 304                         continue;
 305                 if (!S_ISDIR(sb.st_mode)) {
 306                         errno = ENOTDIR;
 307                         continue;
 308                 }
 309
 310                 size_t dirlen = strlen(sockaddr->sun_path);
 311                 if (!ispersonal) {
 312                         /* create subdirectory only accessible to user */
 313                         if (pw && !xsnprintf(sockaddr->sun_path+dirlen, maxlen-dirlen, "%s/", pw->pw_name))
 314                                 continue;
 315                         if (!pw && !xsnprintf(sockaddr->sun_path+dirlen, maxlen-dirlen, "%d/", uid))
 316                                 continue;
 317                         if (mkdir(sockaddr->sun_path, S_IRWXU) != 0 && errno != EEXIST)
 318                                 continue;
 319                         if (lstat(sockaddr->sun_path, &sb) != 0)
 320                                 continue;
 321                         if (!S_ISDIR(sb.st_mode)) {
 322                                 errno = ENOTDIR;
 323                                 continue;
 324                         }
 325                         dirlen = strlen(sockaddr->sun_path);
 326                 }
 327
 328                 if (sb.st_uid != uid || sb.st_mode & (S_IRWXG|S_IRWXO)) {
 329                         errno = EACCES;
 330                         continue;
 331                 }
 332
 333                 if (!xsnprintf(sockaddr->sun_path+dirlen, maxlen-dirlen, ".abduco-%d", getpid()))
 334                         continue;
 335
 336                 socklen_t socklen = offsetof(struct sockaddr_un, sun_path) + strlen(sockaddr->sun_path) + 1;
 337                 if (bind(socketfd, (struct sockaddr*)sockaddr, socklen) == -1)
 338                         continue;
 339                 unlink(sockaddr->sun_path);
 340                 close(socketfd);
 341                 sockaddr->sun_path[dirlen] = '\0';
 342                 return true;
 343         }
 344
 345         close(socketfd);
 346         return false;
 347 }
 348
 349 static bool set_socket_name(struct sockaddr_un *sockaddr, const char *name) {
 350         size_t maxlen = sizeof(sockaddr->sun_path);
 351         if (name[0] == '/') {
 352                 if (strlen(name) >= maxlen) {
 353                         errno = ENAMETOOLONG;
 354                         return false;
 355                 }
 356                 strncpy(sockaddr->sun_path, name, maxlen);
 357         } else if (name[0] == '.' && (name[1] == '.' || name[1] == '/')) {
 358                 char buf[maxlen], *cwd = getcwd(buf, sizeof buf);
 359                 if (!cwd)
 360                         return false;
 361                 if (!xsnprintf(sockaddr->sun_path, maxlen, "%s/%s", cwd, name))
 362                         return false;
 363         } else {
 364                 if (!create_socket_dir(sockaddr))
 365                         return false;
 366                 if (strlen(sockaddr->sun_path) + strlen(name) + strlen(server.host) >= maxlen) {
 367                         errno = ENAMETOOLONG;
 368                         return false;
 369                 }
 370                 strncat(sockaddr->sun_path, name, maxlen - strlen(sockaddr->sun_path) - 1);
 371                 strncat(sockaddr->sun_path, server.host, maxlen - strlen(sockaddr->sun_path) - 1);
 372         }
 373         return true;
 374 }
 375
 376 static bool create_session(const char *name, char * const argv[]) {
 377         /* this uses the well known double fork strategy as described in section 1.7 of
 378          *
 379          *  http://www.faqs.org/faqs/unix-faq/programmer/faq/
 380          *
 381          * pipes are used for synchronization and error reporting i.e. the child sets
 382          * the close on exec flag before calling execvp(3) the parent blocks on a read(2)
 383          * in case of failure the error message is written to the pipe, success is
 384          * indicated by EOF on the pipe.
 385          */
 386         int client_pipe[2], server_pipe[2];
 387         pid_t pid;
 388         char errormsg[255];
 389         struct sigaction sa;
 390
 391         if (session_exists(name)) {
 392                 errno = EADDRINUSE;
 393                 return false;
 394         }
 395
 396         if (pipe(client_pipe) == -1)
 397                 return false;
 398         if ((server.socket = server_create_socket(name)) == -1)
 399                 return false;
 400
 401         switch ((pid = fork())) {
 402         case 0: /* child process */
 403                 setsid();
 404                 close(client_pipe[0]);
 405                 switch ((pid = fork())) {
 406                 case 0: /* child process */
 407                         if (pipe(server_pipe) == -1) {
 408                                 snprintf(errormsg, sizeof(errormsg), "server-pipe: %s\n", strerror(errno));
 409                                 write_all(client_pipe[1], errormsg, strlen(errormsg));
 410                                 close(client_pipe[1]);
 411                                 _exit(EXIT_FAILURE);
 412                         }
 413                         sa.sa_flags = 0;
 414                         sigemptyset(&sa.sa_mask);
 415                         sa.sa_handler = server_pty_died_handler;
 416                         sigaction(SIGCHLD, &sa, NULL);
 417                         switch (server.pid = forkpty(&server.pty, NULL, has_term ? &server.term : NULL, &server.winsize)) {
 418                         case 0: /* child = user application process */
 419                                 close(server.socket);
 420                                 close(server_pipe[0]);
 421                                 if (fcntl(client_pipe[1], F_SETFD, FD_CLOEXEC) == 0 &&
 422                                     fcntl(server_pipe[1], F_SETFD, FD_CLOEXEC) == 0)
 423                                         execvp(argv[0], argv);
 424                                 snprintf(errormsg, sizeof(errormsg), "server-execvp: %s: %s\n",
 425                                                  argv[0], strerror(errno));
 426                                 write_all(client_pipe[1], errormsg, strlen(errormsg));
 427                                 write_all(server_pipe[1], errormsg, strlen(errormsg));
 428                                 close(client_pipe[1]);
 429                                 close(server_pipe[1]);
 430                                 _exit(EXIT_FAILURE);
 431                                 break;
 432                         case -1: /* forkpty failed */
 433                                 snprintf(errormsg, sizeof(errormsg), "server-forkpty: %s\n", strerror(errno));
 434                                 write_all(client_pipe[1], errormsg, strlen(errormsg));
 435                                 close(client_pipe[1]);
 436                                 close(server_pipe[0]);
 437                                 close(server_pipe[1]);
 438                                 _exit(EXIT_FAILURE);
 439                                 break;
 440                         default: /* parent = server process */
 441                                 sa.sa_handler = server_sigterm_handler;
 442                                 sigaction(SIGTERM, &sa, NULL);
 443                                 sigaction(SIGINT, &sa, NULL);
 444                                 sa.sa_handler = server_sigusr1_handler;
 445                                 sigaction(SIGUSR1, &sa, NULL);
 446                                 sa.sa_handler = SIG_IGN;
 447                                 sigaction(SIGPIPE, &sa, NULL);
 448                                 sigaction(SIGHUP, &sa, NULL);
 449                                 chdir("/");
 450                         #ifdef NDEBUG
 451                                 int fd = open("/dev/null", O_RDWR);
 452                                 if (fd != -1) {
 453                                         dup2(fd, 0);
 454                                         dup2(fd, 1);
 455                                         dup2(fd, 2);
 456                                         close(fd);
 457                                 }
 458                         #endif /* NDEBUG */
 459                                 close(client_pipe[1]);
 460                                 close(server_pipe[1]);
 461                                 if (read_all(server_pipe[0], errormsg, sizeof(errormsg)) > 0)
 462                                         _exit(EXIT_FAILURE);
 463                                 close(server_pipe[0]);
 464                                 server_mainloop();
 465                                 break;
 466                         }
 467                         break;
 468                 case -1: /* fork failed */
 469                         snprintf(errormsg, sizeof(errormsg), "server-fork: %s\n", strerror(errno));
 470                         write_all(client_pipe[1], errormsg, strlen(errormsg));
 471                         close(client_pipe[1]);
 472                         _exit(EXIT_FAILURE);
 473                         break;
 474                 default: /* parent = intermediate process */
 475                         close(client_pipe[1]);
 476                         _exit(EXIT_SUCCESS);
 477                         break;
 478                 }
 479                 break;
 480         case -1: /* fork failed */
 481                 close(client_pipe[0]);
 482                 close(client_pipe[1]);
 483                 return false;
 484         default: /* parent = client process */
 485                 close(client_pipe[1]);
 486                 int status;
 487                 wait(&status); /* wait for first fork */
 488                 ssize_t len = read_all(client_pipe[0], errormsg, sizeof(errormsg));
 489                 if (len > 0) {
 490                         write_all(STDERR_FILENO, errormsg, len);
 491                         unlink(sockaddr.sun_path);
 492                         exit(EXIT_FAILURE);
 493                 }
 494                 close(client_pipe[0]);
 495         }
 496         return true;
 497 }
 498
 499 static bool attach_session(const char *name, const bool terminate) {
 500         if (server.socket > 0)
 501                 close(server.socket);
 502         if ((server.socket = session_connect(name)) == -1)
 503                 return false;
 504         if (server_set_socket_non_blocking(server.socket) == -1)
 505                 return false;
 506
 507         struct sigaction sa;
 508         sa.sa_flags = 0;
 509         sigemptyset(&sa.sa_mask);
 510         sa.sa_handler = client_sigwinch_handler;
 511         sigaction(SIGWINCH, &sa, NULL);
 512         sa.sa_handler = SIG_IGN;
 513         sigaction(SIGPIPE, &sa, NULL);
 514
 515         client_setup_terminal();
 516         int status = client_mainloop();
 517         client_restore_terminal();
 518         if (status == -1) {
 519                 info("detached");
 520         } else if (status == -EIO) {
 521                 info("exited due to I/O errors");
 522         } else {
 523                 info("session terminated with exit status %d", status);
 524                 if (terminate)
 525                         exit(status);
 526         }
 527
 528         return terminate;
 529 }
 530
 531 static int session_filter(const struct dirent *d) {
 532         return strstr(d->d_name, server.host) != NULL;
 533 }
 534
 535 static int session_comparator(const struct dirent **a, const struct dirent **b) {
 536         struct stat sa, sb;
 537         if (stat((*a)->d_name, &sa) != 0)
 538                 return -1;
 539         if (stat((*b)->d_name, &sb) != 0)
 540                 return 1;
 541         return sa.st_atime < sb.st_atime ? -1 : 1;
 542 }
 543
 544 static int list_session(void) {
 545         if (!create_socket_dir(&sockaddr))
 546                 return 1;
 547         chdir(sockaddr.sun_path);
 548         struct dirent **namelist;
 549         int n = scandir(sockaddr.sun_path, &namelist, session_filter, session_comparator);
 550         if (n < 0)
 551                 return 1;
 552         printf("Active sessions (on host %s)\n", server.host+1);
 553         while (n--) {
 554                 struct stat sb; char buf[255];
 555                 if (stat(namelist[n]->d_name, &sb) == 0 && S_ISSOCK(sb.st_mode)) {
 556                         strftime(buf, sizeof(buf), "%a%t %F %T", localtime(&sb.st_mtime));
 557                         char status = ' ';
 558                         char *local = strstr(namelist[n]->d_name, server.host);
 559                         if (local) {
 560                                 *local = '\0'; /* truncate hostname if we are local */
 561                                 if (!session_exists(namelist[n]->d_name))
 562                                         continue;
 563                         }
 564                         if (sb.st_mode & S_IXUSR)
 565                                 status = '*';
 566                         else if (sb.st_mode & S_IXGRP)
 567                                 status = '+';
 568                         printf("%c %s\t%s\n", status, buf, namelist[n]->d_name);
 569                 }
 570                 free(namelist[n]);
 571         }
 572         free(namelist);
 573         return 0;
 574 }
 575
 576 int main(int argc, char *argv[]) {
 577         int opt;
 578         bool force = false;
 579         char **cmd = NULL, action = '\0';
 580         server.name = basename(argv[0]);
 581         gethostname(server.host+1, sizeof(server.host) - 1);
 582         if (argc == 1)
 583                 exit(list_session());
 584
 585         while ((opt = getopt(argc, argv, "aAcne:frv")) != -1) {
 586                 switch (opt) {
 587                 case 'a':
 588                 case 'A':
 589                 case 'c':
 590                 case 'n':
 591                         action = opt;
 592                         break;
 593                 case 'e':
 594                         if (!optarg)
 595                                 usage();
 596                         if (optarg[0] == '^' && optarg[1])
 597                                 optarg[0] = CTRL(optarg[1]);
 598                         KEY_DETACH = optarg[0];
 599                         break;
 600                 case 'f':
 601                         force = true;
 602                         break;
 603                 case 'r':
 604                         client.readonly = true;
 605                         break;
 606                 case 'v':
 607                         puts("abduco-"VERSION" © 2013-2015 Marc André Tanner");
 608                         exit(EXIT_SUCCESS);
 609                 default:
 610                         usage();
 611                 }
 612         }
 613
 614         /* collect the session name if trailing args */
 615         if (optind < argc)
 616                 server.session_name = argv[optind];
 617
 618         /* if yet more trailing arguments, they must be the command */
 619         if (optind + 1 < argc)
 620                 cmd = &argv[optind + 1];
 621
 622         if (!cmd) {
 623                 cmd = (char*[]){ getenv("ABDUCO_CMD"), NULL };
 624                 if (!cmd[0])
 625                         cmd[0] = "dvtm";
 626         }
 627
 628         if (!action || !server.session_name)
 629                 usage();
 630
 631         if (tcgetattr(STDIN_FILENO, &orig_term) != -1) {
 632                 server.term = orig_term;
 633                 has_term = true;
 634         }
 635
 636         if (ioctl(STDIN_FILENO, TIOCGWINSZ, &server.winsize) == -1) {
 637                 server.winsize.ws_col = 80;
 638                 server.winsize.ws_row = 25;
 639         }
 640
 641         server.read_pty = (action == 'n');
 642
 643         redo:
 644         switch (action) {
 645         case 'n':
 646         case 'c':
 647                 if (force) {
 648                         if (session_alive(server.session_name)) {
 649                                 info("session exists and has not yet terminated");
 650                                 return 1;
 651                         }
 652                         if (session_exists(server.session_name))
 653                                 attach_session(server.session_name, false);
 654                 }
 655                 if (!create_session(server.session_name, cmd))
 656                         die("create-session");
 657                 if (action == 'n')
 658                         break;
 659         case 'a':
 660                 if (!attach_session(server.session_name, true))
 661                         die("attach-session");
 662                 break;
 663         case 'A':
 664                 if (session_alive(server.session_name)) {
 665                         if (!attach_session(server.session_name, true))
 666                                 die("attach-session");
 667                 } else if (!attach_session(server.session_name, !force)) {
 668                         force = false;
 669                         action = 'c';
 670                         goto redo;
 671                 }
 672                 break;
 673         }
 674
 675         return 0;
 676 }