index/cydir/cydir-sync.c

	cydir-sync.c revision 817d027593510c3ba70ad542ce0011f5f6916d1e
/* Copyright (c) 2007-2015 Dovecot authors, see the included COPYING file */

#include "lib.h"
#include "ioloop.h"
#include "str.h"
#include "cydir-storage.h"
#include "cydir-sync.h"

static void cydir_sync_set_uidvalidity(struct cydir_sync_context *ctx)
{
    uint32_t uid_validity = ioloop_time;

    mail_index_update_header(ctx->trans,
        offsetof(struct mail_index_header, uid_validity),
        &uid_validity, sizeof(uid_validity), TRUE);
    ctx->uid_validity = uid_validity;
}

static string_t *cydir_get_path_prefix(struct cydir_mailbox *mbox)
{
    string_t *path = str_new(default_pool, 256);

    str_append(path, mailbox_get_path(&mbox->box));
    str_append_c(path, '/');
    return path;
}

static void
cydir_sync_expunge(struct cydir_sync_context *ctx, uint32_t seq1, uint32_t seq2)
{
    struct mailbox *box = &ctx->mbox->box;
    uint32_t uid;

    if (ctx->path == NULL) {
        ctx->path = cydir_get_path_prefix(ctx->mbox);
        ctx->path_dir_prefix_len = str_len(ctx->path);
    }

    for (; seq1 <= seq2; seq1++) {
        mail_index_lookup_uid(ctx->sync_view, seq1, &uid);

        str_truncate(ctx->path, ctx->path_dir_prefix_len);
        str_printfa(ctx->path, "%u.", uid);
        if (unlink(str_c(ctx->path)) == 0) {
            if (box->v.sync_notify != NULL) {
                box->v.sync_notify(box, uid,
                           MAILBOX_SYNC_TYPE_EXPUNGE);
            }
            mail_index_expunge(ctx->trans, seq1);
        } else if (errno != ENOENT) {
            mail_storage_set_critical(&ctx->mbox->storage->storage,
                "unlink(%s) failed: %m", str_c(ctx->path));
            /* continue anyway */
        }
    }
}

static void cydir_sync_index(struct cydir_sync_context *ctx)
{
    struct mailbox *box = &ctx->mbox->box;
    const struct mail_index_header *hdr;
    struct mail_index_sync_rec sync_rec;
    uint32_t seq1, seq2;

    hdr = mail_index_get_header(ctx->sync_view);
    if (hdr->uid_validity != 0)
        ctx->uid_validity = hdr->uid_validity;
    else
        cydir_sync_set_uidvalidity(ctx);

    /* mark the newly seen messages as recent */
    if (mail_index_lookup_seq_range(ctx->sync_view, hdr->first_recent_uid,
                    hdr->next_uid, &seq1, &seq2)) {
        mailbox_recent_flags_set_seqs(&ctx->mbox->box, ctx->sync_view,
                          seq1, seq2);
    }

    while (mail_index_sync_next(ctx->index_sync_ctx, &sync_rec)) {
        if (!mail_index_lookup_seq_range(ctx->sync_view,
                         sync_rec.uid1, sync_rec.uid2,
                         &seq1, &seq2)) {
            /* already expunged, nothing to do. */
            continue;
        }

        switch (sync_rec.type) {
        case MAIL_INDEX_SYNC_TYPE_EXPUNGE:
            cydir_sync_expunge(ctx, seq1, seq2);
            break;
        case MAIL_INDEX_SYNC_TYPE_FLAGS:
        case MAIL_INDEX_SYNC_TYPE_KEYWORD_ADD:
        case MAIL_INDEX_SYNC_TYPE_KEYWORD_REMOVE:
            /* FIXME: should be bother calling sync_notify()? */
            break;
        }
    }

    if (box->v.sync_notify != NULL)
        box->v.sync_notify(box, 0, 0);
}

int cydir_sync_begin(struct cydir_mailbox *mbox,
             struct cydir_sync_context **ctx_r, bool force)
{
    struct cydir_sync_context *ctx;
    enum mail_index_sync_flags sync_flags;
    int ret;

    ctx = i_new(struct cydir_sync_context, 1);
    ctx->mbox = mbox;

    sync_flags = index_storage_get_sync_flags(&mbox->box) |
        MAIL_INDEX_SYNC_FLAG_FLUSH_DIRTY;
    if (!force)
        sync_flags |= MAIL_INDEX_SYNC_FLAG_REQUIRE_CHANGES;

    ret = mail_index_sync_begin(mbox->box.index, &ctx->index_sync_ctx,
                    &ctx->sync_view, &ctx->trans,
                    sync_flags);
    if (ret <= 0) {
        if (ret < 0)
            mailbox_set_index_error(&mbox->box);
        i_free(ctx);
        *ctx_r = NULL;
        return ret;
    }

    cydir_sync_index(ctx);
    *ctx_r = ctx;
    return 0;
}

int cydir_sync_finish(struct cydir_sync_context **_ctx, bool success)
{
    struct cydir_sync_context *ctx = *_ctx;
    int ret = success ? 0 : -1;

    *_ctx = NULL;
    if (success) {
        if (mail_index_sync_commit(&ctx->index_sync_ctx) < 0) {
            mailbox_set_index_error(&ctx->mbox->box);
            ret = -1;
        }
    } else {
        mail_index_sync_rollback(&ctx->index_sync_ctx);
    }
    if (ctx->path != NULL)
        str_free(&ctx->path);
    i_free(ctx);
    return ret;
}

static int cydir_sync(struct cydir_mailbox *mbox)
{
    struct cydir_sync_context *sync_ctx;

    if (cydir_sync_begin(mbox, &sync_ctx, FALSE) < 0)
        return -1;

    return sync_ctx == NULL ? 0 :
        cydir_sync_finish(&sync_ctx, TRUE);
}

struct mailbox_sync_context *
cydir_storage_sync_init(struct mailbox *box, enum mailbox_sync_flags flags)
{
    struct cydir_mailbox *mbox = (struct cydir_mailbox *)box;
    int ret = 0;

    if (!box->opened) {
        if (mailbox_open(box) < 0)
            ret = -1;
    }

    if (index_mailbox_want_full_sync(&mbox->box, flags) && ret == 0)
        ret = cydir_sync(mbox);

    return index_mailbox_sync_init(box, flags, ret < 0);
}