Permalink
Fetching contributors…
Cannot retrieve contributors at this time
264 lines (225 sloc) 7.07 KB
/*
* Copyright © 2009 Keith Packard <keithp@keithp.com>
* Copyright © 2010 Michal Sojka <sojkam1@fel.cvut.cz>
*
* 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 3 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.
*/
#include "gmime-filter-headers.h"
#include <string.h>
#include <gmime/gmime-utils.h>
#include <glib/gprintf.h>
#include <stdlib.h>
#include <xutil.h>
/**
* SECTION: gmime-filter-headers
* @title: GMimeFilterHeaders
* @short_description: Add/remove headers markers
*
* A #GMimeFilter for decoding rfc2047 encoded headers to UTF-8
**/
static void g_mime_filter_headers_class_init (GMimeFilterHeadersClass *klass);
static void g_mime_filter_headers_init (GMimeFilterHeaders *filter, GMimeFilterHeadersClass *klass);
static void g_mime_filter_headers_finalize (GObject *object);
static GMimeFilter *filter_copy (GMimeFilter *filter);
static void filter_filter (GMimeFilter *filter, char *in, size_t len, size_t prespace,
char **out, size_t *outlen, size_t *outprespace);
static void filter_complete (GMimeFilter *filter, char *in, size_t len, size_t prespace,
char **out, size_t *outlen, size_t *outprespace);
static void filter_reset (GMimeFilter *filter);
static GMimeFilterClass *parent_class = NULL;
GType
g_mime_filter_headers_get_type (void)
{
static GType type = 0;
if (!type) {
static const GTypeInfo info = {
sizeof (GMimeFilterHeadersClass),
NULL, /* base_class_init */
NULL, /* base_class_finalize */
(GClassInitFunc) g_mime_filter_headers_class_init,
NULL, /* class_finalize */
NULL, /* class_data */
sizeof (GMimeFilterHeaders),
0, /* n_preallocs */
(GInstanceInitFunc) g_mime_filter_headers_init,
NULL /* value_table */
};
type = g_type_register_static (GMIME_TYPE_FILTER, "GMimeFilterHeaders", &info, (GTypeFlags) 0);
}
return type;
}
static void
g_mime_filter_headers_class_init (GMimeFilterHeadersClass *klass)
{
GObjectClass *object_class = G_OBJECT_CLASS (klass);
GMimeFilterClass *filter_class = GMIME_FILTER_CLASS (klass);
parent_class = (GMimeFilterClass *) g_type_class_ref (GMIME_TYPE_FILTER);
object_class->finalize = g_mime_filter_headers_finalize;
filter_class->copy = filter_copy;
filter_class->filter = filter_filter;
filter_class->complete = filter_complete;
filter_class->reset = filter_reset;
}
static void
g_mime_filter_headers_init (GMimeFilterHeaders *filter, GMimeFilterHeadersClass *klass)
{
(void) klass;
filter->saw_nl = TRUE;
filter->line = NULL;
filter->line_size = 0;
filter->lineptr = NULL;
}
static void
g_mime_filter_headers_finalize (GObject *object)
{
free (GMIME_FILTER_HEADERS (object)->line);
G_OBJECT_CLASS (parent_class)->finalize (object);
}
static GMimeFilter *
filter_copy (GMimeFilter *filter)
{
(void) filter;
return g_mime_filter_headers_new ();
}
static void
output_decoded_header (GMimeFilterHeaders *headers, char **outptr)
{
char *colon, *name, *s, *decoded_value;
size_t offset;
gint ret;
colon = strchr (headers->line, ':');
if (colon == NULL)
return;
name = headers->line;
*colon = '\0';
s = colon + 1;
while (*s == ' ' || *s == '\t')
s++;
decoded_value = g_mime_utils_header_decode_text(s);
if (decoded_value == NULL)
return;
offset = *outptr - GMIME_FILTER (headers)->outbuf;
g_mime_filter_set_size (GMIME_FILTER (headers), strlen(name) + 2 +
strlen(decoded_value) + 2, TRUE);
*outptr = GMIME_FILTER (headers)->outbuf + offset;
ret = g_sprintf (*outptr, "%s: %s\n", name, decoded_value);
if (ret > 0)
*outptr += ret;
free (decoded_value);
}
static void
output_final_newline (GMimeFilterHeaders *headers, char **outptr)
{
size_t offset;
offset = *outptr - GMIME_FILTER (headers)->outbuf;
g_mime_filter_set_size (GMIME_FILTER (headers), 1, TRUE);
*outptr = GMIME_FILTER (headers)->outbuf + offset;
*(*outptr)++ = '\n';
}
static void
filter_filter (GMimeFilter *filter, char *inbuf, size_t inlen, size_t prespace,
char **outbuf, size_t *outlen, size_t *outprespace)
{
GMimeFilterHeaders *headers = (GMimeFilterHeaders *) filter;
register const char *inptr = inbuf;
const char *inend = inbuf + inlen;
char *lineptr, *lineend, *outptr;
(void) prespace;
if (headers->line == NULL) {
headers->line_size = 200;
headers->lineptr = headers->line = malloc (headers->line_size);
}
lineptr = headers->lineptr;
lineend = headers->line + headers->line_size - 1;
if (lineptr == NULL)
return;
outptr = filter->outbuf;
while (inptr < inend) {
if (*inptr == '\n') {
if (headers->saw_nl)
output_final_newline(headers, &outptr);
headers->saw_nl = TRUE;
inptr++;
continue;
}
if (lineptr == lineend) {
headers->line_size *= 2;
headers->line = xrealloc (headers->line, headers->line_size);
lineptr = headers->line + (headers->line_size / 2) - 1;
lineend = headers->line + headers->line_size - 1;
}
if (headers->saw_nl && *inptr != ' ' && *inptr != '\t') {
*lineptr = '\0';
output_decoded_header (headers, &outptr);
lineptr = headers->line;
}
if (headers->saw_nl && (*inptr == ' ' || *inptr == '\t')) {
*lineptr = ' ';
lineptr++;
while (inptr < inend && (*inptr == ' ' || *inptr == '\t'))
inptr++;
headers->saw_nl = FALSE;
continue;
}
headers->saw_nl = FALSE;
if (*inptr != '\r')
*lineptr++ = *inptr;
inptr++;
}
if (headers->saw_nl) {
*lineptr = '\0';
output_decoded_header (headers, &outptr);
lineptr = headers->line;
}
headers->lineptr = lineptr;
*outlen = outptr - filter->outbuf;
*outprespace = filter->outpre;
*outbuf = filter->outbuf;
}
static void
filter_complete (GMimeFilter *filter, char *inbuf, size_t inlen, size_t prespace,
char **outbuf, size_t *outlen, size_t *outprespace)
{
if (inbuf && inlen)
filter_filter (filter, inbuf, inlen, prespace, outbuf, outlen, outprespace);
}
static void
filter_reset (GMimeFilter *filter)
{
GMimeFilterHeaders *headers = (GMimeFilterHeaders *) filter;
headers->saw_nl = TRUE;
free(headers->line);
headers->line = NULL;
headers->line_size = 0;
}
/**
* g_mime_filter_headers_new:
* @encode: %TRUE if the filter should encode or %FALSE otherwise
* @dots: encode/decode dots (as for SMTP)
*
* Creates a new #GMimeFilterHeaders filter.
*
* If @encode is %TRUE, then all lines will be prefixed by "> ",
* otherwise any lines starting with "> " will have that removed
*
* Returns: a new #GMimeFilterHeaders filter.
**/
GMimeFilter *
g_mime_filter_headers_new (void)
{
GMimeFilterHeaders *new_headers;
new_headers = (GMimeFilterHeaders *) g_object_newv (GMIME_TYPE_FILTER_HEADERS, 0, NULL);
return (GMimeFilter *) new_headers;
}