src/lib-mail/message-size.c

	message-size.c revision c31661baf87b506c2a90a1fca02968cf82e448bb
/* Copyright (c) 2002-2009 Dovecot authors, see the included COPYING file */

#include "lib.h"
#include "istream.h"
#include "message-parser.h"
#include "message-size.h"

int message_get_header_size(struct istream *input, struct message_size *hdr,
                bool *has_nuls)
{
    const unsigned char *msg;
    size_t i, size, startpos, missing_cr_count;
    int ret = 0;

    memset(hdr, 0, sizeof(struct message_size));
    if (has_nuls != NULL)
        *has_nuls = FALSE;

    missing_cr_count = 0; startpos = 0;
    while (i_stream_read_data(input, &msg, &size, startpos) > 0) {
        for (i = startpos; i < size; i++) {
            if (msg[i] != '\n') {
                if (msg[i] == '\0' && has_nuls != NULL)
                    *has_nuls = TRUE;
                continue;
            }

            hdr->lines++;
            if (i == 0 || msg[i-1] != '\r') {
                /* missing CR */
                missing_cr_count++;
            }

            if (i == 0 || (i == 1 && msg[i-1] == '\r')) {
                /* no headers at all */
                break;
            }

            if ((i > 0 && msg[i-1] == '\n') ||
                (i > 1 && msg[i-2] == '\n' && msg[i-1] == '\r')) {
                /* \n\n or \n\r\n - end of headers */
                break;
            }
        }

        if (i < size) {
            /* end of header */
            startpos = i+1;
            break;
        }

        /* leave the last two characters, they may be \r\n */
        startpos = size == 1 ? 1 : 2;
        i_stream_skip(input, i - startpos);

        hdr->physical_size += i - startpos;
    }
    if (input->stream_errno != 0)
        ret = -1;
    i_stream_skip(input, startpos);
    hdr->physical_size += startpos;

    hdr->virtual_size = hdr->physical_size + missing_cr_count;
    i_assert(hdr->virtual_size >= hdr->physical_size);
    return ret;
}

int message_get_body_size(struct istream *input, struct message_size *body,
              bool *has_nuls)
{
    const unsigned char *msg;
    size_t i, size, missing_cr_count;
    int ret = 0;

    memset(body, 0, sizeof(struct message_size));
    if (has_nuls != NULL)
        *has_nuls = FALSE;

    missing_cr_count = 0;
    if ((ret = i_stream_read_data(input, &msg, &size, 0)) <= 0)
        return ret < 0 && input->stream_errno != 0 ? -1 : 0;

    if (msg[0] == '\n')
        missing_cr_count++;

    do {
        for (i = 1; i < size; i++) {
            if (msg[i] > '\n')
                continue;

            if (msg[i] == '\n') {
                if (msg[i-1] != '\r') {
                    /* missing CR */
                    missing_cr_count++;
                }

                /* increase after making sure we didn't break
                   at virtual \r */
                body->lines++;
            } else if (msg[i] == '\0') {
                if (has_nuls != NULL)
                    *has_nuls = TRUE;
            }
        }

        /* leave the last character, it may be \r */
        i_stream_skip(input, i - 1);
        body->physical_size += i - 1;
    } while (i_stream_read_data(input, &msg, &size, 1) > 0);

    if (input->stream_errno != 0)
        ret = -1;

    i_stream_skip(input, 1);
    body->physical_size++;

    body->virtual_size = body->physical_size + missing_cr_count;
    i_assert(body->virtual_size >= body->physical_size);
    return ret;
}

void message_size_add(struct message_size *dest,
              const struct message_size *src)
{
    dest->virtual_size += src->virtual_size;
    dest->physical_size += src->physical_size;
    dest->lines += src->lines;
}