1 #include "fd.h"

     2 #include "stralloc.h"

     3 #include "readwrite.h"

     4 #include "sgetopt.h"

     5 #include "wait.h"

     6 #include "env.h"

     7 #include "byte.h"

     8 #include "str.h"

     9 #include "alloc.h"

    10 #include "exit.h"

    11 #include "fork.h"

    12 #include "case.h"

    13 #include "subfd.h"

    14 #include "error.h"

    15 #include "substdio.h"

    16 #include "sig.h"

    17 

    18 void die(e,s) int e; char *s; { substdio_putsflush(subfderr,s); _exit(e); }

    19 void die_usage() { die(100,"qsmhook: fatal: incorrect usage\n"); }

    20 void die_temp() { die(111,"qsmhook: fatal: temporary problem\n"); }

    21 void die_read() { die(111,"qsmhook: fatal: unable to read message\n"); }

    22 void die_badcmd() { die(100,"qsmhook: fatal: command not found\n"); }

    23 

    24 int flagrpline = 0; char *rpline;

    25 int flagufline = 1; char *ufline;

    26 int flagdtline = 0; char *dtline;

    27 char *host;

    28 char *sender;

    29 char *recip;

    30 

    31 stralloc newarg = {0};

    32 

    33 substdio ssout;

    34 char outbuf[SUBSTDIO_OUTSIZE];

    35 substdio ssin;

    36 char inbuf[SUBSTDIO_INSIZE];

    37 

    38 void main(argc,argv)

    39 int argc;

    40 char **argv;

    41 {

    42  int pid;

    43  int wstat;

    44  int pi[2];

    45  int opt;

    46  char **arg;

    47  char *x;

    48  int i;

    49  int flagesc;

    50 

    51  sig_pipeignore();

    52 

    53  if (!(dtline = env_get("DTLINE"))) die_usage();

    54  if (!(rpline = env_get("RPLINE"))) die_usage();

    55  if (!(ufline = env_get("UFLINE"))) die_usage();

    56  if (!(recip = env_get("LOCAL"))) die_usage();

    57  if (!(host = env_get("HOST"))) die_usage();

    58  if (!(sender = env_get("SENDER"))) die_usage();

    59 

    60  while ((opt = getopt(argc,argv,"DFlMmnPsx:")) != opteof)

    61    switch(opt)

    62     {

    63      case 'D': case 'F': case 'M': break; /* be serious */

    64      case 'l': flagdtline = 1; break; /* also return-receipt-to? blech */

    65      case 'm': break; /* we only handle one recipient anyway */

    66      case 'n': flagufline = 0; break;

    67      case 's': break; /* could call quote() otherwise, i suppose... */

    68      case 'P': flagrpline = 1; break;

    69      case 'x':

    70        if (case_starts(recip,optarg))

    71 	 recip += str_len(optarg);

    72        break;

    73      default:

    74        _exit(100);

    75     }

    76  argc -= optind;

    77  argv += optind;

    78 

    79  if (!*argv) die_usage();

    80 

    81  for (arg = argv;x = *arg;++arg)

    82   {

    83    if (!stralloc_copys(&newarg,"")) die_temp();

    84    flagesc = 0;

    85    for (i = 0;x[i];++i)

    86      if (flagesc)

    87       {

    88        switch(x[i])

    89 	{

    90 	 case '%': if (!stralloc_cats(&newarg,"%")) die_temp(); break;

    91 	 case 'g': if (!stralloc_cats(&newarg,sender)) die_temp(); break;

    92 	 case 'h': if (!stralloc_cats(&newarg,host)) die_temp(); break;

    93 	 case 'u': if (!stralloc_cats(&newarg,recip)) die_temp(); break;

    94 	}

    95        flagesc = 0;

    96       }

    97      else

    98        if (x[i] == '%')

    99 	 flagesc = 1;

   100        else

   101 	 if (!stralloc_append(&newarg,&x[i])) die_temp();

   102    if (!stralloc_0(&newarg)) die_temp();

   103    i = str_len(newarg.s) + 1;

   104    if (!(x = alloc(i))) die_temp();

   105    byte_copy(x,i,newarg.s);

   106    *arg = x;

   107   }

   108 

   109  if (pipe(pi) == -1) die_temp();

   110 

   111  switch(pid = fork())

   112   {

   113    case -1:

   114      die_temp();

   115    case 0:

   116      close(pi[1]);

   117      if (fd_move(0,pi[0]) == -1) die_temp();

   118      sig_pipedefault();

   119      execvp(*argv,argv);

   120      if (error_temp(errno)) die_temp();

   121      die_badcmd();

   122   }

   123  close(pi[0]);

   124 

   125  substdio_fdbuf(&ssout,write,pi[1],outbuf,sizeof(outbuf));

   126  substdio_fdbuf(&ssin,read,0,inbuf,sizeof(inbuf));

   127  if (flagufline) substdio_bputs(&ssout,ufline);

   128  if (flagrpline) substdio_bputs(&ssout,rpline);

   129  if (flagdtline) substdio_bputs(&ssout,dtline);

   130  if (substdio_copy(&ssout,&ssin) == -2) die_read();

   131  substdio_flush(&ssout);

   132  close(pi[1]);

   133 

   134  if (wait_pid(&wstat,pid) == -1) die_temp();

   135  if (wait_crashed(wstat)) die_temp();

   136  _exit(wait_exitcode(wstat));

   137 }