popen.c

   1 /*@ S-nail - a mail user agent derived from Berkeley Mail.
   2  *@ Handling of pipes, child processes, temporary files, file enwrapping.
   3  *
   4  * Copyright (c) 2000-2004 Gunnar Ritter, Freiburg i. Br., Germany.
   5  * Copyright (c) 2012 - 2015 Steffen (Daode) Nurpmeso <sdaoden@users.sf.net>.
   6  */
   7 /*
   8  * Copyright (c) 1980, 1993
   9  *      The Regents of the University of California.  All rights reserved.
  10  *
  11  * Redistribution and use in source and binary forms, with or without
  12  * modification, are permitted provided that the following conditions
  13  * are met:
  14  * 1. Redistributions of source code must retain the above copyright
  15  *    notice, this list of conditions and the following disclaimer.
  16  * 2. Redistributions in binary form must reproduce the above copyright
  17  *    notice, this list of conditions and the following disclaimer in the
  18  *    documentation and/or other materials provided with the distribution.
  19  * 3. Neither the name of the University nor the names of its contributors
  20  *    may be used to endorse or promote products derived from this software
  21  *    without specific prior written permission.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  33  * SUCH DAMAGE.
  34  */
  35 #undef n_FILE
  36 #define n_FILE popen
  37
  38 #ifndef HAVE_AMALGAMATION
  39 # include "nail.h"
  40 #endif
  41
  42 #include <sys/wait.h>
  43
  44 #define READ               0
  45 #define WRITE              1
  46
  47 struct fp {
  48    FILE        *fp;
  49    struct fp   *link;
  50    char        *realfile;
  51    char        *save_cmd;
  52    struct termios *fp_tios;
  53    long        offset;
  54    int         omode;
  55    int         pid;
  56    enum {
  57       FP_RAW      = 0,
  58       FP_GZIP     = 1<<0,
  59       FP_XZ       = 1<<1,
  60       FP_BZIP2    = 1<<2,
  61       FP_MAILDIR  = 1<<4,
  62       FP_HOOK     = 1<<5,
  63       FP_PIPE     = 1<<6,
  64       FP_MASK     = (1<<7) - 1,
  65       /* TODO FP_UNLINK: should be in a separated process so that unlinking
  66        * TODO the temporary "garbage" is "safe"(r than it is like that) */
  67       FP_UNLINK   = 1<<9,
  68       FP_TERMIOS  = 1<<10
  69    }           flags;
  70 };
  71
  72 struct child {
  73    int         pid;
  74    char        done;
  75    char        free;
  76    int         status;
  77    struct child *link;
  78 };
  79
  80 static struct fp     *fp_head;
  81 static struct child  *_popen_child;
  82
  83 /* TODO Rather temporary: deal with job control with FD_PASS */
  84 static struct termios a_popen_tios;
  85 static sighandler_type a_popen_otstp, a_popen_ottin, a_popen_ottou;
  86 static volatile int a_popen_hadsig;
  87
  88 static int           scan_mode(char const *mode, int *omode);
  89 static void          register_file(FILE *fp, int omode, int pid,
  90                         int flags, char const *realfile, long offset,
  91                         char const *save_cmd, struct termios *tiosp);
  92 static enum okay     _file_save(struct fp *fpp);
  93 static int           _file_load(int flags, int infd, int outfd,
  94                         char const *load_cmd);
  95 static enum okay     unregister_file(FILE *fp, struct termios **tiosp);
  96 static int           file_pid(FILE *fp);
  97
  98 /* TODO Rather temporary: deal with job control with FD_PASS */
  99 static void a_popen_jobsigs_up(void);
 100 static void a_popen_jobsigs_down(void);
 101 static void a_popen_jobsig(int sig);
 102
 103 /* Handle SIGCHLD */
 104 static void          _sigchld(int signo);
 105
 106 static struct child *_findchild(int pid, bool_t create);
 107 static void          _delchild(struct child *cp);
 108
 109 static int
 110 scan_mode(char const *mode, int *omode)
 111 {
 112    static struct {
 113       char const  mode[4];
 114       int         omode;
 115    } const maps[] = {
 116       {"r", O_RDONLY},
 117       {"w", O_WRONLY | O_CREAT | n_O_NOFOLLOW | O_TRUNC},
 118       {"wx", O_WRONLY | O_CREAT | O_EXCL},
 119       {"a", O_WRONLY | O_APPEND | O_CREAT | n_O_NOFOLLOW},
 120       {"a+", O_RDWR | O_APPEND},
 121       {"r+", O_RDWR},
 122       {"w+", O_RDWR | O_CREAT | O_EXCL}
 123    };
 124
 125    int i;
 126    NYD2_ENTER;
 127
 128    for (i = 0; UICMP(z, i, <, NELEM(maps)); ++i)
 129       if (!strcmp(maps[i].mode, mode)) {
 130          *omode = maps[i].omode;
 131          i = 0;
 132          goto jleave;
 133       }
 134
 135    n_alert(_("Internal error: bad stdio open mode %s"), mode);
 136    errno = EINVAL;
 137    *omode = 0; /* (silence CC) */
 138    i = -1;
 139 jleave:
 140    NYD2_LEAVE;
 141    return i;
 142 }
 143
 144 static void
 145 register_file(FILE *fp, int omode, int pid, int flags,
 146    char const *realfile, long offset, char const *save_cmd,
 147    struct termios *tiosp)
 148 {
 149    struct fp *fpp;
 150    NYD_ENTER;
 151
 152    assert(!(flags & FP_UNLINK) || realfile != NULL);
 153    assert(!(flags & FP_TERMIOS) || tiosp != NULL);
 154
 155    fpp = smalloc(sizeof *fpp);
 156    fpp->fp = fp;
 157    fpp->omode = omode;
 158    fpp->pid = pid;
 159    fpp->link = fp_head;
 160    fpp->flags = flags;
 161    fpp->realfile = (realfile != NULL) ? sstrdup(realfile) : NULL;
 162    fpp->save_cmd = (save_cmd != NULL) ? sstrdup(save_cmd) : NULL;
 163    fpp->fp_tios = tiosp;
 164    fpp->offset = offset;
 165    fp_head = fpp;
 166    NYD_LEAVE;
 167 }
 168
 169 static enum okay
 170 _file_save(struct fp *fpp)
 171 {
 172    char const *cmd[3];
 173    int outfd;
 174    enum okay rv;
 175    NYD_ENTER;
 176
 177    if (fpp->omode == O_RDONLY) {
 178       rv = OKAY;
 179       goto jleave;
 180    }
 181    rv = STOP;
 182
 183    fflush(fpp->fp);
 184    clearerr(fpp->fp);
 185
 186    /* Ensure the I/O library doesn't optimize the fseek(3) away! */
 187    if(!n_real_seek(fpp->fp, fpp->offset, SEEK_SET)){
 188       outfd = errno;
 189       n_err(_("Fatal: cannot restore file position and save %s: %s\n"),
 190          n_shell_quote_cp(fpp->realfile, FAL0), strerror(outfd));
 191       goto jleave;
 192    }
 193
 194    if ((fpp->flags & FP_MASK) == FP_MAILDIR) {
 195       rv = maildir_append(fpp->realfile, fpp->fp, fpp->offset);
 196       goto jleave;
 197    }
 198
 199    outfd = open(fpp->realfile,
 200          ((fpp->omode | O_CREAT | (fpp->omode & O_APPEND ? 0 : O_TRUNC) |
 201             n_O_NOFOLLOW) & ~O_EXCL), 0666);
 202    if (outfd == -1) {
 203       outfd = errno;
 204       n_err(_("Fatal: cannot create %s: %s\n"),
 205          n_shell_quote_cp(fpp->realfile, FAL0), strerror(outfd));
 206       goto jleave;
 207    }
 208
 209    cmd[2] = NULL;
 210    switch (fpp->flags & FP_MASK) {
 211    case FP_GZIP:
 212       cmd[0] = "gzip";  cmd[1] = "-c"; break;
 213    case FP_BZIP2:
 214       cmd[0] = "bzip2"; cmd[1] = "-c"; break;
 215    case FP_XZ:
 216       cmd[0] = "xz";    cmd[1] = "-c"; break;
 217    default:
 218       cmd[0] = "cat";   cmd[1] = NULL; break;
 219    case FP_HOOK:
 220       cmd[0] = ok_vlook(SHELL);
 221       cmd[1] = "-c";
 222       cmd[2] = fpp->save_cmd;
 223    }
 224    if (run_command(cmd[0], 0, fileno(fpp->fp), outfd,
 225          cmd[1], cmd[2], NULL, NULL) >= 0)
 226       rv = OKAY;
 227
 228    close(outfd);
 229 jleave:
 230    NYD_LEAVE;
 231    return rv;
 232 }
 233
 234 static int
 235 _file_load(int flags, int infd, int outfd, char const *load_cmd)
 236 {
 237    char const *cmd[3];
 238    int rv;
 239    NYD_ENTER;
 240
 241    cmd[2] = NULL;
 242    switch (flags & FP_MASK) {
 243    case FP_GZIP:     cmd[0] = "gzip";  cmd[1] = "-cd"; break;
 244    case FP_BZIP2:    cmd[0] = "bzip2"; cmd[1] = "-cd"; break;
 245    case FP_XZ:       cmd[0] = "xz";    cmd[1] = "-cd"; break;
 246    default:          cmd[0] = "cat";   cmd[1] = NULL;  break;
 247    case FP_HOOK:
 248       cmd[0] = ok_vlook(SHELL);
 249       cmd[1] = "-c";
 250       cmd[2] = load_cmd;
 251       break;
 252    case FP_MAILDIR:
 253       rv = 0;
 254       goto jleave;
 255    }
 256
 257    rv = run_command(cmd[0], 0, infd, outfd, cmd[1], cmd[2], NULL, NULL);
 258 jleave:
 259    NYD_LEAVE;
 260    return rv;
 261 }
 262
 263 static enum okay
 264 unregister_file(FILE *fp, struct termios **tiosp)
 265 {
 266    struct fp **pp, *p;
 267    enum okay rv = OKAY;
 268    NYD_ENTER;
 269
 270    if (tiosp)
 271       *tiosp = NULL;
 272
 273    for (pp = &fp_head; (p = *pp) != NULL; pp = &p->link)
 274       if (p->fp == fp) {
 275          switch (p->flags & FP_MASK) {
 276          case FP_RAW:
 277          case FP_PIPE:
 278             break;
 279          default:
 280             rv = _file_save(p);
 281             break;
 282          }
 283          if ((p->flags & FP_UNLINK) && unlink(p->realfile))
 284             rv = STOP;
 285
 286          *pp = p->link;
 287          if (p->save_cmd != NULL)
 288             free(p->save_cmd);
 289          if (p->realfile != NULL)
 290             free(p->realfile);
 291          if (p->flags & FP_TERMIOS) {
 292             if (tiosp != NULL)
 293                *tiosp = p->fp_tios;
 294             else
 295                free(p->fp_tios);
 296          }
 297          free(p);
 298          goto jleave;
 299       }
 300    DBGOR(n_panic, n_alert)(_("Invalid file pointer"));
 301    rv = STOP;
 302 jleave:
 303    NYD_LEAVE;
 304    return rv;
 305 }
 306
 307 static int
 308 file_pid(FILE *fp)
 309 {
 310    int rv;
 311    struct fp *p;
 312    NYD2_ENTER;
 313
 314    rv = -1;
 315    for (p = fp_head; p; p = p->link)
 316       if (p->fp == fp) {
 317          rv = p->pid;
 318          break;
 319       }
 320    NYD2_LEAVE;
 321    return rv;
 322 }
 323
 324 static void
 325 a_popen_jobsigs_up(void){
 326    sigset_t nset, oset;
 327    NYD2_ENTER;
 328
 329    sigfillset(&nset);
 330
 331    sigprocmask(SIG_BLOCK, &nset, &oset);
 332    a_popen_otstp = safe_signal(SIGTSTP, &a_popen_jobsig);
 333    a_popen_ottin = safe_signal(SIGTTIN, &a_popen_jobsig);
 334    a_popen_ottou = safe_signal(SIGTTOU, &a_popen_jobsig);
 335
 336    /* This assumes oset contains nothing but SIGCHLD, so to say */
 337    sigdelset(&oset, SIGTSTP);
 338    sigdelset(&oset, SIGTTIN);
 339    sigdelset(&oset, SIGTTOU);
 340    sigprocmask(SIG_SETMASK, &oset, NULL);
 341    NYD2_LEAVE;
 342 }
 343
 344 static void
 345 a_popen_jobsigs_down(void){
 346    sigset_t nset, oset;
 347    NYD2_ENTER;
 348
 349    sigfillset(&nset);
 350
 351    sigprocmask(SIG_BLOCK, &nset, &oset);
 352    safe_signal(SIGTSTP, a_popen_otstp);
 353    safe_signal(SIGTTIN, a_popen_ottin);
 354    safe_signal(SIGTTOU, a_popen_ottou);
 355
 356    sigaddset(&oset, SIGTSTP);
 357    sigaddset(&oset, SIGTTIN);
 358    sigaddset(&oset, SIGTTOU);
 359    sigprocmask(SIG_SETMASK, &oset, NULL);
 360    NYD2_LEAVE;
 361 }
 362
 363 static void
 364 a_popen_jobsig(int sig){
 365    sighandler_type oldact;
 366    sigset_t nset;
 367    bool_t hadsig;
 368    NYD_X; /* Signal handler */
 369
 370    hadsig = (a_popen_hadsig != 0);
 371    a_popen_hadsig = 1;
 372    if(!hadsig)
 373       n_TERMCAP_SUSPEND(TRU1);
 374
 375    oldact = safe_signal(sig, SIG_DFL);
 376
 377    sigemptyset(&nset);
 378    sigaddset(&nset, sig);
 379    sigprocmask(SIG_UNBLOCK, &nset, NULL);
 380    n_raise(sig);
 381    sigprocmask(SIG_BLOCK, &nset, NULL);
 382
 383    safe_signal(sig, oldact);
 384 }
 385
 386 static void
 387 _sigchld(int signo)
 388 {
 389    pid_t pid;
 390    int status;
 391    struct child *cp;
 392    NYD_X; /* Signal handler */
 393    UNUSED(signo);
 394
 395    for (;;) {
 396       pid = waitpid(-1, &status, WNOHANG);
 397       if (pid <= 0) {
 398          if (pid == -1 && errno == EINTR)
 399             continue;
 400          break;
 401       }
 402
 403       if ((cp = _findchild(pid, FAL0)) != NULL) {
 404          if (cp->free)
 405             cp->pid = -1; /* XXX Was _delchild(cp);# */
 406          else {
 407             cp->done = 1;
 408             cp->status = status;
 409          }
 410       }
 411    }
 412 }
 413
 414 static struct child *
 415 _findchild(int pid, bool_t create)
 416 {
 417    struct child **cpp;
 418    NYD_ENTER;
 419
 420    for (cpp = &_popen_child; *cpp != NULL && (*cpp)->pid != pid;
 421          cpp = &(*cpp)->link)
 422       ;
 423
 424    if (*cpp == NULL && create) {
 425       *cpp = smalloc(sizeof **cpp);
 426       (*cpp)->pid = pid;
 427       (*cpp)->done = (*cpp)->free = 0;
 428       (*cpp)->link = NULL;
 429    }
 430    NYD_LEAVE;
 431    return *cpp;
 432 }
 433
 434 static void
 435 _delchild(struct child *cp)
 436 {
 437    struct child **cpp;
 438    NYD_ENTER;
 439
 440    cpp = &_popen_child;
 441    for (;;) {
 442       if (*cpp == cp) {
 443          *cpp = cp->link;
 444          free(cp);
 445          break;
 446       }
 447       if (*(cpp = &(*cpp)->link) == NULL) {
 448          DBG( n_err("! popen.c:_delchild(): implementation error\n"); )
 449          break;
 450       }
 451    }
 452    NYD_LEAVE;
 453 }
 454
 455 FL void
 456 command_manager_start(void)
 457 {
 458    struct sigaction nact, oact;
 459    NYD_ENTER;
 460
 461    nact.sa_handler = &_sigchld;
 462    sigemptyset(&nact.sa_mask);
 463    nact.sa_flags = SA_RESTART
 464 #ifdef SA_NOCLDSTOP
 465          | SA_NOCLDSTOP
 466 #endif
 467          ;
 468    if (sigaction(SIGCHLD, &nact, &oact) != 0)
 469       n_panic(_("Cannot install signal handler for child process management"));
 470    NYD_LEAVE;
 471 }
 472
 473 FL FILE *
 474 safe_fopen(char const *file, char const *oflags, int *xflags)
 475 {
 476    int osflags, fd;
 477    FILE *fp = NULL;
 478    NYD2_ENTER; /* (only for Fopen() and once in lex.c) */
 479
 480    if (scan_mode(oflags, &osflags) < 0)
 481       goto jleave;
 482    osflags |= _O_CLOEXEC;
 483    if (xflags != NULL)
 484       *xflags = osflags;
 485
 486    if ((fd = open(file, osflags, 0666)) == -1)
 487       goto jleave;
 488    _CLOEXEC_SET(fd);
 489
 490    fp = fdopen(fd, oflags);
 491 jleave:
 492    NYD2_LEAVE;
 493    return fp;
 494 }
 495
 496 FL FILE *
 497 Fopen(char const *file, char const *oflags)
 498 {
 499    FILE *fp;
 500    int osflags;
 501    NYD_ENTER;
 502
 503    if ((fp = safe_fopen(file, oflags, &osflags)) != NULL)
 504       register_file(fp, osflags, 0, FP_RAW, NULL, 0L, NULL, NULL);
 505    NYD_LEAVE;
 506    return fp;
 507 }
 508
 509 FL FILE *
 510 Fdopen(int fd, char const *oflags, bool_t nocloexec)
 511 {
 512    FILE *fp;
 513    int osflags;
 514    NYD_ENTER;
 515
 516    scan_mode(oflags, &osflags);
 517    if (!nocloexec)
 518       osflags |= _O_CLOEXEC; /* Ensured to be set by caller as documented! */
 519
 520    if ((fp = fdopen(fd, oflags)) != NULL)
 521       register_file(fp, osflags, 0, FP_RAW, NULL, 0L, NULL, NULL);
 522    NYD_LEAVE;
 523    return fp;
 524 }
 525
 526 FL int
 527 Fclose(FILE *fp)
 528 {
 529    int i = 0;
 530    NYD_ENTER;
 531
 532    if (unregister_file(fp, NULL) == OKAY)
 533       i |= 1;
 534    if (fclose(fp) == 0)
 535       i |= 2;
 536    NYD_LEAVE;
 537    return (i == 3 ? 0 : EOF);
 538 }
 539
 540 FL FILE *
 541 Zopen(char const *file, char const *oflags) /* FIXME MESS! */
 542 {
 543    FILE *rv = NULL;
 544    char const *cload = NULL, *csave = NULL;
 545    int flags, osflags, mode, infd;
 546    enum oflags rof;
 547    long offset;
 548    enum protocol p;
 549    NYD_ENTER;
 550
 551    if (scan_mode(oflags, &osflags) < 0)
 552       goto jleave;
 553
 554    flags = 0;
 555    rof = OF_RDWR | OF_UNLINK;
 556    if (osflags & O_APPEND)
 557       rof |= OF_APPEND;
 558    mode = (osflags == O_RDONLY) ? R_OK : R_OK | W_OK;
 559
 560    if ((osflags & O_APPEND) && ((p = which_protocol(file)) == PROTO_MAILDIR)) {
 561       flags |= FP_MAILDIR;
 562       osflags = O_RDWR | O_APPEND | O_CREAT | n_O_NOFOLLOW;
 563       infd = -1;
 564    } else {
 565       char const *ext;
 566
 567       if ((ext = strrchr(file, '.')) != NULL) {
 568          if (!asccasecmp(ext, ".gz"))
 569             flags |= FP_GZIP;
 570          else if (!asccasecmp(ext, ".xz")) {
 571             flags |= FP_XZ;
 572             osflags &= ~O_APPEND;
 573             rof &= ~OF_APPEND;
 574          } else if (!asccasecmp(ext, ".bz2")) {
 575             flags |= FP_BZIP2;
 576             osflags &= ~O_APPEND;
 577             rof &= ~OF_APPEND;
 578          } else {
 579 #undef _X1
 580 #define _X1 "file-hook-load-"
 581 #undef _X2
 582 #define _X2 "file-hook-save-"
 583             size_t l = strlen(++ext);
 584             char *vbuf = ac_alloc(l + MAX(sizeof(_X1), sizeof(_X2)));
 585
 586             memcpy(vbuf, _X1, sizeof(_X1) -1);
 587             memcpy(vbuf + sizeof(_X1) -1, ext, l);
 588             vbuf[sizeof(_X1) -1 + l] = '\0';
 589             cload = vok_vlook(vbuf);
 590             memcpy(vbuf, _X2, sizeof(_X2) -1);
 591             memcpy(vbuf + sizeof(_X2) -1, ext, l);
 592             vbuf[sizeof(_X2) -1 + l] = '\0';
 593             csave = vok_vlook(vbuf);
 594 #undef _X2
 595 #undef _X1
 596             ac_free(vbuf);
 597
 598             if ((csave != NULL) && (cload != NULL)) {
 599                flags |= FP_HOOK;
 600                osflags &= ~O_APPEND;
 601                rof &= ~OF_APPEND;
 602             } else if ((csave != NULL) | (cload != NULL)) {
 603                n_alert(_("Only one of *mailbox-(load|save)-%s* is set!  "
 604                   "Treating as plain text!"), ext);
 605                goto jraw;
 606             } else
 607                goto jraw;
 608          }
 609       } else {
 610 jraw:
 611          /*flags |= FP_RAW;*/
 612          rv = Fopen(file, oflags);
 613          goto jleave;
 614       }
 615
 616       if ((infd = open(file, (mode & W_OK) ? O_RDWR : O_RDONLY)) == -1 &&
 617             (!(osflags & O_CREAT) || errno != ENOENT))
 618          goto jleave;
 619    }
 620
 621    /* Note rv is not yet register_file()d, fclose() it in error path! */
 622    if ((rv = Ftmp(NULL, "zopen", rof)) == NULL) {
 623       n_perr(_("tmpfile"), 0);
 624       goto jerr;
 625    }
 626
 627    if (flags & FP_MAILDIR)
 628       ;
 629    else if (infd >= 0) {
 630       if (_file_load(flags, infd, fileno(rv), cload) < 0) {
 631 jerr:
 632          if (rv != NULL)
 633             fclose(rv);
 634          rv = NULL;
 635          if (infd >= 0)
 636             close(infd);
 637          goto jleave;
 638       }
 639    } else {
 640       if ((infd = creat(file, 0666)) == -1) {
 641          fclose(rv);
 642          rv = NULL;
 643          goto jleave;
 644       }
 645    }
 646
 647    if (infd >= 0)
 648       close(infd);
 649    fflush(rv);
 650
 651    if (!(osflags & O_APPEND))
 652       rewind(rv);
 653    if ((offset = ftell(rv)) == -1) {
 654       Fclose(rv);
 655       rv = NULL;
 656       goto jleave;
 657    }
 658
 659    register_file(rv, osflags, 0, flags, file, offset, csave, NULL);
 660 jleave:
 661    NYD_LEAVE;
 662    return rv;
 663 }
 664
 665 FL FILE *
 666 Ftmp(char **fn, char const *namehint, enum oflags oflags)
 667 {
 668    /* The 8 is arbitrary but leaves room for a six character suffix (the
 669     * POSIX minimum path length is 14, though we don't check that XXX).
 670     * 8 should be more than sufficient given that we use base64url encoding
 671     * for our random string */
 672    enum {_RANDCHARS = 8};
 673
 674    size_t maxname, xlen, i;
 675    char *cp_base, *cp;
 676    int osoflags, fd, e;
 677    bool_t relesigs;
 678    FILE *fp;
 679    NYD_ENTER;
 680
 681    assert(namehint != NULL);
 682    assert((oflags & OF_WRONLY) || (oflags & OF_RDWR));
 683    assert(!(oflags & OF_RDONLY));
 684    assert(!(oflags & OF_REGISTER_UNLINK) || (oflags & OF_REGISTER));
 685
 686    fp = NULL;
 687    relesigs = FAL0;
 688    e = 0;
 689    maxname = NAME_MAX;
 690 #ifdef HAVE_PATHCONF
 691    {  long pc;
 692
 693       if ((pc = pathconf(tempdir, _PC_NAME_MAX)) != -1)
 694          maxname = (size_t)pc;
 695    }
 696 #endif
 697
 698    if ((oflags & OF_SUFFIX) && *namehint != '\0') {
 699       if ((xlen = strlen(namehint)) > maxname - _RANDCHARS) {
 700          errno = ENAMETOOLONG;
 701          goto jleave;
 702       }
 703    } else
 704       xlen = 0;
 705
 706    /* Prepare the template string once, then iterate over the random range */
 707    cp_base =
 708    cp = smalloc(strlen(tempdir) + 1 + maxname +1);
 709    cp = sstpcpy(cp, tempdir);
 710    *cp++ = '/';
 711    {
 712       char *x = sstpcpy(cp, UAGENT);
 713       *x++ = '-';
 714       if (!(oflags & OF_SUFFIX))
 715          x = sstpcpy(x, namehint);
 716
 717       i = PTR2SIZE(x - cp);
 718       if (i > maxname - xlen - _RANDCHARS) {
 719          size_t j = maxname - xlen - _RANDCHARS;
 720          x -= i - j;
 721          i = j;
 722       }
 723
 724       if ((oflags & OF_SUFFIX) && xlen > 0)
 725          memcpy(x + _RANDCHARS, namehint, xlen);
 726
 727       x[xlen + _RANDCHARS] = '\0';
 728       cp = x;
 729    }
 730
 731    osoflags = O_CREAT | O_EXCL | _O_CLOEXEC;
 732    osoflags |= (oflags & OF_WRONLY) ? O_WRONLY : O_RDWR;
 733    if (oflags & OF_APPEND)
 734       osoflags |= O_APPEND;
 735
 736    for (i = 0;; ++i) {
 737       memcpy(cp, getrandstring(_RANDCHARS), _RANDCHARS);
 738
 739       hold_all_sigs();
 740       relesigs = TRU1;
 741
 742       if ((fd = open(cp_base, osoflags, 0600)) != -1) {
 743          _CLOEXEC_SET(fd);
 744          break;
 745       }
 746       if (i >= FTMP_OPEN_TRIES) {
 747          e = errno;
 748          goto jfree;
 749       }
 750       relesigs = FAL0;
 751       rele_all_sigs();
 752    }
 753
 754    if (oflags & OF_REGISTER) {
 755       char const *osflags = (oflags & OF_RDWR ? "w+" : "w");
 756       int osflagbits;
 757
 758       scan_mode(osflags, &osflagbits); /* TODO osoflags&xy ?!!? */
 759       if ((fp = fdopen(fd, osflags)) != NULL)
 760          register_file(fp, osflagbits | _O_CLOEXEC, 0,
 761             (FP_RAW | (oflags & OF_REGISTER_UNLINK ? FP_UNLINK : 0)),
 762             cp_base, 0L, NULL, NULL);
 763    } else
 764       fp = fdopen(fd, (oflags & OF_RDWR ? "w+" : "w"));
 765
 766    if (fp == NULL || (oflags & OF_UNLINK)) {
 767       e = errno;
 768       unlink(cp_base);
 769       goto jfree;
 770    }
 771
 772    if (fn != NULL)
 773       *fn = cp_base;
 774    else
 775       free(cp_base);
 776 jleave:
 777    if (relesigs && (fp == NULL || !(oflags & OF_HOLDSIGS)))
 778       rele_all_sigs();
 779    if (fp == NULL)
 780       errno = e;
 781    NYD_LEAVE;
 782    return fp;
 783 jfree:
 784    if ((cp = cp_base) != NULL)
 785       free(cp);
 786    goto jleave;
 787 }
 788
 789 FL void
 790 Ftmp_release(char **fn)
 791 {
 792    char *cp;
 793    NYD_ENTER;
 794
 795    cp = *fn;
 796    *fn = NULL;
 797    if (cp != NULL) {
 798       unlink(cp);
 799       rele_all_sigs();
 800       free(cp);
 801    }
 802    NYD_LEAVE;
 803 }
 804
 805 FL void
 806 Ftmp_free(char **fn) /* TODO DROP: OF_REGISTER_FREEPATH! */
 807 {
 808    char *cp;
 809    NYD_ENTER;
 810
 811    cp = *fn;
 812    *fn = NULL;
 813    if (cp != NULL)
 814       free(cp);
 815    NYD_LEAVE;
 816 }
 817
 818 FL bool_t
 819 pipe_cloexec(int fd[2])
 820 {
 821    bool_t rv = FAL0;
 822    NYD_ENTER;
 823
 824 #ifdef HAVE_PIPE2
 825    if (pipe2(fd, O_CLOEXEC) == -1)
 826       goto jleave;
 827 #else
 828    if (pipe(fd) == -1)
 829       goto jleave;
 830    (void)fcntl(fd[0], F_SETFD, FD_CLOEXEC);
 831    (void)fcntl(fd[1], F_SETFD, FD_CLOEXEC);
 832 #endif
 833    rv = TRU1;
 834 jleave:
 835    NYD_LEAVE;
 836    return rv;
 837 }
 838
 839 FL FILE *
 840 Popen(char const *cmd, char const *mode, char const *sh,
 841    char const **env_addon, int newfd1)
 842 {
 843    struct termios *tiosp;
 844    int p[2], myside, hisside, fd0, fd1, pid;
 845    char mod[2] = {'0', '\0'};
 846    sigset_t nset;
 847    FILE *rv = NULL;
 848    NYD_ENTER;
 849
 850    /* First clean up child structures */
 851    {  sigset_t oset;
 852       struct child **cpp, *cp;
 853
 854       sigfillset(&nset);
 855       sigprocmask(SIG_BLOCK, &nset, &oset);
 856
 857       for (cpp = &_popen_child; *cpp != NULL;) {
 858          if ((*cpp)->pid == -1) {
 859             cp = *cpp;
 860             *cpp = cp->link;
 861             free(cp);
 862          } else
 863             cpp = &(*cpp)->link;
 864       }
 865
 866       sigprocmask(SIG_SETMASK, &oset, NULL);
 867    }
 868
 869    if (!pipe_cloexec(p))
 870       goto jleave;
 871
 872    if (*mode == 'r') {
 873       myside = p[READ];
 874       fd0 = COMMAND_FD_PASS;
 875       hisside = fd1 = p[WRITE];
 876       mod[0] = *mode;
 877    } else if (*mode == 'W') {
 878       myside = p[WRITE];
 879       hisside = fd0 = p[READ];
 880       fd1 = newfd1;
 881       mod[0] = 'w';
 882    } else {
 883       myside = p[WRITE];
 884       hisside = fd0 = p[READ];
 885       fd1 = COMMAND_FD_PASS;
 886       mod[0] = 'w';
 887    }
 888
 889    /* In interactive mode both STDIN and STDOUT point to the terminal.  If we
 890     * pass through the TTY restore terminal attributes after pipe returns.
 891     * XXX It shouldn't matter which FD we actually use in this case */
 892    if ((options & OPT_INTERACTIVE) && (fd0 == COMMAND_FD_PASS ||
 893          fd1 == COMMAND_FD_PASS)) {
 894       tiosp = smalloc(sizeof *tiosp);
 895       tcgetattr(STDIN_FILENO, tiosp);
 896       n_TERMCAP_SUSPEND(TRU1);
 897    } else
 898       tiosp = NULL;
 899
 900    sigemptyset(&nset);
 901
 902    if (cmd == (char*)-1) {
 903       if ((pid = fork_child()) == -1)
 904          n_perr(_("fork"), 0);
 905       else if (pid == 0) {
 906          union {char const *ccp; int (*ptf)(void); int es;} u;
 907          prepare_child(&nset, fd0, fd1);
 908          close(p[READ]);
 909          close(p[WRITE]);
 910          u.ccp = sh;
 911          u.es = (*u.ptf)();
 912          _exit(u.es);
 913       }
 914    } else if (sh == NULL) {
 915       pid = start_command(cmd, &nset, fd0, fd1, NULL, NULL, NULL, env_addon);
 916    } else {
 917       pid = start_command(sh, &nset, fd0, fd1, "-c", cmd, NULL, env_addon);
 918    }
 919    if (pid < 0) {
 920       close(p[READ]);
 921       close(p[WRITE]);
 922       goto jleave;
 923    }
 924    close(hisside);
 925    if ((rv = fdopen(myside, mod)) != NULL)
 926       register_file(rv, 0, pid,
 927          (tiosp == NULL ? FP_PIPE : FP_PIPE | FP_TERMIOS),
 928          NULL, 0L, NULL, tiosp);
 929    else
 930       close(myside);
 931 jleave:
 932    NYD_LEAVE;
 933    return rv;
 934 }
 935
 936 FL bool_t
 937 Pclose(FILE *ptr, bool_t dowait)
 938 {
 939    struct termios *tiosp;
 940    sigset_t nset, oset;
 941    int pid;
 942    bool_t rv = FAL0;
 943    NYD_ENTER;
 944
 945    pid = file_pid(ptr);
 946    if (pid < 0)
 947       goto jleave;
 948
 949    unregister_file(ptr, &tiosp);
 950    fclose(ptr);
 951
 952    if (dowait) {
 953       sigemptyset(&nset);
 954       sigaddset(&nset, SIGINT);
 955       sigaddset(&nset, SIGHUP);
 956       sigprocmask(SIG_BLOCK, &nset, &oset);
 957       rv = wait_child(pid, NULL);
 958       if (tiosp != NULL) {
 959          n_TERMCAP_RESUME(TRU1);
 960          tcsetattr(STDIN_FILENO, TCSAFLUSH, tiosp);
 961       }
 962       sigprocmask(SIG_SETMASK, &oset, NULL);
 963    } else {
 964       free_child(pid);
 965       rv = TRU1;
 966    }
 967    if (tiosp != NULL)
 968       free(tiosp);
 969 jleave:
 970    NYD_LEAVE;
 971    return rv;
 972 }
 973
 974 FL FILE *
 975 n_pager_open(void)
 976 {
 977    char const *env_add[2], *pager;
 978    FILE *rv;
 979    NYD_ENTER;
 980
 981    assert(options & OPT_INTERACTIVE);
 982
 983    pager = n_pager_get(env_add + 0);
 984    env_add[1] = NULL;
 985
 986    if ((rv = Popen(pager, "w", NULL, env_add, COMMAND_FD_PASS)) == NULL)
 987       n_perr(pager, 0);
 988    NYD_LEAVE;
 989    return rv;
 990 }
 991
 992 FL bool_t
 993 n_pager_close(FILE *fp)
 994 {
 995    sighandler_type sh;
 996    bool_t rv;
 997    NYD_ENTER;
 998
 999    sh = safe_signal(SIGPIPE, SIG_IGN);
1000    rv = Pclose(fp, TRU1);
1001    safe_signal(SIGPIPE, sh);
1002    NYD_LEAVE;
1003    return rv;
1004 }
1005
1006 FL void
1007 close_all_files(void)
1008 {
1009    NYD_ENTER;
1010    while (fp_head != NULL)
1011       if ((fp_head->flags & FP_MASK) == FP_PIPE)
1012          Pclose(fp_head->fp, TRU1);
1013       else
1014          Fclose(fp_head->fp);
1015    NYD_LEAVE;
1016 }
1017
1018 FL int
1019 fork_child(void)
1020 {
1021    struct child *cp;
1022    int pid;
1023    NYD_ENTER;
1024
1025    cp = _findchild(0, TRU1);
1026
1027    if ((cp->pid = pid = fork()) == -1) {
1028       _delchild(cp);
1029       n_perr(_("fork"), 0);
1030    }
1031    NYD_LEAVE;
1032    return pid;
1033 }
1034
1035 FL int
1036 run_command(char const *cmd, sigset_t *mask, int infd, int outfd,
1037    char const *a0, char const *a1, char const *a2, char const **env_addon)
1038 {
1039    sigset_t nset, oset;
1040    bool_t tio_set;
1041    int rv;
1042    NYD_ENTER;
1043
1044    /* TODO Of course this is a joke given that during a "p*" the PAGER may
1045     * TODO be up and running while we play around like this... but i guess
1046     * TODO this can't be helped at all unless we perform complete and true
1047     * TODO process group separation and ensure we don't deadlock us out
1048     * TODO via TTY jobcontrol signal storms (could this really happen?).
1049     * TODO Or have a builtin pager.  Or query any necessity BEFORE we start
1050     * TODO any action, and shall we find we need to run programs dump it
1051     * TODO all into a temporary file which is then passed through to the
1052     * TODO PAGER.  Ugh.  That still won't help for "needsterminal" anyway */
1053    if ((tio_set = ((options & OPT_INTERACTIVE) &&
1054          (infd == COMMAND_FD_PASS || outfd == COMMAND_FD_PASS)))) {
1055       tcgetattr((options & OPT_TTYIN ? STDIN_FILENO : STDOUT_FILENO), &a_popen_tios);
1056       n_TERMCAP_SUSPEND(FAL0);
1057       sigfillset(&nset);
1058       sigdelset(&nset, SIGCHLD);
1059       /* sigdelset(&nset, SIGPIPE); TODO would need a handler */
1060       sigprocmask(SIG_BLOCK, &nset, &oset);
1061       a_popen_hadsig = 0;
1062       a_popen_jobsigs_up();
1063    }
1064
1065    if ((rv = start_command(cmd, mask, infd, outfd, a0, a1, a2, env_addon)) < 0)
1066       rv = -1;
1067    else {
1068       if (wait_child(rv, NULL))
1069          rv = 0;
1070       else {
1071          if (ok_blook(bsdcompat) || ok_blook(bsdmsgs))
1072             n_err(_("Fatal error in process\n"));
1073          rv = -1;
1074       }
1075    }
1076
1077    if (tio_set) {
1078       a_popen_jobsigs_down();
1079       tio_set = ((options & OPT_TTYIN) != 0);
1080       n_TERMCAP_RESUME(a_popen_hadsig ? TRU1 : FAL0);
1081       tcsetattr((tio_set ? STDIN_FILENO : STDOUT_FILENO),
1082          (tio_set ? TCSAFLUSH : TCSADRAIN), &a_popen_tios);
1083       sigprocmask(SIG_SETMASK, &oset, NULL);
1084    }
1085    NYD_LEAVE;
1086    return rv;
1087 }
1088
1089 FL int
1090 start_command(char const *cmd, sigset_t *mask, int infd, int outfd,
1091    char const *a0, char const *a1, char const *a2,
1092    char const **env_addon)
1093 {
1094    int rv;
1095    NYD_ENTER;
1096
1097    if ((rv = fork_child()) == -1) {
1098       n_perr(_("fork"), 0);
1099       rv = -1;
1100    } else if (rv == 0) {
1101       char *argv[128];
1102       int i;
1103
1104       if (env_addon != NULL) { /* TODO env_addon; should have struct child */
1105          extern char **environ;
1106          size_t ei, ei_orig, ai, ai_orig;
1107          char **env;
1108
1109          /* TODO note we don't check the POSIX limit:
1110           * the total space used to store the environment and the arguments to
1111           * the process is limited to {ARG_MAX} bytes */
1112          for (ei = 0; environ[ei] != NULL; ++ei)
1113             ;
1114          ei_orig = ei;
1115          for (ai = 0; env_addon[ai] != NULL; ++ai)
1116             ;
1117          ai_orig = ai;
1118          env = ac_alloc(sizeof(*env) * (ei + ai +1));
1119          memcpy(env, environ, sizeof(*env) * ei);
1120
1121          /* Replace all those keys that yet exist */
1122          while (ai-- > 0) {
1123             char const *ee, *kvs;
1124             size_t kl;
1125
1126             ee = env_addon[ai];
1127             kvs = strchr(ee, '=');
1128             assert(kvs != NULL);
1129             kl = PTR2SIZE(kvs - ee);
1130             assert(kl > 0);
1131             for (ei = ei_orig; ei-- > 0;) {
1132                char const *ekvs = strchr(env[ei], '=');
1133                if (ekvs != NULL && kl == PTR2SIZE(ekvs - env[ei]) &&
1134                      !memcmp(ee, env[ei], kl)) {
1135                   env[ei] = UNCONST(ee);
1136                   env_addon[ai] = NULL;
1137                   break;
1138                }
1139             }
1140          }
1141
1142          /* And append the rest */
1143          for (ei = ei_orig, ai = ai_orig; ai-- > 0;)
1144             if (env_addon[ai] != NULL)
1145                env[ei++] = UNCONST(env_addon[ai]);
1146
1147          env[ei] = NULL;
1148          environ = env;
1149       }
1150
1151       i = (int)getrawlist(FAL0, argv, NELEM(argv), cmd, strlen(cmd));
1152
1153       if ((argv[i++] = UNCONST(a0)) != NULL &&
1154             (argv[i++] = UNCONST(a1)) != NULL &&
1155             (argv[i++] = UNCONST(a2)) != NULL)
1156          argv[i] = NULL;
1157       prepare_child(mask, infd, outfd);
1158       execvp(argv[0], argv);
1159       perror(argv[0]);
1160       _exit(EXIT_ERR);
1161    }
1162    NYD_LEAVE;
1163    return rv;
1164 }
1165
1166 FL void
1167 prepare_child(sigset_t *nset, int infd, int outfd)
1168 {
1169    int i;
1170    sigset_t fset;
1171    NYD_ENTER;
1172
1173    /* All file descriptors other than 0, 1, and 2 are supposed to be cloexec */
1174    /* TODO WHAT IS WITH STDERR_FILENO DAMN? */
1175    if ((i = (infd == COMMAND_FD_NULL)))
1176       infd = open("/dev/null", O_RDONLY);
1177    if (infd >= 0) {
1178       dup2(infd, STDIN_FILENO);
1179       if (i)
1180          close(infd);
1181    }
1182
1183    if ((i = (outfd == COMMAND_FD_NULL)))
1184       outfd = open("/dev/null", O_WRONLY);
1185    if (outfd >= 0) {
1186       dup2(outfd, STDOUT_FILENO);
1187       if (i)
1188          close(outfd);
1189    }
1190
1191    if (nset) {
1192       for (i = 1; i < NSIG; ++i)
1193          if (sigismember(nset, i))
1194             safe_signal(i, SIG_IGN);
1195       if (!sigismember(nset, SIGINT))
1196          safe_signal(SIGINT, SIG_DFL);
1197    }
1198
1199    sigemptyset(&fset);
1200    sigprocmask(SIG_SETMASK, &fset, NULL);
1201    NYD_LEAVE;
1202 }
1203
1204 FL void
1205 free_child(int pid)
1206 {
1207    sigset_t nset, oset;
1208    struct child *cp;
1209    NYD_ENTER;
1210
1211    sigemptyset(&nset);
1212    sigaddset(&nset, SIGCHLD);
1213    sigprocmask(SIG_BLOCK, &nset, &oset);
1214
1215    if ((cp = _findchild(pid, FAL0)) != NULL) {
1216       if (cp->done)
1217          _delchild(cp);
1218       else
1219          cp->free = 1;
1220    }
1221
1222    sigprocmask(SIG_SETMASK, &oset, NULL);
1223    NYD_LEAVE;
1224 }
1225
1226 FL bool_t
1227 wait_child(int pid, int *wait_status)
1228 {
1229    sigset_t nset, oset;
1230    struct child *cp;
1231    int ws;
1232    bool_t rv;
1233    NYD_ENTER;
1234
1235    sigemptyset(&nset);
1236    sigaddset(&nset, SIGCHLD);
1237    sigprocmask(SIG_BLOCK, &nset, &oset);
1238
1239    cp = _findchild(pid, FAL0);
1240    if (cp != NULL) {
1241       while (!cp->done)
1242          sigsuspend(&oset);
1243       ws = cp->status;
1244       _delchild(cp);
1245    } else
1246       ws = 0;
1247
1248    sigprocmask(SIG_SETMASK, &oset, NULL);
1249
1250    if (wait_status != NULL)
1251       *wait_status = ws;
1252    rv = (WIFEXITED(ws) && WEXITSTATUS(ws) == 0);
1253    NYD_LEAVE;
1254    return rv;
1255 }
1256
1257 /* s-it-mode */