quota-storage.c revision 1f5670bce3e5ec0c1ddf8d95c57ebb428f1339f6
5f5870385cff47efd2f58e7892f251cf13761528Timo Sirainen/* Copyright (c) 2005-2012 Dovecot authors, see the included COPYING file */
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen#include "lib.h"
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen#include "array.h"
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen#include "istream.h"
c4b376dd6e0c423006d7ac83a39253bcaf8e7c47Timo Sirainen#include "mail-search-build.h"
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen#include "mail-storage-private.h"
b039dabf4c53f72454e795930e7643b6e0e625f9Timo Sirainen#include "mailbox-list-private.h"
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen#include "maildir-storage.h"
82995cc154a929f37aa486a72a6485e9f8d34a30Timo Sirainen#include "quota-private.h"
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen#include "quota-plugin.h"
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen#include <sys/stat.h>
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen#define QUOTA_CONTEXT(obj) \
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen MODULE_CONTEXT(obj, quota_storage_module)
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen#define QUOTA_MAIL_CONTEXT(obj) \
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen MODULE_CONTEXT(obj, quota_mail_module)
b039dabf4c53f72454e795930e7643b6e0e625f9Timo Sirainen#define QUOTA_LIST_CONTEXT(obj) \
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen MODULE_CONTEXT(obj, quota_mailbox_list_module)
b039dabf4c53f72454e795930e7643b6e0e625f9Timo Sirainen
b039dabf4c53f72454e795930e7643b6e0e625f9Timo Sirainenstruct quota_mailbox_list {
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen union mailbox_list_module_context module_ctx;
b039dabf4c53f72454e795930e7643b6e0e625f9Timo Sirainen};
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainenstruct quota_mailbox {
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen union mailbox_module_context module_ctx;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen struct mailbox_transaction_context *expunge_trans;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen struct quota_transaction_context *expunge_qt;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen ARRAY_DEFINE(expunge_uids, uint32_t);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen ARRAY_DEFINE(expunge_sizes, uoff_t);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen unsigned int recalculate:1;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen};
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainenstruct quota_user_module quota_user_module =
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen MODULE_CONTEXT_INIT(&mail_user_module_register);
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainenstatic MODULE_CONTEXT_DEFINE_INIT(quota_storage_module,
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen &mail_storage_module_register);
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainenstatic MODULE_CONTEXT_DEFINE_INIT(quota_mail_module, &mail_module_register);
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainenstatic MODULE_CONTEXT_DEFINE_INIT(quota_mailbox_list_module,
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen &mailbox_list_module_register);
b039dabf4c53f72454e795930e7643b6e0e625f9Timo Sirainen
73b50eecfc31750a312e2f940023f522eb07178cTimo Sirainenstatic void quota_mail_expunge(struct mail *_mail)
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen{
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen struct mail_private *mail = (struct mail_private *)_mail;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen struct quota_mailbox *qbox = QUOTA_CONTEXT(_mail->box);
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen union mail_module_context *qmail = QUOTA_MAIL_CONTEXT(mail);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen uoff_t size;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen /* We need to handle the situation where multiple transactions expunged
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen the mail at the same time. In here we'll just save the message's
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen physical size and do the quota freeing later when the message was
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen known to be expunged. */
83bb013a99f0936995f9c7a1077822662d8fefdbTimo Sirainen if (mail_get_physical_size(_mail, &size) == 0) {
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen if (!array_is_created(&qbox->expunge_uids)) {
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen i_array_init(&qbox->expunge_uids, 64);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen i_array_init(&qbox->expunge_sizes, 64);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen }
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen array_append(&qbox->expunge_uids, &_mail->uid, 1);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen array_append(&qbox->expunge_sizes, &size, 1);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen }
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
73b50eecfc31750a312e2f940023f522eb07178cTimo Sirainen qmail->super.expunge(_mail);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen}
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainenstatic int
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainenquota_get_status(struct mailbox *box, enum mailbox_status_items items,
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen struct mailbox_status *status_r)
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen{
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen struct quota_mailbox *qbox = QUOTA_CONTEXT(box);
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen struct quota_transaction_context *qt;
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen bool too_large;
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen int ret = 0;
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen if ((items & STATUS_CHECK_OVER_QUOTA) != 0) {
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen qt = quota_transaction_begin(box);
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen if ((ret = quota_test_alloc(qt, 1, &too_large)) == 0) {
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen mail_storage_set_error(box->storage, MAIL_ERROR_NOSPACE,
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen qt->quota->set->quota_exceeded_msg);
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen ret = -1;
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen }
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen quota_transaction_rollback(&qt);
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen if ((items & ~STATUS_CHECK_OVER_QUOTA) == 0) {
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen /* don't bother calling parent, it may unnecessarily
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen try to open the mailbox */
95d50c6f87501978687a1bc0685d8bc469645e7fTimo Sirainen return ret < 0 ? -1 : 0;
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen }
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen }
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen if (qbox->module_ctx.super.get_status(box, items, status_r) < 0)
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen ret = -1;
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen return ret < 0 ? -1 : 0;
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen}
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainenstatic struct mailbox_transaction_context *
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainenquota_mailbox_transaction_begin(struct mailbox *box,
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen enum mailbox_transaction_flags flags)
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen{
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen struct quota_mailbox *qbox = QUOTA_CONTEXT(box);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen struct mailbox_transaction_context *t;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen struct quota_transaction_context *qt;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen t = qbox->module_ctx.super.transaction_begin(box, flags);
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen qt = quota_transaction_begin(box);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen MODULE_CONTEXT_SET(t, quota_storage_module, qt);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen return t;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen}
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainenstatic int
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainenquota_mailbox_transaction_commit(struct mailbox_transaction_context *ctx,
96308127e006bb3b1108093bcf4cc1fd9481cb7aTimo Sirainen struct mail_transaction_commit_changes *changes_r)
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen{
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen struct quota_mailbox *qbox = QUOTA_CONTEXT(ctx->box);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen struct quota_transaction_context *qt = QUOTA_CONTEXT(ctx);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
12ea3aeeb61df84632f04f86f47902a3750d61f8Timo Sirainen if (qt->tmp_mail != NULL)
12ea3aeeb61df84632f04f86f47902a3750d61f8Timo Sirainen mail_free(&qt->tmp_mail);
12ea3aeeb61df84632f04f86f47902a3750d61f8Timo Sirainen
96308127e006bb3b1108093bcf4cc1fd9481cb7aTimo Sirainen if (qbox->module_ctx.super.transaction_commit(ctx, changes_r) < 0) {
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen quota_transaction_rollback(&qt);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen return -1;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen } else {
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen (void)quota_transaction_commit(&qt);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen return 0;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen }
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen}
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainenstatic void
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainenquota_mailbox_transaction_rollback(struct mailbox_transaction_context *ctx)
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen{
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen struct quota_mailbox *qbox = QUOTA_CONTEXT(ctx->box);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen struct quota_transaction_context *qt = QUOTA_CONTEXT(ctx);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen if (qt->tmp_mail != NULL)
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen mail_free(&qt->tmp_mail);
26550dc10a8c2057d09894c4e3f9dc12b124e90cTimo Sirainen
26550dc10a8c2057d09894c4e3f9dc12b124e90cTimo Sirainen qbox->module_ctx.super.transaction_rollback(ctx);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen quota_transaction_rollback(&qt);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen}
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainenvoid quota_mail_allocated(struct mail *_mail)
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen{
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen struct quota_mailbox *qbox = QUOTA_CONTEXT(_mail->box);
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen struct mail_private *mail = (struct mail_private *)_mail;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen struct mail_vfuncs *v = mail->vlast;
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen union mail_module_context *qmail;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen if (qbox == NULL)
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen return;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen qmail = p_new(mail->pool, union mail_module_context, 1);
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen qmail->super = *v;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen mail->vlast = &qmail->super;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen v->expunge = quota_mail_expunge;
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen MODULE_CONTEXT_SET_SELF(mail, quota_mail_module, qmail);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen}
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainenstatic int quota_check(struct mailbox_transaction_context *t, struct mail *mail)
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen{
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen struct quota_transaction_context *qt = QUOTA_CONTEXT(t);
dcd50ecbfe796bd76f2d63483c534cc0e4e94164Timo Sirainen int ret;
dcd50ecbfe796bd76f2d63483c534cc0e4e94164Timo Sirainen bool too_large;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen ret = quota_try_alloc(qt, mail, &too_large);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen if (ret > 0)
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen return 0;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen else if (ret == 0) {
ff7056842f14fd3b30a2d327dfab165b9d15dd30Timo Sirainen mail_storage_set_error(t->box->storage, MAIL_ERROR_NOSPACE,
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen qt->quota->set->quota_exceeded_msg);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen return -1;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen } else {
ff7056842f14fd3b30a2d327dfab165b9d15dd30Timo Sirainen mail_storage_set_critical(t->box->storage,
ff7056842f14fd3b30a2d327dfab165b9d15dd30Timo Sirainen "Internal quota calculation error");
6f42ea227d5623d7960c8d8cd29e668870a78752Timo Sirainen return qt->quota->set->ignore_save_errors ? 0 : -1;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen }
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen}
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainenstatic int
bd4d0a1a7c0626452b8d82f37e3ec07267ac9896Timo Sirainenquota_copy(struct mail_save_context *ctx, struct mail *mail)
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen{
bd4d0a1a7c0626452b8d82f37e3ec07267ac9896Timo Sirainen struct mailbox_transaction_context *t = ctx->transaction;
2ebeb22b9a8a8bb7fbe2f2e2908478a220792b87Timo Sirainen struct quota_transaction_context *qt = QUOTA_CONTEXT(t);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen struct quota_mailbox *qbox = QUOTA_CONTEXT(t->box);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
bd4d0a1a7c0626452b8d82f37e3ec07267ac9896Timo Sirainen if (ctx->dest_mail == NULL) {
2ebeb22b9a8a8bb7fbe2f2e2908478a220792b87Timo Sirainen /* we always want to know the mail size */
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen if (qt->tmp_mail == NULL) {
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen qt->tmp_mail = mail_alloc(t, MAIL_FETCH_PHYSICAL_SIZE,
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen NULL);
2ebeb22b9a8a8bb7fbe2f2e2908478a220792b87Timo Sirainen }
bd4d0a1a7c0626452b8d82f37e3ec07267ac9896Timo Sirainen ctx->dest_mail = qt->tmp_mail;
2ebeb22b9a8a8bb7fbe2f2e2908478a220792b87Timo Sirainen }
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
bd4d0a1a7c0626452b8d82f37e3ec07267ac9896Timo Sirainen if (qbox->module_ctx.super.copy(ctx, mail) < 0)
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen return -1;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
263e4b2733768062cb0b8b8917cad78fa2a04ff9Timo Sirainen /* if copying used saving internally, we already checked the quota */
abec3f4c5ec486c393e1513950f2d4819dcbc30fTimo Sirainen return ctx->copying_via_save ? 0 : quota_check(t, ctx->dest_mail);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen}
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainenstatic int
e3aeeb634245e80d4f643f8d2eea11d6b72336d8Timo Sirainenquota_save_begin(struct mail_save_context *ctx, struct istream *input)
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen{
e3aeeb634245e80d4f643f8d2eea11d6b72336d8Timo Sirainen struct mailbox_transaction_context *t = ctx->transaction;
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen struct quota_transaction_context *qt = QUOTA_CONTEXT(t);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen struct quota_mailbox *qbox = QUOTA_CONTEXT(t->box);
4b9f99761df5014c659cd87fddaf6854af428cfcTimo Sirainen uoff_t size;
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen int ret;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
4b9f99761df5014c659cd87fddaf6854af428cfcTimo Sirainen if (i_stream_get_size(input, TRUE, &size) > 0) {
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen /* Input size is known, check for quota immediately. This
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen check isn't perfect, especially because input stream's
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen linefeeds may contain CR+LFs while physical message would
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen only contain LFs. With mbox some headers might be skipped
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen entirely.
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen I think these don't really matter though compared to the
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen benefit of giving "out of quota" error before sending the
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen full mail. */
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen bool too_large;
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen
4b9f99761df5014c659cd87fddaf6854af428cfcTimo Sirainen ret = quota_test_alloc(qt, size, &too_large);
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen if (ret == 0) {
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen mail_storage_set_error(t->box->storage,
b87436ebb957a9eb182be72ba00e2c8eae59a2e4Timo Sirainen MAIL_ERROR_NOSPACE,
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen qt->quota->set->quota_exceeded_msg);
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen return -1;
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen } else if (ret < 0) {
ff7056842f14fd3b30a2d327dfab165b9d15dd30Timo Sirainen mail_storage_set_critical(t->box->storage,
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen "Internal quota calculation error");
6f42ea227d5623d7960c8d8cd29e668870a78752Timo Sirainen if (!qt->quota->set->ignore_save_errors)
6f42ea227d5623d7960c8d8cd29e668870a78752Timo Sirainen return -1;
4981827cb5e32cf767b7b0e3070137e6b36f42afTimo Sirainen }
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen }
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
e3aeeb634245e80d4f643f8d2eea11d6b72336d8Timo Sirainen if (ctx->dest_mail == NULL) {
2ebeb22b9a8a8bb7fbe2f2e2908478a220792b87Timo Sirainen /* we always want to know the mail size */
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen if (qt->tmp_mail == NULL) {
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen qt->tmp_mail = mail_alloc(t, MAIL_FETCH_PHYSICAL_SIZE,
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen NULL);
2ebeb22b9a8a8bb7fbe2f2e2908478a220792b87Timo Sirainen }
e3aeeb634245e80d4f643f8d2eea11d6b72336d8Timo Sirainen ctx->dest_mail = qt->tmp_mail;
2ebeb22b9a8a8bb7fbe2f2e2908478a220792b87Timo Sirainen }
2ebeb22b9a8a8bb7fbe2f2e2908478a220792b87Timo Sirainen
e3aeeb634245e80d4f643f8d2eea11d6b72336d8Timo Sirainen return qbox->module_ctx.super.save_begin(ctx, input);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen}
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
2ebeb22b9a8a8bb7fbe2f2e2908478a220792b87Timo Sirainenstatic int quota_save_finish(struct mail_save_context *ctx)
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen{
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen struct quota_mailbox *qbox = QUOTA_CONTEXT(ctx->transaction->box);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen if (qbox->module_ctx.super.save_finish(ctx) < 0)
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen return -1;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
cf942dce0253075911a96cff323b5f30eb654ae0Timo Sirainen return quota_check(ctx->transaction, ctx->dest_mail);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen}
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainenstatic void quota_mailbox_sync_cleanup(struct quota_mailbox *qbox)
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen{
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen if (array_is_created(&qbox->expunge_uids)) {
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen array_clear(&qbox->expunge_uids);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen array_clear(&qbox->expunge_sizes);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen }
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen if (qbox->expunge_qt != NULL && qbox->expunge_qt->tmp_mail != NULL) {
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen mail_free(&qbox->expunge_qt->tmp_mail);
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen mailbox_transaction_rollback(&qbox->expunge_trans);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen }
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen}
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainenstatic void quota_mailbox_sync_commit(struct quota_mailbox *qbox)
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen{
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen quota_mailbox_sync_cleanup(qbox);
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen if (qbox->expunge_qt != NULL)
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen (void)quota_transaction_commit(&qbox->expunge_qt);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen qbox->recalculate = FALSE;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen}
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainenstatic void quota_mailbox_sync_notify(struct mailbox *box, uint32_t uid,
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen enum mailbox_sync_type sync_type)
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen{
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen struct quota_mailbox *qbox = QUOTA_CONTEXT(box);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen const uint32_t *uids;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen const uoff_t *sizep;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen unsigned int i, count;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen uoff_t size;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen if (qbox->module_ctx.super.sync_notify != NULL)
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen qbox->module_ctx.super.sync_notify(box, uid, sync_type);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen if (sync_type != MAILBOX_SYNC_TYPE_EXPUNGE || qbox->recalculate) {
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen if (uid == 0) {
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen /* free the transaction before view syncing begins,
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen otherwise it'll crash. */
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen quota_mailbox_sync_cleanup(qbox);
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen }
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen return;
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen }
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen /* we're in the middle of syncing the mailbox, so it's a bad idea to
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen try and get the message sizes at this point. Rely on sizes that
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen we saved earlier, or recalculate the whole quota if we don't know
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen the size. */
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen if (!array_is_created(&qbox->expunge_uids)) {
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen i = count = 0;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen } else {
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen uids = array_get(&qbox->expunge_uids, &count);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen for (i = 0; i < count; i++) {
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen if (uids[i] == uid)
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen break;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen }
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen }
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen if (qbox->expunge_qt == NULL)
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen qbox->expunge_qt = quota_transaction_begin(box);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen if (i != count) {
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen /* we already know the size */
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen sizep = array_idx(&qbox->expunge_sizes, i);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen quota_free_bytes(qbox->expunge_qt, *sizep);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen return;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen }
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen /* try to look up the size. this works only if it's cached. */
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen if (qbox->expunge_qt->tmp_mail == NULL) {
6ae329de09afb7214c906d762320847e05469d53Timo Sirainen /* FIXME: ugly kludge to open the transaction for sync_view.
6ae329de09afb7214c906d762320847e05469d53Timo Sirainen box->view may not have all the new messages that
6ae329de09afb7214c906d762320847e05469d53Timo Sirainen sync_notify() notifies about, and those messages would
6ae329de09afb7214c906d762320847e05469d53Timo Sirainen cause a quota recalculation. */
6ae329de09afb7214c906d762320847e05469d53Timo Sirainen struct mail_index_view *box_view = box->view;
6ae329de09afb7214c906d762320847e05469d53Timo Sirainen if (box->tmp_sync_view != NULL)
6ae329de09afb7214c906d762320847e05469d53Timo Sirainen box->view = box->tmp_sync_view;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen qbox->expunge_trans = mailbox_transaction_begin(box, 0);
6ae329de09afb7214c906d762320847e05469d53Timo Sirainen box->view = box_view;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen qbox->expunge_qt->tmp_mail =
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen mail_alloc(qbox->expunge_trans,
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen MAIL_FETCH_PHYSICAL_SIZE, NULL);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen }
83bb013a99f0936995f9c7a1077822662d8fefdbTimo Sirainen if (mail_set_uid(qbox->expunge_qt->tmp_mail, uid) &&
83bb013a99f0936995f9c7a1077822662d8fefdbTimo Sirainen mail_get_physical_size(qbox->expunge_qt->tmp_mail, &size) == 0)
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen quota_free_bytes(qbox->expunge_qt, size);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen else {
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen /* there's no way to get the size. recalculate the quota. */
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen quota_recalculate(qbox->expunge_qt);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen qbox->recalculate = TRUE;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen }
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen}
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainenstatic int quota_mailbox_sync_deinit(struct mailbox_sync_context *ctx,
2584e86cc2d8c31ba30a4109cf4ba09d1e37e28aTimo Sirainen struct mailbox_sync_status *status_r)
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen{
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen struct quota_mailbox *qbox = QUOTA_CONTEXT(ctx->box);
c1563f46bd9ae4ad20f37f003220443a6224fb9fTimo Sirainen int ret;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
2584e86cc2d8c31ba30a4109cf4ba09d1e37e28aTimo Sirainen ret = qbox->module_ctx.super.sync_deinit(ctx, status_r);
c1563f46bd9ae4ad20f37f003220443a6224fb9fTimo Sirainen /* update quota only after syncing is finished. the quota commit may
c1563f46bd9ae4ad20f37f003220443a6224fb9fTimo Sirainen recalculate the quota and cause all mailboxes to be synced,
c1563f46bd9ae4ad20f37f003220443a6224fb9fTimo Sirainen including the one we're already syncing. */
3091db1d26fed038841a3a81dd5ff507d0b375ceTimo Sirainen quota_mailbox_sync_commit(qbox);
c1563f46bd9ae4ad20f37f003220443a6224fb9fTimo Sirainen return ret;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen}
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainenstatic void quota_roots_flush(struct quota *quota)
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen{
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen struct quota_root *const *roots;
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen unsigned int i, count;
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen roots = array_get(&quota->roots, &count);
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen for (i = 0; i < count; i++) {
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen if (roots[i]->backend.v.flush != NULL)
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen roots[i]->backend.v.flush(roots[i]);
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen }
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen}
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen
1e9c53531a31e3ad0168872e513046a3222a1afdTimo Sirainenstatic void quota_mailbox_close(struct mailbox *box)
1e9c53531a31e3ad0168872e513046a3222a1afdTimo Sirainen{
1e9c53531a31e3ad0168872e513046a3222a1afdTimo Sirainen struct quota_mailbox *qbox = QUOTA_CONTEXT(box);
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen struct quota_user *quser = QUOTA_USER_CONTEXT(box->storage->user);
1e9c53531a31e3ad0168872e513046a3222a1afdTimo Sirainen
1e9c53531a31e3ad0168872e513046a3222a1afdTimo Sirainen /* sync_notify() may be called outside sync_begin()..sync_deinit().
1e9c53531a31e3ad0168872e513046a3222a1afdTimo Sirainen make sure we apply changes at close time at latest. */
1e9c53531a31e3ad0168872e513046a3222a1afdTimo Sirainen quota_mailbox_sync_commit(qbox);
1e9c53531a31e3ad0168872e513046a3222a1afdTimo Sirainen
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen /* make sure quota backend flushes all data. this could also be done
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen somewhat later, but user.deinit() is too late, since the flushing
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen can trigger quota recalculation which isn't safe to do anymore
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen at user.deinit() when most of the loaded plugins have already been
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen deinitialized. */
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen quota_roots_flush(quser->quota);
048e40f9364fa68482bc276dd4a5d595a3d742e9Timo Sirainen
1e9c53531a31e3ad0168872e513046a3222a1afdTimo Sirainen qbox->module_ctx.super.close(box);
1e9c53531a31e3ad0168872e513046a3222a1afdTimo Sirainen}
1e9c53531a31e3ad0168872e513046a3222a1afdTimo Sirainen
e03d986a74128f5ba30fcfda9f6e36578f5d8decTimo Sirainenstatic void quota_mailbox_free(struct mailbox *box)
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen{
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen struct quota_mailbox *qbox = QUOTA_CONTEXT(box);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen if (array_is_created(&qbox->expunge_uids)) {
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen array_free(&qbox->expunge_uids);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen array_free(&qbox->expunge_sizes);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen }
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen i_assert(qbox->expunge_qt == NULL ||
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen qbox->expunge_qt->tmp_mail == NULL);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
e03d986a74128f5ba30fcfda9f6e36578f5d8decTimo Sirainen qbox->module_ctx.super.free(box);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen}
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
ea9d9d99948cff5f9b881f79b28fa3b80da0f2a7Timo Sirainenvoid quota_mailbox_allocated(struct mailbox *box)
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen{
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen struct mailbox_vfuncs *v = box->vlast;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen struct quota_mailbox *qbox;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
ea9d9d99948cff5f9b881f79b28fa3b80da0f2a7Timo Sirainen if (QUOTA_LIST_CONTEXT(box->list) == NULL)
ea9d9d99948cff5f9b881f79b28fa3b80da0f2a7Timo Sirainen return;
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
40c24f8fe31037b61f2ac594a6dfcd123080b4b5Timo Sirainen if ((box->storage->class_flags & MAIL_STORAGE_CLASS_FLAG_NOQUOTA) != 0)
40c24f8fe31037b61f2ac594a6dfcd123080b4b5Timo Sirainen return;
40c24f8fe31037b61f2ac594a6dfcd123080b4b5Timo Sirainen
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen qbox = p_new(box->pool, struct quota_mailbox, 1);
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen qbox->module_ctx.super = *v;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen box->vlast = &qbox->module_ctx.super;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen
9357960a30d4936de5e261bf522d83f7d3f49986Timo Sirainen v->get_status = quota_get_status;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen v->transaction_begin = quota_mailbox_transaction_begin;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen v->transaction_commit = quota_mailbox_transaction_commit;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen v->transaction_rollback = quota_mailbox_transaction_rollback;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen v->save_begin = quota_save_begin;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen v->save_finish = quota_save_finish;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen v->copy = quota_copy;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen v->sync_notify = quota_mailbox_sync_notify;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen v->sync_deinit = quota_mailbox_sync_deinit;
1e9c53531a31e3ad0168872e513046a3222a1afdTimo Sirainen v->close = quota_mailbox_close;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen v->free = quota_mailbox_free;
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen MODULE_CONTEXT_SET(box, quota_storage_module, qbox);
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen}
1795e934ebcd58175d3b5bbdd811b13c7889efa3Timo Sirainen
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainenstatic void quota_mailbox_list_deinit(struct mailbox_list *list)
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen{
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen struct quota_mailbox_list *qlist = QUOTA_LIST_CONTEXT(list);
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen quota_remove_user_namespace(list->ns);
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen qlist->module_ctx.super.deinit(list);
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen}
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainenstruct quota *quota_get_mail_user_quota(struct mail_user *user)
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen{
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen struct quota_user *quser = QUOTA_USER_CONTEXT(user);
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen return quser->quota;
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen}
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainenstatic void quota_user_deinit(struct mail_user *user)
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen{
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen struct quota_user *quser = QUOTA_USER_CONTEXT(user);
bb34c6538784b564c05eefccc74af2bd7c9505f5Timo Sirainen struct quota_settings *quota_set = quser->quota->set;
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen quota_deinit(&quser->quota);
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen quser->module_ctx.super.deinit(user);
bb34c6538784b564c05eefccc74af2bd7c9505f5Timo Sirainen
bb34c6538784b564c05eefccc74af2bd7c9505f5Timo Sirainen quota_settings_deinit(&quota_set);
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen}
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainenvoid quota_mail_user_created(struct mail_user *user)
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen{
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen struct mail_user_vfuncs *v = user->vlast;
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen struct quota_user *quser;
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen struct quota_settings *set;
4316355ca8b7698516272520a972291378698140Timo Sirainen struct quota *quota;
4316355ca8b7698516272520a972291378698140Timo Sirainen const char *error;
4316355ca8b7698516272520a972291378698140Timo Sirainen int ret;
4316355ca8b7698516272520a972291378698140Timo Sirainen
4316355ca8b7698516272520a972291378698140Timo Sirainen if ((ret = quota_user_read_settings(user, &set, &error)) > 0) {
4316355ca8b7698516272520a972291378698140Timo Sirainen if (quota_init(set, user, &quota, &error) < 0) {
4316355ca8b7698516272520a972291378698140Timo Sirainen quota_settings_deinit(&set);
4316355ca8b7698516272520a972291378698140Timo Sirainen ret = -1;
4316355ca8b7698516272520a972291378698140Timo Sirainen }
4316355ca8b7698516272520a972291378698140Timo Sirainen }
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen
4316355ca8b7698516272520a972291378698140Timo Sirainen if (ret < 0) {
4316355ca8b7698516272520a972291378698140Timo Sirainen user->error = p_strdup_printf(user->pool,
4316355ca8b7698516272520a972291378698140Timo Sirainen "Failed to initialize quota: %s", error);
4316355ca8b7698516272520a972291378698140Timo Sirainen return;
4316355ca8b7698516272520a972291378698140Timo Sirainen }
4316355ca8b7698516272520a972291378698140Timo Sirainen if (ret > 0) {
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen quser = p_new(user->pool, struct quota_user, 1);
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen quser->module_ctx.super = *v;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen user->vlast = &quser->module_ctx.super;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen v->deinit = quota_user_deinit;
4316355ca8b7698516272520a972291378698140Timo Sirainen quser->quota = quota;
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen MODULE_CONTEXT_SET(user, quota_user_module, quser);
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen } else if (user->mail_debug) {
e5acc283bf030b0b5c79ca4e52d315c516a299faPascal Volk i_debug("quota: No quota setting - plugin disabled");
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen }
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen}
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainenstatic struct quota_root *
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainenquota_find_root_for_ns(struct quota *quota, struct mail_namespace *ns)
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen{
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen struct quota_root *const *roots;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen unsigned int i, count;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen roots = array_get(&quota->roots, &count);
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen for (i = 0; i < count; i++) {
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen if (roots[i]->ns_prefix != NULL &&
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen strcmp(roots[i]->ns_prefix, ns->prefix) == 0)
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen return roots[i];
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen }
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen return NULL;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen}
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen
d7c2ae49f036ef256f63bd2d437c53da122c665eTimo Sirainenvoid quota_mailbox_list_created(struct mailbox_list *list)
b039dabf4c53f72454e795930e7643b6e0e625f9Timo Sirainen{
b039dabf4c53f72454e795930e7643b6e0e625f9Timo Sirainen struct quota_mailbox_list *qlist;
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen struct quota *quota = NULL;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen struct quota_root *root;
1f5670bce3e5ec0c1ddf8d95c57ebb428f1339f6Timo Sirainen struct mail_user *quota_user;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen bool add;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen
a9a77f4632b1f00cc3c6664c91ef5f23746e099bTimo Sirainen if (QUOTA_USER_CONTEXT(list->ns->user) == NULL)
a9a77f4632b1f00cc3c6664c91ef5f23746e099bTimo Sirainen return;
a9a77f4632b1f00cc3c6664c91ef5f23746e099bTimo Sirainen
805872b9f885c1cc4bddf0b6ff71ca61f6cdd7bfTimo Sirainen /* see if we have a quota explicitly defined for this namespace */
d7c2ae49f036ef256f63bd2d437c53da122c665eTimo Sirainen quota = quota_get_mail_user_quota(list->ns->user);
d7c2ae49f036ef256f63bd2d437c53da122c665eTimo Sirainen root = quota_find_root_for_ns(quota, list->ns);
1f5670bce3e5ec0c1ddf8d95c57ebb428f1339f6Timo Sirainen if (root != NULL) {
1f5670bce3e5ec0c1ddf8d95c57ebb428f1339f6Timo Sirainen /* explicit quota root */
d7c2ae49f036ef256f63bd2d437c53da122c665eTimo Sirainen root->ns = list->ns;
1f5670bce3e5ec0c1ddf8d95c57ebb428f1339f6Timo Sirainen quota_user = list->ns->user;
1f5670bce3e5ec0c1ddf8d95c57ebb428f1339f6Timo Sirainen } else {
1f5670bce3e5ec0c1ddf8d95c57ebb428f1339f6Timo Sirainen quota_user = list->ns->owner != NULL ?
1f5670bce3e5ec0c1ddf8d95c57ebb428f1339f6Timo Sirainen list->ns->owner : list->ns->user;
1f5670bce3e5ec0c1ddf8d95c57ebb428f1339f6Timo Sirainen }
805872b9f885c1cc4bddf0b6ff71ca61f6cdd7bfTimo Sirainen
40c24f8fe31037b61f2ac594a6dfcd123080b4b5Timo Sirainen if ((list->ns->flags & NAMESPACE_FLAG_NOQUOTA) != 0)
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen add = FALSE;
d7c2ae49f036ef256f63bd2d437c53da122c665eTimo Sirainen else if (list->ns->owner == NULL) {
805872b9f885c1cc4bddf0b6ff71ca61f6cdd7bfTimo Sirainen /* public namespace - add quota only if namespace is
805872b9f885c1cc4bddf0b6ff71ca61f6cdd7bfTimo Sirainen explicitly defined for it */
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen add = root != NULL;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen } else {
1f5670bce3e5ec0c1ddf8d95c57ebb428f1339f6Timo Sirainen /* for shared namespaces add only if the owner has quota
1f5670bce3e5ec0c1ddf8d95c57ebb428f1339f6Timo Sirainen enabled */
1f5670bce3e5ec0c1ddf8d95c57ebb428f1339f6Timo Sirainen add = QUOTA_USER_CONTEXT(quota_user) != NULL;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen }
b039dabf4c53f72454e795930e7643b6e0e625f9Timo Sirainen
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen if (add) {
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen struct mailbox_list_vfuncs *v = list->vlast;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen
f7fcdca120f1b591f885408d08fc90cfd725bf94Timo Sirainen qlist = p_new(list->pool, struct quota_mailbox_list, 1);
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen qlist->module_ctx.super = *v;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen list->vlast = &qlist->module_ctx.super;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen v->deinit = quota_mailbox_list_deinit;
f7fcdca120f1b591f885408d08fc90cfd725bf94Timo Sirainen MODULE_CONTEXT_SET(list, quota_mailbox_list_module, qlist);
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen
1f5670bce3e5ec0c1ddf8d95c57ebb428f1339f6Timo Sirainen quota = quota_get_mail_user_quota(quota_user);
d7c2ae49f036ef256f63bd2d437c53da122c665eTimo Sirainen quota_add_user_namespace(quota, list->ns);
f7fcdca120f1b591f885408d08fc90cfd725bf94Timo Sirainen }
b039dabf4c53f72454e795930e7643b6e0e625f9Timo Sirainen}
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainenstatic void quota_root_set_namespace(struct quota_root *root,
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen struct mail_namespace *namespaces)
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen{
e20e638805c4bd54e039891a3e92760b1dfa189aTimo Sirainen const struct quota_rule *rule;
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen const char *name;
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen if (root->ns_prefix != NULL && root->ns == NULL) {
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen root->ns = mail_namespace_find_prefix(namespaces,
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen root->ns_prefix);
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen if (root->ns == NULL) {
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen i_error("quota: Unknown namespace: %s",
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen root->ns_prefix);
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen }
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen }
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen
e20e638805c4bd54e039891a3e92760b1dfa189aTimo Sirainen array_foreach(&root->set->rules, rule) {
e20e638805c4bd54e039891a3e92760b1dfa189aTimo Sirainen name = rule->mailbox_name;
c0a87e5f3316a57e6f915882fa1951d0fbb74a61Timo Sirainen if (mail_namespace_find(namespaces, name) == NULL)
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen i_error("quota: Unknown namespace: %s", name);
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen }
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen}
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainenvoid quota_mail_namespaces_created(struct mail_namespace *namespaces)
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen{
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen struct quota *quota;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen struct quota_root *const *roots;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen unsigned int i, count;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen
a9a77f4632b1f00cc3c6664c91ef5f23746e099bTimo Sirainen if (QUOTA_USER_CONTEXT(namespaces->user) == NULL)
a9a77f4632b1f00cc3c6664c91ef5f23746e099bTimo Sirainen return;
a9a77f4632b1f00cc3c6664c91ef5f23746e099bTimo Sirainen
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen quota = quota_get_mail_user_quota(namespaces->user);
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen roots = array_get(&quota->roots, &count);
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen for (i = 0; i < count; i++)
eafecb7ad0fbbde0da8be7e8ae735459b7bb688bTimo Sirainen quota_root_set_namespace(roots[i], namespaces);
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen}