From: Timo Sirainen <tss@iki.fi>
Date: Thu, 20 Jan 2011 23:05:00 +0000 (+0200)
Subject: imapc: Call istream_opened() method if it's set for mail.
X-Git-Tag: 2.1.alpha1~416
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=f6605a4195af81847a7c636409a7dd36398daf01;p=thirdparty%2Fdovecot%2Fcore.git

imapc: Call istream_opened() method if it's set for mail.
---

diff --git a/src/lib-storage/index/imapc/imapc-search.c b/src/lib-storage/index/imapc/imapc-search.c
index 9a9c599cdc..03d2c70ea8 100644
--- a/src/lib-storage/index/imapc/imapc-search.c
+++ b/src/lib-storage/index/imapc/imapc-search.c
@@ -96,9 +96,10 @@ imapc_fetch_stream(struct index_mail *imail, const char *value, bool body)
 {
 	struct mail *_mail = &imail->mail.mail;
 	struct istream *input;
-	size_t size = strlen(value);
+	size_t value_len = strlen(value);
+	uoff_t size;
 	const char *path;
-	int fd;
+	int fd, ret;
 
 	if (imail->data.stream != NULL)
 		return;
@@ -106,7 +107,7 @@ imapc_fetch_stream(struct index_mail *imail, const char *value, bool body)
 	fd = create_temp_fd(_mail->box->storage->user, &path);
 	if (fd == -1)
 		return;
-	if (write_full(fd, value, size) < 0) {
+	if (write_full(fd, value, value_len) < 0) {
 		(void)close(fd);
 		return;
 	}
@@ -115,7 +116,21 @@ imapc_fetch_stream(struct index_mail *imail, const char *value, bool body)
 	i_stream_set_name(imail->data.stream, path);
 	index_mail_set_read_buffer_size(_mail, imail->data.stream);
 
+	if (imail->mail.v.istream_opened != NULL) {
+		if (imail->mail.v.istream_opened(_mail,
+						 &imail->data.stream) < 0) {
+			i_stream_unref(&imail->data.stream);
+			return;
+		}
+	}
+
 	if (body) {
+		ret = i_stream_get_size(imail->data.stream, TRUE, &size);
+		if (ret < 0) {
+			i_stream_unref(&imail->data.stream);
+			return;
+		}
+		i_assert(ret != 0);
 		imail->data.physical_size = size;
 		imail->data.virtual_size = size;
 	}