git.marmaro.de Git - mmh/blob - uip/mhstoresbr.c

   1
   2 /*
   3  * mhstoresbr.c -- routines to save/store the contents of MIME messages
   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 <fcntl.h>
  12 #include <h/signals.h>
  13 #include <h/md5.h>
  14 #include <errno.h>
  15 #include <signal.h>
  16 #include <h/mts.h>
  17 #include <h/tws.h>
  18 #include <h/mime.h>
  19 #include <h/mhparse.h>
  20 #include <h/utils.h>
  21
  22
  23 /*
  24  * The list of top-level contents to display
  25  */
  26 extern CT *cts;
  27
  28 int autosw = 0;
  29
  30 /*
  31  * Cache of current directory.  This must be
  32  * set before these routines are called.
  33  */
  34 char *cwd;
  35
  36 /*
  37  * The directory in which to store the contents.
  38  */
  39 static char *dir;
  40
  41 /*
  42  * Type for a compare function for qsort.  This keeps
  43  * the compiler happy.
  44  */
  45 typedef int (*qsort_comp) (const void *, const void *);
  46
  47
  48 /* mhmisc.c */
  49 int part_ok (CT, int);
  50 int type_ok (CT, int);
  51 void flush_errors (void);
  52
  53 /* mhshowsbr.c */
  54 int show_content_aux (CT, int, int, char *, char *);
  55
  56 /*
  57  * prototypes
  58  */
  59 void store_all_messages (CT *);
  60
  61 /*
  62  * static prototypes
  63  */
  64 static void store_single_message (CT);
  65 static int store_switch (CT);
  66 static int store_generic (CT);
  67 static int store_application (CT);
  68 static int store_multi (CT);
  69 static int store_partial (CT);
  70 static int store_external (CT);
  71 static int ct_compar (CT *, CT *);
  72 static int store_content (CT, CT);
  73 static int output_content_file (CT, int);
  74 static int output_content_folder (char *, char *);
  75 static int parse_format_string (CT, char *, char *, int, char *);
  76 static void get_storeproc (CT);
  77 static int copy_some_headers (FILE *, CT);
  78 static char *clobber_check (char *);
  79
  80 /*
  81  * Main entry point to store content
  82  * from a collection of messages.
  83  */
  84
  85 void
  86 store_all_messages (CT *cts)
  87 {
  88     CT ct, *ctp;
  89     char *cp;
  90
  91     /*
  92      * Check for the directory in which to
  93      * store any contents.
  94      */
  95     if ((cp = context_find (nmhstorage)) && *cp)
  96         dir = getcpy (cp);
  97     else
  98         dir = getcpy (cwd);
  99
 100     for (ctp = cts; *ctp; ctp++) {
 101         ct = *ctp;
 102         store_single_message (ct);
 103     }
 104
 105     flush_errors ();
 106 }
 107
 108
 109 /*
 110  * Entry point to store the content
 111  * in a (single) message
 112  */
 113
 114 static void
 115 store_single_message (CT ct)
 116 {
 117     if (type_ok (ct, 1)) {
 118         umask (ct->c_umask);
 119         store_switch (ct);
 120         if (ct->c_fp) {
 121             fclose (ct->c_fp);
 122             ct->c_fp = NULL;
 123         }
 124         if (ct->c_ceclosefnx)
 125             (*ct->c_ceclosefnx) (ct);
 126     }
 127 }
 128
 129
 130 /*
 131  * Switching routine to store different content types
 132  */
 133
 134 static int
 135 store_switch (CT ct)
 136 {
 137     switch (ct->c_type) {
 138         case CT_MULTIPART:
 139             return store_multi (ct);
 140             break;
 141
 142         case CT_MESSAGE:
 143             switch (ct->c_subtype) {
 144                 case MESSAGE_PARTIAL:
 145                     return store_partial (ct);
 146                     break;
 147
 148                 case MESSAGE_EXTERNAL:
 149                     return store_external (ct);
 150
 151                 case MESSAGE_RFC822:
 152                 default:
 153                     return store_generic (ct);
 154                     break;
 155             }
 156             break;
 157
 158         case CT_APPLICATION:
 159             return store_application (ct);
 160             break;
 161
 162         case CT_TEXT:
 163         case CT_AUDIO:
 164         case CT_IMAGE:
 165         case CT_VIDEO:
 166             return store_generic (ct);
 167             break;
 168
 169         default:
 170             adios (NULL, "unknown content type %d", ct->c_type);
 171             break;
 172     }
 173
 174     return OK;  /* NOT REACHED */
 175 }
 176
 177
 178 /*
 179  * Generic routine to store a MIME content.
 180  * (audio, video, image, text, message/rfc922)
 181  */
 182
 183 static int
 184 store_generic (CT ct)
 185 {
 186     /*
 187      * Check if the content specifies a filename.
 188      * Don't bother with this for type "message"
 189      * (only "message/rfc822" will use store_generic).
 190      */
 191     if (autosw && ct->c_type != CT_MESSAGE)
 192         get_storeproc (ct);
 193
 194     return store_content (ct, NULL);
 195 }
 196
 197
 198 /*
 199  * Store content of type "application"
 200  */
 201
 202 static int
 203 store_application (CT ct)
 204 {
 205     char **ap, **ep;
 206     CI ci = &ct->c_ctinfo;
 207
 208     /* Check if the content specifies a filename */
 209     if (autosw)
 210         get_storeproc (ct);
 211
 212     /*
 213      * If storeproc is not defined, and the content is type
 214      * "application/octet-stream", we also check for various
 215      * attribute/value pairs which specify if this a tar file.
 216      */
 217     if (!ct->c_storeproc && ct->c_subtype == APPLICATION_OCTETS) {
 218         int tarP = 0, zP = 0, gzP = 0;
 219
 220         for (ap = ci->ci_attrs, ep = ci->ci_values; *ap; ap++, ep++) {
 221             /* check for "type=tar" attribute */
 222             if (!mh_strcasecmp (*ap, "type")) {
 223                 if (mh_strcasecmp (*ep, "tar"))
 224                     break;
 225
 226                 tarP = 1;
 227                 continue;
 228             }
 229
 230             /* check for "conversions=compress" attribute */
 231             if ((!mh_strcasecmp (*ap, "conversions") || !mh_strcasecmp (*ap, "x-conversions"))
 232                 && (!mh_strcasecmp (*ep, "compress") || !mh_strcasecmp (*ep, "x-compress"))) {
 233                 zP = 1;
 234                 continue;
 235             }
 236             /* check for "conversions=gzip" attribute */
 237             if ((!mh_strcasecmp (*ap, "conversions") || !mh_strcasecmp (*ap, "x-conversions"))
 238                 && (!mh_strcasecmp (*ep, "gzip") || !mh_strcasecmp (*ep, "x-gzip"))) {
 239                 gzP = 1;
 240                 continue;
 241             }
 242         }
 243
 244         if (tarP) {
 245             ct->c_showproc = add (zP ? "%euncompress | tar tvf -"
 246                                   : (gzP ? "%egzip -dc | tar tvf -"
 247                                      : "%etar tvf -"), NULL);
 248             if (!ct->c_storeproc) {
 249                 if (autosw) {
 250                     ct->c_storeproc = add (zP ? "| uncompress | tar xvpf -"
 251                                            : (gzP ? "| gzip -dc | tar xvpf -"
 252                                               : "| tar xvpf -"), NULL);
 253                     ct->c_umask = 0022;
 254                 } else {
 255                     ct->c_storeproc= add (zP ? "%m%P.tar.Z"
 256                                           : (gzP ? "%m%P.tar.gz"
 257                                              : "%m%P.tar"), NULL);
 258                 }
 259             }
 260         }
 261     }
 262
 263     return store_content (ct, NULL);
 264 }
 265
 266
 267 /*
 268  * Store the content of a multipart message
 269  */
 270
 271 static int
 272 store_multi (CT ct)
 273 {
 274     int result;
 275     struct multipart *m = (struct multipart *) ct->c_ctparams;
 276     struct part *part;
 277
 278     result = NOTOK;
 279     for (part = m->mp_parts; part; part = part->mp_next) {
 280         CT  p = part->mp_part;
 281
 282         if (part_ok (p, 1) && type_ok (p, 1)) {
 283             result = store_switch (p);
 284             if (result == OK && ct->c_subtype == MULTI_ALTERNATE)
 285                 break;
 286         }
 287     }
 288
 289     return result;
 290 }
 291
 292
 293 /*
 294  * Reassemble and store the contents of a collection
 295  * of messages of type "message/partial".
 296  */
 297
 298 static int
 299 store_partial (CT ct)
 300 {
 301     int cur, hi, i;
 302     CT p, *ctp, *ctq;
 303     CT *base;
 304     struct partial *pm, *qm;
 305
 306     qm = (struct partial *) ct->c_ctparams;
 307     if (qm->pm_stored)
 308         return OK;
 309
 310     hi = i = 0;
 311     for (ctp = cts; *ctp; ctp++) {
 312         p = *ctp;
 313         if (p->c_type == CT_MESSAGE && p->c_subtype == ct->c_subtype) {
 314             pm = (struct partial *) p->c_ctparams;
 315             if (!pm->pm_stored
 316                     && strcmp (qm->pm_partid, pm->pm_partid) == 0) {
 317                 pm->pm_marked = pm->pm_partno;
 318                 if (pm->pm_maxno)
 319                     hi = pm->pm_maxno;
 320                 pm->pm_stored = 1;
 321                 i++;
 322             }
 323             else
 324                 pm->pm_marked = 0;
 325         }
 326     }
 327
 328     if (hi == 0) {
 329         advise (NULL, "missing (at least) last part of multipart message");
 330         return NOTOK;
 331     }
 332
 333     if ((base = (CT *) calloc ((size_t) (i + 1), sizeof(*base))) == NULL)
 334         adios (NULL, "out of memory");
 335
 336     ctq = base;
 337     for (ctp = cts; *ctp; ctp++) {
 338         p = *ctp;
 339         if (p->c_type == CT_MESSAGE && p->c_subtype == ct->c_subtype) {
 340             pm = (struct partial *) p->c_ctparams;
 341             if (pm->pm_marked)
 342                 *ctq++ = p;
 343         }
 344     }
 345     *ctq = NULL;
 346
 347     if (i > 1)
 348         qsort ((char *) base, i, sizeof(*base), (qsort_comp) ct_compar);
 349
 350     cur = 1;
 351     for (ctq = base; *ctq; ctq++) {
 352         p = *ctq;
 353         pm = (struct partial *) p->c_ctparams;
 354         if (pm->pm_marked != cur) {
 355             if (pm->pm_marked == cur - 1) {
 356                 admonish (NULL,
 357                           "duplicate part %d of %d part multipart message",
 358                           pm->pm_marked, hi);
 359                 continue;
 360             }
 361
 362 missing_part:
 363             advise (NULL,
 364                     "missing %spart %d of %d part multipart message",
 365                     cur != hi ? "(at least) " : "", cur, hi);
 366             goto losing;
 367         }
 368         else
 369             cur++;
 370     }
 371     if (hi != --cur) {
 372         cur = hi;
 373         goto missing_part;
 374     }
 375
 376     /*
 377      * Now cycle through the sorted list of messages of type
 378      * "message/partial" and save/append them to a file.
 379      */
 380
 381     ctq = base;
 382     ct = *ctq++;
 383     if (store_content (ct, NULL) == NOTOK) {
 384 losing:
 385         free ((char *) base);
 386         return NOTOK;
 387     }
 388
 389     for (; *ctq; ctq++) {
 390         p = *ctq;
 391         if (store_content (p, ct) == NOTOK)
 392             goto losing;
 393     }
 394
 395     free ((char *) base);
 396     return OK;
 397 }
 398
 399
 400 /*
 401  * Store content from a message of type "message/external".
 402  */
 403
 404 static int
 405 store_external (CT ct)
 406 {
 407     int result = NOTOK;
 408     struct exbody *e = (struct exbody *) ct->c_ctparams;
 409     CT p = e->eb_content;
 410
 411     if (!type_ok (p, 1))
 412         return OK;
 413
 414     /*
 415      * Check if the parameters for the external body
 416      * specified a filename.
 417      */
 418     if (autosw) {
 419         char *cp;
 420
 421         if ((cp = e->eb_name)
 422             && *cp != '/'
 423             && *cp != '.'
 424             && *cp != '|'
 425             && *cp != '!'
 426             && !strchr (cp, '%')) {
 427             if (!ct->c_storeproc)
 428                 ct->c_storeproc = add (cp, NULL);
 429             if (!p->c_storeproc)
 430                 p->c_storeproc = add (cp, NULL);
 431         }
 432     }
 433
 434     /*
 435      * Since we will let the Content structure for the
 436      * external body substitute for the current content,
 437      * we temporarily change its partno (number inside
 438      * multipart), so everything looks right.
 439      */
 440     p->c_partno = ct->c_partno;
 441
 442     /* we probably need to check if content is really there */
 443     result = store_switch (p);
 444
 445     p->c_partno = NULL;
 446     return result;
 447 }
 448
 449
 450 /*
 451  * Compare the numbering from two different
 452  * message/partials (needed for sorting).
 453  */
 454
 455 static int
 456 ct_compar (CT *a, CT *b)
 457 {
 458     struct partial *am = (struct partial *) ((*a)->c_ctparams);
 459     struct partial *bm = (struct partial *) ((*b)->c_ctparams);
 460
 461     return (am->pm_marked - bm->pm_marked);
 462 }
 463
 464
 465 /*
 466  * Store contents of a message or message part to
 467  * a folder, a file, the standard output, or pass
 468  * the contents to a command.
 469  *
 470  * If the current content to be saved is a followup part
 471  * to a collection of messages of type "message/partial",
 472  * then field "p" is a pointer to the Content structure
 473  * to the first message/partial in the group.
 474  */
 475
 476 static int
 477 store_content (CT ct, CT p)
 478 {
 479     int appending = 0, msgnum = 0;
 480     int is_partial = 0, first_partial = 0;
 481     int last_partial = 0;
 482     char *cp, buffer[BUFSIZ];
 483
 484     /*
 485      * Do special processing for messages of
 486      * type "message/partial".
 487      *
 488      * We first check if this content is of type
 489      * "message/partial".  If it is, then we need to check
 490      * whether it is the first and/or last in the group.
 491      *
 492      * Then if "p" is a valid pointer, it points to the Content
 493      * structure of the first partial in the group.  So we copy
 494      * the file name and/or folder name from that message.  In
 495      * this case, we also note that we will be appending.
 496      */
 497     if (ct->c_type == CT_MESSAGE && ct->c_subtype == MESSAGE_PARTIAL) {
 498         struct partial *pm = (struct partial *) ct->c_ctparams;
 499
 500         /* Yep, it's a message/partial */
 501         is_partial = 1;
 502
 503         /* But is it the first and/or last in the collection? */
 504         if (pm->pm_partno == 1)
 505             first_partial = 1;
 506         if (pm->pm_maxno && pm->pm_partno == pm->pm_maxno)
 507             last_partial = 1;
 508
 509         /*
 510          * If "p" is a valid pointer, then it points to the
 511          * Content structure for the first message in the group.
 512          * So we just copy the filename or foldername information
 513          * from the previous iteration of this function.
 514          */
 515         if (p) {
 516             appending = 1;
 517             ct->c_storage = add (p->c_storage, NULL);
 518
 519             /* record the folder name */
 520             if (p->c_folder) {
 521                 ct->c_folder = add (p->c_folder, NULL);
 522             }
 523             goto got_filename;
 524         }
 525     }
 526
 527     /*
 528      * Get storage formatting string.
 529      *
 530      * 1) If we have storeproc defined, then use that
 531      * 2) Else check for a mhn-store-<type>/<subtype> entry
 532      * 3) Else check for a mhn-store-<type> entry
 533      * 4) Else if content is "message", use "+" (current folder)
 534      * 5) Else use string "%m%P.%s".
 535      */
 536     if ((cp = ct->c_storeproc) == NULL || *cp == '\0') {
 537         CI ci = &ct->c_ctinfo;
 538
 539         snprintf (buffer, sizeof(buffer), "%s-store-%s/%s",
 540                 invo_name, ci->ci_type, ci->ci_subtype);
 541         if ((cp = context_find (buffer)) == NULL || *cp == '\0') {
 542             snprintf (buffer, sizeof(buffer), "%s-store-%s", invo_name, ci->ci_type);
 543             if ((cp = context_find (buffer)) == NULL || *cp == '\0') {
 544                 cp = ct->c_type == CT_MESSAGE ? "+" : "%m%P.%s";
 545             }
 546         }
 547     }
 548
 549     /*
 550      * Check the beginning of storage formatting string
 551      * to see if we are saving content to a folder.
 552      */
 553     if (*cp == '+' || *cp == '@') {
 554         char *tmpfilenam, *folder;
 555
 556         /* Store content in temporary file for now */
 557         tmpfilenam = m_mktemp(invo_name, NULL, NULL);
 558         ct->c_storage = add (tmpfilenam, NULL);
 559
 560         /* Get the folder name */
 561         if (cp[1])
 562             folder = pluspath (cp);
 563         else
 564             folder = getfolder (1);
 565
 566         /* Check if folder exists */
 567         create_folder(m_mailpath(folder), 0, exit);
 568
 569         /* Record the folder name */
 570         ct->c_folder = add (folder, NULL);
 571
 572         if (cp[1])
 573             free (folder);
 574
 575         goto got_filename;
 576     }
 577
 578     /*
 579      * Parse and expand the storage formatting string
 580      * in `cp' into `buffer'.
 581      */
 582     parse_format_string (ct, cp, buffer, sizeof(buffer), dir);
 583
 584     /*
 585      * If formatting begins with '|' or '!', then pass
 586      * content to standard input of a command and return.
 587      */
 588     if (buffer[0] == '|' || buffer[0] == '!')
 589         return show_content_aux (ct, 1, 0, buffer + 1, dir);
 590
 591     /* record the filename */
 592     if ((ct->c_storage = clobber_check (add (buffer, NULL))) == NULL) {
 593       return NOTOK;
 594     }
 595
 596 got_filename:
 597     /* flush the output stream */
 598     fflush (stdout);
 599
 600     /* Now save or append the content to a file */
 601     if (output_content_file (ct, appending) == NOTOK)
 602         return NOTOK;
 603
 604     /*
 605      * If necessary, link the file into a folder and remove
 606      * the temporary file.  If this message is a partial,
 607      * then only do this if it is the last one in the group.
 608      */
 609     if (ct->c_folder && (!is_partial || last_partial)) {
 610         msgnum = output_content_folder (ct->c_folder, ct->c_storage);
 611         unlink (ct->c_storage);
 612         if (msgnum == NOTOK)
 613             return NOTOK;
 614     }
 615
 616     /*
 617      * Now print out the name/number of the message
 618      * that we are storing.
 619      */
 620     if (is_partial) {
 621         if (first_partial)
 622             fprintf (stderr, "reassembling partials ");
 623         if (last_partial)
 624             fprintf (stderr, "%s", ct->c_file);
 625         else
 626             fprintf (stderr, "%s,", ct->c_file);
 627     } else {
 628         fprintf (stderr, "storing message %s", ct->c_file);
 629         if (ct->c_partno)
 630             fprintf (stderr, " part %s", ct->c_partno);
 631     }
 632
 633     /*
 634      * Unless we are in the "middle" of group of message/partials,
 635      * we now print the name of the file, folder, and/or message
 636      * to which we are storing the content.
 637      */
 638     if (!is_partial || last_partial) {
 639         if (ct->c_folder) {
 640             fprintf (stderr, " to folder %s as message %d\n", ct->c_folder, msgnum);
 641         } else if (!strcmp(ct->c_storage, "-")) {
 642             fprintf (stderr, " to stdout\n");
 643         } else {
 644             int cwdlen;
 645
 646             cwdlen = strlen (cwd);
 647             fprintf (stderr, " as file %s\n",
 648                 strncmp (ct->c_storage, cwd, cwdlen)
 649                 || ct->c_storage[cwdlen] != '/'
 650                 ? ct->c_storage : ct->c_storage + cwdlen + 1);
 651         }
 652     }
 653
 654     return OK;
 655 }
 656
 657
 658 /*
 659  * Output content to a file
 660  */
 661
 662 static int
 663 output_content_file (CT ct, int appending)
 664 {
 665     int filterstate;
 666     char *file, buffer[BUFSIZ];
 667     long pos, last;
 668     FILE *fp;
 669
 670     /*
 671      * If the pathname is absolute, make sure
 672      * all the relevant directories exist.
 673      */
 674     if (strchr(ct->c_storage, '/')
 675             && make_intermediates (ct->c_storage) == NOTOK)
 676         return NOTOK;
 677
 678     if (ct->c_encoding != CE_7BIT) {
 679         int cc, fd;
 680
 681         if (!ct->c_ceopenfnx) {
 682             advise (NULL, "don't know how to decode part %s of message %s",
 683                     ct->c_partno, ct->c_file);
 684             return NOTOK;
 685         }
 686
 687         file = appending || !strcmp (ct->c_storage, "-") ? NULL
 688                                                            : ct->c_storage;
 689         if ((fd = (*ct->c_ceopenfnx) (ct, &file)) == NOTOK)
 690             return NOTOK;
 691         if (!strcmp (file, ct->c_storage)) {
 692             (*ct->c_ceclosefnx) (ct);
 693             return OK;
 694         }
 695
 696         /*
 697          * Send to standard output
 698          */
 699         if (!strcmp (ct->c_storage, "-")) {
 700             int gd;
 701
 702             if ((gd = dup (fileno (stdout))) == NOTOK) {
 703                 advise ("stdout", "unable to dup");
 704 losing:
 705                 (*ct->c_ceclosefnx) (ct);
 706                 return NOTOK;
 707             }
 708             if ((fp = fdopen (gd, appending ? "a" : "w")) == NULL) {
 709                 advise ("stdout", "unable to fdopen (%d, \"%s\") from", gd,
 710                         appending ? "a" : "w");
 711                 close (gd);
 712                 goto losing;
 713             }
 714         } else {
 715             /*
 716              * Open output file
 717              */
 718             if ((fp = fopen (ct->c_storage, appending ? "a" : "w")) == NULL) {
 719                 advise (ct->c_storage, "unable to fopen for %s",
 720                         appending ? "appending" : "writing");
 721                 goto losing;
 722             }
 723         }
 724
 725         /*
 726          * Filter the header fields of the initial enclosing
 727          * message/partial into the file.
 728          */
 729         if (ct->c_type == CT_MESSAGE && ct->c_subtype == MESSAGE_PARTIAL) {
 730             struct partial *pm = (struct partial *) ct->c_ctparams;
 731
 732             if (pm->pm_partno == 1)
 733                 copy_some_headers (fp, ct);
 734         }
 735
 736         for (;;) {
 737             switch (cc = read (fd, buffer, sizeof(buffer))) {
 738                 case NOTOK:
 739                     advise (file, "error reading content from");
 740                     break;
 741
 742                 case OK:
 743                     break;
 744
 745                 default:
 746                     fwrite (buffer, sizeof(*buffer), cc, fp);
 747                     continue;
 748             }
 749             break;
 750         }
 751
 752         (*ct->c_ceclosefnx) (ct);
 753
 754         if (cc != NOTOK && fflush (fp))
 755             advise (ct->c_storage, "error writing to");
 756
 757         fclose (fp);
 758
 759         return (cc != NOTOK ? OK : NOTOK);
 760     }
 761
 762     if (!ct->c_fp && (ct->c_fp = fopen (ct->c_file, "r")) == NULL) {
 763         advise (ct->c_file, "unable to open for reading");
 764         return NOTOK;
 765     }
 766
 767     pos = ct->c_begin;
 768     last = ct->c_end;
 769     fseek (ct->c_fp, pos, SEEK_SET);
 770
 771     if (!strcmp (ct->c_storage, "-")) {
 772         int gd;
 773
 774         if ((gd = dup (fileno (stdout))) == NOTOK) {
 775             advise ("stdout", "unable to dup");
 776             return NOTOK;
 777         }
 778         if ((fp = fdopen (gd, appending ? "a" : "w")) == NULL) {
 779             advise ("stdout", "unable to fdopen (%d, \"%s\") from", gd,
 780                     appending ? "a" : "w");
 781             close (gd);
 782             return NOTOK;
 783         }
 784     } else {
 785         if ((fp = fopen (ct->c_storage, appending ? "a" : "w")) == NULL) {
 786             advise (ct->c_storage, "unable to fopen for %s",
 787                     appending ? "appending" : "writing");
 788             return NOTOK;
 789         }
 790     }
 791
 792     /*
 793      * Copy a few of the header fields of the initial
 794      * enclosing message/partial into the file.
 795      */
 796     filterstate = 0;
 797     if (ct->c_type == CT_MESSAGE && ct->c_subtype == MESSAGE_PARTIAL) {
 798         struct partial *pm = (struct partial *) ct->c_ctparams;
 799
 800         if (pm->pm_partno == 1) {
 801             copy_some_headers (fp, ct);
 802             filterstate = 1;
 803         }
 804     }
 805
 806     while (fgets (buffer, sizeof(buffer) - 1, ct->c_fp)) {
 807         if ((pos += strlen (buffer)) > last) {
 808             int diff;
 809
 810             diff = strlen (buffer) - (pos - last);
 811             if (diff >= 0)
 812                 buffer[diff] = '\0';
 813         }
 814         /*
 815          * If this is the first content of a group of
 816          * message/partial contents, then we only copy a few
 817          * of the header fields of the enclosed message.
 818          */
 819         if (filterstate) {
 820             switch (buffer[0]) {
 821                 case ' ':
 822                 case '\t':
 823                     if (filterstate < 0)
 824                         buffer[0] = 0;
 825                     break;
 826
 827                 case '\n':
 828                     filterstate = 0;
 829                     break;
 830
 831                 default:
 832                     if (!uprf (buffer, XXX_FIELD_PRF)
 833                             && !uprf (buffer, VRSN_FIELD)
 834                             && !uprf (buffer, "Subject:")
 835                             && !uprf (buffer, "Encrypted:")
 836                             && !uprf (buffer, "Message-ID:")) {
 837                         filterstate = -1;
 838                         buffer[0] = 0;
 839                         break;
 840                     }
 841                     filterstate = 1;
 842                     break;
 843             }
 844         }
 845         fputs (buffer, fp);
 846         if (pos >= last)
 847             break;
 848     }
 849
 850     if (fflush (fp))
 851         advise (ct->c_storage, "error writing to");
 852
 853     fclose (fp);
 854     fclose (ct->c_fp);
 855     ct->c_fp = NULL;
 856     return OK;
 857 }
 858
 859
 860 /*
 861  * Add a file to a folder.
 862  *
 863  * Return the new message number of the file
 864  * when added to the folder.  Return -1, if
 865  * there is an error.
 866  */
 867
 868 static int
 869 output_content_folder (char *folder, char *filename)
 870 {
 871     int msgnum;
 872     struct msgs *mp;
 873
 874     /* Read the folder. */
 875     if ((mp = folder_read (folder))) {
 876         /* Link file into folder */
 877         msgnum = folder_addmsg (&mp, filename, 0, 0, 0, 0, (char *)0);
 878     } else {
 879         advise (NULL, "unable to read folder %s", folder);
 880         return NOTOK;
 881     }
 882
 883     /* free folder structure */
 884     folder_free (mp);
 885
 886     /*
 887      * Return msgnum.  We are relying on the fact that
 888      * msgnum will be -1, if folder_addmsg() had an error.
 889      */
 890     return msgnum;
 891 }
 892
 893
 894 /*
 895  * Parse and expand the storage formatting string
 896  * pointed to by "cp" into "buffer".
 897  */
 898
 899 static int
 900 parse_format_string (CT ct, char *cp, char *buffer, int buflen, char *dir)
 901 {
 902     int len;
 903     char *bp;
 904     CI ci = &ct->c_ctinfo;
 905
 906     /*
 907      * If storage string is "-", just copy it, and
 908      * return (send content to standard output).
 909      */
 910     if (cp[0] == '-' && cp[1] == '\0') {
 911         strncpy (buffer, cp, buflen);
 912         return 0;
 913     }
 914
 915     bp = buffer;
 916     bp[0] = '\0';
 917
 918     /*
 919      * If formatting string is a pathname that doesn't
 920      * begin with '/', then preface the path with the
 921      * appropriate directory.
 922      */
 923     if (*cp != '/' && *cp != '|' && *cp != '!') {
 924         snprintf (bp, buflen, "%s/", dir[1] ? dir : "");
 925         len = strlen (bp);
 926         bp += len;
 927         buflen -= len;
 928     }
 929
 930     for (; *cp; cp++) {
 931
 932         /* We are processing a storage escape */
 933         if (*cp == '%') {
 934             switch (*++cp) {
 935                 case 'a':
 936                     /*
 937                      * Insert parameters from Content-Type.
 938                      * This is only valid for '|' commands.
 939                      */
 940                     if (buffer[0] != '|' && buffer[0] != '!') {
 941                         *bp++ = *--cp;
 942                         *bp = '\0';
 943                         buflen--;
 944                         continue;
 945                     } else {
 946                         char **ap, **ep;
 947                         char *s = "";
 948
 949                         for (ap = ci->ci_attrs, ep = ci->ci_values;
 950                                  *ap; ap++, ep++) {
 951                             snprintf (bp, buflen, "%s%s=\"%s\"", s, *ap, *ep);
 952                             len = strlen (bp);
 953                             bp += len;
 954                             buflen -= len;
 955                             s = " ";
 956                         }
 957                     }
 958                     break;
 959
 960                 case 'm':
 961                     /* insert message number */
 962                     snprintf (bp, buflen, "%s", r1bindex (ct->c_file, '/'));
 963                     break;
 964
 965                 case 'P':
 966                     /* insert part number with leading dot */
 967                     if (ct->c_partno)
 968                         snprintf (bp, buflen, ".%s", ct->c_partno);
 969                     break;
 970
 971                 case 'p':
 972                     /* insert part number withouth leading dot */
 973                     if (ct->c_partno)
 974                         strncpy (bp, ct->c_partno, buflen);
 975                     break;
 976
 977                 case 't':
 978                     /* insert content type */
 979                     strncpy (bp, ci->ci_type, buflen);
 980                     break;
 981
 982                 case 's':
 983                     /* insert content subtype */
 984                     strncpy (bp, ci->ci_subtype, buflen);
 985                     break;
 986
 987                 case '%':
 988                     /* insert the character % */
 989                     goto raw;
 990
 991                 default:
 992                     *bp++ = *--cp;
 993                     *bp = '\0';
 994                     buflen--;
 995                     continue;
 996             }
 997
 998             /* Advance bp and decrement buflen */
 999             len = strlen (bp);
1000             bp += len;
1001             buflen -= len;
1002
1003         } else {
1004 raw:
1005             *bp++ = *cp;
1006             *bp = '\0';
1007             buflen--;
1008         }
1009     }
1010
1011     return 0;
1012 }
1013
1014
1015 /*
1016  * Check if the content specifies a filename
1017  * in its MIME parameters.
1018  */
1019
1020 static void
1021 get_storeproc (CT ct)
1022 {
1023     char **ap, **ep, *cp;
1024     CI ci = &ct->c_ctinfo;
1025
1026     /*
1027      * If the storeproc has already been defined,
1028      * we just return (for instance, if this content
1029      * is part of a "message/external".
1030      */
1031     if (ct->c_storeproc)
1032         return;
1033
1034     /*
1035      * Check the attribute/value pairs, for the attribute "name".
1036      * If found, do a few sanity checks and copy the value into
1037      * the storeproc.
1038      */
1039     for (ap = ci->ci_attrs, ep = ci->ci_values; *ap; ap++, ep++) {
1040         if (!mh_strcasecmp (*ap, "name")
1041             && *(cp = *ep) != '/'
1042             && *cp != '.'
1043             && *cp != '|'
1044             && *cp != '!'
1045             && !strchr (cp, '%')) {
1046             ct->c_storeproc = add (cp, NULL);
1047             return;
1048         }
1049     }
1050 }
1051
1052
1053 /*
1054  * Copy some of the header fields of the initial message/partial
1055  * message into the header of the reassembled message.
1056  */
1057
1058 static int
1059 copy_some_headers (FILE *out, CT ct)
1060 {
1061     HF hp;
1062
1063     hp = ct->c_first_hf;        /* start at first header field */
1064
1065     while (hp) {
1066         /*
1067          * A few of the header fields of the enclosing
1068          * messages are not copied.
1069          */
1070         if (!uprf (hp->name, XXX_FIELD_PRF)
1071                 && mh_strcasecmp (hp->name, VRSN_FIELD)
1072                 && mh_strcasecmp (hp->name, "Subject")
1073                 && mh_strcasecmp (hp->name, "Encrypted")
1074                 && mh_strcasecmp (hp->name, "Message-ID"))
1075             fprintf (out, "%s:%s", hp->name, hp->value);
1076         hp = hp->next;  /* next header field */
1077     }
1078
1079     return OK;
1080 }
1081
1082 /******************************************************************************/
1083 /* -clobber support */
1084
1085 enum clobber_policy_t {
1086   NMH_CLOBBER_ALWAYS,
1087   NMH_CLOBBER_AUTO,
1088   NMH_CLOBBER_SUFFIX,
1089   NMH_CLOBBER_ASK,
1090   NMH_CLOBBER_NEVER
1091 };
1092
1093 static enum clobber_policy_t clobber_policy = NMH_CLOBBER_ALWAYS;
1094
1095 int files_not_clobbered = 0;
1096
1097 int
1098 save_clobber_policy (const char *value) {
1099   if (! mh_strcasecmp (value, "always")) {
1100     clobber_policy = NMH_CLOBBER_ALWAYS;
1101   } else if (! mh_strcasecmp (value, "auto")) {
1102     clobber_policy = NMH_CLOBBER_AUTO;
1103   } else if (! mh_strcasecmp (value, "suffix")) {
1104     clobber_policy = NMH_CLOBBER_SUFFIX;
1105   } else if (! mh_strcasecmp (value, "ask")) {
1106     clobber_policy = NMH_CLOBBER_ASK;
1107   } else if (! mh_strcasecmp (value, "never")) {
1108     clobber_policy = NMH_CLOBBER_NEVER;
1109   } else {
1110     return 1;
1111   }
1112
1113   return 0;
1114 }
1115
1116
1117 static char *
1118 next_version (char *file, enum clobber_policy_t clobber_policy) {
1119   const size_t max_versions = 1000000;
1120   /* 8 = log max_versions  +  one for - or .  +  one for null terminator */
1121   const size_t buflen = strlen (file) + 8;
1122   char *buffer = mh_xmalloc (buflen);
1123   size_t version;
1124
1125   char *extension = NULL;
1126   if (clobber_policy == NMH_CLOBBER_AUTO  &&
1127       ((extension = strrchr (file, '.')) != NULL)) {
1128     *extension++ = '\0';
1129   }
1130
1131   for (version = 1; version < max_versions; ++version) {
1132     struct stat st;
1133
1134     switch (clobber_policy) {
1135       case NMH_CLOBBER_AUTO: {
1136         snprintf (buffer, buflen, "%s-%ld%s%s", file, (long) version,
1137                   extension == NULL  ?  ""  :  ".",
1138                   extension == NULL  ?  ""  :  extension);
1139         break;
1140       }
1141
1142       case NMH_CLOBBER_SUFFIX:
1143         snprintf (buffer, buflen, "%s.%ld", file, (long) version);
1144         break;
1145
1146       default:
1147         /* Should never get here. */
1148         advise (NULL, "will not overwrite %s, invalid clobber policy", buffer);
1149         free (buffer);
1150         ++files_not_clobbered;
1151         return NULL;
1152     }
1153
1154     if (stat (buffer, &st) == NOTOK) {
1155       break;
1156     }
1157   }
1158
1159   free (file);
1160
1161   if (version >= max_versions) {
1162     advise (NULL, "will not overwrite %s, too many versions", buffer);
1163     free (buffer);
1164     buffer = NULL;
1165     ++files_not_clobbered;
1166   }
1167
1168   return buffer;
1169 }
1170
1171
1172 static char *
1173 clobber_check (char *original_file) {
1174   /* clobber policy        return value
1175    * --------------        ------------
1176    *   -always                 file
1177    *   -auto           file-<digits>.extension
1178    *   -suffix             file.<digits>
1179    *   -ask          file, 0, or another filename/path
1180    *   -never                   0
1181    */
1182
1183   char *file;
1184   char *cwd = NULL;
1185   int check_again;
1186
1187   if (clobber_policy == NMH_CLOBBER_ASK) {
1188     /* Save cwd for possible use in loop below. */
1189     char *slash;
1190
1191     cwd = add (original_file, NULL);
1192     slash = strrchr (cwd, '/');
1193
1194     if (slash) {
1195       *slash = '\0';
1196     } else {
1197       /* original_file wasn't a full path, which shouldn't happen. */
1198       cwd = NULL;
1199     }
1200   }
1201
1202   do {
1203     struct stat st;
1204
1205     file = original_file;
1206     check_again = 0;
1207
1208     switch (clobber_policy) {
1209       case NMH_CLOBBER_ALWAYS:
1210         break;
1211
1212       case NMH_CLOBBER_SUFFIX:
1213       case NMH_CLOBBER_AUTO:
1214         if (stat (file, &st) == OK) {
1215           file = next_version (original_file, clobber_policy);
1216         }
1217         break;
1218
1219       case NMH_CLOBBER_ASK:
1220         if (stat (file, &st) == OK) {
1221           enum answers { NMH_YES, NMH_NO, NMH_RENAME };
1222           static struct swit answer[4] = {
1223             { "yes", 0 }, { "no", 0 }, { "rename", 0 }, { NULL, 0 } };
1224           char **ans;
1225
1226           if (isatty (fileno (stdin))) {
1227             char *prompt =
1228               concat ("Overwrite \"", file, "\" [y/n/rename]? ", NULL);
1229             ans = getans (prompt, answer);
1230             free (prompt);
1231           } else {
1232             /* Overwrite, that's what nmh used to do.  And warn. */
1233             advise (NULL, "-clobber ask but no tty, so overwrite %s", file);
1234             break;
1235           }
1236
1237           switch ((enum answers) smatch (*ans, answer)) {
1238             case NMH_YES:
1239               break;
1240             case NMH_NO:
1241               free (file);
1242               file = NULL;
1243               ++files_not_clobbered;
1244               break;
1245             case NMH_RENAME: {
1246               char buf[PATH_MAX];
1247               printf ("Enter filename or full path of the new file: ");
1248               if (fgets (buf, sizeof buf, stdin) == NULL  ||
1249                   buf[0] == '\0') {
1250                 file = NULL;
1251                 ++files_not_clobbered;
1252               } else {
1253                 char *newline = strchr (buf, '\n');
1254                 if (newline) {
1255                   *newline = '\0';
1256                 }
1257               }
1258
1259               free (file);
1260
1261               if (buf[0] == '/') {
1262                 /* Full path, use it. */
1263                 file = add (buf, NULL);
1264               } else {
1265                 /* Relative path. */
1266                 file = cwd  ?  concat (cwd, "/", buf, NULL)  :  add (buf, NULL);
1267               }
1268
1269               check_again = 1;
1270               break;
1271             }
1272           }
1273         }
1274         break;
1275
1276       case NMH_CLOBBER_NEVER:
1277         if (stat (file, &st) == OK) {
1278           /* Keep count of files that would have been clobbered,
1279              and return that as process exit status. */
1280           advise (NULL, "will not overwrite %s with -clobber never", file);
1281           free (file);
1282           file = NULL;
1283           ++files_not_clobbered;
1284         }
1285         break;
1286     }
1287
1288     original_file = file;
1289   } while (check_again);
1290
1291   if (cwd) {
1292     free (cwd);
1293   }
1294
1295   return file;
1296 }
1297
1298 /* -clobber support */
1299 /******************************************************************************/