ezcopy.c

Go to the documentation of this file.

#include <unistd.h>
#include "stralloc.h"
#include "buffer.h"
#include "str.h"
#include "getln.h"
#include "case.h"
#include "readwrite.h"
#include "qmail.h"
#include "logmsg.h"
#include "error.h"
#include "quote.h"
#include "ezcopy.h"
#include "mime.h"
#include "open.h"
#include "byte.h"
#include "idx.h"
#include "errtxt.h"
 
#define WHO "ezcopy"

 
static stralloc line = {0};
static stralloc outline = {0};
static stralloc qline = {0};
static stralloc outlocal = {0};
static stralloc outhost = {0};
 
static buffer bt;
static char textbuf[256];
static const char *target = "?????";
static const char *verptarget = "?????";
static const char *confirm = "?????";
static unsigned int confirmlocal;
static const char *szmsgnum = "?????";
 
static void die_nomem() { logmsg(WHO,111,FATAL,ERR_NOMEM); }

 
void set_cpoutlocal(const stralloc *ln)
{       
  if (!quote(&outlocal,ln))
    die_nomem();
}
 
void set_cpouthost(const stralloc *ln)
{
  if (!stralloc_copy(&outhost,ln))
    die_nomem();
}
 
void set_cptarget(const char *tg)
{
  target = tg;
}
 
void set_cpverptarget(const char *tg)
{
  verptarget = tg;
}
 
void set_cpconfirm(const char *cf)
{
  confirm = cf;
  confirmlocal = str_chr(cf,'@');
}
 
void set_cpnum(const char *cf)
{
  szmsgnum = cf;
}
 
static struct qmail *qqc;
 
static void codeput(const char *l,unsigned int n,char code)
{
  if (!code || code == 'H')
    qmail_put(qqc,l,n);
  else {
    if (code == 'Q')
      encode_qp(l,n,&qline);
    else
      encode_b64(l,n,&qline,0);
    qmail_put(qqc,qline.s,qline.len);
  }
}
 
static void codeputs(const char *l,char code)
{
  codeput(l,str_len(l),code);
}

 
void ezcopy(struct qmail *qqp,const char *fn, char q)                     
{
  int fd;
  int match, done;
  unsigned int pos, nextpos;
 
  qqc = qqp;
  if ((fd = open_read(fn)) == -1)
    if (errno != ENOENT)
      logmsg(WHO,111,FATAL,B("error opening: ",fn));
    else
      logmsg(WHO,100,FATAL,B("error opening: ",fn));
 
  buffer_init(&bt,buffer_unixread,fd,textbuf,sizeof(textbuf));
 
  for (;;) {
    if (getln(&bt,&line,&match,'\n') == -1)
      logmsg(WHO,111,FATAL,B("error opening: ",fn));
    if (line.len > 0) {            /* line.len is always > 0 if match is true */
                             /* suppress blank line for 'H'eader mode */
      if (line.len == 1 && q == 'H') continue;
      if (line.s[0] == '!') {
        if (line.s[1] == 'R') {
           codeput("   ",3,q);
           codeputs(confirm,q);
           codeput("\n",1,q);
           continue;
        }
        if (line.s[1] == 'A') {
          codeput("   ",3,q);
          codeputs(target,q);
          codeput("\n",1,q);
          continue;
        }
      }
      /* Find tags <#x#>. Replace with for x=R confirm, for x=A */
      /* target, x=l outlocal, x=h outhost. For others, just    */
      /* skip tag. If outlocal/outhost are not set, the tags are*/
      /* skipped. If confirm/taget are not set, the tags are    */
      /* replaced by "???????" */
 
      pos = 0;
      nextpos = 0;
      done = 0;
      outline.len = 0;                             /* zap outline */
 
      while ((pos += byte_chr(line.s+pos,line.len-pos,'<')) != line.len) {
        if (pos + 4 < line.len &&
            line.s[pos+1] == '#' &&
            line.s[pos+3] == '#' &&
            line.s[pos+4] == '>') {        /* tag. Copy first part of line */
          done = 1;                                          /* did something */
          if (!stralloc_catb(&outline,line.s+nextpos,pos-nextpos))
             die_nomem();
          switch(line.s[pos+2]) {
            case 'A':
              if (q == 'H') logmsg(WHO,111,ERROR,ERR_SUBST_UNSAFE);
              if (!stralloc_cats(&outline,target)) die_nomem();
              break;
            case 'R':
              if (!stralloc_cats(&outline,confirm)) die_nomem();
              break;
            case 'r':
              if (!stralloc_catb(&outline,confirm,confirmlocal)) die_nomem();
              break;
            case 'l':
              if (!stralloc_cat(&outline,&outlocal)) die_nomem();
              break;
            case 'h':
              if (!stralloc_cat(&outline,&outhost)) die_nomem();
              break;
            case 't':
              if (q == 'H') logmsg(WHO,111,ERROR,ERR_SUBST_UNSAFE);
              if (!stralloc_cats(&outline,verptarget)) die_nomem();
              break;
            case 'n':
              if (!stralloc_cats(&outline,szmsgnum)) die_nomem();
              break;
            default:
              break;                                 /* unknown tags killed */
          }
          pos += 5;
          nextpos = pos;
        } else
          ++pos;                                               /* try next position */
      }
      if (!done)
        codeput(line.s,line.len,q);
      else {
        if (!stralloc_catb(&outline,line.s+nextpos,line.len-nextpos))
           die_nomem();                             /* remainder */
        codeput(outline.s,outline.len,q);
      }
 
      /* Last line is missing its trailing newline, add one on output. */
      if (!match)
        codeput("\n",1,q);
    } else
        break;
  }
  close(fd);
}