doveadm-connection.c revision 16a5712c1a774b7bd3bbf22032b61ccc9398499e
bcb4e51a409d94ae670de96afb8483a4f7855294Stephan Bosch/* Copyright (c) 2010-2017 Dovecot authors, see the included COPYING file */
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "lib.h"
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen#include "ioloop.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "net.h"
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen#include "istream.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "ostream.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "array.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "str.h"
bdd36cfdba3ff66d25570a9ff568d69e1eb543cfTimo Sirainen#include "strescape.h"
7a60e1dc9e93ef3f7c7fe1af6385a0bfa1e31bc3Timo Sirainen#include "llist.h"
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen#include "time-util.h"
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen#include "master-service.h"
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen#include "user-directory.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "mail-host.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "director.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "director-host.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "director-request.h"
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen#include "director-connection.h"
0cb5a9bfbf40b3b323956792aa13d342a459585eTimo Sirainen#include "doveadm-connection.h"
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
0cb5a9bfbf40b3b323956792aa13d342a459585eTimo Sirainen#include <unistd.h>
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen#define DOVEADM_PROTOCOL_VERSION_MAJOR 1
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen#define DOVEADM_HANDSHAKE "VERSION\tdirector-doveadm\t1\t0\n"
a272994d43de80a306a8ed1f2983960d1f3102d0Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen#define MAX_VALID_VHOST_COUNT 1000
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen#define DEFAULT_MAX_MOVING_USERS 100
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen#define DEFAULT_MAX_KICKING_USERS 100
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
d1414c09cf0d58ac983054e2f4e1a1f329272dcfTimo Sirainen#define DOVEADM_CONNECTION_RING_SYNC_TIMEOUT_MSECS (30*1000)
d1414c09cf0d58ac983054e2f4e1a1f329272dcfTimo Sirainen
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainenenum doveadm_director_cmd_ret {
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen DOVEADM_DIRECTOR_CMD_RET_FAIL = -1,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen DOVEADM_DIRECTOR_CMD_RET_UNFINISHED = 0,
d1414c09cf0d58ac983054e2f4e1a1f329272dcfTimo Sirainen DOVEADM_DIRECTOR_CMD_RET_OK = 1,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen DOVEADM_DIRECTOR_CMD_RET_RING_SYNC_OK,
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen};
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainenenum doveadm_director_cmd_flag {
d1414c09cf0d58ac983054e2f4e1a1f329272dcfTimo Sirainen DOVEADM_DIRECTOR_CMD_FLAG_PRE_RING_SYNC = 0x01,
efe78d3ba24fc866af1c79b9223dc0809ba26cadStephan Bosch};
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainentypedef void
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainendoveadm_connection_ring_sync_callback_t(struct doveadm_connection *);
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainenstruct director_reset_cmd {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct director_reset_cmd *prev, *next;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen struct director *dir;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct doveadm_connection *_conn;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen struct timeval start_time;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen struct director_user_iter *iter;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen unsigned int host_start_idx, host_idx, hosts_count;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen unsigned int max_moving_users;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen unsigned int reset_count;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen bool users_killed;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen};
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstruct director_kick_cmd {
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen struct director_kick_cmd *prev, *next;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct doveadm_connection *_conn;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct director *dir;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen char *mask, *field, *value;
d1414c09cf0d58ac983054e2f4e1a1f329272dcfTimo Sirainen bool alt:1;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen};
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstruct doveadm_connection {
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen struct doveadm_connection *prev, *next;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen int fd;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct io *io;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen struct istream *input;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen struct ostream *output;
9261dbf0675204898c6557591c7aa376e23a52b2Timo Sirainen struct director *dir;
9261dbf0675204898c6557591c7aa376e23a52b2Timo Sirainen
9261dbf0675204898c6557591c7aa376e23a52b2Timo Sirainen struct timeout *to_ring_sync_abort;
9261dbf0675204898c6557591c7aa376e23a52b2Timo Sirainen struct director_reset_cmd *reset_cmd;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct director_kick_cmd *kick_cmd;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen doveadm_connection_ring_sync_callback_t *ring_sync_callback;
d1414c09cf0d58ac983054e2f4e1a1f329272dcfTimo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen const char **cmd_pending_args;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen unsigned int cmd_pending_idx;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen bool handshaked:1;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen};
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic struct doveadm_connection *doveadm_connections;
d1414c09cf0d58ac983054e2f4e1a1f329272dcfTimo Sirainenstatic struct doveadm_connection *doveadm_ring_sync_pending_connections;
9261dbf0675204898c6557591c7aa376e23a52b2Timo Sirainenstatic struct director_reset_cmd *reset_cmds = NULL;
9261dbf0675204898c6557591c7aa376e23a52b2Timo Sirainenstatic struct director_kick_cmd *kick_cmds = NULL;
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainenstatic void doveadm_connection_set_io(struct doveadm_connection *conn);
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainenstatic void doveadm_connection_deinit(struct doveadm_connection **_conn);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic void
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainendoveadm_connection_ring_sync_list_move(struct doveadm_connection *conn);
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainenstatic void doveadm_connection_cmd_run_synced(struct doveadm_connection *conn);
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainenstatic enum doveadm_director_cmd_ret
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainendoveadm_cmd_host_list(struct doveadm_connection *conn,
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen const char *const *args ATTR_UNUSED)
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen{
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen struct mail_host *const *hostp;
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen string_t *str = t_str_new(1024);
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen array_foreach(mail_hosts_get(conn->dir->mail_hosts), hostp) {
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen str_printfa(str, "%s\t%u\t%u\t",
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen (*hostp)->ip_str, (*hostp)->vhost_count,
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen (*hostp)->user_count);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen str_append_tabescaped(str, mail_host_get_tag(*hostp));
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen str_printfa(str, "\t%c\t%ld", (*hostp)->down ? 'D' : 'U',
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen (long)(*hostp)->last_updown_change);
e2ae85924b0ef1a7c97e021a3b901b498f599c18Timo Sirainen str_append_c(str, '\n');
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen }
e2ae85924b0ef1a7c97e021a3b901b498f599c18Timo Sirainen str_append_c(str, '\n');
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen o_stream_nsend(conn->output, str_data(str), str_len(str));
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen return DOVEADM_DIRECTOR_CMD_RET_OK;
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen}
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen
e2ae85924b0ef1a7c97e021a3b901b498f599c18Timo Sirainenstatic enum doveadm_director_cmd_ret
0d1b8b6bec79746c5d89d57dd8c1688946bd9237Josef 'Jeff' Sipekdoveadm_cmd_host_list_removed(struct doveadm_connection *conn,
0cb5a9bfbf40b3b323956792aa13d342a459585eTimo Sirainen const char *const *args ATTR_UNUSED)
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen{
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct mail_host_list *orig_hosts_list;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen struct mail_host *const *orig_hosts, *const *cur_hosts;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen unsigned int i, j, orig_hosts_count, cur_hosts_count;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen string_t *str = t_str_new(1024);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen int ret;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen orig_hosts_list = mail_hosts_init(conn->dir->set->director_user_expire,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen NULL);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen (void)mail_hosts_parse_and_add(orig_hosts_list,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen conn->dir->set->director_mail_servers);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen orig_hosts = array_get(mail_hosts_get(orig_hosts_list),
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen &orig_hosts_count);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen cur_hosts = array_get(mail_hosts_get(conn->dir->mail_hosts),
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen &cur_hosts_count);
9261dbf0675204898c6557591c7aa376e23a52b2Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen /* the hosts are sorted by IP */
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen for (i = j = 0; i < orig_hosts_count && j < cur_hosts_count; ) {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen ret = net_ip_cmp(&orig_hosts[i]->ip, &cur_hosts[j]->ip);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (ret == 0)
a75d470c9223a75801418fcdda258885c36317e0Timo Sirainen i++, j++;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen else if (ret > 0)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen j++;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen else {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen str_printfa(str, "%s\n", orig_hosts[i]->ip_str);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen i++;
a75d470c9223a75801418fcdda258885c36317e0Timo Sirainen }
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen }
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen for (; i < orig_hosts_count; i++)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen str_printfa(str, "%s\n", orig_hosts[i]->ip_str);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen str_append_c(str, '\n');
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen o_stream_nsend(conn->output, str_data(str), str_len(str));
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen mail_hosts_deinit(&orig_hosts_list);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen return DOVEADM_DIRECTOR_CMD_RET_OK;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen}
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic void
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainendoveadm_director_host_append_status(const struct director_host *host,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen const char *type, string_t *str)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen time_t last_failed = I_MAX(host->last_network_failure,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen host->last_protocol_failure);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen str_printfa(str, "%s\t%u\t%s\t%"PRIdTIME_T"\t",
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen host->ip_str, host->port, type,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen last_failed);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen}
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic void doveadm_director_append_status(struct director *dir, string_t *str)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (!dir->ring_handshaked)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen str_append(str, "handshaking");
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen else if (dir->ring_synced)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen str_append(str, "synced");
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen else {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen str_printfa(str, "syncing - last sync %d secs ago",
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen (int)(ioloop_time - dir->ring_last_sync_time));
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen }
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen str_printfa(str, "\t%u", dir->last_sync_msecs);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen}
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic void
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainendoveadm_director_connection_append_status(struct director_connection *conn,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen string_t *str)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct director_connection_status status;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (!director_connection_is_handshaked(conn))
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen str_append(str, "handshaking");
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen else if (director_connection_is_synced(conn))
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen str_append(str, "synced");
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen else
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen str_append(str, "syncing");
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen director_connection_get_status(conn, &status);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen str_printfa(str, "\t%u\t%"PRIuUOFF_T"\t%"PRIuUOFF_T"\t%zu\t%zu\t"
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen "%"PRIdTIME_T"\t%"PRIdTIME_T, status.last_ping_msecs,
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen status.bytes_read, status.bytes_sent,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen status.bytes_buffered, status.peak_bytes_buffered,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen status.last_input.tv_sec, status.last_output.tv_sec);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen}
e5acc283bf030b0b5c79ca4e52d315c516a299faPascal Volk
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic enum doveadm_director_cmd_ret
7a60e1dc9e93ef3f7c7fe1af6385a0bfa1e31bc3Timo Sirainendoveadm_cmd_director_list(struct doveadm_connection *conn,
31fda2fc79d48e5186163972b52eb97b4bd8b6fdTimo Sirainen const char *const *args ATTR_UNUSED)
31fda2fc79d48e5186163972b52eb97b4bd8b6fdTimo Sirainen{
31fda2fc79d48e5186163972b52eb97b4bd8b6fdTimo Sirainen struct director *dir = conn->dir;
31fda2fc79d48e5186163972b52eb97b4bd8b6fdTimo Sirainen struct director_host *const *hostp;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen string_t *str = t_str_new(1024);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen const char *type;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct director_connection *const *connp;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen ARRAY(struct director_host *) hosts;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen t_array_init(&hosts, array_count(&dir->dir_hosts));
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen array_append_array(&hosts, &dir->dir_hosts);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen /* show each individual connection */
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen array_foreach(&dir->connections, connp) {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen const struct director_host *host =
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen director_connection_get_host(*connp);
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen /* NOTE: for incoming connections host is initially NULL */
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen array_foreach(&hosts, hostp) {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (*hostp == host) {
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen array_delete(&hosts,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen array_foreach_idx(&hosts, hostp), 1);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen break;
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen }
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen }
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen if (*connp == dir->left)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen type = "left";
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen else if (*connp == dir->right)
b6e1d85292485a7fb4cfa5f40dd1ec131ab07cc1Timo Sirainen type = "right";
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen else if (director_connection_is_incoming(*connp))
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen type = "in";
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen else
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen type = "out";
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen if (host != NULL)
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen doveadm_director_host_append_status(host, type, str);
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen doveadm_director_connection_append_status(*connp, str);
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen str_append_c(str, '\n');
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen }
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen /* show the rest of the hosts that don't have any connections */
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen array_foreach(&hosts, hostp) {
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen const struct director_host *host = *hostp;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (host->removed)
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen type = "removed";
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen else if (dir->self_host == host)
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen type = "self";
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen else
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen type = "";
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen doveadm_director_host_append_status(host, type, str);
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen if (dir->self_host == host)
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen doveadm_director_append_status(dir, str);
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen str_append_c(str, '\n');
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen }
9261dbf0675204898c6557591c7aa376e23a52b2Timo Sirainen str_append_c(str, '\n');
9261dbf0675204898c6557591c7aa376e23a52b2Timo Sirainen o_stream_nsend(conn->output, str_data(str), str_len(str));
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen return DOVEADM_DIRECTOR_CMD_RET_OK;
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen}
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen
7ab22fc4793e174b3455bc0d2afdb128a73efaa5Timo Sirainenstatic enum doveadm_director_cmd_ret
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainendoveadm_cmd_director_add(struct doveadm_connection *conn,
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen const char *const *args)
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen{
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen struct director_host *host;
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen struct ip_addr ip;
b6e1d85292485a7fb4cfa5f40dd1ec131ab07cc1Timo Sirainen in_port_t port = conn->dir->self_port;
b6e1d85292485a7fb4cfa5f40dd1ec131ab07cc1Timo Sirainen
b6e1d85292485a7fb4cfa5f40dd1ec131ab07cc1Timo Sirainen if (args[0] == NULL ||
19e8adccba16ff419f5675b1575358c2956dce83Timo Sirainen net_addr2ip(args[0], &ip) < 0 ||
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen (args[1] != NULL && net_str2port(args[1], &port) < 0)) {
19e8adccba16ff419f5675b1575358c2956dce83Timo Sirainen i_error("doveadm sent invalid DIRECTOR-ADD parameters");
a272994d43de80a306a8ed1f2983960d1f3102d0Timo Sirainen return DOVEADM_DIRECTOR_CMD_RET_FAIL;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen }
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen if (director_host_lookup(conn->dir, &ip, port) == NULL) {
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen host = director_host_add(conn->dir, &ip, port);
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen director_notify_ring_added(host, conn->dir->self_host, TRUE);
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen }
b6e1d85292485a7fb4cfa5f40dd1ec131ab07cc1Timo Sirainen o_stream_nsend(conn->output, "OK\n", 3);
0cb5a9bfbf40b3b323956792aa13d342a459585eTimo Sirainen return DOVEADM_DIRECTOR_CMD_RET_OK;
0cb5a9bfbf40b3b323956792aa13d342a459585eTimo Sirainen}
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic enum doveadm_director_cmd_ret
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainendoveadm_cmd_director_remove(struct doveadm_connection *conn,
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen const char *const *args)
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen{
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen struct director_host *host;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct ip_addr ip;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen in_port_t port = 0;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen if (args[0] == NULL ||
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen net_addr2ip(args[0], &ip) < 0 ||
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen (args[1] != NULL && net_str2port(args[1], &port) < 0)) {
678d0463849ba777106eb7875f27db07a5d8e3dfTimo Sirainen i_error("doveadm sent invalid DIRECTOR-REMOVE parameters");
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen return DOVEADM_DIRECTOR_CMD_RET_FAIL;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen }
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen host = port != 0 ?
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen director_host_lookup(conn->dir, &ip, port) :
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen director_host_lookup_ip(conn->dir, &ip);
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen if (host == NULL) {
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen o_stream_nsend_str(conn->output, "NOTFOUND\n");
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen return DOVEADM_DIRECTOR_CMD_RET_OK;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen } else {
a75d470c9223a75801418fcdda258885c36317e0Timo Sirainen director_ring_remove(host, conn->dir->self_host);
a75d470c9223a75801418fcdda258885c36317e0Timo Sirainen return DOVEADM_DIRECTOR_CMD_RET_RING_SYNC_OK;
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen }
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen}
d5cebe7f98e63d4e2822863ef2faa4971e8b3a5dTimo Sirainen
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainenstatic enum doveadm_director_cmd_ret
742b9a0fc1d5513552fe99aa726497d1250d762eTimo Sirainendoveadm_cmd_host_set_or_update(struct doveadm_connection *conn,
742b9a0fc1d5513552fe99aa726497d1250d762eTimo Sirainen const char *const *args, bool update)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
a75d470c9223a75801418fcdda258885c36317e0Timo Sirainen struct director *dir = conn->dir;
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen const char *ip_str, *tag = "";
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen struct mail_host *host;
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen struct ip_addr ip;
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen unsigned int vhost_count = UINT_MAX;
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen ip_str = args[0];
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen if (ip_str != NULL) {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen tag = strchr(ip_str, '@');
7c849dbc7be089175c1a83a84ee7249ed695810dTimo Sirainen if (tag == NULL)
7c849dbc7be089175c1a83a84ee7249ed695810dTimo Sirainen tag = "";
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen else
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen ip_str = t_strdup_until(ip_str, tag++);
bf05bf725add3686bdb1cb2723208b1562596d17Timo Sirainen }
84de4076ab3a33b34049a46e72ee0456afad8ad4Timo Sirainen if (ip_str == NULL || net_addr2ip(ip_str, &ip) < 0 ||
84de4076ab3a33b34049a46e72ee0456afad8ad4Timo Sirainen (args[1] != NULL && str_to_uint(args[1], &vhost_count) < 0) ||
84de4076ab3a33b34049a46e72ee0456afad8ad4Timo Sirainen (args[1] == NULL && update)) {
84de4076ab3a33b34049a46e72ee0456afad8ad4Timo Sirainen i_error("doveadm sent invalid %s parameters",
84de4076ab3a33b34049a46e72ee0456afad8ad4Timo Sirainen update ? "HOST-UPDATE" : "HOST-SET");
84de4076ab3a33b34049a46e72ee0456afad8ad4Timo Sirainen return DOVEADM_DIRECTOR_CMD_RET_FAIL;
84de4076ab3a33b34049a46e72ee0456afad8ad4Timo Sirainen }
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (vhost_count > MAX_VALID_VHOST_COUNT && vhost_count != UINT_MAX) {
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen o_stream_nsend_str(conn->output, "vhost count too large\n");
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen return DOVEADM_DIRECTOR_CMD_RET_OK;
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen }
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen host = mail_host_lookup(dir->mail_hosts, &ip);
d2b6189eb61fc0e8a37f8814427282304e39e72cAki Tuomi if (host == NULL) {
d2b6189eb61fc0e8a37f8814427282304e39e72cAki Tuomi if (update) {
d2b6189eb61fc0e8a37f8814427282304e39e72cAki Tuomi o_stream_nsend_str(conn->output, "NOTFOUND\n");
991367db5ec720fa0b764bbd6df21b7f7c654d2cTimo Sirainen return DOVEADM_DIRECTOR_CMD_RET_OK;
991367db5ec720fa0b764bbd6df21b7f7c654d2cTimo Sirainen }
991367db5ec720fa0b764bbd6df21b7f7c654d2cTimo Sirainen host = mail_host_add_ip(dir->mail_hosts, &ip, tag);
991367db5ec720fa0b764bbd6df21b7f7c654d2cTimo Sirainen } else if (tag[0] != '\0' && strcmp(mail_host_get_tag(host), tag) != 0) {
991367db5ec720fa0b764bbd6df21b7f7c654d2cTimo Sirainen o_stream_nsend_str(conn->output, "host tag can't be changed\n");
991367db5ec720fa0b764bbd6df21b7f7c654d2cTimo Sirainen return DOVEADM_DIRECTOR_CMD_RET_OK;
991367db5ec720fa0b764bbd6df21b7f7c654d2cTimo Sirainen } else if (host->desynced) {
991367db5ec720fa0b764bbd6df21b7f7c654d2cTimo Sirainen o_stream_nsend_str(conn->output,
0d1b8b6bec79746c5d89d57dd8c1688946bd9237Josef 'Jeff' Sipek "host is already being updated - try again later\n");
5f1d689131a75c39f064cbd4202373e7edf78f18Josef 'Jeff' Sipek return DOVEADM_DIRECTOR_CMD_RET_OK;
e63aa0fe623cee01975d57755381e2b873e5bf93Timo Sirainen }
e63aa0fe623cee01975d57755381e2b873e5bf93Timo Sirainen if (vhost_count != UINT_MAX)
e63aa0fe623cee01975d57755381e2b873e5bf93Timo Sirainen mail_host_set_vhost_count(host, vhost_count, "doveadm: ");
a272994d43de80a306a8ed1f2983960d1f3102d0Timo Sirainen /* NOTE: we don't support changing a tag for an existing host.
e63aa0fe623cee01975d57755381e2b873e5bf93Timo Sirainen it needs to be removed first. otherwise it would be a bit ugly to
e63aa0fe623cee01975d57755381e2b873e5bf93Timo Sirainen handle. */
e63aa0fe623cee01975d57755381e2b873e5bf93Timo Sirainen director_update_host(dir, dir->self_host, NULL, host);
e63aa0fe623cee01975d57755381e2b873e5bf93Timo Sirainen
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen return DOVEADM_DIRECTOR_CMD_RET_RING_SYNC_OK;
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen}
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic enum doveadm_director_cmd_ret
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainendoveadm_cmd_host_set(struct doveadm_connection *conn, const char *const *args)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen return doveadm_cmd_host_set_or_update(conn, args, FALSE);
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen}
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic enum doveadm_director_cmd_ret
a272994d43de80a306a8ed1f2983960d1f3102d0Timo Sirainendoveadm_cmd_host_update(struct doveadm_connection *conn, const char *const *args)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen return doveadm_cmd_host_set_or_update(conn, args, TRUE);
a272994d43de80a306a8ed1f2983960d1f3102d0Timo Sirainen}
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainenstatic enum doveadm_director_cmd_ret
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainendoveadm_cmd_host_updown(struct doveadm_connection *conn, bool down,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen const char *const *args)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
a272994d43de80a306a8ed1f2983960d1f3102d0Timo Sirainen struct mail_host *host;
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen struct ip_addr ip;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (args[0] == NULL || net_addr2ip(args[0], &ip) < 0) {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen i_error("doveadm sent invalid %s parameters: %s",
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen down ? "HOST-DOWN" : "HOST-UP",
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen args[0] == NULL ? "" : args[0]);
a272994d43de80a306a8ed1f2983960d1f3102d0Timo Sirainen return DOVEADM_DIRECTOR_CMD_RET_FAIL;
a272994d43de80a306a8ed1f2983960d1f3102d0Timo Sirainen }
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen host = mail_host_lookup(conn->dir->mail_hosts, &ip);
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen if (host == NULL) {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen o_stream_nsend_str(conn->output, "NOTFOUND\n");
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen return DOVEADM_DIRECTOR_CMD_RET_OK;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen }
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (host->down == down) {
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen o_stream_nsend_str(conn->output, "OK\n");
742b9a0fc1d5513552fe99aa726497d1250d762eTimo Sirainen return DOVEADM_DIRECTOR_CMD_RET_OK;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen } else if (host->desynced) {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen o_stream_nsend_str(conn->output,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen "host is already being updated - try again later\n");
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen return DOVEADM_DIRECTOR_CMD_RET_OK;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen } else {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen mail_host_set_down(host, down, ioloop_time, "doveadm: ");
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen director_update_host(conn->dir, conn->dir->self_host,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen NULL, host);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen return DOVEADM_DIRECTOR_CMD_RET_RING_SYNC_OK;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen }
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen}
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic enum doveadm_director_cmd_ret
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainendoveadm_cmd_host_up(struct doveadm_connection *conn,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen const char *const *args)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
21c1655dbc5fe861a152dc9a8a388d0d64f5ae20Timo Sirainen return doveadm_cmd_host_updown(conn, FALSE, args);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen}
d2b6189eb61fc0e8a37f8814427282304e39e72cAki Tuomi
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic enum doveadm_director_cmd_ret
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainendoveadm_cmd_host_down(struct doveadm_connection *conn,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen const char *const *args)
0d1b8b6bec79746c5d89d57dd8c1688946bd9237Josef 'Jeff' Sipek{
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen return doveadm_cmd_host_updown(conn, TRUE, args);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen}
21c1655dbc5fe861a152dc9a8a388d0d64f5ae20Timo Sirainen
21c1655dbc5fe861a152dc9a8a388d0d64f5ae20Timo Sirainenstatic enum doveadm_director_cmd_ret
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainendoveadm_cmd_host_remove(struct doveadm_connection *conn,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen const char *const *args)
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen{
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct mail_host *host;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct ip_addr ip;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (args[0] == NULL || net_addr2ip(args[0], &ip) < 0) {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen i_error("doveadm sent invalid HOST-REMOVE parameters");
d1414c09cf0d58ac983054e2f4e1a1f329272dcfTimo Sirainen return DOVEADM_DIRECTOR_CMD_RET_FAIL;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen }
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen host = mail_host_lookup(conn->dir->mail_hosts, &ip);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (host == NULL) {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen o_stream_nsend_str(conn->output, "NOTFOUND\n");
e93184a9055c2530366dfe617e07199603c399ddMartti Rannanjärvi return DOVEADM_DIRECTOR_CMD_RET_OK;
e93184a9055c2530366dfe617e07199603c399ddMartti Rannanjärvi } else {
d2b6189eb61fc0e8a37f8814427282304e39e72cAki Tuomi director_remove_host(conn->dir, conn->dir->self_host,
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen NULL, host);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen return DOVEADM_DIRECTOR_CMD_RET_RING_SYNC_OK;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen }
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen}
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic void
0f3d4fbcf88e2ffd674893aed8cc1288fe17d290Timo Sirainendoveadm_cmd_host_flush_all(struct doveadm_connection *conn)
0f3d4fbcf88e2ffd674893aed8cc1288fe17d290Timo Sirainen{
9ddd3d7d8651985e373a6c48e0ddc76b8a4ef1c7Timo Sirainen struct mail_host *const *hostp;
0f3d4fbcf88e2ffd674893aed8cc1288fe17d290Timo Sirainen unsigned int total_user_count = 0;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
f540662dd02e930cf704c1468cccd05d512bc663Timo Sirainen array_foreach(mail_hosts_get(conn->dir->mail_hosts), hostp) {
f540662dd02e930cf704c1468cccd05d512bc663Timo Sirainen total_user_count += (*hostp)->user_count;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen director_flush_host(conn->dir, conn->dir->self_host,
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen NULL, *hostp);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen }
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen i_warning("Flushed all backend hosts with %u users. This is an unsafe "
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen "operation and may cause the same users to end up in multiple backends.",
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen total_user_count);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen o_stream_nsend(conn->output, "OK\n", 3);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen}
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainenstatic enum doveadm_director_cmd_ret
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainendoveadm_cmd_host_flush(struct doveadm_connection *conn, const char *const *args)
05a4fe649671ac5ca6b3beb085c73e6e85bc4f8bAki Tuomi{
05a4fe649671ac5ca6b3beb085c73e6e85bc4f8bAki Tuomi struct mail_host *host;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen struct ip_addr ip;
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (args[0] == NULL || args[0][0] == '\0') {
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen doveadm_cmd_host_flush_all(conn);
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen return DOVEADM_DIRECTOR_CMD_RET_OK;
fd05c47210cad1f6d8effbda5cba7f7d938ca9a0Timo Sirainen }
a75d470c9223a75801418fcdda258885c36317e0Timo Sirainen
9137c55411aa39d41c1e705ddc34d5bd26c65021Timo Sirainen if (net_addr2ip(args[0], &ip) < 0) {
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen i_error("doveadm sent invalid HOST-FLUSH parameters");
a9b135760aea6d1790d447d351c56b78889dac22Aki Tuomi return DOVEADM_DIRECTOR_CMD_RET_FAIL;
a9b135760aea6d1790d447d351c56b78889dac22Aki Tuomi }
a9b135760aea6d1790d447d351c56b78889dac22Aki Tuomi host = mail_host_lookup(conn->dir->mail_hosts, &ip);
a9b135760aea6d1790d447d351c56b78889dac22Aki Tuomi if (host == NULL) {
a9b135760aea6d1790d447d351c56b78889dac22Aki Tuomi o_stream_nsend_str(conn->output, "NOTFOUND\n");
a9b135760aea6d1790d447d351c56b78889dac22Aki Tuomi return DOVEADM_DIRECTOR_CMD_RET_OK;
a9b135760aea6d1790d447d351c56b78889dac22Aki Tuomi } else {
director_flush_host(conn->dir, conn->dir->self_host,
NULL, host);
return DOVEADM_DIRECTOR_CMD_RET_RING_SYNC_OK;
}
}
static void doveadm_reset_cmd_free(struct director_reset_cmd *cmd)
{
DLLIST_REMOVE(&reset_cmds, cmd);
if (cmd->iter != NULL)
director_iterate_users_deinit(&cmd->iter);
if (cmd->_conn != NULL)
cmd->_conn->reset_cmd = NULL;
i_free(cmd);
}
static bool
director_host_reset_users(struct director_reset_cmd *cmd,
struct mail_host *host)
{
struct director *dir = cmd->dir;
struct user *user;
struct mail_host *new_host;
if (dir->users_moving_count >= cmd->max_moving_users)
return FALSE;
if (dir->right != NULL)
director_connection_cork(dir->right);
if (cmd->iter == NULL) {
cmd->iter = director_iterate_users_init(dir);
cmd->users_killed = FALSE;
}
while ((user = director_iterate_users_next(cmd->iter)) != NULL) {
if (user->host != host)
continue;
new_host = mail_host_get_by_hash(dir->mail_hosts,
user->username_hash,
mail_host_get_tag(host));
if (new_host != host) T_BEGIN {
if (new_host != NULL) {
director_move_user(dir, dir->self_host, NULL,
user->username_hash, new_host);
} else {
/* there are no more available backends.
kick the user instead. */
director_kill_user(dir, dir->self_host, user,
user->host->tag, user->host,
TRUE);
cmd->users_killed = TRUE;
}
cmd->reset_count++;
} T_END;
if (dir->users_moving_count >= cmd->max_moving_users)
break;
}
if (user == NULL) {
int msecs = timeval_diff_msecs(&ioloop_timeval, &cmd->start_time);
i_info("Moved %u users in %u hosts in %u.%03u secs (max parallel=%u)",
cmd->reset_count, cmd->hosts_count - cmd->host_start_idx,
msecs / 1000, msecs % 1000, cmd->max_moving_users);
director_iterate_users_deinit(&cmd->iter);
if (cmd->users_killed) {
/* no more backends. we already sent kills. now remove
the users entirely from the host. */
director_flush_host(dir, dir->self_host, NULL, host);
}
}
if (dir->right != NULL)
director_connection_uncork(dir->right);
return user == NULL;
}
static bool
director_reset_cmd_run(struct director_reset_cmd *cmd)
{
struct mail_host *const *hosts;
unsigned int count;
hosts = array_get(mail_hosts_get(cmd->dir->mail_hosts), &count);
if (count > cmd->hosts_count)
count = cmd->hosts_count;
while (cmd->host_idx < count) {
if (!director_host_reset_users(cmd, hosts[cmd->host_idx]))
return FALSE;
cmd->host_idx++;
}
if (cmd->_conn != NULL) {
struct doveadm_connection *conn = cmd->_conn;
o_stream_nsend(conn->output, "OK\n", 3);
if (conn->io == NULL)
doveadm_connection_set_io(conn);
}
doveadm_reset_cmd_free(cmd);
return TRUE;
}
static enum doveadm_director_cmd_ret
doveadm_cmd_host_reset_users(struct doveadm_connection *conn,
const char *const *args)
{
struct director_reset_cmd *cmd;
struct ip_addr ip;
struct mail_host *const *hosts;
unsigned int i = 0, count;
unsigned int max_moving_users = DEFAULT_MAX_MOVING_USERS;
if (args[0] != NULL && args[1] != NULL &&
(str_to_uint(args[1], &max_moving_users) < 0 ||
max_moving_users == 0)) {
i_error("doveadm sent invalid HOST-RESET-USERS parameters");
return DOVEADM_DIRECTOR_CMD_RET_FAIL;
}
hosts = array_get(mail_hosts_get(conn->dir->mail_hosts), &count);
if (args[0] != NULL && args[0][0] != '\0') {
if (net_addr2ip(args[0], &ip) < 0) {
i_error("doveadm sent invalid HOST-RESET-USERS ip: %s",
args[0]);
return DOVEADM_DIRECTOR_CMD_RET_FAIL;
}
for (i = 0; i < count; i++) {
if (net_ip_compare(&hosts[i]->ip, &ip))
break;
}
if (i == count) {
o_stream_nsend_str(conn->output, "NOTFOUND\n");
return DOVEADM_DIRECTOR_CMD_RET_OK;
}
count = i+1;
}
conn->reset_cmd = cmd = i_new(struct director_reset_cmd, 1);
cmd->dir = conn->dir;
cmd->_conn = conn;
cmd->max_moving_users = max_moving_users;
cmd->host_start_idx = i;
cmd->host_idx = i;
cmd->hosts_count = count;
cmd->start_time = ioloop_timeval;
DLLIST_PREPEND(&reset_cmds, cmd);
if (!director_reset_cmd_run(cmd)) {
/* we still have work to do. don't handle any more doveadm
input until we're finished. */
io_remove(&conn->io);
return DOVEADM_DIRECTOR_CMD_RET_UNFINISHED;
}
return DOVEADM_DIRECTOR_CMD_RET_OK;
}
static enum doveadm_director_cmd_ret
doveadm_cmd_user_lookup(struct doveadm_connection *conn,
const char *const *args)
{
struct user *user;
struct mail_host *host;
const char *username, *tag;
unsigned int username_hash;
struct mail_tag *mail_tag;
string_t *str = t_str_new(256);
if (args[0] == NULL) {
username = "";
tag = "";
} else {
username = args[0];
tag = args[1] != NULL ? args[1] : "";
}
if (str_to_uint(username, &username_hash) < 0) {
if (!director_get_username_hash(conn->dir,
username, &username_hash)) {
o_stream_nsend_str(conn->output, "TRYAGAIN\n");
return DOVEADM_DIRECTOR_CMD_RET_OK;
}
}
/* get user's current host */
mail_tag = mail_tag_find(conn->dir->mail_hosts, tag);
user = mail_tag == NULL ? NULL :
user_directory_lookup(mail_tag->users, username_hash);
if (user == NULL)
str_append(str, "\t0");
else {
str_printfa(str, "%s\t%u", user->host->ip_str,
user->timestamp +
conn->dir->set->director_user_expire);
}
/* get host if it wasn't in user directory */
host = mail_host_get_by_hash(conn->dir->mail_hosts, username_hash, tag);
if (host == NULL)
str_append(str, "\t");
else
str_printfa(str, "\t%s", host->ip_str);
/* get host with default configuration */
host = mail_host_get_by_hash(conn->dir->orig_config_hosts,
username_hash, tag);
if (host == NULL)
str_append(str, "\t\n");
else
str_printfa(str, "\t%s\n", host->ip_str);
o_stream_nsend(conn->output, str_data(str), str_len(str));
return DOVEADM_DIRECTOR_CMD_RET_OK;
}
static enum doveadm_director_cmd_ret
doveadm_cmd_user_list(struct doveadm_connection *conn, const char *const *args)
{
struct director_user_iter *iter;
struct user *user;
struct ip_addr ip;
if (args[0] != NULL && args[0][0] != '\0') {
if (net_addr2ip(args[0], &ip) < 0) {
i_error("doveadm sent invalid USER-LIST parameters");
return DOVEADM_DIRECTOR_CMD_RET_FAIL;
}
} else {
ip.family = 0;
}
iter = director_iterate_users_init(conn->dir);
while ((user = director_iterate_users_next(iter)) != NULL) {
if (ip.family == 0 ||
net_ip_compare(&ip, &user->host->ip)) T_BEGIN {
unsigned int expire_time = user->timestamp +
conn->dir->set->director_user_expire;
o_stream_nsend_str(conn->output, t_strdup_printf(
"%u\t%u\t%s\n",
user->username_hash, expire_time,
user->host->ip_str));
} T_END;
}
director_iterate_users_deinit(&iter);
o_stream_nsend(conn->output, "\n", 1);
return DOVEADM_DIRECTOR_CMD_RET_OK;
}
static enum doveadm_director_cmd_ret
doveadm_cmd_user_move(struct doveadm_connection *conn, const char *const *args)
{
unsigned int username_hash;
struct user *user;
struct mail_host *host;
struct ip_addr ip;
if (args[0] == NULL || args[1] == NULL ||
net_addr2ip(args[1], &ip) < 0) {
i_error("doveadm sent invalid USER-MOVE parameters");
return DOVEADM_DIRECTOR_CMD_RET_FAIL;
}
host = mail_host_lookup(conn->dir->mail_hosts, &ip);
if (host == NULL) {
o_stream_nsend_str(conn->output, "NOTFOUND\n");
return DOVEADM_DIRECTOR_CMD_RET_OK;
}
if (str_to_uint(args[0], &username_hash) < 0) {
if (!director_get_username_hash(conn->dir,
args[0], &username_hash)) {
o_stream_nsend_str(conn->output, "TRYAGAIN\n");
return DOVEADM_DIRECTOR_CMD_RET_OK;
}
}
user = user_directory_lookup(host->tag->users, username_hash);
if (user != NULL && USER_IS_BEING_KILLED(user)) {
o_stream_nsend_str(conn->output, "TRYAGAIN\n");
return DOVEADM_DIRECTOR_CMD_RET_OK;
}
if (user == NULL || user->host != host) {
director_move_user(conn->dir, conn->dir->self_host, NULL,
username_hash, host);
} else {
/* already the correct host. reset the user's timeout. */
user_directory_refresh(host->tag->users, user);
director_update_user(conn->dir, conn->dir->self_host, user);
}
o_stream_nsend(conn->output, "OK\n", 3);
return DOVEADM_DIRECTOR_CMD_RET_OK;
}
static void doveadm_kick_cmd_free(struct director_kick_cmd **_cmd)
{
struct director_kick_cmd *cmd = *_cmd;
*_cmd = NULL;
if (cmd->_conn != NULL)
cmd->_conn->kick_cmd = NULL;
i_free(cmd->field);
i_free(cmd->value);
i_free(cmd->mask);
i_free(cmd);
}
static bool doveadm_cmd_user_kick_run(struct director_kick_cmd *cmd)
{
if (cmd->dir->users_kicking_count >= DEFAULT_MAX_KICKING_USERS)
return FALSE;
if (cmd->alt)
director_kick_user_alt(cmd->dir, cmd->dir->self_host,
NULL, cmd->field, cmd->value);
else
director_kick_user(cmd->dir, cmd->dir->self_host,
NULL, cmd->mask);
if (cmd->_conn != NULL) {
struct doveadm_connection *conn = cmd->_conn;
o_stream_nsend(conn->output, "OK\n", 3);
if (conn->io == NULL)
doveadm_connection_set_io(conn);
}
DLLIST_REMOVE(&kick_cmds, cmd);
doveadm_kick_cmd_free(&cmd);
return TRUE;
}
static enum doveadm_director_cmd_ret
doveadm_cmd_user_kick(struct doveadm_connection *conn, const char *const *args)
{
struct director_kick_cmd *cmd;
bool wait = TRUE;
if (args[0] == NULL) {
i_error("doveadm sent invalid USER-KICK parameters");
return DOVEADM_DIRECTOR_CMD_RET_FAIL;
}
if (null_strcmp(args[1], "nowait") == 0)
wait = FALSE;
cmd = conn->kick_cmd = i_new(struct director_kick_cmd, 1);
cmd->alt = FALSE;
cmd->mask = i_strdup(args[0]);
cmd->dir = conn->dir;
cmd->_conn = conn;
DLLIST_PREPEND(&kick_cmds, cmd);
if (!doveadm_cmd_user_kick_run(cmd)) {
if (wait) {
/* we have work to do, wait until it finishes */
io_remove(&conn->io);
return DOVEADM_DIRECTOR_CMD_RET_UNFINISHED;
} else {
o_stream_nsend_str(conn->output, "TRYAGAIN\n");
/* need to remove it here */
DLLIST_REMOVE(&kick_cmds, cmd);
doveadm_kick_cmd_free(&cmd);
}
}
return DOVEADM_DIRECTOR_CMD_RET_OK;
}
static enum doveadm_director_cmd_ret
doveadm_cmd_user_kick_alt(struct doveadm_connection *conn, const char *const *args)
{
bool wait = TRUE;
struct director_kick_cmd *cmd;
if (str_array_length(args) < 2) {
i_error("doveadm sent invalid USER-KICK-ALT parameters");
return DOVEADM_DIRECTOR_CMD_RET_FAIL;
}
if (null_strcmp(args[2], "nowait") == 0)
wait = FALSE;
conn->kick_cmd = cmd = i_new(struct director_kick_cmd, 1);
cmd->alt = TRUE;
cmd->field = i_strdup(args[0]);
cmd->value = i_strdup(args[1]);
cmd->dir = conn->dir;
cmd->_conn = conn;
DLLIST_PREPEND(&kick_cmds, cmd);
if (!doveadm_cmd_user_kick_run(cmd)) {
if (wait) {
/* we have work to do, wait until it finishes */
io_remove(&conn->io);
return DOVEADM_DIRECTOR_CMD_RET_UNFINISHED;
} else {
o_stream_nsend_str(conn->output, "TRYAGAIN\n");
DLLIST_REMOVE(&kick_cmds, cmd);
doveadm_kick_cmd_free(&cmd);
}
}
return DOVEADM_DIRECTOR_CMD_RET_OK;
}
struct {
const char *name;
enum doveadm_director_cmd_ret (*cmd)
(struct doveadm_connection *conn, const char *const *args);
enum doveadm_director_cmd_flag flags;
} doveadm_director_commands[] = {
{ "HOST-LIST", doveadm_cmd_host_list, 0 },
{ "HOST-LIST-REMOVED", doveadm_cmd_host_list_removed, 0 },
{ "DIRECTOR-LIST", doveadm_cmd_director_list, 0 },
{ "DIRECTOR-ADD", doveadm_cmd_director_add, DOVEADM_DIRECTOR_CMD_FLAG_PRE_RING_SYNC },
{ "DIRECTOR-REMOVE", doveadm_cmd_director_remove, DOVEADM_DIRECTOR_CMD_FLAG_PRE_RING_SYNC },
{ "HOST-SET", doveadm_cmd_host_set, DOVEADM_DIRECTOR_CMD_FLAG_PRE_RING_SYNC },
{ "HOST-UPDATE", doveadm_cmd_host_update, DOVEADM_DIRECTOR_CMD_FLAG_PRE_RING_SYNC },
{ "HOST-UP", doveadm_cmd_host_up, DOVEADM_DIRECTOR_CMD_FLAG_PRE_RING_SYNC },
{ "HOST-DOWN", doveadm_cmd_host_down, DOVEADM_DIRECTOR_CMD_FLAG_PRE_RING_SYNC },
{ "HOST-REMOVE", doveadm_cmd_host_remove, DOVEADM_DIRECTOR_CMD_FLAG_PRE_RING_SYNC },
{ "HOST-FLUSH", doveadm_cmd_host_flush, DOVEADM_DIRECTOR_CMD_FLAG_PRE_RING_SYNC },
{ "HOST-RESET-USERS", doveadm_cmd_host_reset_users, 0 },
{ "USER-LOOKUP", doveadm_cmd_user_lookup, 0 },
{ "USER-LIST", doveadm_cmd_user_list, 0 },
{ "USER-MOVE", doveadm_cmd_user_move, 0 },
{ "USER-KICK", doveadm_cmd_user_kick, 0 },
{ "USER-KICK-ALT", doveadm_cmd_user_kick_alt, 0 },
};
static void
doveadm_connection_ring_sync_timeout(struct doveadm_connection *conn)
{
doveadm_connection_ring_sync_list_move(conn);
o_stream_nsend_str(conn->output, "Ring sync timed out\n");
doveadm_connection_set_io(conn);
io_set_pending(conn->io);
i_free_and_null(conn->cmd_pending_args);
}
static void
doveadm_connection_set_ring_sync_callback(struct doveadm_connection *conn,
doveadm_connection_ring_sync_callback_t *callback)
{
i_assert(conn->ring_sync_callback == NULL);
i_assert(conn->to_ring_sync_abort == NULL);
conn->ring_sync_callback = callback;
io_remove(&conn->io);
DLLIST_REMOVE(&doveadm_connections, conn);
DLLIST_PREPEND(&doveadm_ring_sync_pending_connections, conn);
conn->to_ring_sync_abort =
timeout_add(DOVEADM_CONNECTION_RING_SYNC_TIMEOUT_MSECS,
doveadm_connection_ring_sync_timeout, conn);
}
static void doveadm_connection_ret_ok(struct doveadm_connection *conn)
{
o_stream_nsend(conn->output, "OK\n", 3);
}
static enum doveadm_director_cmd_ret
doveadm_connection_cmd_run(struct doveadm_connection *conn,
const char *const *args, unsigned int i)
{
enum doveadm_director_cmd_ret ret;
if ((doveadm_director_commands[i].flags &
DOVEADM_DIRECTOR_CMD_FLAG_PRE_RING_SYNC) != 0 &&
!conn->dir->ring_synced) {
/* wait for ring to be synced before running the command */
conn->cmd_pending_args = p_strarray_dup(default_pool, args);
conn->cmd_pending_idx = i;
doveadm_connection_set_ring_sync_callback(conn,
doveadm_connection_cmd_run_synced);
return DOVEADM_DIRECTOR_CMD_RET_UNFINISHED;
}
ret = doveadm_director_commands[i].cmd(conn, args);
if (ret != DOVEADM_DIRECTOR_CMD_RET_RING_SYNC_OK)
return ret;
/* Delay sending OK until ring is synced. This way doveadm will know
whether the call actually succeeded or not. */
if (conn->dir->ring_synced) {
/* director is alone */
i_assert(conn->dir->right == NULL && conn->dir->left == NULL);
o_stream_nsend(conn->output, "OK\n", 3);
return DOVEADM_DIRECTOR_CMD_RET_OK;
}
doveadm_connection_set_ring_sync_callback(conn, doveadm_connection_ret_ok);
return DOVEADM_DIRECTOR_CMD_RET_RING_SYNC_OK;
}
static void doveadm_connection_cmd_run_synced(struct doveadm_connection *conn)
{
const char **args = conn->cmd_pending_args;
conn->cmd_pending_args = NULL;
(void)doveadm_connection_cmd_run(conn, args, conn->cmd_pending_idx);
i_free(args);
}
static enum doveadm_director_cmd_ret
doveadm_connection_cmd(struct doveadm_connection *conn, const char *line)
{
const char *cmd, *const *args;
args = t_strsplit_tabescaped(line);
if (args[0] == NULL) {
i_error("doveadm sent empty command line");
return DOVEADM_DIRECTOR_CMD_RET_FAIL;
}
cmd = args[0];
args++;
for (unsigned int i = 0; i < N_ELEMENTS(doveadm_director_commands); i++) {
if (strcmp(doveadm_director_commands[i].name, cmd) == 0)
return doveadm_connection_cmd_run(conn, args, i);
}
i_error("doveadm sent unknown command: %s", line);
return DOVEADM_DIRECTOR_CMD_RET_FAIL;
}
static void doveadm_connection_input(struct doveadm_connection *conn)
{
const char *line;
enum doveadm_director_cmd_ret ret = DOVEADM_DIRECTOR_CMD_RET_OK;
if (!conn->handshaked) {
if ((line = i_stream_read_next_line(conn->input)) == NULL) {
if (conn->input->eof || conn->input->stream_errno != 0)
doveadm_connection_deinit(&conn);
return;
}
if (!version_string_verify(line, "director-doveadm",
DOVEADM_PROTOCOL_VERSION_MAJOR)) {
i_error("doveadm not compatible with this server "
"(mixed old and new binaries?)");
doveadm_connection_deinit(&conn);
return;
}
conn->handshaked = TRUE;
}
while ((line = i_stream_read_next_line(conn->input)) != NULL &&
ret == DOVEADM_DIRECTOR_CMD_RET_OK) {
T_BEGIN {
ret = doveadm_connection_cmd(conn, line);
} T_END;
}
if (conn->input->eof || conn->input->stream_errno != 0 ||
ret == DOVEADM_DIRECTOR_CMD_RET_FAIL)
doveadm_connection_deinit(&conn);
}
static void doveadm_connection_set_io(struct doveadm_connection *conn)
{
conn->io = io_add(conn->fd, IO_READ, doveadm_connection_input, conn);
}
struct doveadm_connection *
doveadm_connection_init(struct director *dir, int fd)
{
struct doveadm_connection *conn;
conn = i_new(struct doveadm_connection, 1);
conn->fd = fd;
conn->dir = dir;
conn->input = i_stream_create_fd(conn->fd, 1024);
conn->output = o_stream_create_fd(conn->fd, (size_t)-1);
o_stream_set_no_error_handling(conn->output, TRUE);
doveadm_connection_set_io(conn);
o_stream_nsend_str(conn->output, DOVEADM_HANDSHAKE);
DLLIST_PREPEND(&doveadm_connections, conn);
return conn;
}
static void doveadm_connection_deinit(struct doveadm_connection **_conn)
{
struct doveadm_connection *conn = *_conn;
*_conn = NULL;
i_assert(conn->to_ring_sync_abort == NULL);
if (conn->reset_cmd != NULL) {
/* finish the move even if doveadm disconnected */
conn->reset_cmd->_conn = NULL;
}
if (conn->kick_cmd != NULL) {
/* finish the kick even if doveadm disconnected */
conn->kick_cmd->_conn = NULL;
}
DLLIST_REMOVE(&doveadm_connections, conn);
io_remove(&conn->io);
i_stream_unref(&conn->input);
o_stream_unref(&conn->output);
if (close(conn->fd) < 0)
i_error("close(doveadm connection) failed: %m");
i_free(conn);
master_service_client_connection_destroyed(master_service);
}
static void
doveadm_connection_ring_sync_list_move(struct doveadm_connection *conn)
{
timeout_remove(&conn->to_ring_sync_abort);
DLLIST_REMOVE(&doveadm_ring_sync_pending_connections, conn);
DLLIST_PREPEND(&doveadm_connections, conn);
}
void doveadm_connections_deinit(void)
{
while (reset_cmds != NULL)
doveadm_reset_cmd_free(reset_cmds);
unsigned int pending_count = 0;
while (doveadm_ring_sync_pending_connections != NULL) {
doveadm_connection_ring_sync_list_move(doveadm_ring_sync_pending_connections);
pending_count++;
}
if (pending_count > 0)
i_warning("Shutting down while %u doveadm connections were waiting for ring sync", pending_count);
while (doveadm_connections != NULL) {
struct doveadm_connection *conn = doveadm_connections;
doveadm_connection_deinit(&conn);
}
}
void doveadm_connections_kick_callback(struct director *dir ATTR_UNUSED)
{
while(kick_cmds != NULL)
if (!doveadm_cmd_user_kick_run(kick_cmds))
break;
}
static void doveadm_connections_continue_reset_cmds(void)
{
while (reset_cmds != NULL) {
if (!director_reset_cmd_run(reset_cmds))
break;
}
}
void doveadm_connections_ring_synced(void)
{
while (doveadm_ring_sync_pending_connections != NULL) {
struct doveadm_connection *conn =
doveadm_ring_sync_pending_connections;
doveadm_connection_ring_sync_callback_t *callback =
conn->ring_sync_callback;
conn->ring_sync_callback = NULL;
doveadm_connection_ring_sync_list_move(conn);
doveadm_connection_set_io(conn);
io_set_pending(conn->io);
callback(conn);
}
doveadm_connections_continue_reset_cmds();
}