1 files changed, 47 insertions, 0 deletions
diff --git a/src/lib-fs/istream-fs-stats.c b/src/lib-fs/istream-fs-stats.c
new file mode 100644
index 0000000..3f77430
--- /dev/null
+++ b/src/lib-fs/istream-fs-stats.c
@@ -0,0 +1,47 @@
+/* Copyright (c) 2015-2018 Dovecot authors, see the included COPYING file */
+
+#include "lib.h"
+#include "fs-api-private.h"
+#include "istream-private.h"
+#include "istream-fs-stats.h"
+
+struct fs_stats_istream {
+	struct istream_private istream;
+	struct fs_file *file;
+};
+
+static ssize_t
+i_stream_fs_stats_read(struct istream_private *stream)
+{
+	struct fs_stats_istream *sstream = (struct fs_stats_istream *)stream;
+	ssize_t ret;
+
+	i_stream_seek(stream->parent, stream->parent_start_offset +
+		      stream->istream.v_offset);
+
+	ret = i_stream_read_copy_from_parent(&stream->istream);
+	if (ret > 0) {
+		/* count the first returned bytes as the finish time, since
+		   we don't want to count the time caller spends on processing
+		   this stream. (only the first fs_file_timing_end() call
+		   actually does anything - the others are ignored.) */
+		fs_file_timing_end(sstream->file, FS_OP_READ);
+	}
+	return ret;
+}
+
+struct istream *
+i_stream_create_fs_stats(struct istream *input, struct fs_file *file)
+{
+	struct fs_stats_istream *sstream;
+
+	sstream = i_new(struct fs_stats_istream, 1);
+	sstream->file = file;
+	sstream->istream.max_buffer_size = input->real_stream->max_buffer_size;
+	sstream->istream.stream_size_passthrough = TRUE;
+	sstream->istream.read = i_stream_fs_stats_read;
+	sstream->istream.istream.blocking = input->blocking;
+	sstream->istream.istream.seekable = input->seekable;
+	return i_stream_create(&sstream->istream, input,
+			       i_stream_get_fd(input), 0);
+}