int quota_count(struct quota_root *root, uint64_t *bytes_r, uint64_t *count_r)
{
struct quota_mailbox_iter *iter;
const struct mailbox_info *info;
int ret = 0;
*bytes_r = *count_r = 0;
if (root->recounting)
return 0;
root->recounting = TRUE;
iter = quota_mailbox_iter_begin(root);
while (ret >= 0 && (info = quota_mailbox_iter_next(iter)) != NULL) {
ret = quota_count_mailbox(root, info->ns, info->vname,
bytes_r, count_r);
}
quota_mailbox_iter_deinit(&iter);
root->recounting = FALSE;
return ret;
}
static const struct mailbox_info *
quota_mailbox_iter_next(struct quota_mailbox_iter *iter)
{
struct mail_namespace *const *namespaces;
const struct mailbox_info *info;
unsigned int count;
if (iter->iter == NULL) {
namespaces = array_get(&iter->root->quota->namespaces, &count);
if (iter->ns_idx >= count)
return NULL;
iter->ns = namespaces[iter->ns_idx++];
iter->iter = mailbox_list_iter_init(iter->ns->list, "*",
MAILBOX_LIST_ITER_SKIP_ALIASES |
MAILBOX_LIST_ITER_RETURN_NO_FLAGS |
MAILBOX_LIST_ITER_NO_AUTO_BOXES);
}
while ((info = mailbox_list_iter_next(iter->iter)) != NULL) {
if ((info->flags & (MAILBOX_NONEXISTENT |
MAILBOX_NOSELECT)) == 0)
return info;
}
if (mailbox_list_iter_deinit(&iter->iter) < 0) {
i_error("quota: Listing namespace '%s' failed: %s",
iter->ns->prefix,
mailbox_list_get_last_error(iter->ns->list, NULL));
iter->failed = TRUE;
}
if (iter->ns->prefix_len > 0 &&
(iter->ns->prefix_len != 6 ||
strncasecmp(iter->ns->prefix, "INBOX", 5) != 0)) {
/* if the namespace prefix itself exists, count it also */
iter->info.ns = iter->ns;
iter->info.vname = t_strndup(iter->ns->prefix,
iter->ns->prefix_len-1);
return &iter->info;
}
/* try the next namespace */
return quota_mailbox_iter_next(iter);
}
