Mercurial > dovecot > original-hg > dovecot-1.2
view src/lib-mail/message-size.c @ 9532:00cd9aacd03c HEAD
Updated copyright notices to include year 2010.
author | Timo Sirainen <tss@iki.fi> |
---|---|
date | Mon, 25 Jan 2010 01:18:58 +0200 |
parents | 94dbe538cdb7 |
children |
line wrap: on
line source
/* Copyright (c) 2002-2010 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; }