src/lib/connection.c

	connection.c revision 54c6a8135c0ded324f7ae7d0cdf1ef177004ee2d
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi/* Copyright (c) 2013-2018 Dovecot authors, see the included COPYING file */
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi#include "lib.h"
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi#include "ioloop.h"
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi#include "istream.h"
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi#include "istream-unix.h"
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi#include "ostream.h"
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi#include "ostream-unix.h"
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi#include "iostream.h"
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi#include "net.h"
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi#include "strescape.h"
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi#include "llist.h"
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi#include "time-util.h"
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi#include "connection.h"
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi#include <unistd.h>
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomistatic void connection_idle_timeout(struct connection *conn)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->disconnect_reason = CONNECTION_DISCONNECT_IDLE_TIMEOUT;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->list->v.destroy(conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomistatic void connection_connect_timeout(struct connection *conn)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->disconnect_reason = CONNECTION_DISCONNECT_CONNECT_TIMEOUT;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->list->v.destroy(conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomivoid connection_input_default(struct connection *conn)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    const char *line;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    struct istream *input;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    struct ostream *output;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    int ret = 0;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    switch (connection_input_read(conn)) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    case -1:
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        return;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    case 0: /* allow calling this function for buffered input */
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    case 1:
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        break;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    default:
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        i_unreached();
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    input = conn->input;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    output = conn->output;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_stream_ref(input);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (output != NULL) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        o_stream_ref(output);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        o_stream_cork(output);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    while (!input->closed && (line = i_stream_next_line(input)) != NULL) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        T_BEGIN {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            ret = conn->list->v.input_line(conn, line);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        } T_END;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        if (ret <= 0)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            break;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (output != NULL) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        o_stream_uncork(output);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        o_stream_unref(&output);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (ret < 0 && !input->closed) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        conn->disconnect_reason = CONNECTION_DISCONNECT_DEINIT;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        conn->list->v.destroy(conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_stream_unref(&input);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomiint connection_verify_version(struct connection *conn,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                  const char *const *args)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    unsigned int recv_major_version;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    /* VERSION <tab> service_name <tab> major version <tab> minor version */
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (str_array_length(args) != 4 ||
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        strcmp(args[0], "VERSION") != 0 ||
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        str_to_uint(args[2], &recv_major_version) < 0 ||
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        str_to_uint(args[3], &conn->minor_version) < 0) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        i_error("%s didn't reply with a valid VERSION line: %s",
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            conn->name, t_strarray_join(args, "\t"));
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        return -1;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (strcmp(args[1], conn->list->set.service_name_in) != 0) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        i_error("%s: Connected to wrong socket type. "
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            "We want '%s', but received '%s'", conn->name,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            conn->list->set.service_name_in, args[1]);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        return -1;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (recv_major_version != conn->list->set.major_version) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        i_error("%s: Socket supports major version %u, "
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            "but we support only %u (mixed old and new binaries?)",
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            conn->name, recv_major_version,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            conn->list->set.major_version);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        return -1;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    return 0;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomiint connection_input_line_default(struct connection *conn, const char *line)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    const char *const *args;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    args = t_strsplit_tabescaped(line);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (!conn->version_received) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        if (connection_verify_version(conn, args) < 0)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            return -1;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        conn->version_received = TRUE;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        return 1;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (args[0] == NULL && !conn->list->set.allow_empty_args_input) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        i_error("%s: Unexpectedly received empty line", conn->name);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        return -1;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    return conn->list->v.input_args(conn, args);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomivoid connection_input_halt(struct connection *conn)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    io_remove(&conn->io);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomivoid connection_input_resume(struct connection *conn)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    const struct connection_settings *set = &conn->list->set;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (conn->io != NULL)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        return;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (conn->from_streams || set->input_max_size != 0) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        conn->io = io_add_istream_to(conn->ioloop, conn->input,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                         *conn->list->v.input, conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    } else {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        conn->io = io_add_to(conn->ioloop, conn->fd_in, IO_READ,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                     *conn->list->v.input, conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomistatic void connection_init_streams(struct connection *conn)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    const struct connection_settings *set = &conn->list->set;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(conn->io == NULL);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(conn->input == NULL);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(conn->output == NULL);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(conn->to == NULL);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->version_received = set->major_version == 0;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (set->input_max_size != 0) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        if (conn->unix_socket)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            conn->input = i_stream_create_unix(conn->fd_in,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                               set->input_max_size);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        else
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            conn->input = i_stream_create_fd(conn->fd_in,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                             set->input_max_size);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        i_stream_set_name(conn->input, conn->name);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        i_stream_switch_ioloop_to(conn->input, conn->ioloop);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (set->output_max_size != 0) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        if (conn->unix_socket)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            conn->output = o_stream_create_unix(conn->fd_out,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                                set->output_max_size);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        else
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            conn->output = o_stream_create_fd(conn->fd_out,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                              set->output_max_size);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        o_stream_set_no_error_handling(conn->output, TRUE);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        o_stream_set_finish_via_child(conn->output, FALSE);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        o_stream_set_name(conn->output, conn->name);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        o_stream_switch_ioloop_to(conn->output, conn->ioloop);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    connection_input_resume(conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (set->input_idle_timeout_secs != 0) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        conn->to = timeout_add_to(conn->ioloop,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                      set->input_idle_timeout_secs*1000,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                      connection_idle_timeout, conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (set->major_version != 0 && !set->dont_send_version) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        o_stream_nsend_str(conn->output, t_strdup_printf(
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            "VERSION\t%s\t%u\t%u\n", set->service_name_out,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            set->major_version, set->minor_version));
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomivoid connection_streams_changed(struct connection *conn)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    const struct connection_settings *set = &conn->list->set;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (set->input_max_size != 0 && conn->io != NULL) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        connection_input_halt(conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        connection_input_resume(conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomistatic void connection_client_connected(struct connection *conn, bool success)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(conn->list->set.client);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->connect_finished = ioloop_timeval;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (success)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        connection_init_streams(conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (conn->list->v.client_connected != NULL)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        conn->list->v.client_connected(conn, success);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (!success) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        conn->disconnect_reason =
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            CONNECTION_DISCONNECT_CONN_CLOSED;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        conn->list->v.destroy(conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomivoid connection_init(struct connection_list *list,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi             struct connection *conn)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->ioloop = current_ioloop;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->fd_in = -1;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->fd_out = -1;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->name = NULL;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (conn->list != NULL) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        i_assert(conn->list == list);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    } else {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        conn->list = list;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        DLLIST_PREPEND(&list->connections, conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        list->connections_count++;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomivoid connection_init_server(struct connection_list *list,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                struct connection *conn, const char *name,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                int fd_in, int fd_out)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(name != NULL);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(!list->set.client);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    connection_init(list, conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->name = i_strdup(name);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->fd_in = fd_in;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->fd_out = fd_out;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    connection_init_streams(conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomivoid connection_init_client_ip_from(struct connection_list *list,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                    struct connection *conn,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                    const struct ip_addr *ip, in_port_t port,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                    const struct ip_addr *my_ip)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(list->set.client);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    connection_init(list, conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->fd_in = conn->fd_out = -1;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->name = i_strdup_printf("%s:%u", net_ip2addr(ip), port);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->ip = *ip;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->port = port;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (my_ip != NULL)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        conn->my_ip = *my_ip;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    else
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        i_zero(&conn->my_ip);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomivoid connection_init_client_ip(struct connection_list *list,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                   struct connection *conn,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                   const struct ip_addr *ip, in_port_t port)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    connection_init_client_ip_from(list, conn, ip, port, NULL);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomivoid connection_init_client_unix(struct connection_list *list,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                 struct connection *conn, const char *path)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(list->set.client);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    connection_init(list, conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->fd_in = conn->fd_out = -1;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->name = i_strdup(path);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->unix_socket = TRUE;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomivoid connection_init_from_streams(struct connection_list *list,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                struct connection *conn, const char *name,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                struct istream *input, struct ostream *output)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(name != NULL);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    connection_init(list, conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->name = i_strdup(name);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->from_streams = TRUE;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->fd_in = i_stream_get_fd(input);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->fd_out = o_stream_get_fd(output);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(conn->fd_in >= 0);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(conn->fd_out >= 0);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(conn->io == NULL);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(conn->input == NULL);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(conn->output == NULL);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(conn->to == NULL);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->input = input;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_stream_ref(conn->input);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_stream_set_name(conn->input, conn->name);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->output = output;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    o_stream_ref(conn->output);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    o_stream_set_no_error_handling(conn->output, TRUE);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    o_stream_set_name(conn->output, conn->name);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    connection_input_resume(conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (list->v.client_connected != NULL)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        list->v.client_connected(conn, TRUE);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomistatic void connection_socket_connected(struct connection *conn)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    io_remove(&conn->io);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    timeout_remove(&conn->to);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    errno = net_geterror(conn->fd_in);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    connection_client_connected(conn, errno == 0);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomiint connection_client_connect(struct connection *conn)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    const struct connection_settings *set = &conn->list->set;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    int fd;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(conn->list->set.client);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    i_assert(conn->fd_in == -1);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (conn->port != 0) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        fd = net_connect_ip(&conn->ip, conn->port,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            (conn->my_ip.family != 0 ? &conn->my_ip : NULL));
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    } else if (conn->list->set.unix_client_connect_msecs == 0)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        fd = net_connect_unix(conn->name);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    else
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        fd = net_connect_unix_with_retries(conn->name, conn->list->set.unix_client_connect_msecs);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (fd == -1)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        return -1;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->fd_in = conn->fd_out = fd;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->connect_started = ioloop_timeval;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    if (conn->port != 0 ||
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        conn->list->set.delayed_unix_client_connected_callback) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        conn->io = io_add_to(conn->ioloop, conn->fd_out, IO_WRITE,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                     connection_socket_connected, conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        if (set->client_connect_timeout_msecs != 0) {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi            conn->to = timeout_add_to(conn->ioloop,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                          set->client_connect_timeout_msecs,
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi                          connection_connect_timeout, conn);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    } else {
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi        connection_client_connected(conn, TRUE);
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    }
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    return 0;
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi}
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomivoid connection_disconnect(struct connection *conn)
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi{
1de7b73a11afda43323410a4481d866930b1e632Aki Tuomi    conn->last_input = 0;
    i_zero(&conn->last_input_tv);
    timeout_remove(&conn->to);
    io_remove(&conn->io);
    i_stream_close(conn->input);
    i_stream_destroy(&conn->input);
    o_stream_close(conn->output);
    o_stream_destroy(&conn->output);
    fd_close_maybe_stdio(&conn->fd_in, &conn->fd_out);
}

void connection_deinit(struct connection *conn)
{
    i_assert(conn->list->connections_count > 0);

    conn->list->connections_count--;
    DLLIST_REMOVE(&conn->list->connections, conn);

    connection_disconnect(conn);
    i_free(conn->name);
}

int connection_input_read(struct connection *conn)
{
    conn->last_input = ioloop_time;
    conn->last_input_tv = ioloop_timeval;
    if (conn->to != NULL)
        timeout_reset(conn->to);

    switch (i_stream_read(conn->input)) {
    case -2:
        /* buffer full */
        switch (conn->list->set.input_full_behavior) {
        case CONNECTION_BEHAVIOR_DESTROY:
            conn->disconnect_reason =
                CONNECTION_DISCONNECT_BUFFER_FULL;
            conn->list->v.destroy(conn);
            return -1;
        case CONNECTION_BEHAVIOR_ALLOW:
            return -2;
        }
        i_unreached();
    case -1:
        /* disconnected */
        conn->disconnect_reason =
            CONNECTION_DISCONNECT_CONN_CLOSED;
        conn->list->v.destroy(conn);
        return -1;
    case 0:
        /* nothing new read */
        return 0;
    default:
        /* something was read */
        return 1;
    }
}

const char *connection_disconnect_reason(struct connection *conn)
{
    switch (conn->disconnect_reason) {
    case CONNECTION_DISCONNECT_DEINIT:
        return "Deinitializing";
    case CONNECTION_DISCONNECT_CONNECT_TIMEOUT: {
        unsigned int msecs =
            conn->list->set.client_connect_timeout_msecs;
        return t_strdup_printf("connect() timed out in %u.%03u secs",
                       msecs/1000, msecs%1000);
    }
    case CONNECTION_DISCONNECT_IDLE_TIMEOUT:
        return "Idle timeout";
    case CONNECTION_DISCONNECT_CONN_CLOSED:
        if (conn->input == NULL)
            return t_strdup_printf("connect() failed: %m");
        /* fall through */
    case CONNECTION_DISCONNECT_NOT:
    case CONNECTION_DISCONNECT_BUFFER_FULL:
        return io_stream_get_disconnect_reason(conn->input, conn->output);
    }
    i_unreached();
}

const char *connection_input_timeout_reason(struct connection *conn)
{
    if (conn->last_input_tv.tv_sec != 0) {
        int diff = timeval_diff_msecs(&ioloop_timeval, &conn->last_input_tv);
        return t_strdup_printf("No input for %u.%03u secs",
                       diff/1000, diff%1000);
    } else if (conn->connect_finished.tv_sec != 0) {
        int diff = timeval_diff_msecs(&ioloop_timeval, &conn->connect_finished);
        return t_strdup_printf(
            "No input since connected %u.%03u secs ago",
            diff/1000, diff%1000);
    } else {
        int diff = timeval_diff_msecs(&ioloop_timeval, &conn->connect_started);
        return t_strdup_printf("connect() timed out after %u.%03u secs",
                       diff/1000, diff%1000);
    }
}

void connection_switch_ioloop_to(struct connection *conn,
                 struct ioloop *ioloop)
{
    conn->ioloop = ioloop;
    if (conn->io != NULL)
        conn->io = io_loop_move_io_to(ioloop, &conn->io);
    if (conn->to != NULL)
        conn->to = io_loop_move_timeout_to(ioloop, &conn->to);
    if (conn->input != NULL)
        i_stream_switch_ioloop_to(conn->input, ioloop);
    if (conn->output != NULL)
        o_stream_switch_ioloop_to(conn->output, ioloop);
}

void connection_switch_ioloop(struct connection *conn)
{
    connection_switch_ioloop_to(conn, current_ioloop);
}

struct connection_list *
connection_list_init(const struct connection_settings *set,
             const struct connection_vfuncs *vfuncs)
{
    struct connection_list *list;

    i_assert(vfuncs->input != NULL ||
         set->input_full_behavior != CONNECTION_BEHAVIOR_ALLOW);
    i_assert(set->major_version == 0 ||
         (set->service_name_in != NULL &&
          set->service_name_out != NULL &&
          set->output_max_size != 0));

    list = i_new(struct connection_list, 1);
    list->set = *set;
    list->v = *vfuncs;

    if (list->v.input == NULL)
        list->v.input = connection_input_default;
    if (list->v.input_line == NULL)
        list->v.input_line = connection_input_line_default;

    return list;
}

void connection_list_deinit(struct connection_list **_list)
{
    struct connection_list *list = *_list;
    struct connection *conn;

    *_list = NULL;

    while (list->connections != NULL) {
        conn = list->connections;
        conn->disconnect_reason = CONNECTION_DISCONNECT_DEINIT;
        list->v.destroy(conn);
        i_assert(conn != list->connections);
    }
    i_free(list);
}