maildir-keywords.c revision 79fcd3f95a6266cc62ceaa753e56dd4456ab7c4b
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen/* Copyright (C) 2005 Timo Sirainen */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen/* note that everything here depends on uidlist file being locked the whole
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen time. that's why we don't have any locking of our own, or that we do things
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen that would be racy otherwise. */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "lib.h"
16f816d3f3c32ae3351834253f52ddd0212bcbf3Timo Sirainen#include "array.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "ioloop.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "hash.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "str.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "istream.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "file-dotlock.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "write-full.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "maildir-storage.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "maildir-uidlist.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "maildir-keywords.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include <stdlib.h>
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include <sys/stat.h>
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include <utime.h>
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstruct maildir_keywords {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct maildir_mailbox *mbox;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen char *path;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen pool_t pool;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen array_t ARRAY_DEFINE(list, const char *);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct hash_table *hash; /* name -> idx+1 */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct dotlock_settings dotlock_settings;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen time_t synced_mtime;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int synced:1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int changed:1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen};
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstruct maildir_keywords_sync_ctx {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct maildir_keywords *mk;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct mail_index *index;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen const array_t *ARRAY_DEFINE_PTR(keywords, const char *);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen array_t ARRAY_DEFINE(idx_to_chr, char);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int chridx_to_idx[MAILDIR_MAX_KEYWORDS];
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen};
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstruct maildir_keywords *maildir_keywords_init(struct maildir_mailbox *mbox)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct maildir_keywords *mk;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk = i_new(struct maildir_keywords, 1);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->mbox = mbox;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->path = i_strconcat(mbox->control_dir,
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen "/" MAILDIR_KEYWORDS_NAME, NULL);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->pool = pool_alloconly_create("maildir keywords", 256);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ARRAY_CREATE(&mk->list, default_pool,
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen const char *, MAILDIR_MAX_KEYWORDS);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->hash = hash_create(default_pool, mk->pool, 0,
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen strcase_hash, (hash_cmp_callback_t *)strcasecmp);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return mk;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenvoid maildir_keywords_deinit(struct maildir_keywords *mk)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen hash_destroy(mk->hash);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen array_free(&mk->list);
79fcd3f95a6266cc62ceaa753e56dd4456ab7c4bTimo Sirainen pool_unref(mk->pool);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_free(mk->path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_free(mk);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstatic void maildir_keywords_clear(struct maildir_keywords *mk)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen array_clear(&mk->list);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen hash_clear(mk->hash, FALSE);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen p_clear(mk->pool);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstatic int maildir_keywords_sync(struct maildir_keywords *mk)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct istream *input;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct stat st;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen char *line, *p, *new_name;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen const char **strp;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen int fd, idx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
1e47cfede3a0b62654105daab00e97b5d660bc6bTimo Sirainen /* Remember that we rely on uidlist file locking in here. That's why
1e47cfede3a0b62654105daab00e97b5d660bc6bTimo Sirainen we rely on stat()'s timestamp and don't bother handling ESTALE
1e47cfede3a0b62654105daab00e97b5d660bc6bTimo Sirainen errors. */
1e47cfede3a0b62654105daab00e97b5d660bc6bTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (stat(mk->path, &st) < 0) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (errno == ENOENT) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen maildir_keywords_clear(mk);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->synced = TRUE;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return 0;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mail_storage_set_critical(STORAGE(mk->mbox->storage),
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen "stat(%s) failed: %m", mk->path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (st.st_mtime == mk->synced_mtime) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* hasn't changed */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->synced = TRUE;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return 0;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->synced_mtime = st.st_mtime;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen fd = open(mk->path, O_RDONLY);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (fd == -1) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (errno == ENOENT) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen maildir_keywords_clear(mk);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->synced = TRUE;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return 0;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mail_storage_set_critical(STORAGE(mk->mbox->storage),
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen "open(%s) failed: %m", mk->path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen maildir_keywords_clear(mk);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen input = i_stream_create_file(fd, default_pool, 1024, FALSE);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen while ((line = i_stream_read_next_line(input)) != NULL) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen p = strchr(line, ' ');
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (p == NULL) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* note that when converting .customflags file this
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen case happens in the first line. */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen continue;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen *p++ = '\0';
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen idx = atoi(line);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (idx < 0 || idx >= MAILDIR_MAX_KEYWORDS) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* shouldn't happen */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen continue;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* save it */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen new_name = p_strdup(mk->pool, p);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen hash_insert(mk->hash, new_name, POINTER_CAST(idx + 1));
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen strp = array_idx_modifyable(&mk->list, idx);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen *strp = new_name;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cd56a23e21f1df3f79648cf07e2f4385e2fadebbTimo Sirainen i_stream_destroy(&input);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (close(fd) < 0) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mail_storage_set_critical(STORAGE(mk->mbox->storage),
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen "close(%s) failed: %m", mk->path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->synced = TRUE;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return 0;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstatic int
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenmaildir_keywords_lookup(struct maildir_keywords *mk, const char *name)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen void *p;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_assert(maildir_uidlist_is_locked(mk->mbox->uidlist));
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen p = hash_lookup(mk->hash, name);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (p == NULL) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (mk->synced)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (maildir_keywords_sync(mk) < 0)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen p = hash_lookup(mk->hash, name);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (p == NULL)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return POINTER_CAST_TO(p, int)-1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstatic void
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenmaildir_keywords_create(struct maildir_keywords *mk, const char *name,
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int chridx)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen const char **strp;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen char *new_name;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_assert(chridx < MAILDIR_MAX_KEYWORDS);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen new_name = p_strdup(mk->pool, name);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen hash_insert(mk->hash, new_name, POINTER_CAST(chridx + 1));
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen strp = array_idx_modifyable(&mk->list, chridx);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen *strp = new_name;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->changed = TRUE;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstatic int
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenmaildir_keywords_lookup_or_create(struct maildir_keywords *mk, const char *name)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen const char *const *keywords;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int i, count;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen int idx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen idx = maildir_keywords_lookup(mk, name);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (idx >= 0)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return idx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_assert(mk->synced);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* see if we are full */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen keywords = array_get(&mk->list, &count);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen for (i = 0; i < count; i++) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (keywords[i] == NULL)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen break;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (i == count && count >= MAILDIR_MAX_KEYWORDS)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen maildir_keywords_create(mk, name, i);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return i;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstatic const char *
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenmaildir_keywords_idx(struct maildir_keywords *mk, unsigned int idx)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen const char *const *keywords;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int count;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_assert(maildir_uidlist_is_locked(mk->mbox->uidlist));
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen keywords = array_get(&mk->list, &count);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (idx >= count) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (mk->synced)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return NULL;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (maildir_keywords_sync(mk) < 0)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return NULL;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen keywords = array_get(&mk->list, &count);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return idx >= count ? NULL : keywords[idx];
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstatic int maildir_keywords_commit(struct maildir_keywords *mk)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct dotlock *dotlock;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen const char *lock_path, *const *keywords;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int i, count;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct utimbuf ut;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen string_t *str;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen int fd;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->synced = FALSE;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (!mk->changed)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return 0;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* we could just create the temp file directly, but doing it this
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ways avoids potential problems with overwriting contents in
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen malicious symlinks */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen t_push();
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen lock_path = t_strconcat(mk->path, ".lock", NULL);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen (void)unlink(lock_path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen fd = file_dotlock_open(&mk->dotlock_settings, mk->path,
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen DOTLOCK_CREATE_FLAG_NONBLOCK, &dotlock);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (fd == -1) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mail_storage_set_critical(STORAGE(mk->mbox->storage),
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen "file_dotlock_open(%s) failed: %m", mk->path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen t_pop();
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen str = t_str_new(256);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen keywords = array_get(&mk->list, &count);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen for (i = 0; i < count; i++) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (keywords[i] != NULL)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen str_printfa(str, "%u %s\n", i, keywords[i]);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (write_full(fd, str_data(str), str_len(str)) < 0) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mail_storage_set_critical(STORAGE(mk->mbox->storage),
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen "write_full(%s) failed: %m", mk->path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen (void)file_dotlock_delete(&dotlock);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen t_pop();
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* mtime must grow every time */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->synced_mtime = ioloop_time <= mk->synced_mtime ?
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->synced_mtime + 1 : ioloop_time;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ut.actime = ioloop_time;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ut.modtime = mk->synced_mtime;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (utime(lock_path, &ut) < 0) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mail_storage_set_critical(STORAGE(mk->mbox->storage),
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen "utime(%s) failed: %m", lock_path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (file_dotlock_replace(&dotlock, 0) < 0) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mail_storage_set_critical(STORAGE(mk->mbox->storage),
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen "file_dotlock_replace(%s) failed: %m", mk->path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen t_pop();
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->changed = FALSE;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen t_pop();
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return 0;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstruct maildir_keywords_sync_ctx *
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenmaildir_keywords_sync_init(struct maildir_keywords *mk,
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct mail_index *index)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct maildir_keywords_sync_ctx *ctx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ctx = i_new(struct maildir_keywords_sync_ctx, 1);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ctx->mk = mk;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ctx->index = index;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ctx->keywords = mail_index_get_keywords(index);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ARRAY_CREATE(&ctx->idx_to_chr, default_pool,
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen char, MAILDIR_MAX_KEYWORDS);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return ctx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenvoid maildir_keywords_sync_deinit(struct maildir_keywords_sync_ctx *ctx)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen maildir_keywords_commit(ctx->mk);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen array_free(&ctx->idx_to_chr);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_free(ctx);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenunsigned int maildir_keywords_char_idx(struct maildir_keywords_sync_ctx *ctx,
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen char keyword)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen const char *name;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int chridx, idx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_assert(keyword >= MAILDIR_KEYWORD_FIRST &&
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen keyword <= MAILDIR_KEYWORD_LAST);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen chridx = keyword - MAILDIR_KEYWORD_FIRST;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (ctx->chridx_to_idx[chridx] != 0)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return ctx->chridx_to_idx[chridx];
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* lookup / create */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen name = maildir_keywords_idx(ctx->mk, chridx);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (name == NULL) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* name is lost. just generate one ourself. */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen name = t_strdup_printf("unknown-%u", chridx);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen while (maildir_keywords_lookup(ctx->mk, name) >= 0) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* don't create a duplicate name.
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen keep changing the name until it doesn't exist */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen name = t_strconcat(name, "?", NULL);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen maildir_keywords_create(ctx->mk, name, chridx);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (!mail_index_keyword_lookup(ctx->index, name, TRUE, &idx))
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_unreached();
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ctx->chridx_to_idx[chridx] = idx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return idx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenchar maildir_keywords_idx_char(struct maildir_keywords_sync_ctx *ctx,
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int idx)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen const char *const *name_p;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen char *chr_p;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen int chridx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen chr_p = array_idx_modifyable(&ctx->idx_to_chr, idx);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (*chr_p != '\0')
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return *chr_p;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen name_p = array_idx(ctx->keywords, idx);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen chridx = maildir_keywords_lookup_or_create(ctx->mk, *name_p);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (chridx < 0)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return '\0';
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen *chr_p = chridx + MAILDIR_KEYWORD_FIRST;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return *chr_p;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}