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

   1 /* fmttest.c -- A program to help test and debug format instructions
   2  *
   3  * This code is Copyright (c) 2012, by the authors of nmh.  See the
   4  * COPYRIGHT file in the root directory of the nmh distribution for
   5  * complete copyright information.
   6  */
   7
   8 #include "h/mh.h"
   9 #include "sbr/fmt_new.h"
  10 #include "scansbr.h"
  11 #include "sbr/m_name.h"
  12 #include "sbr/m_getfld.h"
  13 #include "sbr/getarguments.h"
  14 #include "sbr/seq_setprev.h"
  15 #include "sbr/seq_save.h"
  16 #include "sbr/smatch.h"
  17 #include "sbr/snprintb.h"
  18 #include "sbr/getcpy.h"
  19 #include "sbr/m_convert.h"
  20 #include "sbr/getfolder.h"
  21 #include "sbr/folder_read.h"
  22 #include "sbr/folder_free.h"
  23 #include "sbr/context_save.h"
  24 #include "sbr/context_replace.h"
  25 #include "sbr/context_find.h"
  26 #include "sbr/brkstring.h"
  27 #include "sbr/ambigsw.h"
  28 #include "sbr/path.h"
  29 #include "sbr/print_version.h"
  30 #include "sbr/print_help.h"
  31 #include "sbr/seq_getnum.h"
  32 #include "sbr/error.h"
  33 #include "h/fmt_scan.h"
  34 #include "h/fmt_compile.h"
  35 #include "h/utils.h"
  36 #include "h/addrsbr.h"
  37 #include "h/done.h"
  38 #include "sbr/m_maildir.h"
  39 #include "sbr/terminal.h"
  40
  41 #define FMTTEST_SWITCHES \
  42     X("form formatfile", 0, FORMSW) \
  43     X("format string", 5, FMTSW) \
  44     X("address", 0, ADDRSW) \
  45     X("raw", 0, RAWSW) \
  46     X("date", 0, DATESW) \
  47     X("message", 0, MESSAGESW) \
  48     X("file", 0, FILESW) \
  49     X("nofile", 0, NFILESW) \
  50     X("-component-name component-text", 0, OTHERSW) \
  51     X("dupaddrs", 0, DUPADDRSW) \
  52     X("nodupaddrs", 0, NDUPADDRSW) \
  53     X("ccme", 0, CCMESW) \
  54     X("noccme", 0, NCCMESW) \
  55     X("outsize size-in-characters", 0, OUTSIZESW) \
  56     X("width column-width", 0, WIDTHSW) \
  57     X("msgnum number", 0, MSGNUMSW) \
  58     X("msgcur flag", 0, MSGCURSW) \
  59     X("msgsize size", 0, MSGSIZESW) \
  60     X("unseen flag", 0, UNSEENSW) \
  61     X("dump", 0, DUMPSW) \
  62     X("nodump", 0, NDUMPSW) \
  63     X("trace", 0, TRACESW) \
  64     X("notrace", 0, NTRACESW) \
  65     X("version", 0, VERSIONSW) \
  66     X("help", 0, HELPSW) \
  67
  68 #define X(sw, minchars, id) id,
  69 DEFINE_SWITCH_ENUM(FMTTEST);
  70 #undef X
  71
  72 #define X(sw, minchars, id) { sw, minchars, id },
  73 DEFINE_SWITCH_ARRAY(FMTTEST, switches);
  74 #undef X
  75
  76 /*
  77  * An array containing labels used for branch instructions
  78  */
  79
  80 static struct format **lvec = NULL;
  81 static int lused = 0;
  82 static int lallocated = 0;
  83
  84 enum mode_t { MESSAGE, ADDRESS, DATE, RAW };
  85 #define DEFADDRFORMAT "%<{error}%{error}: %{text}%|%(putstr(proper{text}))%>"
  86 #define DEFDATEFORMAT "%<(nodate{text})error: %{text}%|%(putstr(pretty{text}))%>"
  87
  88 /*
  89  * Context structure used by the tracing routines
  90  */
  91
  92 struct trace_context {
  93     int num;
  94     char *str;
  95     char *outbuf;
  96 };
  97
  98 /*
  99  * static prototypes
 100  */
 101 static void fmt_dump (char *, struct format *);
 102 static void dumpone(struct format *);
 103 static void initlabels(struct format *);
 104 static int findlabel(struct format *);
 105 static void assignlabel(struct format *);
 106 static char *f_typestr(int);
 107 static char *c_typestr(int);
 108 static char *c_flagsstr(int);
 109 static void litputs(const char *);
 110 static void litputc(char);
 111 static void process_addresses(struct format *, struct msgs_array *,
 112                               charstring_t, int, int *,
 113                               struct fmt_callbacks *);
 114 static void process_raw(struct format *, struct msgs_array *, charstring_t,
 115                         int, int *, struct fmt_callbacks *);
 116 static void process_messages(struct format *, struct msgs_array *,
 117                              struct msgs_array *, charstring_t, char *, int,
 118                              int, int *, struct fmt_callbacks *);
 119 static void process_single_file(FILE *, struct msgs_array *, int *, int,
 120                                 struct format *, charstring_t, int,
 121                                 struct fmt_callbacks *);
 122 static void test_trace(void *, struct format *, int, char *, const char *);
 123 static char *test_formataddr(char *, char *);
 124 static char *test_concataddr(char *, char *);
 125 static int insert(struct mailname *);
 126 static void mlistfree(void);
 127
 128 static bool nodupcheck;         /* If set, no check for duplicates */
 129 static bool ccme;               /* Should I cc myself? */
 130 static struct mailname mq;      /* Mail addresses to check for duplicates */
 131 static char *dummy = "dummy";
 132
 133 int
 134 main (int argc, char **argv)
 135 {
 136     char *cp, *form = NULL, *format = NULL, *defformat = FORMAT, *folder = NULL;
 137     char buf[BUFSIZ], *nfs, **argp, **arguments;
 138     charstring_t buffer;
 139     struct format *fmt;
 140     struct comp *cptr;
 141     struct msgs_array msgs = { 0, 0, NULL }, compargs = { 0, 0, NULL};
 142     bool dump = false;
 143     int i;
 144     bool dupaddrs = true;
 145     bool trace = false;
 146     int files = 0;
 147     int colwidth = -1, msgnum = -1, msgcur = -1, msgsize = -1, msgunseen = -1;
 148     enum mode_t mode = MESSAGE;
 149     int dat[5];
 150     struct fmt_callbacks cb, *cbp = NULL;
 151
 152     if (nmh_init(argv[0], true, true)) { return 1; }
 153
 154     arguments = getarguments (invo_name, argc, argv, 1);
 155     argp = arguments;
 156
 157     bool outputsize_given = false;
 158     int outputsize;
 159     while ((cp = *argp++)) {
 160         if (*cp == '-') {
 161             /*
 162              * A -- means that we have a component name (like pick);
 163              * save the component name and the next argument for the text.
 164              */
 165             if (*++cp == '-') {
 166                 if (*++cp == '\0')
 167                     die("missing component name after --");
 168                 app_msgarg(&compargs, cp);
 169                 /* Grab next argument for component text */
 170                 if (!(cp = *argp++))
 171                     die("missing argument to %s", argp[-2]);
 172                 app_msgarg(&compargs, cp);
 173                 continue;
 174             }
 175             switch (smatch (cp, switches)) {
 176                 case AMBIGSW:
 177                     ambigsw (cp, switches);
 178                     done (1);
 179                 case UNKWNSW:
 180                     die("-%s unknown", cp);
 181
 182                 case HELPSW:
 183                     snprintf (buf, sizeof(buf), "%s [switches]", invo_name);
 184                     print_help (buf, switches, 1);
 185                     done (0);
 186                 case VERSIONSW:
 187                     print_version(invo_name);
 188                     done (0);
 189                 case OTHERSW:
 190                     die("internal argument error!");
 191                     continue;
 192
 193                 case OUTSIZESW:
 194                     outputsize_given = true;
 195                     if (!(cp = *argp++) || *cp == '-')
 196                         die("missing argument to %s", argp[-2]);
 197                     if (strcmp(cp, "max") == 0)
 198                         outputsize = INT_MAX;
 199                     else if (strcmp(cp, "width") == 0)
 200                         outputsize = sc_width();
 201                     else
 202                         outputsize = atoi(cp);
 203                     continue;
 204
 205                 case FORMSW:
 206                     if (!(form = *argp++) || *form == '-')
 207                         die("missing argument to %s", argp[-2]);
 208                     format = NULL;
 209                     continue;
 210                 case FMTSW:
 211                     if (!(format = *argp++) || *format == '-')
 212                         die("missing argument to %s", argp[-2]);
 213                     form = NULL;
 214                     continue;
 215
 216                 case TRACESW:
 217                     trace = true;
 218                     continue;
 219                 case NTRACESW:
 220                     trace = false;
 221                     continue;
 222
 223                 case ADDRSW:
 224                     mode = ADDRESS;
 225                     defformat = DEFADDRFORMAT;
 226                     continue;
 227                 case RAWSW:
 228                     mode = RAW;
 229                     continue;
 230                 case MESSAGESW:
 231                     mode = MESSAGE;
 232                     defformat = FORMAT;
 233                     dupaddrs = false;
 234                     continue;
 235                 case DATESW:
 236                     mode = DATE;
 237                     defformat = DEFDATEFORMAT;
 238                     continue;
 239
 240                 case FILESW:
 241                     files++;
 242                     continue;
 243                 case NFILESW:
 244                     files = 0;
 245                     continue;
 246
 247                 case DUPADDRSW:
 248                     dupaddrs = true;
 249                     continue;
 250                 case NDUPADDRSW:
 251                     dupaddrs = false;
 252                     continue;
 253
 254                 case CCMESW:
 255                     ccme = true;
 256                     continue;
 257                 case NCCMESW:
 258                     ccme = false;
 259                     continue;
 260
 261                 case WIDTHSW:
 262                     if (!(cp = *argp++) || *cp == '-')
 263                         die("missing argument to %s", argp[-2]);
 264                     colwidth = atoi(cp);
 265                     continue;
 266                 case MSGNUMSW:
 267                     if (!(cp = *argp++) || *cp == '-')
 268                         die("missing argument to %s", argp[-2]);
 269                     msgnum = atoi(cp);
 270                     continue;
 271                 case MSGCURSW:
 272                     if (!(cp = *argp++) || *cp == '-')
 273                         die("missing argument to %s", argp[-2]);
 274                     msgcur = atoi(cp);
 275                     continue;
 276                 case MSGSIZESW:
 277                     if (!(cp = *argp++) || *cp == '-')
 278                         die("missing argument to %s", argp[-2]);
 279                     msgsize = atoi(cp);
 280                     continue;
 281                 case UNSEENSW:
 282                     if (!(cp = *argp++) || *cp == '-')
 283                         die("missing argument to %s", argp[-2]);
 284                     msgunseen = atoi(cp);
 285                     continue;
 286
 287                 case DUMPSW:
 288                     dump = true;
 289                     continue;
 290                 case NDUMPSW:
 291                     dump = false;
 292                     continue;
 293
 294             }
 295         }
 296
 297         /*
 298          * Only interpret as a folder if we're in message mode
 299          */
 300
 301         if (mode == MESSAGE && !files && (*cp == '+' || *cp == '@')) {
 302             if (folder)
 303                 die("only one folder at a time!");
 304             folder = pluspath (cp);
 305         } else
 306             app_msgarg(&msgs, cp);
 307     }
 308
 309     /*
 310      * Here's our weird heuristic:
 311      *
 312      * - We allow -dump without any other arguments.
 313      * - If you've given any component arguments, we don't require any
 314      *   other arguments.
 315      * - The arguments are interpreted as folders/messages _if_ we're in
 316      *   message mode, otherwise pass as strings in the text component.
 317      */
 318
 319    if (!dump && compargs.size == 0 && msgs.size == 0) {
 320         die("usage: [switches] [+folder] msgs | strings...");
 321    }
 322
 323    /*
 324     * If you're picking "raw" as a mode, then you have to select
 325     * a format.
 326     */
 327
 328    if (mode == RAW && form == NULL && format == NULL) {
 329         die("You must specify a format with -form or -format when "
 330                "using -raw");
 331    }
 332
 333     /*
 334      * Get new format string.  Must be before chdir().
 335      */
 336     nfs = new_fs (form, format, defformat);
 337     (void) fmt_compile(nfs, &fmt, 1);
 338
 339     if (dump || trace) {
 340         initlabels(fmt);
 341         if (dump) {
 342             fmt_dump(nfs, fmt);
 343             if (compargs.size == 0 && msgs.size == 0)
 344                 done(0);
 345         }
 346     }
 347
 348     buffer = charstring_create(BUFSIZ);
 349
 350     if (!outputsize_given) {
 351         outputsize = mode == MESSAGE ? sc_width() : INT_MAX;
 352     }
 353
 354     dat[0] = msgnum;
 355     dat[1] = msgcur;
 356     dat[2] = msgsize;
 357     dat[3] = colwidth == -1 ? outputsize : colwidth;
 358     dat[4] = msgunseen;
 359
 360     /*
 361      * If we want to provide our own formataddr, concactaddr, or tracing
 362      * callback, do that now.  Also, prime ismymbox if we use it.
 363      */
 364
 365     if (!dupaddrs || trace) {
 366         ZERO(&cb);
 367         cbp = &cb;
 368
 369         if (!dupaddrs) {
 370             cb.formataddr = test_formataddr;
 371             cb.concataddr = test_concataddr;
 372             if (!ccme)
 373                 ismymbox(NULL);
 374         }
 375
 376         if (trace) {
 377             struct trace_context *ctx;
 378
 379             NEW(ctx);
 380             ctx->num = -1;
 381             ctx->str = dummy;
 382             ctx->outbuf = mh_xstrdup("");
 383
 384             cb.trace_func = test_trace;
 385             cb.trace_context = ctx;
 386         }
 387     }
 388
 389     if (mode == MESSAGE) {
 390         process_messages(fmt, &compargs, &msgs, buffer, folder, outputsize,
 391                          files, dat, cbp);
 392     } else {
 393         if (compargs.size) {
 394             for (i = 0; i < compargs.size; i += 2) {
 395                 cptr = fmt_findcomp(compargs.msgs[i]);
 396                 if (cptr)
 397                     cptr->c_text = getcpy(compargs.msgs[i + 1]);
 398             }
 399         }
 400
 401         if (mode == ADDRESS) {
 402             process_addresses(fmt, &msgs, buffer, outputsize, dat, cbp);
 403         } else /* Fall-through for RAW or DATE */
 404             process_raw(fmt, &msgs, buffer, outputsize, dat, cbp);
 405     }
 406
 407     charstring_free(buffer);
 408     fmt_free(fmt, 1);
 409
 410     done(0);
 411     return 1;
 412 }
 413
 414 /*
 415  * Process each address with fmt_scan().
 416  */
 417
 418 struct pqpair {
 419     char *pq_text;
 420     char *pq_error;
 421     struct pqpair *pq_next;
 422 };
 423
 424 static void
 425 process_addresses(struct format *fmt, struct msgs_array *addrs,
 426                   charstring_t buffer, int outwidth, int *dat,
 427                   struct fmt_callbacks *cb)
 428 {
 429     int i;
 430     char *cp, error[BUFSIZ];
 431     struct mailname *mp;
 432     struct pqpair *p, *q;
 433     struct pqpair pq;
 434     struct comp *c;
 435
 436     if (dat[0] == -1)
 437         dat[0] = 0;
 438     if (dat[1] == -1)
 439         dat[1] = 0;
 440     if (dat[2] == -1)
 441         dat[2] = 0;
 442     if (dat[4] == -1)
 443         dat[4] = 0;
 444
 445     for (i = 0; i < addrs->size; i++) {
 446         (q = &pq)->pq_next = NULL;
 447         while ((cp = getname(addrs->msgs[i]))) {
 448             NEW0(p);
 449             if ((mp = getm(cp, NULL, 0, error, sizeof(error))) == NULL) {
 450                 p->pq_text = mh_xstrdup(cp);
 451                 p->pq_error = mh_xstrdup(error);
 452             } else {
 453                 p->pq_text = getcpy(mp->m_text);
 454                 mnfree(mp);
 455             }
 456             q = (q->pq_next = p);
 457         }
 458
 459         for (p = pq.pq_next; p; p = q) {
 460             c = fmt_findcomp("text");
 461             if (c) {
 462                 free(c->c_text);
 463                 c->c_text = p->pq_text;
 464                 p->pq_text = NULL;
 465             }
 466             c = fmt_findcomp("error");
 467             if (c) {
 468                 free(c->c_text);
 469                 c->c_text = p->pq_error;
 470                 p->pq_error = NULL;
 471             }
 472
 473             fmt_scan(fmt, buffer, outwidth, dat, cb);
 474             fputs(charstring_buffer(buffer), stdout);
 475             charstring_clear(buffer);
 476             mlistfree();
 477
 478             free(p->pq_text);
 479             free(p->pq_error);
 480             q = p->pq_next;
 481             free(p);
 482         }
 483     }
 484 }
 485
 486 /*
 487  * Process messages and run them through the format engine.  A lot taken
 488  * from scan.c.
 489  */
 490
 491 static void
 492 process_messages(struct format *fmt, struct msgs_array *comps,
 493                  struct msgs_array *msgs, charstring_t buffer, char *folder,
 494                  int outwidth, int files, int *dat,
 495                  struct fmt_callbacks *cb)
 496 {
 497     int i, msgnum, msgsize = dat[2], num = dat[0], cur = dat[1];
 498     int num_unseen_seq = 0;
 499     ivector_t seqnum = ivector_create (0);
 500     char *maildir, *cp;
 501     struct msgs *mp;
 502     FILE *in;
 503
 504     /*
 505      * If 'files' is set, short-circuit everything else and just process
 506      * everything now.
 507      */
 508
 509     if (files) {
 510         for (i = 0; i < msgs->size; i++) {
 511             if ((in = fopen(cp = msgs->msgs[i], "r")) == NULL) {
 512                 admonish(cp, "unable to open file");
 513                 continue;
 514             }
 515             process_single_file(in, comps, dat, msgsize, fmt, buffer,
 516                                 outwidth, cb);
 517         }
 518
 519         return;
 520     }
 521
 522     if (! folder)
 523         folder = getfolder(1);
 524
 525     maildir = m_maildir(folder);
 526
 527     if (chdir(maildir) < 0)
 528         adios(maildir, "unable to change directory to");
 529
 530     if (!(mp = folder_read(folder, 1)))
 531         die("unable to read folder %s", folder);
 532
 533     if (mp->nummsg == 0)
 534         die("no messages in %s", folder);
 535
 536     for (i = 0; i < msgs->size; i++)
 537         if (!m_convert(mp, msgs->msgs[i]))
 538             done(1);
 539     seq_setprev(mp);                    /* set the Previous-Sequence */
 540
 541     context_replace(pfolder, folder);   /* update current folder */
 542     seq_save(mp);                       /* synchronize message sequences */
 543     context_save();                     /* save the context file */
 544
 545     /*
 546      * We want to set the unseen flag if requested, so we have to check
 547      * the unseen sequence as well.
 548      */
 549
 550     if (dat[4] == -1) {
 551         if ((cp = context_find(usequence)) && *cp) {
 552             char **ap, *dp;
 553
 554             dp = mh_xstrdup(cp);
 555             ap = brkstring(dp, " ", "\n");
 556             for (i = 0; ap && *ap; i++, ap++)
 557                 ivector_push_back (seqnum, seq_getnum(mp, *ap));
 558
 559             num_unseen_seq = i;
 560             free(dp);
 561         }
 562     }
 563
 564     for (msgnum = mp->lowsel; msgnum <= mp->hghsel; msgnum++) {
 565         if (is_selected(mp, msgnum)) {
 566             if ((in = fopen(cp = m_name(msgnum), "r")) == NULL) {
 567                 admonish(cp, "unable to open message");
 568                 continue;
 569             }
 570
 571             fmt_freecomptext();
 572
 573             if (num == -1)
 574                 dat[0] = msgnum;
 575
 576             if (cur == -1)
 577                 dat[1] = msgnum == mp->curmsg;
 578
 579             /*
 580              * Check to see if this is in the unseen sequence
 581              */
 582
 583             dat[4] = 0;
 584             for (i = 0; i < num_unseen_seq; i++) {
 585                 if (in_sequence(mp, ivector_at (seqnum, i), msgnum)) {
 586                     dat[4] = 1;
 587                     break;
 588                 }
 589             }
 590
 591             /*
 592              * Read in the message and process the components
 593              */
 594
 595             process_single_file(in, comps, dat, msgsize, fmt, buffer,
 596                                 outwidth, cb);
 597         }
 598     }
 599
 600     ivector_free (seqnum);
 601     folder_free(mp);
 602 }
 603
 604 /*
 605  * Process a single file in message mode
 606  */
 607
 608 static void
 609 process_single_file(FILE *in, struct msgs_array *comps, int *dat, int msgsize,
 610                     struct format *fmt, charstring_t buffer, int outwidth,
 611                     struct fmt_callbacks *cb)
 612 {
 613     int i, state;
 614     char name[NAMESZ], rbuf[NMH_BUFSIZ];
 615     m_getfld_state_t gstate;
 616     struct comp *c;
 617     int bufsz;
 618
 619     /*
 620      * Get our size if we didn't include one
 621      */
 622
 623     if (msgsize == -1) {
 624         struct stat st;
 625
 626         if (fstat(fileno(in), &st) < 0)
 627             dat[2] = 0;
 628         else
 629             dat[2] = st.st_size;
 630     }
 631
 632     /*
 633      * Initialize everything else
 634      */
 635
 636     if (dat[0] == -1)
 637         dat[0] = 0;
 638     if (dat[1] == -1)
 639         dat[1] = 0;
 640     if (dat[4] == -1)
 641         dat[4] = 0;
 642
 643     /*
 644      * Read in the message and process the components
 645      */
 646
 647     gstate = m_getfld_state_init(in);
 648     for (;;) {
 649         bufsz = sizeof(rbuf);
 650         state = m_getfld2(&gstate, name, rbuf, &bufsz);
 651         switch (state) {
 652         case FLD:
 653         case FLDPLUS:
 654             i = fmt_addcomptext(name, rbuf);
 655             if (i != -1) {
 656                 while (state == FLDPLUS) {
 657                     bufsz = sizeof(rbuf);
 658                     state = m_getfld2(&gstate, name, rbuf, &bufsz);
 659                     fmt_appendcomp(i, name, rbuf);
 660                 }
 661             }
 662
 663             while (state == FLDPLUS) {
 664                 bufsz = sizeof(rbuf);
 665                 state = m_getfld2(&gstate, name, rbuf, &bufsz);
 666             }
 667             break;
 668
 669         case BODY:
 670             if (fmt_findcomp("body")) {
 671                 if ((i = strlen(rbuf)) < outwidth) {
 672                     bufsz = min (outwidth, (int) sizeof rbuf - i);
 673                     m_getfld2(&gstate, name, rbuf + i, &bufsz);
 674                 }
 675
 676                 fmt_addcomptext("body", rbuf);
 677             }
 678             goto finished;
 679
 680         default:
 681             goto finished;
 682         }
 683     }
 684 finished:
 685     fclose(in);
 686     m_getfld_state_destroy(&gstate);
 687
 688     /*
 689      * Do this now to override any components in the original message
 690      */
 691     if (comps->size) {
 692         for (i = 0; i < comps->size; i += 2) {
 693             c = fmt_findcomp(comps->msgs[i]);
 694             if (c) {
 695                 free(c->c_text);
 696                 c->c_text = getcpy(comps->msgs[i + 1]);
 697             }
 698         }
 699     }
 700     fmt_scan(fmt, buffer, outwidth, dat, cb);
 701     fputs(charstring_buffer (buffer), stdout);
 702     charstring_clear(buffer);
 703     mlistfree();
 704 }
 705
 706 /*
 707  * Run text through the format engine with no special processing
 708  */
 709
 710 static void
 711 process_raw(struct format *fmt, struct msgs_array *text, charstring_t buffer,
 712             int outwidth, int *dat, struct fmt_callbacks *cb)
 713 {
 714     int i;
 715     struct comp *c;
 716
 717     if (dat[0] == -1)
 718         dat[0] = 0;
 719     if (dat[1] == -1)
 720         dat[1] = 0;
 721     if (dat[2] == -1)
 722         dat[2] = 0;
 723     if (dat[4] == -1)
 724         dat[4] = 0;
 725
 726     c = fmt_findcomp("text");
 727
 728     for (i = 0; i < text->size; i++) {
 729         if (c != NULL) {
 730             free(c->c_text);
 731             c->c_text = getcpy(text->msgs[i]);
 732         }
 733
 734         fmt_scan(fmt, buffer, outwidth, dat, cb);
 735         fputs(charstring_buffer (buffer), stdout);
 736         charstring_clear(buffer);
 737         mlistfree();
 738     }
 739 }
 740
 741 /*
 742  * Our basic tracing support callback.
 743  *
 744  * Print out each instruction as it's executed, including the values of
 745  * the num and str registers if they've changed.
 746  */
 747
 748 static void
 749 test_trace(void *context, struct format *fmt, int num, char *str,
 750            const char *outbuf)
 751 {
 752     struct trace_context *ctx = (struct trace_context *) context;
 753     bool changed = false;
 754
 755     dumpone(fmt);
 756
 757     if (num != ctx->num) {
 758         printf("num=%d", num);
 759         ctx->num = num;
 760         changed = true;
 761     }
 762
 763     if (str != ctx->str) {
 764         if (changed)
 765             putchar(' ');
 766         changed = true;
 767         fputs("str=", stdout);
 768         litputs(str);
 769         ctx->str = str;
 770     }
 771
 772     if (changed)
 773         putchar('\n');
 774
 775     if (strcmp(outbuf, ctx->outbuf) != 0) {
 776         fputs("outbuf=", stdout);
 777         litputs(outbuf);
 778         putchar('\n');
 779         free(ctx->outbuf);
 780         ctx->outbuf = mh_xstrdup(outbuf);
 781     }
 782 }
 783
 784 static void
 785 fmt_dump (char *nfs, struct format *fmth)
 786 {
 787         struct format *fmt;
 788
 789         printf("Instruction dump of format string: \n%s\n", nfs);
 790
 791         /* Dump them out! */
 792         for (fmt = fmth; fmt; ++fmt) {
 793                 dumpone(fmt);
 794                 if (fmt->f_type == FT_DONE && fmt->f_value == 0)
 795                         break;
 796         }
 797 }
 798
 799 static void
 800 dumpone(struct format *fmt)
 801 {
 802         int i;
 803
 804         if ((i = findlabel(fmt)) >= 0)
 805                 printf("L%d:", i);
 806         putchar('\t');
 807
 808         fputs(f_typestr((int)fmt->f_type), stdout);
 809
 810         switch (fmt->f_type) {
 811
 812         case FT_COMP:
 813         case FT_LS_COMP:
 814         case FT_LV_COMPFLAG:
 815         case FT_LV_COMP:
 816                 fputs(", comp ", stdout);
 817                 litputs(fmt->f_comp->c_name);
 818                 if (fmt->f_comp->c_type)
 819                         printf(", c_type %s", c_typestr(fmt->f_comp->c_type));
 820                 if (fmt->f_comp->c_flags)
 821                         printf(", c_flags %s", c_flagsstr(fmt->f_comp->c_flags));
 822                 break;
 823
 824         case FT_LV_SEC:
 825         case FT_LV_MIN:
 826         case FT_LV_HOUR:
 827         case FT_LV_MDAY:
 828         case FT_LV_MON:
 829         case FT_LS_MONTH:
 830         case FT_LS_LMONTH:
 831         case FT_LS_ZONE:
 832         case FT_LV_YEAR:
 833         case FT_LV_WDAY:
 834         case FT_LS_DAY:
 835         case FT_LS_WEEKDAY:
 836         case FT_LV_YDAY:
 837         case FT_LV_ZONE:
 838         case FT_LV_CLOCK:
 839         case FT_LV_RCLOCK:
 840         case FT_LV_DAYF:
 841         case FT_LV_ZONEF:
 842         case FT_LV_DST:
 843         case FT_LS_822DATE:
 844         case FT_LS_PRETTY:
 845         case FT_LOCALDATE:
 846         case FT_GMTDATE:
 847         case FT_PARSEDATE:
 848                 fputs(", c_name ", stdout);
 849                 litputs(fmt->f_comp->c_name);
 850                 if (fmt->f_comp->c_type)
 851                         printf(", c_type %s", c_typestr(fmt->f_comp->c_type));
 852                 if (fmt->f_comp->c_flags)
 853                         printf(", c_flags %s", c_flagsstr(fmt->f_comp->c_flags));
 854                 break;
 855
 856         case FT_LS_ADDR:
 857         case FT_LS_PERS:
 858         case FT_LS_MBOX:
 859         case FT_LS_HOST:
 860         case FT_LS_PATH:
 861         case FT_LS_GNAME:
 862         case FT_LS_NOTE:
 863         case FT_LS_822ADDR:
 864         case FT_LV_HOSTTYPE:
 865         case FT_LV_INGRPF:
 866         case FT_LV_NOHOSTF:
 867         case FT_LS_FRIENDLY:
 868         case FT_PARSEADDR:
 869         case FT_MYMBOX:
 870         case FT_GETMYMBOX:
 871         case FT_GETMYADDR:
 872                 fputs(", c_name ", stdout);
 873                 litputs(fmt->f_comp->c_name);
 874                 if (fmt->f_comp->c_type)
 875                         printf(", c_type %s", c_typestr(fmt->f_comp->c_type));
 876                 if (fmt->f_comp->c_flags)
 877                         printf(", c_flags %s", c_flagsstr(fmt->f_comp->c_flags));
 878                 break;
 879
 880         case FT_COMPF:
 881                 printf(", width %d, fill '", fmt->f_width);
 882                 litputc(fmt->f_fill);
 883                 fputs("' name ", stdout);
 884                 litputs(fmt->f_comp->c_name);
 885                 if (fmt->f_comp->c_type)
 886                         printf(", c_type %s", c_typestr(fmt->f_comp->c_type));
 887                 if (fmt->f_comp->c_flags)
 888                         printf(", c_flags %s", c_flagsstr(fmt->f_comp->c_flags));
 889                 break;
 890
 891         case FT_STRF:
 892         case FT_NUMF:
 893                 printf(", width %d, fill '", fmt->f_width);
 894                 litputc(fmt->f_fill);
 895                 putchar('\'');
 896                 break;
 897
 898         case FT_LIT:
 899                 putchar(' ');
 900                 litputs(fmt->f_text);
 901                 break;
 902
 903         case FT_LITF:
 904                 printf(", width %d, fill '", fmt->f_width);
 905                 litputc(fmt->f_fill);
 906                 fputs("' ", stdout);
 907                 litputs(fmt->f_text);
 908                 break;
 909
 910         case FT_CHAR:
 911                 putchar(' ');
 912                 putchar('\'');
 913                 litputc(fmt->f_char);
 914                 putchar('\'');
 915                 break;
 916
 917
 918         case FT_IF_S:
 919         case FT_IF_S_NULL:
 920         case FT_IF_MATCH:
 921         case FT_IF_AMATCH:
 922                 fputs(" continue else goto", stdout);
 923                 /* FALLTHRU */
 924         case FT_GOTO:
 925                 i = findlabel(fmt + fmt->f_skip);
 926                 printf(" L%d", i);
 927                 break;
 928
 929         case FT_IF_V_EQ:
 930         case FT_IF_V_NE:
 931         case FT_IF_V_GT:
 932                 i = findlabel(fmt + fmt->f_skip);
 933                 printf(" %d continue else goto L%d", fmt->f_value, i);
 934                 break;
 935
 936         case FT_V_EQ:
 937         case FT_V_NE:
 938         case FT_V_GT:
 939         case FT_LV_LIT:
 940         case FT_LV_PLUS_L:
 941         case FT_LV_MINUS_L:
 942         case FT_LV_MULTIPLY_L:
 943         case FT_LV_DIVIDE_L:
 944         case FT_LV_MODULO_L:
 945                 printf(" value %d", fmt->f_value);
 946                 break;
 947
 948         case FT_LS_LIT:
 949                 fputs(" str ", stdout);
 950                 litputs(fmt->f_text);
 951                 break;
 952
 953         case FT_LS_GETENV:
 954                 fputs(" getenv ", stdout);
 955                 litputs(fmt->f_text);
 956                 break;
 957
 958         case FT_LS_DECODECOMP:
 959                 fputs(", comp ", stdout);
 960                 litputs(fmt->f_comp->c_name);
 961                 break;
 962
 963         case FT_LS_DECODE:
 964                 break;
 965
 966         case FT_LS_TRIM:
 967                 printf(", width %d", fmt->f_width);
 968                 break;
 969
 970         case FT_LV_DAT:
 971                 printf(", value dat[%d]", fmt->f_value);
 972                 break;
 973         }
 974         putchar('\n');
 975 }
 976
 977 /*
 978  * Iterate over all instructions and assign labels to the targets of
 979  * branch statements
 980  */
 981
 982 static void
 983 initlabels(struct format *fmth)
 984 {
 985         struct format *fmt, *addr;
 986         int i;
 987
 988         /* Assign labels */
 989         for (fmt = fmth; fmt; ++fmt) {
 990                 i = fmt->f_type;
 991                 if (i == FT_IF_S ||
 992                     i == FT_IF_S_NULL ||
 993                     i == FT_IF_V_EQ ||
 994                     i == FT_IF_V_NE ||
 995                     i == FT_IF_V_GT ||
 996                     i == FT_IF_MATCH ||
 997                     i == FT_IF_AMATCH ||
 998                     i == FT_GOTO) {
 999                         addr = fmt + fmt->f_skip;
1000                         if (findlabel(addr) < 0)
1001                                 assignlabel(addr);
1002                 }
1003                 if (fmt->f_type == FT_DONE && fmt->f_value == 0)
1004                         break;
1005         }
1006 }
1007
1008
1009 static int
1010 findlabel(struct format *addr)
1011 {
1012         int i;
1013
1014         for (i = 0; i < lused; ++i)
1015                 if (addr == lvec[i])
1016                         return i;
1017         return -1;
1018 }
1019
1020 static void
1021 assignlabel(struct format *addr)
1022 {
1023         if (lused >= lallocated) {
1024                 lallocated += 64;
1025                 lvec = (struct format **)
1026                         mh_xrealloc(lvec, sizeof(struct format *) * lallocated);
1027         }
1028
1029         lvec[lused++] = addr;
1030 }
1031
1032 static char *
1033 f_typestr(int t)
1034 {
1035         static char buf[32];
1036
1037         switch (t) {
1038         case FT_COMP: return "COMP";
1039         case FT_COMPF: return "COMPF";
1040         case FT_LIT: return "LIT";
1041         case FT_LITF: return "LITF";
1042         case FT_CHAR: return "CHAR";
1043         case FT_NUM: return "NUM";
1044         case FT_NUMF: return "NUMF";
1045         case FT_STR: return "STR";
1046         case FT_STRF: return "STRF";
1047         case FT_STRFW: return "STRFW";
1048         case FT_STRLIT: return "STRLIT";
1049         case FT_STRLITZ: return "STRLITZ";
1050         case FT_PUTADDR: return "PUTADDR";
1051         case FT_LS_COMP: return "LS_COMP";
1052         case FT_LS_LIT: return "LS_LIT";
1053         case FT_LS_GETENV: return "LS_GETENV";
1054         case FT_LS_CFIND: return "LS_CFIND";
1055         case FT_LS_DECODECOMP: return "LS_DECODECOMP";
1056         case FT_LS_DECODE: return "LS_DECODE";
1057         case FT_LS_TRIM: return "LS_TRIM";
1058         case FT_LV_COMP: return "LV_COMP";
1059         case FT_LV_COMPFLAG: return "LV_COMPFLAG";
1060         case FT_LV_LIT: return "LV_LIT";
1061         case FT_LV_DAT: return "LV_DAT";
1062         case FT_LV_STRLEN: return "LV_STRLEN";
1063         case FT_LV_PLUS_L: return "LV_PLUS_L";
1064         case FT_LV_MINUS_L: return "LV_MINUS_L";
1065         case FT_LV_MULTIPLY_L: return "LV_MULTIPLY_L";
1066         case FT_LV_DIVIDE_L: return "LV_DIVIDE_L";
1067         case FT_LV_MODULO_L: return "LV_MODULO_L";
1068         case FT_LV_CHAR_LEFT: return "LV_CHAR_LEFT";
1069         case FT_LS_MONTH: return "LS_MONTH";
1070         case FT_LS_LMONTH: return "LS_LMONTH";
1071         case FT_LS_ZONE: return "LS_ZONE";
1072         case FT_LS_DAY: return "LS_DAY";
1073         case FT_LS_WEEKDAY: return "LS_WEEKDAY";
1074         case FT_LS_822DATE: return "LS_822DATE";
1075         case FT_LS_PRETTY: return "LS_PRETTY";
1076         case FT_LS_KILO: return "LS_KILO";
1077         case FT_LS_KIBI: return "LS_KIBI";
1078         case FT_LV_SEC: return "LV_SEC";
1079         case FT_LV_MIN: return "LV_MIN";
1080         case FT_LV_HOUR: return "LV_HOUR";
1081         case FT_LV_MDAY: return "LV_MDAY";
1082         case FT_LV_MON: return "LV_MON";
1083         case FT_LV_YEAR: return "LV_YEAR";
1084         case FT_LV_YDAY: return "LV_YDAY";
1085         case FT_LV_WDAY: return "LV_WDAY";
1086         case FT_LV_ZONE: return "LV_ZONE";
1087         case FT_LV_CLOCK: return "LV_CLOCK";
1088         case FT_LV_RCLOCK: return "LV_RCLOCK";
1089         case FT_LV_DAYF: return "LV_DAYF";
1090         case FT_LV_DST: return "LV_DST";
1091         case FT_LV_ZONEF: return "LV_ZONEF";
1092         case FT_LS_PERS: return "LS_PERS";
1093         case FT_LS_MBOX: return "LS_MBOX";
1094         case FT_LS_HOST: return "LS_HOST";
1095         case FT_LS_PATH: return "LS_PATH";
1096         case FT_LS_GNAME: return "LS_GNAME";
1097         case FT_LS_NOTE: return "LS_NOTE";
1098         case FT_LS_ADDR: return "LS_ADDR";
1099         case FT_LS_822ADDR: return "LS_822ADDR";
1100         case FT_LS_FRIENDLY: return "LS_FRIENDLY";
1101         case FT_LV_HOSTTYPE: return "LV_HOSTTYPE";
1102         case FT_LV_INGRPF: return "LV_INGRPF";
1103         case FT_LS_UNQUOTE: return "LS_UNQUOTE";
1104         case FT_LV_NOHOSTF: return "LV_NOHOSTF";
1105         case FT_LOCALDATE: return "LOCALDATE";
1106         case FT_GMTDATE: return "GMTDATE";
1107         case FT_PARSEDATE: return "PARSEDATE";
1108         case FT_PARSEADDR: return "PARSEADDR";
1109         case FT_FORMATADDR: return "FORMATADDR";
1110         case FT_CONCATADDR: return "CONCATADDR";
1111         case FT_MYMBOX: return "MYMBOX";
1112         case FT_GETMYMBOX: return "GETMYMBOX";
1113         case FT_GETMYADDR: return "GETMYADDR";
1114         case FT_SAVESTR: return "SAVESTR";
1115         case FT_DONE: return "DONE";
1116         case FT_PAUSE: return "PAUSE";
1117         case FT_NOP: return "NOP";
1118         case FT_GOTO: return "GOTO";
1119         case FT_IF_S_NULL: return "IF_S_NULL";
1120         case FT_IF_S: return "IF_S";
1121         case FT_IF_V_EQ: return "IF_V_EQ";
1122         case FT_IF_V_NE: return "IF_V_NE";
1123         case FT_IF_V_GT: return "IF_V_GT";
1124         case FT_IF_MATCH: return "IF_MATCH";
1125         case FT_IF_AMATCH: return "IF_AMATCH";
1126         case FT_S_NULL: return "S_NULL";
1127         case FT_S_NONNULL: return "S_NONNULL";
1128         case FT_V_EQ: return "V_EQ";
1129         case FT_V_NE: return "V_NE";
1130         case FT_V_GT: return "V_GT";
1131         case FT_V_MATCH: return "V_MATCH";
1132         case FT_V_AMATCH: return "V_AMATCH";
1133         default:
1134                 snprintf(buf, sizeof(buf), "/* ??? #%d */", t);
1135                 return buf;
1136         }
1137 }
1138
1139 static char *
1140 c_typestr(int t)
1141 {
1142         static char buf[64];
1143
1144         snprintb(buf, sizeof(buf), t, CT_BITS);
1145         return buf;
1146 }
1147
1148 static char *
1149 c_flagsstr(int t)
1150 {
1151         static char buf[64];
1152
1153         snprintb(buf, sizeof(buf), t, CF_BITS);
1154         return buf;
1155 }
1156
1157 static void
1158 litputs(const char *s)
1159 {
1160         if (s) {
1161                 putchar('"');
1162                 while (*s)
1163                         litputc(*s++);
1164                 putchar('"');
1165         } else
1166                 fputs("<nil>", stdout);
1167 }
1168
1169 static void
1170 litputc(char c)
1171 {
1172         if (c & ~ 0177) {
1173                 printf("\\x%02x", (unsigned char) c);
1174         } else if (c < 0x20 || c == 0177) {
1175                 if (c == '\b') {
1176                         putchar('\\');
1177                         putchar('b');
1178                 } else if (c == '\f') {
1179                         putchar('\\');
1180                         putchar('f');
1181                 } else if (c == '\n') {
1182                         putchar('\\');
1183                         putchar('n');
1184                 } else if (c == '\r') {
1185                         putchar('\\');
1186                         putchar('r');
1187                 } else if (c == '\t') {
1188                         putchar('\\');
1189                         putchar('t');
1190                 } else {
1191                         putchar('^');
1192                         putchar(c ^ 0x40);      /* DEL to ?, others to alpha */
1193                 }
1194         } else
1195                 putchar(c);
1196 }
1197
1198 /*
1199  * Routines/code to support the duplicate address suppression code, adapted
1200  * from replsbr.c
1201  */
1202
1203 static char *buf;               /* our current working buffer */
1204 static char *bufend;            /* end of working buffer */
1205 static char *last_dst;          /* buf ptr at end of last call */
1206 static unsigned int bufsiz=0;   /* current size of buf */
1207
1208 #define BUFINCR 512             /* how much to expand buf when if fills */
1209
1210 #define CPY(s) { cp = (s); while ((*dst++ = *cp++)) ; --dst; }
1211
1212 /*
1213  * check if there's enough room in buf for str.
1214  * add more mem if needed
1215  */
1216 #define CHECKMEM(str) \
1217             if ((len = strlen (str)) >= bufend - dst) {\
1218                 int i = dst - buf;\
1219                 int n = last_dst - buf;\
1220                 bufsiz += ((dst + len - bufend) / BUFINCR + 1) * BUFINCR;\
1221                 buf = mh_xrealloc (buf, bufsiz);\
1222                 dst = buf + i;\
1223                 last_dst = buf + n;\
1224                 bufend = buf + bufsiz;\
1225             }
1226
1227
1228 /*
1229  * These are versions of similar routines from replsbr.c; the purpose is
1230  * to suppress duplicate addresses from being added to a list when building
1231  * up addresses for the %(formataddr) format function.  This is used by
1232  * repl to prevent duplicate addresses from being added to the "to" line.
1233  * See replsbr.c for more information.
1234  *
1235  * We can't use the functions in replsbr.c directly because they are slightly
1236  * different and depend on the rest of replsbr.c
1237  */
1238 static char *
1239 test_formataddr (char *orig, char *str)
1240 {
1241     int len;
1242     char error[BUFSIZ];
1243     bool isgroup;
1244     char *dst;
1245     char *cp;
1246     char *sp;
1247     struct mailname *mp = NULL;
1248
1249     /* if we don't have a buffer yet, get one */
1250     if (bufsiz == 0) {
1251         buf = mh_xmalloc (BUFINCR);
1252         last_dst = buf;         /* XXX */
1253         bufsiz = BUFINCR - 6;  /* leave some slop */
1254         bufend = buf + bufsiz;
1255     }
1256     /*
1257      * If "orig" points to our buffer we can just pick up where we
1258      * left off.  Otherwise we have to copy orig into our buffer.
1259      */
1260     if (orig == buf)
1261         dst = last_dst;
1262     else if (!orig || !*orig) {
1263         dst = buf;
1264         *dst = '\0';
1265     } else {
1266         dst = last_dst;         /* XXX */
1267         CHECKMEM (orig);
1268         CPY (orig);
1269     }
1270
1271     /* concatenate all the new addresses onto 'buf' */
1272     for (isgroup = false; (cp = getname (str)); ) {
1273         if ((mp = getm (cp, NULL, 0, error, sizeof(error))) == NULL) {
1274             fprintf(stderr, "bad address \"%s\" -- %s\n", cp, error);
1275             continue;
1276         }
1277         if (isgroup && (mp->m_gname || !mp->m_ingrp)) {
1278             *dst++ = ';';
1279             isgroup = false;
1280         }
1281         if (insert (mp)) {
1282             /* if we get here we're going to add an address */
1283             if (dst != buf) {
1284                 *dst++ = ',';
1285                 *dst++ = ' ';
1286             }
1287             if (mp->m_gname) {
1288                 CHECKMEM (mp->m_gname);
1289                 CPY (mp->m_gname);
1290                 isgroup = true;
1291             }
1292             sp = adrformat (mp);
1293             CHECKMEM (sp);
1294             CPY (sp);
1295         }
1296     }
1297
1298     if (isgroup)
1299         *dst++ = ';';
1300
1301     *dst = '\0';
1302     last_dst = dst;
1303     return buf;
1304 }
1305
1306
1307 /*
1308  * The companion to test_formataddr(); it behaves the same way, except doesn't
1309  * do duplicate address detection.
1310  */
1311 static char *
1312 test_concataddr(char *orig, char *str)
1313 {
1314     char *cp;
1315
1316     nodupcheck = true;
1317     cp = test_formataddr(orig, str);
1318     nodupcheck = false;
1319     return cp;
1320 }
1321
1322 static int
1323 insert (struct mailname *np)
1324 {
1325     struct mailname *mp;
1326
1327     if (nodupcheck)
1328         return 1;
1329
1330     if (np->m_mbox == NULL)
1331         return 0;
1332
1333     for (mp = &mq; mp->m_next; mp = mp->m_next) {
1334         if (!strcasecmp (FENDNULL(np->m_host),
1335                          FENDNULL(mp->m_next->m_host)) &&
1336             !strcasecmp (FENDNULL(np->m_mbox),
1337                          FENDNULL(mp->m_next->m_mbox)))
1338             return 0;
1339     }
1340     if (!ccme && ismymbox (np))
1341         return 0;
1342
1343     mp->m_next = np;
1344
1345     return 1;
1346 }
1347
1348 /*
1349  * Reset our duplicate address list
1350  */
1351
1352 void
1353 mlistfree(void)
1354 {
1355     struct mailname *mp, *mp2;
1356
1357     for (mp = mq.m_next; mp; mp = mp2) {
1358         mp2 = mp->m_next;
1359         mnfree(mp);
1360     }
1361 }