istream-concat.c revision beffc30d933c5e134c45cc871852a8427eba7e70
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen/* Copyright (c) 2007-2009 Dovecot authors, see the included COPYING file */
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen#include "lib.h"
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen#include "buffer.h"
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen#include "istream-internal.h"
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen#include "istream-concat.h"
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainenstruct concat_istream {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen struct istream_private istream;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen struct istream **input, *cur_input;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen uoff_t *input_size;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen unsigned int cur_idx, unknown_size_idx;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen size_t prev_stream_left, prev_skip;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen};
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainenstatic void i_stream_concat_close(struct iostream_private *stream)
252db51b6c0a605163326b3ea5d09e9936ca3b29Timo Sirainen{
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen struct concat_istream *cstream = (struct concat_istream *)stream;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen unsigned int i;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen for (i = 0; cstream->input[i] != NULL; i++)
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_stream_close(cstream->input[i]);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen}
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainenstatic void i_stream_concat_destroy(struct iostream_private *stream)
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen{
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen struct concat_istream *cstream = (struct concat_istream *)stream;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen unsigned int i;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
252db51b6c0a605163326b3ea5d09e9936ca3b29Timo Sirainen for (i = 0; cstream->input[i] != NULL; i++)
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_stream_unref(&cstream->input[i]);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen}
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainenstatic void
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Siraineni_stream_concat_set_max_buffer_size(struct iostream_private *stream,
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen size_t max_size)
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen{
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen struct concat_istream *cstream = (struct concat_istream *)stream;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen unsigned int i;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->istream.max_buffer_size = max_size;
252db51b6c0a605163326b3ea5d09e9936ca3b29Timo Sirainen for (i = 0; cstream->input[i] != NULL; i++)
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_stream_set_max_buffer_size(cstream->input[i], max_size);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen}
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainenstatic void i_stream_concat_read_next(struct concat_istream *cstream)
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen{
0f66f12eb4cdbf47670975044c88d8f388bf92dfTimo Sirainen const unsigned char *data;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen size_t data_size, size;
d5cebe7f98e63d4e2822863ef2faa4971e8b3a5dTimo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_assert(cstream->cur_input->eof);
d5cebe7f98e63d4e2822863ef2faa4971e8b3a5dTimo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen data = i_stream_get_data(cstream->cur_input, &data_size);
fe363b433b8038a69b55169da9dca27892ad7d18Timo Sirainen cstream->cur_idx++;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->cur_input = cstream->input[cstream->cur_idx];
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_stream_seek(cstream->cur_input, 0);
252db51b6c0a605163326b3ea5d09e9936ca3b29Timo Sirainen
252db51b6c0a605163326b3ea5d09e9936ca3b29Timo Sirainen if (cstream->prev_stream_left > 0 || cstream->istream.pos == 0) {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->prev_stream_left += data_size;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_assert(cstream->prev_stream_left ==
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->istream.pos - cstream->istream.skip);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen return;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen }
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen /* we already verified that the data size is less than the
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen maximum buffer size */
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->istream.pos = 0;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen if (!i_stream_get_buffer_space(&cstream->istream, data_size, &size))
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_unreached();
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_assert(size >= data_size);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->prev_stream_left = data_size;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen memcpy(cstream->istream.w_buffer, data, data_size);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->istream.skip = 0;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->istream.pos = data_size;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen}
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainenstatic ssize_t i_stream_concat_read(struct istream_private *stream)
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen{
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen struct concat_istream *cstream = (struct concat_istream *)stream;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen const unsigned char *data;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen size_t size, pos, cur_pos, bytes_skipped;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen ssize_t ret;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen bool last_stream;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen if (cstream->cur_input == NULL) {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen stream->istream.eof = TRUE;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen return -1;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen }
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_assert(stream->skip >= cstream->prev_skip);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen bytes_skipped = stream->skip - cstream->prev_skip;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen if (cstream->prev_stream_left == 0) {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen /* no need to worry about buffers, skip everything */
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_assert(cstream->prev_skip == 0);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen } else if (bytes_skipped < cstream->prev_stream_left) {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen /* we're still skipping inside buffer */
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->prev_stream_left -= bytes_skipped;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen bytes_skipped = 0;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen } else {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen /* done with the buffer */
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen bytes_skipped -= cstream->prev_stream_left;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->prev_stream_left = 0;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen }
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_stream_skip(cstream->cur_input, bytes_skipped);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen stream->pos -= bytes_skipped;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen stream->skip -= bytes_skipped;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cur_pos = stream->pos - stream->skip - cstream->prev_stream_left;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen data = i_stream_get_data(cstream->cur_input, &pos);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen if (pos > cur_pos)
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen ret = 0;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen else {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen /* need to read more */
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_assert(cur_pos == pos);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen ret = i_stream_read(cstream->cur_input);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen if (ret == -2 || ret == 0)
0f66f12eb4cdbf47670975044c88d8f388bf92dfTimo Sirainen return ret;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen if (ret == -1 && stream->istream.stream_errno != 0) {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen stream->istream.stream_errno =
2526d52441ef368215ab6bf04fd0356d3b09d235Timo Sirainen cstream->cur_input->stream_errno;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen return -1;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen }
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen /* we either read something or we're at EOF */
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen last_stream = cstream->input[cstream->cur_idx+1] == NULL;
5069adb2f5b3609fff9a0a705c6edeae56e0030aTimo Sirainen if (ret == -1 && !last_stream) {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen if (stream->pos >= stream->max_buffer_size)
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen return -2;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_stream_concat_read_next(cstream);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->prev_skip = stream->skip;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen return i_stream_concat_read(stream);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen }
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen stream->istream.eof = cstream->cur_input->eof && last_stream;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_assert(ret != -1 || stream->istream.eof);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen data = i_stream_get_data(cstream->cur_input, &pos);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen }
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen if (cstream->prev_stream_left == 0) {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen stream->buffer = data;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen stream->pos -= stream->skip;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen stream->skip = 0;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen } else if (pos == cur_pos) {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen stream->buffer = stream->w_buffer;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen } else {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen stream->buffer = stream->w_buffer;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen if (!i_stream_get_buffer_space(stream, pos - cur_pos, &size))
5069adb2f5b3609fff9a0a705c6edeae56e0030aTimo Sirainen return -2;
5069adb2f5b3609fff9a0a705c6edeae56e0030aTimo Sirainen
5069adb2f5b3609fff9a0a705c6edeae56e0030aTimo Sirainen if (pos > size)
5069adb2f5b3609fff9a0a705c6edeae56e0030aTimo Sirainen pos = size;
5069adb2f5b3609fff9a0a705c6edeae56e0030aTimo Sirainen memcpy(stream->w_buffer + stream->pos,
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen data + cur_pos, pos - cur_pos);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen }
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen pos += stream->skip + cstream->prev_stream_left;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
6ef7e31619edfaa17ed044b45861d106a86191efTimo Sirainen ret = pos > stream->pos ? (ssize_t)(pos - stream->pos) :
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen (ret == 0 ? 0 : -1);
252db51b6c0a605163326b3ea5d09e9936ca3b29Timo Sirainen stream->pos = pos;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->prev_skip = stream->skip;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen return ret;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen}
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainenstatic unsigned int
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainenfind_v_offset(struct concat_istream *cstream, uoff_t *v_offset)
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen{
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen const struct stat *st;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen unsigned int i;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen for (i = 0; cstream->input[i] != NULL; i++) {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen if (*v_offset == 0) {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen /* seek to beginning of this stream */
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen break;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen }
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen if (i == cstream->unknown_size_idx) {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen /* we'll need to figure out this stream's size */
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen st = i_stream_stat(cstream->input[i], TRUE);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen if (st == NULL) {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->istream.istream.stream_errno =
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->input[i]->stream_errno;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen return (unsigned int)-1;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen }
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen /* @UNSAFE */
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->input_size[i] = st->st_size;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->unknown_size_idx = i + 1;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen }
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen if (*v_offset < cstream->input_size[i])
252db51b6c0a605163326b3ea5d09e9936ca3b29Timo Sirainen break;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen *v_offset -= cstream->input_size[i];
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen }
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
7efee0bb408b0d5253e41997857bdda57855cdc7Timo Sirainen return i;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen}
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
8d235b34022f0bf0f7db54fdbda2d4fd204ed864Timo Sirainenstatic void i_stream_concat_seek(struct istream_private *stream,
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen uoff_t v_offset, bool mark ATTR_UNUSED)
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen{
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen struct concat_istream *cstream = (struct concat_istream *)stream;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen stream->istream.v_offset = v_offset;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen stream->skip = stream->pos = 0;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->prev_stream_left = 0;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->prev_skip = 0;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->cur_idx = find_v_offset(cstream, &v_offset);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen if (cstream->cur_idx == (unsigned int)-1) {
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->cur_input = NULL;
7efee0bb408b0d5253e41997857bdda57855cdc7Timo Sirainen return;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen }
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->cur_input = cstream->input[cstream->cur_idx];
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen if (cstream->cur_input != NULL)
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_stream_seek(cstream->cur_input, v_offset);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen}
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainenstatic const struct stat *
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Siraineni_stream_concat_stat(struct istream_private *stream, bool exact ATTR_UNUSED)
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen{
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen struct concat_istream *cstream = (struct concat_istream *)stream;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen uoff_t v_offset = (uoff_t)-1;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen unsigned int i;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen /* make sure we have all sizes */
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen (void)find_v_offset(cstream, &v_offset);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
7efee0bb408b0d5253e41997857bdda57855cdc7Timo Sirainen stream->statbuf.st_size = 0;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen for (i = 0; i < cstream->unknown_size_idx; i++)
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen stream->statbuf.st_size += cstream->input_size[i];
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen return &stream->statbuf;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen}
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainenstruct istream *i_stream_create_concat(struct istream *input[])
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen{
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen struct concat_istream *cstream;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen unsigned int count;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen size_t max_buffer_size = I_STREAM_MIN_SIZE;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen bool blocking = TRUE, seekable = TRUE;
7efee0bb408b0d5253e41997857bdda57855cdc7Timo Sirainen
7efee0bb408b0d5253e41997857bdda57855cdc7Timo Sirainen /* if any of the streams isn't blocking or seekable, set ourself also
7efee0bb408b0d5253e41997857bdda57855cdc7Timo Sirainen nonblocking/nonseekable */
7efee0bb408b0d5253e41997857bdda57855cdc7Timo Sirainen for (count = 0; input[count] != NULL; count++) {
7efee0bb408b0d5253e41997857bdda57855cdc7Timo Sirainen size_t cur_max = input[count]->real_stream->max_buffer_size;
4da70fe8c9cb6e57b36103d78ab1e9c8654f76d9Timo Sirainen
7efee0bb408b0d5253e41997857bdda57855cdc7Timo Sirainen if (cur_max > max_buffer_size)
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen max_buffer_size = cur_max;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen if (!input[count]->blocking)
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen blocking = FALSE;
252db51b6c0a605163326b3ea5d09e9936ca3b29Timo Sirainen if (!input[count]->seekable)
43d32cbe60fdaef2699d99f1ca259053e9350411Timo Sirainen seekable = FALSE;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_stream_ref(input[count]);
8d235b34022f0bf0f7db54fdbda2d4fd204ed864Timo Sirainen }
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_assert(count != 0);
8d235b34022f0bf0f7db54fdbda2d4fd204ed864Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream = i_new(struct concat_istream, 1);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->input = i_new(struct istream *, count + 1);
252db51b6c0a605163326b3ea5d09e9936ca3b29Timo Sirainen cstream->input_size = i_new(uoff_t, count + 1);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen memcpy(cstream->input, input, sizeof(*input) * count);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->cur_input = cstream->input[0];
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_stream_seek(cstream->cur_input, 0);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->istream.iostream.close = i_stream_concat_close;
8255a22cccf3b0ccf38206c594941820ac1c9e00Timo Sirainen cstream->istream.iostream.destroy = i_stream_concat_destroy;
8255a22cccf3b0ccf38206c594941820ac1c9e00Timo Sirainen cstream->istream.iostream.set_max_buffer_size =
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen i_stream_concat_set_max_buffer_size;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->istream.max_buffer_size = max_buffer_size;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->istream.read = i_stream_concat_read;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->istream.seek = i_stream_concat_seek;
7efee0bb408b0d5253e41997857bdda57855cdc7Timo Sirainen cstream->istream.stat = i_stream_concat_stat;
7efee0bb408b0d5253e41997857bdda57855cdc7Timo Sirainen
7efee0bb408b0d5253e41997857bdda57855cdc7Timo Sirainen cstream->istream.istream.readable_fd = FALSE;
7efee0bb408b0d5253e41997857bdda57855cdc7Timo Sirainen cstream->istream.istream.blocking = blocking;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen cstream->istream.istream.seekable = seekable;
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen return i_stream_create(&cstream->istream, NULL, -1);
2201e2cc1b3f744dac61c2bf8095bcb6b5719540Timo Sirainen}
8255a22cccf3b0ccf38206c594941820ac1c9e00Timo Sirainen