test-istream-seekable.c revision a05250bd599fca0cb8b761c4fd69f91bc1c3667e
2454dfa32c93c20a8522c6ed42fe057baaac9f9aStephan Bosch/* Copyright (c) 2009-2017 Dovecot authors, see the included COPYING file */
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen#include "test-lib.h"
0536ccb51d41e3078c3a9fa33e509fb4b2420f95Timo Sirainen#include "istream-private.h"
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen#include "istream-seekable.h"
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen#include <fcntl.h>
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen#include <unistd.h>
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainenstatic int fd_callback(const char **path_r, void *context ATTR_UNUSED)
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen{
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen int fd;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen *path_r = "test-lib.tmp";
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen fd = open(*path_r, O_RDWR | O_CREAT | O_TRUNC, 0600);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen if (fd == -1)
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen i_error("creat(%s) failed: %m", *path_r);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen else
46b823ac3bce2c0f9f0fc73911e48d3a77b04fbeTimo Sirainen i_unlink(*path_r);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen return fd;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen}
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainenstatic void test_istream_seekable_one(unsigned int buffer_size)
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen{
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen static const char *input_string = "xyz";
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen#define STREAM_COUNT 5
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen#define STREAM_BYTES 3
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen struct istream *streams[STREAM_COUNT+1];
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen struct istream *input;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen const unsigned char *data;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen size_t size;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen unsigned int i, j;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen for (i = 0; i < STREAM_COUNT; i++) {
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen streams[i] = test_istream_create(input_string);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen streams[i]->seekable = FALSE;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_istream_set_allow_eof(streams[i], TRUE);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_istream_set_size(streams[i], 0);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen }
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen streams[i] = NULL;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen input = i_stream_create_seekable(streams, buffer_size, fd_callback, NULL);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen for (i = 0; i/STREAM_BYTES < STREAM_COUNT; i++) {
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_istream_set_size(streams[i/STREAM_BYTES], (i%STREAM_BYTES) + 1);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen if (i < buffer_size) {
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_assert(i_stream_read(input) == 1);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen data = i_stream_get_data(input, &size);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_assert(size == i+1);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen } else {
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_assert(i_stream_read(input) == -2);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen i_stream_skip(input, 1);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_assert(i_stream_read(input) == 1);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen data = i_stream_get_data(input, &size);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_assert(size == buffer_size);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen }
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen for (j = 0; j < size; j++) {
4efba37e4f27b93832f6147c3a353d6d22c855c7Timo Sirainen test_assert((char)data[j] == input_string[(input->v_offset + j) % STREAM_BYTES]);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen }
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen }
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen test_assert(i_stream_read(input) == -1);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen for (i = 0; i < STREAM_COUNT; i++) {
a05250bd599fca0cb8b761c4fd69f91bc1c3667eTimo Sirainen test_assert(streams[i]->eof && streams[i]->stream_errno == 0);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen i_stream_unref(&streams[i]);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen }
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen i_stream_unref(&input);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen}
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainenstatic void test_istream_seekable_random(void)
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen{
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen struct istream **streams, *input;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen const unsigned char *data;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen unsigned char *w_data;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen size_t size;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen unsigned int i, j, offset, stream_count, data_len, buffer_size;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen
191153d1a5b0eb0c129139570e3aa5212f28d2acJosef 'Jeff' Sipek stream_count = i_rand_minmax(2, 10 + 2 - 1);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen streams = t_new(struct istream *, stream_count + 1);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen for (i = 0, offset = 0; i < stream_count; i++) {
191153d1a5b0eb0c129139570e3aa5212f28d2acJosef 'Jeff' Sipek data_len = i_rand_minmax(1, 100);
0175d37a5ae5a4d146ca41b684bd38d9b03683cbMartti Rannanjärvi w_data = t_malloc_no0(data_len);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen for (j = 0; j < data_len; j++)
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen w_data[j] = offset++;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen streams[i] = test_istream_create_data(w_data, data_len);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen streams[i]->seekable = FALSE;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_istream_set_allow_eof(streams[i], TRUE);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen }
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen streams[i] = NULL;
e376e08040b5f21ff79a15ae728d2532a34207f6Timo Sirainen i_assert(offset > 0);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen
191153d1a5b0eb0c129139570e3aa5212f28d2acJosef 'Jeff' Sipek buffer_size = i_rand_minmax(1, 100); size = 0;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen input = i_stream_create_seekable(streams, buffer_size, fd_callback, NULL);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen /* first read it through */
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen while (i_stream_read(input) > 0) {
3785910c303507db5f629684e6dde2cc7f83668eTimo Sirainen size = i_stream_get_data_size(input);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen i_stream_skip(input, size);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen }
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen i_stream_seek(input, 0);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen for (i = 0; i < 100; i++) {
191153d1a5b0eb0c129139570e3aa5212f28d2acJosef 'Jeff' Sipek if (i_rand_limit(3) == 0) {
191153d1a5b0eb0c129139570e3aa5212f28d2acJosef 'Jeff' Sipek i_stream_seek(input, i_rand_limit(offset));
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen } else {
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen ssize_t ret = i_stream_read(input);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen if (input->v_offset + size == offset)
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_assert(ret < 0);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen else if (ret == -2) {
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_assert(size == buffer_size);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen } else {
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_assert(ret > 0);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_assert(input->v_offset + ret <= offset);
191153d1a5b0eb0c129139570e3aa5212f28d2acJosef 'Jeff' Sipek i_stream_skip(input, i_rand_limit(ret + 1));
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen data = i_stream_get_data(input, &size);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen for (j = 0; j < size; j++) {
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_assert(data[j] == (input->v_offset + j) % 256);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen }
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen }
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen }
3785910c303507db5f629684e6dde2cc7f83668eTimo Sirainen size = i_stream_get_data_size(input);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen }
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen for (i = 0; i < stream_count; i++) {
a05250bd599fca0cb8b761c4fd69f91bc1c3667eTimo Sirainen test_assert(streams[i]->eof && streams[i]->stream_errno == 0);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen i_stream_unref(&streams[i]);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen }
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen i_stream_unref(&input);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen}
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainenstatic void test_istream_seekable_eof(void)
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen{
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen static const char *in_str = "foo";
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen unsigned int in_str_len = strlen(in_str);
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen struct istream *streams[2], *input;
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen const unsigned char *data;
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen size_t size;
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen test_begin("istream seekable eof");
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen streams[0] = i_stream_create_from_data(in_str, in_str_len);
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen streams[0]->seekable = FALSE;
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen streams[1] = NULL;
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen input = i_stream_create_seekable(streams, in_str_len, fd_callback, NULL);
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen test_assert(i_stream_read(input) == (ssize_t)in_str_len);
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen data = i_stream_get_data(input, &size);
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen test_assert(size == in_str_len);
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen test_assert(memcmp(data, in_str, in_str_len) == 0);
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen test_assert(i_stream_read(input) == -1);
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen data = i_stream_get_data(input, &size);
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen test_assert(size == in_str_len);
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen test_assert(memcmp(data, in_str, in_str_len) == 0);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen i_stream_seek(input, size);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen i_stream_unref(&input);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen test_assert(streams[0]->v_offset == in_str_len);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen test_assert(streams[0]->eof);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen i_stream_unref(&streams[0]);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen test_end();
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen}
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainenstatic void test_istream_seekable_early_end(void)
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen{
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen struct istream *input, *streams[2];
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen test_begin("istream seekable early end");
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen streams[0] = test_istream_create("stream");
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen test_istream_set_size(streams[0], 3);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen test_istream_set_allow_eof(streams[0], FALSE);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen streams[0]->seekable = FALSE;
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen streams[1] = NULL;
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen input = i_stream_create_seekable(streams, 1000, fd_callback, NULL);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen test_assert(i_stream_read(input) == 3);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen test_istream_set_size(streams[0], 5);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen test_assert(i_stream_read(input) == 2);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen i_stream_skip(input, 5);
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen i_stream_unref(&input);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen test_assert(streams[0]->v_offset == 5);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen i_stream_unref(&streams[0]);
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen test_end();
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen}
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainenvoid test_istream_seekable(void)
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen{
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen unsigned int i;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_begin("istream seekable");
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen for (i = 1; i <= STREAM_BYTES*STREAM_COUNT; i++)
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_istream_seekable_one(i);
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_end();
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_begin("istream seekable random");
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen for (i = 0; i < 100; i++) T_BEGIN {
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_istream_seekable_random();
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen } T_END;
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen test_end();
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen
7bcb308d0e13dfa48b483b0addccd889a77bb598Timo Sirainen test_istream_seekable_eof();
96af3e787c15929853b77bac0451b7b8f343510cTimo Sirainen test_istream_seekable_early_end();
7d548d4984752d74bfea8dc4bf1781a8996f43e7Timo Sirainen}