plugins/imap-quota/imap-quota-plugin.c

	imap-quota-plugin.c revision 378d924da5853145a6df9a299074f04be69986c7
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen/* Copyright (C) 2005 Timo Sirainen */
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include "common.h"
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include "str.h"
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include "imap-quote.h"
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include "commands.h"
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include "quota.h"
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include "quota-plugin.h"
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include "imap-quota-plugin.h"
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen#include <stdlib.h>
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainenstatic void
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainenquota_send(struct client_command_context *cmd, struct quota_root *root)
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen{
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen        const char *const *list;
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    string_t *str;
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    unsigned int i;
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    uint64_t value, limit;
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    int ret;
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    t_push();
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    str = t_str_new(128);
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    str_append(str, "* QUOTA ");
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    imap_quote_append_string(str, quota_root_get_name(root), FALSE);
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    str_append(str, " (");
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    list = quota_root_get_resources(root);
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen    for (i = 0; *list != NULL; list++, i++) {
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen        ret = quota_get_resource(root, *list, &value, &limit);
e5fb952c6d49d3b6bff1746551566202e92947daTimo Sirainen        if (ret > 0) {
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen            if (i > 0)
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen                str_append_c(str, ' ');
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen            str_printfa(str, "%s %llu %llu", *list,
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen                    (unsigned long long)value,
1ac7c8e9040e0d0b7e9f849e45b94bfe919595a9Timo Sirainen                    (unsigned long long)limit);
        } else if (ret < 0) {
            client_send_line(cmd->client, t_strconcat(
                "* BAD ", quota_last_error(quota), NULL));
        }
    }
    str_append_c(str, ')');
    client_send_line(cmd->client, str_c(str));

    t_pop();
}

static bool cmd_getquotaroot(struct client_command_context *cmd)
{
    struct mail_storage *storage;
    struct mailbox *box;
    struct quota_root_iter *iter;
        struct quota_root *root;
    const char *mailbox;
    string_t *str;

    /* <mailbox> */
    if (!client_read_string_args(cmd, 1, &mailbox))
        return FALSE;

    storage = client_find_storage(cmd, &mailbox);
    if (storage == NULL)
        return TRUE;

    box = mailbox_open(storage, mailbox, NULL, (MAILBOX_OPEN_READONLY |
                            MAILBOX_OPEN_FAST |
                            MAILBOX_OPEN_KEEP_RECENT));
    if (box == NULL) {
        client_send_storage_error(cmd, storage);
        return TRUE;
    }

    if (quota == NULL) {
        client_send_tagline(cmd, "OK No quota.");
        return TRUE;
    }

    /* send QUOTAROOT reply */
    str = t_str_new(128);
    str_append(str, "* QUOTAROOT ");
    imap_quote_append_string(str, mailbox, FALSE);

    iter = quota_root_iter_init(quota, box);
    while ((root = quota_root_iter_next(iter)) != NULL) {
        str_append_c(str, ' ');
        imap_quote_append_string(str, quota_root_get_name(root), FALSE);
    }
    if (quota_root_iter_deinit(iter) < 0) {
        /* some failure, send as untagged error */
        client_send_line(cmd->client, t_strconcat(
            "* BAD ", quota_last_error(quota), NULL));
    }
    client_send_line(cmd->client, str_c(str));

    /* send QUOTA reply for each quotaroot */
    iter = quota_root_iter_init(quota, box);
    while ((root = quota_root_iter_next(iter)) != NULL)
        quota_send(cmd, root);
    if (quota_root_iter_deinit(iter) < 0) {
        /* some failure, send as untagged error */
        client_send_line(cmd->client, t_strconcat(
            "* BAD ", quota_last_error(quota), NULL));
    }

    mailbox_close(&box);

    client_send_tagline(cmd, "OK Getquotaroot completed.");
    return TRUE;
}

static bool cmd_getquota(struct client_command_context *cmd)
{
    const char *root_name;
        struct quota_root *root;

    /* <quota root> */
    if (!client_read_string_args(cmd, 1, &root_name))
        return FALSE;

    if (quota == NULL) {
        client_send_tagline(cmd, "OK No quota.");
        return TRUE;
    }

    root = quota_root_lookup(quota, root_name);
    if (root == NULL) {
        client_send_tagline(cmd, "NO Quota root doesn't exist.");
        return TRUE;
    }

    quota_send(cmd, root);
    client_send_tagline(cmd, "OK Getquota completed.");
    return TRUE;
}

static bool cmd_setquota(struct client_command_context *cmd)
{
    struct quota_root *root;
        struct imap_arg *args, *arg;
    const char *root_name, *name;
    uint64_t value;

    /* <quota root> <resource limits> */
    if (!client_read_args(cmd, 2, 0, &args))
        return FALSE;

    root_name = imap_arg_string(&args[0]);
    if (args[1].type != IMAP_ARG_LIST || root_name == NULL) {
        client_send_command_error(cmd, "Invalid arguments.");
        return TRUE;
    }

    if (quota == NULL) {
        client_send_tagline(cmd, "OK No quota.");
        return TRUE;
    }

    root = quota_root_lookup(quota, root_name);
    if (root == NULL) {
        client_send_tagline(cmd, "NO Quota root doesn't exist.");
        return TRUE;
    }

        arg = IMAP_ARG_LIST(&args[1])->args;
    for (; arg->type != IMAP_ARG_EOL; arg += 2) {
        name = imap_arg_string(arg);
        if (name == NULL || arg[1].type != IMAP_ARG_ATOM ||
            !is_numeric(IMAP_ARG_STR(&arg[1]), '\0')) {
            client_send_command_error(cmd, "Invalid arguments.");
            return TRUE;
        }

                value = strtoull(IMAP_ARG_STR_NONULL(&arg[1]), NULL, 10);
        if (quota_set_resource(root, name, value) < 0) {
            client_send_command_error(cmd,
                          quota_last_error(quota));
            return TRUE;
        }
    }

    client_send_tagline(cmd, "OK Setquota completed.");
    return TRUE;
}

void imap_quota_plugin_init(void)
{
    command_register("GETQUOTAROOT", cmd_getquotaroot);
    command_register("GETQUOTA", cmd_getquota);
    command_register("SETQUOTA", cmd_setquota);
    str_append(capability_string, " QUOTA");
}

void imap_quota_plugin_deinit(void)
{
    command_unregister("GETQUOTAROOT");
    command_unregister("GETQUOTA");
    command_unregister("SETQUOTA");
}