plugins/virtual/virtual-config.c

	virtual-config.c revision 84078771687fabf75819918f0f3aecdc3ed08b36
5a580c3a38ced62d4bcc95b8ac7c4f2935b5d294Timo Sirainen/* Copyright (c) 2008-2010 Dovecot authors, see the included COPYING file */
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch#include "lib.h"
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch#include "array.h"
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch#include "crc32.h"
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch#include "istream.h"
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch#include "str.h"
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch#include "imap-parser.h"
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch#include "imap-match.h"
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch#include "mail-search-build.h"
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch#include "mail-search-parser.h"
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch#include "virtual-storage.h"
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch#include "virtual-plugin.h"
5afe997e79978b7e989aa3b0bfdf4a813ecdc6f6Timo Sirainen
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch#include <unistd.h>
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch#include <fcntl.h>
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Boschstruct virtual_parse_context {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    struct virtual_mailbox *mbox;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    struct istream *input;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    pool_t pool;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    string_t *rule;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    unsigned int rule_idx;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    char sep;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    bool have_wildcards;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch};
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Boschstatic struct mail_search_args *
a10ed8c47534b4c6b6bf2711ccfe577e720a47b4Timo Sirainenvirtual_search_args_parse(const string_t *rule, const char **error_r)
a10ed8c47534b4c6b6bf2711ccfe577e720a47b4Timo Sirainen{
a10ed8c47534b4c6b6bf2711ccfe577e720a47b4Timo Sirainen    struct istream *input;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    struct imap_parser *imap_parser;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    const struct imap_arg *args;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    struct mail_search_parser *parser;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    struct mail_search_args *sargs;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    bool fatal;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    int ret;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    if (str_len(rule) == 0) {
5afe997e79978b7e989aa3b0bfdf4a813ecdc6f6Timo Sirainen        sargs = mail_search_build_init();
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        mail_search_build_add_all(sargs);
5afe997e79978b7e989aa3b0bfdf4a813ecdc6f6Timo Sirainen        return sargs;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    }
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    input = i_stream_create_from_data(str_data(rule), str_len(rule));
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    (void)i_stream_read(input);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    imap_parser = imap_parser_create(input, NULL, (size_t)-1);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    ret = imap_parser_finish_line(imap_parser, 0,  0, &args);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    if (ret < 0) {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        sargs = NULL;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        *error_r = t_strdup(imap_parser_get_error(imap_parser, &fatal));
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    } else {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        parser = mail_search_parser_init_imap(args);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        if (mail_search_build(mail_search_register_get_imap(),
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch                      parser, "UTF-8", &sargs, error_r) < 0)
a10ed8c47534b4c6b6bf2711ccfe577e720a47b4Timo Sirainen            sargs = NULL;
a10ed8c47534b4c6b6bf2711ccfe577e720a47b4Timo Sirainen        mail_search_parser_deinit(&parser);
9a5980c7bb836f69a63082f4699c30596ea4ee74Timo Sirainen    }
a10ed8c47534b4c6b6bf2711ccfe577e720a47b4Timo Sirainen
a10ed8c47534b4c6b6bf2711ccfe577e720a47b4Timo Sirainen    imap_parser_destroy(&imap_parser);
a10ed8c47534b4c6b6bf2711ccfe577e720a47b4Timo Sirainen    i_stream_destroy(&input);
a10ed8c47534b4c6b6bf2711ccfe577e720a47b4Timo Sirainen    return sargs;
9a5980c7bb836f69a63082f4699c30596ea4ee74Timo Sirainen}
a10ed8c47534b4c6b6bf2711ccfe577e720a47b4Timo Sirainen
a10ed8c47534b4c6b6bf2711ccfe577e720a47b4Timo Sirainenstatic int
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Boschvirtual_config_add_rule(struct virtual_parse_context *ctx, const char **error_r)
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch{
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    struct virtual_backend_box *const *bboxes;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    struct mail_search_args *search_args;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    unsigned int i, count;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    if (ctx->rule_idx == array_count(&ctx->mbox->backend_boxes)) {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        i_assert(str_len(ctx->rule) == 0);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        return 0;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    }
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    ctx->mbox->search_args_crc32 =
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        crc32_str_more(ctx->mbox->search_args_crc32, str_c(ctx->rule));
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    search_args = virtual_search_args_parse(ctx->rule, error_r);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    str_truncate(ctx->rule, 0);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    if (search_args == NULL) {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        *error_r = t_strconcat("Previous search rule is invalid: ",
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch                       *error_r, NULL);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        return -1;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    }
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    /* update at all the mailboxes that were introduced since the previous
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch       rule. */
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    bboxes = array_get(&ctx->mbox->backend_boxes, &count);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    i_assert(ctx->rule_idx < count);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    for (i = ctx->rule_idx; i < count; i++) {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        i_assert(bboxes[i]->search_args == NULL);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        mail_search_args_ref(search_args);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        bboxes[i]->search_args = search_args;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    }
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    mail_search_args_unref(&search_args);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    ctx->rule_idx = array_count(&ctx->mbox->backend_boxes);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    return 0;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch}
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Boschstatic int
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Boschvirtual_config_parse_line(struct virtual_parse_context *ctx, const char *line,
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch              const char **error_r)
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch{
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    struct mail_user *user = ctx->mbox->storage->storage.user;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    struct virtual_backend_box *bbox;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    const char *name;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    if (*line == ' ' || *line == '\t') {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        /* continues the previous search rule */
5afe997e79978b7e989aa3b0bfdf4a813ecdc6f6Timo Sirainen        if (ctx->rule_idx == array_count(&ctx->mbox->backend_boxes)) {
5afe997e79978b7e989aa3b0bfdf4a813ecdc6f6Timo Sirainen            *error_r = "Search rule without a mailbox";
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch            return -1;
5afe997e79978b7e989aa3b0bfdf4a813ecdc6f6Timo Sirainen        }
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        str_append(ctx->rule, line);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        return 0;
d730192e34fbedbc590a5abc7351e5af5e120c5fTimo Sirainen    }
d730192e34fbedbc590a5abc7351e5af5e120c5fTimo Sirainen    /* if there is no rule yet, it means we want the previous mailboxes
d730192e34fbedbc590a5abc7351e5af5e120c5fTimo Sirainen       to use the rule that comes later */
d730192e34fbedbc590a5abc7351e5af5e120c5fTimo Sirainen    if (str_len(ctx->rule) > 0) {
d730192e34fbedbc590a5abc7351e5af5e120c5fTimo Sirainen        if (virtual_config_add_rule(ctx, error_r) < 0)
d730192e34fbedbc590a5abc7351e5af5e120c5fTimo Sirainen            return -1;
d730192e34fbedbc590a5abc7351e5af5e120c5fTimo Sirainen    }
d730192e34fbedbc590a5abc7351e5af5e120c5fTimo Sirainen
d730192e34fbedbc590a5abc7351e5af5e120c5fTimo Sirainen    /* new mailbox. the search args are added to it later. */
d730192e34fbedbc590a5abc7351e5af5e120c5fTimo Sirainen    bbox = p_new(ctx->pool, struct virtual_backend_box, 1);
d730192e34fbedbc590a5abc7351e5af5e120c5fTimo Sirainen    if (strcasecmp(line, "INBOX") == 0)
d730192e34fbedbc590a5abc7351e5af5e120c5fTimo Sirainen        line = "INBOX";
d730192e34fbedbc590a5abc7351e5af5e120c5fTimo Sirainen    bbox->name = p_strdup(ctx->pool, line);
d730192e34fbedbc590a5abc7351e5af5e120c5fTimo Sirainen    if (*line == '-') line++;
d730192e34fbedbc590a5abc7351e5af5e120c5fTimo Sirainen    bbox->ns = strcasecmp(line, "!INBOX") != 0 ?
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        mail_namespace_find(user->namespaces, &line) :
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        mail_namespace_find_inbox(user->namespaces);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    if (bbox->ns == NULL) {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        *error_r = t_strdup_printf("Namespace not found for %s",
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch                       bbox->name);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        return -1;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    }
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    if (strchr(bbox->name, '*') != NULL ||
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        strchr(bbox->name, '%') != NULL) {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        name = bbox->name[0] == '-' ? bbox->name + 1 : bbox->name;
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen        bbox->glob = imap_match_init(ctx->pool, name, TRUE, ctx->sep);
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen        ctx->have_wildcards = TRUE;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    } else if (bbox->name[0] == '!') {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        /* save messages here */
5afe997e79978b7e989aa3b0bfdf4a813ecdc6f6Timo Sirainen        if (ctx->mbox->save_bbox != NULL) {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch            *error_r = "Multiple save mailboxes defined";
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch            return -1;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        }
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        bbox->name++;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        ctx->mbox->save_bbox = bbox;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    }
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    array_append(&ctx->mbox->backend_boxes, &bbox, 1);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    return 0;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch}
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Boschstatic void
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Boschvirtual_mailbox_get_list_patterns(struct virtual_parse_context *ctx)
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch{
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    struct virtual_mailbox *mbox = ctx->mbox;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    ARRAY_TYPE(mailbox_virtual_patterns) *dest;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    struct mailbox_virtual_pattern pattern;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    struct virtual_backend_box *const *bboxes;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    unsigned int i, count;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    memset(&pattern, 0, sizeof(pattern));
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen    bboxes = array_get_modifiable(&mbox->backend_boxes, &count);
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen    p_array_init(&mbox->list_include_patterns, ctx->pool, count);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    p_array_init(&mbox->list_exclude_patterns, ctx->pool, count);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    for (i = 0; i < count; i++) {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        pattern.ns = bboxes[i]->ns;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        pattern.pattern = bboxes[i]->name;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        if (*pattern.pattern != '-')
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch            dest = &mbox->list_include_patterns;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        else {
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen            dest = &mbox->list_exclude_patterns;
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen            pattern.pattern++;
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen        }
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen        array_append(dest, &pattern, 1);
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen    }
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch}
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Boschstatic void
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainenseparate_wildcard_mailboxes(struct virtual_mailbox *mbox,
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch                ARRAY_TYPE(virtual_backend_box) *wildcard_boxes,
5afe997e79978b7e989aa3b0bfdf4a813ecdc6f6Timo Sirainen                ARRAY_TYPE(virtual_backend_box) *neg_boxes)
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch{
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    struct virtual_backend_box *const *bboxes;
5afe997e79978b7e989aa3b0bfdf4a813ecdc6f6Timo Sirainen    ARRAY_TYPE(virtual_backend_box) *dest;
5afe997e79978b7e989aa3b0bfdf4a813ecdc6f6Timo Sirainen    unsigned int i, count;
5afe997e79978b7e989aa3b0bfdf4a813ecdc6f6Timo Sirainen
5afe997e79978b7e989aa3b0bfdf4a813ecdc6f6Timo Sirainen    bboxes = array_get_modifiable(&mbox->backend_boxes, &count);
5afe997e79978b7e989aa3b0bfdf4a813ecdc6f6Timo Sirainen    t_array_init(wildcard_boxes, I_MIN(16, count));
5afe997e79978b7e989aa3b0bfdf4a813ecdc6f6Timo Sirainen    t_array_init(neg_boxes, 4);
5afe997e79978b7e989aa3b0bfdf4a813ecdc6f6Timo Sirainen    for (i = 0; i < count;) {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        if (*bboxes[i]->name == '-')
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch            dest = neg_boxes;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        else if (bboxes[i]->glob != NULL)
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch            dest = wildcard_boxes;
5afe997e79978b7e989aa3b0bfdf4a813ecdc6f6Timo Sirainen        else {
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen            dest = NULL;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch            i++;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        }
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen        if (dest != NULL) {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch            array_append(dest, &bboxes[i], 1);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch            array_delete(&mbox->backend_boxes, i, 1);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch            bboxes = array_get_modifiable(&mbox->backend_boxes,
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen                              &count);
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen        }
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen    }
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen}
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainenstatic void virtual_config_copy_expanded(struct virtual_parse_context *ctx,
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch                     struct virtual_backend_box *wbox,
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen                     const char *name)
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen{
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen    struct virtual_backend_box *bbox;
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen    bbox = p_new(ctx->pool, struct virtual_backend_box, 1);
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen    *bbox = *wbox;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    bbox->name = p_strdup(ctx->pool, name);
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen    bbox->glob = NULL;
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen    bbox->wildcard = TRUE;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    mail_search_args_ref(bbox->search_args);
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen    array_append(&ctx->mbox->backend_boxes, &bbox, 1);
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen}
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainenstatic bool virtual_config_match(const struct mailbox_info *info,
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen                 ARRAY_TYPE(virtual_backend_box) *boxes_arr,
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch                 unsigned int *idx_r)
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch{
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen    struct virtual_backend_box *const *boxes;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    unsigned int i, count;
4afd5082f38342fd688acb5796912329f57dd02cTimo Sirainen
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    boxes = array_get_modifiable(boxes_arr, &count);
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    for (i = 0; i < count; i++) {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        if (boxes[i]->glob != NULL) {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch            /* we match only one namespace for each pattern. */
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch            if (boxes[i]->ns == info->ns &&
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch                imap_match(boxes[i]->glob,
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch                       info->name) == IMAP_MATCH_YES) {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch                *idx_r = i;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch                return TRUE;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch            }
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        } else {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch            i_assert(boxes[i]->name[0] == '-');
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch            if (strcmp(boxes[i]->name + 1, info->name) == 0) {
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch                *idx_r = i;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch                return TRUE;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch            }
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        }
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    }
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    return FALSE;
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch}
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Boschstatic int virtual_config_expand_wildcards(struct virtual_parse_context *ctx)
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch{
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch    const enum namespace_type iter_ns_types =
0fe2992e4d09f3ae4cceea88c9871c832d67b461Stephan Bosch        NAMESPACE_PRIVATE | NAMESPACE_SHARED | NAMESPACE_PUBLIC;
    const enum mailbox_list_iter_flags iter_flags =
        MAILBOX_LIST_ITER_VIRTUAL_NAMES |
        MAILBOX_LIST_ITER_RETURN_NO_FLAGS;
    struct mail_user *user = ctx->mbox->storage->storage.user;
    ARRAY_TYPE(virtual_backend_box) wildcard_boxes, neg_boxes;
    struct mailbox_list_iterate_context *iter;
    struct virtual_backend_box *const *wboxes;
    const char **patterns;
    const struct mailbox_info *info;
    unsigned int i, j, count;

    separate_wildcard_mailboxes(ctx->mbox, &wildcard_boxes, &neg_boxes);

    /* get patterns we want to list */
    wboxes = array_get_modifiable(&wildcard_boxes, &count);
    if (count == 0) {
        /* only negative wildcards - doesn't really make sense.
           just ignore. */
        return 0;
    }
    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,
                         iter_ns_types, iter_flags);
    while ((info = mailbox_list_iter_next(iter)) != NULL) {
        /* skip non-selectable mailboxes (especially mbox
           directories) */
        if ((info->flags & MAILBOX_NOSELECT) != 0)
            continue;

        if (virtual_config_match(info, &wildcard_boxes, &i) &&
            !virtual_config_match(info, &neg_boxes, &j) &&
            virtual_backend_box_lookup_name(ctx->mbox,
                            info->name) == NULL) {
            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);
}

static void virtual_config_search_args_dup(struct virtual_mailbox *mbox)
{
    struct virtual_backend_box *const *bboxes;
    struct mail_search_args *old_args;
    unsigned int i, count;

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

int virtual_config_read(struct virtual_mailbox *mbox)
{
    struct mail_user *user = mbox->storage->storage.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->box.path, "/"VIRTUAL_CONFIG_FNAME, NULL);
    fd = open(path, O_RDONLY);
    if (fd == -1) {
        if (errno == ENOENT) {
            mail_storage_set_error(mbox->box.storage,
                MAIL_ERROR_NOTPOSSIBLE,
                "Virtual mailbox missing configuration file");
            return -1;
        }
        mail_storage_set_critical(mbox->box.storage,
                      "open(%s) failed: %m", path);
        return -1;
    }

    memset(&ctx, 0, sizeof(ctx));
    ctx.sep = mail_namespaces_get_root_sep(user->namespaces);
    ctx.mbox = mbox;
    ctx.pool = mbox->box.pool;
    ctx.rule = t_str_new(256);
    ctx.input = i_stream_create_fd(fd, (size_t)-1, FALSE);
    i_stream_set_return_partial_line(ctx.input, TRUE);
    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->box.storage,
                          "%s: Error at line %u: %s",
                          path, linenum, error);
            break;
        }
    }
    if (ret == 0) {
        ret = virtual_config_add_rule(&ctx, &error);
        if (ret < 0) {
            mail_storage_set_critical(mbox->box.storage,
                          "%s: Error at line %u: %s",
                          path, linenum, error);
        }
    }

    virtual_mailbox_get_list_patterns(&ctx);
    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->box.storage,
                      "%s: No mailboxes defined", path);
        ret = -1;
    }
    if (ret == 0)
        virtual_config_search_args_dup(mbox);
    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);
    }
    array_free(&mbox->backend_boxes);
}