doveadm-mail.c revision 09304bdc1e5e416aa343802732a15064f49495cc
5f5870385cff47efd2f58e7892f251cf13761528Timo Sirainen/* Copyright (c) 2009-2010 Dovecot authors, see the included COPYING file */
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen#include "lib.h"
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen#include "array.h"
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen#include "lib-signals.h"
cbc61fcb33b370d049c16a3c44568b4deb4e2b33Timo Sirainen#include "ioloop.h"
a1607f6001a9949e1cf9d49eb0aa936dbb4c77ffTimo Sirainen#include "str.h"
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen#include "module-dir.h"
00d58fcfe8191d6ce7efa801d289a5c0fe88d1aeTimo Sirainen#include "wildcard-match.h"
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen#include "master-service.h"
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen#include "imap-utf7.h"
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen#include "mail-user.h"
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen#include "mail-namespace.h"
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen#include "mail-storage.h"
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen#include "mail-storage-settings.h"
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen#include "mail-storage-service.h"
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen#include "mail-search-build.h"
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen#include "mail-search-parser.h"
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen#include "doveadm.h"
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen#include "doveadm-settings.h"
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen#include "doveadm-print.h"
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen#include "doveadm-mail.h"
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen#include <stdio.h>
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen#include <stdlib.h>
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo SirainenARRAY_TYPE(doveadm_mail_cmd) doveadm_mail_cmds;
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainenvoid (*hook_doveadm_mail_init)(struct doveadm_mail_cmd_context *ctx);
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainenstruct doveadm_mail_cmd_module_register
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen doveadm_mail_cmd_module_register = { 0 };
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainenstatic int killed_signo = 0;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainenstruct doveadm_mail_cmd_context *
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainendoveadm_mail_cmd_alloc_size(size_t size)
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen{
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen struct doveadm_mail_cmd_context *ctx;
cbc61fcb33b370d049c16a3c44568b4deb4e2b33Timo Sirainen pool_t pool;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen i_assert(size >= sizeof(struct doveadm_mail_cmd_context));
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
4462bd7b4c7ef3de006f060e155a90e5de7cae21Timo Sirainen pool = pool_alloconly_create("doveadm mail cmd", 1024);
f6c1297c26b355c4aec2a08978f51ec3efecb351Timo Sirainen ctx = p_malloc(pool, size);
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen ctx->pool = pool;
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen return ctx;
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen}
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainenstatic void
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainencmd_purge_run(struct doveadm_mail_cmd_context *ctx ATTR_UNUSED,
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen struct mail_user *user)
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen{
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen struct mail_namespace *ns;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen for (ns = user->namespaces; ns != NULL; ns = ns->next) {
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen if (ns->type != NAMESPACE_PRIVATE || ns->alias_for != NULL)
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen continue;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen if (mail_storage_purge(ns->storage) < 0) {
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen i_error("Purging namespace '%s' failed: %s", ns->prefix,
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen mail_storage_get_last_error(ns->storage, NULL));
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen }
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen }
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen}
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
cd83124e5d070a016c590bb0b1096d7828c7b6adTimo Sirainenstatic struct doveadm_mail_cmd_context *cmd_purge_alloc(void)
73b50eecfc31750a312e2f940023f522eb07178cTimo Sirainen{
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen struct doveadm_mail_cmd_context *ctx;
cd83124e5d070a016c590bb0b1096d7828c7b6adTimo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen ctx = doveadm_mail_cmd_alloc(struct doveadm_mail_cmd_context);
cbc61fcb33b370d049c16a3c44568b4deb4e2b33Timo Sirainen ctx->v.run = cmd_purge_run;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen return ctx;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen}
cd83124e5d070a016c590bb0b1096d7828c7b6adTimo Sirainen
cbc61fcb33b370d049c16a3c44568b4deb4e2b33Timo Sirainenstatic int mailbox_find_and_open(struct mail_user *user, const char *mailbox,
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen struct mailbox **box_r)
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen{
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen struct mail_namespace *ns;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen struct mailbox *box;
73b50eecfc31750a312e2f940023f522eb07178cTimo Sirainen string_t *str;
cbc61fcb33b370d049c16a3c44568b4deb4e2b33Timo Sirainen const char *orig_mailbox = mailbox;
e07bf3772a2bc075de4915ad0961beb8d083c22dTimo Sirainen
e07bf3772a2bc075de4915ad0961beb8d083c22dTimo Sirainen str = t_str_new(128);
73b50eecfc31750a312e2f940023f522eb07178cTimo Sirainen if (imap_utf8_to_utf7(mailbox, str) < 0)
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen i_fatal("Mailbox name not valid UTF-8: %s", mailbox);
cbc61fcb33b370d049c16a3c44568b4deb4e2b33Timo Sirainen mailbox = str_c(str);
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
e07bf3772a2bc075de4915ad0961beb8d083c22dTimo Sirainen ns = mail_namespace_find(user->namespaces, &mailbox);
e07bf3772a2bc075de4915ad0961beb8d083c22dTimo Sirainen if (ns == NULL)
e07bf3772a2bc075de4915ad0961beb8d083c22dTimo Sirainen i_fatal("Can't find namespace for mailbox %s", mailbox);
e07bf3772a2bc075de4915ad0961beb8d083c22dTimo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen box = mailbox_alloc(ns->list, mailbox, MAILBOX_FLAG_KEEP_RECENT |
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen MAILBOX_FLAG_IGNORE_ACLS);
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen if (mailbox_open(box) < 0) {
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen i_error("Opening mailbox %s failed: %s", orig_mailbox,
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen mail_storage_get_last_error(mailbox_get_storage(box),
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen NULL));
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen mailbox_free(&box);
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen return -1;
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen }
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen *box_r = box;
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen return 0;
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen}
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainenint doveadm_mailbox_find_and_sync(struct mail_user *user, const char *mailbox,
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen struct mailbox **box_r)
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen{
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen if (mailbox_find_and_open(user, mailbox, box_r) < 0)
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen return -1;
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen if (mailbox_sync(*box_r, MAILBOX_SYNC_FLAG_FULL_READ) < 0) {
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen i_error("Syncing mailbox %s failed: %s", mailbox,
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen mail_storage_get_last_error(mailbox_get_storage(*box_r),
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen NULL));
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen return -1;
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen }
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen return 0;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen}
96308127e006bb3b1108093bcf4cc1fd9481cb7aTimo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainenstruct mail_search_args *
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainendoveadm_mail_build_search_args(const char *const args[])
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen{
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen struct mail_search_parser *parser;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen struct mail_search_args *sargs;
d66be2bebfa96e7d3d20e2153f60e6e25dcc9a18Timo Sirainen const char *error;
e376e08040b5f21ff79a15ae728d2532a34207f6Timo Sirainen
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen parser = mail_search_parser_init_cmdline(args);
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen if (mail_search_build(mail_search_register_get_human(),
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen parser, "UTF-8", &sargs, &error) < 0)
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen i_fatal("%s", error);
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen mail_search_parser_deinit(&parser);
73b50eecfc31750a312e2f940023f522eb07178cTimo Sirainen return sargs;
fd35227c47190afc832579ca5c76524792701bf7Timo Sirainen}
fd35227c47190afc832579ca5c76524792701bf7Timo Sirainen
fd35227c47190afc832579ca5c76524792701bf7Timo Sirainenstruct force_resync_cmd_context {
fd35227c47190afc832579ca5c76524792701bf7Timo Sirainen struct doveadm_mail_cmd_context ctx;
fd35227c47190afc832579ca5c76524792701bf7Timo Sirainen const char *mailbox;
2e263a9d901483a902720a30c474761bd3324fe8Timo Sirainen};
2e263a9d901483a902720a30c474761bd3324fe8Timo Sirainen
2e263a9d901483a902720a30c474761bd3324fe8Timo Sirainenstatic void cmd_force_resync_run(struct doveadm_mail_cmd_context *_ctx,
2e263a9d901483a902720a30c474761bd3324fe8Timo Sirainen struct mail_user *user)
2e263a9d901483a902720a30c474761bd3324fe8Timo Sirainen{
73b50eecfc31750a312e2f940023f522eb07178cTimo Sirainen struct force_resync_cmd_context *ctx =
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen (struct force_resync_cmd_context *)_ctx;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen struct mail_storage *storage;
96308127e006bb3b1108093bcf4cc1fd9481cb7aTimo Sirainen struct mailbox *box;
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen if (mailbox_find_and_open(user, ctx->mailbox, &box) < 0) {
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen _ctx->failed = TRUE;
d66be2bebfa96e7d3d20e2153f60e6e25dcc9a18Timo Sirainen return;
d66be2bebfa96e7d3d20e2153f60e6e25dcc9a18Timo Sirainen }
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen storage = mailbox_get_storage(box);
cbc61fcb33b370d049c16a3c44568b4deb4e2b33Timo Sirainen if (mailbox_sync(box, MAILBOX_SYNC_FLAG_FORCE_RESYNC |
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen MAILBOX_SYNC_FLAG_FIX_INCONSISTENT) < 0) {
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen i_error("Forcing a resync on mailbox %s failed: %s",
783278c012551bc2f25f065a8d3ec1a3cfc0d296Timo Sirainen ctx->mailbox,
2615df45a8027948a474abe5e817b34b0499c171Timo Sirainen mail_storage_get_last_error(storage, NULL));
5d49cbcf87354f0ddf3b71bc5f0cefdc02b14f68Timo Sirainen _ctx->failed = TRUE;
ce3faaaaab3e2d45b023396287e02f88e5c76e74Timo Sirainen }
fd35227c47190afc832579ca5c76524792701bf7Timo Sirainen mailbox_free(&box);
ce3faaaaab3e2d45b023396287e02f88e5c76e74Timo Sirainen}
ce3faaaaab3e2d45b023396287e02f88e5c76e74Timo Sirainen
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainenstatic void cmd_force_resync_init(struct doveadm_mail_cmd_context *_ctx,
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen const char *const args[])
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen{
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen struct force_resync_cmd_context *ctx =
3e7565a7b39694bcdf448d8eb2a7f0774733297bTimo Sirainen (struct force_resync_cmd_context *)_ctx;
3e7565a7b39694bcdf448d8eb2a7f0774733297bTimo Sirainen const char *mailbox = args[0];
3e7565a7b39694bcdf448d8eb2a7f0774733297bTimo Sirainen
3e7565a7b39694bcdf448d8eb2a7f0774733297bTimo Sirainen if (mailbox == NULL || args[1] != NULL)
3e7565a7b39694bcdf448d8eb2a7f0774733297bTimo Sirainen doveadm_mail_help_name("force-resync");
3e7565a7b39694bcdf448d8eb2a7f0774733297bTimo Sirainen
3e7565a7b39694bcdf448d8eb2a7f0774733297bTimo Sirainen ctx->mailbox = p_strdup(ctx->ctx.pool, mailbox);
3e7565a7b39694bcdf448d8eb2a7f0774733297bTimo Sirainen}
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainenstatic struct doveadm_mail_cmd_context *cmd_force_resync_alloc(void)
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen{
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen struct force_resync_cmd_context *ctx;
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen ctx = doveadm_mail_cmd_alloc(struct force_resync_cmd_context);
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen ctx->ctx.v.init = cmd_force_resync_init;
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen ctx->ctx.v.run = cmd_force_resync_run;
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen return &ctx->ctx;
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen}
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainenstatic int
2e263a9d901483a902720a30c474761bd3324fe8Timo Sirainendoveadm_mail_next_user(struct doveadm_mail_cmd_context *ctx,
2e263a9d901483a902720a30c474761bd3324fe8Timo Sirainen const struct mail_storage_service_input *input,
2e263a9d901483a902720a30c474761bd3324fe8Timo Sirainen const char **error_r)
2e263a9d901483a902720a30c474761bd3324fe8Timo Sirainen{
2e263a9d901483a902720a30c474761bd3324fe8Timo Sirainen struct mail_storage_service_user *service_user;
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen const char *error;
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen int ret;
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen i_set_failure_prefix(t_strdup_printf("doveadm(%s): ", input->username));
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen ret = mail_storage_service_lookup(ctx->storage_service, input,
bbd4c4cf902539c25c471157eb9849459734759cTimo Sirainen &service_user, &error);
3e7565a7b39694bcdf448d8eb2a7f0774733297bTimo Sirainen if (ret <= 0) {
3e7565a7b39694bcdf448d8eb2a7f0774733297bTimo Sirainen if (ret < 0) {
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen *error_r = t_strdup_printf("User lookup failed: %s",
19e8adccba16ff419f5675b1575358c2956dce83Timo Sirainen error);
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen }
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen return ret;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen }
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen ret = mail_storage_service_next(ctx->storage_service, service_user,
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen &ctx->cur_mail_user);
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen if (ret < 0) {
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen *error_r = "User init failed";
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen mail_storage_service_user_free(&service_user);
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen return ret;
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen }
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen ctx->v.run(ctx, ctx->cur_mail_user);
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen mail_user_unref(&ctx->cur_mail_user);
73b50eecfc31750a312e2f940023f522eb07178cTimo Sirainen mail_storage_service_user_free(&service_user);
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen return 1;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen}
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainenstatic void
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainendoveadm_mail_single_user(struct doveadm_mail_cmd_context *ctx,
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen const char *username,
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen enum mail_storage_service_flags service_flags)
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen{
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen struct mail_storage_service_input input;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen const char *error;
73b50eecfc31750a312e2f940023f522eb07178cTimo Sirainen int ret;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen if (username == NULL)
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen i_fatal("USER environment is missing and -u option not used");
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen memset(&input, 0, sizeof(input));
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen input.username = username;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen ctx->storage_service = mail_storage_service_init(master_service, NULL,
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen service_flags);
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen if (hook_doveadm_mail_init != NULL)
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen hook_doveadm_mail_init(ctx);
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen ret = doveadm_mail_next_user(ctx, &input, &error);
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen if (ret < 0)
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen i_fatal("%s", error);
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen else if (ret == 0)
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen i_fatal("User no longer exists");
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen mail_storage_service_deinit(&ctx->storage_service);
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen}
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
cbc61fcb33b370d049c16a3c44568b4deb4e2b33Timo Sirainenstatic void sig_die(const siginfo_t *si, void *context ATTR_UNUSED)
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen{
cbc61fcb33b370d049c16a3c44568b4deb4e2b33Timo Sirainen killed_signo = si->si_signo;
cbc61fcb33b370d049c16a3c44568b4deb4e2b33Timo Sirainen}
cbc61fcb33b370d049c16a3c44568b4deb4e2b33Timo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainenstatic void
cbc61fcb33b370d049c16a3c44568b4deb4e2b33Timo Sirainendoveadm_mail_all_users(struct doveadm_mail_cmd_context *ctx,
cbc61fcb33b370d049c16a3c44568b4deb4e2b33Timo Sirainen const char *wildcard_user,
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen enum mail_storage_service_flags service_flags)
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen{
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen struct mail_storage_service_input input;
bd4d0a1a7c0626452b8d82f37e3ec07267ac9896Timo Sirainen unsigned int user_idx, user_count, interval, n;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen const char *user, *error;
bd4d0a1a7c0626452b8d82f37e3ec07267ac9896Timo Sirainen int ret;
bd4d0a1a7c0626452b8d82f37e3ec07267ac9896Timo Sirainen
bd4d0a1a7c0626452b8d82f37e3ec07267ac9896Timo Sirainen service_flags |= MAIL_STORAGE_SERVICE_FLAG_USERDB_LOOKUP;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
cbc61fcb33b370d049c16a3c44568b4deb4e2b33Timo Sirainen memset(&input, 0, sizeof(input));
bd4d0a1a7c0626452b8d82f37e3ec07267ac9896Timo Sirainen input.service = "doveadm";
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen ctx->storage_service = mail_storage_service_init(master_service, NULL,
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen service_flags);
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen lib_signals_set_handler(SIGINT, FALSE, sig_die, NULL);
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen lib_signals_set_handler(SIGTERM, FALSE, sig_die, NULL);
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen if (hook_doveadm_mail_init != NULL)
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen hook_doveadm_mail_init(ctx);
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen user_count = mail_storage_service_all_init(ctx->storage_service);
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen n = user_count / 10000;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen for (interval = 10; n > 0 && interval < 1000; interval *= 10)
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen n /= 10;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen user_idx = 0;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen while ((ret = ctx->v.get_next_user(ctx, &user)) > 0) {
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen if (wildcard_user != NULL) {
8e371a3ce32bd64288786855b8ce0cb63f19f7d1Timo Sirainen if (!wildcard_match_icase(user, wildcard_user))
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen continue;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen }
56963ffad65b860c827553dfaf09fb766cb7e20eTimo Sirainen input.username = user;
56963ffad65b860c827553dfaf09fb766cb7e20eTimo Sirainen doveadm_print_sticky("username", user);
56963ffad65b860c827553dfaf09fb766cb7e20eTimo Sirainen T_BEGIN {
56963ffad65b860c827553dfaf09fb766cb7e20eTimo Sirainen ret = doveadm_mail_next_user(ctx, &input, &error);
56963ffad65b860c827553dfaf09fb766cb7e20eTimo Sirainen if (ret < 0)
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen i_error("%s", error);
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen else if (ret == 0)
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen i_info("User no longer exists, skipping");
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen } T_END;
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen if (ret == -1)
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen break;
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen if (doveadm_verbose) {
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen if (++user_idx % interval == 0) {
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen printf("\r%d / %d", user_idx, user_count);
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen fflush(stdout);
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen }
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen }
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen if (killed_signo != 0) {
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen i_warning("Killed with signal %d", killed_signo);
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen ret = -1;
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen break;
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen }
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen }
e34d170f8f0e084bd94bfbc1a7085ece67e508dfTimo Sirainen if (doveadm_verbose)
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen printf("\n");
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen i_set_failure_prefix("doveadm: ");
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen if (ret < 0)
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen i_error("Failed to iterate through some users");
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen mail_storage_service_deinit(&ctx->storage_service);
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen}
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen
e34d170f8f0e084bd94bfbc1a7085ece67e508dfTimo Sirainenstatic void
e34d170f8f0e084bd94bfbc1a7085ece67e508dfTimo Sirainendoveadm_mail_cmd_init_noop(struct doveadm_mail_cmd_context *ctx ATTR_UNUSED,
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen const char *const args[] ATTR_UNUSED)
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen{
31a574fda352ef4f71dbff9c30e15e4744e132c0Timo Sirainen}
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainenstatic int
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainendoveadm_mail_cmd_get_next_user(struct doveadm_mail_cmd_context *ctx,
9514dbd41faad453406fc3a10aca791417a9ce75Timo Sirainen const char **username_r)
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen{
9514dbd41faad453406fc3a10aca791417a9ce75Timo Sirainen return mail_storage_service_all_next(ctx->storage_service, username_r);
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen}
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainenstatic void
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainendoveadm_mail_cmd_deinit_noop(struct doveadm_mail_cmd_context *ctx ATTR_UNUSED)
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen{
b87daa509bf5b306189282a9df795d094a6d7150Timo Sirainen}
b87daa509bf5b306189282a9df795d094a6d7150Timo Sirainen
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainenstatic void
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainendoveadm_mail_cmd(const struct doveadm_mail_cmd *cmd, int argc, char *argv[])
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen{
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen enum mail_storage_service_flags service_flags =
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen MAIL_STORAGE_SERVICE_FLAG_NO_LOG_INIT;
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen struct doveadm_mail_cmd_context *ctx;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen const char *getopt_args, *username, *wildcard_user;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen int c;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen if (doveadm_debug)
f6d57a2c182f63cd52819f0abb3c3d9f828afe19Timo Sirainen service_flags |= MAIL_STORAGE_SERVICE_FLAG_DEBUG;
783278c012551bc2f25f065a8d3ec1a3cfc0d296Timo Sirainen
783278c012551bc2f25f065a8d3ec1a3cfc0d296Timo Sirainen ctx = cmd->alloc();
4073f0dbf3277f981a8fcee3b89ea15aaf380a7fTimo Sirainen if (ctx->v.init == NULL)
4073f0dbf3277f981a8fcee3b89ea15aaf380a7fTimo Sirainen ctx->v.init = doveadm_mail_cmd_init_noop;
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen if (ctx->v.get_next_user == NULL)
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen ctx->v.get_next_user = doveadm_mail_cmd_get_next_user;
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen if (ctx->v.deinit == NULL)
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen ctx->v.deinit = doveadm_mail_cmd_deinit_noop;
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen p_array_init(&ctx->module_contexts, ctx->pool, 5);
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen
5666a3d6a7ea89362b8d9e8b39b15424cd9d6388Timo Sirainen getopt_args = t_strconcat("Au:", ctx->getopt_args, NULL);
5666a3d6a7ea89362b8d9e8b39b15424cd9d6388Timo Sirainen username = getenv("USER");
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen wildcard_user = NULL;
539977f9257bd8985be5a8093658da266ae9cd19Timo Sirainen while ((c = getopt(argc, argv, getopt_args)) > 0) {
5666a3d6a7ea89362b8d9e8b39b15424cd9d6388Timo Sirainen switch (c) {
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen case 'A':
5666a3d6a7ea89362b8d9e8b39b15424cd9d6388Timo Sirainen ctx->iterate_all_users = TRUE;
5666a3d6a7ea89362b8d9e8b39b15424cd9d6388Timo Sirainen if (doveadm_print_is_initialized()) {
5666a3d6a7ea89362b8d9e8b39b15424cd9d6388Timo Sirainen doveadm_print_header("username", "Username",
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen DOVEADM_PRINT_HEADER_FLAG_STICKY |
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen DOVEADM_PRINT_HEADER_FLAG_HIDE_TITLE);
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen }
0df9428baed48afaff90b4d4f03792d2fd756a43Timo Sirainen break;
5666a3d6a7ea89362b8d9e8b39b15424cd9d6388Timo Sirainen case 'u':
cf7164ece50797a67fc4bfb5889022ac93a36a8aTimo Sirainen service_flags |=
MAIL_STORAGE_SERVICE_FLAG_USERDB_LOOKUP;
username = optarg;
if (strchr(username, '*') != NULL ||
strchr(username, '?') != NULL)
wildcard_user = username;
break;
default:
if (ctx->v.parse_arg == NULL ||
!ctx->v.parse_arg(ctx, c))
doveadm_mail_help(cmd);
}
}
argv += optind;
if (argv[0] != NULL && cmd->usage_args == NULL) {
i_fatal("doveadm %s: Unknown parameter: %s",
cmd->name, argv[0]);
}
ctx->v.init(ctx, (const void *)argv);
if (!ctx->iterate_all_users && wildcard_user == NULL) {
doveadm_mail_single_user(ctx, username, service_flags);
} else {
service_flags |= MAIL_STORAGE_SERVICE_FLAG_TEMP_PRIV_DROP;
doveadm_mail_all_users(ctx, wildcard_user, service_flags);
}
ctx->v.deinit(ctx);
doveadm_print_flush();
if (ctx->failed)
exit(FATAL_DEFAULT);
}
static bool
doveadm_mail_try_run_multi_word(const struct doveadm_mail_cmd *cmd,
const char *cmdname, int argc, char *argv[])
{
unsigned int len;
if (argc < 2)
return FALSE;
len = strlen(argv[1]);
if (strncmp(cmdname, argv[1], len) != 0)
return FALSE;
if (cmdname[len] == ' ') {
/* more args */
return doveadm_mail_try_run_multi_word(cmd, cmdname + len + 1,
argc - 1, argv + 1);
}
if (cmdname[len] != '\0')
return FALSE;
/* match */
doveadm_mail_cmd(cmd, argc - 1, argv + 1);
return TRUE;
}
bool doveadm_mail_try_run(const char *cmd_name, int argc, char *argv[])
{
const struct doveadm_mail_cmd *cmd;
unsigned int cmd_name_len;
i_assert(argc > 0);
cmd_name_len = strlen(cmd_name);
array_foreach(&doveadm_mail_cmds, cmd) {
if (strcmp(cmd->name, cmd_name) == 0) {
doveadm_mail_cmd(cmd, argc, argv);
return TRUE;
}
/* see if it matches a multi-word command */
if (strncmp(cmd->name, cmd_name, cmd_name_len) == 0 &&
cmd->name[cmd_name_len] == ' ') {
const char *subcmd = cmd->name + cmd_name_len + 1;
if (doveadm_mail_try_run_multi_word(cmd, subcmd,
argc, argv))
return TRUE;
}
}
return FALSE;
}
void doveadm_mail_register_cmd(const struct doveadm_mail_cmd *cmd)
{
/* for now we'll just assume that cmd will be permanently in memory */
array_append(&doveadm_mail_cmds, cmd, 1);
}
void doveadm_mail_usage(string_t *out)
{
const struct doveadm_mail_cmd *cmd;
array_foreach(&doveadm_mail_cmds, cmd) {
str_printfa(out, "%s\t[-u <user>|-A]", cmd->name);
if (cmd->usage_args != NULL)
str_printfa(out, " %s", cmd->usage_args);
str_append_c(out, '\n');
}
}
void doveadm_mail_help(const struct doveadm_mail_cmd *cmd)
{
fprintf(stderr, "doveadm %s [-u <user>|-A] %s\n", cmd->name,
cmd->usage_args == NULL ? "" : cmd->usage_args);
exit(0);
}
void doveadm_mail_try_help_name(const char *cmd_name)
{
const struct doveadm_mail_cmd *cmd;
array_foreach(&doveadm_mail_cmds, cmd) {
if (strcmp(cmd->name, cmd_name) == 0)
doveadm_mail_help(cmd);
}
}
bool doveadm_mail_has_subcommands(const char *cmd_name)
{
const struct doveadm_mail_cmd *cmd;
unsigned int len = strlen(cmd_name);
array_foreach(&doveadm_mail_cmds, cmd) {
if (strncmp(cmd->name, cmd_name, len) == 0 &&
cmd->name[len] == ' ')
return TRUE;
}
return FALSE;
}
void doveadm_mail_help_name(const char *cmd_name)
{
doveadm_mail_try_help_name(cmd_name);
i_fatal("Missing help for command %s", cmd_name);
}
static struct doveadm_mail_cmd cmd_force_resync = {
cmd_force_resync_alloc, "force-resync", "<mailbox>"
};
static struct doveadm_mail_cmd cmd_purge = {
cmd_purge_alloc, "purge", NULL
};
static struct doveadm_mail_cmd *mail_commands[] = {
&cmd_force_resync,
&cmd_purge,
&cmd_expunge,
&cmd_search,
&cmd_fetch,
&cmd_altmove,
&cmd_mailbox_list,
&cmd_mailbox_create,
&cmd_mailbox_delete,
&cmd_mailbox_rename,
&cmd_mailbox_subscribe,
&cmd_mailbox_unsubscribe,
&cmd_mailbox_status
};
void doveadm_mail_init(void)
{
struct module_dir_load_settings mod_set;
unsigned int i;
i_array_init(&doveadm_mail_cmds, 32);
for (i = 0; i < N_ELEMENTS(mail_commands); i++)
doveadm_mail_register_cmd(mail_commands[i]);
memset(&mod_set, 0, sizeof(mod_set));
mod_set.version = master_service_get_version_string(master_service);
mod_set.require_init_funcs = TRUE;
mod_set.debug = doveadm_debug;
/* load all configured mail plugins */
mail_storage_service_modules =
module_dir_load_missing(mail_storage_service_modules,
doveadm_settings->mail_plugin_dir,
doveadm_settings->mail_plugins,
&mod_set);
}
void doveadm_mail_deinit(void)
{
array_free(&doveadm_mail_cmds);
}