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