chiark - git - mdw - qmail/blob - preline.c

   1 #include "fd.h"
   2 #include "sgetopt.h"
   3 #include "readwrite.h"
   4 #include "strerr.h"
   5 #include "substdio.h"
   6 #include "exit.h"
   7 #include "fork.h"
   8 #include "wait.h"
   9 #include "env.h"
  10 #include "sig.h"
  11 #include "error.h"
  12
  13 #define FATAL "preline: fatal: "
  14
  15 void die_usage()
  16 {
  17   strerr_die1x(100,"preline: usage: preline cmd [ arg ... ]");
  18 }
  19
  20 int flagufline = 1; char *ufline;
  21 int flagrpline = 1; char *rpline;
  22 int flagdtline = 1; char *dtline;
  23
  24 char outbuf[SUBSTDIO_OUTSIZE];
  25 char inbuf[SUBSTDIO_INSIZE];
  26 substdio ssout = SUBSTDIO_FDBUF(write,1,outbuf,sizeof outbuf);
  27 substdio ssin = SUBSTDIO_FDBUF(read,0,inbuf,sizeof inbuf);
  28
  29 void main(argc,argv)
  30 int argc;
  31 char **argv;
  32 {
  33   int opt;
  34   int pi[2];
  35   int pid;
  36   int wstat;
  37
  38   sig_pipeignore();
  39
  40   if (!(ufline = env_get("UFLINE"))) die_usage();
  41   if (!(rpline = env_get("RPLINE"))) die_usage();
  42   if (!(dtline = env_get("DTLINE"))) die_usage();
  43
  44   while ((opt = getopt(argc,argv,"frdFRD")) != opteof)
  45     switch(opt) {
  46       case 'f': flagufline = 0; break;
  47       case 'r': flagrpline = 0; break;
  48       case 'd': flagdtline = 0; break;
  49       case 'F': flagufline = 1; break;
  50       case 'R': flagrpline = 1; break;
  51       case 'D': flagdtline = 1; break;
  52       default: die_usage();
  53     }
  54   argc -= optind;
  55   argv += optind;
  56   if (!*argv) die_usage();
  57
  58   if (pipe(pi) == -1)
  59     strerr_die2sys(111,FATAL,"unable to create pipe: ");
  60
  61   pid = fork();
  62   if (pid == -1)
  63     strerr_die2sys(111,FATAL,"unable to fork: ");
  64
  65   if (pid == 0) {
  66     close(pi[1]);
  67     if (fd_move(0,pi[0]) == -1)
  68       strerr_die2sys(111,FATAL,"unable to set up fds: ");
  69     sig_pipedefault();
  70     execvp(*argv,argv);
  71     strerr_die4sys(error_temp(errno) ? 111 : 100,FATAL,"unable to run ",*argv,": ");
  72   }
  73   close(pi[0]);
  74   if (fd_move(1,pi[1]) == -1)
  75     strerr_die2sys(111,FATAL,"unable to set up fds: ");
  76
  77   if (flagufline) substdio_bputs(&ssout,ufline);
  78   if (flagrpline) substdio_bputs(&ssout,rpline);
  79   if (flagdtline) substdio_bputs(&ssout,dtline);
  80   if (substdio_copy(&ssout,&ssin) != 0)
  81     strerr_die2sys(111,FATAL,"unable to copy input: ");
  82   substdio_flush(&ssout);
  83   close(1);
  84
  85   if (wait_pid(&wstat,pid) == -1)
  86     strerr_die2sys(111,FATAL,"wait failed: ");
  87   if (wait_crashed(wstat))
  88     strerr_die2x(111,FATAL,"child crashed");
  89   _exit(wait_exitcode(wstat));
  90 }