# HG changeset patch
# User Timo Sirainen <tss@iki.fi>
# Date 1165448595 -7200
# Node ID b0ada6e57b0768c7bc67cb557666759c248a865c
# Parent  88a91d9a867b906207c1a73b2c78e5ee3bca129f
Fixes

diff -r 88a91d9a867b -r b0ada6e57b07 src/plugins/fts-squat/fts-backend-squat.c
--- a/src/plugins/fts-squat/fts-backend-squat.c	Wed Dec 06 17:45:46 2006 +0200
+++ b/src/plugins/fts-squat/fts-backend-squat.c	Thu Dec 07 01:43:15 2006 +0200
@@ -23,7 +23,9 @@
 {
 	struct squat_fts_backend *backend;
 	struct mail_storage *storage;
+	struct mailbox_status status;
 	const char *path;
+	bool mmap_disable;
 
 	storage = mailbox_get_storage(box);
 	path = mail_storage_get_mailbox_index_dir(storage,
@@ -33,11 +35,17 @@
 		return NULL;
 	}
 
+	if (mailbox_get_status(box, STATUS_UIDVALIDITY, &status) < 0)
+		return NULL;
+
+	mmap_disable = (storage->flags & MAIL_STORAGE_FLAG_MMAP_DISABLE) != 0;
+
 	backend = i_new(struct squat_fts_backend, 1);
 	backend->backend = fts_backend_squat;
 	backend->trie =
 		squat_trie_open(t_strconcat(path, "/"SQUAT_FILE_PREFIX, NULL),
-				storage->lock_method);
+				status.uidvalidity, storage->lock_method,
+				mmap_disable);
 	return &backend->backend;
 }
 
diff -r 88a91d9a867b -r b0ada6e57b07 src/plugins/fts-squat/squat-test.c
--- a/src/plugins/fts-squat/squat-test.c	Wed Dec 06 17:45:46 2006 +0200
+++ b/src/plugins/fts-squat/squat-test.c	Thu Dec 07 01:43:15 2006 +0200
@@ -48,8 +48,8 @@
 	lib_init();
 	(void)unlink("/tmp/squat-test-index.search");
 	(void)unlink("/tmp/squat-test-index.search.uids");
-	trie = squat_trie_open("/tmp/squat-test-index.search",
-			       FILE_LOCK_METHOD_FCNTL);
+	trie = squat_trie_open("/tmp/squat-test-index.search", time(NULL),
+			       FILE_LOCK_METHOD_FCNTL, FALSE);
 
 	clock_start = clock();
 	gettimeofday(&tv_start, NULL);
diff -r 88a91d9a867b -r b0ada6e57b07 src/plugins/fts-squat/squat-trie.c
--- a/src/plugins/fts-squat/squat-trie.c	Wed Dec 06 17:45:46 2006 +0200
+++ b/src/plugins/fts-squat/squat-trie.c	Thu Dec 07 01:43:15 2006 +0200
@@ -6,6 +6,7 @@
 #include "file-lock.h"
 #include "istream.h"
 #include "ostream.h"
+#include "write-full.h"
 #include "mmap-util.h"
 #include "squat-uidlist.h"
 #include "squat-trie.h"
@@ -42,12 +43,15 @@
 	struct file_lock *file_lock;
 	int lock_count;
 	int lock_type; /* F_RDLCK / F_WRLCK */
+	bool mmap_disable;
 
 	void *mmap_base;
 	size_t mmap_size;
 
 	const struct squat_trie_header *hdr;
+	uint32_t uidvalidity;
 
+	char *uidlist_filepath;
 	struct squat_uidlist *uidlist;
 	struct trie_node *root;
 	buffer_t *buf;
@@ -236,6 +240,7 @@
 	i_error("Corrupted index search file %s: %s", trie->filepath, reason);
 
 	(void)unlink(trie->filepath);
+	(void)unlink(trie->uidlist_filepath);
 	trie->corrupted = TRUE;
 }
 
@@ -358,18 +363,24 @@
 	return 0;
 }
 
-static void trie_close(struct squat_trie *trie)
+static void squat_trie_unmap(struct squat_trie *trie)
 {
-	if (trie->file_lock != NULL)
-		file_lock_free(&trie->file_lock);
-
 	if (trie->mmap_base != NULL) {
 		if (munmap(trie->mmap_base, trie->mmap_size) < 0)
 			squat_trie_set_syscall_error(trie, "munmap()");
 		trie->mmap_base = NULL;
 	}
+
 	trie->mmap_size = 0;
+	trie->hdr = NULL;
+}
 
+static void trie_file_close(struct squat_trie *trie)
+{
+	if (trie->file_lock != NULL)
+		file_lock_free(&trie->file_lock);
+
+	squat_trie_unmap(trie);
 	if (trie->fd != -1) {
 		if (close(trie->fd) < 0)
 			squat_trie_set_syscall_error(trie, "close()");
@@ -390,20 +401,29 @@
 		squat_trie_set_corrupted(trie, "used_file_size too large");
 		return -1;
 	}
-	if (hdr->root_offset > trie->mmap_size ||
-	    hdr->root_offset < sizeof(*hdr)) {
+	if (hdr->root_offset != 0 &&
+	    (hdr->root_offset > trie->mmap_size ||
+	     hdr->root_offset < sizeof(*hdr))) {
 		squat_trie_set_corrupted(trie, "invalid root_offset");
 		return -1;
 	}
+	if (hdr->uidvalidity != trie->uidvalidity) {
+		squat_trie_set_corrupted(trie, "uidvalidity changed");
+		return -1;
+	}
 
 	return 0;
 }
 
-static int trie_map(struct squat_trie *trie)
+static int squat_trie_map(struct squat_trie *trie)
 {
 	struct stat st;
 
-	i_assert(trie->lock_count > 0);
+	if (trie->hdr != NULL &&
+	    trie->hdr->used_file_size <= trie->mmap_size) {
+		/* everything is already mapped */
+		return 1;
+	}
 
 	if (fstat(trie->fd, &st) < 0) {
 		squat_trie_set_syscall_error(trie, "fstat()");
@@ -412,12 +432,9 @@
 	trie->dev = st.st_dev;
 	trie->ino = st.st_ino;
 
-	if (trie->mmap_base != NULL) {
-		if (munmap(trie->mmap_base, trie->mmap_size) < 0)
-			squat_trie_set_syscall_error(trie, "munmap()");
-	}
+	squat_trie_unmap(trie);
+
 	trie->mmap_size = st.st_size;
-
 	trie->mmap_base = mmap(NULL, trie->mmap_size, PROT_READ | PROT_WRITE,
 			       MAP_SHARED, trie->fd, 0);
 	if (trie->mmap_base == MAP_FAILED) {
@@ -427,18 +444,18 @@
 		return -1;
 	}
 
+	if (trie_map_check_header(trie, trie->mmap_base) < 0)
+		return -1;
 	trie->hdr = trie->mmap_base;
-	if (trie_map_check_header(trie, trie->hdr) < 0)
-		return -1;
 
 	if (trie_map_node(trie, trie->hdr->root_offset, 1, &trie->root) < 0) {
-		trie_close(trie);
+		trie_file_close(trie);
 		return 0;
 	}
 	return 1;
 }
 
-static int trie_open(struct squat_trie *trie)
+static int trie_file_open(struct squat_trie *trie)
 {
 	trie->fd = open(trie->filepath, O_RDWR);
 	if (trie->fd == -1) {
@@ -449,25 +466,79 @@
 		return -1;
 	}
 
-	return trie_map(trie);
+	return 1;
+}
+
+static int trie_file_create(struct squat_trie *trie)
+{
+	struct stat st;
+
+	trie->fd = open(trie->filepath, O_RDWR | O_CREAT, 0660);
+	if (trie->fd == -1) {
+		squat_trie_set_syscall_error(trie, "open()");
+		return -1;
+	}
+
+	if (fstat(trie->fd, &st) < 0) {
+		squat_trie_set_syscall_error(trie, "fstat()");
+		return -1;
+	}
+	trie->dev = st.st_dev;
+	trie->ino = st.st_ino;
+	return 0;
+}
+
+static int trie_file_create_finish(struct squat_trie *trie)
+{
+	struct squat_trie_header hdr;
+	struct stat st;
+
+	if (fstat(trie->fd, &st) < 0) {
+		squat_trie_set_syscall_error(trie, "fstat()");
+		return -1;
+	}
+
+	if (st.st_size <= sizeof(hdr)) {
+		memset(&hdr, 0, sizeof(hdr));
+		hdr.version = SQUAT_TRIE_VERSION;
+		hdr.uidvalidity = trie->uidvalidity;
+		hdr.used_file_size = sizeof(hdr);
+
+		if (pwrite_full(trie->fd, &hdr, sizeof(hdr), 0) < 0) {
+			squat_trie_set_syscall_error(trie, "pwrite_full()");
+			return -1;
+		}
+	}
+
+	return 0;
+}
+
+static int squat_trie_reopen(struct squat_trie *trie)
+{
+	trie_file_close(trie);
+	if (trie_file_open(trie) < 0)
+		return -1;
+
+	return 0;
 }
 
 struct squat_trie *
-squat_trie_open(const char *path, enum file_lock_method lock_method)
+squat_trie_open(const char *path, uint32_t uidvalidity,
+		enum file_lock_method lock_method, bool mmap_disable)
 {
 	struct squat_trie *trie;
-	const char *uidlist_path;
 
 	trie = i_new(struct squat_trie, 1);
 	trie->fd = -1;
 	trie->filepath = i_strdup(path);
+	trie->uidvalidity = uidvalidity;
 	trie->lock_method = lock_method;
+	trie->mmap_disable = mmap_disable;
 	trie->buf = buffer_create_dynamic(default_pool, 1024);
 
-	uidlist_path = t_strconcat(path, ".uids", NULL);
-	trie->uidlist = squat_uidlist_init(trie, uidlist_path);
-
-	(void)trie_open(trie);
+	trie->uidlist_filepath = i_strconcat(path, ".uids", NULL);
+	trie->uidlist =
+		squat_uidlist_init(trie, trie->uidlist_filepath, uidvalidity);
 	return trie;
 }
 
@@ -475,6 +546,7 @@
 {
 	buffer_free(trie->buf);
 	squat_uidlist_deinit(trie->uidlist);
+	i_free(trie->uidlist_filepath);
 	i_free(trie->filepath);
 	i_free(trie);
 }
@@ -484,8 +556,22 @@
 	return squat_uidlist_get_last_uid(trie->uidlist, uid_r);
 }
 
+static int squat_trie_is_file_stale(struct squat_trie *trie)
+{
+	struct stat st;
+
+	if (stat(trie->filepath, &st) < 0) {
+		squat_trie_set_syscall_error(trie, "stat()");
+		return -1;
+	}
+
+	return st.st_ino == trie->ino &&
+		CMP_DEV_T(st.st_dev, trie->dev) ? 0 : 1;
+}
+
 int squat_trie_lock(struct squat_trie *trie, int lock_type)
 {
+	bool created = FALSE;
 	int ret;
 
 	i_assert(lock_type == F_RDLCK || lock_type == F_WRLCK);
@@ -498,12 +584,58 @@
 		return 1;
 	}
 
-	i_assert(trie->file_lock == NULL);
-	ret = file_wait_lock(trie->fd, trie->filepath, lock_type,
-			     trie->lock_method, SQUAT_TRIE_LOCK_TIMEOUT,
-			     &trie->file_lock);
-	if (ret <= 0)
-		return ret;
+	if (trie->fd == -1 || trie->corrupted) {
+		trie_file_close(trie);
+		if (lock_type == F_WRLCK) {
+			if ((ret = trie_file_open(trie)) < 0)
+				return -1;
+			if (ret == 0) {
+				if (trie_file_create(trie) < 0)
+					return -1;
+				created = TRUE;
+			}
+		} else {
+			if (trie_file_open(trie) <= 0)
+				return -1;
+		}
+	}
+
+	for (;;) {
+		i_assert(trie->file_lock == NULL);
+		ret = file_wait_lock(trie->fd, trie->filepath, lock_type,
+				     trie->lock_method, SQUAT_TRIE_LOCK_TIMEOUT,
+				     &trie->file_lock);
+		if (ret <= 0)
+			return ret;
+
+		/* if the trie has been compressed, we need to reopen the
+		   file and try to lock again */
+		ret = squat_trie_is_file_stale(trie);
+		if (ret == 0)
+			break;
+
+		file_unlock(&trie->file_lock);
+		if (ret < 0)
+			return -1;
+
+		if (squat_trie_reopen(trie) < 0)
+			return -1;
+	}
+
+	if (created) {
+		/* we possibly created this file. now that we've locked the
+		   file, we can safely check if someone else already wrote the
+		   header or if we should do it now */
+		if (trie_file_create_finish(trie) < 0) {
+			file_unlock(&trie->file_lock);
+			return -1;
+		}
+	}
+
+	if (squat_trie_map(trie) <= 0) {
+		file_unlock(&trie->file_lock);
+		return -1;
+	}
 
 	trie->lock_count++;
 	trie->lock_type = lock_type;
@@ -1054,23 +1186,10 @@
 	struct squat_trie *trie = ctx->trie;
 	struct squat_trie_header hdr;
 
-	if (trie->fd == -1) {
-		trie->fd = open(trie->filepath,
-				O_RDWR | O_CREAT | O_TRUNC, 0600);
-		if (trie->fd == -1) {
-			squat_trie_set_syscall_error(trie, "open()");
-			return -1;
-		}
-
-		memset(&hdr, 0, sizeof(hdr));
-		hdr.version = SQUAT_TRIE_VERSION;
-		hdr.uidvalidity = 0; // FIXME
-	} else {
-		hdr = *trie->hdr;
-		if (lseek(trie->fd, hdr.used_file_size, SEEK_SET) < 0) {
-			squat_trie_set_syscall_error(trie, "lseek()");
-			return -1;
-		}
+	hdr = *trie->hdr;
+	if (lseek(trie->fd, hdr.used_file_size, SEEK_SET) < 0) {
+		squat_trie_set_syscall_error(trie, "lseek()");
+		return -1;
 	}
 
 	ctx->output = o_stream_create_file(trie->fd, default_pool, 0, FALSE);
@@ -1125,7 +1244,9 @@
 		return -1;
 	if (squat_uidlist_flush(trie->uidlist, uidvalidity) < 0)
 		return -1;
-	if (trie_map(trie) <= 0)
+
+	squat_trie_unmap(trie);
+	if (squat_trie_map(trie) <= 0)
 		return -1;
 
 	if (squat_trie_need_compress(trie, (unsigned int)-1)) {
@@ -1338,7 +1459,7 @@
 
 	memset(&hdr, 0, sizeof(hdr));
 	hdr.version = SQUAT_TRIE_VERSION;
-	hdr.uidvalidity = ctx->trie->hdr->uidvalidity;
+	hdr.uidvalidity = ctx->trie->uidvalidity;
 	hdr.root_offset = root_node->file_offset;
 	hdr.used_file_size = ctx->output->offset;
 	hdr.node_count = ctx->node_count;
@@ -1390,9 +1511,8 @@
 	if (ret < 0)
 		(void)unlink(ctx.tmp_path);
 	else {
-		trie_close(trie);
-		if (trie_open(trie) <= 0)
-			ret = -1;
+		if (squat_trie_reopen(trie) < 0)
+			return -1;
 	}
 	return ret;
 }
@@ -1414,7 +1534,7 @@
 
 size_t squat_trie_mem_used(struct squat_trie *trie, unsigned int *count_r)
 {
-	*count_r = trie->hdr->node_count;
+	*count_r = trie->hdr == NULL ? 0 : trie->hdr->node_count;
 
 	return trie->mmap_size;
 }
diff -r 88a91d9a867b -r b0ada6e57b07 src/plugins/fts-squat/squat-trie.h
--- a/src/plugins/fts-squat/squat-trie.h	Wed Dec 06 17:45:46 2006 +0200
+++ b/src/plugins/fts-squat/squat-trie.h	Thu Dec 07 01:43:15 2006 +0200
@@ -6,7 +6,8 @@
 #include "seq-range-array.h"
 
 struct squat_trie *
-squat_trie_open(const char *path, enum file_lock_method lock_method);
+squat_trie_open(const char *path, uint32_t uidvalidity,
+		enum file_lock_method lock_method, bool mmap_disable);
 void squat_trie_close(struct squat_trie *trie);
 
 int squat_trie_get_last_uid(struct squat_trie *trie, uint32_t *last_uid_r);
diff -r 88a91d9a867b -r b0ada6e57b07 src/plugins/fts-squat/squat-uidlist.c
--- a/src/plugins/fts-squat/squat-uidlist.c	Wed Dec 06 17:45:46 2006 +0200
+++ b/src/plugins/fts-squat/squat-uidlist.c	Thu Dec 07 01:43:15 2006 +0200
@@ -21,8 +21,7 @@
 #define UID_LIST_IDX_FLAG_SINGLE 0x80000000
 
 struct squat_uidlist_header {
-	uint32_t uidvalidity; // FIXME
-	uint32_t header_size;
+	uint32_t uidvalidity; 
 	uint32_t used_file_size;
 	uint32_t deleted_space;
 
@@ -48,6 +47,7 @@
 
 struct squat_uidlist {
 	struct squat_trie *trie;
+	uint32_t uidvalidity; 
 
 	char *filepath;
 	int fd;
@@ -92,6 +92,23 @@
 		function, uidlist->filepath);
 }
 
+static int squat_uidlist_check_header(struct squat_uidlist *uidlist,
+				      const struct squat_uidlist_header *hdr)
+{
+	if (hdr->uidvalidity != uidlist->uidvalidity) {
+		squat_trie_set_corrupted(uidlist->trie,
+					 "uidlist: uidvalidity changed");
+		return -1;
+	}
+	if (hdr->used_file_size > uidlist->mmap_size) {
+		squat_trie_set_corrupted(uidlist->trie,
+					 "uidlist: used_file_size too large");
+		return -1;
+	}
+
+	return 0;
+}
+
 static int squat_uidlist_map(struct squat_uidlist *uidlist)
 {
 	struct stat st;
@@ -103,7 +120,6 @@
 
 	if (st.st_size <= sizeof(uidlist->hdr)) {
 		memset(&uidlist->hdr, 0, sizeof(uidlist->hdr));
-		uidlist->hdr.header_size = sizeof(uidlist->hdr);
 		uidlist->hdr.used_file_size = sizeof(uidlist->hdr);
 		return 0;
 	}
@@ -125,7 +141,8 @@
 	}
 
 	memcpy(&uidlist->hdr, uidlist->mmap_base, sizeof(uidlist->hdr));
-	// FIXME: verify header
+	if (squat_uidlist_check_header(uidlist, &uidlist->hdr) < 0)
+		return -1;
 
 	if (uidlist->hdr.uids_expunged)
 		uidlist->check_expunges = TRUE;
@@ -164,13 +181,15 @@
 }
 
 struct squat_uidlist *
-squat_uidlist_init(struct squat_trie *trie, const char *path)
+squat_uidlist_init(struct squat_trie *trie, const char *path,
+		   uint32_t uidvalidity)
 {
 	struct squat_uidlist *uidlist;
 
 	uidlist = i_new(struct squat_uidlist, 1);
 	uidlist->trie = trie;
 	uidlist->filepath = i_strdup(path);
+	uidlist->uidvalidity = uidvalidity;
 	uidlist->fd = -1;
 	uidlist->first_new_list_idx = 1;
 	i_array_init(&uidlist->lists, 65536);
@@ -755,8 +774,7 @@
 	}
 
 	/* write the header */
-	ctx->hdr.uidvalidity = ctx->uidlist->hdr.uidvalidity;
-	ctx->hdr.header_size = sizeof(ctx->hdr);
+	ctx->hdr.uidvalidity = ctx->uidlist->uidvalidity;
 	ctx->hdr.used_file_size = ctx->output->offset;
 
 	if (ctx->existing_uids == NULL) {
diff -r 88a91d9a867b -r b0ada6e57b07 src/plugins/fts-squat/squat-uidlist.h
--- a/src/plugins/fts-squat/squat-uidlist.h	Wed Dec 06 17:45:46 2006 +0200
+++ b/src/plugins/fts-squat/squat-uidlist.h	Thu Dec 07 01:43:15 2006 +0200
@@ -7,7 +7,8 @@
 struct squat_uidlist;
 
 struct squat_uidlist *
-squat_uidlist_init(struct squat_trie *trie, const char *path);
+squat_uidlist_init(struct squat_trie *trie, const char *path,
+		   uint32_t uidvalidity);
 void squat_uidlist_deinit(struct squat_uidlist *uidlist);
 
 int squat_uidlist_get_last_uid(struct squat_uidlist *uidlist, uint32_t *uid_r);