quota-maildir.c revision 2615df45a8027948a474abe5e817b34b0499c171
45312f52ff3a3d4c137447be4c7556500c2f8bf2Timo Sirainen/* Copyright (c) 2006-2009 Dovecot authors, see the included COPYING file */
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen#include "lib.h"
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen#include "array.h"
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen#include "ioloop.h"
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen#include "nfs-workarounds.h"
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen#include "file-dotlock.h"
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen#include "mkdir-parents.h"
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen#include "read-full.h"
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen#include "write-full.h"
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen#include "str.h"
677e22747b82bf15b339e31d1d0106d62bf806daTimo Sirainen#include "maildir-storage.h"
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen#include "quota-private.h"
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen#include <stdio.h>
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen#include <stdlib.h>
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen#include <dirent.h>
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen#include <sys/stat.h>
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen#define MAILDIRSIZE_FILENAME "maildirsize"
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen#define MAILDIRSIZE_STALE_SECS (60*15)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainenstruct maildir_quota_root {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen struct quota_root root;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen struct mail_namespace *maildirsize_ns;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen const char *maildirsize_path;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen uint64_t total_bytes;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen uint64_t total_count;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen int fd;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen time_t recalc_last_stamp;
ed22384e18b005c6a518478c515e7bfdb9e110eeTimo Sirainen off_t last_size;
2d71e0ea3006576961b47d91d564d31771676624Timo Sirainen
c73bb576f140e1c1aae57fd607d02b418afe2c3fTimo Sirainen unsigned int limits_initialized:1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen};
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainenstruct maildir_list_context {
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen struct mailbox_list *list;
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainen struct maildir_quota_root *root;
a27e065f1a1f91c7fbdf7c2ea1c387441af0cbb3Timo Sirainen struct mailbox_list_iterate_context *iter;
61e84692827b6a64912343f515c984853021483aTimo Sirainen const struct mailbox_info *info;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen string_t *path;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen int state;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen};
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainenextern struct quota_backend quota_backend_maildir;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
ab0d9eecd85f74acae18fe88529302e0776cc500Timo Sirainenstatic struct dotlock_settings dotlock_settings = {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen MEMBER(temp_prefix) NULL,
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen MEMBER(lock_suffix) NULL,
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen MEMBER(timeout) 0,
8da095519878426b012058e6f331a669f327f47fTimo Sirainen MEMBER(stale_timeout) 30
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen};
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainenstatic int maildir_sum_dir(const char *dir, uint64_t *total_bytes,
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen uint64_t *total_count)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen DIR *dirp;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen struct dirent *dp;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen string_t *path;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen const char *p;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen size_t len;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen uoff_t num;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen int ret = 0;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen dirp = opendir(dir);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (dirp == NULL) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (errno == ENOENT || errno == ESTALE)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return 0;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen i_error("opendir(%s) failed: %m", dir);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return -1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen path = t_str_new(256);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen str_append(path, dir);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen str_append_c(path, '/');
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen len = str_len(path);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen while ((dp = readdir(dirp)) != NULL) {
b83f52b3ac3736d7f509db7ba80360d89f8a776dTimo Sirainen if (dp->d_name[0] == '.' &&
b83f52b3ac3736d7f509db7ba80360d89f8a776dTimo Sirainen (dp->d_name[1] == '\0' || dp->d_name[1] == '.'))
b83f52b3ac3736d7f509db7ba80360d89f8a776dTimo Sirainen continue;
b83f52b3ac3736d7f509db7ba80360d89f8a776dTimo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen p = strstr(dp->d_name, ",S=");
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen num = (uoff_t)-1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (p != NULL) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen /* ,S=nnnn[:,] */
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen p += 3;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen for (num = 0; *p >= '0' && *p <= '9'; p++)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen num = num * 10 + (*p - '0');
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (*p != ':' && *p != '\0' && *p != ',') {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen /* not in expected format, fallback to stat() */
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen num = (uoff_t)-1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen } else {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen *total_bytes += num;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen *total_count += 1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (num == (uoff_t)-1) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen struct stat st;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen str_truncate(path, len);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen str_append(path, dp->d_name);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (stat(str_c(path), &st) == 0) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen *total_bytes += st.st_size;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen *total_count += 1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen } else if (errno != ENOENT && errno != ESTALE) {
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen i_error("stat(%s) failed: %m", str_c(path));
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen ret = -1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (closedir(dirp) < 0) {
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen i_error("closedir(%s) failed: %m", dir);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return -1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return ret;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainenstatic struct maildir_list_context *
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainenmaildir_list_init(struct maildir_quota_root *root, struct mailbox_list *list)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen struct maildir_list_context *ctx;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen ctx = i_new(struct maildir_list_context, 1);
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainen ctx->root = root;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen ctx->path = str_new(default_pool, 512);
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen ctx->list = list;
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen ctx->iter = mailbox_list_iter_init(list, "*",
18ffea71d9beeec3cc1d400f751926ee72807f62Timo Sirainen MAILBOX_LIST_ITER_RETURN_NO_FLAGS);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return ctx;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainenstatic const char *
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainenmaildir_list_next(struct maildir_list_context *ctx, time_t *mtime_r)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainen struct quota_rule *rule;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen struct stat st;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen for (;;) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (ctx->state == 0) {
a27e065f1a1f91c7fbdf7c2ea1c387441af0cbb3Timo Sirainen ctx->info = mailbox_list_iter_next(ctx->iter);
a27e065f1a1f91c7fbdf7c2ea1c387441af0cbb3Timo Sirainen if (ctx->info == NULL)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return NULL;
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainen
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen rule = quota_root_rule_find(ctx->root->root.set,
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainen ctx->info->name);
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainen if (rule != NULL && rule->ignore) {
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainen /* mailbox not included in quota */
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainen continue;
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
19e8adccba16ff419f5675b1575358c2956dce83Timo Sirainen T_BEGIN {
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen const char *path;
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen path = mailbox_list_get_path(ctx->list, ctx->info->name,
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen MAILBOX_LIST_PATH_TYPE_MAILBOX);
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen str_truncate(ctx->path, 0);
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen str_append(ctx->path, path);
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen str_append(ctx->path, ctx->state == 0 ?
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen "/new" : "/cur");
19e8adccba16ff419f5675b1575358c2956dce83Timo Sirainen } T_END;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (++ctx->state == 2)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen ctx->state = 0;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (stat(str_c(ctx->path), &st) == 0)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen break;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen /* ignore if the directory got lost, stale or if it was
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen actually a file and not a directory */
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (errno != ENOENT && errno != ESTALE && errno != ENOTDIR) {
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen i_error("stat(%s) failed: %m", str_c(ctx->path));
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen ctx->state = 0;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
f037abbef2cc57979356127e3b2bb456351460f7Timo Sirainen *mtime_r = st.st_mtime;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return str_c(ctx->path);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainenstatic int maildir_list_deinit(struct maildir_list_context *ctx)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
a27e065f1a1f91c7fbdf7c2ea1c387441af0cbb3Timo Sirainen int ret = mailbox_list_iter_deinit(&ctx->iter);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen str_free(&ctx->path);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen i_free(ctx);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return ret;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainenstatic int
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainenmaildirs_check_have_changed(struct maildir_quota_root *root,
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen struct mail_namespace *ns, time_t latest_mtime)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen struct maildir_list_context *ctx;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen time_t mtime;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen int ret = 0;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen ctx = maildir_list_init(root, ns->list);
ccb77e2f63626ec46e5745ef4f38baa8e8e504fcTimo Sirainen while (maildir_list_next(ctx, &mtime) != NULL) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (mtime > latest_mtime) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen ret = 1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen break;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (maildir_list_deinit(ctx) < 0)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return -1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return ret;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainenstatic int maildirsize_write(struct maildir_quota_root *root, const char *path)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen const struct mail_storage_settings *set =
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen root->maildirsize_ns->mail_set;
b1678954f83e1059b981e2def52a70054fa71399Timo Sirainen struct quota_root *_root = &root->root;
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen struct mail_namespace *const *namespaces;
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen unsigned int i, count;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen struct dotlock *dotlock;
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen const char *p, *dir;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen string_t *str;
263457e5d159bd461d34ff4796ade7ba434f02b7Timo Sirainen mode_t mode, dir_mode;
263457e5d159bd461d34ff4796ade7ba434f02b7Timo Sirainen gid_t gid, dir_gid;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen int fd;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
b83f52b3ac3736d7f509db7ba80360d89f8a776dTimo Sirainen i_assert(root->fd == -1);
b83f52b3ac3736d7f509db7ba80360d89f8a776dTimo Sirainen
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen /* figure out what permissions we should use for maildirsize.
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen use the inbox namespace's permissions if possible. */
263457e5d159bd461d34ff4796ade7ba434f02b7Timo Sirainen mode = 0600; dir_mode = 0700;
263457e5d159bd461d34ff4796ade7ba434f02b7Timo Sirainen gid = dir_gid = (gid_t)-1;
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen namespaces = array_get(&root->root.quota->namespaces, &count);
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen i_assert(count > 0);
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen for (i = 0; i < count; i++) {
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen if ((namespaces[i]->flags & NAMESPACE_FLAG_INBOX) != 0) {
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen mailbox_list_get_permissions(namespaces[i]->list,
b82d6d7f02734007c129fa25bc876049c8d9bddeTimo Sirainen NULL, &mode, &gid);
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen mailbox_list_get_dir_permissions(namespaces[i]->list,
b82d6d7f02734007c129fa25bc876049c8d9bddeTimo Sirainen NULL,
263457e5d159bd461d34ff4796ade7ba434f02b7Timo Sirainen &dir_mode, &dir_gid);
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen break;
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen }
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen }
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen dotlock_settings.use_excl_lock = set->dotlock_use_excl;
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen dotlock_settings.nfs_flush = set->mail_nfs_storage;
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen fd = file_dotlock_open_mode(&dotlock_settings, path,
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen DOTLOCK_CREATE_FLAG_NONBLOCK,
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen mode, (uid_t)-1, gid, &dotlock);
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen if (fd == -1 && errno == ENOENT) {
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen /* the control directory doesn't exist yet? create it */
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen p = strrchr(path, '/');
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen dir = t_strdup_until(path, p);
263457e5d159bd461d34ff4796ade7ba434f02b7Timo Sirainen if (mkdir_parents(dir, dir_mode) < 0 && errno != EEXIST) {
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen i_error("mkdir_parents(%s) failed: %m", dir);
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen return -1;
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen }
263457e5d159bd461d34ff4796ade7ba434f02b7Timo Sirainen if (dir_gid != (gid_t)-1) {
263457e5d159bd461d34ff4796ade7ba434f02b7Timo Sirainen if (chown(dir, (uid_t)-1, dir_gid) < 0) {
263457e5d159bd461d34ff4796ade7ba434f02b7Timo Sirainen i_error("chown(%s,-1,%ld) failed: %m",
263457e5d159bd461d34ff4796ade7ba434f02b7Timo Sirainen dir, (long)dir_gid);
263457e5d159bd461d34ff4796ade7ba434f02b7Timo Sirainen }
263457e5d159bd461d34ff4796ade7ba434f02b7Timo Sirainen }
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen fd = file_dotlock_open_mode(&dotlock_settings, path,
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen DOTLOCK_CREATE_FLAG_NONBLOCK,
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen mode, (uid_t)-1, gid, &dotlock);
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (fd == -1) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (errno == EAGAIN) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen /* someone's just in the middle of updating it */
25b0229f40dc59d0b3f31e01b7afc57304bead5cTimo Sirainen return 1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen i_error("file_dotlock_open(%s) failed: %m", path);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return -1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen str = t_str_new(128);
b76bc28de1c65b7ed6b19c7873b294d3742f5d57Timo Sirainen /* if we have no limits, write 0S instead of an empty line */
b76bc28de1c65b7ed6b19c7873b294d3742f5d57Timo Sirainen if (_root->bytes_limit != 0 || _root->count_limit == 0) {
2d71e0ea3006576961b47d91d564d31771676624Timo Sirainen str_printfa(str, "%lluS",
b1678954f83e1059b981e2def52a70054fa71399Timo Sirainen (unsigned long long)_root->bytes_limit);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
b1678954f83e1059b981e2def52a70054fa71399Timo Sirainen if (_root->count_limit != 0) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (str_len(str) > 0)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen str_append_c(str, ',');
2d71e0ea3006576961b47d91d564d31771676624Timo Sirainen str_printfa(str, "%lluC",
b1678954f83e1059b981e2def52a70054fa71399Timo Sirainen (unsigned long long)_root->count_limit);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen str_printfa(str, "\n%llu %llu\n",
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen (unsigned long long)root->total_bytes,
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen (unsigned long long)root->total_count);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (write_full(fd, str_data(str), str_len(str)) < 0) {
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen i_error("write_full(%s) failed: %m", path);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen file_dotlock_delete(&dotlock);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return -1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
b83f52b3ac3736d7f509db7ba80360d89f8a776dTimo Sirainen /* keep the fd open since we might want to update it later */
b83f52b3ac3736d7f509db7ba80360d89f8a776dTimo Sirainen if (file_dotlock_replace(&dotlock,
b83f52b3ac3736d7f509db7ba80360d89f8a776dTimo Sirainen DOTLOCK_REPLACE_FLAG_DONT_CLOSE_FD) < 0) {
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen i_error("file_dotlock_replace(%s) failed: %m", path);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return -1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
b83f52b3ac3736d7f509db7ba80360d89f8a776dTimo Sirainen root->fd = fd;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return 0;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainenstatic void maildirsize_recalculate_init(struct maildir_quota_root *root)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen root->total_bytes = root->total_count = 0;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen root->recalc_last_stamp = 0;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainenstatic int maildirsize_recalculate_namespace(struct maildir_quota_root *root,
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen struct mail_namespace *ns)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen struct maildir_list_context *ctx;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen const char *dir;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen time_t mtime;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen int ret = 0;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen ctx = maildir_list_init(root, ns->list);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen while ((dir = maildir_list_next(ctx, &mtime)) != NULL) {
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen if (mtime > root->recalc_last_stamp)
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen root->recalc_last_stamp = mtime;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
19e8adccba16ff419f5675b1575358c2956dce83Timo Sirainen T_BEGIN {
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen if (maildir_sum_dir(dir, &root->total_bytes,
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen &root->total_count) < 0)
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen ret = -1;
19e8adccba16ff419f5675b1575358c2956dce83Timo Sirainen } T_END;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (maildir_list_deinit(ctx) < 0)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen ret = -1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen return ret;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainenstatic void maildirsize_rebuild_later(struct maildir_quota_root *root)
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen{
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen if (!root->root.set->force_default_rule) {
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen /* FIXME: can't unlink(), because the limits would be lost. */
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen return;
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen }
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen if (unlink(root->maildirsize_path) < 0 &&
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen errno != ENOENT && errno != ESTALE)
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen i_error("unlink(%s) failed: %m", root->maildirsize_path);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen}
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainenstatic int maildirsize_recalculate_finish(struct maildir_quota_root *root,
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen int ret)
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen{
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (ret == 0) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen /* maildir didn't change, we can write the maildirsize file */
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen ret = maildirsize_write(root, root->maildirsize_path);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen if (ret != 0)
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen maildirsize_rebuild_later(root);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return ret;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainenstatic int maildirsize_recalculate(struct maildir_quota_root *root)
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen{
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen struct mail_namespace *const *namespaces;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen unsigned int i, count;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen int ret = 0;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen maildirsize_recalculate_init(root);
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen /* count mails from all namespaces */
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen namespaces = array_get(&root->root.quota->namespaces, &count);
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen for (i = 0; i < count; i++) {
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen if (!quota_root_is_namespace_visible(&root->root, namespaces[i]))
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen continue;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen if (maildirsize_recalculate_namespace(root, namespaces[i]) < 0) {
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen ret = -1;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen break;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen }
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen }
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen if (ret == 0) {
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen /* check if any of the directories have changed */
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen for (i = 0; i < count; i++) {
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen if (!quota_root_is_namespace_visible(&root->root,
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen namespaces[i]))
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen continue;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen ret = maildirs_check_have_changed(root, namespaces[i],
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen root->recalc_last_stamp);
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen if (ret != 0)
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen break;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen }
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen }
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen return maildirsize_recalculate_finish(root, ret);
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen}
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainenstatic bool
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainenmaildir_parse_limit(const char *str, uint64_t *bytes_r, uint64_t *count_r)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen const char *const *limit;
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen unsigned long long value;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen char *pos;
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen bool ret = TRUE;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
3851ad9fcb25635f02b46d44586742ef1081876bTimo Sirainen *bytes_r = 0;
3851ad9fcb25635f02b46d44586742ef1081876bTimo Sirainen *count_r = 0;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen /* 0 values mean unlimited */
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen for (limit = t_strsplit(str, ","); *limit != NULL; limit++) {
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen value = strtoull(*limit, &pos, 10);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (pos[0] != '\0' && pos[1] == '\0') {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen switch (pos[0]) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen case 'C':
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen if (value != 0)
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen *count_r = value;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen break;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen case 'S':
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen if (value != 0)
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen *bytes_r = value;
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen break;
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen default:
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen ret = FALSE;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen break;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen } else {
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen ret = FALSE;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen return ret;
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen}
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainenstatic int maildirsize_parse(struct maildir_quota_root *root,
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen int fd, const char *const *lines)
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen{
b1678954f83e1059b981e2def52a70054fa71399Timo Sirainen struct quota_root *_root = &root->root;
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen uint64_t message_bytes_limit, message_count_limit;
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen long long bytes_diff, total_bytes;
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen int count_diff, total_count;
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen unsigned int line_count = 0;
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen if (*lines == NULL)
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen return -1;
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen /* first line contains the limits */
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen (void)maildir_parse_limit(lines[0], &message_bytes_limit,
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen &message_count_limit);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
3851ad9fcb25635f02b46d44586742ef1081876bTimo Sirainen /* truncate too high limits to signed 64bit int range */
3851ad9fcb25635f02b46d44586742ef1081876bTimo Sirainen if (message_bytes_limit >= (1ULL << 63))
3851ad9fcb25635f02b46d44586742ef1081876bTimo Sirainen message_bytes_limit = (1ULL << 63) - 1;
3851ad9fcb25635f02b46d44586742ef1081876bTimo Sirainen if (message_count_limit >= (1ULL << 63))
3851ad9fcb25635f02b46d44586742ef1081876bTimo Sirainen message_count_limit = (1ULL << 63) - 1;
3851ad9fcb25635f02b46d44586742ef1081876bTimo Sirainen
b1678954f83e1059b981e2def52a70054fa71399Timo Sirainen if (root->root.bytes_limit == (int64_t)message_bytes_limit &&
b1678954f83e1059b981e2def52a70054fa71399Timo Sirainen root->root.count_limit == (int64_t)message_count_limit) {
3851ad9fcb25635f02b46d44586742ef1081876bTimo Sirainen /* limits haven't changed */
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen } else if (root->root.set->force_default_rule) {
2d71e0ea3006576961b47d91d564d31771676624Timo Sirainen /* we know the limits and they've changed.
2d71e0ea3006576961b47d91d564d31771676624Timo Sirainen the file must be rewritten. */
2d71e0ea3006576961b47d91d564d31771676624Timo Sirainen return 0;
3851ad9fcb25635f02b46d44586742ef1081876bTimo Sirainen } else {
3851ad9fcb25635f02b46d44586742ef1081876bTimo Sirainen /* we're using limits from the file. */
b1678954f83e1059b981e2def52a70054fa71399Timo Sirainen root->root.bytes_limit = message_bytes_limit;
b1678954f83e1059b981e2def52a70054fa71399Timo Sirainen root->root.count_limit = message_count_limit;
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen quota_root_recalculate_relative_rules(root->root.set);
2d71e0ea3006576961b47d91d564d31771676624Timo Sirainen }
2d71e0ea3006576961b47d91d564d31771676624Timo Sirainen
ba3f68dfb8475299f43125c3e86985a713013c5dTimo Sirainen if (*lines == NULL) {
ba3f68dfb8475299f43125c3e86985a713013c5dTimo Sirainen /* no quota lines. rebuild it. */
ba3f68dfb8475299f43125c3e86985a713013c5dTimo Sirainen return 0;
ba3f68dfb8475299f43125c3e86985a713013c5dTimo Sirainen }
ba3f68dfb8475299f43125c3e86985a713013c5dTimo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen /* rest of the lines contains <bytes> <count> diffs */
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen total_bytes = 0; total_count = 0;
0563ad5c7f179554623682f6fd7b98596901b49fTimo Sirainen for (lines++; *lines != NULL; lines++, line_count++) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (sscanf(*lines, "%lld %d", &bytes_diff, &count_diff) != 2)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return -1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen total_bytes += bytes_diff;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen total_count += count_diff;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (total_bytes < 0 || total_count < 0) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen /* corrupted */
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return -1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
b1678954f83e1059b981e2def52a70054fa71399Timo Sirainen if ((total_bytes > _root->bytes_limit && _root->bytes_limit != 0) ||
b1678954f83e1059b981e2def52a70054fa71399Timo Sirainen (total_count > _root->count_limit && _root->count_limit != 0)) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen /* we're over quota. don't trust these values if the file
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen contains more than the initial summary line, or if the file
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen is older than 15 minutes. */
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen struct stat st;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (line_count > 1)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return 0;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (fstat(fd, &st) < 0 ||
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen st.st_mtime < ioloop_time - MAILDIRSIZE_STALE_SECS)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return 0;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen root->total_bytes = (uint64_t)total_bytes;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen root->total_count = (uint64_t)total_count;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return 1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainenstatic int maildirsize_open(struct maildir_quota_root *root)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (root->fd != -1) {
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen if (close(root->fd) < 0)
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen i_error("close(%s) failed: %m", root->maildirsize_path);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen root->fd = nfs_safe_open(root->maildirsize_path, O_RDWR | O_APPEND);
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen if (root->fd == -1) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (errno == ENOENT)
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen return 0;
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen i_error("open(%s) failed: %m", root->maildirsize_path);
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen return -1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen return 1;
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen}
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainenstatic bool maildirsize_has_changed(struct maildir_quota_root *root)
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen{
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen struct stat st1, st2;
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen if (dotlock_settings.nfs_flush) {
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen nfs_flush_file_handle_cache(root->maildirsize_path);
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen nfs_flush_attr_cache_unlocked(root->maildirsize_path);
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen }
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen if (root->fd == -1)
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen return TRUE;
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen if (stat(root->maildirsize_path, &st1) < 0)
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen return TRUE;
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen if (fstat(root->fd, &st2) < 0)
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen return TRUE;
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen
ed22384e18b005c6a518478c515e7bfdb9e110eeTimo Sirainen return root->last_size != st2.st_size || st1.st_ino != st2.st_ino ||
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen !CMP_DEV_T(st1.st_dev, st2.st_dev);
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen}
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainenstatic int maildirsize_read(struct maildir_quota_root *root)
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen{
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen char buf[5120+1];
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen unsigned int i, size;
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen int ret;
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen if (!maildirsize_has_changed(root))
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen return 1;
7be291ff7fcb7d10a5f96db00282a40287b471c5Timo Sirainen
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen if ((ret = maildirsize_open(root)) <= 0)
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen return ret;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
4c5272b168b4d71a95a08d48cc5b29cd7b27f193Timo Sirainen /* @UNSAFE */
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen size = 0;
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen while ((ret = read(root->fd, buf + size, sizeof(buf)-1 - size)) != 0) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (ret < 0) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (errno == ESTALE)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen break;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen i_error("read(%s) failed: %m", root->maildirsize_path);
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen break;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen size += ret;
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen if (size >= sizeof(buf)-1) {
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen /* we'll need to recalculate the quota */
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen break;
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen /* try to use the file even if we ran into some error. if we don't have
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen forced limits, we'll need to read the header to get them */
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen root->total_bytes = root->total_count = 0;
ed22384e18b005c6a518478c515e7bfdb9e110eeTimo Sirainen root->last_size = size;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
0563ad5c7f179554623682f6fd7b98596901b49fTimo Sirainen /* skip the last line if there's no LF at the end. Remove the last LF
0563ad5c7f179554623682f6fd7b98596901b49fTimo Sirainen so we don't get one empty line in the strsplit. */
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen while (size > 0 && buf[size-1] != '\n') size--;
0563ad5c7f179554623682f6fd7b98596901b49fTimo Sirainen if (size > 0) size--;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen buf[size] = '\0';
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen if (ret < 0 && size == 0) {
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen /* the read failed and there's no usable header, fail. */
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen (void)close(root->fd);
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen root->fd = -1;
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen return -1;
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen }
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen
0563ad5c7f179554623682f6fd7b98596901b49fTimo Sirainen /* If there are any NUL bytes, the file is broken. */
0563ad5c7f179554623682f6fd7b98596901b49fTimo Sirainen for (i = 0; i < size; i++) {
0563ad5c7f179554623682f6fd7b98596901b49fTimo Sirainen if (buf[i] == '\0')
0563ad5c7f179554623682f6fd7b98596901b49fTimo Sirainen break;
0563ad5c7f179554623682f6fd7b98596901b49fTimo Sirainen }
0563ad5c7f179554623682f6fd7b98596901b49fTimo Sirainen
0563ad5c7f179554623682f6fd7b98596901b49fTimo Sirainen if (i == size &&
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen maildirsize_parse(root, root->fd, t_strsplit(buf, "\n")) > 0 &&
37f5e4f22d8f0d6aeca6536fba84b396c2196364Timo Sirainen ret == 0)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen ret = 1;
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen else {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen /* broken file / need recalculation */
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen (void)close(root->fd);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen root->fd = -1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen ret = 0;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return ret;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
3851ad9fcb25635f02b46d44586742ef1081876bTimo Sirainenstatic bool maildirquota_limits_init(struct maildir_quota_root *root)
c73bb576f140e1c1aae57fd607d02b418afe2c3fTimo Sirainen{
7ca2a9f1cca63cbc2ebffc185c7e5a2b32bc2780Timo Sirainen if (!root->limits_initialized) {
7ca2a9f1cca63cbc2ebffc185c7e5a2b32bc2780Timo Sirainen root->limits_initialized = TRUE;
7ca2a9f1cca63cbc2ebffc185c7e5a2b32bc2780Timo Sirainen
7ca2a9f1cca63cbc2ebffc185c7e5a2b32bc2780Timo Sirainen if (root->maildirsize_path == NULL) {
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen i_warning("quota maildir: No maildir namespaces, "
7ca2a9f1cca63cbc2ebffc185c7e5a2b32bc2780Timo Sirainen "ignoring quota.");
7ca2a9f1cca63cbc2ebffc185c7e5a2b32bc2780Timo Sirainen }
9d254c7c7b715d7fe1206e9b228e9c5b23b99f28Timo Sirainen }
7ca2a9f1cca63cbc2ebffc185c7e5a2b32bc2780Timo Sirainen
e394583a7c1121599799a35f205bf237f6a1ee04Timo Sirainen return root->maildirsize_path != NULL;
e394583a7c1121599799a35f205bf237f6a1ee04Timo Sirainen}
e394583a7c1121599799a35f205bf237f6a1ee04Timo Sirainen
e394583a7c1121599799a35f205bf237f6a1ee04Timo Sirainenstatic int maildirquota_refresh(struct maildir_quota_root *root)
e394583a7c1121599799a35f205bf237f6a1ee04Timo Sirainen{
e394583a7c1121599799a35f205bf237f6a1ee04Timo Sirainen int ret;
e394583a7c1121599799a35f205bf237f6a1ee04Timo Sirainen
e394583a7c1121599799a35f205bf237f6a1ee04Timo Sirainen if (!maildirquota_limits_init(root))
e394583a7c1121599799a35f205bf237f6a1ee04Timo Sirainen return 0;
c73bb576f140e1c1aae57fd607d02b418afe2c3fTimo Sirainen
19e8adccba16ff419f5675b1575358c2956dce83Timo Sirainen T_BEGIN {
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen ret = maildirsize_read(root);
19e8adccba16ff419f5675b1575358c2956dce83Timo Sirainen } T_END;
2d71e0ea3006576961b47d91d564d31771676624Timo Sirainen if (ret == 0) {
9f018a85c93145e789ea5365718cfc6dd954a913Timo Sirainen if (root->root.bytes_limit == 0 &&
9f018a85c93145e789ea5365718cfc6dd954a913Timo Sirainen root->root.count_limit == 0 &&
9f018a85c93145e789ea5365718cfc6dd954a913Timo Sirainen root->root.set->default_rule.bytes_limit == 0 &&
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen root->root.set->default_rule.count_limit == 0) {
2d71e0ea3006576961b47d91d564d31771676624Timo Sirainen /* no quota */
53101f0e1e998d9bf718689fdb1363c5f0a07b0fTimo Sirainen if (!root->root.set->force_default_rule)
53101f0e1e998d9bf718689fdb1363c5f0a07b0fTimo Sirainen return 0;
53101f0e1e998d9bf718689fdb1363c5f0a07b0fTimo Sirainen /* explicitly specified 0 as quota. keep the quota
53101f0e1e998d9bf718689fdb1363c5f0a07b0fTimo Sirainen updated even if it's not enforced. */
2d71e0ea3006576961b47d91d564d31771676624Timo Sirainen }
2d71e0ea3006576961b47d91d564d31771676624Timo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen ret = maildirsize_recalculate(root);
2d71e0ea3006576961b47d91d564d31771676624Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return ret < 0 ? -1 : 0;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainenstatic int maildirsize_update(struct maildir_quota_root *root,
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen int count_diff, int64_t bytes_diff)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen char str[MAX_INT_STRLEN*2 + 2];
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen int ret = 0;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
0461b5dfd1ac6b5ae49a2f1ab6ffdc2e5e2ee7c2Timo Sirainen if (count_diff == 0 && bytes_diff == 0)
0461b5dfd1ac6b5ae49a2f1ab6ffdc2e5e2ee7c2Timo Sirainen return 0;
0461b5dfd1ac6b5ae49a2f1ab6ffdc2e5e2ee7c2Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen /* We rely on O_APPEND working in here. That isn't NFS-safe, but it
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen isn't necessarily that bad because the file is recreated once in
a8fcd55e88550ebb905249825bdb1eec7b9667ffTimo Sirainen a while, and sooner if corruption causes calculations to go
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen over quota. This is also how Maildir++ spec specifies it should be
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen done.. */
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen i_snprintf(str, sizeof(str), "%lld %d\n",
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen (long long)bytes_diff, count_diff);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (write_full(root->fd, str, strlen(str)) < 0) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen ret = -1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen if (errno == ESTALE) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen /* deleted/replaced already, ignore */
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen } else {
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen i_error("write_full(%s) failed: %m",
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen root->maildirsize_path);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return ret;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainenstatic struct quota_root *maildir_quota_alloc(void)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen struct maildir_quota_root *root;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen root = i_new(struct maildir_quota_root, 1);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen root->fd = -1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return &root->root;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainenstatic int maildir_quota_init(struct quota_root *_root, const char *args)
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen{
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen const char *const *tmp;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen if (args == NULL)
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen return 0;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen for (tmp = t_strsplit(args, ":"); *tmp != NULL; tmp++) {
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen if (strcmp(*tmp, "noenforcing") == 0)
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen _root->no_enforcing = TRUE;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen else if (strncmp(*tmp, "ns=", 3) == 0)
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen _root->ns_prefix = p_strdup(_root->pool, *tmp + 3);
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen else {
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen i_error("maildir quota: Invalid parameter: %s", *tmp);
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen return -1;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen }
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen }
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen return 0;
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen}
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainenstatic void maildir_quota_deinit(struct quota_root *_root)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen struct maildir_quota_root *root = (struct maildir_quota_root *)_root;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
6ce34c7a2462412af024a0320911a3fe1c04b264Timo Sirainen if (root->fd != -1)
6ce34c7a2462412af024a0320911a3fe1c04b264Timo Sirainen (void)close(root->fd);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen i_free(root);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainenstatic bool
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainenmaildir_quota_parse_rule(struct quota_root_settings *root_set ATTR_UNUSED,
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen struct quota_rule *rule,
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen const char *str, const char **error_r)
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen{
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen uint64_t bytes, count;
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen if (!maildir_parse_limit(str, &bytes, &count)) {
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen *error_r = "Invalid Maildir++ quota rule";
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen return FALSE;
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen }
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen rule->bytes_limit = bytes;
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen rule->count_limit = count;
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen return TRUE;
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen}
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainenstatic void
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainenmaildir_quota_root_namespace_added(struct quota_root *_root,
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen struct mail_namespace *ns)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen struct maildir_quota_root *root = (struct maildir_quota_root *)_root;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen const char *control_dir;
677e22747b82bf15b339e31d1d0106d62bf806daTimo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen if (root->maildirsize_path != NULL)
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen return;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen control_dir = mailbox_list_get_path(ns->list, NULL,
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen MAILBOX_LIST_PATH_TYPE_CONTROL);
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen root->maildirsize_ns = ns;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen root->maildirsize_path =
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen p_strconcat(_root->pool, control_dir,
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen "/"MAILDIRSIZE_FILENAME, NULL);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainenstatic void
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainenmaildir_quota_namespace_added(struct quota *quota, struct mail_namespace *ns)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen struct quota_root **roots;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen unsigned int i, count;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen if (strcmp(ns->storage->name, "maildir") != 0)
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen return;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen roots = array_get_modifiable(&quota->roots, &count);
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen for (i = 0; i < count; i++) {
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen if (roots[i]->backend.name == quota_backend_maildir.name &&
e6dfe937343e7d5e2a9f5a188a44d922460fbbefTimo Sirainen ((roots[i]->ns_prefix == NULL &&
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen ns->type == NAMESPACE_PRIVATE) || roots[i]->ns == ns))
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen maildir_quota_root_namespace_added(roots[i], ns);
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainenstatic const char *const *
43d32cbe60fdaef2699d99f1ca259053e9350411Timo Sirainenmaildir_quota_root_get_resources(struct quota_root *root ATTR_UNUSED)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen static const char *resources_both[] = {
e392fcb39a06609af20a9e79017683f194de3ddeTimo Sirainen QUOTA_NAME_STORAGE_KILOBYTES,
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen QUOTA_NAME_MESSAGES,
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen NULL
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen };
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return resources_both;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainenstatic int
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainenmaildir_quota_get_resource(struct quota_root *_root, const char *name,
3851ad9fcb25635f02b46d44586742ef1081876bTimo Sirainen uint64_t *value_r)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen struct maildir_quota_root *root = (struct maildir_quota_root *)_root;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen if (maildirquota_refresh(root) < 0)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen return -1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
5a001dd0848310e9ed6622442329ff346b189257Timo Sirainen if (strcmp(name, QUOTA_NAME_STORAGE_BYTES) == 0) {
e392fcb39a06609af20a9e79017683f194de3ddeTimo Sirainen *value_r = root->total_bytes;
5a001dd0848310e9ed6622442329ff346b189257Timo Sirainen } else if (strcmp(name, QUOTA_NAME_MESSAGES) == 0) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen *value_r = root->total_count;
5a001dd0848310e9ed6622442329ff346b189257Timo Sirainen } else
2d71e0ea3006576961b47d91d564d31771676624Timo Sirainen return 0;
8b10d65deb01f351172408ab9540413d60d5fe3bTimo Sirainen return 1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainenstatic int
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainenmaildir_quota_update(struct quota_root *_root,
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen struct quota_transaction_context *ctx)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen struct maildir_quota_root *root =
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen (struct maildir_quota_root *) _root;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
e394583a7c1121599799a35f205bf237f6a1ee04Timo Sirainen if (!maildirquota_limits_init(root)) {
e394583a7c1121599799a35f205bf237f6a1ee04Timo Sirainen /* no limits */
e394583a7c1121599799a35f205bf237f6a1ee04Timo Sirainen return 0;
e394583a7c1121599799a35f205bf237f6a1ee04Timo Sirainen }
63af428111bdf1975b06da83f98d072210514a36Timo Sirainen
63af428111bdf1975b06da83f98d072210514a36Timo Sirainen /* even though we don't really care about the limits in here ourself,
63af428111bdf1975b06da83f98d072210514a36Timo Sirainen we do want to make sure the header gets updated if the limits have
63af428111bdf1975b06da83f98d072210514a36Timo Sirainen changed. also this makes sure the maildirsize file is created if
63af428111bdf1975b06da83f98d072210514a36Timo Sirainen it doesn't exist. */
63af428111bdf1975b06da83f98d072210514a36Timo Sirainen if (maildirquota_refresh(root) < 0)
63af428111bdf1975b06da83f98d072210514a36Timo Sirainen return -1;
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen if (root->fd == -1 || ctx->recalculate ||
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen maildirsize_update(root, ctx->count_used, ctx->bytes_used) < 0)
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen maildirsize_rebuild_later(root);
bd7b1a9000b12349e2a99bb43b3ce8b96a18e92bTimo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen return 0;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen}
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainenstruct quota_backend quota_backend_maildir = {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen "maildir",
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen {
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen maildir_quota_alloc,
2d8b23805db6f06b8b38174fb6e135386694f429Timo Sirainen maildir_quota_init,
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen maildir_quota_deinit,
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen maildir_quota_parse_rule,
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen maildir_quota_namespace_added,
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen maildir_quota_root_get_resources,
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen maildir_quota_get_resource,
6d6bbe8787354bbb69d0c03187adfe0f497d70b8Timo Sirainen maildir_quota_update,
6d6bbe8787354bbb69d0c03187adfe0f497d70b8Timo Sirainen NULL
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen};