src/procmime.c

   1 /*
   2  * Sylpheed -- a GTK+ based, lightweight, and fast e-mail client
   3  * Copyright (C) 1999-2002 Hiroyuki Yamamoto
   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-1307, USA.
  18  */
  19
  20 #ifdef HAVE_CONFIG_H
  21 #  include "config.h"
  22 #endif
  23
  24 #include "defs.h"
  25
  26 #include <glib.h>
  27 #include <stdio.h>
  28 #include <string.h>
  29 #include <locale.h>
  30 #include <ctype.h>
  31
  32 #include "intl.h"
  33 #include "procmime.h"
  34 #include "procheader.h"
  35 #include "base64.h"
  36 #include "uuencode.h"
  37 #include "unmime.h"
  38 #include "html.h"
  39 #include "enriched.h"
  40 #include "codeconv.h"
  41 #include "utils.h"
  42 #include "prefs_common.h"
  43
  44 #if USE_GPGME
  45 #  include "rfc2015.h"
  46 #endif
  47
  48 #include "prefs.h"
  49
  50 static GHashTable *procmime_get_mime_type_table (void);
  51
  52 MimeInfo *procmime_mimeinfo_new(void)
  53 {
  54         MimeInfo *mimeinfo;
  55
  56         mimeinfo = g_new0(MimeInfo, 1);
  57         mimeinfo->mime_type     = MIME_UNKNOWN;
  58         mimeinfo->encoding_type = ENC_UNKNOWN;
  59
  60         return mimeinfo;
  61 }
  62
  63 void procmime_mimeinfo_free(MimeInfo *mimeinfo)
  64 {
  65         if (!mimeinfo) return;
  66
  67         g_free(mimeinfo->encoding);
  68         g_free(mimeinfo->content_type);
  69         g_free(mimeinfo->charset);
  70         g_free(mimeinfo->name);
  71         g_free(mimeinfo->boundary);
  72         g_free(mimeinfo->content_disposition);
  73         g_free(mimeinfo->filename);
  74 #if USE_GPGME
  75         g_free(mimeinfo->plaintextfile);
  76         g_free(mimeinfo->sigstatus);
  77         g_free(mimeinfo->sigstatus_full);
  78 #endif
  79
  80         procmime_mimeinfo_free(mimeinfo->sub);
  81
  82         g_free(mimeinfo);
  83 }
  84
  85 void procmime_mimeinfo_free_all(MimeInfo *mimeinfo)
  86 {
  87         while (mimeinfo != NULL) {
  88                 MimeInfo *next;
  89
  90                 g_free(mimeinfo->encoding);
  91                 g_free(mimeinfo->content_type);
  92                 g_free(mimeinfo->charset);
  93                 g_free(mimeinfo->name);
  94                 g_free(mimeinfo->boundary);
  95                 g_free(mimeinfo->content_disposition);
  96                 g_free(mimeinfo->filename);
  97 #if USE_GPGME
  98                 g_free(mimeinfo->plaintextfile);
  99                 g_free(mimeinfo->sigstatus);
 100                 g_free(mimeinfo->sigstatus_full);
 101 #endif
 102
 103                 procmime_mimeinfo_free_all(mimeinfo->sub);
 104                 procmime_mimeinfo_free_all(mimeinfo->children);
 105 #if USE_GPGME
 106                 procmime_mimeinfo_free_all(mimeinfo->plaintext);
 107 #endif
 108
 109                 next = mimeinfo->next;
 110                 g_free(mimeinfo);
 111                 mimeinfo = next;
 112         }
 113 }
 114
 115 MimeInfo *procmime_mimeinfo_insert(MimeInfo *parent, MimeInfo *mimeinfo)
 116 {
 117         MimeInfo *child = parent->children;
 118
 119         if (!child)
 120                 parent->children = mimeinfo;
 121         else {
 122                 while (child->next != NULL)
 123                         child = child->next;
 124
 125                 child->next = mimeinfo;
 126         }
 127
 128         mimeinfo->parent = parent;
 129         mimeinfo->level = parent->level + 1;
 130
 131         return mimeinfo;
 132 }
 133
 134 void procmime_mimeinfo_replace(MimeInfo *old, MimeInfo *new)
 135 {
 136         MimeInfo *parent = old->parent;
 137         MimeInfo *child;
 138
 139         g_return_if_fail(parent != NULL);
 140         g_return_if_fail(new->next == NULL);
 141
 142         for (child = parent->children; child && child != old;
 143              child = child->next)
 144                 ;
 145         if (!child) {
 146                 g_warning("oops: parent can't find it's own child");
 147                 return;
 148         }
 149         procmime_mimeinfo_free_all(old);
 150
 151         if (child == parent->children) {
 152                 new->next = parent->children->next;
 153                 parent->children = new;
 154         } else {
 155                 new->next = child->next;
 156                 child = new;
 157         }
 158 }
 159
 160 MimeInfo *procmime_mimeinfo_next(MimeInfo *mimeinfo)
 161 {
 162         if (!mimeinfo) return NULL;
 163
 164         if (mimeinfo->children)
 165                 return mimeinfo->children;
 166         if (mimeinfo->sub)
 167                 return mimeinfo->sub;
 168         if (mimeinfo->next)
 169                 return mimeinfo->next;
 170
 171         if (mimeinfo->main) {
 172                 mimeinfo = mimeinfo->main;
 173                 if (mimeinfo->next)
 174                         return mimeinfo->next;
 175         }
 176
 177         for (mimeinfo = mimeinfo->parent; mimeinfo != NULL;
 178              mimeinfo = mimeinfo->parent) {
 179                 if (mimeinfo->next)
 180                         return mimeinfo->next;
 181                 if (mimeinfo->main) {
 182                         mimeinfo = mimeinfo->main;
 183                         if (mimeinfo->next)
 184                                 return mimeinfo->next;
 185                 }
 186         }
 187
 188         return NULL;
 189 }
 190
 191 MimeInfo *procmime_scan_message(MsgInfo *msginfo)
 192 {
 193         FILE *fp;
 194         MimeInfo *mimeinfo;
 195
 196         g_return_val_if_fail(msginfo != NULL, NULL);
 197
 198         if ((fp = procmsg_open_message(msginfo)) == NULL) return NULL;
 199         mimeinfo = procmime_scan_mime_header(fp);
 200
 201         if (mimeinfo) {
 202                 if (mimeinfo->mime_type != MIME_MULTIPART) {
 203                         if (fseek(fp, mimeinfo->fpos, SEEK_SET) < 0)
 204                                 perror("fseek");
 205                 }
 206                 if (mimeinfo->mime_type != MIME_TEXT)
 207                         procmime_scan_multipart_message(mimeinfo, fp);
 208         }
 209
 210 #if USE_GPGME
 211         if (prefs_common.auto_check_signatures)
 212                 rfc2015_check_signature(mimeinfo, fp);
 213 #endif
 214         fclose(fp);
 215
 216         return mimeinfo;
 217 }
 218
 219 void procmime_scan_multipart_message(MimeInfo *mimeinfo, FILE *fp)
 220 {
 221         gchar *p;
 222         gchar *boundary;
 223         gint boundary_len = 0;
 224         gchar buf[BUFFSIZE];
 225         glong fpos, prev_fpos;
 226         gint npart;
 227
 228         g_return_if_fail(mimeinfo != NULL);
 229         g_return_if_fail(mimeinfo->mime_type != MIME_TEXT);
 230
 231         if (mimeinfo->mime_type == MIME_MULTIPART) {
 232                 g_return_if_fail(mimeinfo->boundary != NULL);
 233                 g_return_if_fail(mimeinfo->sub == NULL);
 234         }
 235         g_return_if_fail(fp != NULL);
 236
 237         boundary = mimeinfo->boundary;
 238
 239         if (boundary) {
 240                 boundary_len = strlen(boundary);
 241
 242                 /* look for first boundary */
 243                 while ((p = fgets(buf, sizeof(buf), fp)) != NULL)
 244                         if (IS_BOUNDARY(buf, boundary, boundary_len)) break;
 245                 if (!p) return;
 246         }
 247
 248         if ((fpos = ftell(fp)) < 0) {
 249                 perror("ftell");
 250                 return;
 251         }
 252
 253         for (npart = 0;; npart++) {
 254                 MimeInfo *partinfo;
 255                 gboolean eom = FALSE;
 256
 257                 prev_fpos = fpos;
 258                 debug_print("prev_fpos: %ld\n", fpos);
 259
 260                 partinfo = procmime_scan_mime_header(fp);
 261                 if (!partinfo) break;
 262                 procmime_mimeinfo_insert(mimeinfo, partinfo);
 263
 264                 if (partinfo->mime_type == MIME_MULTIPART) {
 265                         if (partinfo->level < 8)
 266                                 procmime_scan_multipart_message(partinfo, fp);
 267                 } else if (partinfo->mime_type == MIME_MESSAGE_RFC822) {
 268                         MimeInfo *sub;
 269
 270                         partinfo->sub = sub = procmime_scan_mime_header(fp);
 271                         if (!sub) break;
 272
 273                         sub->level = partinfo->level + 1;
 274                         sub->parent = partinfo->parent;
 275                         sub->main = partinfo;
 276
 277                         if (sub->level < 8) {
 278                                 if (sub->mime_type == MIME_MULTIPART) {
 279                                         procmime_scan_multipart_message
 280                                                 (sub, fp);
 281                                 } else if (sub->mime_type == MIME_MESSAGE_RFC822) {
 282                                         fseek(fp, sub->fpos, SEEK_SET);
 283                                         procmime_scan_multipart_message
 284                                                 (sub, fp);
 285                                 }
 286                         }
 287                 }
 288
 289                 /* look for next boundary */
 290                 buf[0] = '\0';
 291                 while ((p = fgets(buf, sizeof(buf), fp)) != NULL) {
 292                         if (IS_BOUNDARY(buf, boundary, boundary_len)) {
 293                                 if (buf[2 + boundary_len]     == '-' &&
 294                                     buf[2 + boundary_len + 1] == '-')
 295                                         eom = TRUE;
 296                                 break;
 297                         }
 298                 }
 299                 if (p == NULL) {
 300                         /* broken MIME, or single part MIME message */
 301                         buf[0] = '\0';
 302                         eom = TRUE;
 303                 }
 304                 fpos = ftell(fp);
 305                 debug_print("fpos: %ld\n", fpos);
 306
 307                 partinfo->size = fpos - prev_fpos - strlen(buf);
 308                 debug_print("partinfo->size: %d\n", partinfo->size);
 309                 if (partinfo->sub && !partinfo->sub->sub &&
 310                     !partinfo->sub->children) {
 311                         partinfo->sub->size = fpos - partinfo->sub->fpos - strlen(buf);
 312                         debug_print("partinfo->sub->size: %d\n",
 313                                     partinfo->sub->size);
 314                 }
 315                 debug_print("boundary: %s\n", buf);
 316
 317                 if (eom) break;
 318         }
 319 }
 320
 321 void procmime_scan_encoding(MimeInfo *mimeinfo, const gchar *encoding)
 322 {
 323         gchar *buf;
 324
 325         Xstrdup_a(buf, encoding, return);
 326
 327         g_free(mimeinfo->encoding);
 328
 329         mimeinfo->encoding = g_strdup(g_strstrip(buf));
 330         if (!strcasecmp(buf, "7bit"))
 331                 mimeinfo->encoding_type = ENC_7BIT;
 332         else if (!strcasecmp(buf, "8bit"))
 333                 mimeinfo->encoding_type = ENC_8BIT;
 334         else if (!strcasecmp(buf, "quoted-printable"))
 335                 mimeinfo->encoding_type = ENC_QUOTED_PRINTABLE;
 336         else if (!strcasecmp(buf, "base64"))
 337                 mimeinfo->encoding_type = ENC_BASE64;
 338         else if (!strcasecmp(buf, "x-uuencode"))
 339                 mimeinfo->encoding_type = ENC_X_UUENCODE;
 340         else
 341                 mimeinfo->encoding_type = ENC_UNKNOWN;
 342
 343 }
 344
 345 void procmime_scan_content_type(MimeInfo *mimeinfo, const gchar *content_type)
 346 {
 347         gchar *delim, *p, *cnttype;
 348         gchar *buf;
 349
 350         if (conv_get_current_charset() == C_EUC_JP &&
 351             strchr(content_type, '\033')) {
 352                 gint len;
 353                 len = strlen(content_type) * 2 + 1;
 354                 Xalloca(buf, len, return);
 355                 conv_jistoeuc(buf, len, content_type);
 356         } else
 357                 Xstrdup_a(buf, content_type, return);
 358
 359         g_free(mimeinfo->content_type);
 360         g_free(mimeinfo->charset);
 361         /* g_free(mimeinfo->name); */
 362         mimeinfo->content_type = NULL;
 363         mimeinfo->charset      = NULL;
 364         /* mimeinfo->name      = NULL; */
 365
 366         if ((delim = strchr(buf, ';'))) *delim = '\0';
 367         mimeinfo->content_type = cnttype = g_strdup(g_strstrip(buf));
 368
 369         mimeinfo->mime_type = procmime_scan_mime_type(cnttype);
 370
 371         if (!delim) return;
 372         p = delim + 1;
 373
 374         for (;;) {
 375                 gchar *eq;
 376                 gchar *attr, *value;
 377
 378                 if ((delim = strchr(p, ';'))) *delim = '\0';
 379
 380                 if (!(eq = strchr(p, '='))) break;
 381
 382                 *eq = '\0';
 383                 attr = p;
 384                 g_strstrip(attr);
 385                 value = eq + 1;
 386                 g_strstrip(value);
 387
 388                 if (*value == '"')
 389                         extract_quote(value, '"');
 390                 else {
 391                         eliminate_parenthesis(value, '(', ')');
 392                         g_strstrip(value);
 393                 }
 394
 395                 if (*value) {
 396                         if (!strcasecmp(attr, "charset"))
 397                                 mimeinfo->charset = g_strdup(value);
 398                         else if (!strcasecmp(attr, "name")) {
 399                                 gchar *tmp;
 400                                 size_t len;
 401
 402                                 len = strlen(value) + 1;
 403                                 Xalloca(tmp, len, return);
 404                                 conv_unmime_header(tmp, len, value, NULL);
 405                                 g_free(mimeinfo->name);
 406                                 mimeinfo->name = g_strdup(tmp);
 407                         } else if (!strcasecmp(attr, "boundary"))
 408                                 mimeinfo->boundary = g_strdup(value);
 409                 }
 410
 411                 if (!delim) break;
 412                 p = delim + 1;
 413         }
 414
 415         if (mimeinfo->mime_type == MIME_MULTIPART && !mimeinfo->boundary)
 416                 mimeinfo->mime_type = MIME_TEXT;
 417 }
 418
 419 void procmime_scan_content_disposition(MimeInfo *mimeinfo,
 420                                        const gchar *content_disposition)
 421 {
 422         gchar *delim, *p, *dispos;
 423         gchar *buf;
 424
 425         if (conv_get_current_charset() == C_EUC_JP &&
 426             strchr(content_disposition, '\033')) {
 427                 gint len;
 428                 len = strlen(content_disposition) * 2 + 1;
 429                 Xalloca(buf, len, return);
 430                 conv_jistoeuc(buf, len, content_disposition);
 431         } else
 432                 Xstrdup_a(buf, content_disposition, return);
 433
 434         if ((delim = strchr(buf, ';'))) *delim = '\0';
 435         mimeinfo->content_disposition = dispos = g_strdup(g_strstrip(buf));
 436
 437         if (!delim) return;
 438         p = delim + 1;
 439
 440         for (;;) {
 441                 gchar *eq;
 442                 gchar *attr, *value;
 443
 444                 if ((delim = strchr(p, ';'))) *delim = '\0';
 445
 446                 if (!(eq = strchr(p, '='))) break;
 447
 448                 *eq = '\0';
 449                 attr = p;
 450                 g_strstrip(attr);
 451                 value = eq + 1;
 452                 g_strstrip(value);
 453
 454                 if (*value == '"')
 455                         extract_quote(value, '"');
 456                 else {
 457                         eliminate_parenthesis(value, '(', ')');
 458                         g_strstrip(value);
 459                 }
 460
 461                 if (*value) {
 462                         if (!strcasecmp(attr, "filename")) {
 463                                 gchar *tmp;
 464                                 size_t len;
 465
 466                                 len = strlen(value) + 1;
 467                                 Xalloca(tmp, len, return);
 468                                 conv_unmime_header(tmp, len, value, NULL);
 469                                 g_free(mimeinfo->filename);
 470                                 mimeinfo->filename = g_strdup(tmp);
 471                                 break;
 472                         }
 473                 }
 474
 475                 if (!delim) break;
 476                 p = delim + 1;
 477         }
 478 }
 479
 480 void procmime_scan_content_description(MimeInfo *mimeinfo,
 481                                        const gchar *content_description)
 482 {
 483         gchar *buf;
 484
 485         gchar *tmp;
 486         size_t blen;
 487
 488         if (conv_get_current_charset() == C_EUC_JP &&
 489             strchr(content_description, '\033')) {
 490                 gint len;
 491                 len = strlen(content_description) * 2 + 1;
 492                 Xalloca(buf, len, return);
 493                 conv_jistoeuc(buf, len, content_description);
 494         } else
 495                 Xstrdup_a(buf, content_description, return);
 496
 497         blen = strlen(buf) + 1;
 498         Xalloca(tmp, blen, return);
 499         conv_unmime_header(tmp, blen, buf, NULL);
 500         g_free(mimeinfo->name);
 501         mimeinfo->name = g_strdup(tmp);
 502 }
 503
 504 void procmime_scan_subject(MimeInfo *mimeinfo,
 505                            const gchar *subject)
 506 {
 507         gchar *buf;
 508
 509         gchar *tmp;
 510         size_t blen;
 511
 512         if (conv_get_current_charset() == C_EUC_JP &&
 513             strchr(subject, '\033')) {
 514                 gint len;
 515                 len = strlen(subject) * 2 + 1;
 516                 Xalloca(buf, len, return);
 517                 conv_jistoeuc(buf, len, subject);
 518         } else
 519                 Xstrdup_a(buf, subject, return);
 520
 521         blen = strlen(buf) + 1;
 522         Xalloca(tmp, blen, return);
 523         conv_unmime_header(tmp, blen, buf, NULL);
 524         g_free(mimeinfo->name);
 525         mimeinfo->name = g_strdup(tmp);
 526 }
 527
 528 enum
 529 {
 530         H_CONTENT_TRANSFER_ENCODING = 0,
 531         H_CONTENT_TYPE              = 1,
 532         H_CONTENT_DISPOSITION       = 2,
 533         H_CONTENT_DESCRIPTION       = 3,
 534         H_SUBJECT                   = 4
 535 };
 536
 537 MimeInfo *procmime_scan_mime_header(FILE *fp)
 538 {
 539         static HeaderEntry hentry[] = {{"Content-Transfer-Encoding:",
 540                                                           NULL, FALSE},
 541                                        {"Content-Type:", NULL, TRUE},
 542                                        {"Content-Disposition:",
 543                                                           NULL, TRUE},
 544                                        {"Content-description:",
 545                                                           NULL, TRUE},
 546                                        {"Subject:",
 547                                                           NULL, TRUE},
 548                                        {NULL,             NULL, FALSE}};
 549         gchar buf[BUFFSIZE];
 550         gint hnum;
 551         HeaderEntry *hp;
 552         MimeInfo *mimeinfo;
 553
 554         g_return_val_if_fail(fp != NULL, NULL);
 555
 556         mimeinfo = procmime_mimeinfo_new();
 557         mimeinfo->mime_type = MIME_TEXT;
 558         mimeinfo->encoding_type = ENC_7BIT;
 559         mimeinfo->fpos = ftell(fp);
 560
 561         while ((hnum = procheader_get_one_field(buf, sizeof(buf), fp, hentry))
 562                != -1) {
 563                 hp = hentry + hnum;
 564
 565                 if (H_CONTENT_TRANSFER_ENCODING == hnum) {
 566                         procmime_scan_encoding
 567                                 (mimeinfo, buf + strlen(hp->name));
 568                 } else if (H_CONTENT_TYPE == hnum) {
 569                         procmime_scan_content_type
 570                                 (mimeinfo, buf + strlen(hp->name));
 571                 } else if (H_CONTENT_DISPOSITION == hnum) {
 572                         procmime_scan_content_disposition
 573                                 (mimeinfo, buf + strlen(hp->name));
 574                 } else if (H_CONTENT_DESCRIPTION == hnum) {
 575                         procmime_scan_content_description
 576                                 (mimeinfo, buf + strlen(hp->name));
 577                 } else if (H_SUBJECT == hnum) {
 578                         procmime_scan_subject
 579                                 (mimeinfo, buf + strlen(hp->name));
 580                 }
 581         }
 582
 583         if (mimeinfo->mime_type == MIME_APPLICATION_OCTET_STREAM &&
 584             mimeinfo->name) {
 585                 const gchar *type;
 586                 type = procmime_get_mime_type(mimeinfo->name);
 587                 if (type)
 588                         mimeinfo->mime_type = procmime_scan_mime_type(type);
 589         }
 590
 591         if (!mimeinfo->content_type)
 592                         mimeinfo->content_type = g_strdup("text/plain");
 593
 594         return mimeinfo;
 595 }
 596
 597 FILE *procmime_decode_content(FILE *outfp, FILE *infp, MimeInfo *mimeinfo)
 598 {
 599         gchar buf[BUFFSIZE];
 600         gchar *boundary = NULL;
 601         gint boundary_len = 0;
 602         gboolean tmp_file = FALSE;
 603
 604         g_return_val_if_fail(infp != NULL, NULL);
 605         g_return_val_if_fail(mimeinfo != NULL, NULL);
 606
 607         if (!outfp) {
 608                 outfp = my_tmpfile();
 609                 if (!outfp) {
 610                         perror("tmpfile");
 611                         return NULL;
 612                 }
 613                 tmp_file = TRUE;
 614         }
 615
 616         if (mimeinfo->parent && mimeinfo->parent->boundary) {
 617                 boundary = mimeinfo->parent->boundary;
 618                 boundary_len = strlen(boundary);
 619         }
 620
 621         if (mimeinfo->encoding_type == ENC_QUOTED_PRINTABLE) {
 622                 gboolean softline = FALSE;
 623
 624                 while (fgets(buf, sizeof(buf), infp) != NULL &&
 625                        (!boundary ||
 626                         !IS_BOUNDARY(buf, boundary, boundary_len))) {
 627                         guchar *p = buf;
 628
 629                         softline = DoOneQPLine(&p, FALSE, softline);
 630                         fwrite(buf, p - (guchar *)buf, 1, outfp);
 631                 }
 632         } else if (mimeinfo->encoding_type == ENC_BASE64) {
 633                 gchar outbuf[BUFFSIZE];
 634                 gint len;
 635                 Base64Decoder *decoder;
 636
 637                 decoder = base64_decoder_new();
 638                 while (fgets(buf, sizeof(buf), infp) != NULL &&
 639                        (!boundary ||
 640                         !IS_BOUNDARY(buf, boundary, boundary_len))) {
 641                         len = base64_decoder_decode(decoder, buf, outbuf);
 642                         if (len < 0) {
 643                                 g_warning("Bad BASE64 content\n");
 644                                 break;
 645                         }
 646                         fwrite(outbuf, sizeof(gchar), len, outfp);
 647                 }
 648                 base64_decoder_free(decoder);
 649         } else if (mimeinfo->encoding_type == ENC_X_UUENCODE) {
 650                 gchar outbuf[BUFFSIZE];
 651                 gint len;
 652                 gboolean flag = FALSE;
 653
 654                 while (fgets(buf, sizeof(buf), infp) != NULL &&
 655                        (!boundary ||
 656                         !IS_BOUNDARY(buf, boundary, boundary_len))) {
 657                         if(!flag && strncmp(buf,"begin ", 6)) continue;
 658
 659                         if (flag) {
 660                                 len = fromuutobits(outbuf, buf);
 661                                 if (len <= 0) {
 662                                         if (len < 0)
 663                                                 g_warning("Bad UUENCODE content(%d)\n", len);
 664                                         break;
 665                                 }
 666                                 fwrite(outbuf, sizeof(gchar), len, outfp);
 667                         } else
 668                                 flag = TRUE;
 669                 }
 670         } else {
 671                 while (fgets(buf, sizeof(buf), infp) != NULL &&
 672                        (!boundary ||
 673                         !IS_BOUNDARY(buf, boundary, boundary_len))) {
 674                         fputs(buf, outfp);
 675                 }
 676         }
 677
 678         if (tmp_file) rewind(outfp);
 679         return outfp;
 680 }
 681
 682 gint procmime_get_part(const gchar *outfile, const gchar *infile,
 683                        MimeInfo *mimeinfo)
 684 {
 685         FILE *infp, *outfp;
 686         gchar buf[BUFFSIZE];
 687
 688         g_return_val_if_fail(outfile != NULL, -1);
 689         g_return_val_if_fail(infile != NULL, -1);
 690         g_return_val_if_fail(mimeinfo != NULL, -1);
 691
 692         if ((infp = fopen(infile, "rb")) == NULL) {
 693                 FILE_OP_ERROR(infile, "fopen");
 694                 return -1;
 695         }
 696         if (fseek(infp, mimeinfo->fpos, SEEK_SET) < 0) {
 697                 FILE_OP_ERROR(infile, "fseek");
 698                 fclose(infp);
 699                 return -1;
 700         }
 701         if ((outfp = fopen(outfile, "wb")) == NULL) {
 702                 FILE_OP_ERROR(outfile, "fopen");
 703                 fclose(infp);
 704                 return -1;
 705         }
 706
 707         while (fgets(buf, sizeof(buf), infp) != NULL)
 708                 if (buf[0] == '\r' || buf[0] == '\n') break;
 709
 710         procmime_decode_content(outfp, infp, mimeinfo);
 711
 712         fclose(infp);
 713         if (fclose(outfp) == EOF) {
 714                 FILE_OP_ERROR(outfile, "fclose");
 715                 unlink(outfile);
 716                 return -1;
 717         }
 718
 719         return 0;
 720 }
 721
 722 struct ContentRenderer {
 723         char * content_type;
 724         char * renderer;
 725 };
 726
 727 static GList * renderer_list = NULL;
 728
 729 static struct ContentRenderer *
 730 content_renderer_new(char * content_type, char * renderer)
 731 {
 732         struct ContentRenderer * cr;
 733
 734         cr = g_new(struct ContentRenderer, 1);
 735         if (cr == NULL)
 736                 return NULL;
 737
 738         cr->content_type = g_strdup(content_type);
 739         cr->renderer = g_strdup(renderer);
 740
 741         return cr;
 742 }
 743
 744 static void content_renderer_free(struct ContentRenderer * cr)
 745 {
 746         g_free(cr->content_type);
 747         g_free(cr->renderer);
 748         g_free(cr);
 749 }
 750
 751 void renderer_read_config(void)
 752 {
 753         gchar buf[BUFFSIZE];
 754         FILE * f;
 755         gchar * rcpath;
 756
 757         g_list_foreach(renderer_list, (GFunc) content_renderer_free, NULL);
 758         renderer_list = NULL;
 759
 760         rcpath = g_strconcat(get_rc_dir(), G_DIR_SEPARATOR_S, RENDERER_RC, NULL);
 761         f = fopen(rcpath, "rb");
 762         g_free(rcpath);
 763
 764         if (f == NULL)
 765                 return;
 766
 767         while (fgets(buf, BUFFSIZE, f)) {
 768                 char * p;
 769                 struct ContentRenderer * cr;
 770
 771                 strretchomp(buf);
 772                 p = strchr(buf, ' ');
 773                 if (p == NULL)
 774                         continue;
 775                 * p = 0;
 776
 777                 cr = content_renderer_new(buf, p + 1);
 778                 if (cr == NULL)
 779                         continue;
 780
 781                 renderer_list = g_list_append(renderer_list, cr);
 782         }
 783
 784         fclose(f);
 785 }
 786
 787 void renderer_write_config(void)
 788 {
 789         gchar * rcpath;
 790         PrefFile *pfile;
 791         GList * cur;
 792
 793         rcpath = g_strconcat(get_rc_dir(), G_DIR_SEPARATOR_S, RENDERER_RC, NULL);
 794
 795         if ((pfile = prefs_write_open(rcpath)) == NULL) {
 796                 g_warning(_("failed to write configuration to file\n"));
 797                 g_free(rcpath);
 798                 return;
 799         }
 800
 801         g_free(rcpath);
 802
 803         for(cur = renderer_list ; cur != NULL ; cur = cur->next) {
 804                 struct ContentRenderer * renderer;
 805                 renderer = cur->data;
 806                 fprintf(pfile->fp, "%s %s\n", renderer->content_type,
 807                         renderer->renderer);
 808         }
 809
 810         if (prefs_write_close(pfile) < 0) {
 811                 g_warning(_("failed to write configuration to file\n"));
 812                 return;
 813         }
 814 }
 815
 816 FILE *procmime_get_text_content(MimeInfo *mimeinfo, FILE *infp)
 817 {
 818         FILE *tmpfp, *outfp;
 819         gchar *src_codeset;
 820         gboolean conv_fail = FALSE;
 821         gchar buf[BUFFSIZE];
 822         gchar *str;
 823         struct ContentRenderer * renderer;
 824         GList * cur;
 825
 826         g_return_val_if_fail(mimeinfo != NULL, NULL);
 827         g_return_val_if_fail(infp != NULL, NULL);
 828         g_return_val_if_fail(mimeinfo->mime_type == MIME_TEXT ||
 829                              mimeinfo->mime_type == MIME_TEXT_HTML ||
 830                              mimeinfo->mime_type == MIME_TEXT_ENRICHED, NULL);
 831
 832         if (fseek(infp, mimeinfo->fpos, SEEK_SET) < 0) {
 833                 perror("fseek");
 834                 return NULL;
 835         }
 836
 837         while (fgets(buf, sizeof(buf), infp) != NULL)
 838                 if (buf[0] == '\r' || buf[0] == '\n') break;
 839
 840         tmpfp = procmime_decode_content(NULL, infp, mimeinfo);
 841         if (!tmpfp)
 842                 return NULL;
 843
 844         if ((outfp = my_tmpfile()) == NULL) {
 845                 perror("tmpfile");
 846                 fclose(tmpfp);
 847                 return NULL;
 848         }
 849
 850         src_codeset = prefs_common.force_charset
 851                 ? prefs_common.force_charset : mimeinfo->charset;
 852
 853         renderer = NULL;
 854
 855         for(cur = renderer_list ; cur != NULL ; cur = cur->next) {
 856                 struct ContentRenderer * cr;
 857                 cr = cur->data;
 858                 if (g_strcasecmp(cr->content_type,
 859                                  mimeinfo->content_type) == 0) {
 860                         renderer = cr;
 861                         break;
 862                 }
 863         }
 864
 865         if (renderer != NULL) {
 866                 FILE * p;
 867                 int oldout;
 868
 869                 oldout = dup(1);
 870
 871                 dup2(fileno(outfp), 1);
 872
 873                 p = popen(renderer->renderer, "w");
 874                 if (p != NULL) {
 875                         size_t count;
 876
 877                         while ((count =
 878                                 fread(buf, sizeof(char), sizeof(buf),
 879                                       tmpfp)) > 0)
 880                                 fwrite(buf, sizeof(char), count, p);
 881                         pclose(p);
 882                 }
 883
 884                 dup2(oldout, 1);
 885         } else if (mimeinfo->mime_type == MIME_TEXT) {
 886                 while (fgets(buf, sizeof(buf), tmpfp) != NULL) {
 887                         str = conv_codeset_strdup(buf, src_codeset, NULL);
 888                         if (str) {
 889                                 fputs(str, outfp);
 890                                 g_free(str);
 891                         } else {
 892                                 conv_fail = TRUE;
 893                                 fputs(buf, outfp);
 894                         }
 895                 }
 896         } else if (mimeinfo->mime_type == MIME_TEXT_HTML) {
 897                 HTMLParser *parser;
 898                 CodeConverter *conv;
 899
 900                 conv = conv_code_converter_new(src_codeset);
 901                 parser = html_parser_new(tmpfp, conv);
 902                 while ((str = html_parse(parser)) != NULL) {
 903                         fputs(str, outfp);
 904                 }
 905                 html_parser_destroy(parser);
 906                 conv_code_converter_destroy(conv);
 907         } else if (mimeinfo->mime_type == MIME_TEXT_ENRICHED) {
 908                 ERTFParser *parser;
 909                 CodeConverter *conv;
 910
 911                 conv = conv_code_converter_new(src_codeset);
 912                 parser = ertf_parser_new(tmpfp, conv);
 913                 while ((str = ertf_parse(parser)) != NULL) {
 914                         fputs(str, outfp);
 915                 }
 916                 ertf_parser_destroy(parser);
 917                 conv_code_converter_destroy(conv);
 918         }
 919
 920         if (conv_fail)
 921                 g_warning(_("procmime_get_text_content(): Code conversion failed.\n"));
 922
 923         fclose(tmpfp);
 924         rewind(outfp);
 925
 926         return outfp;
 927 }
 928
 929 /* search the first text part of (multipart) MIME message,
 930    decode, convert it and output to outfp. */
 931 FILE *procmime_get_first_text_content(MsgInfo *msginfo)
 932 {
 933         FILE *infp, *outfp = NULL;
 934         MimeInfo *mimeinfo, *partinfo;
 935
 936         g_return_val_if_fail(msginfo != NULL, NULL);
 937
 938         mimeinfo = procmime_scan_message(msginfo);
 939         if (!mimeinfo) return NULL;
 940
 941         if ((infp = procmsg_open_message(msginfo)) == NULL) {
 942                 procmime_mimeinfo_free_all(mimeinfo);
 943                 return NULL;
 944         }
 945
 946         partinfo = mimeinfo;
 947         while (partinfo && partinfo->mime_type != MIME_TEXT)
 948                 partinfo = procmime_mimeinfo_next(partinfo);
 949         if (!partinfo) {
 950                 partinfo = mimeinfo;
 951                 while (partinfo && partinfo->mime_type != MIME_TEXT_HTML &&
 952                                 partinfo->mime_type != MIME_TEXT_ENRICHED)
 953                         partinfo = procmime_mimeinfo_next(partinfo);
 954         }
 955
 956
 957         if (partinfo)
 958                 outfp = procmime_get_text_content(partinfo, infp);
 959
 960         fclose(infp);
 961         procmime_mimeinfo_free_all(mimeinfo);
 962
 963         return outfp;
 964 }
 965
 966 gboolean procmime_find_string_part(MimeInfo *mimeinfo, const gchar *filename,
 967                                    const gchar *str, gboolean case_sens)
 968 {
 969
 970         FILE *infp, *outfp;
 971         gchar buf[BUFFSIZE];
 972         gchar *(* StrFindFunc) (const gchar *haystack, const gchar *needle);
 973
 974         g_return_val_if_fail(mimeinfo != NULL, FALSE);
 975         g_return_val_if_fail(mimeinfo->mime_type == MIME_TEXT ||
 976                              mimeinfo->mime_type == MIME_TEXT_HTML ||
 977                              mimeinfo->mime_type == MIME_TEXT_ENRICHED, FALSE);
 978         g_return_val_if_fail(str != NULL, FALSE);
 979
 980         if ((infp = fopen(filename, "rb")) == NULL) {
 981                 FILE_OP_ERROR(filename, "fopen");
 982                 return FALSE;
 983         }
 984
 985         outfp = procmime_get_text_content(mimeinfo, infp);
 986         fclose(infp);
 987
 988         if (!outfp)
 989                 return FALSE;
 990
 991         if (case_sens)
 992                 StrFindFunc = strstr;
 993         else
 994                 StrFindFunc = strcasestr;
 995
 996         while (fgets(buf, sizeof(buf), outfp) != NULL) {
 997                 if (StrFindFunc(buf, str) != NULL) {
 998                         fclose(outfp);
 999                         return TRUE;
1000                 }
1001         }
1002
1003         fclose(outfp);
1004
1005         return FALSE;
1006 }
1007
1008 gboolean procmime_find_string(MsgInfo *msginfo, const gchar *str,
1009                               gboolean case_sens)
1010 {
1011         MimeInfo *mimeinfo;
1012         MimeInfo *partinfo;
1013         gchar *filename;
1014         gboolean found = FALSE;
1015
1016         g_return_val_if_fail(msginfo != NULL, FALSE);
1017         g_return_val_if_fail(str != NULL, FALSE);
1018
1019         filename = procmsg_get_message_file(msginfo);
1020         if (!filename) return FALSE;
1021         mimeinfo = procmime_scan_message(msginfo);
1022
1023         for (partinfo = mimeinfo; partinfo != NULL;
1024              partinfo = procmime_mimeinfo_next(partinfo)) {
1025                 if (partinfo->mime_type == MIME_TEXT ||
1026                     partinfo->mime_type == MIME_TEXT_HTML ||
1027                     partinfo->mime_type == MIME_TEXT_ENRICHED) {
1028                         if (procmime_find_string_part
1029                                 (partinfo, filename, str, case_sens) == TRUE) {
1030                                 found = TRUE;
1031                                 break;
1032                         }
1033                 }
1034         }
1035
1036         procmime_mimeinfo_free_all(mimeinfo);
1037         g_free(filename);
1038
1039         return found;
1040 }
1041
1042 gchar *procmime_get_tmp_file_name(MimeInfo *mimeinfo)
1043 {
1044         static guint32 id = 0;
1045         gchar *base;
1046         gchar *filename;
1047         gchar f_prefix[10];
1048
1049         g_return_val_if_fail(mimeinfo != NULL, NULL);
1050
1051         g_snprintf(f_prefix, sizeof(f_prefix), "%08x.", id++);
1052
1053         if (MIME_TEXT_HTML == mimeinfo->mime_type)
1054                 base = "mimetmp.html";
1055         else {
1056                 base = mimeinfo->filename ? mimeinfo->filename
1057                         : mimeinfo->name ? mimeinfo->name : "mimetmp";
1058                 base = g_basename(base);
1059                 if (*base == '\0') base = "mimetmp";
1060                 Xstrdup_a(base, base, return NULL);
1061                 subst_for_filename(base);
1062         }
1063
1064         filename = g_strconcat(get_mime_tmp_dir(), G_DIR_SEPARATOR_S,
1065                                f_prefix, base, NULL);
1066
1067         return filename;
1068 }
1069
1070 ContentType procmime_scan_mime_type(const gchar *mime_type)
1071 {
1072         ContentType type;
1073
1074         if (!strncasecmp(mime_type, "text/html", 9))
1075                 type = MIME_TEXT_HTML;
1076         else if (!strncasecmp(mime_type, "text/enriched", 13))
1077                 type = MIME_TEXT_ENRICHED;
1078         else if (!strncasecmp(mime_type, "text/", 5))
1079                 type = MIME_TEXT;
1080         else if (!strncasecmp(mime_type, "message/rfc822", 14))
1081                 type = MIME_MESSAGE_RFC822;
1082         else if (!strncasecmp(mime_type, "message/", 8))
1083                 type = MIME_TEXT;
1084         else if (!strncasecmp(mime_type, "application/octet-stream", 24))
1085                 type = MIME_APPLICATION_OCTET_STREAM;
1086         else if (!strncasecmp(mime_type, "application/", 12))
1087                 type = MIME_APPLICATION;
1088         else if (!strncasecmp(mime_type, "multipart/", 10))
1089                 type = MIME_MULTIPART;
1090         else if (!strncasecmp(mime_type, "image/", 6))
1091                 type = MIME_IMAGE;
1092         else if (!strncasecmp(mime_type, "audio/", 6))
1093                 type = MIME_AUDIO;
1094         else if (!strcasecmp(mime_type, "text"))
1095                 type = MIME_TEXT;
1096         else
1097                 type = MIME_UNKNOWN;
1098
1099         return type;
1100 }
1101
1102 static GList *mime_type_list = NULL;
1103
1104 gchar *procmime_get_mime_type(const gchar *filename)
1105 {
1106         static GHashTable *mime_type_table = NULL;
1107         MimeType *mime_type;
1108         const gchar *p;
1109         gchar *ext;
1110
1111         if (!mime_type_table) {
1112                 mime_type_table = procmime_get_mime_type_table();
1113                 if (!mime_type_table) return NULL;
1114         }
1115
1116         filename = g_basename(filename);
1117         p = strrchr(filename, '.');
1118         if (!p) return NULL;
1119
1120         Xstrdup_a(ext, p + 1, return NULL);
1121         g_strdown(ext);
1122         mime_type = g_hash_table_lookup(mime_type_table, ext);
1123         if (mime_type) {
1124                 gchar *str;
1125
1126                 str = g_strconcat(mime_type->type, "/", mime_type->sub_type,
1127                                   NULL);
1128                 return str;
1129         }
1130
1131         return NULL;
1132 }
1133
1134 static guint procmime_str_hash(gconstpointer gptr)
1135 {
1136         guint hash_result = 0;
1137         const char *str;
1138
1139         for (str = gptr; str && *str; str++) {
1140                 if (isupper(*str)) hash_result += (*str + ' ');
1141                 else hash_result += *str;
1142         }
1143
1144         return hash_result;
1145 }
1146
1147 static gint procmime_str_equal(gconstpointer gptr1, gconstpointer gptr2)
1148 {
1149         const char *str1 = gptr1;
1150         const char *str2 = gptr2;
1151
1152         return !strcasecmp(str1, str2);
1153 }
1154
1155 static GHashTable *procmime_get_mime_type_table(void)
1156 {
1157         GHashTable *table = NULL;
1158         GList *cur;
1159         MimeType *mime_type;
1160         gchar **exts;
1161
1162         if (!mime_type_list) {
1163                 mime_type_list = procmime_get_mime_type_list();
1164                 if (!mime_type_list) return NULL;
1165         }
1166
1167         table = g_hash_table_new(procmime_str_hash, procmime_str_equal);
1168
1169         for (cur = mime_type_list; cur != NULL; cur = cur->next) {
1170                 gint i;
1171                 gchar *key;
1172
1173                 mime_type = (MimeType *)cur->data;
1174
1175                 if (!mime_type->extension) continue;
1176
1177                 exts = g_strsplit(mime_type->extension, " ", 16);
1178                 for (i = 0; exts[i] != NULL; i++) {
1179                         /* make the key case insensitive */
1180                         g_strdown(exts[i]);
1181                         /* use previously dup'd key on overwriting */
1182                         if (g_hash_table_lookup(table, exts[i]))
1183                                 key = exts[i];
1184                         else
1185                                 key = g_strdup(exts[i]);
1186                         g_hash_table_insert(table, key, mime_type);
1187                 }
1188                 g_strfreev(exts);
1189         }
1190
1191         return table;
1192 }
1193
1194 GList *procmime_get_mime_type_list(void)
1195 {
1196         GList *list = NULL;
1197         FILE *fp;
1198         gchar buf[BUFFSIZE];
1199         gchar *p, *delim;
1200         MimeType *mime_type;
1201
1202         if (mime_type_list)
1203                 return mime_type_list;
1204
1205         if ((fp = fopen("/etc/mime.types", "rb")) == NULL) {
1206                 if ((fp = fopen(SYSCONFDIR "/mime.types", "rb")) == NULL) {
1207                         FILE_OP_ERROR(SYSCONFDIR "/mime.types", "fopen");
1208                         return NULL;
1209                 }
1210         }
1211
1212         while (fgets(buf, sizeof(buf), fp) != NULL) {
1213                 p = strchr(buf, '#');
1214                 if (p) *p = '\0';
1215                 g_strstrip(buf);
1216
1217                 p = buf;
1218                 while (*p && !isspace(*p)) p++;
1219                 if (*p) {
1220                         *p = '\0';
1221                         p++;
1222                 }
1223                 delim = strchr(buf, '/');
1224                 if (delim == NULL) continue;
1225                 *delim = '\0';
1226
1227                 mime_type = g_new(MimeType, 1);
1228                 mime_type->type = g_strdup(buf);
1229                 mime_type->sub_type = g_strdup(delim + 1);
1230
1231                 while (*p && isspace(*p)) p++;
1232                 if (*p)
1233                         mime_type->extension = g_strdup(p);
1234                 else
1235                         mime_type->extension = NULL;
1236
1237                 list = g_list_append(list, mime_type);
1238         }
1239
1240         fclose(fp);
1241
1242         if (!list)
1243                 g_warning("Can't read mime.types\n");
1244
1245         return list;
1246 }
1247
1248 EncodingType procmime_get_encoding_for_charset(const gchar *charset)
1249 {
1250         if (!charset)
1251                 return ENC_8BIT;
1252         else if (!strncasecmp(charset, "ISO-2022-", 9) ||
1253                  !strcasecmp(charset, "US-ASCII"))
1254                 return ENC_7BIT;
1255         else
1256                 return ENC_8BIT;
1257                 /* return ENC_BASE64; */
1258                 /* return ENC_QUOTED_PRINTABLE; */
1259 }
1260
1261 EncodingType procmime_get_encoding_for_file(const gchar *file)
1262 {
1263         FILE *fp;
1264         guchar buf[BUFSIZ];
1265         size_t len;
1266
1267         if ((fp = fopen(file, "rb")) == NULL) {
1268                 FILE_OP_ERROR(file, "fopen");
1269                 return ENC_UNKNOWN;
1270         }
1271
1272         while ((len = fread(buf, sizeof(gchar), sizeof(buf), fp)) > 0) {
1273                 guchar *p;
1274                 gint i;
1275
1276                 for (p = buf, i = 0; i < len; p++, i++) {
1277                         if (*p & 0x80) {
1278                                 fclose(fp);
1279                                 return ENC_BASE64;
1280                         }
1281                 }
1282         }
1283
1284         fclose(fp);
1285         return ENC_7BIT;
1286 }
1287
1288 const gchar *procmime_get_encoding_str(EncodingType encoding)
1289 {
1290         static const gchar *encoding_str[] = {
1291                 "7bit", "8bit", "quoted-printable", "base64", "x-uuencode",
1292                 NULL
1293         };
1294
1295         if (encoding >= ENC_7BIT && encoding <= ENC_UNKNOWN)
1296                 return encoding_str[encoding];
1297         else
1298                 return NULL;
1299 }