index/maildir/maildir-keywords.c

	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}