Fixes several potential infinite loops with corrupted account data
[modest] / src / modest-tny-msg.c
1 /* Copyright (c) 2006, Nokia Corporation
2  * All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are
6  * met:
7  *
8  * * Redistributions of source code must retain the above copyright
9  *   notice, this list of conditions and the following disclaimer.
10  * * Redistributions in binary form must reproduce the above copyright
11  *   notice, this list of conditions and the following disclaimer in the
12  *   documentation and/or other materials provided with the distribution.
13  * * Neither the name of the Nokia Corporation nor the names of its
14  *   contributors may be used to endorse or promote products derived from
15  *   this software without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
18  * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
19  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
20  * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
21  * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
22  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
23  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
24  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
25  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
26  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
27  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28  */
29
30 #include <string.h>
31 #include <gtkhtml/gtkhtml.h>
32 #include <tny-gtk-text-buffer-stream.h>
33 #include <tny-simple-list.h>
34 #include <tny-folder.h>
35 #include <modest-runtime.h>
36 #include "modest-formatter.h"
37 #include <tny-camel-mem-stream.h>
38 #include <tny-camel-mime-part.h>
39 #include <glib/gprintf.h>
40 #include <modest-tny-folder.h>
41 #include "modest-tny-mime-part.h"
42 #include <modest-error.h>
43
44
45 #ifdef HAVE_CONFIG_H
46 #include <config.h>
47 #endif /*HAVE_CONFIG_H */
48
49 #include <modest-tny-msg.h>
50 #include "modest-text-utils.h"
51
52 static TnyMimePart * add_body_part (TnyMsg *msg, const gchar *body,
53                                     const gchar *content_type);
54 static TnyMimePart * add_html_body_part (TnyMsg *msg, const gchar *body);
55 static gint add_attachments (TnyMimePart *part, GList *attachments_list, gboolean add_inline, GError **err);
56 static void add_images (TnyMsg *msg, GList *attachments_list, GError **err);
57 static char * get_content_type(const gchar *s);
58 static gboolean is_ascii(const gchar *s);
59
60
61 TnyMsg*
62 modest_tny_msg_new (const gchar* mailto, const gchar* from, const gchar *cc,
63                     const gchar *bcc, const gchar* subject, const gchar *body,
64                     GList *attachments, gint *attached, GError **err)
65 {
66         TnyMsg *new_msg;
67         TnyHeader *header;
68         gchar *content_type;
69         gint tmp_attached = 0;
70         
71         /* Create new msg */
72         new_msg = modest_formatter_create_message (NULL, TRUE, (attachments != NULL), FALSE);
73         header  = tny_msg_get_header (new_msg);
74         
75         if ((from != NULL) && (strlen(from) > 0)) {
76                 tny_header_set_from (TNY_HEADER (header), from);
77                 tny_header_set_replyto (TNY_HEADER (header), from);
78         }
79         if ((mailto != NULL) && (strlen(mailto) > 0)) {
80                 gchar *removed_to = modest_text_utils_remove_duplicate_addresses (mailto);
81                 tny_header_set_to (TNY_HEADER (header), removed_to);
82                 g_free (removed_to);
83         }
84         if ((cc != NULL) && (strlen(cc) > 0)) 
85                 tny_header_set_cc (TNY_HEADER (header), cc);
86         if ((bcc != NULL) && (strlen(bcc) > 0)) 
87                 tny_header_set_bcc (TNY_HEADER (header), bcc);
88         
89         if ((subject != NULL) && (strlen(subject) > 0)) 
90                 tny_header_set_subject (TNY_HEADER (header), subject);
91
92         content_type = get_content_type(body);
93
94         /* set modest as the X-Mailer
95          * we could this in the platform factory, but then the header
96          * would show up before all the others.
97          */
98         tny_mime_part_set_header_pair (TNY_MIME_PART (new_msg), "X-Mailer", "Modest "
99                                        VERSION);
100         
101         /* Add the body of the new mail */
102         /* This is needed even if body is NULL or empty. */
103         add_body_part (new_msg, body, content_type);
104         g_free (content_type);
105                        
106         /* Add attachments */
107         if (attachments)
108                 tmp_attached = add_attachments (TNY_MIME_PART (new_msg), attachments, FALSE, err);
109         if (attached)
110                 *attached = tmp_attached;
111         if (header)
112                 g_object_unref(header);
113
114         return new_msg;
115 }
116
117 TnyMsg*
118 modest_tny_msg_new_html_plain (const gchar* mailto, const gchar* from, const gchar *cc,
119                                const gchar *bcc, const gchar* subject, 
120                                const gchar *html_body, const gchar *plain_body,
121                                GList *attachments, GList *images, gint *attached, GError **err)
122 {
123         TnyMsg *new_msg;
124         TnyHeader *header;
125         gchar *content_type;
126         gint tmp_attached;
127         
128         /* Create new msg */
129         new_msg = modest_formatter_create_message (NULL, FALSE, (attachments != NULL), (images != NULL));
130         header  = tny_msg_get_header (new_msg);
131         
132         if ((from != NULL) && (strlen(from) > 0)) {
133                 tny_header_set_from (TNY_HEADER (header), from);
134                 tny_header_set_replyto (TNY_HEADER (header), from);
135         }
136         if ((mailto != NULL) && (strlen(mailto) > 0)) 
137                 tny_header_set_to (TNY_HEADER (header), mailto);
138         if ((cc != NULL) && (strlen(cc) > 0)) 
139                 tny_header_set_cc (TNY_HEADER (header), cc);
140         if ((bcc != NULL) && (strlen(bcc) > 0)) 
141                 tny_header_set_bcc (TNY_HEADER (header), bcc);
142         
143         if ((subject != NULL) && (strlen(subject) > 0)) 
144                 tny_header_set_subject (TNY_HEADER (header), subject);
145
146         content_type = get_content_type(plain_body);
147         
148         /* set modest as the X-Mailer
149          * we could this in the platform factory, but then the header
150          * would show up before all the others.
151          */
152         tny_mime_part_set_header_pair (TNY_MIME_PART (new_msg), "X-Mailer", "Modest "
153                                        VERSION);
154
155         /* Add the body of the new mail */      
156         add_body_part (new_msg, plain_body, content_type);
157         add_html_body_part (new_msg, html_body);
158         g_free (content_type);
159                        
160         /* Add attachments */
161         tmp_attached = add_attachments (TNY_MIME_PART (new_msg), attachments, FALSE, err);
162         if (attached)
163                 *attached = tmp_attached;
164         add_images (new_msg, images, err);
165         if (header)
166                 g_object_unref(header);
167
168         return new_msg;
169 }
170
171
172 /* FIXME: this func copy from modest-mail-operation: refactor */
173 static TnyMimePart *
174 add_body_part (TnyMsg *msg, 
175                const gchar *body,
176                const gchar *content_type)
177 {
178         TnyMimePart *text_body_part = NULL;
179         TnyStream *text_body_stream;
180
181         /* Create the stream */
182         text_body_stream = TNY_STREAM (tny_camel_mem_stream_new_with_buffer
183                                         (body, (body ? strlen(body) : 0)));
184
185         text_body_part = modest_formatter_create_body_part (NULL, msg);
186
187         /* Construct MIME part */
188         tny_stream_reset (text_body_stream);
189         tny_mime_part_construct (text_body_part,
190                                  text_body_stream,
191                                  content_type, "7bit");
192         tny_stream_reset (text_body_stream);
193
194         g_object_unref (G_OBJECT(text_body_part));
195
196         /* Clean */
197         g_object_unref (text_body_stream);
198
199         return text_body_part;
200 }
201
202 static TnyMimePart *
203 add_html_body_part (TnyMsg *msg, 
204                     const gchar *body)
205 {
206         TnyMimePart *html_body_part = NULL;
207         TnyStream *html_body_stream;
208
209         /* Create the stream */
210         html_body_stream = TNY_STREAM (tny_camel_mem_stream_new_with_buffer
211                                         (body, strlen(body)));
212
213         /* Create body part if needed */
214         html_body_part = modest_formatter_create_body_part (NULL, msg);
215
216         /* Construct MIME part */
217         tny_stream_reset (html_body_stream);
218         tny_mime_part_construct (html_body_part,
219                                  html_body_stream,
220                                  "text/html; charset=utf-8", 
221                                  "7bit"); /* Sometimes it might be needed 
222                                              to make this one a 8bit! */
223         tny_stream_reset (html_body_stream);
224
225         g_object_unref (G_OBJECT(html_body_part));
226
227         /* Clean */
228         g_object_unref (html_body_stream);
229
230         return html_body_part;
231 }
232
233 static TnyMimePart *
234 copy_mime_part (TnyMimePart *part, GError **err)
235 {
236         TnyMimePart *result = NULL;
237         const gchar *attachment_content_type;
238         const gchar *attachment_filename;
239         const gchar *attachment_cid;
240         TnyList *parts;
241         TnyIterator *iterator;
242         TnyStream *attachment_stream;
243         const gchar *enc;
244         gint ret;
245         
246         if (TNY_IS_MSG (part)) {
247                 g_object_ref (part);
248                 return part;
249         }
250
251         result = tny_platform_factory_new_mime_part (
252                 modest_runtime_get_platform_factory());
253
254         attachment_content_type = tny_mime_part_get_content_type (part);
255
256         /* get mime part headers */
257         attachment_filename = tny_mime_part_get_filename (part);
258         attachment_cid = tny_mime_part_get_content_id (part);
259         
260         /* fill the stream */
261         attachment_stream = tny_mime_part_get_decoded_stream (part);
262         enc = tny_mime_part_get_transfer_encoding (part);
263         if (attachment_stream == NULL) {
264                 if (err != NULL && *err == NULL)
265                         g_set_error (err, MODEST_MAIL_OPERATION_ERROR, MODEST_MAIL_OPERATION_ERROR_FILE_IO, _("TODO: couldn't retrieve attachment"));
266                 g_object_unref (result);
267                 return NULL;
268         } else {
269                 ret = tny_stream_reset (attachment_stream);
270                 ret = tny_mime_part_construct (result,
271                                                attachment_stream,
272                                                attachment_content_type, 
273                                                enc);
274                 ret = tny_stream_reset (attachment_stream);
275         }
276         
277         /* set other mime part fields */
278         tny_mime_part_set_filename (result, attachment_filename);
279         tny_mime_part_set_content_id (result, attachment_cid);
280
281         /* copy subparts */
282         parts = tny_simple_list_new ();
283         tny_mime_part_get_parts (part, parts);
284         iterator = tny_list_create_iterator (parts);
285         while (!tny_iterator_is_done (iterator)) {
286                 TnyMimePart *subpart = TNY_MIME_PART (tny_iterator_get_current (iterator));
287                 if (subpart) {
288                         const gchar *subpart_cid;
289                         TnyMimePart *subpart_copy = copy_mime_part (subpart, err);
290                         if (subpart_copy != NULL) {
291                                 subpart_cid = tny_mime_part_get_content_id (subpart);
292                                 tny_mime_part_add_part (result, subpart_copy);
293                                 if (subpart_cid)
294                                         tny_mime_part_set_content_id (result, subpart_cid);
295                                 g_object_unref (subpart_copy);
296                         }
297                         g_object_unref (subpart);
298                 }
299
300                 tny_iterator_next (iterator);
301         }
302         g_object_unref (iterator);
303         g_object_unref (parts);
304         g_object_unref (attachment_stream);
305
306         return result;
307 }
308
309 static gint
310 add_attachments (TnyMimePart *part, GList *attachments_list, gboolean add_inline, GError **err)
311 {
312         GList *pos;
313         TnyMimePart *attachment_part, *old_attachment;
314         gint ret;
315         gint attached = 0;
316
317         for (pos = (GList *)attachments_list; pos; pos = pos->next) {
318
319                 old_attachment = pos->data;
320                 if (!tny_mime_part_is_purged (old_attachment)) {
321                         const gchar *old_cid;
322                         old_cid = tny_mime_part_get_content_id (old_attachment);
323                         attachment_part = copy_mime_part (old_attachment, err);
324                         if (attachment_part != NULL) {
325                                 if (add_inline) {
326                                         tny_mime_part_set_header_pair (attachment_part, "Content-Disposition",
327                                                                        "inline");
328                                 } else {
329                                         const gchar *filename;
330                                         filename = tny_mime_part_get_filename (old_attachment);
331                                         if (filename)
332                                                 tny_mime_part_set_filename (attachment_part, filename);
333                                         else
334                                                 tny_mime_part_set_header_pair (attachment_part, "Content-Disposition",
335                                                                                "attachment");
336                                 }
337                                 tny_mime_part_set_transfer_encoding (TNY_MIME_PART (attachment_part), "base64");
338                                 ret = tny_mime_part_add_part (TNY_MIME_PART (part), attachment_part);
339                                 attached++;
340                                 if (old_cid)
341                                         tny_mime_part_set_content_id (attachment_part, old_cid);
342                                 g_object_unref (attachment_part);
343                         }
344                 }
345         }
346         return attached;
347 }
348
349 static void
350 add_images (TnyMsg *msg, GList *images_list, GError **err)
351 {
352         TnyMimePart *related_part = NULL;
353         const gchar *content_type;
354
355         content_type = tny_mime_part_get_content_type (TNY_MIME_PART (msg));
356
357         if ((content_type != NULL) && !strcasecmp (content_type, "multipart/related")) {
358                 related_part = g_object_ref (msg);
359         } else if ((content_type != NULL) && !strcasecmp (content_type, "multipart/mixed")) {
360                 TnyList *parts = TNY_LIST (tny_simple_list_new ());
361                 TnyIterator *iter = NULL;
362                 tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
363                 iter = tny_list_create_iterator (parts);
364
365                 while (!tny_iterator_is_done (iter)) {
366                         TnyMimePart *part = TNY_MIME_PART (tny_iterator_get_current (iter));
367                         if (part && !g_strcasecmp (tny_mime_part_get_content_type (part), "multipart/related")) {
368                                 related_part = part;
369                                 break;
370                         } else {
371                                 g_object_unref (part);
372                         }
373                         tny_iterator_next (iter);
374                 }
375                 g_object_unref (iter);
376                 g_object_unref (parts);
377         }
378
379         if (related_part != NULL) {
380                 /* TODO: attach images in their proper place */
381                 add_attachments (related_part, images_list, TRUE, err);
382                 g_object_unref (related_part);
383         }
384 }
385
386
387 gchar * 
388 modest_tny_msg_get_body (TnyMsg *msg, gboolean want_html, gboolean *is_html)
389 {
390         TnyStream *stream;
391         TnyMimePart *body;
392         GtkTextBuffer *buf;
393         GtkTextIter start, end;
394         gchar *to_quote;
395         gboolean result_was_html = TRUE;
396
397         g_return_val_if_fail (msg && TNY_IS_MSG(msg), NULL);
398         
399         body = modest_tny_msg_find_body_part(msg, want_html);
400         if (!body)
401                 return NULL;
402
403         buf = gtk_text_buffer_new (NULL);
404         stream = TNY_STREAM (tny_gtk_text_buffer_stream_new (buf));
405         tny_stream_reset (stream);
406         tny_mime_part_decode_to_stream (body, stream, NULL);
407         tny_stream_reset (stream);
408         
409         gtk_text_buffer_get_bounds (buf, &start, &end);
410         to_quote = gtk_text_buffer_get_text (buf, &start, &end, FALSE);
411         if (tny_mime_part_content_type_is (body, "text/plain")) {
412                 gchar *to_quote_converted = modest_text_utils_convert_to_html (to_quote);
413                 g_free (to_quote);
414                 to_quote = to_quote_converted;
415                 result_was_html = FALSE;
416         }
417
418         g_object_unref (buf);
419         g_object_unref (G_OBJECT(stream));
420         g_object_unref (G_OBJECT(body));
421
422         if (is_html != NULL)
423                 *is_html = result_was_html;
424
425         return to_quote;
426 }
427
428
429 TnyMimePart*
430 modest_tny_msg_find_body_part_from_mime_part (TnyMimePart *msg, gboolean want_html)
431 {
432         const gchar *desired_mime_type = want_html ? "text/html" : "text/plain";
433         TnyMimePart *part = NULL;
434         TnyList *parts = NULL;
435         TnyIterator *iter = NULL;
436         gchar *header_content_type;
437         gchar *header_content_type_lower = NULL;
438         
439         if (!msg)
440                 return NULL;
441
442         /* If it's an application multipart, then we don't get into as we don't
443          * support them (for example application/sml or wap messages */
444         header_content_type = modest_tny_mime_part_get_header_value (msg, "Content-Type");
445         if (header_content_type) {
446                 header_content_type = g_strstrip (header_content_type);
447                 header_content_type_lower = g_ascii_strdown (header_content_type, -1);
448         }
449         if (header_content_type_lower && 
450             g_str_has_prefix (header_content_type_lower, "multipart/") &&
451             !g_str_has_prefix (header_content_type_lower, "multipart/signed") &&
452             strstr (header_content_type_lower, "application/")) {
453                 g_free (header_content_type_lower);
454                 g_free (header_content_type);
455                 return NULL;
456         }       
457         g_free (header_content_type_lower);
458         g_free (header_content_type);
459
460         parts = TNY_LIST (tny_simple_list_new());
461         tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
462
463         iter  = tny_list_create_iterator(parts);
464
465         /* no parts? assume it's single-part message */
466         if (tny_iterator_is_done(iter)) {
467                 gchar *content_type;
468                 gboolean is_text_part;
469                 g_object_unref (G_OBJECT(iter));
470                 content_type = modest_tny_mime_part_get_content_type (msg);
471                 if (content_type == NULL)
472                         return NULL;
473                 is_text_part = 
474                         g_str_has_prefix (content_type, "text/");
475                 g_free (content_type);
476                 /* if this part cannot be a supported body return NULL */
477                 if (!is_text_part) {
478                         return NULL;
479                 } else {
480                         return TNY_MIME_PART (g_object_ref(G_OBJECT(msg)));
481                 }
482         } else {
483                 do {
484                         gchar *tmp, *content_type = NULL;
485                         gboolean has_content_disp_name = FALSE;
486
487                         part = TNY_MIME_PART(tny_iterator_get_current (iter));
488
489                         if (!part) {
490                                 g_warning ("%s: not a valid mime part", __FUNCTION__);
491                                 tny_iterator_next (iter);
492                                 continue;
493                         }
494
495                         /* it's a message --> ignore */
496                         if (part && TNY_IS_MSG (part)) {
497                                 g_object_unref (part);
498                                 part = NULL;
499                                 tny_iterator_next (iter);
500                                 continue;
501                         }                       
502
503                         /* we need to strdown the content type, because
504                          * tny_mime_part_has_content_type does not do it...
505                          */
506                         content_type = g_ascii_strdown (tny_mime_part_get_content_type (part), -1);
507                         
508                         /* mime-parts with a content-disposition header (either 'inline' or 'attachment')
509                          * and a 'name=' thingy cannot be body parts
510                          */
511                                 
512                         tmp = modest_tny_mime_part_get_header_value (part, "Content-Disposition");
513                         if (tmp) {
514                                 gchar *content_disp = g_ascii_strdown(tmp, -1);
515                                 g_free (tmp);
516                                 has_content_disp_name = g_strstr_len (content_disp, strlen(content_disp), "name=") != NULL;
517                                 g_free (content_disp);
518                         }
519                         
520                         if (g_str_has_prefix (content_type, desired_mime_type) && 
521                             !has_content_disp_name &&
522                             !modest_tny_mime_part_is_attachment_for_modest (part)) {
523                                 /* we found the desired mime-type! */
524                                 g_free (content_type);
525                                 break;
526
527                         } else  if (g_str_has_prefix(content_type, "multipart")) {
528
529                                 /* multipart? recurse! */
530                                 g_object_unref (part);
531                                 g_free (content_type);
532                                 part = modest_tny_msg_find_body_part_from_mime_part (part, want_html);
533                                 if (part)
534                                         break;
535                         } else
536                                 g_free (content_type);
537                         
538                         if (part) {
539                                 g_object_unref (G_OBJECT(part));
540                                 part = NULL;
541                         }
542                         
543                         tny_iterator_next (iter);
544                         
545                 } while (!tny_iterator_is_done(iter));
546         }
547         
548         g_object_unref (G_OBJECT(iter));
549         g_object_unref (G_OBJECT(parts));
550
551         /* if were trying to find an HTML part and couldn't find it,
552          * try to find a text/plain part instead
553          */
554         if (!part && want_html) 
555                 return modest_tny_msg_find_body_part_from_mime_part (msg, FALSE);
556         else
557                 return part; /* this maybe NULL, this is not an error; some message just don't have a body
558                               * part */
559 }
560
561
562 TnyMimePart*
563 modest_tny_msg_find_body_part (TnyMsg *msg, gboolean want_html)
564 {
565         g_return_val_if_fail (msg && TNY_IS_MSG(msg), NULL);
566         
567         return modest_tny_msg_find_body_part_from_mime_part (TNY_MIME_PART(msg),
568                                                              want_html);
569 }
570
571
572 #define MODEST_TNY_MSG_PARENT_UID "parent-uid"
573
574 static TnyMsg *
575 create_reply_forward_mail (TnyMsg *msg, TnyHeader *header, const gchar *from,
576                            const gchar *signature, gboolean is_reply,
577                            guint type /*ignored*/, GList *attachments)
578 {
579         TnyMsg *new_msg;
580         TnyHeader *new_header;
581         gchar *old_subject;
582         gchar *new_subject;
583         TnyMimePart *body = NULL;
584         ModestFormatter *formatter;
585         gchar *subject_prefix;
586         gboolean no_text_part;
587         
588         if (header)
589                 g_object_ref (header);
590         else
591                 header = tny_msg_get_header (msg);
592
593         /* Get body from original msg. Always look for the text/plain
594            part of the message to create the reply/forwarded mail */
595         if (msg != NULL)
596                 body   = modest_tny_msg_find_body_part (msg, FALSE);
597
598         if (modest_conf_get_bool (modest_runtime_get_conf (), MODEST_CONF_PREFER_FORMATTED_TEXT,
599                                   NULL))
600                 formatter = modest_formatter_new ("text/html", signature);
601         else
602                 formatter = modest_formatter_new ("text/plain", signature);
603
604
605         /* if we don't have a text-part */
606         no_text_part = (!body) || (strcmp (tny_mime_part_get_content_type (body), "text/html")==0);
607
608         /* when we're reply, include the text part if we have it, or nothing otherwise. */
609         if (is_reply)
610                 new_msg = modest_formatter_quote  (formatter, no_text_part ? NULL: body, header,
611                                                     attachments);
612         else {
613                 /* for attachements; inline if there is a text part, and include the
614                  * full old mail if there was none */
615                 if (no_text_part) {
616                         new_msg = modest_formatter_attach (formatter, msg, header);
617                 } else { 
618                         new_msg = modest_formatter_inline  (formatter, body, header,
619                                                             attachments);
620                 }
621         }
622
623         g_object_unref (G_OBJECT(formatter));
624         if (body)
625                 g_object_unref (G_OBJECT(body));
626
627         /* Fill the header */
628         new_header = tny_msg_get_header (new_msg);      
629         tny_header_set_from (new_header, from);
630         tny_header_set_replyto (new_header, from);
631
632         /* Change the subject */
633         if (is_reply)
634                 subject_prefix = g_strconcat (_("mail_va_re"), ":", NULL);
635         else
636                 subject_prefix = g_strconcat (_("mail_va_fw"), ":", NULL);
637         old_subject = tny_header_dup_subject (header);
638         new_subject =
639                 (gchar *) modest_text_utils_derived_subject (old_subject,
640                                                              subject_prefix);
641         g_free (old_subject);
642         g_free (subject_prefix);
643         tny_header_set_subject (new_header, (const gchar *) new_subject);
644         g_free (new_subject);
645         
646         /* get the parent uid, and set it as a gobject property on the new msg */
647         if (new_msg) {
648                 gchar* parent_uid = modest_tny_folder_get_header_unique_id (header);
649                 g_object_set_data_full (G_OBJECT(new_msg), MODEST_TNY_MSG_PARENT_UID,
650                                         parent_uid, g_free);
651         }
652
653         /* set modest as the X-Mailer
654          * we could this in the platform factory, but then the header
655          * would show up before all the others.
656          */
657         tny_mime_part_set_header_pair (TNY_MIME_PART (msg), "X-Mailer", "Modest "
658                                        VERSION);
659
660         /* Clean */
661         g_object_unref (G_OBJECT (new_header));
662         g_object_unref (G_OBJECT (header));
663         /* ugly to unref it here instead of in the calling func */
664
665         if (!is_reply & !no_text_part) {
666                 add_attachments (TNY_MIME_PART (new_msg), attachments, FALSE, NULL);
667         }
668
669         return new_msg;
670 }
671
672 const gchar*
673 modest_tny_msg_get_parent_uid (TnyMsg *msg)
674 {
675         g_return_val_if_fail (msg && TNY_IS_MSG(msg), NULL);
676         
677         return g_object_get_data (G_OBJECT(msg), MODEST_TNY_MSG_PARENT_UID);
678 }
679
680
681
682 static void
683 add_if_attachment (gpointer data, gpointer user_data)
684 {
685         TnyMimePart *part;
686         GList **attachments_list;
687
688         part = TNY_MIME_PART (data);
689         attachments_list = ((GList **) user_data);
690
691         if ((tny_mime_part_is_attachment (part))||(TNY_IS_MSG (part))) {
692                 *attachments_list = g_list_prepend (*attachments_list, part);
693                 g_object_ref (part);
694         }
695 }
696
697 TnyMsg* 
698 modest_tny_msg_create_forward_msg (TnyMsg *msg, 
699                                    const gchar *from,
700                                    const gchar *signature,
701                                    ModestTnyMsgForwardType forward_type)
702 {
703         TnyMsg *new_msg;
704         TnyList *parts = NULL;
705         GList *attachments_list = NULL;
706
707         g_return_val_if_fail (msg && TNY_IS_MSG(msg), NULL);
708         
709         /* Add attachments */
710         parts = TNY_LIST (tny_simple_list_new());
711         tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
712         tny_list_foreach (parts, add_if_attachment, &attachments_list);
713
714         new_msg = create_reply_forward_mail (msg, NULL, from, signature, FALSE, forward_type,
715                                              attachments_list);
716
717         /* Clean */
718         if (attachments_list) {
719                 g_list_foreach (attachments_list, (GFunc) g_object_unref, NULL);
720                 g_list_free (attachments_list);
721         }
722         g_object_unref (G_OBJECT (parts));
723
724         return new_msg;
725 }
726
727
728
729 static gint
730 count_addresses (const gchar* addresses)
731 {
732         gint count = 1;
733
734         if (!addresses)
735                 return 0;
736         
737         while (*addresses) {
738                 if (*addresses == ',' || *addresses == ';')
739                         ++count;
740                 ++addresses;
741         }
742         
743         return count;
744 }
745
746
747 /* get the new To:, based on the old header,
748  * result is newly allocated or NULL in case of error
749  * */
750 static gchar*
751 get_new_to (TnyMsg *msg, TnyHeader *header, const gchar* from,
752             ModestTnyMsgReplyMode reply_mode)
753 {
754         gchar* old_reply_to;
755         gchar* old_from;
756         gchar* new_to;
757         
758         /* according to RFC2369 (http://www.faqs.org/rfcs/rfc2369.html), we
759          * can identify Mailing-List posts by the List-Help header.
760          * for mailing lists, both the Reply-To: and From: should be included
761          * in the new To:; for now, we're ignoring List-Post
762          */
763         gchar* list_help = modest_tny_mime_part_get_header_value (TNY_MIME_PART(msg), 
764                                                                   "List-Help");
765         gboolean is_mailing_list = (list_help != NULL);
766         g_free (list_help);
767
768
769         /* reply to sender, use ReplyTo or From */
770         old_reply_to = modest_tny_mime_part_get_header_value (TNY_MIME_PART(msg), 
771                                                               "Reply-To"); 
772         old_from     = tny_header_dup_from (header);
773         
774         if (!old_from && !old_reply_to) {
775                 g_warning ("%s: failed to get either Reply-To: or From: from header",
776                            __FUNCTION__);
777                 return NULL;
778         }
779         
780         /* for mailing lists, use both Reply-To and From if we did a
781          * 'Reply All:'
782          * */
783         if (is_mailing_list && reply_mode == MODEST_TNY_MSG_REPLY_MODE_ALL &&
784             old_reply_to && old_from && strcmp (old_from, old_reply_to) != 0)
785                 new_to = g_strjoin (",", old_reply_to, old_from, NULL);
786         else
787                 /* otherwise use either Reply-To: (preferred) or From: */
788                 new_to = g_strdup (old_reply_to ? old_reply_to : old_from);
789         g_free (old_from);
790         g_free (old_reply_to);
791
792         /* in case of ReplyAll, we need to add the Recipients in the old To: */
793         if (reply_mode == MODEST_TNY_MSG_REPLY_MODE_ALL) {
794                 gchar *old_to = tny_header_dup_to (header);
795                 if (!old_to) 
796                         g_warning ("%s: no To: address found in source mail",
797                                    __FUNCTION__);
798                 else {
799                         /* append the old To: */
800                         gchar *tmp = g_strjoin (",", new_to, old_to, NULL);
801                         g_free (new_to);
802                         new_to = tmp;
803                         g_free (old_to);
804                 }
805
806                 /* remove duplicate entries */
807                 gchar *tmp = modest_text_utils_remove_duplicate_addresses (new_to);
808                 g_free (new_to);
809                 new_to = tmp;
810                 
811                 /* now, strip me (the new From:) from the new_to, but only if
812                  * there are >1 addresses there */
813                 if (count_addresses (new_to) > 1) {
814                         gchar *tmp = modest_text_utils_remove_address (new_to, from);
815                         g_free (new_to);
816                         new_to = tmp;
817                 }
818         }
819
820         return new_to;
821 }
822
823
824 /* get the new Cc:, based on the old header,
825  * result is newly allocated or NULL in case of error */
826 static gchar*
827 get_new_cc (TnyHeader *header, const gchar* from)
828 {
829         gchar *old_cc, *result, *dup;
830
831         old_cc = tny_header_dup_cc (header);
832         if (!old_cc)
833                 return NULL;
834
835         /* remove me (the new From:) from the Cc: list */
836         dup =  modest_text_utils_remove_address (old_cc, from);
837         result = modest_text_utils_remove_duplicate_addresses (dup);
838         g_free (dup);
839         g_free (old_cc);
840         return result;
841 }
842
843
844 TnyMsg* 
845 modest_tny_msg_create_reply_msg (TnyMsg *msg,
846                                  TnyHeader *header,
847                                  const gchar *from,
848                                  const gchar *signature,
849                                  ModestTnyMsgReplyType reply_type,
850                                  ModestTnyMsgReplyMode reply_mode)
851 {
852         TnyMsg *new_msg = NULL;
853         TnyHeader *new_header;
854         gchar *new_to = NULL;
855         TnyList *parts = NULL;
856         GList *attachments_list = NULL;
857
858         g_return_val_if_fail (msg && TNY_IS_MSG(msg), NULL);
859         
860         parts = TNY_LIST (tny_simple_list_new());
861         tny_mime_part_get_parts (TNY_MIME_PART (msg), parts);
862         tny_list_foreach (parts, add_if_attachment, &attachments_list);
863
864         new_msg = create_reply_forward_mail (msg, header, from, signature, TRUE, reply_type,
865                                              attachments_list);
866         if (attachments_list != NULL) {
867                 g_list_foreach (attachments_list, (GFunc) g_object_unref, NULL);
868                 g_list_free (attachments_list);
869         }
870         if (parts)
871                 g_object_unref (G_OBJECT (parts));
872
873         /* Fill the header */
874         if (header)
875                 g_object_ref (header);
876         else
877                 header = tny_msg_get_header (msg);
878
879         
880         new_header = tny_msg_get_header(new_msg);
881         new_to = get_new_to (msg, header, from, reply_mode);
882         if (!new_to)
883                 g_warning ("%s: failed to get new To:", __FUNCTION__);
884         else {
885                 tny_header_set_to (new_header, new_to);
886                 g_free (new_to);
887         }
888
889         if (reply_mode == MODEST_TNY_MSG_REPLY_MODE_ALL) {
890                 gchar *new_cc = get_new_cc (header, from);
891                 if (new_cc) { 
892                         tny_header_set_cc (new_header, new_cc);
893                         g_free (new_cc);
894                 }
895         }
896
897         /* Clean */
898         g_object_unref (G_OBJECT (new_header));
899         g_object_unref (G_OBJECT (header));
900
901         return new_msg;
902 }
903
904
905 static gboolean
906 is_ascii(const gchar *s)
907 {
908         if (!s)
909                 return TRUE;
910         while (s[0]) {
911                 if (s[0] & 128 || s[0] < 32)
912                         return FALSE;
913                 s++;
914         }
915         return TRUE;
916 }
917
918 static char *
919 get_content_type(const gchar *s)
920 {
921         GString *type;
922         
923         type = g_string_new("text/plain");
924         if (!is_ascii(s)) {
925                 if (g_utf8_validate(s, -1, NULL)) {
926                         g_string_append(type, "; charset=\"utf-8\"");
927                 } else {
928                         /* it should be impossible to reach this, but better safe than sorry */
929                         g_warning("invalid utf8 in message");
930                         g_string_append(type, "; charset=\"latin1\"");
931                 }
932         }
933         return g_string_free(type, FALSE);
934 }
935
936 guint64
937 modest_tny_msg_estimate_size (const gchar *plain_body, const gchar *html_body,
938                               guint64 parts_count,
939                               guint64 parts_size)
940 {
941         guint64 result;
942
943         /* estimation of headers size */
944         result = 1024;
945
946         /* We add a 20% of size due to the increase in 7bit encoding */
947         if (plain_body) {
948                 result += strlen (plain_body) * 120 / 100;
949         }
950         if (html_body) {
951                 result += strlen (html_body) * 120 / 100;
952         }
953
954         /* 256 bytes per additional part because of their headers */
955         result += parts_count * 256;
956
957         /* 150% of increase per encoding */
958         result += parts_size * 3 / 2;
959
960         return result;
961 }
962
963 GSList *
964 modest_tny_msg_header_get_all_recipients_list (TnyHeader *header)
965 {
966         GSList *recipients = NULL;
967         gchar *from = NULL, *to = NULL, *cc = NULL, *bcc = NULL;
968
969         if (header == NULL)
970                 return NULL;
971
972         from = tny_header_dup_from (header);
973         to = tny_header_dup_to (header);
974         cc = tny_header_dup_cc (header);
975         bcc = tny_header_dup_bcc (header);
976
977         recipients = NULL;
978         if (from)
979                 recipients = g_slist_concat (recipients, modest_text_utils_split_addresses_list (from));
980         if (to)
981                 recipients = g_slist_concat (recipients, modest_text_utils_split_addresses_list (to));
982         if (cc)
983                 recipients = g_slist_concat (recipients, modest_text_utils_split_addresses_list (cc));
984         if (bcc)
985                 recipients = g_slist_concat (recipients, modest_text_utils_split_addresses_list (bcc));
986
987         g_free (from);
988         g_free (to);
989         g_free (cc);
990         g_free (bcc);
991
992         return recipients;
993 }
994
995 GSList *
996 modest_tny_msg_get_all_recipients_list (TnyMsg *msg)
997 {
998         TnyHeader *header = NULL;
999         GSList *recipients = NULL;
1000
1001         if (msg == NULL)
1002                 return NULL;
1003
1004         header = tny_msg_get_header (msg);
1005         if (header == NULL)
1006                 return NULL;
1007
1008         recipients = modest_tny_msg_header_get_all_recipients_list (header);
1009         g_object_unref (header);
1010
1011         return recipients;
1012 }
1013