srsforward.c

Go to the documentation of this file.

#include <unistd.h>
#include <sys/types.h>
#include "control.h"
#include "sig.h"
#include "constmap.h"
#include "readwrite.h"
#include "exit.h"
#include "env.h"
#include "qmail.h"
#include "auto_qmail.h"
#include "getoptb.h"
#include "buffer.h"
#include "str.h"
#include "fmt.h"
#include "stralloc.h"
#include "logmsg.h"
#include "srsq.h"
#include "qmail.h"
 
#define WHO "srsforward"
 
void die_nomem() { logmsg(WHO,111,FATAL,"out of memory"); } 
void die_control() { logmsg(WHO,110,FATAL,"Unable to read control files"); }
 
struct qmail qqt;
 
char *srsdomaininfo = 0;
stralloc srsdomains = {0};
struct constmap mapsrsdomains;
stralloc srshost = {0};
stralloc srssender = {0};
stralloc cookie = {0};
srs_ctx srs = {0};
 
int flagforward = 1;
int flagpassthrough = 1;
 
ssize_t mywrite(int fd,char *buf,size_t len)
{
  qmail_put(&qqt,buf,len);
  return len;
}
 
char inbuf[BUFSIZE_LINE];
char outbuf[BUFSIZE_LINE];
buffer bi = BUFFER_INIT(buffer_unixread,0,inbuf,sizeof(inbuf));
buffer bo = BUFFER_INIT(mywrite,-1,outbuf,sizeof(outbuf));
 
char num[FMT_ULONG];
 
int main(int argc,char * const argv[])
{
  int i, j, r;
  int opt;
  char *qqx;
  char separator = '=';
  char *host = 0;
  char *sender = 0;
  char *dtline = 0;
  char *sendhost = 0;
  int alwaysrewrite = 0;
  char ch;
 
  sig_pipeignore();
 
  sender = env_get("NEWSENDER");
  if (!sender)
    logmsg(WHO,100,FATAL,"NEWSENDER not set");
  host = env_get("HOST");
  if (!host)
    logmsg(WHO,100,FATAL,"HOST not set");
  dtline = env_get("DTLINE");
  if (!dtline)
    logmsg(WHO,100,FATAL,"DTLINE not set");
 
  if (chdir(auto_qmail) == -1)
    logmsg(WHO,111,FATAL,B("unable to chdir to: ",auto_qmail));
 
  if (!stralloc_cats(&srshost,"!")) die_nomem();
  if (!stralloc_cats(&srshost,host)) die_nomem();
 
  switch (control_readfile(&srsdomains,"control/srsdomains",0)) {
    case -1: die_control();
    case  0: if (!constmap_init(&mapsrsdomains,"",0,1)) die_nomem(); break;
    case  1: if (!constmap_init(&mapsrsdomains,srsdomains.s,srsdomains.len,1)) die_nomem(); break;
  }
  if (constmap(&mapsrsdomains,srshost.s,srshost.len)) flagforward = 0;    // domain blacklisted, bounce
  if ((srsdomaininfo = constmap(&mapsrsdomains,host,str_len(host))) == 0) {
    if ((srsdomaininfo = constmap(&mapsrsdomains,"*",1)) == 0) return 0;  // '*' means always SRS
    else alwaysrewrite = 1;
  }
 
  if (*srsdomaininfo) {
    i = str_chr(srsdomaininfo,'|');              // multiple cookies; separated by  ' '
    if (srsdomaininfo[i] == '|') {
      srsdomaininfo[i] = 0;
      j = str_chr(srsdomaininfo + i + 1,'|'); 
      if (srsdomaininfo[i + j + 1] == '|') {
        srsdomaininfo[i + j + 1] = 0;
        sendhost = srsdomaininfo + i + j + 2;    // separator: - + =
      }
      separator = srsdomaininfo[i + 1];
    }
    if (!stralloc_copys(&cookie,srsdomaininfo)) die_nomem();
    if (!stralloc_0(&cookie)) die_nomem();
    if (!stralloc_copys(&srshost,"")) die_nomem();
    if (*sendhost) {
      j = str_len(sendhost);
      if (sendhost[j - 1] == '.') {
        if (!stralloc_copys(&srshost,sendhost)) die_nomem();
        if (!stralloc_cats(&srshost,host)) die_nomem();
      } else
        if (!stralloc_copys(&srshost,sendhost)) die_nomem();
    } else
      if (!stralloc_copys(&srshost,host)) die_nomem();
    if (!stralloc_0(&srshost)) die_nomem();
  } else
    die_control();
 
  /* Check options */
 
  while ((opt = getoptb(argc,(char **)argv,"pP")) != opteof)
    switch (opt) {
      case 'p': flagpassthrough = 0; break;
      case 'P': flagpassthrough = 100; break;
    }
  argv += optind - 1;
 
  if (!flagforward) {
    logmsg(WHO,flagpassthrough,INFO,B("not srsforwarding: ",sender));
     _exit(flagpassthrough);
  }
 
  /* Let's go SRS rewrite */
 
  srs_init(&srs);
 
  if (separator == '-' || separator == '+' || separator == '=')      // '=' is default
    srs.separator = separator;
  if (alwaysrewrite) 
    srs.alwaysrewrite = alwaysrewrite;
 
  if (!stralloc_copys(&srs.secrets,"")) die_nomem();
  for (j = 0; j <= cookie.len; j++) {
    ch = cookie.s[j];
    if (ch == ' ') { ch = '\0'; srs.numsecrets++; }
    if (!stralloc_append(&srs.secrets,&ch)) die_nomem();
  }
  if (!stralloc_append(&srs.secrets,&ch)) die_nomem();
 
  if ((r = srs_forward(&srs,&srssender,sender,srshost.s)) != SRS_SUCCESS)
    logmsg(WHO,100,FATAL,B("Unable to srsforward: ",sender," ",srs_errorstr(r)));
 
  if (qmail_open(&qqt) == -1)
    logmsg(WHO,111,FATAL,"unable to fork: ");
  qmail_puts(&qqt,dtline);
  if (buffer_copy(&bo,&bi) != 0)
    logmsg(WHO,111,FATAL,"unable to read message: ");
  buffer_flush(&bo);
 
  num[fmt_ulong(num,qmail_qp(&qqt))] = 0;
 
  qmail_from(&qqt,srssender.s);
  while (*++argv) qmail_to(&qqt,*argv);
  qqx = qmail_close(&qqt);
  if (*qqx) logmsg(WHO,*qqx == 'D' ? 100 : 111,FATAL,qqx + 1);
  logmsg(WHO,0,LOG,B(srssender.s,": qp ",num));
}