nightmaremail

nmail-qmqpd.c (5834B)

---

 #include <unistd.h>
 #include <stdio.h>
 //#define dprintf(x, ...)
 #include "noreturn.h"
 #include "auto_qmail.h"
 #include "qmail.h"
 #include "received.h"
 #include "sig.h"
 #include "substdio.h"
 #include "readwrite.h"
 #include "exit.h"
 #include "now.h"
 #include "fmt.h"
 #include "byte.h"
 #include "env.h"
 #include "str.h"
 #include "netstrings.h"
 
 
 ssize_t safewrite(int fd, const void *buf, size_t len)
 {
   ssize_t r;
   r = write(fd,buf,len);
   if (r == 0 || r == -1) _exit(0);
   return r;
 }
 ssize_t saferead(int fd, void *buf, size_t len)
 {
   ssize_t r;
   r = read(fd,buf,len);
   if (r == 0 || r == -1) _exit(0);
   return r;
 }
 
 char buf[1000];
 char strnum[FMT_ULONG];
 
 char ssinbuf[512];
 substdio ssin = SUBSTDIO_FDBUF(saferead,0,ssinbuf,sizeof(ssinbuf));
 char ssoutbuf[256];
 substdio ssout = SUBSTDIO_FDBUF(safewrite,1,ssoutbuf,sizeof(ssoutbuf));
 
 void _noreturn_ resources() {
   char *result = "DOut of resources";
 
   substdio_put(&ssout,strnum,fmt_ulong(strnum,(unsigned long) str_len(result)));
   substdio_puts(&ssout,":");
   substdio_puts(&ssout,result);
   substdio_puts(&ssout,",");
   substdio_flush(&ssout);
 _exit(111); }
 void _noreturn_ badproto() {
   char *result = "DBad protocol";
 
   substdio_put(&ssout,strnum,fmt_ulong(strnum,(unsigned long) str_len(result)));
   substdio_puts(&ssout,":");
   substdio_puts(&ssout,result);
   substdio_puts(&ssout,",");
   substdio_flush(&ssout);
 _exit(100); }
 void _noreturn_ bytesout() {
   char *result = "DBytes Out";
 
   substdio_put(&ssout,strnum,fmt_ulong(strnum,(unsigned long) str_len(result)));
   substdio_puts(&ssout,":");
   substdio_puts(&ssout,result);
   substdio_puts(&ssout,",");
   substdio_flush(&ssout);
 _exit(100); }
 void _noreturn_ badcomma() {
   char *result = "DBad comma";
 
   substdio_put(&ssout,strnum,fmt_ulong(strnum,(unsigned long) str_len(result)));
   substdio_puts(&ssout,":");
   substdio_puts(&ssout,result);
   substdio_puts(&ssout,",");
   substdio_flush(&ssout);
 _exit(100); }
 void _noreturn_ die_wtf() {
   char *result = "DWTF?";
 
   substdio_put(&ssout,strnum,fmt_ulong(strnum,(unsigned long) str_len(result)));
   substdio_puts(&ssout,":");
   substdio_puts(&ssout,result);
   substdio_puts(&ssout,",");
   substdio_flush(&ssout);
 _exit(100); }
 
 unsigned long long bytesleft = 100;
 
 void getbyte(ch)
 char *ch;
 {
   if (!bytesleft--) bytesout();
   substdio_get(&ssin,ch,1);
 }
 
 int getlen_times_called = 0;
 
 unsigned long long getlen()
 {
   unsigned long long len = 0;
   dprintf(2, "getlen has been called %d times\n", getlen_times_called++);
   switch (ns_getlength(&ssin,&len)) {
     case NS_SUCCESS: return len; break;
     case NS_TOOMUCH: resources(); break;
     case NS_PROTOCOL: badproto(); break;
     case NS_INVAL: die_wtf(); break;
     case NS_BIOERR: die_wtf(); break;
   }
   // we won't reach here.
   return len;
 #if 0
   char ch;
 
   for (;;) {
     getbyte(&ch);
     if (ch == ':') return len;
     if (len > 200000000) resources();
     len = 10 * len + (ch - '0');
   }
 #endif
 }
 
 void getcomma()
 {
   switch (ns_getcomma(&ssin,&bytesleft)) {
     case NS_SUCCESS: break;
     case NS_BYTES: bytesout(); break;
     case NS_PROTOCOL: badcomma(); break;
   }
 
 #if 0
   char ch;
   getbyte(&ch);
   if (ch != ',') _exit(100);
 #endif
 }
 
 struct qmail qq;
 
 void identify()
 {
   char *remotehost;
   char *remoteinfo;
   char *remoteip;
   char *local;
 
   remotehost = env_get("TCPREMOTEHOST");
   if (!remotehost) remotehost = "unknown";
   remoteinfo = env_get("TCPREMOTEINFO");
   remoteip = env_get("TCPREMOTEIP");
   if (!remoteip) remoteip = "unknown";
   local = env_get("TCPLOCALHOST");
   if (!local) local = env_get("TCPLOCALIP");
   if (!local) local = "unknown";
  
   received(&qq,"QMQP",local,remoteip,remotehost,remoteinfo,NULL);
 }
 
 int getbuf()
 {
   unsigned long long len;
   int i;
 
   len = getlen();
   if (len >= 1000) {
     for (i = 0;i < len;++i) getbyte(buf);
     getcomma();
     buf[0] = 0;
     return 0;
   }
 
   for (i = 0;i < len;++i) getbyte(buf + i);
   getcomma();
   buf[len] = 0;
   return byte_chr(buf,len,'\0') == len;
 }
 
 int flagok = 1;
 
 int
 main()
 {
   char *result;
   unsigned long long qp;
   unsigned long long len, gb;
   char ch;
 
   sig_pipeignore();
   sig_alarmcatch(resources);
   alarm(3600);
   dprintf(2, "%s\n", "stopwatch set: die in 3600 s");
 
   bytesleft = getlen();
 
   len = getlen();
   dprintf(2, "bytesleft %llu len %llu\n", bytesleft, len);
 
   if (chdir(auto_qmail) == -1) resources();
   if (qmail_open(&qq) == -1) resources();
   qp = qmail_qp(&qq);
   identify();
 
   while (len > 0) { /* XXX: could speed this up */
     getbyte(&ch);
     --len;
     qmail_put(&qq,&ch,1);
   }
   getcomma();
 
   dprintf(2, "pre-from bytesleft %llu len %llu\n", bytesleft, len);
 
   if (getbuf()) {
     dprintf(2, "from bytesleft %llu len %llu\n", bytesleft, len);
     qmail_from(&qq,buf);
   } else {
     qmail_from(&qq,"");
     qmail_fail(&qq);
     flagok = 0;
     dprintf(2, "dying due to failed getbuf for from\n");
   }
 
   while (bytesleft)
     if (getbuf()) {
       dprintf(2, "to bytesleft %llu len %llu\n", bytesleft, len);
       qmail_to(&qq,buf);
     } else {
       dprintf(2, "to bytesleft %llu len %llu\n", bytesleft, len);
       dprintf(2, "dying due to failed getbuf for to\n");
       qmail_fail(&qq);
       flagok = 0;
     }
 
   bytesleft = 1;
   getcomma();
   dprintf(2, "bytesleft %llu len %llu\n", bytesleft, len);
 
   result = qmail_close(&qq);
 
   if (!*result) {
     len = fmt_str(buf,"Kok ");
     len += fmt_ulong(buf + len,(unsigned long) now());
     len += fmt_str(buf + len," qp ");
     len += fmt_ulong(buf + len,qp);
     buf[len] = 0;
     result = buf;
   }
 
   if (!flagok)
     result = "Dsorry, I can't accept addresses like that (#5.1.3)";
 
   substdio_put(&ssout,strnum,fmt_ulong(strnum,(unsigned long) str_len(result)));
   substdio_puts(&ssout,":");
   substdio_puts(&ssout,result);
   substdio_puts(&ssout,",");
   substdio_flush(&ssout);
   _exit(0);
 }