qsmhook.c
changeset 0 068428edee47
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/qsmhook.c	Fri Oct 19 14:06:22 2007 +0200
@@ -0,0 +1,137 @@
+#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));
+}