void iostream_pump_switch_ioloop(struct iostream_pump *pump)
{
i_assert(pump != NULL);
if (pump->io != NULL)
pump->io = io_loop_move_io(&pump->io);
o_stream_switch_ioloop(pump->output);
i_stream_switch_ioloop(pump->input);
}
void connection_switch_ioloop(struct connection *conn)
{
if (conn->io != NULL)
conn->io = io_loop_move_io(&conn->io);
if (conn->to != NULL)
conn->to = io_loop_move_timeout(&conn->to);
if (conn->output != NULL)
o_stream_switch_ioloop(conn->output);
}
static void pop3c_client_ioloop_changed(struct pop3c_client *client)
{
if (client->to != NULL)
client->to = io_loop_move_timeout(&client->to);
if (client->io != NULL)
client->io = io_loop_move_io(&client->io);
if (client->output != NULL)
o_stream_switch_ioloop(client->output);
}
static int doveadm_cmd_handle(struct client_connection *conn,
const char *cmd_name,
const struct mail_storage_service_input *input,
int argc, char *argv[])
{
struct ioloop *ioloop, *prev_ioloop = current_ioloop;
const struct doveadm_cmd *cmd;
const struct doveadm_mail_cmd *mail_cmd;
struct doveadm_mail_cmd_context *ctx;
cmd = doveadm_cmd_find(cmd_name, &argc, &argv);
if (cmd == NULL) {
mail_cmd = doveadm_mail_cmd_find(cmd_name);
if (mail_cmd == NULL) {
i_error("doveadm: Client sent unknown command: %s", cmd_name);
return -1;
}
if (doveadm_mail_cmd_server_parse(mail_cmd, conn->set, input,
argc, argv, &ctx) < 0)
return -1;
}
/* some commands will want to call io_loop_run(), but we're already
running one and we can't call the original one recursively, so
create a new ioloop. */
ioloop = io_loop_create();
lib_signals_reset_ioloop();
if (cmd != NULL)
doveadm_cmd_server_run(conn, cmd, argc, argv);
else
doveadm_mail_cmd_server_run(conn, ctx, input);
io_loop_set_current(prev_ioloop);
lib_signals_reset_ioloop();
o_stream_switch_ioloop(conn->output);
io_loop_set_current(ioloop);
io_loop_destroy(&ioloop);
/* clear all headers */
doveadm_print_deinit();
doveadm_print_init(DOVEADM_PRINT_TYPE_SERVER);
return 0;
}
static void o_stream_ssl_switch_ioloop(struct ostream_private *stream)
{
struct ssl_ostream *sstream = (struct ssl_ostream *)stream;
o_stream_switch_ioloop(sstream->ssl_io->plain_output);
}
