From ae2bad232435e5f81f37dd4b8ddb631a564d3dc2 Mon Sep 17 00:00:00 2001
From: Aki Tuomi <aki.tuomi@dovecot.fi>
Date: Tue, 3 Jan 2017 17:21:33 +0200
Subject: [PATCH] lib-storage: Add vsize extension to index

Keep virtual size in index, instead of cache
when it's less than 2^32-1. This helps when
cache becomes corrupted, and goes away, we
still have virtual sizes for quota calculations.
---
 src/lib-storage/index/index-mail.c | 51 +++++++++++++++++++++++++++++-
 src/lib-storage/index/index-mail.h |  1 +
 2 files changed, 51 insertions(+), 1 deletion(-)

diff --git a/src/lib-storage/index/index-mail.c b/src/lib-storage/index/index-mail.c
index 27f4e85af1..842cd5da6a 100644
--- a/src/lib-storage/index/index-mail.c
+++ b/src/lib-storage/index/index-mail.c
@@ -448,7 +448,20 @@ static bool get_cached_msgpart_sizes(struct index_mail *mail)
 bool index_mail_get_cached_virtual_size(struct index_mail *mail, uoff_t *size_r)
 {
 	struct index_mail_data *data = &mail->data;
+	struct mail *_mail = &mail->mail.mail;
 	uoff_t size;
+	const void *idata;
+	bool expunged ATTR_UNUSED;
+	unsigned int idx ATTR_UNUSED;
+
+	/* see if we can get it from index */
+	mail_index_lookup_ext(_mail->transaction->view, _mail->seq,
+			      mail->vsize_ext_id, &idata, &expunged);
+	const uint32_t *vsize = idata;
+
+	if (vsize != NULL && *vsize > 0) {
+		data->virtual_size = (*vsize)-1;
+	}
 
 	data->cache_fetch_fields |= MAIL_FETCH_VIRTUAL_SIZE;
 	if (data->virtual_size == (uoff_t)-1) {
@@ -469,6 +482,16 @@ bool index_mail_get_cached_virtual_size(struct index_mail *mail, uoff_t *size_r)
 		data->body_size_set = TRUE;
 	}
 	*size_r = data->virtual_size;
+
+	/* if vsize is present and wanted for index, but missing from index
+	   add it to index. */
+	if (vsize != NULL && *vsize == 0 &&
+	    data->body_size.virtual_size < (uint32_t)-1) {
+		uint32_t vsize = data->body_size.virtual_size+1;
+		mail_index_update_ext(_mail->transaction->itrans, _mail->seq,
+				      mail->vsize_ext_id, &vsize, NULL);
+	}
+
 	return TRUE;
 }
 
@@ -833,16 +856,40 @@ static void index_mail_body_parsed_cache_body_snippet(struct index_mail *mail)
 
 static void index_mail_cache_sizes(struct index_mail *mail)
 {
+	struct mail *_mail = &mail->mail.mail;
+	struct mail_index_view *view = _mail->transaction->view;
+
 	static enum index_cache_field size_fields[] = {
 		MAIL_CACHE_VIRTUAL_FULL_SIZE,
 		MAIL_CACHE_PHYSICAL_FULL_SIZE
 	};
 	uoff_t sizes[N_ELEMENTS(size_fields)];
 	unsigned int i;
+	uint32_t vsize;
+	uint32_t idx ATTR_UNUSED;
 
 	sizes[0] = mail->data.virtual_size;
 	sizes[1] = mail->data.physical_size;
 
+	/* store the virtual size in index if
+		extension for it exists or
+		extension for box virtual size exists
+	*/
+
+	if ((mail_index_map_get_ext_idx(view->map, mail->vsize_ext_id, &idx) ||
+	     mail_index_map_get_ext_idx(view->map, _mail->box->vsize_hdr_ext_id, &idx)) &&
+	    (sizes[0] != (uoff_t)-1 &&
+	     sizes[0] < (uint32_t)-1)) {
+		/* vsize = 0 means it's not present in index, consult cache.
+		   we store vsize for every +4GB-1 mail to cache because
+		   index can only hold 2^32-1 size. Cache will not be used
+		   when vsize is stored in index. */
+		vsize = sizes[0] + 1;
+		sizes[0] = (uoff_t)-1;
+		mail_index_update_ext(_mail->transaction->itrans, _mail->seq,
+				      mail->vsize_ext_id, &vsize, NULL);
+	}
+
 	for (i = 0; i < N_ELEMENTS(size_fields); i++) {
 		if (sizes[i] != (uoff_t)-1 &&
 		    index_mail_want_cache(mail, size_fields[i])) {
@@ -1493,7 +1540,9 @@ void index_mail_init(struct index_mail *mail,
 	mail->mail.v = *t->box->mail_vfuncs;
 	mail->mail.mail.box = t->box;
 	mail->mail.mail.transaction = t;
-
+	mail->vsize_ext_id = mail_index_ext_register(t->box->index, "vsize", 0,
+						     sizeof(uint32_t),
+						     sizeof(uint32_t));
 	t->mail_ref_count++;
 	mail->mail.data_pool = pool_alloconly_create("index_mail", 16384);
 	mail->ibox = INDEX_STORAGE_CONTEXT(t->box);
diff --git a/src/lib-storage/index/index-mail.h b/src/lib-storage/index/index-mail.h
index 7daf60837d..99b65dc587 100644
--- a/src/lib-storage/index/index-mail.h
+++ b/src/lib-storage/index/index-mail.h
@@ -137,6 +137,7 @@ struct index_mail {
 	struct index_mailbox_context *ibox;
 
 	int pop3_state;
+	uint32_t vsize_ext_id;
 
 	/* per-mail variables, here for performance reasons: */
 	uint32_t header_seq;