quota-clone-plugin.c revision bcb4e51a409d94ae670de96afb8483a4f7855294
e59faf65ce864fe95dc00f5d52b8323cdbd0608aTimo Sirainen/* Copyright (c) 2015-2018 Dovecot authors, see the included COPYING file */
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "lib.h"
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen#include "module-context.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "ioloop.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "dict.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "mail-storage-private.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "quota.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "quota-clone-plugin.h"
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen/* If mailbox is kept open for this many milliseconds after quota update,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen flush quota-clone. */
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#define QUOTA_CLONE_FLUSH_DELAY_MSECS (10*1000)
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#define DICT_QUOTA_CLONE_PATH DICT_PATH_PRIVATE"quota/"
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen#define DICT_QUOTA_CLONE_BYTES_PATH DICT_QUOTA_CLONE_PATH"storage"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#define DICT_QUOTA_CLONE_COUNT_PATH DICT_QUOTA_CLONE_PATH"messages"
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
0cb5a9bfbf40b3b323956792aa13d342a459585eTimo Sirainen#define QUOTA_CLONE_USER_CONTEXT(obj) \
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen MODULE_CONTEXT(obj, quota_clone_user_module)
0cb5a9bfbf40b3b323956792aa13d342a459585eTimo Sirainen#define QUOTA_CLONE_CONTEXT(obj) \
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen MODULE_CONTEXT(obj, quota_clone_storage_module)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
a272994d43de80a306a8ed1f2983960d1f3102d0Timo Sirainenstatic MODULE_CONTEXT_DEFINE_INIT(quota_clone_user_module,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen &mail_user_module_register);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic MODULE_CONTEXT_DEFINE_INIT(quota_clone_storage_module,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen &mail_storage_module_register);
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
d1414c09cf0d58ac983054e2f4e1a1f329272dcfTimo Sirainenstruct quota_clone_user {
d1414c09cf0d58ac983054e2f4e1a1f329272dcfTimo Sirainen union mail_user_module_context module_ctx;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen struct dict *dict;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen bool quota_flushing;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen};
d1414c09cf0d58ac983054e2f4e1a1f329272dcfTimo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstruct quota_clone_mailbox {
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen union mailbox_module_context module_ctx;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct timeout *to_quota_flush;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen bool quota_changed;
d1414c09cf0d58ac983054e2f4e1a1f329272dcfTimo Sirainen};
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic void quota_clone_flush_real(struct mailbox *box)
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen{
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen struct quota_clone_mailbox *qbox = QUOTA_CLONE_CONTEXT(box);
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen struct quota_clone_user *quser =
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen QUOTA_CLONE_USER_CONTEXT(box->storage->user);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct dict_transaction_context *trans;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct quota_root_iter *iter;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen struct quota_root *root;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen uint64_t bytes_value, count_value, limit;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen const char *error;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen enum quota_get_result bytes_res, count_res;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen /* we'll clone the first quota root */
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen iter = quota_root_iter_init(box);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen root = quota_root_iter_next(iter);
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen quota_root_iter_deinit(&iter);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (root == NULL) {
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen /* no quota roots defined for this mailbox - ignore */
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen qbox->quota_changed = FALSE;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen return;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen }
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen /* get new values first */
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen bytes_res = quota_get_resource(root, "", QUOTA_NAME_STORAGE_BYTES,
d1414c09cf0d58ac983054e2f4e1a1f329272dcfTimo Sirainen &bytes_value, &limit, &error);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (bytes_res == QUOTA_GET_RESULT_INTERNAL_ERROR) {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen i_error("quota_clone_plugin: "
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen "Failed to get quota resource "QUOTA_NAME_STORAGE_BYTES": %s",
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen error);
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen return;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen }
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen count_res = quota_get_resource(root, "", QUOTA_NAME_MESSAGES,
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen &count_value, &limit, &error);
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen if (count_res == QUOTA_GET_RESULT_INTERNAL_ERROR) {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen i_error("quota_clone_plugin: "
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen "Failed to get quota resource "QUOTA_NAME_MESSAGES": %s",
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen error);
d1414c09cf0d58ac983054e2f4e1a1f329272dcfTimo Sirainen return;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen }
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (bytes_res == QUOTA_GET_RESULT_UNKNOWN_RESOURCE &&
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen count_res == QUOTA_GET_RESULT_UNKNOWN_RESOURCE) {
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen /* quota resources don't exist - no point in updating it */
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen return;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen }
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (bytes_res == QUOTA_GET_RESULT_BACKGROUND_CALC &&
d1414c09cf0d58ac983054e2f4e1a1f329272dcfTimo Sirainen count_res == QUOTA_GET_RESULT_BACKGROUND_CALC) {
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen /* Blocked by an ongoing quota calculation - try again later */
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen return;
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen }
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen
d1414c09cf0d58ac983054e2f4e1a1f329272dcfTimo Sirainen /* Then update the resources that exist. The resources' existence can't
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen change unless the quota backend is changed, so we don't worry about
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen the special case of lookup changing from
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen RESULT_LIMITED/RESULT_UNLIMITED to RESULT_UNKNOWN_RESOURCE, which
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen leaves the old value unchanged. */
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen trans = dict_transaction_begin(quser->dict);
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen if (bytes_res == QUOTA_GET_RESULT_LIMITED ||
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen bytes_res == QUOTA_GET_RESULT_UNLIMITED) {
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen dict_set(trans, DICT_QUOTA_CLONE_BYTES_PATH,
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen t_strdup_printf("%"PRIu64, bytes_value));
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen }
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen if (count_res == QUOTA_GET_RESULT_LIMITED ||
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen count_res == QUOTA_GET_RESULT_UNLIMITED) {
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen dict_set(trans, DICT_QUOTA_CLONE_COUNT_PATH,
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen t_strdup_printf("%"PRIu64, count_value));
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen }
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen if (dict_transaction_commit(&trans, &error) < 0)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen i_error("quota_clone_plugin: Failed to commit dict update: %s", error);
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen else
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen qbox->quota_changed = FALSE;
e2ae85924b0ef1a7c97e021a3b901b498f599c18Timo Sirainen}
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
e2ae85924b0ef1a7c97e021a3b901b498f599c18Timo Sirainenstatic void quota_clone_flush(struct mailbox *box)
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen{
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen struct quota_clone_mailbox *qbox = QUOTA_CLONE_CONTEXT(box);
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen struct quota_clone_user *quser =
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen QUOTA_CLONE_USER_CONTEXT(box->storage->user);
e2ae85924b0ef1a7c97e021a3b901b498f599c18Timo Sirainen
0cb5a9bfbf40b3b323956792aa13d342a459585eTimo Sirainen timeout_remove(&qbox->to_quota_flush);
0cb5a9bfbf40b3b323956792aa13d342a459585eTimo Sirainen
0cb5a9bfbf40b3b323956792aa13d342a459585eTimo Sirainen if (quser->quota_flushing) {
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen /* recursing back from quota recalculation */
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen } else if (qbox->quota_changed) {
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen quser->quota_flushing = TRUE;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen quota_clone_flush_real(box);
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen quser->quota_flushing = FALSE;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen }
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen}
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic void quota_clone_changed(struct mailbox *box)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct quota_clone_mailbox *qbox = QUOTA_CLONE_CONTEXT(box);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen qbox->quota_changed = TRUE;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (qbox->to_quota_flush == NULL) {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen qbox->to_quota_flush = timeout_add(QUOTA_CLONE_FLUSH_DELAY_MSECS,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen quota_clone_flush, box);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen }
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen}
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic int quota_clone_save_finish(struct mail_save_context *ctx)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct quota_clone_mailbox *qbox =
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen QUOTA_CLONE_CONTEXT(ctx->transaction->box);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen quota_clone_changed(ctx->transaction->box);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen return qbox->module_ctx.super.save_finish(ctx);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen}
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic int
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenquota_clone_copy(struct mail_save_context *ctx, struct mail *mail)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct quota_clone_mailbox *qbox =
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen QUOTA_CLONE_CONTEXT(ctx->transaction->box);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen quota_clone_changed(ctx->transaction->box);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen return qbox->module_ctx.super.copy(ctx, mail);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen}
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic void
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenquota_clone_mailbox_sync_notify(struct mailbox *box, uint32_t uid,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen enum mailbox_sync_type sync_type)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct quota_clone_mailbox *qbox = QUOTA_CLONE_CONTEXT(box);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (qbox->module_ctx.super.sync_notify != NULL)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen qbox->module_ctx.super.sync_notify(box, uid, sync_type);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (sync_type == MAILBOX_SYNC_TYPE_EXPUNGE)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen quota_clone_changed(box);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen}
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic void quota_clone_mailbox_close(struct mailbox *box)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct quota_clone_mailbox *qbox = QUOTA_CLONE_CONTEXT(box);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen qbox->module_ctx.super.close(box);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen quota_clone_flush(box);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen}
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic void quota_clone_mailbox_allocated(struct mailbox *box)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct quota_clone_user *quser =
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen QUOTA_CLONE_USER_CONTEXT(box->storage->user);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct mailbox_vfuncs *v = box->vlast;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct quota_clone_mailbox *qbox;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (quser == NULL)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen return;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen qbox = p_new(box->pool, struct quota_clone_mailbox, 1);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen qbox->module_ctx.super = *v;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen box->vlast = &qbox->module_ctx.super;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen v->save_finish = quota_clone_save_finish;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen v->copy = quota_clone_copy;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen v->sync_notify = quota_clone_mailbox_sync_notify;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen v->close = quota_clone_mailbox_close;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen MODULE_CONTEXT_SET(box, quota_clone_storage_module, qbox);
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen}
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic void quota_clone_mail_user_deinit(struct mail_user *user)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
e5acc283bf030b0b5c79ca4e52d315c516a299faPascal Volk struct quota_clone_user *quser = QUOTA_CLONE_USER_CONTEXT(user);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen dict_deinit(&quser->dict);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen quser->module_ctx.super.deinit(user);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen}
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic void quota_clone_mail_user_created(struct mail_user *user)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct quota_clone_user *quser;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct mail_user_vfuncs *v = user->vlast;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct dict_settings dict_set;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct dict *dict;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen const char *uri, *error;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen uri = mail_user_plugin_getenv(user, "quota_clone_dict");
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen if (uri == NULL || uri[0] == '\0') {
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen if (user->mail_debug) {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen i_debug("The quota_clone_dict setting is missing from configuration");
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen }
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen return;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen }
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen i_zero(&dict_set);
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen dict_set.username = user->username;
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen dict_set.base_dir = user->set->base_dir;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen (void)mail_user_get_home(user, &dict_set.home_dir);
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen if (dict_init(uri, &dict_set, &dict, &error) < 0) {
b6e1d85292485a7fb4cfa5f40dd1ec131ab07cc1Timo Sirainen i_error("quota_clone_dict: Failed to initialize '%s': %s",
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen uri, error);
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen return;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen }
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen quser = p_new(user->pool, struct quota_clone_user, 1);
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen quser->module_ctx.super = *v;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen user->vlast = &quser->module_ctx.super;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen v->deinit = quota_clone_mail_user_deinit;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen quser->dict = dict;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen MODULE_CONTEXT_SET(user, quota_clone_user_module, quser);
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen}
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainenstatic struct mail_storage_hooks quota_clone_mail_storage_hooks = {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen .mailbox_allocated = quota_clone_mailbox_allocated,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen .mail_user_created = quota_clone_mail_user_created
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen};
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainenvoid quota_clone_plugin_init(struct module *module ATTR_UNUSED)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen mail_storage_hooks_add(module, &quota_clone_mail_storage_hooks);
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen}
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainenvoid quota_clone_plugin_deinit(void)
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen{
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen mail_storage_hooks_remove(&quota_clone_mail_storage_hooks);
4b8c92b4773677a7b4064816e469eeafc976ba75Timo Sirainen}
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainenconst char *quota_clone_plugin_dependencies[] = { "quota", NULL };
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen