Milan Crha
9cea547597
If there was a known formatter for the attachment's mime type, the attachment content was shown inline, without the attachment button, instead of it being shown collapsed and with the corresponding button. All that caused by a variable overwrite and a related memory leak.
417 lines
11 KiB
C
417 lines
11 KiB
C
/*
|
|
* e-mail-formatter-attachment.c
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU Lesser General Public License as published by
|
|
* the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
|
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
* for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public License
|
|
* along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include <config.h>
|
|
#endif
|
|
|
|
#include <glib/gi18n-lib.h>
|
|
|
|
#include <shell/e-shell.h>
|
|
#include <shell/e-shell-window.h>
|
|
|
|
#include "e-mail-formatter-extension.h"
|
|
#include "e-mail-inline-filter.h"
|
|
#include "e-mail-part-attachment-bar.h"
|
|
#include "e-mail-part-attachment.h"
|
|
#include "e-mail-part-utils.h"
|
|
|
|
#define d(x)
|
|
|
|
typedef EMailFormatterExtension EMailFormatterAttachment;
|
|
typedef EMailFormatterExtensionClass EMailFormatterAttachmentClass;
|
|
|
|
GType e_mail_formatter_attachment_get_type (void);
|
|
|
|
G_DEFINE_TYPE (
|
|
EMailFormatterAttachment,
|
|
e_mail_formatter_attachment,
|
|
E_TYPE_MAIL_FORMATTER_EXTENSION)
|
|
|
|
static const gchar *formatter_mime_types[] = {
|
|
E_MAIL_PART_ATTACHMENT_MIME_TYPE,
|
|
"application/vnd.evolution.widget.attachment-button",
|
|
NULL
|
|
};
|
|
|
|
static EAttachmentStore *
|
|
find_attachment_store (EMailPartList *part_list,
|
|
EMailPart *start)
|
|
{
|
|
EAttachmentStore *store = NULL;
|
|
GQueue queue = G_QUEUE_INIT;
|
|
GList *head, *link;
|
|
const gchar *start_id;
|
|
gchar *tmp, *pos;
|
|
EMailPart *part;
|
|
gchar *id;
|
|
|
|
start_id = e_mail_part_get_id (start);
|
|
|
|
e_mail_part_list_queue_parts (part_list, NULL, &queue);
|
|
|
|
head = g_queue_peek_head_link (&queue);
|
|
|
|
id = g_strconcat (start_id, ".attachment-bar", NULL);
|
|
tmp = g_strdup (id);
|
|
part = NULL;
|
|
do {
|
|
d (printf ("Looking up attachment bar as %s\n", id));
|
|
|
|
for (link = head; link != NULL; link = g_list_next (link)) {
|
|
EMailPart *p = link->data;
|
|
const gchar *p_id;
|
|
|
|
p_id = e_mail_part_get_id (p);
|
|
|
|
if (g_strcmp0 (p_id, id) == 0) {
|
|
part = p;
|
|
break;
|
|
}
|
|
}
|
|
|
|
pos = g_strrstr (tmp, ".");
|
|
if (!pos)
|
|
break;
|
|
|
|
g_free (id);
|
|
g_free (tmp);
|
|
tmp = g_strndup (start_id, pos - tmp);
|
|
id = g_strdup_printf ("%s.attachment-bar", tmp);
|
|
|
|
} while (pos && !part);
|
|
|
|
g_free (id);
|
|
g_free (tmp);
|
|
|
|
if (part != NULL)
|
|
store = e_mail_part_attachment_bar_get_store (
|
|
E_MAIL_PART_ATTACHMENT_BAR (part));
|
|
|
|
while (!g_queue_is_empty (&queue))
|
|
g_object_unref (g_queue_pop_head (&queue));
|
|
|
|
return store;
|
|
}
|
|
|
|
static gboolean
|
|
emfe_attachment_format (EMailFormatterExtension *extension,
|
|
EMailFormatter *formatter,
|
|
EMailFormatterContext *context,
|
|
EMailPart *part,
|
|
GOutputStream *stream,
|
|
GCancellable *cancellable)
|
|
{
|
|
gchar *text, *html;
|
|
gchar *button_id;
|
|
EAttachmentStore *store;
|
|
EMailExtensionRegistry *registry;
|
|
GQueue *extensions;
|
|
EMailPartAttachment *empa;
|
|
CamelMimePart *mime_part;
|
|
CamelMimeFilterToHTMLFlags flags;
|
|
GString *buffer;
|
|
const gchar *attachment_part_id;
|
|
const gchar *part_id;
|
|
|
|
g_return_val_if_fail (E_IS_MAIL_PART_ATTACHMENT (part), FALSE);
|
|
|
|
empa = (EMailPartAttachment *) part;
|
|
part_id = e_mail_part_get_id (part);
|
|
|
|
if ((context->mode == E_MAIL_FORMATTER_MODE_NORMAL) ||
|
|
(context->mode == E_MAIL_FORMATTER_MODE_PRINTING) ||
|
|
(context->mode == E_MAIL_FORMATTER_MODE_ALL_HEADERS)) {
|
|
EAttachment *attachment;
|
|
GList *head, *link;
|
|
|
|
attachment = e_mail_part_attachment_ref_attachment (
|
|
E_MAIL_PART_ATTACHMENT (part));
|
|
|
|
head = g_queue_peek_head_link (&part->validities);
|
|
|
|
for (link = head; link != NULL; link = g_list_next (link)) {
|
|
EMailPartValidityPair *pair = link->data;
|
|
|
|
if (pair == NULL)
|
|
continue;
|
|
|
|
if ((pair->validity_type & E_MAIL_PART_VALIDITY_SIGNED) != 0)
|
|
e_attachment_set_signed (
|
|
attachment,
|
|
pair->validity->sign.status);
|
|
|
|
if ((pair->validity_type & E_MAIL_PART_VALIDITY_ENCRYPTED) != 0)
|
|
e_attachment_set_encrypted (
|
|
attachment,
|
|
pair->validity->encrypt.status);
|
|
}
|
|
|
|
store = find_attachment_store (context->part_list, part);
|
|
if (store) {
|
|
GList *attachments = e_attachment_store_get_attachments (store);
|
|
if (!g_list_find (attachments, attachment)) {
|
|
e_attachment_store_add_attachment (
|
|
store, attachment);
|
|
}
|
|
g_list_free (attachments);
|
|
} else {
|
|
g_warning ("Failed to locate attachment-bar for %s", part_id);
|
|
}
|
|
|
|
g_object_unref (attachment);
|
|
}
|
|
|
|
registry = e_mail_formatter_get_extension_registry (formatter);
|
|
|
|
extensions = e_mail_extension_registry_get_for_mime_type (
|
|
registry, empa->snoop_mime_type);
|
|
if (extensions == NULL)
|
|
extensions = e_mail_extension_registry_get_fallback (
|
|
registry, empa->snoop_mime_type);
|
|
|
|
/* If the attachment is requested as RAW, then call the
|
|
* handler directly and do not append any other code. */
|
|
if ((context->mode == E_MAIL_FORMATTER_MODE_RAW) ||
|
|
(context->mode == E_MAIL_FORMATTER_MODE_PRINTING)) {
|
|
GList *head, *link;
|
|
gboolean success = FALSE;
|
|
|
|
if (extensions == NULL)
|
|
return FALSE;
|
|
|
|
if (context->mode == E_MAIL_FORMATTER_MODE_PRINTING) {
|
|
gchar *name;
|
|
EAttachment *attachment;
|
|
GFileInfo *file_info;
|
|
const gchar *display_name;
|
|
gchar *description;
|
|
|
|
attachment = e_mail_part_attachment_ref_attachment (
|
|
E_MAIL_PART_ATTACHMENT (part));
|
|
|
|
file_info = e_attachment_ref_file_info (attachment);
|
|
display_name = g_file_info_get_display_name (file_info);
|
|
|
|
description = e_attachment_dup_description (attachment);
|
|
if (description != NULL && *description != '\0') {
|
|
name = g_strdup_printf (
|
|
"<h2>Attachment: %s (%s)</h2>\n",
|
|
description, display_name);
|
|
} else {
|
|
name = g_strdup_printf (
|
|
"<h2>Attachment: %s</h2>\n",
|
|
display_name);
|
|
}
|
|
|
|
g_output_stream_write_all (
|
|
stream, name, strlen (name),
|
|
NULL, cancellable, NULL);
|
|
|
|
g_free (description);
|
|
g_free (name);
|
|
|
|
g_object_unref (attachment);
|
|
g_object_unref (file_info);
|
|
}
|
|
|
|
head = g_queue_peek_head_link (extensions);
|
|
|
|
for (link = head; link != NULL; link = g_list_next (link)) {
|
|
success = e_mail_formatter_extension_format (
|
|
E_MAIL_FORMATTER_EXTENSION (link->data),
|
|
formatter, context, part, stream, cancellable);
|
|
if (success)
|
|
break;
|
|
}
|
|
|
|
return success;
|
|
}
|
|
|
|
/* E_MAIL_FORMATTER_MODE_NORMAL: */
|
|
|
|
mime_part = e_mail_part_ref_mime_part (part);
|
|
text = e_mail_part_describe (mime_part, empa->snoop_mime_type);
|
|
flags = e_mail_formatter_get_text_format_flags (formatter);
|
|
html = camel_text_to_html (
|
|
text, flags & CAMEL_MIME_FILTER_TOHTML_CONVERT_URLS, 0);
|
|
g_free (text);
|
|
g_object_unref (mime_part);
|
|
|
|
if (empa->attachment_view_part_id)
|
|
attachment_part_id = empa->attachment_view_part_id;
|
|
else
|
|
attachment_part_id = part_id;
|
|
|
|
button_id = g_strconcat (attachment_part_id, ".attachment_button", NULL);
|
|
|
|
/* XXX Wild guess at the initial size. */
|
|
buffer = g_string_sized_new (8192);
|
|
|
|
g_string_append_printf (
|
|
buffer,
|
|
"<div class=\"attachment\">"
|
|
"<table width=\"100%%\" border=\"0\">"
|
|
"<tr valign=\"middle\">"
|
|
"<td align=\"left\" width=\"100\">"
|
|
"<object type=\"application/vnd.evolution.widget.attachment-button\" "
|
|
"height=\"20\" width=\"100\" data=\"%s\" id=\"%s\"></object>"
|
|
"</td>"
|
|
"<td align=\"left\">%s</td>"
|
|
"</tr>", part_id, button_id, html);
|
|
|
|
g_free (button_id);
|
|
g_free (html);
|
|
|
|
if (extensions != NULL) {
|
|
GOutputStream *content_stream;
|
|
gboolean success = FALSE;
|
|
|
|
content_stream = g_memory_output_stream_new_resizable ();
|
|
|
|
if (empa->attachment_view_part_id != NULL) {
|
|
EMailPart *attachment_view_part;
|
|
|
|
attachment_view_part = e_mail_part_list_ref_part (
|
|
context->part_list,
|
|
empa->attachment_view_part_id);
|
|
|
|
/* Avoid recursion. */
|
|
if (attachment_view_part == part)
|
|
g_clear_object (&attachment_view_part);
|
|
|
|
if (attachment_view_part != NULL) {
|
|
success = e_mail_formatter_format_as (
|
|
formatter, context,
|
|
attachment_view_part,
|
|
content_stream, NULL,
|
|
cancellable);
|
|
g_object_unref (attachment_view_part);
|
|
}
|
|
|
|
} else {
|
|
GList *head, *link;
|
|
|
|
head = g_queue_peek_head_link (extensions);
|
|
|
|
for (link = head; link != NULL; link = g_list_next (link)) {
|
|
success = e_mail_formatter_extension_format (
|
|
E_MAIL_FORMATTER_EXTENSION (link->data),
|
|
formatter, context,
|
|
part, content_stream,
|
|
cancellable);
|
|
if (success)
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (success) {
|
|
gchar *wrapper_element_id;
|
|
gconstpointer data;
|
|
gsize size;
|
|
|
|
wrapper_element_id = g_strconcat (
|
|
attachment_part_id, ".wrapper", NULL);
|
|
|
|
data = g_memory_output_stream_get_data (
|
|
G_MEMORY_OUTPUT_STREAM (content_stream));
|
|
size = g_memory_output_stream_get_data_size (
|
|
G_MEMORY_OUTPUT_STREAM (content_stream));
|
|
|
|
g_string_append_printf (
|
|
buffer,
|
|
"<tr><td colspan=\"2\">"
|
|
"<div class=\"attachment-wrapper\" id=\"%s\">",
|
|
wrapper_element_id);
|
|
|
|
g_string_append_len (buffer, data, size);
|
|
|
|
g_string_append (buffer, "</div></td></tr>");
|
|
|
|
g_free (wrapper_element_id);
|
|
}
|
|
|
|
g_object_unref (content_stream);
|
|
}
|
|
|
|
g_string_append (buffer, "</table></div>");
|
|
|
|
g_output_stream_write_all (
|
|
stream, buffer->str, buffer->len, NULL, cancellable, NULL);
|
|
|
|
g_string_free (buffer, TRUE);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static GtkWidget *
|
|
emfe_attachment_get_widget (EMailFormatterExtension *extension,
|
|
EMailPartList *context,
|
|
EMailPart *part,
|
|
GHashTable *params)
|
|
{
|
|
EAttachment *attachment;
|
|
EAttachmentStore *store;
|
|
EAttachmentView *view;
|
|
GtkWidget *widget;
|
|
const gchar *part_id;
|
|
|
|
g_return_val_if_fail (E_IS_MAIL_PART_ATTACHMENT (part), NULL);
|
|
|
|
attachment = e_mail_part_attachment_ref_attachment (
|
|
E_MAIL_PART_ATTACHMENT (part));
|
|
|
|
part_id = e_mail_part_get_id (part);
|
|
|
|
store = find_attachment_store (context, part);
|
|
widget = e_attachment_button_new ();
|
|
g_object_set_data_full (
|
|
G_OBJECT (widget),
|
|
"uri", g_strdup (part_id),
|
|
(GDestroyNotify) g_free);
|
|
e_attachment_button_set_attachment (
|
|
E_ATTACHMENT_BUTTON (widget), attachment);
|
|
|
|
view = g_object_get_data (G_OBJECT (store), "attachment-bar");
|
|
if (view != NULL)
|
|
e_attachment_button_set_view (
|
|
E_ATTACHMENT_BUTTON (widget), view);
|
|
|
|
gtk_widget_set_can_focus (widget, TRUE);
|
|
gtk_widget_show (widget);
|
|
|
|
g_object_unref (attachment);
|
|
|
|
return widget;
|
|
}
|
|
|
|
static void
|
|
e_mail_formatter_attachment_class_init (EMailFormatterExtensionClass *class)
|
|
{
|
|
class->display_name = _("Attachment");
|
|
class->description = _("Display as attachment");
|
|
class->mime_types = formatter_mime_types;
|
|
class->priority = G_PRIORITY_LOW;
|
|
class->format = emfe_attachment_format;
|
|
class->get_widget = emfe_attachment_get_widget;
|
|
}
|
|
|
|
static void
|
|
e_mail_formatter_attachment_init (EMailFormatterExtension *extension)
|
|
{
|
|
}
|