doveadm-mail.c revision 2e57fc359c4520bb6ddba32095475e5f518c4066
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch/* Copyright (c) 2009-2010 Dovecot authors, see the included COPYING file */
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan BoschARRAY_TYPE(doveadm_mail_cmd) doveadm_mail_cmds;
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschvoid (*hook_doveadm_mail_init)(struct doveadm_mail_cmd_context *ctx);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch i_assert(size >= sizeof(struct doveadm_mail_cmd_context));
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch pool = pool_alloconly_create("doveadm mail cmd", 1024);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschcmd_purge_run(struct doveadm_mail_cmd_context *ctx ATTR_UNUSED,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch for (ns = user->namespaces; ns != NULL; ns = ns->next) {
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch if (ns->type != NAMESPACE_PRIVATE || ns->alias_for != NULL)
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch i_error("Purging namespace '%s' failed: %s", ns->prefix,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch mail_storage_get_last_error(ns->storage, NULL));
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschstatic struct doveadm_mail_cmd_context *cmd_purge_alloc(void)
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch ctx = doveadm_mail_cmd_alloc(struct doveadm_mail_cmd_context);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschstatic struct mailbox *
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschmailbox_find_and_open(struct mail_user *user, const char *mailbox)
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch ns = mail_namespace_find(user->namespaces, &mailbox);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch i_fatal("Can't find namespace for mailbox %s", mailbox);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch box = mailbox_alloc(ns->list, mailbox, MAILBOX_FLAG_KEEP_RECENT |
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch i_fatal("Opening mailbox %s failed: %s", orig_mailbox,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch mail_storage_get_last_error(mailbox_get_storage(box),
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschdoveadm_mailbox_find_and_sync(struct mail_user *user, const char *mailbox)
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch if (mailbox_sync(box, MAILBOX_SYNC_FLAG_FULL_READ) < 0) {
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch i_fatal("Syncing mailbox %s failed: %s", mailbox,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch mail_storage_get_last_error(mailbox_get_storage(box),
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschdoveadm_mail_build_search_args(const char *const args[])
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch parser = mail_search_parser_init_cmdline(args);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch if (mail_search_build(mail_search_register_get_human(),
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschstatic void cmd_force_resync_run(struct doveadm_mail_cmd_context *_ctx,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch box = mailbox_find_and_open(user, ctx->mailbox);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch if (mailbox_sync(box, MAILBOX_SYNC_FLAG_FORCE_RESYNC |
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch i_fatal("Forcing a resync on mailbox %s failed: %s",
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschstatic void cmd_force_resync_init(struct doveadm_mail_cmd_context *_ctx,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch const char *const args[])
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch ctx->mailbox = p_strdup(ctx->ctx.pool, mailbox);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschstatic struct doveadm_mail_cmd_context *cmd_force_resync_alloc(void)
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch ctx = doveadm_mail_cmd_alloc(struct force_resync_cmd_context);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschdoveadm_mail_next_user(struct doveadm_mail_cmd_context *ctx,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch const struct mail_storage_service_input *input,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch const char **error_r)
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch struct mail_storage_service_user *service_user;
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch i_set_failure_prefix(t_strdup_printf("doveadm(%s): ", input->username));
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch ret = mail_storage_service_lookup(ctx->storage_service, input,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch *error_r = t_strdup_printf("User lookup failed: %s",
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch ret = mail_storage_service_next(ctx->storage_service, service_user,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch mail_storage_service_user_free(&service_user);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch mail_storage_service_user_free(&service_user);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschdoveadm_mail_single_user(struct doveadm_mail_cmd_context *ctx,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch enum mail_storage_service_flags service_flags)
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch i_fatal("USER environment is missing and -u option not used");
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch ctx->storage_service = mail_storage_service_init(master_service, NULL,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch ret = doveadm_mail_next_user(ctx, &input, &error);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch else if (ret == 0)
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch mail_storage_service_deinit(&ctx->storage_service);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschstatic void sig_die(const siginfo_t *si, void *context ATTR_UNUSED)
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschdoveadm_mail_all_users(struct doveadm_mail_cmd_context *ctx,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch enum mail_storage_service_flags service_flags)
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch unsigned int user_idx, user_count, interval, n;
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch service_flags |= MAIL_STORAGE_SERVICE_FLAG_USERDB_LOOKUP;
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch ctx->storage_service = mail_storage_service_init(master_service, NULL,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch lib_signals_set_handler(SIGINT, FALSE, sig_die, NULL);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch lib_signals_set_handler(SIGTERM, FALSE, sig_die, NULL);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch user_count = mail_storage_service_all_init(ctx->storage_service);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch for (interval = 10; n > 0 && interval < 1000; interval *= 10)
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch while ((ret = ctx->v.get_next_user(ctx, &user)) > 0) {
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch if (!wildcard_match_icase(user, wildcard_user))
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch ret = doveadm_mail_next_user(ctx, &input, &error);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch else if (ret == 0)
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch i_warning("Killed with signal %d", killed_signo);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch i_error("Failed to iterate through some users");
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch mail_storage_service_deinit(&ctx->storage_service);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschdoveadm_mail_cmd_init_noop(struct doveadm_mail_cmd_context *ctx ATTR_UNUSED,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschdoveadm_mail_cmd_get_next_user(struct doveadm_mail_cmd_context *ctx,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch return mail_storage_service_all_next(ctx->storage_service, username_r);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschdoveadm_mail_cmd_deinit_noop(struct doveadm_mail_cmd_context *ctx ATTR_UNUSED)
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschdoveadm_mail_cmd(const struct doveadm_mail_cmd *cmd, int argc, char *argv[])
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch enum mail_storage_service_flags service_flags =
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch const char *getopt_args, *username, *wildcard_user;
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch service_flags |= MAIL_STORAGE_SERVICE_FLAG_DEBUG;
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch ctx->v.get_next_user = doveadm_mail_cmd_get_next_user;
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch p_array_init(&ctx->module_contexts, ctx->pool, 5);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch getopt_args = t_strconcat("Au:", ctx->getopt_args, NULL);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch while ((c = getopt(argc, argv, getopt_args)) > 0) {
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch if (argv[0] != NULL && cmd->usage_args == NULL) {
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch doveadm_mail_single_user(ctx, username, service_flags);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch service_flags |= MAIL_STORAGE_SERVICE_FLAG_TEMP_PRIV_DROP;
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch doveadm_mail_all_users(ctx, wildcard_user, service_flags);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschdoveadm_mail_try_run_multi_word(const struct doveadm_mail_cmd *cmd,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch unsigned int len;
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch /* more args */
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch return doveadm_mail_try_run_multi_word(cmd, cmdname + len + 1,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschbool doveadm_mail_try_run(const char *cmd_name, int argc, char *argv[])
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch /* see if it matches a multi-word command */
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch if (strncmp(cmd->name, cmd_name, cmd_name_len) == 0 &&
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch const char *subcmd = cmd->name + cmd_name_len + 1;
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch if (doveadm_mail_try_run_multi_word(cmd, subcmd,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschvoid doveadm_mail_register_cmd(const struct doveadm_mail_cmd *cmd)
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch /* for now we'll just assume that cmd will be permanently in memory */
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch fprintf(stderr, USAGE_CMDNAME_FMT" [-u <user>|-A]", cmd->name);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschvoid doveadm_mail_help(const struct doveadm_mail_cmd *cmd)
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch fprintf(stderr, "doveadm %s [-u <user>|-A] %s\n", cmd->name,
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Bosch cmd->usage_args == NULL ? "" : cmd->usage_args);
5394bed8aaef2a6c1c870a34a23a7824e1f370bbStephan Boschvoid doveadm_mail_try_help_name(const char *cmd_name)
void doveadm_mail_init(void)
&mod_set);
void doveadm_mail_deinit(void)