From 330ef6ca44df4111d6f6b33ba0684e865ba8c30e Mon Sep 17 00:00:00 2001 From: thomascube <thomas@roundcube.net> Date: Fri, 31 Jul 2009 04:55:56 -0400 Subject: [PATCH] Create plugin hook for encrypted message parts + add size property to text part in order to display it --- program/include/rcube_message.php | 103 ++++++++++++++++++++++++++++++++++++++------------- 1 files changed, 76 insertions(+), 27 deletions(-) diff --git a/program/include/rcube_message.php b/program/include/rcube_message.php index d810ef3..0c46945 100644 --- a/program/include/rcube_message.php +++ b/program/include/rcube_message.php @@ -5,7 +5,7 @@ | program/include/rcube_message.php | | | | This file is part of the RoundCube Webmail client | - | Copyright (C) 2008, RoundCube Dev. - Switzerland | + | Copyright (C) 2008-2009, RoundCube Dev. - Switzerland | | Licensed under the GNU GPL | | | | PURPOSE: | @@ -42,34 +42,51 @@ public $mime_parts = array(); public $attachments = array(); public $subject = ''; + public $sender = null; public $is_safe = false; - + + /** + * __construct + * + * Provide a uid, and parse message structure. + * + * @param string $uid The message UID. + * + * @uses rcmail::get_instance() + * @uses rcube_imap::decode_mime_string() + * @uses self::set_safe() + * + * @see self::$app, self::$imap, self::$opt, self::$structure + */ function __construct($uid) { $this->app = rcmail::get_instance(); $this->imap = $this->app->imap; $this->uid = $uid; - $this->headers = $this->imap->get_headers($uid); + $this->headers = $this->imap->get_headers($uid, NULL, true, true); + $this->subject = rcube_imap::decode_mime_string($this->headers->subject, $this->headers->charset); + list(, $this->sender) = each($this->imap->decode_address_list($this->headers->from)); - $this->is_safe = (intval($_GET['_safe']) || $_SESSION['safe_messages'][$uid]) ? true : false; - $_SESSION['safe_messages'][$uid] = $this->is_safe; - + $this->set_safe((intval($_GET['_safe']) || $_SESSION['safe_messages'][$uid])); $this->opt = array( 'safe' => $this->is_safe, 'prefer_html' => $this->app->config->get('prefer_html'), 'get_url' => rcmail_url('get', array('_mbox' => $this->imap->get_mailbox_name(), '_uid' => $uid)) ); - - if ($this->structure = $this->imap->get_structure($uid)) { + + if ($this->structure = $this->imap->get_structure($uid, $this->headers->body_structure)) { $this->get_mime_numbers($this->structure); $this->parse_structure($this->structure); } else { $this->body = $this->imap->get_body($uid); } + + // notify plugins and let them analyze this structured message object + $this->app->plugins->exec_hook('message_load', array('object' => $this)); } @@ -116,12 +133,13 @@ * Get content of a specific part of this message * * @param string Part MIME-ID + * @param resource File pointer to save the message part * @return string Part content */ - public function get_part_content($mime_id) + public function get_part_content($mime_id, $fp=NULL) { if ($part = $this->mime_parts[$mime_id]) - return $this->imap->get_message_part($this->uid, $mime_id, $part); + return $this->imap->get_message_part($this->uid, $mime_id, $part, NULL, $fp); else return null; } @@ -151,17 +169,13 @@ */ function first_html_part() { - $html_part = null; - // check all message parts foreach ($this->mime_parts as $mime_id => $part) { $mimetype = strtolower($part->ctype_primary . '/' . $part->ctype_secondary); if ($mimetype == 'text/html') { - $html_part = $this->imap->get_message_part($this->uid, $mime_id, $part); + return $this->imap->get_message_part($this->uid, $mime_id, $part); } } - - return $html_part; } @@ -229,6 +243,11 @@ $structure->type = 'content'; $this->parts[] = &$structure; } + // the same for pgp signed messages + else if ($message_ctype_primary == 'application' && $message_ctype_secondary == 'pgp' && !$recursive) { + $structure->type = 'content'; + $this->parts[] = &$structure; + } // message contains alternative parts else if ($message_ctype_primary == 'multipart' && ($message_ctype_secondary == 'alternative') && is_array($structure->parts)) { // get html/plaintext parts @@ -246,7 +265,7 @@ $html_part = $p; else if ($sub_ctype_primary=='text' && $sub_ctype_secondary=='enriched') $enriched_part = $p; - else if ($sub_ctype_primary=='multipart' && ($sub_ctype_secondary=='related' || $sub_ctype_secondary=='mixed')) + else if ($sub_ctype_primary=='multipart' && in_array($sub_ctype_secondary, array('related', 'mixed', 'alternative'))) $related_part = $p; } @@ -278,7 +297,7 @@ $this->parts[] = $print_part; } // show plaintext warning - else if ($html_part !== nullL && empty($this->parts)) { + else if ($html_part !== null && empty($this->parts)) { $c = new stdClass; $c->type = 'content'; $c->body = rcube_label('htmlmessage'); @@ -304,8 +323,16 @@ $p->ctype_primary = 'text'; $p->ctype_secondary = 'plain'; $p->body = rcube_label('encryptedmessage'); + $p->size = strlen($p->body); - $this->parts[] = $p; + // maybe some plugins are able to decode this encrypted message part + $data = $this->app->plugins->exec_hook('message_part_encrypted', array('object' => $this, 'struct' => $structure, 'part' => $p)); + if (is_array($data['parts'])) { + $this->parts = array_merge($this->parts, $data['parts']); + } + else if ($data['part']) { + $this->parts[] = $p; + } } // message contains multiple parts else if (is_array($structure->parts) && !empty($structure->parts)) { @@ -322,7 +349,7 @@ // part text/[plain|html] OR message/delivery-status else if (($primary_type == 'text' && ($secondary_type == 'plain' || $secondary_type == 'html') && $mail_part->disposition != 'attachment') || ($primary_type == 'message' && ($secondary_type == 'delivery-status' || $secondary_type == 'disposition-notification'))) { - + // add text part if we're not in alternative mode or if it matches the prefs if (!$this->parse_alternative || ($secondary_type == 'html' && $this->opt['prefer_html']) || @@ -330,7 +357,7 @@ $mail_part->type = 'content'; $this->parts[] = $mail_part; } - + // list as attachment as well if (!empty($mail_part->filename)) $this->attachments[] = $mail_part; @@ -338,24 +365,42 @@ // part message/* else if ($primary_type=='message') { $this->parse_structure($mail_part, true); + + // list as attachment as well (mostly .eml) + if (!empty($mail_part->filename)) + $this->attachments[] = $mail_part; } // ignore "virtual" protocol parts else if ($primary_type == 'protocol') continue; + + // part is Microsoft Outlook TNEF (winmail.dat) + else if ($primary_type == 'application' && $secondary_type == 'ms-tnef') { + foreach ((array)$this->imap->tnef_decode($mail_part, $structure->headers['uid']) as $tnef_part) { + $this->mime_parts[$tnef_part->mime_id] = $tnef_part; + $this->attachments[] = $tnef_part; + } + } - // part is file/attachment - else if ($mail_part->disposition == 'attachment' || $mail_part->disposition == 'inline' || + // part is a file/attachment + else if (preg_match('/^(inline|attach)/', $mail_part->disposition) || $mail_part->headers['content-id'] || (empty($mail_part->disposition) && $mail_part->filename)) { + // skip apple resource forks if ($message_ctype_secondary == 'appledouble' && $secondary_type == 'applefile') continue; - // part belongs to a related message - if ($message_ctype_secondary == 'related' && $mail_part->headers['content-id']) { - $mail_part->content_id = preg_replace(array('/^</', '/>$/'), '', $mail_part->headers['content-id']); + // part belongs to a related message and is linked + if ($message_ctype_secondary == 'related' + && ($mail_part->headers['content-id'] || $mail_part->headers['content-location'])) { + if ($mail_part->headers['content-id']) + $mail_part->content_id = preg_replace(array('/^</', '/>$/'), '', $mail_part->headers['content-id']); + if ($mail_part->headers['content-location']) + $mail_part->content_location = $mail_part->headers['content-base'] . $mail_part->headers['content-location']; + $this->inline_parts[] = $mail_part; } - // is regular attachment + // is a regular attachment else { if (!$mail_part->filename) $mail_part->filename = 'Part '.$mail_part->mime_id; @@ -369,7 +414,11 @@ $a_replaces = array(); foreach ($this->inline_parts as $inline_object) { - $a_replaces['cid:'.$inline_object->content_id] = $this->get_part_url($inline_object->mime_id); + $part_url = $this->get_part_url($inline_object->mime_id); + if ($inline_object->content_id) + $a_replaces['cid:'.$inline_object->content_id] = $part_url; + if ($inline_object->content_location) + $a_replaces[$inline_object->content_location] = $part_url; } // add replace array to each content part -- Gitblit v1.9.1