Mercurial > dovecot > original-hg > dovecot-1.2
view src/plugins/mbox-snarf/mbox-snarf-plugin.c @ 9658:8ba4253adc9b HEAD tip
*-login: SSL connections didn't get closed when the client got destroyed.
author | Timo Sirainen <tss@iki.fi> |
---|---|
date | Thu, 08 May 2014 16:41:29 +0300 |
parents | 00cd9aacd03c |
children |
line wrap: on
line source
/* Copyright (c) 2007-2010 Dovecot authors, see the included COPYING file */ #include "lib.h" #include "array.h" #include "home-expand.h" #include "mail-search-build.h" #include "mail-storage-private.h" #include "mailbox-list-private.h" #include "mbox-snarf-plugin.h" #include <stdlib.h> #include <sys/stat.h> #define MBOX_SNARF_CONTEXT(obj) \ MODULE_CONTEXT(obj, mbox_snarf_storage_module) struct mbox_snarf_mail_storage { union mail_storage_module_context module_ctx; const char *snarf_inbox_path; bool open_spool_inbox; }; struct mbox_snarf_mailbox { union mailbox_module_context module_ctx; }; const char *mbox_snarf_plugin_version = PACKAGE_VERSION; static void (*mbox_snarf_next_hook_mail_storage_created) (struct mail_storage *storage); static MODULE_CONTEXT_DEFINE_INIT(mbox_snarf_storage_module, &mail_storage_module_register); static int mbox_snarf(struct mailbox *srcbox, struct mailbox *destbox) { struct mail_search_args *search_args; struct mail_search_context *search_ctx; struct mailbox_transaction_context *src_trans, *dest_trans; struct mail_save_context *save_ctx; struct mail *mail; enum mail_error error; int ret; if (mailbox_sync(srcbox, MAILBOX_SYNC_FLAG_FULL_READ, 0, NULL) < 0) return -1; src_trans = mailbox_transaction_begin(srcbox, 0); dest_trans = mailbox_transaction_begin(destbox, MAILBOX_TRANSACTION_FLAG_EXTERNAL); search_args = mail_search_build_init(); mail_search_build_add_all(search_args); search_ctx = mailbox_search_init(src_trans, search_args, NULL); mail_search_args_unref(&search_args); mail = mail_alloc(src_trans, MAIL_FETCH_STREAM_HEADER | MAIL_FETCH_STREAM_BODY, NULL); while ((ret = mailbox_search_next(search_ctx, mail)) > 0) { if (mail->expunged) continue; save_ctx = mailbox_save_alloc(dest_trans); if (mailbox_copy(&save_ctx, mail) < 0 && !mail->expunged) { (void)mail_storage_get_last_error(destbox->storage, &error); /* if we failed because of out of disk space, just move those messages we managed to move so far. */ if (error != MAIL_ERROR_NOSPACE) ret = -1; break; } mail_expunge(mail); } mail_free(&mail); if (mailbox_search_deinit(&search_ctx) < 0) ret = -1; /* commit the copied messages to the destination mailbox. if we crash between that and between expunging the messages from the source mailbox, we're left with duplicates. */ if (ret < 0) mailbox_transaction_rollback(&dest_trans); else if (mailbox_transaction_commit(&dest_trans) < 0) ret = -1; if (ret < 0) mailbox_transaction_rollback(&src_trans); else { if (mailbox_transaction_commit(&src_trans) < 0) ret = -1; } if (ret == 0) { if (mailbox_sync(srcbox, 0, 0, NULL) < 0) ret = -1; } return ret; } static struct mailbox_sync_context * mbox_snarf_sync_init(struct mailbox *box, enum mailbox_sync_flags flags) { struct mbox_snarf_mail_storage *mstorage = MBOX_SNARF_CONTEXT(box->storage); struct mail_storage *storage; struct mbox_snarf_mailbox *mbox = MBOX_SNARF_CONTEXT(box); struct mailbox *spool_mbox; struct mailbox_sync_context *ctx; /* try to open the spool mbox */ mstorage->open_spool_inbox = TRUE; storage = box->storage; spool_mbox = mailbox_open(&storage, "INBOX", NULL, MAILBOX_OPEN_KEEP_LOCKED | MAILBOX_OPEN_KEEP_RECENT | MAILBOX_OPEN_NO_INDEX_FILES); mstorage->open_spool_inbox = FALSE; if (spool_mbox != NULL) mbox_snarf(spool_mbox, box); ctx = mbox->module_ctx.super.sync_init(box, flags); if (spool_mbox != NULL) mailbox_close(&spool_mbox); return ctx; } static struct mailbox * mbox_snarf_mailbox_open(struct mail_storage *storage, const char *name, struct istream *input, enum mailbox_open_flags flags) { struct mbox_snarf_mail_storage *mstorage = MBOX_SNARF_CONTEXT(storage); struct mailbox_list *list; struct mailbox *box; struct mbox_snarf_mailbox *mbox; struct stat st; enum mail_storage_flags old_flags = storage->flags; enum mailbox_list_flags old_list_flags; bool use_snarfing = FALSE; list = mail_storage_get_list(storage); old_list_flags = list->flags; if (strcasecmp(name, "INBOX") == 0 && !mstorage->open_spool_inbox) { if (stat(mstorage->snarf_inbox_path, &st) == 0) { /* use ~/mbox as the INBOX */ name = mstorage->snarf_inbox_path; use_snarfing = TRUE; storage->flags |= MAIL_STORAGE_FLAG_FULL_FS_ACCESS; list->flags |= MAILBOX_LIST_FLAG_FULL_FS_ACCESS; } else if (errno != ENOENT) { mail_storage_set_critical(storage, "stat(%s) failed: %m", mstorage->snarf_inbox_path); } } box = mstorage->module_ctx.super. mailbox_open(storage, name, input, flags); storage->flags = old_flags; list->flags = old_list_flags; if (box == NULL || !use_snarfing) return box; mbox = p_new(box->pool, struct mbox_snarf_mailbox, 1); mbox->module_ctx.super = box->v; box->v.sync_init = mbox_snarf_sync_init; MODULE_CONTEXT_SET(box, mbox_snarf_storage_module, mbox); return box; } static void mbox_snarf_mail_storage_created(struct mail_storage *storage) { struct mbox_snarf_mail_storage *mstorage; const char *path; path = mail_user_home_expand(storage->ns->user, getenv("MBOX_SNARF")); mstorage = p_new(storage->pool, struct mbox_snarf_mail_storage, 1); mstorage->snarf_inbox_path = p_strdup(storage->pool, path); mstorage->module_ctx.super = storage->v; storage->v.mailbox_open = mbox_snarf_mailbox_open; MODULE_CONTEXT_SET(storage, mbox_snarf_storage_module, mstorage); if (mbox_snarf_next_hook_mail_storage_created != NULL) mbox_snarf_next_hook_mail_storage_created(storage); } void mbox_snarf_plugin_init(void) { const char *path; path = getenv("MBOX_SNARF"); if (path != NULL) { mbox_snarf_next_hook_mail_storage_created = hook_mail_storage_created; hook_mail_storage_created = mbox_snarf_mail_storage_created; } else if (getenv("DEBUG") != NULL) i_info("mbox_snarf: No mbox_snarf setting - plugin disabled"); } void mbox_snarf_plugin_deinit(void) { if (getenv("MBOX_SNARF") != NULL) { hook_mail_storage_created = mbox_snarf_next_hook_mail_storage_created; } }