/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
/* camel-multipart.c : Abstract class for a multipart */
#ifndef NO_WARNINGS
#warning This should be a mostly abstract class, but it is not!
#endif
/*
*
* Author :
* Bertrand Guiheneuf <bertrand@helixcode.com>
*
* Copyright 1999, 2000 Helix Code, Inc. (http://www.helixcode.com)
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License as
* published by the Free Software Foundation; either version 2 of the
* License, or (at your option) any later version.
*
* 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 General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
* USA
*/
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <string.h> /* strlen() */
#include <unistd.h> /* for getpid */
#include <time.h> /* for time */
#include "camel-stream-mem.h"
#include "camel-multipart.h"
#include "camel-mime-part.h"
#include "camel-exception.h"
#include "md5-utils.h"
#define d(x)
static void add_part (CamelMultipart *multipart,
CamelMimePart *part);
static void add_part_at (CamelMultipart *multipart,
CamelMimePart *part,
guint index);
static void remove_part (CamelMultipart *multipart,
CamelMimePart *part);
static CamelMimePart * remove_part_at (CamelMultipart *multipart,
guint index);
static CamelMimePart * get_part (CamelMultipart *multipart,
guint index);
static guint get_number (CamelMultipart *multipart);
static void set_boundary (CamelMultipart *multipart,
gchar *boundary);
static const gchar * get_boundary (CamelMultipart *multipart);
static int write_to_stream (CamelDataWrapper *data_wrapper,
CamelStream *stream);
static void unref_part (gpointer data, gpointer user_data);
static CamelDataWrapperClass *parent_class = NULL;
/* Returns the class for a CamelMultipart */
#define CMP_CLASS(so) CAMEL_MULTIPART_CLASS (CAMEL_OBJECT_GET_CLASS(so))
/* Returns the class for a CamelDataWrapper */
#define CDW_CLASS(so) CAMEL_DATA_WRAPPER_CLASS (CAMEL_OBJECT_GET_CLASS(so))
static void
camel_multipart_class_init (CamelMultipartClass *camel_multipart_class)
{
CamelDataWrapperClass *camel_data_wrapper_class =
CAMEL_DATA_WRAPPER_CLASS (camel_multipart_class);
parent_class = CAMEL_DATA_WRAPPER_CLASS (camel_type_get_global_classfuncs (camel_data_wrapper_get_type ()));
/* virtual method definition */
camel_multipart_class->add_part = add_part;
camel_multipart_class->add_part_at = add_part_at;
camel_multipart_class->remove_part = remove_part;
camel_multipart_class->remove_part_at = remove_part_at;
camel_multipart_class->get_part = get_part;
camel_multipart_class->get_number = get_number;
camel_multipart_class->set_boundary = set_boundary;
camel_multipart_class->get_boundary = get_boundary;
/* virtual method overload */
camel_data_wrapper_class->write_to_stream = write_to_stream;
}
static void
camel_multipart_init (gpointer object, gpointer klass)
{
CamelMultipart *multipart = CAMEL_MULTIPART (object);
camel_data_wrapper_set_mime_type (CAMEL_DATA_WRAPPER (multipart),
"multipart/mixed");
multipart->preface = NULL;
multipart->postface = NULL;
}
static void
camel_multipart_finalize (CamelObject *object)
{
CamelMultipart *multipart = CAMEL_MULTIPART (object);
g_list_foreach (multipart->parts, unref_part, NULL);
if (multipart->boundary)
g_free (multipart->boundary);
if (multipart->preface)
g_free (multipart->preface);
if (multipart->postface)
g_free (multipart->postface);
}
CamelType
camel_multipart_get_type (void)
{
static CamelType camel_multipart_type = CAMEL_INVALID_TYPE;
if (camel_multipart_type == CAMEL_INVALID_TYPE) {
camel_multipart_type = camel_type_register (camel_data_wrapper_get_type (), "CamelMultipart",
sizeof (CamelMultipart),
sizeof (CamelMultipartClass),
(CamelObjectClassInitFunc) camel_multipart_class_init,
NULL,
(CamelObjectInitFunc) camel_multipart_init,
(CamelObjectFinalizeFunc) camel_multipart_finalize);
}
return camel_multipart_type;
}
static void
unref_part (gpointer data, gpointer user_data)
{
CamelObject *part = CAMEL_OBJECT (data);
camel_object_unref (part);
}
/**
* camel_multipart_new:
*
* Create a new CamelMultipart object.
*
* Return value: a new CamelMultipart
**/
CamelMultipart *
camel_multipart_new (void)
{
CamelMultipart *multipart;
multipart = (CamelMultipart *)camel_object_new (CAMEL_MULTIPART_TYPE);
multipart->preface = NULL;
multipart->postface = NULL;
return multipart;
}
static void
add_part (CamelMultipart *multipart, CamelMimePart *part)
{
multipart->parts = g_list_append (multipart->parts, part);
camel_object_ref (CAMEL_OBJECT (part));
}
/**
* camel_multipart_add_part:
* @multipart: a CamelMultipart
* @part: the part to add
*
* Appends the part to the multipart object.
**/
void
camel_multipart_add_part (CamelMultipart *multipart, CamelMimePart *part)
{
g_return_if_fail (CAMEL_IS_MULTIPART (multipart));
g_return_if_fail (CAMEL_IS_MIME_PART (part));
CMP_CLASS (multipart)->add_part (multipart, part);
}
static void
add_part_at (CamelMultipart *multipart, CamelMimePart *part, guint index)
{
multipart->parts = g_list_insert (multipart->parts, part, index);
camel_object_ref (CAMEL_OBJECT (part));
}
/**
* camel_multipart_add_part_at:
* @multipart: a CamelMultipart
* @part: the part to add
* @index: index to add the multipart at
*
* Adds the part to the multipart object after the @index'th
* element. If @index is greater than the number of parts, it is
* equivalent to camel_multipart_add_part().
**/
void
camel_multipart_add_part_at (CamelMultipart *multipart,
CamelMimePart *part, guint index)
{
g_return_if_fail (CAMEL_IS_MULTIPART (multipart));
g_return_if_fail (CAMEL_IS_MIME_PART (part));
CMP_CLASS (multipart)->add_part_at (multipart, part, index);
}
static void
remove_part (CamelMultipart *multipart, CamelMimePart *part)
{
if (!multipart->parts)
return;
multipart->parts = g_list_remove (multipart->parts, part);
camel_object_unref (CAMEL_OBJECT (part));
}
/**
* camel_multipart_remove_part:
* @multipart: a CamelMultipart
* @part: the part to remove
*
* Removes @part from @multipart.
**/
void
camel_multipart_remove_part (CamelMultipart *multipart,
CamelMimePart *part)
{
g_return_if_fail (CAMEL_IS_MULTIPART (multipart));
g_return_if_fail (CAMEL_IS_MIME_PART (part));
CMP_CLASS (multipart)->remove_part (multipart, part);
}
static CamelMimePart *
remove_part_at (CamelMultipart *multipart, guint index)
{
GList *parts_list;
GList *part_to_remove;
CamelMimePart *removed_part;
if (!(multipart->parts))
return NULL;
parts_list = multipart->parts;
part_to_remove = g_list_nth (parts_list, index);
if (!part_to_remove) {
g_warning ("CamelMultipart::remove_part_at: "
"part to remove is NULL\n");
return NULL;
}
removed_part = CAMEL_MIME_PART (part_to_remove->data);
multipart->parts = g_list_remove_link (parts_list, part_to_remove);
if (part_to_remove->data)
camel_object_unref (CAMEL_OBJECT (part_to_remove->data));
g_list_free_1 (part_to_remove);
return removed_part;
}
/**
* camel_multipart_remove_part_at:
* @multipart: a CamelMultipart
* @index: a zero-based index indicating the part to remove
*
* Remove the indicated part from the multipart object.
*
* Return value: the removed part. Note that it is camel_object_unref()ed
* before being returned, which may cause it to be destroyed.
**/
CamelMimePart *
camel_multipart_remove_part_at (CamelMultipart *multipart, guint index)
{
g_return_val_if_fail (CAMEL_IS_MULTIPART (multipart), NULL);
return CMP_CLASS (multipart)->remove_part_at (multipart, index);
}
static CamelMimePart *
get_part (CamelMultipart *multipart, guint index)
{
GList *part;
if (!(multipart->parts))
return NULL;
part = g_list_nth (multipart->parts, index);
if (part)
return CAMEL_MIME_PART (part->data);
else
return NULL;
}
/**
* camel_multipart_get_part:
* @multipart: a CamelMultipart
* @index: a zero-based index indicating the part to get
*
* Return value: the indicated subpart, or %NULL
**/
CamelMimePart *
camel_multipart_get_part (CamelMultipart *multipart, guint index)
{
g_return_val_if_fail (CAMEL_IS_MULTIPART (multipart), NULL);
return CMP_CLASS (multipart)->get_part (multipart, index);
}
static guint
get_number (CamelMultipart *multipart)
{
return g_list_length (multipart->parts);
}
/**
* camel_multipart_get_number:
* @multipart: a CamelMultipart
*
* Return value: the number of subparts in @multipart
**/
guint
camel_multipart_get_number (CamelMultipart *multipart)
{
g_return_val_if_fail (CAMEL_IS_MULTIPART (multipart), 0);
return CMP_CLASS (multipart)->get_number (multipart);
}
static void
set_boundary (CamelMultipart *multipart, gchar *boundary)
{
CamelDataWrapper *cdw = CAMEL_DATA_WRAPPER (multipart);
char *bgen, digest[16], bbuf[27], *p;
int state, save;
g_return_if_fail (cdw->mime_type != NULL);
if (!boundary) {
/* Generate a fairly random boundary string. */
bgen = g_strdup_printf ("%p:%lu:%lu", multipart,
(unsigned long) getpid(),
(unsigned long) time(0));
md5_get_digest (bgen, strlen (bgen), digest);
g_free (bgen);
strcpy (bbuf, "=-");
p = bbuf + 2;
state = save = 0;
p += base64_encode_step (digest, 16, FALSE, p, &state, &save);
*p = '\0';
boundary = bbuf;
}
header_content_type_set_param (cdw->mime_type, "boundary", boundary);
}
/**
* camel_multipart_set_boundary:
* @multipart: a CamelMultipart
* @boundary: the message boundary, or %NULL
*
* Sets the message boundary for @multipart to @boundary. This should
* be a string which does not occur anywhere in any of @multipart's
* subparts. If @boundary is %NULL, a randomly-generated boundary will
* be used.
**/
void
camel_multipart_set_boundary (CamelMultipart *multipart, gchar *boundary)
{
g_return_if_fail (CAMEL_IS_MULTIPART (multipart));
CMP_CLASS (multipart)->set_boundary (multipart, boundary);
}
static const gchar *
get_boundary (CamelMultipart *multipart)
{
CamelDataWrapper *cdw = CAMEL_DATA_WRAPPER (multipart);
g_return_val_if_fail (cdw->mime_type != NULL, NULL);
return header_content_type_param (cdw->mime_type, "boundary");
}
/**
* camel_multipart_get_boundary:
* @multipart: a CamelMultipart
*
* Return value: @multipart's message boundary
**/
const gchar *
camel_multipart_get_boundary (CamelMultipart *multipart)
{
return CMP_CLASS (multipart)->get_boundary (multipart);
}
/* this is MIME specific, doesn't belong here really */
static int
write_to_stream (CamelDataWrapper *data_wrapper, CamelStream *stream)
{
CamelMultipart *multipart = CAMEL_MULTIPART (data_wrapper);
const gchar *boundary;
int total = 0;
int count;
GList *node;
/* get the bundary text */
boundary = camel_multipart_get_boundary (multipart);
/* we cannot write a multipart without a boundary string */
g_return_val_if_fail (boundary && *boundary, -1);
/*
* write the preface text (usually something like
* "This is a mime message, if you see this, then
* your mail client probably doesn't support ...."
*/
if (multipart->preface) {
count = camel_stream_write_string (stream, multipart->preface);
if (count == -1)
return -1;
total += count;
}
/*
* Now, write all the parts, separated by the boundary
* delimiter
*/
node = multipart->parts;
while (node) {
count = camel_stream_printf (stream, "\n--%s\n", boundary);
if (count == -1)
return -1;
total += count;
count = camel_data_wrapper_write_to_stream (CAMEL_DATA_WRAPPER (node->data), stream);
if (count == -1)
return -1;
total += count;
node = node->next;
}
/* write the terminating boudary delimiter */
count = camel_stream_printf (stream, "\n--%s--\n", boundary);
if (count == -1)
return -1;
total += count;
/* and finally the postface */
if (multipart->postface) {
count = camel_stream_write_string (stream, multipart->postface);
if (count == -1)
return -1;
total += count;
}
return total;
}
/**
* camel_multipart_set_preface:
* @multipart:
* @preface:
*
* Set the preface text for this multipart. Will be written out infront
* of the multipart. This text should only include US-ASCII strings, and
* be relatively short, and will be ignored by any MIME mail client.
**/
void
camel_multipart_set_preface(CamelMultipart *multipart, const char *preface)
{
if (multipart->preface != preface) {
g_free(multipart->preface);
if (preface)
multipart->preface = g_strdup(preface);
else
multipart->preface = NULL;
}
}
/**
* camel_multipart_set_postface:
* @multipart:
* @postface:
*
* Set the postfix text for this multipart. Will be written out after
* the last boundary of the multipart, and ignored by any MIME mail
* client.
*
* Generally postface texts should not be sent with multipart messages.
**/
void
camel_multipart_set_postface(CamelMultipart *multipart, const char *postface)
{
if (multipart->postface != postface) {
g_free(multipart->postface);
if (postface)
multipart->postface = g_strdup(postface);
else
multipart->postface = NULL;
}
}
