istream-crlf.c revision da2aa032ccfa8e7e4a4380ef738014549f4d2c2d
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen/* Copyright (c) 2007-2009 Dovecot authors, see the included COPYING file */
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen#include "lib.h"
ecc81625167ed96c04c02aa190a1ea5baa65b474Timo Sirainen#include "istream-internal.h"
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen#include "istream-crlf.h"
fb8e400512210d49339775747a520411987d4353Timo Sirainen
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainenstruct crlf_istream {
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen struct istream_private istream;
c0435c854a0e7246373b9752d163095cc4fbe985Timo Sirainen
c0435c854a0e7246373b9752d163095cc4fbe985Timo Sirainen unsigned int pending_cr:1;
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen unsigned int last_cr:1;
c0435c854a0e7246373b9752d163095cc4fbe985Timo Sirainen};
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainenstatic int i_stream_crlf_read_common(struct crlf_istream *cstream)
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen{
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen struct istream_private *stream = &cstream->istream;
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen size_t size;
c0435c854a0e7246373b9752d163095cc4fbe985Timo Sirainen ssize_t ret;
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen (void)i_stream_get_data(stream->parent, &size);
c0435c854a0e7246373b9752d163095cc4fbe985Timo Sirainen if (size == 0) {
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen ret = i_stream_read(stream->parent);
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen if (ret <= 0 && (ret != -2 || stream->skip == 0)) {
ecc81625167ed96c04c02aa190a1ea5baa65b474Timo Sirainen stream->istream.stream_errno =
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen stream->parent->stream_errno;
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen stream->istream.eof = stream->parent->eof;
c0435c854a0e7246373b9752d163095cc4fbe985Timo Sirainen return ret;
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen }
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen (void)i_stream_get_data(stream->parent, &size);
c0435c854a0e7246373b9752d163095cc4fbe985Timo Sirainen i_assert(size != 0);
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen }
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen
ecc81625167ed96c04c02aa190a1ea5baa65b474Timo Sirainen if (!i_stream_get_buffer_space(stream, size, NULL))
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen return -2;
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen return 1;
c0435c854a0e7246373b9752d163095cc4fbe985Timo Sirainen}
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainenstatic ssize_t i_stream_crlf_read_crlf(struct istream_private *stream)
d756ebcfa96bd7cff02097c8f26df9df368b81b1Timo Sirainen{
c2bb1764c359ce85a7f7f789ead11dd613ff9769Timo Sirainen struct crlf_istream *cstream = (struct crlf_istream *)stream;
const unsigned char *data;
size_t i, dest, size;
ssize_t ret;
ret = i_stream_crlf_read_common(cstream);
if (ret <= 0)
return ret;
data = i_stream_get_data(stream->parent, &size);
/* @UNSAFE: add missing CRs */
dest = stream->pos;
for (i = 0; i < size && dest < stream->buffer_size; i++) {
if (data[i] == '\n') {
if (i == 0) {
if (!cstream->last_cr)
stream->w_buffer[dest++] = '\r';
} else {
if (data[i-1] != '\r')
stream->w_buffer[dest++] = '\r';
}
if (dest == stream->buffer_size)
break;
}
stream->w_buffer[dest++] = data[i];
}
cstream->last_cr = stream->w_buffer[dest-1] == '\r';
i_stream_skip(stream->parent, i);
ret = dest - stream->pos;
i_assert(ret > 0);
stream->pos = dest;
return ret;
}
static ssize_t i_stream_crlf_read_lf(struct istream_private *stream)
{
struct crlf_istream *cstream = (struct crlf_istream *)stream;
const unsigned char *data;
size_t i, dest, size;
ssize_t ret;
int diff;
ret = i_stream_crlf_read_common(cstream);
if (ret <= 0)
return ret;
data = i_stream_get_data(stream->parent, &size);
/* @UNSAFE */
dest = stream->pos;
if (data[0] == '\n') {
stream->w_buffer[dest++] = '\n';
cstream->pending_cr = FALSE;
} else {
if (cstream->pending_cr) {
/* CR without LF */
stream->w_buffer[dest++] = '\r';
if (dest == stream->buffer_size) {
cstream->pending_cr = FALSE;
return 1;
}
}
if (data[0] != '\r')
stream->w_buffer[dest++] = data[0];
}
diff = -1;
for (i = 1; i < size && dest < stream->buffer_size; i++) {
if (data[i] == '\r') {
if (data[i-1] != '\r')
continue;
} else if (data[i-1] == '\r' && data[i] != '\n') {
stream->w_buffer[dest++] = '\r';
if (dest == stream->buffer_size) {
diff = 0;
break;
}
}
stream->w_buffer[dest++] = data[i];
}
cstream->pending_cr = data[i+diff] == '\r';
i_stream_skip(stream->parent, i);
ret = dest - stream->pos;
if (ret == 0) {
i_assert(cstream->pending_cr && size == 1);
return i_stream_crlf_read_lf(stream);
}
i_assert(ret > 0);
stream->pos = dest;
return ret;
}
static const struct stat *
i_stream_crlf_stat(struct istream_private *stream, bool exact)
{
return i_stream_stat(stream->parent, exact);
}
static struct istream *
i_stream_create_crlf_full(struct istream *input, bool crlf)
{
struct crlf_istream *cstream;
cstream = i_new(struct crlf_istream, 1);
cstream->istream.max_buffer_size = input->real_stream->max_buffer_size;
cstream->istream.read = crlf ? i_stream_crlf_read_crlf :
i_stream_crlf_read_lf;
cstream->istream.stat = i_stream_crlf_stat;
cstream->istream.istream.readable_fd = FALSE;
cstream->istream.istream.blocking = input->blocking;
cstream->istream.istream.seekable = FALSE;
return i_stream_create(&cstream->istream, input,
i_stream_get_fd(input));
}
struct istream *i_stream_create_crlf(struct istream *input)
{
return i_stream_create_crlf_full(input, TRUE);
}
struct istream *i_stream_create_lf(struct istream *input)
{
return i_stream_create_crlf_full(input, FALSE);
}