attach.c

   1 /*
   2  * Copyright (C) 1996-2000 Michael R. Elkins <me@mutt.org>
   3  * Copyright (C) 1999-2000 Thomas Roessler <roessler@does-not-exist.org>
   4  *
   5  *     This program is free software; you can redistribute it and/or modify
   6  *     it under the terms of the GNU General Public License as published by
   7  *     the Free Software Foundation; either version 2 of the License, or
   8  *     (at your option) any later version.
   9  *
  10  *     This program is distributed in the hope that it will be useful,
  11  *     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *     GNU General Public License for more details.
  14  *
  15  *     You should have received a copy of the GNU General Public License
  16  *     along with this program; if not, write to the Free Software
  17  *     Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  18  */
  19
  20 #if HAVE_CONFIG_H
  21 # include "config.h"
  22 #endif
  23
  24 #include "mutt.h"
  25 #include "mutt_menu.h"
  26 #include "mutt_curses.h"
  27 #include "keymap.h"
  28 #include "rfc1524.h"
  29 #include "mime.h"
  30 #include "pager.h"
  31 #include "mailbox.h"
  32 #include "copy.h"
  33 #include "mx.h"
  34 #include "mutt_crypt.h"
  35
  36 #include <ctype.h>
  37 #include <stdlib.h>
  38 #include <unistd.h>
  39 #include <sys/wait.h>
  40 #include <sys/stat.h>
  41 #include <fcntl.h>
  42 #include <string.h>
  43 #include <errno.h>
  44
  45 int mutt_get_tmp_attachment (BODY * a)
  46 {
  47   char type[STRING];
  48   char tempfile[_POSIX_PATH_MAX];
  49   rfc1524_entry *entry = rfc1524_new_entry ();
  50   FILE *fpin = NULL, *fpout = NULL;
  51   struct stat st;
  52
  53   if (a->unlink)
  54     return 0;
  55
  56   snprintf (type, sizeof (type), "%s/%s", TYPE (a), a->subtype);
  57   rfc1524_mailcap_lookup (a, type, entry, 0);
  58   rfc1524_expand_filename (entry->nametemplate, a->filename,
  59                            tempfile, sizeof (tempfile));
  60
  61   rfc1524_free_entry (&entry);
  62
  63   if (stat (a->filename, &st) == -1)
  64     return -1;
  65
  66   if ((fpin = fopen (a->filename, "r")) && (fpout = safe_fopen (tempfile, "w"))) {      /* __FOPEN_CHECKED__ */
  67     mutt_copy_stream (fpin, fpout);
  68     mutt_str_replace (&a->filename, tempfile);
  69     a->unlink = 1;
  70
  71     if (a->stamp >= st.st_mtime)
  72       mutt_stamp_attachment (a);
  73   }
  74   else
  75     mutt_perror (fpin ? tempfile : a->filename);
  76
  77   if (fpin)
  78     fclose (fpin);
  79   if (fpout)
  80     fclose (fpout);
  81
  82   return a->unlink ? 0 : -1;
  83 }
  84
  85
  86 /* return 1 if require full screen redraw, 0 otherwise */
  87 int mutt_compose_attachment (BODY * a)
  88 {
  89   char type[STRING];
  90   char command[STRING];
  91   char newfile[_POSIX_PATH_MAX] = "";
  92   rfc1524_entry *entry = rfc1524_new_entry ();
  93   short unlink_newfile = 0;
  94   int rc = 0;
  95
  96   snprintf (type, sizeof (type), "%s/%s", TYPE (a), a->subtype);
  97   if (rfc1524_mailcap_lookup (a, type, entry, M_COMPOSE)) {
  98     if (entry->composecommand || entry->composetypecommand) {
  99
 100       if (entry->composetypecommand)
 101         strfcpy (command, entry->composetypecommand, sizeof (command));
 102       else
 103         strfcpy (command, entry->composecommand, sizeof (command));
 104       if (rfc1524_expand_filename (entry->nametemplate,
 105                                    a->filename, newfile, sizeof (newfile))) {
 106         dprint (1, (debugfile, "oldfile: %s\t newfile: %s\n",
 107                     a->filename, newfile));
 108         if (safe_symlink (a->filename, newfile) == -1) {
 109           if (mutt_yesorno (_("Can't match nametemplate, continue?"), M_YES)
 110               != M_YES)
 111             goto bailout;
 112         }
 113         else
 114           unlink_newfile = 1;
 115       }
 116       else
 117         strfcpy (newfile, a->filename, sizeof (newfile));
 118
 119       if (rfc1524_expand_command (a, newfile, type,
 120                                   command, sizeof (command))) {
 121         /* For now, editing requires a file, no piping */
 122         mutt_error _("Mailcap compose entry requires %%s");
 123       }
 124       else {
 125         int r;
 126
 127         mutt_endwin (NULL);
 128         if ((r = mutt_system (command)) == -1)
 129           mutt_error (_("Error running \"%s\"!"), command);
 130
 131         if (r != -1 && entry->composetypecommand) {
 132           BODY *b;
 133           FILE *fp, *tfp;
 134           char tempfile[_POSIX_PATH_MAX];
 135
 136           if ((fp = safe_fopen (a->filename, "r")) == NULL) {
 137             mutt_perror _("Failure to open file to parse headers.");
 138
 139             goto bailout;
 140           }
 141
 142           b = mutt_read_mime_header (fp, 0);
 143           if (b) {
 144             if (b->parameter) {
 145               mutt_free_parameter (&a->parameter);
 146               a->parameter = b->parameter;
 147               b->parameter = NULL;
 148             }
 149             if (b->description) {
 150               FREE (&a->description);
 151               a->description = b->description;
 152               b->description = NULL;
 153             }
 154             if (b->form_name) {
 155               FREE (&a->form_name);
 156               a->form_name = b->form_name;
 157               b->form_name = NULL;
 158             }
 159
 160             /* Remove headers by copying out data to another file, then
 161              * copying the file back */
 162             fseek (fp, b->offset, 0);
 163             mutt_mktemp (tempfile);
 164             if ((tfp = safe_fopen (tempfile, "w")) == NULL) {
 165               mutt_perror _("Failure to open file to strip headers.");
 166
 167               goto bailout;
 168             }
 169             mutt_copy_stream (fp, tfp);
 170             fclose (fp);
 171             fclose (tfp);
 172             mutt_unlink (a->filename);
 173             if (mutt_rename_file (tempfile, a->filename) != 0) {
 174               mutt_perror _("Failure to rename file.");
 175
 176               goto bailout;
 177             }
 178
 179             mutt_free_body (&b);
 180           }
 181         }
 182       }
 183     }
 184   }
 185   else {
 186     rfc1524_free_entry (&entry);
 187     mutt_message (_("No mailcap compose entry for %s, creating empty file."),
 188                   type);
 189     return 1;
 190   }
 191
 192   rc = 1;
 193
 194 bailout:
 195
 196   if (unlink_newfile)
 197     unlink (newfile);
 198
 199   rfc1524_free_entry (&entry);
 200   return rc;
 201 }
 202
 203 /*
 204  * Currently, this only works for send mode, as it assumes that the
 205  * BODY->filename actually contains the information.  I'm not sure
 206  * we want to deal with editing attachments we've already received,
 207  * so this should be ok.
 208  *
 209  * Returns 1 if editor found, 0 if not (useful to tell calling menu to
 210  * redraw)
 211  */
 212 int mutt_edit_attachment (BODY * a)
 213 {
 214   char type[STRING];
 215   char command[STRING];
 216   char newfile[_POSIX_PATH_MAX] = "";
 217   rfc1524_entry *entry = rfc1524_new_entry ();
 218   short unlink_newfile = 0;
 219   int rc = 0;
 220
 221   snprintf (type, sizeof (type), "%s/%s", TYPE (a), a->subtype);
 222   if (rfc1524_mailcap_lookup (a, type, entry, M_EDIT)) {
 223     if (entry->editcommand) {
 224
 225       strfcpy (command, entry->editcommand, sizeof (command));
 226       if (rfc1524_expand_filename (entry->nametemplate,
 227                                    a->filename, newfile, sizeof (newfile))) {
 228         dprint (1, (debugfile, "oldfile: %s\t newfile: %s\n",
 229                     a->filename, newfile));
 230         if (safe_symlink (a->filename, newfile) == -1) {
 231           if (mutt_yesorno (_("Can't match nametemplate, continue?"), M_YES)
 232               != M_YES)
 233             goto bailout;
 234         }
 235         else
 236           unlink_newfile = 1;
 237       }
 238       else
 239         strfcpy (newfile, a->filename, sizeof (newfile));
 240
 241       if (rfc1524_expand_command (a, newfile, type,
 242                                   command, sizeof (command))) {
 243         /* For now, editing requires a file, no piping */
 244         mutt_error _("Mailcap Edit entry requires %%s");
 245       }
 246       else {
 247         mutt_endwin (NULL);
 248         if (mutt_system (command) == -1)
 249           mutt_error (_("Error running \"%s\"!"), command);
 250       }
 251     }
 252   }
 253   else if (a->type == TYPETEXT) {
 254     /* On text, default to editor */
 255     mutt_edit_file (NONULL (Editor), newfile);
 256   }
 257   else {
 258     rfc1524_free_entry (&entry);
 259     mutt_error (_("No mailcap edit entry for %s"), type);
 260     return 0;
 261   }
 262
 263   rc = 1;
 264
 265 bailout:
 266
 267   if (unlink_newfile)
 268     unlink (newfile);
 269
 270   rfc1524_free_entry (&entry);
 271   return rc;
 272 }
 273
 274
 275 /* for compatibility with metamail */
 276 static int is_mmnoask (const char *buf)
 277 {
 278   char tmp[LONG_STRING], *p, *q;
 279   int lng;
 280
 281   if ((p = getenv ("MM_NOASK")) != NULL && *p) {
 282     if (mutt_strcmp (p, "1") == 0)
 283       return (1);
 284
 285     strfcpy (tmp, p, sizeof (tmp));
 286     p = tmp;
 287
 288     while ((p = strtok (p, ",")) != NULL) {
 289       if ((q = strrchr (p, '/')) != NULL) {
 290         if (*(q + 1) == '*') {
 291           if (ascii_strncasecmp (buf, p, q - p) == 0)
 292             return (1);
 293         }
 294         else {
 295           if (ascii_strcasecmp (buf, p) == 0)
 296             return (1);
 297         }
 298       }
 299       else {
 300         lng = mutt_strlen (p);
 301         if (buf[lng] == '/' && mutt_strncasecmp (buf, p, lng) == 0)
 302           return (1);
 303       }
 304
 305       p = NULL;
 306     }
 307   }
 308
 309   return (0);
 310 }
 311
 312 void mutt_check_lookup_list (BODY * b, char *type, int len)
 313 {
 314   LIST *t = MimeLookupList;
 315   int i;
 316
 317   for (; t; t = t->next) {
 318     i = mutt_strlen (t->data) - 1;
 319     if ((i > 0 && t->data[i - 1] == '/' && t->data[i] == '*' &&
 320          ascii_strncasecmp (type, t->data, i) == 0) ||
 321         ascii_strcasecmp (type, t->data) == 0) {
 322
 323       BODY tmp = { 0 };
 324       int n;
 325
 326       if ((n = mutt_lookup_mime_type (&tmp, b->filename)) != TYPEOTHER) {
 327         snprintf (type, len, "%s/%s",
 328                   n == TYPEAUDIO ? "audio" :
 329                   n == TYPEAPPLICATION ? "application" :
 330                   n == TYPEIMAGE ? "image" :
 331                   n == TYPEMESSAGE ? "message" :
 332                   n == TYPEMODEL ? "model" :
 333                   n == TYPEMULTIPART ? "multipart" :
 334                   n == TYPETEXT ? "text" :
 335                   n == TYPEVIDEO ? "video" : "other", tmp.subtype);
 336         dprint (1, (debugfile, "mutt_check_lookup_list: \"%s\" -> %s\n",
 337                     b->filename, type));
 338       }
 339       if (tmp.subtype)
 340         FREE (&tmp.subtype);
 341       if (tmp.xtype)
 342         FREE (&tmp.xtype);
 343     }
 344   }
 345 }
 346
 347 int mutt_is_autoview (BODY * b, const char *type)
 348 {
 349   LIST *t = AutoViewList;
 350   char _type[SHORT_STRING];
 351   int i;
 352
 353   if (!type)
 354     snprintf (_type, sizeof (_type), "%s/%s", TYPE (b), b->subtype);
 355   else
 356     strncpy (_type, type, sizeof (_type));
 357
 358   mutt_check_lookup_list (b, _type, sizeof (_type));
 359   type = _type;
 360
 361   if (mutt_needs_mailcap (b)) {
 362     if (option (OPTIMPLICITAUTOVIEW))
 363       return 1;
 364
 365     if (is_mmnoask (type))
 366       return 1;
 367   }
 368
 369   for (; t; t = t->next) {
 370     i = mutt_strlen (t->data) - 1;
 371     if ((i > 0 && t->data[i - 1] == '/' && t->data[i] == '*' &&
 372          ascii_strncasecmp (type, t->data, i) == 0) ||
 373         ascii_strcasecmp (type, t->data) == 0)
 374       return 1;
 375   }
 376
 377   return 0;
 378 }
 379
 380 /* returns -1 on error, 0 or the return code from mutt_do_pager() on success */
 381 int mutt_view_attachment (FILE * fp, BODY * a, int flag, HEADER * hdr,
 382                           ATTACHPTR ** idx, short idxlen)
 383 {
 384   char tempfile[_POSIX_PATH_MAX] = "";
 385   char pagerfile[_POSIX_PATH_MAX] = "";
 386   int is_message;
 387   int use_mailcap;
 388   int use_pipe = 0;
 389   int use_pager = 1;
 390   char type[STRING];
 391   char command[STRING];
 392   char descrip[STRING];
 393   char *fname;
 394   rfc1524_entry *entry = NULL;
 395   int rc = -1;
 396   int unlink_tempfile = 0;
 397
 398   is_message = mutt_is_message_type (a->type, a->subtype);
 399   if (WithCrypto && is_message && a->hdr && (a->hdr->security & ENCRYPT) &&
 400       !crypt_valid_passphrase (a->hdr->security))
 401     return (rc);
 402   use_mailcap = (flag == M_MAILCAP ||
 403                  (flag == M_REGULAR && mutt_needs_mailcap (a)));
 404   snprintf (type, sizeof (type), "%s/%s", TYPE (a), a->subtype);
 405
 406   if (use_mailcap) {
 407     entry = rfc1524_new_entry ();
 408     if (!rfc1524_mailcap_lookup (a, type, entry, 0)) {
 409       if (flag == M_REGULAR) {
 410         /* fallback to view as text */
 411         rfc1524_free_entry (&entry);
 412         mutt_error _("No matching mailcap entry found.  Viewing as text.");
 413
 414         flag = M_AS_TEXT;
 415         use_mailcap = 0;
 416       }
 417       else
 418         goto return_error;
 419     }
 420   }
 421
 422   if (use_mailcap) {
 423     if (!entry->command) {
 424       mutt_error _("MIME type not defined.  Cannot view attachment.");
 425
 426       goto return_error;
 427     }
 428     strfcpy (command, entry->command, sizeof (command));
 429
 430     if (fp) {
 431       fname = safe_strdup (a->filename);
 432       mutt_sanitize_filename (fname, 1);
 433     }
 434     else
 435       fname = a->filename;
 436
 437     if (rfc1524_expand_filename (entry->nametemplate, fname,
 438                                  tempfile, sizeof (tempfile))) {
 439       if (fp == NULL && mutt_strcmp (tempfile, a->filename)) {
 440         /* send case: the file is already there */
 441         if (safe_symlink (a->filename, tempfile) == -1) {
 442           if (mutt_yesorno (_("Can't match nametemplate, continue?"), M_YES)
 443               == M_YES)
 444             strfcpy (tempfile, a->filename, sizeof (tempfile));
 445           else
 446             goto return_error;
 447         }
 448         else
 449           unlink_tempfile = 1;
 450       }
 451     }
 452     else if (fp == NULL)        /* send case */
 453       strfcpy (tempfile, a->filename, sizeof (tempfile));
 454
 455     if (fp) {
 456       /* recv case: we need to save the attachment to a file */
 457       FREE (&fname);
 458       if (mutt_save_attachment (fp, a, tempfile, 0, NULL) == -1)
 459         goto return_error;
 460     }
 461
 462     use_pipe = rfc1524_expand_command (a, tempfile, type,
 463                                        command, sizeof (command));
 464     use_pager = entry->copiousoutput;
 465   }
 466
 467   if (use_pager) {
 468     if (fp && !use_mailcap && a->filename) {
 469       /* recv case */
 470       strfcpy (pagerfile, a->filename, sizeof (pagerfile));
 471       mutt_adv_mktemp (pagerfile, sizeof (pagerfile));
 472     }
 473     else
 474       mutt_mktemp (pagerfile);
 475   }
 476
 477   if (use_mailcap) {
 478     pid_t thepid = 0;
 479     int tempfd = -1, pagerfd = -1;
 480
 481     if (!use_pager)
 482       mutt_endwin (NULL);
 483
 484     if (use_pager || use_pipe) {
 485       if (use_pager
 486           && ((pagerfd = safe_open (pagerfile, O_CREAT | O_EXCL | O_WRONLY))
 487               == -1)) {
 488         mutt_perror ("open");
 489         goto return_error;
 490       }
 491       if (use_pipe && ((tempfd = open (tempfile, 0)) == -1)) {
 492         if (pagerfd != -1)
 493           close (pagerfd);
 494         mutt_perror ("open");
 495         goto return_error;
 496       }
 497
 498       if ((thepid = mutt_create_filter_fd (command, NULL, NULL, NULL,
 499                                            use_pipe ? tempfd : -1,
 500                                            use_pager ? pagerfd : -1,
 501                                            -1)) == -1) {
 502         if (pagerfd != -1)
 503           close (pagerfd);
 504
 505         if (tempfd != -1)
 506           close (tempfd);
 507
 508         mutt_error _("Cannot create filter");
 509
 510         goto return_error;
 511       }
 512
 513       if (use_pager) {
 514         if (a->description)
 515           snprintf (descrip, sizeof (descrip),
 516                     "---Command: %-20.20s Description: %s",
 517                     command, a->description);
 518         else
 519           snprintf (descrip, sizeof (descrip),
 520                     "---Command: %-30.30s Attachment: %s", command, type);
 521       }
 522
 523       if ((mutt_wait_filter (thepid) || (entry->needsterminal &&
 524                                          option (OPTWAITKEY))) && !use_pager)
 525         mutt_any_key_to_continue (NULL);
 526
 527       close (tempfd);
 528       close (pagerfd);
 529
 530     }
 531     else {
 532       /* interactive command */
 533       if (mutt_system (command) ||
 534           (entry->needsterminal && option (OPTWAITKEY)))
 535         mutt_any_key_to_continue (NULL);
 536     }
 537   }
 538   else {
 539     /* Don't use mailcap; the attachment is viewed in the pager */
 540
 541     if (flag == M_AS_TEXT) {
 542       /* just let me see the raw data */
 543       if (mutt_save_attachment (fp, a, pagerfile, 0, NULL))
 544         goto return_error;
 545     }
 546     else {
 547       /* Use built-in handler */
 548       set_option (OPTVIEWATTACH);       /* disable the "use 'v' to view this part"
 549                                          * message in case of error */
 550       if (mutt_decode_save_attachment (fp, a, pagerfile, M_DISPLAY, 0)) {
 551         unset_option (OPTVIEWATTACH);
 552         goto return_error;
 553       }
 554       unset_option (OPTVIEWATTACH);
 555     }
 556
 557     if (a->description)
 558       strfcpy (descrip, a->description, sizeof (descrip));
 559     else if (a->filename)
 560       snprintf (descrip, sizeof (descrip), "---Attachment: %s : %s",
 561                 a->filename, type);
 562     else
 563       snprintf (descrip, sizeof (descrip), "---Attachment: %s", type);
 564   }
 565
 566   /* We only reach this point if there have been no errors */
 567
 568   if (use_pager) {
 569     pager_t info;
 570
 571     memset (&info, 0, sizeof (info));
 572     info.fp = fp;
 573     info.bdy = a;
 574     info.ctx = Context;
 575     info.idx = idx;
 576     info.idxlen = idxlen;
 577     info.hdr = hdr;
 578
 579     rc = mutt_do_pager (descrip, pagerfile,
 580                         M_PAGER_ATTACHMENT | (is_message ? M_PAGER_MESSAGE :
 581                                               0), &info);
 582     *pagerfile = '\0';
 583   }
 584   else
 585     rc = 0;
 586
 587 return_error:
 588
 589   if (entry)
 590     rfc1524_free_entry (&entry);
 591   if (fp && tempfile[0])
 592     mutt_unlink (tempfile);
 593   else if (unlink_tempfile)
 594     unlink (tempfile);
 595
 596   if (pagerfile[0])
 597     mutt_unlink (pagerfile);
 598
 599   return rc;
 600 }
 601
 602 /* returns 1 on success, 0 on error */
 603 int mutt_pipe_attachment (FILE * fp, BODY * b, const char *path,
 604                           char *outfile)
 605 {
 606   pid_t thepid;
 607   int out = -1;
 608   int rv = 0;
 609
 610   if (outfile && *outfile)
 611     if ((out = safe_open (outfile, O_CREAT | O_EXCL | O_WRONLY)) < 0) {
 612       mutt_perror ("open");
 613       return 0;
 614     }
 615
 616   mutt_endwin (NULL);
 617
 618   if (fp) {
 619     /* recv case */
 620
 621     STATE s;
 622
 623     memset (&s, 0, sizeof (STATE));
 624
 625     if (outfile && *outfile)
 626       thepid =
 627         mutt_create_filter_fd (path, &s.fpout, NULL, NULL, -1, out, -1);
 628     else
 629       thepid = mutt_create_filter (path, &s.fpout, NULL, NULL);
 630
 631     if (thepid < 0) {
 632       mutt_perror _("Can't create filter");
 633
 634       goto bail;
 635     }
 636
 637     s.fpin = fp;
 638     mutt_decode_attachment (b, &s);
 639     safe_fclose (&s.fpout);
 640   }
 641   else {
 642     /* send case */
 643
 644     FILE *ifp, *ofp;
 645
 646     if ((ifp = fopen (b->filename, "r")) == NULL) {
 647       mutt_perror ("fopen");
 648       if (outfile && *outfile) {
 649         close (out);
 650         unlink (outfile);
 651       }
 652       return 0;
 653     }
 654
 655     if (outfile && *outfile)
 656       thepid = mutt_create_filter_fd (path, &ofp, NULL, NULL, -1, out, -1);
 657     else
 658       thepid = mutt_create_filter (path, &ofp, NULL, NULL);
 659
 660     if (thepid < 0) {
 661       mutt_perror _("Can't create filter");
 662
 663       safe_fclose (&ifp);
 664       goto bail;
 665     }
 666
 667     mutt_copy_stream (ifp, ofp);
 668     safe_fclose (&ofp);
 669     safe_fclose (&ifp);
 670   }
 671
 672   rv = 1;
 673
 674 bail:
 675
 676   if (outfile && *outfile)
 677     close (out);
 678
 679   /*
 680    * check for error exit from child process
 681    */
 682   if (mutt_wait_filter (thepid) != 0)
 683     rv = 0;
 684
 685   if (rv == 0 || option (OPTWAITKEY))
 686     mutt_any_key_to_continue (NULL);
 687   return rv;
 688 }
 689
 690 static FILE *mutt_save_attachment_open (char *path, int flags)
 691 {
 692   if (flags == M_SAVE_APPEND)
 693     return fopen (path, "a");
 694   /* be sure not to change the following fopen to safe_fopen
 695    * as safe_fopen returns w/ an error if path exists
 696    */
 697   if (flags == M_SAVE_OVERWRITE)
 698     return fopen (path, "w");   /* __FOPEN_CHECKED__ */
 699
 700   return safe_fopen (path, "w");
 701 }
 702
 703 /* returns 0 on success, -1 on error */
 704 int mutt_save_attachment (FILE * fp, BODY * m, char *path, int flags,
 705                           HEADER * hdr)
 706 {
 707   if (fp) {
 708
 709     /* recv mode */
 710
 711     if (hdr &&
 712         m->hdr &&
 713         m->encoding != ENCBASE64 &&
 714         m->encoding != ENCQUOTEDPRINTABLE &&
 715         mutt_is_message_type (m->type, m->subtype)) {
 716       /* message type attachments are written to mail folders. */
 717
 718       char buf[HUGE_STRING];
 719       HEADER *hn;
 720       CONTEXT ctx;
 721       MESSAGE *msg;
 722       int chflags = 0;
 723       int r = -1;
 724
 725       hn = m->hdr;
 726       hn->msgno = hdr->msgno;   /* required for MH/maildir */
 727       hn->read = 1;
 728
 729       fseek (fp, m->offset, 0);
 730       if (fgets (buf, sizeof (buf), fp) == NULL)
 731         return -1;
 732       if (mx_open_mailbox (path, M_APPEND | M_QUIET, &ctx) == NULL)
 733         return -1;
 734       if ((msg =
 735            mx_open_new_message (&ctx, hn,
 736                                 is_from (buf, NULL, 0,
 737                                          NULL) ? 0 : M_ADD_FROM)) == NULL) {
 738         mx_close_mailbox (&ctx, NULL);
 739         return -1;
 740       }
 741       if (ctx.magic == M_MBOX || ctx.magic == M_MMDF)
 742         chflags = CH_FROM;
 743       chflags |= (ctx.magic == M_MAILDIR ? CH_NOSTATUS : CH_UPDATE);
 744       if (_mutt_copy_message (msg->fp, fp, hn, hn->content, 0, chflags) == 0
 745           && mx_commit_message (msg, &ctx) == 0)
 746         r = 0;
 747       else
 748         r = -1;
 749
 750       mx_close_message (&msg);
 751       mx_close_mailbox (&ctx, NULL);
 752       return r;
 753     }
 754     else {
 755       /* In recv mode, extract from folder and decode */
 756
 757       STATE s;
 758
 759       memset (&s, 0, sizeof (s));
 760       if ((s.fpout = mutt_save_attachment_open (path, flags)) == NULL) {
 761         mutt_perror ("fopen");
 762         return (-1);
 763       }
 764       fseek ((s.fpin = fp), m->offset, 0);
 765       mutt_decode_attachment (m, &s);
 766
 767       if (fclose (s.fpout) != 0) {
 768         mutt_perror ("fclose");
 769         return (-1);
 770       }
 771     }
 772   }
 773   else {
 774     /* In send mode, just copy file */
 775
 776     FILE *ofp, *nfp;
 777
 778     if ((ofp = fopen (m->filename, "r")) == NULL) {
 779       mutt_perror ("fopen");
 780       return (-1);
 781     }
 782
 783     if ((nfp = mutt_save_attachment_open (path, flags)) == NULL) {
 784       mutt_perror ("fopen");
 785       safe_fclose (&ofp);
 786       return (-1);
 787     }
 788
 789     if (mutt_copy_stream (ofp, nfp) == -1) {
 790       mutt_error _("Write fault!");
 791
 792       safe_fclose (&ofp);
 793       safe_fclose (&nfp);
 794       return (-1);
 795     }
 796     safe_fclose (&ofp);
 797     safe_fclose (&nfp);
 798   }
 799
 800   return 0;
 801 }
 802
 803 /* returns 0 on success, -1 on error */
 804 int mutt_decode_save_attachment (FILE * fp, BODY * m, char *path,
 805                                  int displaying, int flags)
 806 {
 807   STATE s;
 808   unsigned int saved_encoding = 0;
 809   BODY *saved_parts = NULL;
 810   HEADER *saved_hdr = NULL;
 811
 812   memset (&s, 0, sizeof (s));
 813   s.flags = displaying;
 814
 815   if (flags == M_SAVE_APPEND)
 816     s.fpout = fopen (path, "a");
 817   else if (flags == M_SAVE_OVERWRITE)
 818     s.fpout = safe_fopen (path, "w");   /* __FOPEN_CHECKED__ */
 819   else
 820     s.fpout = safe_fopen (path, "w");
 821
 822   if (s.fpout == NULL) {
 823     mutt_perror ("fopen");
 824     return (-1);
 825   }
 826
 827   if (fp == NULL) {
 828     /* When called from the compose menu, the attachment isn't parsed,
 829      * so we need to do it here. */
 830     struct stat st;
 831
 832     if (stat (m->filename, &st) == -1) {
 833       mutt_perror ("stat");
 834       fclose (s.fpout);
 835       return (-1);
 836     }
 837
 838     if ((s.fpin = fopen (m->filename, "r")) == NULL) {
 839       mutt_perror ("fopen");
 840       return (-1);
 841     }
 842
 843     saved_encoding = m->encoding;
 844     if (!is_multipart (m))
 845       m->encoding = ENC8BIT;
 846
 847     m->length = st.st_size;
 848     m->offset = 0;
 849     saved_parts = m->parts;
 850     saved_hdr = m->hdr;
 851     mutt_parse_part (s.fpin, m);
 852
 853     if (m->noconv || is_multipart (m))
 854       s.flags |= M_CHARCONV;
 855   }
 856   else {
 857     s.fpin = fp;
 858     s.flags |= M_CHARCONV;
 859   }
 860
 861   mutt_body_handler (m, &s);
 862
 863   fclose (s.fpout);
 864   if (fp == NULL) {
 865     m->length = 0;
 866     m->encoding = saved_encoding;
 867     if (saved_parts) {
 868       mutt_free_header (&m->hdr);
 869       m->parts = saved_parts;
 870       m->hdr = saved_hdr;
 871     }
 872     fclose (s.fpin);
 873   }
 874
 875   return (0);
 876 }
 877
 878 /* Ok, the difference between send and receive:
 879  * recv: BODY->filename is a suggested name, and Context|HEADER points
 880  *       to the attachment in mailbox which is encooded
 881  * send: BODY->filename points to the un-encoded file which contains the
 882  *       attachment
 883  */
 884
 885 int mutt_print_attachment (FILE * fp, BODY * a)
 886 {
 887   char newfile[_POSIX_PATH_MAX] = "";
 888   char type[STRING];
 889   pid_t thepid;
 890   FILE *ifp, *fpout;
 891   short unlink_newfile = 0;
 892
 893   snprintf (type, sizeof (type), "%s/%s", TYPE (a), a->subtype);
 894
 895   if (rfc1524_mailcap_lookup (a, type, NULL, M_PRINT)) {
 896     char command[_POSIX_PATH_MAX + STRING];
 897     rfc1524_entry *entry;
 898     int piped = FALSE;
 899
 900     dprint (2, (debugfile, "Using mailcap...\n"));
 901
 902     entry = rfc1524_new_entry ();
 903     rfc1524_mailcap_lookup (a, type, entry, M_PRINT);
 904     if (rfc1524_expand_filename (entry->nametemplate, a->filename,
 905                                  newfile, sizeof (newfile))) {
 906       if (!fp) {
 907         if (safe_symlink (a->filename, newfile) == -1) {
 908           if (mutt_yesorno (_("Can't match nametemplate, continue?"), M_YES)
 909               != M_YES) {
 910             rfc1524_free_entry (&entry);
 911             return 0;
 912           }
 913           strfcpy (newfile, a->filename, sizeof (newfile));
 914         }
 915         else
 916           unlink_newfile = 1;
 917       }
 918     }
 919
 920     /* in recv mode, save file to newfile first */
 921     if (fp)
 922       mutt_save_attachment (fp, a, newfile, 0, NULL);
 923
 924     strfcpy (command, entry->printcommand, sizeof (command));
 925     piped =
 926       rfc1524_expand_command (a, newfile, type, command, sizeof (command));
 927
 928     mutt_endwin (NULL);
 929
 930     /* interactive program */
 931     if (piped) {
 932       if ((ifp = fopen (newfile, "r")) == NULL) {
 933         mutt_perror ("fopen");
 934         rfc1524_free_entry (&entry);
 935         return (0);
 936       }
 937
 938       if ((thepid = mutt_create_filter (command, &fpout, NULL, NULL)) < 0) {
 939         mutt_perror _("Can't create filter");
 940
 941         rfc1524_free_entry (&entry);
 942         safe_fclose (&ifp);
 943         return 0;
 944       }
 945       mutt_copy_stream (ifp, fpout);
 946       safe_fclose (&fpout);
 947       safe_fclose (&ifp);
 948       if (mutt_wait_filter (thepid) || option (OPTWAITKEY))
 949         mutt_any_key_to_continue (NULL);
 950     }
 951     else {
 952       if (mutt_system (command) || option (OPTWAITKEY))
 953         mutt_any_key_to_continue (NULL);
 954     }
 955
 956     if (fp)
 957       mutt_unlink (newfile);
 958     else if (unlink_newfile)
 959       unlink (newfile);
 960
 961     rfc1524_free_entry (&entry);
 962     return (1);
 963   }
 964
 965   if (!ascii_strcasecmp ("text/plain", type) ||
 966       !ascii_strcasecmp ("application/postscript", type)) {
 967     return (mutt_pipe_attachment (fp, a, NONULL (PrintCmd), NULL));
 968   }
 969   else if (mutt_can_decode (a)) {
 970     /* decode and print */
 971
 972     int rc = 0;
 973
 974     ifp = NULL;
 975     fpout = NULL;
 976
 977     mutt_mktemp (newfile);
 978     if (mutt_decode_save_attachment (fp, a, newfile, M_PRINTING, 0) == 0) {
 979
 980       dprint (2,
 981               (debugfile, "successfully decoded %s type attachment to %s\n",
 982                type, newfile));
 983
 984       if ((ifp = fopen (newfile, "r")) == NULL) {
 985         mutt_perror ("fopen");
 986         goto bail0;
 987       }
 988
 989       dprint (2, (debugfile, "successfully opened %s read-only\n", newfile));
 990
 991       mutt_endwin (NULL);
 992       if ((thepid =
 993            mutt_create_filter (NONULL (PrintCmd), &fpout, NULL, NULL)) < 0) {
 994         mutt_perror _("Can't create filter");
 995
 996         goto bail0;
 997       }
 998
 999       dprint (2, (debugfile, "Filter created.\n"));
1000
1001       mutt_copy_stream (ifp, fpout);
1002
1003       safe_fclose (&fpout);
1004       safe_fclose (&ifp);
1005
1006       if (mutt_wait_filter (thepid) != 0 || option (OPTWAITKEY))
1007         mutt_any_key_to_continue (NULL);
1008       rc = 1;
1009     }
1010   bail0:
1011     safe_fclose (&ifp);
1012     safe_fclose (&fpout);
1013     mutt_unlink (newfile);
1014     return rc;
1015   }
1016   else {
1017     mutt_error _("I don't know how to print that!");
1018
1019     return 0;
1020   }
1021 }