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 }