src/procmime.c

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