2005-07-01 [colin] 1.9.12cvs8
[claws.git] / src / procmime.c
index da754128b24d96fb1f8e9176fa23240020437993..3a40553d293f5cda1908f3cdc80456982a6470cd 100644 (file)
@@ -25,6 +25,7 @@
 
 #include <stdio.h>
 #include <glib.h>
+#include <glib/gi18n.h>
 #include <stdio.h>
 #include <string.h>
 #include <locale.h>
@@ -33,7 +34,6 @@
 #include <sys/stat.h>
 #include <unistd.h>
 
-#include "intl.h"
 #include "procmime.h"
 #include "procheader.h"
 #include "base64.h"
@@ -308,12 +308,13 @@ gboolean procmime_decode_content(MimeInfo *mimeinfo)
                while ((ftell(infp) < readend) && (fgets(buf, sizeof(buf), infp) != NULL)) {
                        gint len;
                        len = qp_decode_line(buf);
-                       fwrite(buf, len, 1, outfp);
+                       fwrite(buf, 1, len, outfp);
                }
        } else if (encoding == ENC_BASE64) {
                gchar outbuf[BUFFSIZE];
                gint len;
                Base64Decoder *decoder;
+               gboolean got_error = FALSE;
                gboolean uncanonicalize = FALSE;
                FILE *tmpfp = outfp;
 
@@ -331,11 +332,20 @@ gboolean procmime_decode_content(MimeInfo *mimeinfo)
                decoder = base64_decoder_new();
                while ((ftell(infp) < readend) && (fgets(buf, sizeof(buf), infp) != NULL)) {
                        len = base64_decoder_decode(decoder, buf, outbuf);
-                       if (len < 0) {
-                               g_warning("Bad BASE64 content\n");
-                               break;
+                       if (len < 0 && !got_error) {
+                               g_warning("Bad BASE64 content.\n");
+                               fwrite(_("[Error decoding BASE64]\n"),
+                                       sizeof(gchar),
+                                       strlen(_("[Error decoding BASE64]\n")),
+                                       tmpfp);
+                               got_error = TRUE;
+                               continue;
+                       } else if (len >= 0) {
+                               /* print out the error message only once 
+                                * per block */
+                               fwrite(outbuf, sizeof(gchar), len, tmpfp);
+                               got_error = FALSE;
                        }
-                       fwrite(outbuf, sizeof(gchar), len, tmpfp);
                }
                base64_decoder_free(decoder);
 
@@ -394,7 +404,7 @@ gboolean procmime_decode_content(MimeInfo *mimeinfo)
 
 gboolean procmime_encode_content(MimeInfo *mimeinfo, EncodingType encoding)
 {
-       FILE *infp, *outfp;
+       FILE *infp = NULL, *outfp;
        gint len;
        gchar *tmpfilename;
        struct stat statbuf;
@@ -448,6 +458,7 @@ gboolean procmime_encode_content(MimeInfo *mimeinfo, EncodingType encoding)
                                gchar *out = canonicalize_str(mimeinfo->data.mem);
                                fclose(infp);
                                infp = str_open_as_stream(out);
+                               tmp_fp = infp;
                                g_free(out);
                                if (infp == NULL)
                                        return FALSE;
@@ -477,7 +488,16 @@ gboolean procmime_encode_content(MimeInfo *mimeinfo, EncodingType encoding)
 
                while (fgets(inbuf, sizeof(inbuf), infp) != NULL) {
                        qp_encode_line(outbuf, inbuf);
-                       fputs(outbuf, outfp);
+
+                       if (!strncmp("From ", outbuf, sizeof("From ")-1)) {
+                               gchar *tmpbuf = outbuf;
+                               
+                               tmpbuf += sizeof("From ")-1;
+                               
+                               fputs("=46rom ", outfp);
+                               fputs(tmpbuf, outfp);
+                       } else 
+                               fputs(outbuf, outfp);
                }
        } else {
                gchar buf[BUFFSIZE];
@@ -726,7 +746,6 @@ FILE *procmime_get_text_content(MimeInfo *mimeinfo)
                }
                
                dup2(oldout, 1);
-#warning FIXME_GTK2 HTML/RTF not yet utf8
 /* CodeConverter seems to have no effect here */
        } else if (mimeinfo->type == MIMETYPE_TEXT && !g_ascii_strcasecmp(mimeinfo->subtype, "html")) {
                HTMLParser *parser;
@@ -787,8 +806,65 @@ FILE *procmime_get_first_text_content(MsgInfo *msginfo)
        if (!mimeinfo) return NULL;
 
        partinfo = mimeinfo;
-       while (partinfo && partinfo->type != MIMETYPE_TEXT)
+       while (partinfo && partinfo->type != MIMETYPE_TEXT) {
+               partinfo = procmime_mimeinfo_next(partinfo);
+       }
+       if (partinfo)
+               outfp = procmime_get_text_content(partinfo);
+
+       procmime_mimeinfo_free_all(mimeinfo);
+
+       return outfp;
+}
+
+
+static gboolean find_encrypted_func(GNode *node, gpointer data)
+{
+       MimeInfo *mimeinfo = (MimeInfo *) node->data;
+       MimeInfo **encinfo = (MimeInfo **) data;
+       
+       if (privacy_mimeinfo_is_encrypted(mimeinfo)) {
+               *encinfo = mimeinfo;
+               return TRUE;
+       }
+       
+       return FALSE;
+}
+
+static MimeInfo *find_encrypted_part(MimeInfo *rootinfo)
+{
+       MimeInfo *encinfo = NULL;
+
+       g_node_traverse(rootinfo->node, G_IN_ORDER, G_TRAVERSE_ALL, -1,
+               find_encrypted_func, &encinfo);
+       
+       return encinfo;
+}
+
+/* search the first encrypted text part of (multipart) MIME message,
+   decode, convert it and output to outfp. */
+FILE *procmime_get_first_encrypted_text_content(MsgInfo *msginfo)
+{
+       FILE *outfp = NULL;
+       MimeInfo *mimeinfo, *partinfo, *encinfo;
+
+       g_return_val_if_fail(msginfo != NULL, NULL);
+
+       mimeinfo = procmime_scan_message(msginfo);
+       if (!mimeinfo) {
+               return NULL;
+       }
+
+       partinfo = mimeinfo;
+       if ((encinfo = find_encrypted_part(partinfo)) != NULL) {
+               debug_print("decrypting message part\n");
+               if (privacy_mimeinfo_decrypt(encinfo) < 0)
+                       return NULL;
+       }
+       partinfo = mimeinfo;
+       while (partinfo && partinfo->type != MIMETYPE_TEXT) {
                partinfo = procmime_mimeinfo_next(partinfo);
+       }
 
        if (partinfo)
                outfp = procmime_get_text_content(partinfo);
@@ -798,6 +874,25 @@ FILE *procmime_get_first_text_content(MsgInfo *msginfo)
        return outfp;
 }
 
+gboolean procmime_msginfo_is_encrypted(MsgInfo *msginfo)
+{
+       MimeInfo *mimeinfo, *partinfo;
+       gboolean result = FALSE;
+
+       g_return_val_if_fail(msginfo != NULL, FALSE);
+
+       mimeinfo = procmime_scan_message(msginfo);
+       if (!mimeinfo) {
+               return FALSE;
+       }
+
+       partinfo = mimeinfo;
+       result = (find_encrypted_part(partinfo) != NULL);
+       procmime_mimeinfo_free_all(mimeinfo);
+
+       return result;
+}
+
 gboolean procmime_find_string_part(MimeInfo *mimeinfo, const gchar *filename,
                                   const gchar *str, StrFindFunc find_func)
 {
@@ -881,8 +976,9 @@ gchar *procmime_get_tmp_file_name(MimeInfo *mimeinfo)
                        basetmp = procmime_mimeinfo_get_parameter(mimeinfo, "name");
                if (basetmp == NULL)
                        basetmp = "mimetmp";
-               base = g_path_get_basename(basetmp);
-               if (*base == '\0') base = g_strdup("mimetmp");
+               basetmp = g_path_get_basename(basetmp);
+               if (*basetmp == '\0') basetmp = g_strdup("mimetmp");
+               base = conv_filename_from_utf8(basetmp);
                subst_for_shellsafe_filename(base);
        }
 
@@ -890,6 +986,7 @@ gchar *procmime_get_tmp_file_name(MimeInfo *mimeinfo)
                               f_prefix, base, NULL);
 
        g_free(base);
+       
        return filename;
 }
 
@@ -900,7 +997,7 @@ gchar *procmime_get_mime_type(const gchar *filename)
        static GHashTable *mime_type_table = NULL;
        MimeType *mime_type;
        const gchar *p;
-       gchar *ext;
+       gchar *ext = NULL;
        gchar *base;
 
        if (!mime_type_table) {
@@ -909,11 +1006,11 @@ gchar *procmime_get_mime_type(const gchar *filename)
        }
 
        base = g_path_get_basename(filename);
-       p = strrchr(base, '.');
+       if ((p = strrchr(base, '.')) != NULL)
+               Xstrdup_a(ext, p + 1, p = NULL );
        g_free(base);
        if (!p) return NULL;
 
-       Xstrdup_a(ext, p + 1, return NULL);
        g_strdown(ext);
        mime_type = g_hash_table_lookup(mime_type_table, ext);
        if (mime_type) {
@@ -995,14 +1092,20 @@ GList *procmime_get_mime_type_list(void)
        guchar *p;
        gchar *delim;
        MimeType *mime_type;
+       gboolean fp_is_glob_file = TRUE;
 
        if (mime_type_list) 
                return mime_type_list;
-
-       if ((fp = fopen("/etc/mime.types", "rb")) == NULL) {
-               if ((fp = fopen(SYSCONFDIR "/mime.types", "rb")) == NULL) {
-                       FILE_OP_ERROR(SYSCONFDIR "/mime.types", "fopen");
-                       return NULL;
+       
+       if ((fp = fopen("/usr/share/mime/globs", "rb")) == NULL) {
+               fp_is_glob_file = FALSE;
+               if ((fp = fopen("/etc/mime.types", "rb")) == NULL) {
+                       if ((fp = fopen(SYSCONFDIR "/mime.types", "rb")) 
+                               == NULL) {
+                               FILE_OP_ERROR(SYSCONFDIR "/mime.types", 
+                                       "fopen");
+                               return NULL;
+                       }
                }
        }
 
@@ -1012,7 +1115,13 @@ GList *procmime_get_mime_type_list(void)
                g_strstrip(buf);
 
                p = buf;
-               while (*p && !isspace(*p)) p++;
+               
+               if (fp_is_glob_file) {
+                       while (*p && !isspace(*p) && (*p!=':')) p++;
+               } else {
+                       while (*p && !isspace(*p)) p++;
+               }
+
                if (*p) {
                        *p = '\0';
                        p++;
@@ -1025,7 +1134,12 @@ GList *procmime_get_mime_type_list(void)
                mime_type->type = g_strdup(buf);
                mime_type->sub_type = g_strdup(delim + 1);
 
-               while (*p && isspace(*p)) p++;
+               if (fp_is_glob_file) {
+                       while (*p && (isspace(*p)||(*p=='*')||(*p=='.'))) p++;
+               } else {
+                       while (*p && isspace(*p)) p++;
+               }
+
                if (*p)
                        mime_type->extension = g_strdup(p);
                else
@@ -1189,7 +1303,7 @@ gchar *procmime_get_content_type_str(MimeMediaType type,
        return g_strdup_printf("%s/%s", type_str, subtype);
 }
 
-void procmime_parse_mimepart(MimeInfo *parent,
+int procmime_parse_mimepart(MimeInfo *parent,
                             gchar *content_type,
                             gchar *content_encoding,
                             gchar *content_description,
@@ -1215,7 +1329,7 @@ void procmime_parse_message_rfc822(MimeInfo *mimeinfo)
                                {NULL,             NULL, FALSE}};
        guint content_start, i;
        FILE *fp;
-       gint mime_major, mime_minor;
+        gchar *tmp;
 
        procmime_decode_content(mimeinfo);
 
@@ -1226,38 +1340,31 @@ void procmime_parse_message_rfc822(MimeInfo *mimeinfo)
        }
        fseek(fp, mimeinfo->offset, SEEK_SET);
        procheader_get_header_fields(fp, hentry);
-       if (hentry[0].body != NULL)
-               conv_unmime_header_overwrite(hentry[0].body);
-       if (hentry[2].body != NULL)
-               conv_unmime_header_overwrite(hentry[2].body);
-       if (hentry[4].body != NULL)
-               conv_unmime_header_overwrite(hentry[4].body);
+       if (hentry[0].body != NULL) {
+                tmp = conv_unmime_header(hentry[0].body, NULL);
+                g_free(hentry[0].body);
+                hentry[0].body = tmp;
+        }                
+       if (hentry[2].body != NULL) {
+                tmp = conv_unmime_header(hentry[2].body, NULL);
+                g_free(hentry[2].body);
+                hentry[2].body = tmp;
+        }                
+       if (hentry[4].body != NULL) {
+                tmp = conv_unmime_header(hentry[4].body, NULL);
+                g_free(hentry[4].body);
+                hentry[4].body = tmp;
+        }                
        content_start = ftell(fp);
        fclose(fp);
-
-       if ((hentry[5].body != NULL) &&
-           (sscanf(hentry[5].body, "%d.%d", &mime_major, &mime_minor) == 2) &&
-           (mime_major == 1) && (mime_minor == 0)) {
-               procmime_parse_mimepart(mimeinfo,
-                                       hentry[0].body, hentry[1].body,
-                                       hentry[2].body, hentry[3].body, 
-                                       hentry[4].body, 
-                                       mimeinfo->data.filename, content_start,
-                                       mimeinfo->length - (content_start - mimeinfo->offset));
-       } else {
-               MimeInfo *subinfo;
-
-               subinfo = procmime_mimeinfo_new();
-               subinfo->content = MIMECONTENT_FILE;
-               subinfo->encoding_type = ENC_UNKNOWN;
-               subinfo->type = MIMETYPE_TEXT;
-               subinfo->subtype = g_strdup("plain");
-               subinfo->data.filename = g_strdup(mimeinfo->data.filename);
-               subinfo->offset = content_start;
-               subinfo->length = mimeinfo->length - (content_start - mimeinfo->offset);
-
-               g_node_append(mimeinfo->node, subinfo->node);
-       }
+       
+       procmime_parse_mimepart(mimeinfo,
+                               hentry[0].body, hentry[1].body,
+                               hentry[2].body, hentry[3].body,
+                               hentry[4].body,
+                               mimeinfo->data.filename, content_start,
+                               mimeinfo->length - (content_start - mimeinfo->offset));
+       
        for (i = 0; i < (sizeof hentry / sizeof hentry[0]); i++) {
                g_free(hentry[i].body);
                hentry[i].body = NULL;
@@ -1276,11 +1383,12 @@ void procmime_parse_multipart(MimeInfo *mimeinfo)
                                {"Content-Disposition:",
                                                   NULL, TRUE},
                                {NULL,             NULL, FALSE}};
-       gchar *p;
+       gchar *p, *tmp;
        gchar *boundary;
        gint boundary_len = 0, lastoffset = -1, i;
        gchar buf[BUFFSIZE];
        FILE *fp;
+       int result = 0;
 
        boundary = g_hash_table_lookup(mimeinfo->typeparameters, "boundary");
        if (!boundary)
@@ -1295,13 +1403,13 @@ void procmime_parse_multipart(MimeInfo *mimeinfo)
                return;
        }
        fseek(fp, mimeinfo->offset, SEEK_SET);
-       while ((p = fgets(buf, sizeof(buf), fp)) != NULL) {
+       while ((p = fgets(buf, sizeof(buf), fp)) != NULL && result == 0) {
                if (ftell(fp) > (mimeinfo->offset + mimeinfo->length))
                        break;
 
                if (IS_BOUNDARY(buf, boundary, boundary_len)) {
                        if (lastoffset != -1) {
-                               procmime_parse_mimepart(mimeinfo,
+                               result = procmime_parse_mimepart(mimeinfo,
                                                        hentry[0].body, hentry[1].body,
                                                        hentry[2].body, hentry[3].body, 
                                                        hentry[4].body, 
@@ -1318,12 +1426,21 @@ void procmime_parse_multipart(MimeInfo *mimeinfo)
                                hentry[i].body = NULL;
                        }
                        procheader_get_header_fields(fp, hentry);
-                       if (hentry[0].body != NULL)
-                               conv_unmime_header_overwrite(hentry[0].body);
-                       if (hentry[2].body != NULL)
-                               conv_unmime_header_overwrite(hentry[2].body);
-                       if (hentry[4].body != NULL)
-                               conv_unmime_header_overwrite(hentry[4].body);
+                        if (hentry[0].body != NULL) {
+                                tmp = conv_unmime_header(hentry[0].body, NULL);
+                                g_free(hentry[0].body);
+                                hentry[0].body = tmp;
+                        }                
+                        if (hentry[2].body != NULL) {
+                                tmp = conv_unmime_header(hentry[2].body, NULL);
+                                g_free(hentry[2].body);
+                                hentry[2].body = tmp;
+                        }                
+                        if (hentry[4].body != NULL) {
+                                tmp = conv_unmime_header(hentry[4].body, NULL);
+                                g_free(hentry[4].body);
+                                hentry[4].body = tmp;
+                        }                
                        lastoffset = ftell(fp);
                }
        }
@@ -1559,7 +1676,7 @@ static void procmime_parse_content_encoding(const gchar *content_encoding, MimeI
        return;
 }
 
-void procmime_parse_mimepart(MimeInfo *parent,
+int procmime_parse_mimepart(MimeInfo *parent,
                             gchar *content_type,
                             gchar *content_encoding,
                             gchar *content_description,
@@ -1574,8 +1691,17 @@ void procmime_parse_mimepart(MimeInfo *parent,
        /* Create MimeInfo */
        mimeinfo = procmime_mimeinfo_new();
        mimeinfo->content = MIMECONTENT_FILE;
-       if (parent != NULL)
+       if (parent != NULL) {
+               if (g_node_depth(parent->node) > 32) {
+                       /* 32 is an arbitrary value
+                        * this avoids DOSsing ourselves 
+                        * with enormous messages
+                        */
+                       procmime_mimeinfo_free_all(mimeinfo);
+                       return -1;                      
+               }
                g_node_append(parent->node, mimeinfo->node);
+       }
        mimeinfo->data.filename = g_strdup(filename);
        mimeinfo->offset = offset;
        mimeinfo->length = length;
@@ -1626,6 +1752,8 @@ void procmime_parse_mimepart(MimeInfo *parent,
                default:
                        break;
        }
+
+       return 0;
 }
 
 static gchar *typenames[] = {
@@ -1779,8 +1907,12 @@ static void write_parameters(gpointer key, gpointer value, gpointer user_data)
                break;
 
        case ENC_AS_EXTENDED:
-               g_string_append_printf(buf, "%s*=%s''", param,
-                       conv_get_locale_charset_str());
+               if (!g_utf8_validate(val, -1, NULL))
+                       g_string_append_printf(buf, "%s*=%s''", param,
+                               conv_get_locale_charset_str());
+               else
+                       g_string_append_printf(buf, "%s*=%s''", param,
+                               CS_INTERNAL);
                for (valpos = val; *valpos != '\0'; valpos++) {
                        if (IS_ASCII(*valpos) && isalnum(*valpos)) {
                                g_string_append_printf(buf, "%c", *valpos);
@@ -1898,7 +2030,10 @@ gint procmime_write_message_rfc822(MimeInfo *mimeinfo, FILE *fp)
                break;
 
        case MIMECONTENT_MEM:
-               fwrite(mimeinfo->data.mem, strlen(mimeinfo->data.mem), sizeof(gchar), fp);
+               fwrite(mimeinfo->data.mem, 
+                               sizeof(gchar), 
+                               strlen(mimeinfo->data.mem), 
+                               fp);
                break;
 
        default:
@@ -1947,7 +2082,7 @@ gint procmime_write_multipart(MimeInfo *mimeinfo, FILE *fp)
                if (((str2 = strstr(str, boundary)) != NULL) && ((str2 - str) >= 2) &&
                    (*(str2 - 1) == '-') && (*(str2 - 2) == '-'))
                        *(str2 - 2) = '\0';
-               fwrite(str, strlen(str), sizeof(gchar), fp);
+               fwrite(str, sizeof(gchar), strlen(str), fp);
                g_free(str);
                break;
 
@@ -1995,7 +2130,10 @@ gint procmime_write_mimeinfo(MimeInfo *mimeinfo, FILE *fp)
                        return 0;
 
                case MIMECONTENT_MEM:
-                       fwrite(mimeinfo->data.mem, strlen(mimeinfo->data.mem), sizeof(gchar), fp);
+                       fwrite(mimeinfo->data.mem, 
+                                       sizeof(gchar), 
+                                       strlen(mimeinfo->data.mem), 
+                                       fp);
                        return 0;
 
                default:
@@ -2021,3 +2159,32 @@ gint procmime_write_mimeinfo(MimeInfo *mimeinfo, FILE *fp)
 
        return 0;
 }
+
+gchar *procmime_get_part_file_name(MimeInfo *mimeinfo)
+{
+       gchar *base;
+
+       if ((mimeinfo->type == MIMETYPE_TEXT) && !g_ascii_strcasecmp(mimeinfo->subtype, "html"))
+               base = g_strdup("mimetmp.html");
+       else {
+               const gchar *basetmp;
+               gchar *basename;
+
+               basetmp = procmime_mimeinfo_get_parameter(mimeinfo, "filename");
+               if (basetmp == NULL)
+                       basetmp = procmime_mimeinfo_get_parameter(mimeinfo, "name");
+               if (basetmp == NULL)
+                       basetmp = "mimetmp";
+               basename = g_path_get_basename(basetmp);
+               if (*basename == '\0') {
+                       g_free(basename);
+                       basename = g_strdup("mimetmp");
+               }
+               base = conv_filename_from_utf8(basename);
+               g_free(basename);
+               subst_for_shellsafe_filename(base);
+       }
+       
+       return base;
+}
+