diplodocus.org Git - nmh/blob - uip/replsbr.c

   1
   2 /*
   3  * replsbr.c -- routines to help repl along...
   4  *
   5  * This code is Copyright (c) 2002, by the authors of nmh.  See the
   6  * COPYRIGHT file in the root directory of the nmh distribution for
   7  * complete copyright information.
   8  */
   9
  10 #include <h/mh.h>
  11 #include <h/addrsbr.h>
  12 #include <h/fmt_scan.h>
  13 #include <h/utils.h>
  14 #include <sys/file.h>           /* L_SET */
  15 #include <errno.h>
  16
  17 extern short ccto;              /* from repl.c */
  18 extern short cccc;
  19 extern short ccme;
  20 extern short querysw;
  21
  22 static int dftype=0;
  23
  24 static char *badaddrs = NULL;
  25 static char *dfhost = NULL;
  26
  27 static struct mailname mq;
  28 static int nodupcheck = 0;              /* If set, no check for duplicates */
  29
  30 /*
  31  * Buffer size for content part of header fields.
  32  * We want this to be large enough so that we don't
  33  * do a lot of extra FLDPLUS calls on m_getfld but
  34  * small enough so that we don't snarf the entire
  35  * message body when we're not going to use any of it.
  36  */
  37 #define SBUFSIZ 256
  38
  39 static char *addrcomps[] = {
  40     "from",
  41     "sender",
  42     "reply-to",
  43     "to",
  44     "cc",
  45     "bcc",
  46     "resent-from",
  47     "resent-sender",
  48     "resent-reply-to",
  49     "resent-to",
  50     "resent-cc",
  51     "resent-bcc",
  52     NULL
  53 };
  54
  55 /*
  56  * static prototypes
  57  */
  58 static int insert (struct mailname *);
  59 static void replfilter (FILE *, FILE *, char *, int);
  60 static char *replformataddr(char *, char *);
  61 static char *replconcataddr(char *, char *);
  62
  63
  64 void
  65 replout (FILE *inb, char *msg, char *drft, struct msgs *mp, int outputlinelen,
  66         int mime, char *form, char *filter, char *fcc, int fmtproc)
  67 {
  68     register int state, i;
  69     register struct comp *cptr;
  70     char tmpbuf[SBUFSIZ];
  71     struct format *fmt;
  72     register char **ap;
  73     int char_read = 0, format_len, mask;
  74     char name[NAMESZ], *scanl, *cp;
  75     static int dat[5];                  /* aux. data for format routine */
  76     m_getfld_state_t gstate = 0;
  77     struct fmt_callbacks cb;
  78
  79     FILE *out;
  80     NMH_UNUSED (msg);
  81
  82     mask = umask(~m_gmprot());
  83     if ((out = fopen (drft, "w")) == NULL)
  84         adios (drft, "unable to create");
  85
  86     umask(mask);
  87
  88     /* get new format string */
  89     cp = new_fs (form, NULL, NULL);
  90     format_len = strlen (cp);
  91
  92     /* compile format string */
  93     fmt_compile (cp, &fmt, 1);
  94
  95     for (ap = addrcomps; *ap; ap++) {
  96         cptr = fmt_findcomp (*ap);
  97         if (cptr)
  98             cptr->c_type |= CT_ADDR;
  99     }
 100
 101     /*
 102      * ignore any components killed by command line switches
 103      *
 104      * This prevents the component from being found via fmt_findcomp(),
 105      * which makes sure no text gets added to it when the message is processed.
 106      *
 107      * getcpy(NULL) returns a malloc'd zero-length string, so it can safely
 108      * be free()'d later.
 109      */
 110     if (!ccto) {
 111         cptr = fmt_findcomp ("to");
 112         if (cptr)
 113             cptr->c_name = getcpy(NULL);
 114     }
 115     if (!cccc) {
 116          cptr = fmt_findcomp("cc");
 117         if (cptr)
 118             cptr->c_name = getcpy(NULL);
 119     }
 120     /* set up the "fcc" pseudo-component */
 121     if (fcc) {
 122         cptr = fmt_findcomp ("fcc");
 123         if (cptr)
 124             cptr->c_text = getcpy (fcc);
 125     }
 126     if ((cp = getenv("USER"))) {
 127         cptr = fmt_findcomp ("user");
 128         if (cptr)
 129             cptr->c_text = getcpy(cp);
 130     }
 131     if (!ccme)
 132         ismymbox (NULL);
 133
 134     /*
 135      * pick any interesting stuff out of msg "inb"
 136      */
 137     for (;;) {
 138         int msg_count = sizeof tmpbuf;
 139         state = m_getfld (&gstate, name, tmpbuf, &msg_count, inb);
 140         switch (state) {
 141             case FLD:
 142             case FLDPLUS:
 143                 /*
 144                  * if we're interested in this component, save a pointer
 145                  * to the component text, then start using our next free
 146                  * buffer as the component temp buffer (buffer switching
 147                  * saves an extra copy of the component text).
 148                  */
 149
 150                 i = fmt_addcomptext(name, tmpbuf);
 151                 if (i != -1) {
 152                     char_read += msg_count;
 153                     while (state == FLDPLUS) {
 154                         msg_count= sizeof tmpbuf;
 155                         state = m_getfld (&gstate, name, tmpbuf, &msg_count, inb);
 156                         fmt_appendcomp(i, name, tmpbuf);
 157                         char_read += msg_count;
 158                     }
 159                 }
 160
 161                 while (state == FLDPLUS) {
 162                     msg_count= sizeof tmpbuf;
 163                     state = m_getfld (&gstate, name, tmpbuf, &msg_count, inb);
 164                 }
 165                 break;
 166
 167             case LENERR:
 168             case FMTERR:
 169             case BODY:
 170             case FILEEOF:
 171                 goto finished;
 172
 173             default:
 174                 adios (NULL, "m_getfld() returned %d", state);
 175         }
 176     }
 177
 178     /*
 179      * format and output the header lines.
 180      */
 181 finished:
 182     m_getfld_state_destroy (&gstate);
 183
 184     /*
 185      * if there's a "Subject" component, strip any "Re:"s off it
 186      */
 187     cptr = fmt_findcomp ("subject");
 188     if (cptr && (cp = cptr->c_text)) {
 189         register char *sp = cp;
 190
 191         for (;;) {
 192             while (isspace((unsigned char) *cp))
 193                 cp++;
 194             if(uprf(cp, "re:"))
 195                 cp += 3;
 196             else
 197                 break;
 198             sp = cp;
 199         }
 200         if (sp != cptr->c_text) {
 201             cp = cptr->c_text;
 202             cptr->c_text = getcpy (sp);
 203             free (cp);
 204         }
 205     }
 206     i = format_len + char_read + 256;
 207     scanl = mh_xmalloc ((size_t) i + 2);
 208     dat[0] = 0;
 209     dat[1] = 0;
 210     dat[2] = 0;
 211     dat[3] = outputlinelen;
 212     dat[4] = 0;
 213     memset(&cb, 0, sizeof(cb));
 214     cb.formataddr = replformataddr;
 215     cb.concataddr = replconcataddr;
 216     fmt_scan (fmt, scanl, i + 1, i, dat, &cb);
 217     fputs (scanl, out);
 218     if (badaddrs) {
 219         fputs ("\nrepl: bad addresses:\n", out);
 220         fputs ( badaddrs, out);
 221     }
 222
 223     /*
 224      * Check if we should filter the message
 225      * or add mhn directives
 226      */
 227     if (filter) {
 228         fflush(out);
 229         if (ferror (out))
 230             adios (drft, "error writing");
 231
 232         replfilter (inb, out, filter, fmtproc);
 233     } else if (mime && mp) {
 234             fprintf (out, "#forw [original message] +%s %s\n",
 235                      mp->foldpath, m_name (mp->lowsel));
 236     }
 237
 238     fflush(out);
 239     if (ferror (out))
 240         adios (drft, "error writing");
 241     fclose (out);
 242
 243     /* return dynamically allocated buffers */
 244     free (scanl);
 245     fmt_free(fmt, 1);
 246 }
 247
 248 static char *buf;               /* our current working buffer */
 249 static char *bufend;            /* end of working buffer */
 250 static char *last_dst;          /* buf ptr at end of last call */
 251 static unsigned int bufsiz=0;   /* current size of buf */
 252
 253 #define BUFINCR 512             /* how much to expand buf when if fills */
 254
 255 #define CPY(s) { cp = (s); while ((*dst++ = *cp++)) ; --dst; }
 256
 257 /*
 258  * check if there's enough room in buf for str.
 259  * add more mem if needed
 260  */
 261 #define CHECKMEM(str) \
 262             if ((len = strlen (str)) >= bufend - dst) {\
 263                 int i = dst - buf;\
 264                 int n = last_dst - buf;\
 265                 bufsiz += ((dst + len - bufend) / BUFINCR + 1) * BUFINCR;\
 266                 buf = mh_xrealloc (buf, bufsiz);\
 267                 dst = buf + i;\
 268                 last_dst = buf + n;\
 269                 bufend = buf + bufsiz;\
 270             }
 271
 272
 273 /*
 274  * fmt_scan will call this routine if the user includes the function
 275  * "(formataddr {component})" in a format string.  "orig" is the
 276  * original contents of the string register.  "str" is the address
 277  * string to be formatted and concatenated onto orig.  This routine
 278  * returns a pointer to the concatenated address string.
 279  *
 280  * We try to not do a lot of malloc/copy/free's (which is why we
 281  * don't call "getcpy") but still place no upper limit on the
 282  * length of the result string.
 283  */
 284 static char *
 285 replformataddr (char *orig, char *str)
 286 {
 287     register int len;
 288     char baddr[BUFSIZ], error[BUFSIZ];
 289     register int isgroup;
 290     register char *dst;
 291     register char *cp;
 292     register char *sp;
 293     register struct mailname *mp = NULL;
 294
 295     /* if we don't have a buffer yet, get one */
 296     if (bufsiz == 0) {
 297         buf = mh_xmalloc (BUFINCR);
 298         last_dst = buf;         /* XXX */
 299         bufsiz = BUFINCR - 6;  /* leave some slop */
 300         bufend = buf + bufsiz;
 301     }
 302     /*
 303      * If "orig" points to our buffer we can just pick up where we
 304      * left off.  Otherwise we have to copy orig into our buffer.
 305      */
 306     if (orig == buf)
 307         dst = last_dst;
 308     else if (!orig || !*orig) {
 309         dst = buf;
 310         *dst = '\0';
 311     } else {
 312         dst = last_dst;         /* XXX */
 313         CHECKMEM (orig);
 314         CPY (orig);
 315     }
 316
 317     /* concatenate all the new addresses onto 'buf' */
 318     for (isgroup = 0; (cp = getname (str)); ) {
 319         if ((mp = getm (cp, dfhost, dftype, AD_NAME, error)) == NULL) {
 320             snprintf (baddr, sizeof(baddr), "\t%s -- %s\n", cp, error);
 321             badaddrs = add (baddr, badaddrs);
 322             continue;
 323         }
 324         if (isgroup && (mp->m_gname || !mp->m_ingrp)) {
 325             *dst++ = ';';
 326             isgroup = 0;
 327         }
 328         if (insert (mp)) {
 329             /* if we get here we're going to add an address */
 330             if (dst != buf) {
 331                 *dst++ = ',';
 332                 *dst++ = ' ';
 333             }
 334             if (mp->m_gname) {
 335                 CHECKMEM (mp->m_gname);
 336                 CPY (mp->m_gname);
 337                 isgroup++;
 338             }
 339             sp = adrformat (mp);
 340             CHECKMEM (sp);
 341             CPY (sp);
 342         }
 343     }
 344
 345     if (isgroup)
 346         *dst++ = ';';
 347
 348     *dst = '\0';
 349     last_dst = dst;
 350     return (buf);
 351 }
 352
 353
 354 /*
 355  * fmt_scan will call this routine if the user includes the function
 356  * "(concataddr {component})" in a format string.  This behaves exactly
 357  * like formataddr, except that it does NOT suppress duplicate addresses
 358  * between calls.
 359  *
 360  * As an implementation detail: I thought about splitting out replformataddr()
 361  * into the generic part and duplicate-suppressing part, but the call to
 362  * insert() was buried deep within a couple of loops and I didn't see a
 363  * way to do it easily.  So instead we simply set a special flag to stop
 364  * the duplicate check and call replformataddr().
 365  */
 366 static char *
 367 replconcataddr(char *orig, char *str)
 368 {
 369     char *cp;
 370
 371     nodupcheck = 1;
 372     cp = replformataddr(orig, str);
 373     nodupcheck = 0;
 374     return cp;
 375 }
 376
 377 static int
 378 insert (struct mailname *np)
 379 {
 380     char buffer[BUFSIZ];
 381     register struct mailname *mp;
 382
 383     if (nodupcheck)
 384         return 1;
 385
 386     if (np->m_mbox == NULL)
 387         return 0;
 388
 389     for (mp = &mq; mp->m_next; mp = mp->m_next) {
 390         if (!strcasecmp (np->m_host ? np->m_host : "",
 391                          mp->m_next->m_host ? mp->m_next->m_host : "")  &&
 392             !strcasecmp (np->m_mbox ? np->m_mbox : "",
 393                          mp->m_next->m_mbox ? mp->m_next->m_mbox : ""))
 394             return 0;
 395     }
 396     if (!ccme && ismymbox (np))
 397         return 0;
 398
 399     if (querysw) {
 400         snprintf (buffer, sizeof(buffer), "Reply to %s? ", adrformat (np));
 401         if (!gans (buffer, anoyes))
 402         return 0;
 403     }
 404     mp->m_next = np;
 405
 406     return 1;
 407 }
 408
 409
 410 /*
 411  * Call the mhlproc
 412  *
 413  * This function expects that argument out has been fflushed by the caller.
 414  */
 415
 416 static void
 417 replfilter (FILE *in, FILE *out, char *filter, int fmtproc)
 418 {
 419     int pid;
 420     char *mhl;
 421     char *errstr;
 422     char **arglist;
 423     int argnum;
 424
 425     if (filter == NULL)
 426         return;
 427
 428     if (access (filter, R_OK) == NOTOK)
 429         adios (filter, "unable to read");
 430
 431     rewind (in);
 432     lseek (fileno(in), (off_t) 0, SEEK_SET);
 433
 434     switch (pid = fork()) {
 435         case NOTOK:
 436             adios ("fork", "unable to");
 437
 438         case OK:
 439             dup2 (fileno (in), fileno (stdin));
 440             dup2 (fileno (out), fileno (stdout));
 441             closefds (3);
 442
 443             /*
 444              * We're not allocating the memory for the extra arguments,
 445              * because we never call arglist_free().  But if we ever change
 446              * that be sure to use getcpy() for the extra arguments.
 447              */
 448             arglist = argsplit(mhlproc, &mhl, &argnum);
 449             arglist[argnum++] = "-form";
 450             arglist[argnum++] = filter;
 451             arglist[argnum++] = "-noclear";
 452
 453             switch (fmtproc) {
 454             case 1:
 455                 arglist[argnum++] = "-fmtproc";
 456                 arglist[argnum++] = formatproc;
 457                 break;
 458             case 0:
 459                 arglist[argnum++] = "-nofmtproc";
 460                 break;
 461             }
 462
 463             arglist[argnum++] = NULL;
 464
 465             execvp (mhl, arglist);
 466             errstr = strerror(errno);
 467             write(2, "unable to exec ", 15);
 468             write(2, mhlproc, strlen(mhlproc));
 469             write(2, ": ", 2);
 470             write(2, errstr, strlen(errstr));
 471             write(2, "\n", 1);
 472             _exit (-1);
 473
 474         default:
 475             if (pidXwait (pid, mhl))
 476                 done (1);
 477             fseek (out, 0L, SEEK_END);
 478             break;
 479     }
 480 }