chiark - git - mdw - qmail/blob - maildir2mbox.c

   1 #include "readwrite.h"
   2 #include "prioq.h"
   3 #include "env.h"
   4 #include "stralloc.h"
   5 #include "subfd.h"
   6 #include "substdio.h"
   7 #include "getln.h"
   8 #include "error.h"
   9 #include "open.h"
  10 #include "lock.h"
  11 #include "gfrom.h"
  12 #include "str.h"
  13 #include "exit.h"
  14 #include "myctime.h"
  15 #include "maildir.h"
  16
  17 char *mbox;
  18 char *mboxtmp;
  19
  20 stralloc filenames = {0};
  21 prioq pq = {0};
  22 prioq pq2 = {0};
  23
  24 stralloc line = {0};
  25
  26 stralloc ufline = {0};
  27
  28 char inbuf[SUBSTDIO_INSIZE];
  29 char outbuf[SUBSTDIO_OUTSIZE];
  30
  31 #define FATAL "maildir2mbox: fatal: "
  32 #define WARNING "maildir2mbox: warning: "
  33
  34 void die_nomem() { strerr_die2x(111,FATAL,"out of memory"); }
  35
  36 void main()
  37 {
  38  substdio ssin;
  39  substdio ssout;
  40  struct prioq_elt pe;
  41  int fdoldmbox;
  42  int fdnewmbox;
  43  int fd;
  44  int match;
  45  int fdlock;
  46
  47  umask(077);
  48
  49  mbox = env_get("MAIL");
  50  if (!mbox) strerr_die2x(111,FATAL,"MAIL not set");
  51  mboxtmp = env_get("MAILTMP");
  52  if (!mboxtmp) strerr_die2x(111,FATAL,"MAILTMP not set");
  53
  54  if (maildir_chdir() == -1)
  55    strerr_die1(111,FATAL,&maildir_chdir_err);
  56  maildir_clean(&filenames);
  57  if (maildir_scan(&pq,&filenames,1,1) == -1)
  58    strerr_die1(111,FATAL,&maildir_scan_err);
  59
  60  if (!prioq_min(&pq,&pe)) _exit(0); /* nothing new */
  61
  62  fdlock = open_append(mbox);
  63  if (fdlock == -1)
  64    strerr_die4sys(111,FATAL,"unable to lock ",mbox,": ");
  65  if (lock_ex(fdlock) == -1)
  66    strerr_die4sys(111,FATAL,"unable to lock ",mbox,": ");
  67
  68  fdoldmbox = open_read(mbox);
  69  if (fdoldmbox == -1)
  70    strerr_die4sys(111,FATAL,"unable to read ",mbox,": ");
  71
  72  fdnewmbox = open_trunc(mboxtmp);
  73  if (fdnewmbox == -1)
  74    strerr_die4sys(111,FATAL,"unable to create ",mboxtmp,": ");
  75
  76  substdio_fdbuf(&ssin,read,fdoldmbox,inbuf,sizeof(inbuf));
  77  substdio_fdbuf(&ssout,write,fdnewmbox,outbuf,sizeof(outbuf));
  78
  79  switch(substdio_copy(&ssout,&ssin))
  80   {
  81    case -2: strerr_die4sys(111,FATAL,"unable to read ",mbox,": ");
  82    case -3: strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": ");
  83   }
  84
  85  while (prioq_min(&pq,&pe))
  86   {
  87    prioq_delmin(&pq);
  88    if (!prioq_insert(&pq2,&pe)) die_nomem();
  89
  90    fd = open_read(filenames.s + pe.id);
  91    if (fd == -1)
  92      strerr_die4sys(111,FATAL,"unable to read $MAILDIR/",filenames.s + pe.id,": ");
  93    substdio_fdbuf(&ssin,read,fd,inbuf,sizeof(inbuf));
  94
  95    if (getln(&ssin,&line,&match,'\n') != 0)
  96      strerr_die4sys(111,FATAL,"unable to read $MAILDIR/",filenames.s + pe.id,": ");
  97
  98    if (!stralloc_copys(&ufline,"From XXX ")) die_nomem();
  99    if (match)
 100      if (stralloc_starts(&line,"Return-Path: <"))
 101       {
 102        if (line.s[14] == '>')
 103         {
 104          if (!stralloc_copys(&ufline,"From MAILER-DAEMON ")) die_nomem();
 105         }
 106        else
 107         {
 108          int i;
 109          if (!stralloc_ready(&ufline,line.len)) die_nomem();
 110          if (!stralloc_copys(&ufline,"From ")) die_nomem();
 111          for (i = 14;i < line.len - 2;++i)
 112            if ((line.s[i] == ' ') || (line.s[i] == '\t'))
 113              ufline.s[ufline.len++] = '-';
 114            else
 115              ufline.s[ufline.len++] = line.s[i];
 116          if (!stralloc_cats(&ufline," ")) die_nomem();
 117         }
 118       }
 119    if (!stralloc_cats(&ufline,myctime(pe.dt))) die_nomem();
 120    if (substdio_put(&ssout,ufline.s,ufline.len) == -1)
 121      strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": ");
 122
 123    while (match && line.len)
 124     {
 125      if (gfrom(line.s,line.len))
 126        if (substdio_puts(&ssout,">") == -1)
 127          strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": ");
 128      if (substdio_put(&ssout,line.s,line.len) == -1)
 129        strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": ");
 130      if (!match)
 131       {
 132        if (substdio_puts(&ssout,"\n") == -1)
 133          strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": ");
 134        break;
 135       }
 136      if (getln(&ssin,&line,&match,'\n') != 0)
 137        strerr_die4sys(111,FATAL,"unable to read $MAILDIR/",filenames.s + pe.id,": ");
 138     }
 139    if (substdio_puts(&ssout,"\n"))
 140      strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": ");
 141
 142    close(fd);
 143   }
 144
 145  if (substdio_flush(&ssout) == -1)
 146    strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": ");
 147  if (fsync(fdnewmbox) == -1)
 148    strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": ");
 149  if (close(fdnewmbox) == -1) /* NFS dorks */
 150    strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": ");
 151  if (rename(mboxtmp,mbox) == -1)
 152    strerr_die6(111,FATAL,"unable to move ",mboxtmp," to ",mbox,": ",&strerr_sys);
 153
 154  while (prioq_min(&pq2,&pe))
 155   {
 156    prioq_delmin(&pq2);
 157    if (unlink(filenames.s + pe.id) == -1)
 158      strerr_warn4(WARNING,"$MAILDIR/",filenames.s + pe.id," will be delivered twice; unable to unlink: ",&strerr_sys);
 159   }
 160
 161  _exit(0);
 162 }