plugins/quota/quota-maildir.c

	quota-maildir.c revision cb07373a0afd4b0635e10a6757bd22e46dc9cf61
76b43e4417bab52e913da39b5f5bc2a130d3f149Timo Sirainen/* Copyright (c) 2006-2008 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
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 {
a27e065f1a1f91c7fbdf7c2ea1c387441af0cbb3Timo Sirainen    struct mail_storage *storage;
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 *
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainenmaildir_list_init(struct maildir_quota_root *root,
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainen          struct mail_storage *storage)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen    struct maildir_list_context *ctx;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen    ctx = i_new(struct maildir_list_context, 1);
a27e065f1a1f91c7fbdf7c2ea1c387441af0cbb3Timo Sirainen    ctx->storage = storage;
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainen    ctx->root = root;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen    ctx->path = str_new(default_pool, 512);
e392fcb39a06609af20a9e79017683f194de3ddeTimo Sirainen    ctx->iter = mailbox_list_iter_init(mail_storage_get_list(storage), "*",
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    bool is_file;
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
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen            path = mail_storage_get_mailbox_path(ctx->storage,
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen                                 ctx->info->name,
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen                                 &is_file);
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
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen    *mtime_r = st.st_size;
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,
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainen                struct mail_storage *storage, 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
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainen    ctx = maildir_list_init(root, storage);
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{
b1678954f83e1059b981e2def52a70054fa71399Timo Sirainen    struct quota_root *_root = &root->root;
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen    struct mail_storage *const *storages;
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen    unsigned int i, count;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen    struct dotlock *dotlock;
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen    const char *p, *dir;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen    string_t *str;
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen    mode_t mode;
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen    gid_t 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. */
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen    mode = 0600;
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen    gid = (gid_t)-1;
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen    storages = array_get(&root->root.quota->storages, &count);
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen    for (i = 0; i < count; i++) {
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen        if ((storages[i]->ns->flags & NAMESPACE_FLAG_INBOX) != 0) {
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen            mailbox_list_get_permissions(storages[i]->ns->list,
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen                             &mode, &gid);
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen            break;
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen        }
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen    }
fea7b8b3fc182e415b1875d79587c0aa1adb09d7Timo Sirainen
8887bf3757d51d73887dd20b1db3334d867d3817Timo Sirainen    dotlock_settings.use_excl_lock = getenv("DOTLOCK_USE_EXCL") != NULL;
8da095519878426b012058e6f331a669f327f47fTimo Sirainen    dotlock_settings.nfs_flush = getenv("MAIL_NFS_STORAGE") != NULL;
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);
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen        if (mkdir_parents(dir, 0700) < 0 && errno != EEXIST) {
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen            i_error("mkdir_parents(%s) failed: %m", dir);
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo Sirainen            return -1;
cb07373a0afd4b0635e10a6757bd22e46dc9cf61Timo 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);
b1678954f83e1059b981e2def52a70054fa71399Timo Sirainen    if (_root->bytes_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
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainenstatic int maildirsize_recalculate_storage(struct maildir_quota_root *root,
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen                       struct mail_storage *storage)
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
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainen    ctx = maildir_list_init(root, storage);
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{
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen    struct mail_storage *const *storages;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen    unsigned int i, count;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen    int ret = 0;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen    maildirsize_recalculate_init(root);
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen    /* count mails from all storages */
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen    storages = array_get(&root->root.quota->storages, &count);
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen    for (i = 0; i < count; i++) {
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen        if (maildirsize_recalculate_storage(root, storages[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++) {
2b682d8d3661800f16aceaa45fa4de9b6b140a59Timo Sirainen            ret = maildirs_check_have_changed(root, storages[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;
4c5272b168b4d71a95a08d48cc5b29cd7b27f193Timo Sirainen    while (size < sizeof(buf)-1 &&
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen           (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);
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen        }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen        size += ret;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen    }
4c5272b168b4d71a95a08d48cc5b29cd7b27f193Timo Sirainen    if (ret < 0 || size >= sizeof(buf)-1) {
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen        /* error / recalculation needed. */
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen        (void)close(root->fd);
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen        root->fd = -1;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen        return ret < 0 ? -1 : 0;
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen    }
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen    /* file is smaller than 5120 bytes, which means we can use it */
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
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 &&
0223007c8a72339fe40a732710203afb82f108a6Timo Sirainen        maildirsize_parse(root, root->fd, t_strsplit(buf, "\n")) > 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) {
7ca2a9f1cca63cbc2ebffc185c7e5a2b32bc2780Timo Sirainen            i_warning("quota maildir: No maildir storages, "
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) {
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen        if (root->root.set->default_rule.bytes_limit == 0 &&
d23c747de9d33966483fbdd41f08ad7766da7c5cTimo Sirainen            root->root.set->default_rule.count_limit == 0) {
2d71e0ea3006576961b47d91d564d31771676624Timo Sirainen            /* no quota */
2d71e0ea3006576961b47d91d564d31771676624Timo Sirainen            return 0;
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
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
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainenmaildir_quota_root_storage_added(struct quota_root *_root,
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen                 struct mail_storage *storage)
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
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen    control_dir = mail_storage_get_mailbox_control_dir(storage, "");
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
ccd44abfe14f51cc1f6d8c0ec1aa6dc31242e2d3Timo Sirainenmaildir_quota_storage_added(struct quota *quota, struct mail_storage *storage)
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen{
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen    struct quota_root **roots;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen    unsigned int i, count;
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen
ccd44abfe14f51cc1f6d8c0ec1aa6dc31242e2d3Timo Sirainen    if (strcmp(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++) {
3fe44a0df5a0bdd80c495f79cbf0e384441d6fccTimo Sirainen        if (roots[i]->backend.name == quota_backend_maildir.name)
ccd44abfe14f51cc1f6d8c0ec1aa6dc31242e2d3Timo Sirainen            maildir_quota_root_storage_added(roots[i], storage);
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,
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen        NULL,
cb951d3282610a9a0960230865bc5f3e3347b203Timo Sirainen        maildir_quota_deinit,
12dc81583d1958cb301a617e19fbd40e8d376397Timo Sirainen        maildir_quota_parse_rule,
3ccfcf0856958cb9208a9fc51c3bdf13c58ad52aTimo Sirainen        maildir_quota_storage_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};