From cd219a45dd64d7d6916656e5e1b3a4ead68974c2 Mon Sep 17 00:00:00 2001
From: Aleksander Machniak <alec@alec.pl>
Date: Fri, 18 Dec 2015 05:42:05 -0500
Subject: [PATCH] Fix regression where some message attachments could be missing on edit/forward (#1490608)

---
 program/steps/mail/func.inc |   34 ++++++++++++++++++++++------------
 1 files changed, 22 insertions(+), 12 deletions(-)

diff --git a/program/steps/mail/func.inc b/program/steps/mail/func.inc
index 5b92793..5436368 100644
--- a/program/steps/mail/func.inc
+++ b/program/steps/mail/func.inc
@@ -1233,17 +1233,6 @@
                 // fetch part body
                 $body = $MESSAGE->get_part_body($part->mime_id, true);
 
-                // extract headers from message/rfc822 parts
-                if ($part->mimetype == 'message/rfc822') {
-                    $msgpart = rcube_mime::parse_message($body);
-                    $body    = rcube_message::format_part_body($msgpart->body, $msgpart);
-
-                    if (!empty($msgpart->headers) && !empty($body)) {
-                        $part = $msgpart;
-                        $out .= html::div('message-partheaders', rcmail_message_headers(sizeof($header_attrib) ? $header_attrib : null, $part->headers));
-                    }
-                }
-
                 // message is cached but not exists (#1485443), or other error
                 if ($body === false) {
                     rcmail_message_error($MESSAGE->uid);
@@ -2214,6 +2203,8 @@
 
 function rcmail_save_attachment($message, $pid, $compose_id, $params = array())
 {
+    global $COMPOSE;
+
     $rcmail  = rcmail::get_instance();
     $storage = $rcmail->get_storage();
 
@@ -2276,7 +2267,26 @@
 
     if ($attachment['status']) {
         unset($attachment['data'], $attachment['status'], $attachment['content_id'], $attachment['abort']);
-        $rcmail->session->append('compose_data_' . $compose_id . '.attachments', $attachment['id'], $attachment);
+
+        $session_key = 'compose_data_' . $compose_id;
+
+        // rcube_session::append() replaces current session data with the old values
+        // (in rcube_session::reload()). This is a problem in 'compose' action, because before
+        // the first append() use we set some important data in the session.
+        // It also overwrites attachments list. Fixing reload() is not so simple if possible
+        // as we don't really know what has been added and what removed in meantime.
+        // So, for now we'll do not use append() on 'compose' action (#1490608).
+
+        if ($rcmail->action == 'compose') {
+            $_SESSION[$session_key]['attachments'][$attachment['id']] = $attachment;
+        }
+        else {
+            $rcmail->session->append($session_key . '.attachments', $attachment['id'], $attachment);
+        }
+
+        // Fix reference to compose session data after rcube_session::append()
+        $COMPOSE =& $_SESSION[$session_key];
+
         return $attachment;
     }
     else if ($path) {

--
Gitblit v1.9.1