137 lines
3.4 KiB
C
137 lines
3.4 KiB
C
#include "fd.h"
|
|
#include "stralloc.h"
|
|
#include "readwrite.h"
|
|
#include "sgetopt.h"
|
|
#include "wait.h"
|
|
#include "env.h"
|
|
#include "byte.h"
|
|
#include "str.h"
|
|
#include "alloc.h"
|
|
#include "exit.h"
|
|
#include "fork.h"
|
|
#include "case.h"
|
|
#include "subfd.h"
|
|
#include "error.h"
|
|
#include "substdio.h"
|
|
#include "sig.h"
|
|
|
|
void die(e,s) int e; char *s; { substdio_putsflush(subfderr,s); _exit(e); }
|
|
void die_usage() { die(100,"qsmhook: fatal: incorrect usage\n"); }
|
|
void die_temp() { die(111,"qsmhook: fatal: temporary problem\n"); }
|
|
void die_read() { die(111,"qsmhook: fatal: unable to read message\n"); }
|
|
void die_badcmd() { die(100,"qsmhook: fatal: command not found\n"); }
|
|
|
|
int flagrpline = 0; char *rpline;
|
|
int flagufline = 1; char *ufline;
|
|
int flagdtline = 0; char *dtline;
|
|
char *host;
|
|
char *sender;
|
|
char *recip;
|
|
|
|
stralloc newarg = {0};
|
|
|
|
substdio ssout;
|
|
char outbuf[SUBSTDIO_OUTSIZE];
|
|
substdio ssin;
|
|
char inbuf[SUBSTDIO_INSIZE];
|
|
|
|
void main(argc,argv)
|
|
int argc;
|
|
char **argv;
|
|
{
|
|
int pid;
|
|
int wstat;
|
|
int pi[2];
|
|
int opt;
|
|
char **arg;
|
|
char *x;
|
|
int i;
|
|
int flagesc;
|
|
|
|
sig_pipeignore();
|
|
|
|
if (!(dtline = env_get("DTLINE"))) die_usage();
|
|
if (!(rpline = env_get("RPLINE"))) die_usage();
|
|
if (!(ufline = env_get("UFLINE"))) die_usage();
|
|
if (!(recip = env_get("LOCAL"))) die_usage();
|
|
if (!(host = env_get("HOST"))) die_usage();
|
|
if (!(sender = env_get("SENDER"))) die_usage();
|
|
|
|
while ((opt = getopt(argc,argv,"DFlMmnPsx:")) != opteof)
|
|
switch(opt)
|
|
{
|
|
case 'D': case 'F': case 'M': break; /* be serious */
|
|
case 'l': flagdtline = 1; break; /* also return-receipt-to? blech */
|
|
case 'm': break; /* we only handle one recipient anyway */
|
|
case 'n': flagufline = 0; break;
|
|
case 's': break; /* could call quote() otherwise, i suppose... */
|
|
case 'P': flagrpline = 1; break;
|
|
case 'x':
|
|
if (case_starts(recip,optarg))
|
|
recip += str_len(optarg);
|
|
break;
|
|
default:
|
|
_exit(100);
|
|
}
|
|
argc -= optind;
|
|
argv += optind;
|
|
|
|
if (!*argv) die_usage();
|
|
|
|
for (arg = argv;x = *arg;++arg)
|
|
{
|
|
if (!stralloc_copys(&newarg,"")) die_temp();
|
|
flagesc = 0;
|
|
for (i = 0;x[i];++i)
|
|
if (flagesc)
|
|
{
|
|
switch(x[i])
|
|
{
|
|
case '%': if (!stralloc_cats(&newarg,"%")) die_temp(); break;
|
|
case 'g': if (!stralloc_cats(&newarg,sender)) die_temp(); break;
|
|
case 'h': if (!stralloc_cats(&newarg,host)) die_temp(); break;
|
|
case 'u': if (!stralloc_cats(&newarg,recip)) die_temp(); break;
|
|
}
|
|
flagesc = 0;
|
|
}
|
|
else
|
|
if (x[i] == '%')
|
|
flagesc = 1;
|
|
else
|
|
if (!stralloc_append(&newarg,&x[i])) die_temp();
|
|
if (!stralloc_0(&newarg)) die_temp();
|
|
i = str_len(newarg.s) + 1;
|
|
if (!(x = alloc(i))) die_temp();
|
|
byte_copy(x,i,newarg.s);
|
|
*arg = x;
|
|
}
|
|
|
|
if (pipe(pi) == -1) die_temp();
|
|
|
|
switch(pid = fork())
|
|
{
|
|
case -1:
|
|
die_temp();
|
|
case 0:
|
|
close(pi[1]);
|
|
if (fd_move(0,pi[0]) == -1) die_temp();
|
|
sig_pipedefault();
|
|
execvp(*argv,argv);
|
|
if (error_temp(errno)) die_temp();
|
|
die_badcmd();
|
|
}
|
|
close(pi[0]);
|
|
|
|
substdio_fdbuf(&ssout,write,pi[1],outbuf,sizeof(outbuf));
|
|
substdio_fdbuf(&ssin,read,0,inbuf,sizeof(inbuf));
|
|
if (flagufline) substdio_bputs(&ssout,ufline);
|
|
if (flagrpline) substdio_bputs(&ssout,rpline);
|
|
if (flagdtline) substdio_bputs(&ssout,dtline);
|
|
if (substdio_copy(&ssout,&ssin) == -2) die_read();
|
|
substdio_flush(&ssout);
|
|
close(pi[1]);
|
|
|
|
if (wait_pid(&wstat,pid) == -1) die_temp();
|
|
if (wait_crashed(wstat)) die_temp();
|
|
_exit(wait_exitcode(wstat));
|
|
}
|