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