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