pjs/lib/libmime/mimemsg.c

593 строки
16 KiB
C

/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
*
* The contents of this file are subject to the Netscape Public License
* Version 1.0 (the "NPL"); you may not use this file except in
* compliance with the NPL. You may obtain a copy of the NPL at
* http://www.mozilla.org/NPL/
*
* Software distributed under the NPL is distributed on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the NPL
* for the specific language governing rights and limitations under the
* NPL.
*
* The Initial Developer of this code under the NPL is Netscape
* Communications Corporation. Portions created by Netscape are
* Copyright (C) 1998 Netscape Communications Corporation. All Rights
* Reserved.
*/
/* mimemsg.c --- definition of the MimeMessage class (see mimei.h)
Created: Jamie Zawinski <jwz@netscape.com>, 15-May-96.
*/
#include "rosetta.h"
#include "mimemsg.h"
#define MIME_SUPERCLASS mimeContainerClass
MimeDefClass(MimeMessage, MimeMessageClass, mimeMessageClass,
&MIME_SUPERCLASS);
static int MimeMessage_initialize (MimeObject *);
static void MimeMessage_finalize (MimeObject *);
static int MimeMessage_add_child (MimeObject *, MimeObject *);
static int MimeMessage_parse_begin (MimeObject *);
static int MimeMessage_parse_line (char *, int32, MimeObject *);
static int MimeMessage_parse_eof (MimeObject *, XP_Bool);
static int MimeMessage_close_headers (MimeObject *obj);
static int MimeMessage_write_headers_html (MimeObject *);
#ifndef MOZ_MAIL_COMPOSE
extern MimeObjectClass mimeFilterClass;
#endif /* MOZ_MAIL_COMPOSE */
#ifdef XP_UNIX
extern void MimeHeaders_do_unix_display_hook_hack(MimeHeaders *);
#endif /* XP_UNIX */
#if defined(DEBUG) && defined(XP_UNIX)
static int MimeMessage_debug_print (MimeObject *, FILE *, int32 depth);
#endif
extern MimeObjectClass mimeMultipartClass;
static int
MimeMessageClassInitialize(MimeMessageClass *class)
{
MimeObjectClass *oclass = (MimeObjectClass *) class;
MimeContainerClass *cclass = (MimeContainerClass *) class;
XP_ASSERT(!oclass->class_initialized);
oclass->initialize = MimeMessage_initialize;
oclass->finalize = MimeMessage_finalize;
oclass->parse_begin = MimeMessage_parse_begin;
oclass->parse_line = MimeMessage_parse_line;
oclass->parse_eof = MimeMessage_parse_eof;
cclass->add_child = MimeMessage_add_child;
#if defined(DEBUG) && defined(XP_UNIX)
oclass->debug_print = MimeMessage_debug_print;
#endif
return 0;
}
static int
MimeMessage_initialize (MimeObject *object)
{
return ((MimeObjectClass*)&MIME_SUPERCLASS)->initialize(object);
}
static void
MimeMessage_finalize (MimeObject *object)
{
MimeMessage *msg = (MimeMessage *)object;
if (msg->hdrs)
MimeHeaders_free(msg->hdrs);
msg->hdrs = 0;
((MimeObjectClass*)&MIME_SUPERCLASS)->finalize(object);
}
static int
MimeMessage_parse_begin (MimeObject *obj)
{
int status = ((MimeObjectClass*)&MIME_SUPERCLASS)->parse_begin(obj);
if (status < 0) return status;
/* Messages have separators before the headers, except for the outermost
message. */
return MimeObject_write_separator(obj);
}
static int
MimeMessage_parse_line (char *line, int32 length, MimeObject *obj)
{
MimeMessage *msg = (MimeMessage *) obj;
int status = 0;
XP_ASSERT(line && *line);
if (!line || !*line) return -1;
HG09869
/* If we already have a child object, then we're done parsing headers,
and all subsequent lines get passed to the inferior object without
further processing by us. (Our parent will stop feeding us lines
when this MimeMessage part is out of data.)
*/
if (msg->container.nchildren)
{
MimeObject *kid = msg->container.children[0];
XP_Bool nl;
XP_ASSERT(kid);
if (!kid) return -1;
/* Don't allow MimeMessage objects to not end in a newline, since it
would be inappropriate for any following part to appear on the same
line as the last line of the message.
#### This assumes that the only time the `parse_line' method is
called with a line that doesn't end in a newline is when that line
is the last line.
*/
nl = (length > 0 && (line[length-1] == CR || line[length-1] == LF));
#ifdef MIME_DRAFTS
if ( !mime_typep (kid, (MimeObjectClass*) &mimeMessageClass) &&
obj->options &&
obj->options->decompose_file_p &&
! obj->options->is_multipart_msg &&
obj->options->decompose_file_output_fn )
{
HG09870
{
status = obj->options->decompose_file_output_fn (line,
length,
obj->options->stream_closure);
if (status < 0) return status;
if (!nl) {
status = obj->options->decompose_file_output_fn (LINEBREAK,
LINEBREAK_LEN,
obj->options->stream_closure);
if (status < 0) return status;
}
return status;
}
}
#endif /* MIME_DRAFTS */
if (nl)
return kid->class->parse_buffer (line, length, kid);
else
{
/* Hack a newline onto the end. */
char *s = XP_ALLOC(length + LINEBREAK_LEN + 1);
if (!s) return MK_OUT_OF_MEMORY;
XP_MEMCPY(s, line, length);
XP_STRCPY(s + length, LINEBREAK);
status = kid->class->parse_buffer (s, length + LINEBREAK_LEN, kid);
XP_FREE(s);
return status;
}
}
/* Otherwise we don't yet have a child object, which means we're not
done parsing our headers yet.
*/
if (!msg->hdrs)
{
msg->hdrs = MimeHeaders_new();
if (!msg->hdrs) return MK_OUT_OF_MEMORY;
}
#ifdef MIME_DRAFTS
if ( obj->options &&
obj->options->decompose_file_p &&
! obj->options->is_multipart_msg &&
obj->options->done_parsing_outer_headers &&
obj->options->decompose_file_output_fn ) {
status = obj->options->decompose_file_output_fn ( line,
length,
obj->options->stream_closure );
if (status < 0) return status;
}
#endif /* MIME_DRAFTS */
status = MimeHeaders_parse_line(line, length, msg->hdrs);
if (status < 0) return status;
/* If this line is blank, we're now done parsing headers, and should
examine our content-type to create our "body" part.
*/
if (*line == CR || *line == LF)
{
status = MimeMessage_close_headers(obj);
if (status < 0) return status;
}
return 0;
}
static int
MimeMessage_close_headers (MimeObject *obj)
{
MimeMessage *msg = (MimeMessage *) obj;
int status = 0;
char *ct = 0; /* Content-Type header */
MimeObject *body;
if (msg->hdrs)
{
XP_Bool outer_p = !obj->headers; /* is this the outermost message? */
#ifdef MIME_DRAFTS
if (outer_p &&
obj->options &&
obj->options->decompose_file_p &&
obj->options->decompose_headers_info_fn)
{
HG09868
status = obj->options->decompose_headers_info_fn (
obj->options->stream_closure,
msg->hdrs );
}
#endif /* MIME_DRAFTS */
/* If this is the outermost message, we need to run the
`generate_header' callback. This happens here instead of
in `parse_begin', because it's only now that we've parsed
our headers. However, since this is the outermost message,
we have yet to write any HTML, so that's fine.
*/
if (outer_p &&
obj->output_p &&
obj->options &&
obj->options->write_html_p &&
obj->options->generate_header_html_fn)
{
int status = 0;
char *html = 0;
/* The generate_header_html_fn might return HTML, so it's important
that the output stream be set up with the proper type before we
make the MimeObject_write() call below. */
if (!obj->options->state->first_data_written_p)
{
status = MimeObject_output_init (obj, TEXT_HTML);
if (status < 0) return status;
XP_ASSERT(obj->options->state->first_data_written_p);
}
html = obj->options->generate_header_html_fn(NULL,
obj->options->html_closure,
msg->hdrs);
if (html)
{
status = MimeObject_write(obj, html, XP_STRLEN(html), FALSE);
XP_FREE(html);
if (status < 0) return status;
}
}
/* Find the content-type of the body of this message.
*/
{
XP_Bool ok = TRUE;
char *mv = MimeHeaders_get (msg->hdrs, HEADER_MIME_VERSION,
TRUE, FALSE);
#ifdef REQUIRE_MIME_VERSION_HEADER
/* If this is the outermost message, it must have a MIME-Version
header with the value 1.0 for us to believe what might be in
the Content-Type header. If the MIME-Version header is not
present, we must treat this message as untyped.
*/
ok = (mv && !XP_STRCMP(mv, "1.0"));
#else
/* #### actually, we didn't check this in Mozilla 2.0, and checking
it now could cause some compatibility nonsense, so for now, let's
just believe any Content-Type header we see.
*/
ok = TRUE;
#endif
if (ok)
{
ct = MimeHeaders_get (msg->hdrs, HEADER_CONTENT_TYPE, TRUE, FALSE);
/* If there is no Content-Type header, but there is a MIME-Version
header, then assume that this *is* in fact a MIME message.
(I've seen messages with
MIME-Version: 1.0
Content-Transfer-Encoding: quoted-printable
and no Content-Type, and we should treat those as being of type
MimeInlineTextPlain rather than MimeUntypedText.)
*/
if (mv && !ct)
ct = XP_STRDUP(TEXT_PLAIN);
}
FREEIF(mv); /* done with this now. */
}
HG09871
/* Emit the HTML for this message's headers. Do this before
creating the object representing the body.
*/
if (obj->output_p &&
obj->options &&
obj->options->write_html_p)
{
/* If citation headers are on, and this is not the outermost message,
turn them off. */
if (obj->options->headers == MimeHeadersCitation && !outer_p)
obj->options->headers = MimeHeadersSome;
/* Emit a normal header block. */
status = MimeMessage_write_headers_html(obj);
if (status < 0) return status;
}
else if (obj->output_p)
{
/* Dump the headers, raw. */
status = MimeObject_write(obj, "", 0, FALSE); /* initialize */
if (status < 0) return status;
HG09872
if (status < 0) return status;
}
#ifdef XP_UNIX
if (outer_p && obj->output_p)
/* Kludge from mimehdrs.c */
MimeHeaders_do_unix_display_hook_hack(msg->hdrs);
#endif /* XP_UNIX */
}
/* Never put out a separator after a message header block. */
if (obj->options && obj->options->state)
obj->options->state->separator_suppressed_p = TRUE;
#ifdef MIME_DRAFTS
if ( !obj->headers && /* outer most message header */
obj->options &&
obj->options->decompose_file_p &&
ct )
obj->options->is_multipart_msg = XP_STRSTR(ct, "multipart/") != NULL;
#endif /* MIME_DRAFTS */
body = mime_create(ct, msg->hdrs, obj->options);
FREEIF(ct);
if (!body) return MK_OUT_OF_MEMORY;
status = ((MimeContainerClass *) obj->class)->add_child (obj, body);
if (status < 0)
{
mime_free(body);
return status;
}
/* Now that we've added this new object to our list of children,
start its parser going. */
status = body->class->parse_begin(body);
if (status < 0) return status;
return 0;
}
static int
MimeMessage_parse_eof (MimeObject *obj, XP_Bool abort_p)
{
int status;
XP_Bool outer_p;
MimeMessage *msg = (MimeMessage *)obj;
if (obj->closed_p) return 0;
/* Run parent method first, to flush out any buffered data. */
status = ((MimeObjectClass*)&MIME_SUPERCLASS)->parse_eof(obj, abort_p);
if (status < 0) return status;
outer_p = !obj->headers; /* is this the outermost message? */
if (outer_p &&
obj->options &&
obj->options->write_html_p &&
obj->options->generate_footer_html_fn)
{
char *html =
obj->options->generate_footer_html_fn (NULL,
obj->options->html_closure,
msg->hdrs);
if (html)
{
int status = MimeObject_write(obj, html, XP_STRLEN(html), FALSE);
XP_FREE(html);
if (status < 0) return status;
}
}
#ifdef MIME_DRAFTS
if ( obj->options &&
obj->options->decompose_file_p &&
obj->options->done_parsing_outer_headers &&
! obj->options->is_multipart_msg &&
#ifndef MOZ_MAIL_COMPOSE
! mime_typep(obj, (MimeObjectClass*) &mimeFilterClass) &&
#endif /* MOZ_MAIL_COMPOSE */
obj->options->decompose_file_close_fn ) {
status = obj->options->decompose_file_close_fn (
obj->options->stream_closure );
if ( status < 0 ) return status;
}
#endif /* MIME_DRAFTS */
/* Put out a separator after every message/rfc822 object. */
if (!abort_p)
{
status = MimeObject_write_separator(obj);
if (status < 0) return status;
}
return 0;
}
static int
MimeMessage_add_child (MimeObject *parent, MimeObject *child)
{
MimeContainer *cont = (MimeContainer *) parent;
XP_ASSERT(parent && child);
if (!parent || !child) return -1;
/* message/rfc822 containers can only have one child. */
XP_ASSERT(cont->nchildren == 0);
if (cont->nchildren != 0) return -1;
#ifdef MIME_DRAFTS
if ( parent->options &&
parent->options->decompose_file_p &&
! parent->options->is_multipart_msg &&
#ifndef MOZ_MAIL_COMPOSE
! mime_typep(child, (MimeObjectClass*) &mimeFilterClass) &&
#endif /* MOZ_MAIL_COMPOSE */
parent->options->decompose_file_init_fn ) {
int status = 0;
status = parent->options->decompose_file_init_fn (
parent->options->stream_closure,
((MimeMessage*)parent)->hdrs );
if ( status < 0 ) return status;
}
#endif /* MIME_DRAFTS */
return ((MimeContainerClass*)&MIME_SUPERCLASS)->add_child (parent, child);
}
static int
MimeMessage_write_headers_html (MimeObject *obj)
{
MimeMessage *msg = (MimeMessage *) obj;
XP_Bool altform_p;
int status;
if (!obj->options ||
!obj->options->output_fn)
return 0;
XP_ASSERT(obj->output_p && obj->options->write_html_p);
if (!obj->options->state->first_data_written_p)
{
status = MimeObject_output_init (obj, TEXT_HTML);
if (status < 0) return status;
XP_ASSERT(obj->options->state->first_data_written_p);
}
HG09873
HG43771
status = MimeHeaders_write_headers_html (msg->hdrs, obj->options);
if (status < 0) return status;
if (HG56232)
{
HG78388
}
else
{
HG80200
if (obj->options &&
obj->options->state &&
obj->options->generate_post_header_html_fn &&
!obj->options->state->post_header_html_run_p)
{
char *html = 0;
XP_ASSERT(obj->options->state->first_data_written_p);
html = obj->options->generate_post_header_html_fn(NULL,
obj->options->html_closure,
msg->hdrs);
obj->options->state->post_header_html_run_p = TRUE;
if (html)
{
status = MimeObject_write(obj, html, XP_STRLEN(html), FALSE);
XP_FREE(html);
if (status < 0) return status;
}
}
/* Write out a paragraph break between the headers and body. */
{
char s[] = "<P>";
status = MimeObject_write(obj, s, XP_STRLEN(s), FALSE);
if (status < 0) return status;
}
}
return 0;
}
#if defined(DEBUG) && defined(XP_UNIX)
static int
MimeMessage_debug_print (MimeObject *obj, FILE *stream, int32 depth)
{
MimeMessage *msg = (MimeMessage *) obj;
char *addr = mime_part_address(obj);
int i;
for (i=0; i < depth; i++)
fprintf(stream, " ");
fprintf(stream, "<%s %s%s 0x%08X>\n",
obj->class->class_name,
addr ? addr : "???",
(msg->container.nchildren == 0 ? " (no body)" : ""),
(uint32) msg);
FREEIF(addr);
#if 0
if (msg->hdrs)
{
char *s;
depth++;
# define DUMP(HEADER) \
for (i=0; i < depth; i++) \
fprintf(stream, " "); \
s = MimeHeaders_get (msg->hdrs, HEADER, FALSE, TRUE); \
fprintf(stream, HEADER ": %s\n", s ? s : ""); \
FREEIF(s)
DUMP(HEADER_SUBJECT);
DUMP(HEADER_DATE);
DUMP(HEADER_FROM);
DUMP(HEADER_TO);
/* DUMP(HEADER_CC); */
DUMP(HEADER_NEWSGROUPS);
DUMP(HEADER_MESSAGE_ID);
# undef DUMP
fprintf(stream, "\n");
}
#endif
XP_ASSERT(msg->container.nchildren <= 1);
if (msg->container.nchildren == 1)
{
MimeObject *kid = msg->container.children[0];
int status = kid->class->debug_print (kid, stream, depth+1);
if (status < 0) return status;
}
return 0;
}
#endif