test-istream-dot.c revision d4488f314d0c799c5f5b8a7890e17eef868f7a2c
2ronwalf/* Copyright (c) 2007-2016 Dovecot authors, see the included COPYING file */
2ronwalf
2ronwalf#include "lib.h"
2ronwalf#include "str.h"
2ronwalf#include "istream.h"
2ronwalf#include "istream-dot.h"
2ronwalf#include "test-common.h"
2ronwalf
38daenzeroramastruct dot_test {
2ronwalf const char *input;
2ronwalf const char *output;
2ronwalf const char *parent_input;
2ronwalf};
2ronwalf
2ronwalfstatic void test_istream_dot_one(const struct dot_test *test,
38daenzerorama bool send_last_lf, bool test_bufsize)
2ronwalf{
2ronwalf struct istream *test_input, *input;
2ronwalf const unsigned char *data;
2ronwalf size_t size;
2ronwalf unsigned int i, outsize, input_len, output_len;
2ronwalf string_t *str;
2ronwalf uoff_t offset;
2ronwalf int ret;
2ronwalf
2ronwalf test_input = test_istream_create(test->input);
2ronwalf input = i_stream_create_dot(test_input, send_last_lf);
2ronwalf
2ronwalf input_len = strlen(test->input);
2ronwalf output_len = strlen(test->output);
2ronwalf if (!send_last_lf &&
2ronwalf (test->input[input_len-1] == '\n' ||
2ronwalf strstr(test->input, "\n.\n") != NULL ||
2ronwalf strstr(test->input, "\n.\r\n") != NULL)) {
2ronwalf if (output_len > 0 &&
2ronwalf test->output[output_len-1] == '\n') {
2ronwalf output_len--;
2ronwalf if (output_len > 0 &&
2ronwalf test->output[output_len-1] == '\r')
2ronwalf output_len--;
2ronwalf }
2ronwalf }
2ronwalf
2ronwalf str = t_str_new(256);
2ronwalf if (!test_bufsize) {
2ronwalf outsize = 1; i = 0;
2ronwalf i_stream_set_max_buffer_size(input, outsize);
2ronwalf test_istream_set_size(test_input, 1);
2ronwalf while ((ret = i_stream_read(input)) != -1) {
2ronwalf switch (ret) {
2ronwalf case -2:
2ronwalf i_stream_set_max_buffer_size(input, ++outsize);
2ronwalf offset = test_input->v_offset;
2ronwalf /* seek one byte backwards so stream gets
2ronwalf reset */
2ronwalf i_stream_seek(test_input, offset - 1);
2ronwalf /* go back to original position */
2ronwalf test_istream_set_size(test_input, offset);
2ronwalf i_stream_skip(test_input, 1);
2ronwalf /* and finally allow reading one more byte */
2ronwalf test_istream_set_size(test_input, offset + 1);
2ronwalf break;
2ronwalf case 0:
2ronwalf test_istream_set_size(test_input, ++i);
2ronwalf break;
2ronwalf default:
2ronwalf test_assert(ret > 0);
2ronwalf
2ronwalf data = i_stream_get_data(input, &size);
2ronwalf str_append_n(str, data, size);
2ronwalf i_stream_skip(input, size);
2ronwalf }
2ronwalf }
2ronwalf test_istream_set_size(test_input, input_len);
2ronwalf (void)i_stream_read(test_input);
2ronwalf } else {
2ronwalf test_istream_set_size(test_input, input_len);
2ronwalf size = 0;
2ronwalf for (i = 1; i < output_len; i++) {
2ronwalf i_stream_set_max_buffer_size(input, i);
2ronwalf test_assert(i_stream_read(input) == 1);
2ronwalf test_assert(i_stream_read(input) == -2);
2ronwalf data = i_stream_get_data(input, &size);
2ronwalf test_assert(memcmp(data, test->output, size) == 0);
2ronwalf }
2ronwalf i_stream_set_max_buffer_size(input, i+2);
2ronwalf if (size < output_len)
2ronwalf test_assert(i_stream_read(input) == 1);
2ronwalf test_assert(i_stream_read(input) == -1);
2ronwalf
2ronwalf data = i_stream_get_data(input, &size);
2ronwalf if (size > 0)
2ronwalf str_append_n(str, data, size);
2ronwalf }
2ronwalf test_assert(input->stream_errno == 0);
2ronwalf test_assert(str_len(str) == output_len);
2ronwalf test_assert(memcmp(str_data(str), test->output, output_len) == 0);
2ronwalf
2ronwalf /* read the data after the '.' line and verify it's still there */
2ronwalf i_stream_set_max_buffer_size(test_input, (size_t)-1);
2ronwalf (void)i_stream_read(test_input);
2ronwalf data = i_stream_get_data(test_input, &size);
2ronwalf test_assert(size == strlen(test->parent_input));
2ronwalf if (size > 0)
2ronwalf test_assert(memcmp(data, test->parent_input, size) == 0);
2ronwalf
2ronwalf i_stream_unref(&test_input);
2ronwalf i_stream_unref(&input);
2ronwalf}
2ronwalf
2ronwalfstatic void test_istream_dot_error(const char *input_str, bool test_bufsize)
2ronwalf{
2ronwalf struct istream *test_input, *input;
2ronwalf unsigned int i, outsize, input_len;
2ronwalf uoff_t offset;
2ronwalf int ret;
2ronwalf
2ronwalf test_input = test_istream_create(input_str);
2ronwalf input = i_stream_create_dot(test_input, FALSE);
2ronwalf
2ronwalf input_len = strlen(input_str);
2ronwalf
2ronwalf if (!test_bufsize) {
2ronwalf outsize = 1; i = 0;
2ronwalf i_stream_set_max_buffer_size(input, outsize);
2ronwalf test_istream_set_size(test_input, 1);
2ronwalf while ((ret = i_stream_read(input)) != -1) {
2ronwalf switch (ret) {
2ronwalf case -2:
2ronwalf i_stream_set_max_buffer_size(input, ++outsize);
2ronwalf offset = test_input->v_offset;
2ronwalf /* seek one byte backwards so stream gets
2ronwalf reset */
2ronwalf i_stream_seek(test_input, offset - 1);
2ronwalf /* go back to original position */
2ronwalf test_istream_set_size(test_input, offset);
2ronwalf i_stream_skip(test_input, 1);
2ronwalf /* and finally allow reading one more byte */
2ronwalf test_istream_set_size(test_input, offset + 1);
2ronwalf break;
2ronwalf case 0:
2ronwalf test_istream_set_size(test_input, ++i);
2ronwalf break;
2ronwalf default:
2ronwalf test_assert(ret > 0);
2ronwalf }
2ronwalf }
2ronwalf test_istream_set_size(test_input, input_len);
2ronwalf (void)i_stream_read(test_input);
2ronwalf } else {
2ronwalf test_istream_set_size(test_input, input_len);
2ronwalf for (i = 1; i <= input_len; i++) {
2ronwalf i_stream_set_max_buffer_size(input, i);
2ronwalf (void)i_stream_read(input);
2ronwalf (void)i_stream_read(input);
2ronwalf }
2ronwalf i_stream_set_max_buffer_size(input, i+1);
2ronwalf (void)i_stream_read(input);
2ronwalf }
2ronwalf test_assert(input->stream_errno == EPIPE);
2ronwalf
2ronwalf i_stream_unref(&test_input);
2ronwalf i_stream_unref(&input);
38daenzerorama}
38daenzerorama
38daenzeroramastatic void test_istream_dot(void)
38daenzerorama{
38daenzerorama static struct dot_test tests[] = {
38daenzerorama { "..foo\n..\n.foo\n.\nfoo", ".foo\n.\nfoo\n", "foo" },
38daenzerorama { "..foo\r\n..\r\n.foo\r\n.\r\nfoo", ".foo\r\n.\r\nfoo\r\n", "foo" },
38daenzerorama { "\r.\r\n.\r\n", "\r.\r\n", "" },
38daenzerorama { "\n\r.\r\r\n.\r\n", "\n\r.\r\r\n", "" },
2ronwalf { "\r\n.\rfoo\n.\n", "\r\n\rfoo\n", "" },
{ "\r\n.\r\n", "\r\n", "" },
{ "\n.\r\n", "\n", "" },
{ "\n.\n", "\n", "" },
{ ".\r\n", "", "" },
{ ".\n", "", "" }
};
static const char *error_tests[] = {
"",
".",
"..",
".\r",
".\rx",
"..\r\n",
"\r.",
"\r.\r",
"\r.\rx",
"\r.\r\n",
"\r.\n",
"\r..\n",
"\r\n",
"\r\n.",
"\r\n.\r",
"\r\n.\rx",
"\r\n.\rx\n",
"\r\n..\r\n",
"\n",
"\n.",
"\n.\r",
"\n.\rx",
"\n..\r\n"
};
unsigned int i;
test_begin("dot istream");
for (i = 0; i < N_ELEMENTS(tests); i++) {
test_istream_dot_one(&tests[i], TRUE, TRUE);
test_istream_dot_one(&tests[i], TRUE, FALSE);
test_istream_dot_one(&tests[i], FALSE, TRUE);
test_istream_dot_one(&tests[i], FALSE, FALSE);
}
for (i = 0; i < N_ELEMENTS(error_tests); i++) {
test_istream_dot_error(error_tests[i], FALSE);
test_istream_dot_error(error_tests[i], TRUE);
}
test_end();
}
int main(void)
{
static void (*test_functions[])(void) = {
test_istream_dot,
NULL
};
return test_run(test_functions);
}