imap-quota-plugin.c revision d337f4291d9ea9cd1adee0f468bc0c3baeabb4fe
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen/* Copyright (c) 2005-2010 Dovecot authors, see the included COPYING file */
c9099b08b3cae8a849098ca776b4363c6d5f5f36Timo Sirainenconst char *imap_quota_plugin_version = DOVECOT_VERSION;
c9099b08b3cae8a849098ca776b4363c6d5f5f36Timo Sirainenstatic void (*next_hook_client_created)(struct client **client);
c9099b08b3cae8a849098ca776b4363c6d5f5f36Timo Sirainenstatic const char *
c9099b08b3cae8a849098ca776b4363c6d5f5f36Timo Sirainenimap_quota_root_get_name(struct mail_user *user, struct mail_user *owner,
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen return t_strdup_printf("%s%c%s", owner->username,
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainenquota_reply_write(string_t *str, struct mail_user *user,
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen struct mail_user *owner, struct quota_root *root)
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen unsigned int i;
7000810786f2959f02cd6d2f4151a9eb61ff5db8Timo Sirainen name = imap_quota_root_get_name(user, owner, root);
7000810786f2959f02cd6d2f4151a9eb61ff5db8Timo Sirainen ret = quota_get_resource(root, "", *list, &value, &limit);
7000810786f2959f02cd6d2f4151a9eb61ff5db8Timo Sirainen (unsigned long long)value,
7000810786f2959f02cd6d2f4151a9eb61ff5db8Timo Sirainen (unsigned long long)limit);
d6a7cb184cc882a90aa3d9312082e0029f354ff6Timo Sirainen str_append(str, "* BAD Internal quota calculation error\r\n");
d6a7cb184cc882a90aa3d9312082e0029f354ff6Timo Sirainenstatic bool cmd_getquotaroot(struct client_command_context *cmd)
d6a7cb184cc882a90aa3d9312082e0029f354ff6Timo Sirainen struct quota_user *quser = QUOTA_USER_CONTEXT(client->user);
d6a7cb184cc882a90aa3d9312082e0029f354ff6Timo Sirainen /* <mailbox> */
f5e8a76a128d4e92f0641135183c164fd5c5ce5eTimo Sirainen if (!client_read_string_args(cmd, 1, &mailbox))
f5e8a76a128d4e92f0641135183c164fd5c5ce5eTimo Sirainen ns = client_find_namespace(cmd, mailbox, &storage_name, NULL);
cc935aff970ed6c24d136cc560c7e705a49d536cTimo Sirainen if (ns->owner != NULL && ns->owner != client->user &&
cc935aff970ed6c24d136cc560c7e705a49d536cTimo Sirainen client_send_tagline(cmd, "NO Not showing other users' quota.");
cc935aff970ed6c24d136cc560c7e705a49d536cTimo Sirainen box = mailbox_alloc(ns->list, storage_name, MAILBOX_FLAG_READONLY |
cc935aff970ed6c24d136cc560c7e705a49d536cTimo Sirainen /* build QUOTAROOT reply and QUOTA reply for all quota roots */
cc935aff970ed6c24d136cc560c7e705a49d536cTimo Sirainen imap_quote_append_string(quotaroot_reply, mailbox, FALSE);
cc935aff970ed6c24d136cc560c7e705a49d536cTimo Sirainen while ((root = quota_root_iter_next(iter)) != NULL) {
279c6b6d0b0a159c8533102e7e914db21dadcb03Timo Sirainen name = imap_quota_root_get_name(client->user, ns->owner, root);
279c6b6d0b0a159c8533102e7e914db21dadcb03Timo Sirainen imap_quote_append_string(quotaroot_reply, name, FALSE);
279c6b6d0b0a159c8533102e7e914db21dadcb03Timo Sirainen quota_reply_write(quota_reply, client->user, ns->owner, root);
279c6b6d0b0a159c8533102e7e914db21dadcb03Timo Sirainen /* send replies */
27e859cee42654bff801ba96677cfc4e4e0108c7Timo Sirainen client_send_line(client, str_c(quotaroot_reply));
27e859cee42654bff801ba96677cfc4e4e0108c7Timo Sirainen o_stream_send(client->output, str_data(quota_reply),
071543cc13df9600d2e97aa35f28907be5a79477Timo Sirainen client_send_tagline(cmd, "OK Getquotaroot completed.");
632018810af689442569cbb0139c55868923ccfeTimo Sirainenstatic bool cmd_getquota(struct client_command_context *cmd)
632018810af689442569cbb0139c55868923ccfeTimo Sirainen const char *root_name, *p;
632018810af689442569cbb0139c55868923ccfeTimo Sirainen /* <quota root> */
387f9e3b4120273ad0213206a0e9cc2dc0e62ccaTimo Sirainen if (!client_read_string_args(cmd, 1, &root_name))
632018810af689442569cbb0139c55868923ccfeTimo Sirainen root = quota_root_lookup(cmd->client->user, root_name);
632018810af689442569cbb0139c55868923ccfeTimo Sirainen if (root == NULL && cmd->client->user->admin) {
8d3af185ae454653fad60e41c5f36edb1d45c868Timo Sirainen /* we're an admin. see if there's a quota root for another
dea8bfa31729dbdde1b12718f1ef98fac4e99db9Timo Sirainen client_send_tagline(cmd, "NO Quota root doesn't exist.");
dea8bfa31729dbdde1b12718f1ef98fac4e99db9Timo Sirainen quota_reply_write(quota_reply, cmd->client->user, owner, root);
dea8bfa31729dbdde1b12718f1ef98fac4e99db9Timo Sirainen o_stream_send(cmd->client->output, str_data(quota_reply),
dea8bfa31729dbdde1b12718f1ef98fac4e99db9Timo Sirainen client_send_tagline(cmd, "OK Getquota completed.");
dea8bfa31729dbdde1b12718f1ef98fac4e99db9Timo Sirainenstatic bool cmd_setquota(struct client_command_context *cmd)
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen const char *root_name, *name, *value_str, *error;
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen /* <quota root> <resource limits> */
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen if (!imap_arg_get_astring(&args[0], &root_name) ||
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen client_send_command_error(cmd, "Invalid arguments.");
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen root = quota_root_lookup(cmd->client->user, root_name);
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen client_send_tagline(cmd, "NO Quota root doesn't exist.");
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen for (; !IMAP_ARG_IS_EOL(list_args); list_args += 2) {
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen if (!imap_arg_get_atom(&list_args[0], &name) ||
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen !imap_arg_get_atom(&list_args[1], &value_str) ||
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen client_send_command_error(cmd, "Invalid arguments.");
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen if (quota_set_resource(root, name, value, &error) < 0) {
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen client_send_tagline(cmd, "OK Setquota completed.");
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainenstatic void imap_quota_client_created(struct client **client)
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen if (mail_user_is_plugin_loaded((*client)->user, imap_quota_module))
f8740ac53310cd28ba4ec6dc9e9ce6e9a3688f39Timo Sirainen str_append((*client)->capability_string, " QUOTA");
void imap_quota_plugin_deinit(void)