From 1c63ec7031926fcd06ac3293d9e402af069ea6ac Mon Sep 17 00:00:00 2001 From: Carl Worth Date: Wed, 14 Oct 2009 12:48:58 -0700 Subject: [PATCH] notmuch-index-message: Correctly parse and index encoded mime parts. This cleans up some old code that was very ugly, (separately opening the mail file and seeking to the end of the headers to parse the body). I knew gmime must have had support for transparently decoding mime content, but I just couldn't find it previously. Note: Multipart and MultipartSigned parts are not handled yet. Things are quite happy now. The few differences I see with sup are: 1. sup forces email address domains to lowercase, (I don't think I care) 2. sup and notmuch disagree on ordering of multiple thread_id values (another thing that's of no concern) We are still doing one thing wrong when a message belongs to multiple threads. We've got a nice comma-separated thread-value just like sup, but then we're also putting in a comma-separated thread-term where sup does multiple thread terms. That should be an easy fix. Beyond that, sup and notmuch are still disagreeing on the term lists for some messages, (I think attachment vs. inline content-disposition is at least one piece of this). But there are likley still differences in the heuristics for which chunks of the message body to index. I'll be looking into this more. --- notmuch-index-message.cc | 130 +++++++++++++++++++-------------------- 1 file changed, 65 insertions(+), 65 deletions(-) diff --git a/notmuch-index-message.cc b/notmuch-index-message.cc index 0aa374c8..f7ea340f 100644 --- a/notmuch-index-message.cc +++ b/notmuch-index-message.cc @@ -333,89 +333,90 @@ parse_references (GPtrArray *array, } } -/* Generate terms for the body of a message, given the filename of the - * message and the offset at which the headers of the message end, - * (and hence the body begins). */ +/* Given a string representing the body of a message, generate terms + * for it, (skipping quoted portions and signatures). */ static void -gen_terms_body (Xapian::TermGenerator term_gen, - const char * filename, - gint64 body_offset) +gen_terms_body_str (Xapian::TermGenerator term_gen, + char *body) { - GIOChannel *channel; - GIOStatus gio_status; - GError *error = NULL; - char *p, *body_line = NULL, *prev_line = NULL; + char *line, *line_end, *next_line; - channel = g_io_channel_new_file (filename, "r", &error); - if (channel == NULL) { - fprintf (stderr, "Error: %s\n", error->message); - exit (1); - } + if (body == NULL) + return; - gio_status = g_io_channel_seek_position (channel, body_offset, - G_SEEK_SET, &error); - if (gio_status != G_IO_STATUS_NORMAL) { - fprintf (stderr, "Error: %s\n", error->message); - exit (1); - } + next_line = body; while (1) { - if (body_line) - g_free (body_line); - - gio_status = g_io_channel_read_line (channel, &body_line, - NULL, NULL, &error); - if (gio_status == G_IO_STATUS_EOF) + line = next_line; + if (*line == '\0') break; - if (gio_status != G_IO_STATUS_NORMAL) { - fprintf (stderr, "Error: %s\n", error->message); - exit (1); + + next_line = strchr (line, '\n'); + if (next_line == NULL) { + next_line = line + strlen (line); } + line_end = next_line - 1; - if (strlen (body_line) == 0) - continue; + /* Trim whitespace. */ + while (*next_line && isspace (*next_line)) + next_line++; - /* If the line looks like it might be introducing a quote, - * save it until we see if the next line begins a quote. */ - p = body_line + strlen (body_line) - 1; - while (p > body_line and isspace (*p)) - p--; - if (*p == ':') { - prev_line = body_line; - body_line = NULL; + /* Skip lines that are quotes. */ + if (*line == '>') continue; - } - /* Skip quoted lines, (and previous lines that introduced them) */ - if (body_line[0] == '>') { - if (prev_line) { - g_free (prev_line); - prev_line = NULL; - } - continue; - } + /* Also skip lines introducing a quote on the next line. */ + while (line_end > line && isspace (*line_end)) + line_end--; - /* Now that we're not looking at a quote we can add the prev_line */ - if (prev_line) { - gen_terms (term_gen, "body", prev_line); - g_free (prev_line); - prev_line = NULL; - } + if (*line_end == ':' && *next_line == '>') + continue; - /* Skip signatures */ + /* Finally, bail as soon as we see a signature. */ /* XXX: Should only do this if "near" the end of the message. */ - if (strncmp (body_line, "-- ", 3) == 0 || - strncmp (body_line, "----------", 10) == 0 || - strncmp (body_line, "__________", 10) == 0) + if (strncmp (line, "-- ", 3) == 0 || + strncmp (line, "----------", 10) == 0 || + strncmp (line, "__________", 10) == 0) break; - gen_terms (term_gen, "body", body_line); + *(line_end + 1) = '\0'; + gen_terms (term_gen, "body", line); } +} + + +/* Callback to generate terms for each mime part of a message. */ +static void +gen_terms_part (GMimeObject *parent, + GMimeObject *part, + gpointer user_data) +{ + Xapian::TermGenerator *term_gen = (Xapian::TermGenerator *) user_data; + GMimeStream *stream; + GMimeDataWrapper *wrapper; + GByteArray *byte_array; + char *body; + + if (! GMIME_IS_PART (part)) { + fprintf (stderr, "Warning: Not indexing unknown mime part: %s.\n", + g_type_name (G_OBJECT_TYPE (part))); + return; + } + + byte_array = g_byte_array_new (); + + stream = g_mime_stream_mem_new_with_byte_array (byte_array); + g_mime_stream_mem_set_owner (GMIME_STREAM_MEM (stream), FALSE); + wrapper = g_mime_part_get_content_object (GMIME_PART (part)); + g_mime_data_wrapper_write_to_stream (wrapper, stream); + + g_object_unref (stream); + + body = (char *) g_byte_array_free (byte_array, FALSE); - if (body_line) - g_free (body_line); + gen_terms_body_str (*term_gen, body); - g_io_channel_close (channel); + free (body); } static void @@ -474,8 +475,7 @@ index_file (Xapian::WritableDatabase db, gen_terms (term_gen, "subject", subject); gen_terms (term_gen, "body", subject); - gen_terms_body (term_gen, filename, - g_mime_parser_get_headers_end (parser)); + g_mime_message_foreach (message, gen_terms_part, &term_gen); parents = g_ptr_array_new (); -- 2.26.2