/* -*- 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 , 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[] = "

"; 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