plugins/virtual/virtual-config.c

	virtual-config.c revision 7e209b78ca757294dbbc15604c88673b3a6b0c39
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen/* Copyright (c) 2008 Dovecot authors, see the included COPYING file */
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include "lib.h"
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include "array.h"
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include "crc32.h"
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include "istream.h"
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include "str.h"
5b62dea2f88165f3f4d87bba9011343f3ff415ffTimo Sirainen#include "imap-parser.h"
b9f564d00b7a115f465ffd6840341c7b8f9bfc8aTimo Sirainen#include "imap-match.h"
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include "mail-search-build.h"
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include "virtual-storage.h"
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include "virtual-plugin.h"
6c073ea23cd0e3760b072727cc18114860c27647Timo Sirainen
1fb81cb0b622cef1690c96bcc6a3e183e0b1e6ffTimo Sirainen#include <unistd.h>
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include <fcntl.h>
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainenstruct virtual_parse_context {
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    struct virtual_mailbox *mbox;
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    struct istream *input;
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    pool_t pool;
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    string_t *rule;
2271d1a3dfc7191e610f039e86b9245bbc5dfb8cTimo Sirainen    unsigned int rule_idx;
2271d1a3dfc7191e610f039e86b9245bbc5dfb8cTimo Sirainen
2271d1a3dfc7191e610f039e86b9245bbc5dfb8cTimo Sirainen    char sep;
9865d9e7c5713e41db939222ed9c0225a11fb99eTimo Sirainen    bool have_wildcards;
9865d9e7c5713e41db939222ed9c0225a11fb99eTimo Sirainen};
9865d9e7c5713e41db939222ed9c0225a11fb99eTimo Sirainen
9865d9e7c5713e41db939222ed9c0225a11fb99eTimo Sirainenstatic struct mail_search_args *
9865d9e7c5713e41db939222ed9c0225a11fb99eTimo Sirainenvirtual_search_args_parse(const string_t *rule, const char **error_r)
9865d9e7c5713e41db939222ed9c0225a11fb99eTimo Sirainen{
9865d9e7c5713e41db939222ed9c0225a11fb99eTimo Sirainen    struct istream *input;
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    struct imap_parser *parser;
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    const struct imap_arg *args;
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    struct mail_search_args *sargs;
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    bool fatal;
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    int ret;
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    input = i_stream_create_from_data(str_data(rule), str_len(rule));
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    (void)i_stream_read(input);
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    parser = imap_parser_create(input, NULL, (size_t)-1);
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    ret = imap_parser_finish_line(parser, 0,  0, &args);
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    if (ret < 0) {
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen        sargs = NULL;
dd4f30895ebbddd77e000472fbadcb3128ae2883Timo Sirainen        *error_r = t_strdup(imap_parser_get_error(parser, &fatal));
dd4f30895ebbddd77e000472fbadcb3128ae2883Timo Sirainen    } else if (mail_search_build_from_imap_args(args, "UTF-8", &sargs,
b9f564d00b7a115f465ffd6840341c7b8f9bfc8aTimo Sirainen                            error_r) < 0)
b9f564d00b7a115f465ffd6840341c7b8f9bfc8aTimo Sirainen        sargs = NULL;
b9f564d00b7a115f465ffd6840341c7b8f9bfc8aTimo Sirainen
57bf90f66f393c2807b2fc543655013f61d1d9e4Timo Sirainen    imap_parser_destroy(&parser);
57bf90f66f393c2807b2fc543655013f61d1d9e4Timo Sirainen    i_stream_destroy(&input);
5b62dea2f88165f3f4d87bba9011343f3ff415ffTimo Sirainen    return sargs;
5b62dea2f88165f3f4d87bba9011343f3ff415ffTimo Sirainen}
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainenstatic int
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainenvirtual_config_add_rule(struct virtual_parse_context *ctx, const char **error_r)
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen{
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    struct virtual_backend_box *const *bboxes;
b9f564d00b7a115f465ffd6840341c7b8f9bfc8aTimo Sirainen    struct mail_search_args *search_args;
b9f564d00b7a115f465ffd6840341c7b8f9bfc8aTimo Sirainen    unsigned int i, count;
b9f564d00b7a115f465ffd6840341c7b8f9bfc8aTimo Sirainen
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    if (ctx->rule_idx == array_count(&ctx->mbox->backend_boxes)) {
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen        i_assert(str_len(ctx->rule) == 0);
804fa3f03bd9170272168a5ad214053bbe3160c7Josef 'Jeff' Sipek        return 0;
804fa3f03bd9170272168a5ad214053bbe3160c7Josef 'Jeff' Sipek    }
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    ctx->mbox->search_args_crc32 =
b9f564d00b7a115f465ffd6840341c7b8f9bfc8aTimo Sirainen        crc32_str_more(ctx->mbox->search_args_crc32, str_c(ctx->rule));
b9f564d00b7a115f465ffd6840341c7b8f9bfc8aTimo Sirainen    search_args = virtual_search_args_parse(ctx->rule, error_r);
b9f564d00b7a115f465ffd6840341c7b8f9bfc8aTimo Sirainen    str_truncate(ctx->rule, 0);
b9f564d00b7a115f465ffd6840341c7b8f9bfc8aTimo Sirainen    if (search_args == NULL) {
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen        *error_r = t_strconcat("Previous search rule is invalid: ",
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen                       *error_r, NULL);
7b15788793354ca0fd4fdb4dda8e426ca1e9e9d7Timo Sirainen        return -1;
7b15788793354ca0fd4fdb4dda8e426ca1e9e9d7Timo Sirainen    }
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    /* update at all the mailboxes that were introduced since the previous
48d8312488089dc1a8360991f0881d91095c21eaTimo Sirainen       rule. */
48d8312488089dc1a8360991f0881d91095c21eaTimo Sirainen    bboxes = array_get(&ctx->mbox->backend_boxes, &count);
755abfa2bc6c5f072519b545faa1487357046b27Timo Sirainen    i_assert(ctx->rule_idx < count);
9865d9e7c5713e41db939222ed9c0225a11fb99eTimo Sirainen    for (i = ctx->rule_idx; i < count; i++) {
9865d9e7c5713e41db939222ed9c0225a11fb99eTimo Sirainen        i_assert(bboxes[i]->search_args == NULL);
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen        mail_search_args_ref(search_args);
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen        bboxes[i]->search_args = search_args;
    }
    mail_search_args_unref(&search_args);

    ctx->rule_idx = array_count(&ctx->mbox->backend_boxes);
    return 0;
}

static int
virtual_config_parse_line(struct virtual_parse_context *ctx, const char *line,
              const char **error_r)
{
    struct mail_user *user = ctx->mbox->storage->storage.ns->user;
    struct virtual_backend_box *bbox;

    if (*line == ' ') {
        /* continues the previous search rule */
        if (ctx->rule_idx == array_count(&ctx->mbox->backend_boxes)) {
            *error_r = "Search rule without a mailbox";
            return -1;
        }
        str_append(ctx->rule, line);
        return 0;
    }
    if (virtual_config_add_rule(ctx, error_r) < 0)
        return -1;

    /* new mailbox. the search args are added to it later. */
    bbox = p_new(ctx->pool, struct virtual_backend_box, 1);
    bbox->name = p_strdup(ctx->pool, line);
    if (strchr(bbox->name, '*') != NULL ||
        strchr(bbox->name, '%') != NULL) {
        bbox->glob = imap_match_init(ctx->pool, bbox->name,
                         TRUE, ctx->sep);
        bbox->ns = mail_namespace_find(user->namespaces, &line);
        ctx->have_wildcards = TRUE;
    }
    array_append(&ctx->mbox->backend_boxes, &bbox, 1);
    return 0;
}

static void
separate_wildcard_mailboxes(struct virtual_mailbox *mbox,
                ARRAY_TYPE(virtual_backend_box) *wildcard_boxes)
{
    struct virtual_backend_box *const *bboxes;
    unsigned int i, count;

    bboxes = array_get_modifiable(&mbox->backend_boxes, &count);
    t_array_init(wildcard_boxes, I_MIN(16, count));
    for (i = 0; i < count;) {
        if (bboxes[i]->glob == NULL)
            i++;
        else {
            array_append(wildcard_boxes, &bboxes[i], 1);
            array_delete(&mbox->backend_boxes, i, 1);
            bboxes = array_get_modifiable(&mbox->backend_boxes,
                              &count);
        }
    }
}

static void virtual_config_copy_expanded(struct virtual_parse_context *ctx,
                     struct virtual_backend_box *wbox,
                     const char *name)
{
    struct virtual_backend_box *bbox;

    bbox = p_new(ctx->pool, struct virtual_backend_box, 1);
    *bbox = *wbox;
    bbox->name = p_strdup(ctx->pool, name);
    bbox->glob = NULL;
    mail_search_args_ref(bbox->search_args);
    array_append(&ctx->mbox->backend_boxes, &bbox, 1);
}

static int virtual_config_expand_wildcards(struct virtual_parse_context *ctx)
{
    struct mail_user *user = ctx->mbox->storage->storage.ns->user;
    ARRAY_TYPE(virtual_backend_box) wildcard_boxes;
    struct mailbox_list_iterate_context *iter;
    struct virtual_backend_box *const *wboxes;
    const char **patterns;
    const struct mailbox_info *info;
    unsigned int i, count;

    separate_wildcard_mailboxes(ctx->mbox, &wildcard_boxes);

    /* get patterns we want to list */
    wboxes = array_get_modifiable(&wildcard_boxes, &count);
    patterns = t_new(const char *, count + 1);
    for (i = 0; i < count; i++)
        patterns[i] = wboxes[i]->name;

    /* match listed mailboxes to wildcards */
    iter = mailbox_list_iter_init_namespaces(user->namespaces, patterns,
                    MAILBOX_LIST_ITER_VIRTUAL_NAMES |
                    MAILBOX_LIST_ITER_RETURN_NO_FLAGS);
    while ((info = mailbox_list_iter_next(iter)) != NULL) {
        for (i = 0; i < count; i++) {
            if (wboxes[i]->ns == info->ns &&
                imap_match(wboxes[i]->glob,
                       info->name) == IMAP_MATCH_YES) {
                virtual_config_copy_expanded(ctx, wboxes[i],
                                 info->name);
            }
        }
    }
    for (i = 0; i < count; i++)
        mail_search_args_unref(&wboxes[i]->search_args);
    return mailbox_list_iter_deinit(&iter);
}

int virtual_config_read(struct virtual_mailbox *mbox)
{
    struct mail_user *user = mbox->storage->storage.ns->user;
    struct virtual_parse_context ctx;
    const char *path, *line, *error;
    unsigned int linenum = 0;
    int fd, ret = 0;

    i_array_init(&mbox->backend_boxes, 8);
    mbox->search_args_crc32 = (uint32_t)-1;

    path = t_strconcat(mbox->path, "/"VIRTUAL_CONFIG_FNAME, NULL);
    fd = open(path, O_RDWR);
    if (fd == -1) {
        if (errno == ENOENT) {
            mail_storage_set_error(mbox->ibox.storage,
                MAIL_ERROR_NOTPOSSIBLE,
                "Virtual mailbox missing configuration file");
            return -1;
        }
        mail_storage_set_critical(mbox->ibox.storage,
                      "open(%s) failed: %m", path);
        return -1;
    }

    memset(&ctx, 0, sizeof(ctx));
    ctx.sep = mail_namespace_get_root_sep(user->namespaces);
    ctx.mbox = mbox;
    ctx.pool = mbox->ibox.box.pool;
    ctx.rule = t_str_new(256);
    ctx.input = i_stream_create_fd(fd, (size_t)-1, FALSE);
    while ((line = i_stream_read_next_line(ctx.input)) != NULL) {
        linenum++;
        if (*line == '#')
            continue;
        if (*line == '\0')
            ret = virtual_config_add_rule(&ctx, &error);
        else
            ret = virtual_config_parse_line(&ctx, line, &error);
        if (ret < 0) {
            mail_storage_set_critical(mbox->ibox.storage,
                          "%s: Error at line %u: %s",
                          path, linenum, error);
            break;
        }
    }
    if (ret == 0)
        ret = virtual_config_add_rule(&ctx, &error);

    if (ret == 0 && ctx.have_wildcards)
        ret = virtual_config_expand_wildcards(&ctx);

    if (ret == 0 && array_count(&mbox->backend_boxes) == 0) {
        mail_storage_set_critical(mbox->ibox.storage,
                      "%s: No mailboxes defined", path);
        ret = -1;
    }
    i_stream_unref(&ctx.input);
    (void)close(fd);
    return ret;
}

void virtual_config_free(struct virtual_mailbox *mbox)
{
    struct virtual_backend_box *const *bboxes;
    unsigned int i, count;

    bboxes = array_get_modifiable(&mbox->backend_boxes, &count);
    for (i = 0; i < count; i++) {
        if (bboxes[i]->search_args != NULL)
            mail_search_args_unref(&bboxes[i]->search_args);
    }
}