| 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 | } |