src/login-common/client-common-auth.c

	client-common-auth.c revision e2a88d59c0d47d63ce1ad5b1fd95e487124a3fd4
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina/* Copyright (c) 2002-2012 Dovecot authors, see the included COPYING file */
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina#include "hostpid.h"
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina#include "login-common.h"
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina#include "istream.h"
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina#include "ostream.h"
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina#include "str.h"
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina#include "safe-memset.h"
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina#include "time-util.h"
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina#include "login-proxy.h"
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina#include "auth-client.h"
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina#include "client-common.h"
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina#include <stdlib.h>
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina#define PROXY_FAILURE_MSG "Account is temporarily unavailable."
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina/* If we've been waiting auth server to respond for over this many milliseconds,
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina   send a "waiting" message. */
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina#define AUTH_WAITING_TIMEOUT_MSECS (30*1000)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina#define AUTH_WAITING_WARNING_TIMEOUT_MSECS (10*1000)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březinastatic void client_auth_failed(struct client *client)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina{
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina    i_free_and_null(client->master_data_prefix);
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina    if (client->auth_response != NULL)
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        str_truncate(client->auth_response, 0);
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina    if (client->auth_initializing || client->destroyed)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina        return;
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina    if (client->io != NULL)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina        io_remove(&client->io);
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina    client->io = io_add(client->fd, IO_READ, client_input, client);
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina    client_input(client);
968e2bc3bb28458b25c3b19868ea50f0325d8b0aPavel Březina}
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březinastatic void client_auth_waiting_timeout(struct client *client)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina{
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina    if (!client->notified_auth_ready) {
968e2bc3bb28458b25c3b19868ea50f0325d8b0aPavel Březina        client_log_warn(client, "Auth process not responding, "
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina                "delayed sending initial response (greeting)");
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina    }
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina    client_notify_status(client, FALSE, client->master_tag == 0 ?
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina                 AUTH_SERVER_WAITING_MSG : AUTH_MASTER_WAITING_MSG);
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina    timeout_remove(&client->to_auth_waiting);
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina}
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březinavoid client_set_auth_waiting(struct client *client)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina{
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina    i_assert(client->to_auth_waiting == NULL);
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina    client->to_auth_waiting =
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        timeout_add(!client->notified_auth_ready ?
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina                AUTH_WAITING_WARNING_TIMEOUT_MSECS :
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina                AUTH_WAITING_TIMEOUT_MSECS,
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina                client_auth_waiting_timeout, client);
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina}
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březinastatic void client_auth_parse_args(struct client *client,
e6092759dc4268b105b79478aa9b1398f21ed7a9Pavel Březina                   const char *const *args,
e6092759dc4268b105b79478aa9b1398f21ed7a9Pavel Březina                   struct client_auth_reply *reply_r)
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina{
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina    const char *key, *value, *p;
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina    memset(reply_r, 0, sizeof(*reply_r));
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina    for (; *args != NULL; args++) {
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        p = strchr(*args, '=');
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        if (p == NULL) {
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina            key = *args;
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina            value = "";
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        } else {
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina            key = t_strdup_until(*args, p);
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina            value = p + 1;
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        }
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        if (strcmp(key, "nologin") == 0)
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina            reply_r->nologin = TRUE;
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        else if (strcmp(key, "proxy") == 0)
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina            reply_r->proxy = TRUE;
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        else if (strcmp(key, "temp") == 0)
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina            reply_r->temp = TRUE;
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        else if (strcmp(key, "authz") == 0)
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina            reply_r->authz_failure = TRUE;
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        else if (strcmp(key, "user_disabled") == 0)
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina            client->auth_user_disabled = TRUE;
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        else if (strcmp(key, "pass_expired") == 0)
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina            client->auth_pass_expired = TRUE;
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        else if (strcmp(key, "reason") == 0)
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina            reply_r->reason = value;
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        else if (strcmp(key, "host") == 0)
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina            reply_r->host = value;
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        else if (strcmp(key, "hostip") == 0)
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina            reply_r->hostip = value;
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        else if (strcmp(key, "port") == 0)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina            reply_r->port = atoi(value);
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina        else if (strcmp(key, "destuser") == 0)
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina            reply_r->destuser = value;
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina        else if (strcmp(key, "pass") == 0)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina            reply_r->password = value;
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina        else if (strcmp(key, "proxy_timeout") == 0)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina            reply_r->proxy_timeout_msecs = 1000*atoi(value);
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina        else if (strcmp(key, "proxy_refresh") == 0)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina            reply_r->proxy_refresh_secs = atoi(value);
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina        else if (strcmp(key, "master") == 0)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina            reply_r->master_user = value;
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina        else if (strcmp(key, "ssl") == 0) {
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina            reply_r->ssl_flags |= PROXY_SSL_FLAG_YES;
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina            if (strcmp(value, "any-cert") == 0)
46d3d2c731e8c7e138462e5b60a39a279dc77d81Pavel Březina                reply_r->ssl_flags |= PROXY_SSL_FLAG_ANY_CERT;
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina            if (reply_r->port == 0)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina                reply_r->port = login_binary->default_ssl_port;
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina        } else if (strcmp(key, "starttls") == 0) {
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina            reply_r->ssl_flags |= PROXY_SSL_FLAG_YES |
968e2bc3bb28458b25c3b19868ea50f0325d8b0aPavel Březina                PROXY_SSL_FLAG_STARTTLS;
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina            if (strcmp(value, "any-cert") == 0)
968e2bc3bb28458b25c3b19868ea50f0325d8b0aPavel Březina                reply_r->ssl_flags |= PROXY_SSL_FLAG_ANY_CERT;
968e2bc3bb28458b25c3b19868ea50f0325d8b0aPavel Březina        } else if (strcmp(key, "user") == 0) {
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina            /* already handled in login-common */
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina        } else if (client->set->auth_debug)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina            i_debug("Ignoring unknown passdb extra field: %s", key);
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina    }
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina    if (reply_r->port == 0)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina        reply_r->port = login_binary->default_port;
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina    if (reply_r->destuser == NULL)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina        reply_r->destuser = client->virtual_user;
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina}
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březinastatic void proxy_free_password(struct client *client)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina{
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina    if (client->proxy_password == NULL)
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina        return;
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina    safe_memset(client->proxy_password, 0, strlen(client->proxy_password));
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina    i_free_and_null(client->proxy_password);
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina}
f8c829e72968b574e1c9bda96f4d5f206622358fPavel Březina
void client_proxy_finish_destroy_client(struct client *client)
{
    string_t *str = t_str_new(128);

    if (client->input->closed) {
        /* input stream got closed in client_send_raw_data().
           In most places we don't have to check for this explicitly,
           but login_proxy_detach() attempts to get and use the
           istream's fd, which is now -1. */
        client_destroy(client, "Disconnected");
        return;
    }

    str_printfa(str, "proxy(%s): started proxying to %s:%u",
            client->virtual_user,
            login_proxy_get_host(client->login_proxy),
            login_proxy_get_port(client->login_proxy));
    if (strcmp(client->virtual_user, client->proxy_user) != 0) {
        /* remote username is different, log it */
        str_append_c(str, '/');
        str_append(str, client->proxy_user);
    }
    if (client->proxy_master_user != NULL)
        str_printfa(str, " (master %s)", client->proxy_master_user);

    login_proxy_detach(client->login_proxy);
    client_destroy_success(client, str_c(str));
}

static void client_proxy_error(struct client *client, const char *text)
{
    client->v.proxy_error(client, text);
}

void client_proxy_log_failure(struct client *client, const char *line)
{
    string_t *str = t_str_new(128);

    str_printfa(str, "proxy(%s): Login failed to %s:%u",
            client->virtual_user,
            login_proxy_get_host(client->login_proxy),
            login_proxy_get_port(client->login_proxy));
    if (strcmp(client->virtual_user, client->proxy_user) != 0) {
        /* remote username is different, log it */
        str_append_c(str, '/');
        str_append(str, client->proxy_user);
    }
    if (client->proxy_master_user != NULL)
        str_printfa(str, " (master %s)", client->proxy_master_user);
    str_append(str, ": ");
    str_append(str, line);
    i_info("%s", str_c(str));
}

void client_proxy_failed(struct client *client, bool send_line)
{
    if (send_line) {
        client_proxy_error(client, PROXY_FAILURE_MSG);
    }

    login_proxy_free(&client->login_proxy);
    proxy_free_password(client);
    i_free_and_null(client->proxy_user);
    i_free_and_null(client->proxy_master_user);

    /* call this last - it may destroy the client */
    client_auth_failed(client);
}

static const char *get_disconnect_reason(struct istream *input)
{
    errno = input->stream_errno;
    return errno == 0 || errno == EPIPE ? "Connection closed" :
        t_strdup_printf("Connection closed: %m");
}

static void proxy_input(struct client *client)
{
    struct istream *input;
    const char *line;
    unsigned int duration;

    if (client->login_proxy == NULL) {
        /* we're just freeing the proxy */
        return;
    }

    input = login_proxy_get_istream(client->login_proxy);
    if (input == NULL) {
        if (client->destroyed) {
            /* we came here from client_destroy() */
            return;
        }

        /* failed for some reason, probably server disconnected */
        client_proxy_failed(client, TRUE);
        return;
    }

    i_assert(!client->destroyed);

    switch (i_stream_read(input)) {
    case -2:
        client_log_err(client, "proxy: Remote input buffer full");
        client_proxy_failed(client, TRUE);
        return;
    case -1:
        line = i_stream_next_line(input);
        duration = ioloop_time - client->created;
        client_log_err(client, t_strdup_printf(
            "proxy: Remote %s:%u disconnected: %s "
            "(state=%u, duration=%us)%s",
            login_proxy_get_host(client->login_proxy),
            login_proxy_get_port(client->login_proxy),
            get_disconnect_reason(input),
            client->proxy_state, duration,
            line == NULL ? "" : t_strdup_printf(
                " - BUG: line not read: %s", line)));
        client_proxy_failed(client, TRUE);
        return;
    }

    while ((line = i_stream_next_line(input)) != NULL) {
        if (client->v.proxy_parse_line(client, line) != 0)
            break;
    }
}

static int proxy_start(struct client *client,
               const struct client_auth_reply *reply)
{
    struct login_proxy_settings proxy_set;

    i_assert(reply->destuser != NULL);
    i_assert(!client->destroyed);

    client->v.proxy_reset(client);

    if (reply->password == NULL) {
        client_log_err(client, "proxy: password not given");
        client_proxy_error(client, PROXY_FAILURE_MSG);
        return -1;
    }
    if (reply->host == NULL || *reply->host == '\0') {
        client_log_err(client, "proxy: host not given");
        client_proxy_error(client, PROXY_FAILURE_MSG);
        return -1;
    }

    i_assert(client->refcount > 1);

    if (client->destroyed) {
        /* connection_queue_add() decided that we were the oldest
           connection and killed us. */
        return -1;
    }
    if (login_proxy_is_ourself(client, reply->host, reply->port,
                   reply->destuser)) {
        client_log_err(client, "Proxying loops to itself");
        client_proxy_error(client, PROXY_FAILURE_MSG);
        return -1;
    }

    memset(&proxy_set, 0, sizeof(proxy_set));
    proxy_set.host = reply->host;
    if (reply->hostip != NULL &&
        net_addr2ip(reply->hostip, &proxy_set.ip) < 0)
        proxy_set.ip.family = 0;
    proxy_set.port = reply->port;
    proxy_set.connect_timeout_msecs = reply->proxy_timeout_msecs;
    proxy_set.notify_refresh_secs = reply->proxy_refresh_secs;
    proxy_set.ssl_flags = reply->ssl_flags;

    if (login_proxy_new(client, &proxy_set, proxy_input) < 0) {
        client_proxy_error(client, PROXY_FAILURE_MSG);
        return -1;
    }

    client->proxy_user = i_strdup(reply->destuser);
    client->proxy_master_user = i_strdup(reply->master_user);
    client->proxy_password = i_strdup(reply->password);

    /* disable input until authentication is finished */
    if (client->io != NULL)
        io_remove(&client->io);
    return 0;
}

static void ATTR_NULL(3, 4)
client_auth_result(struct client *client, enum client_auth_result result,
           const struct client_auth_reply *reply, const char *text)
{
    client->v.auth_result(client, result, reply, text);
}

static bool
client_auth_handle_reply(struct client *client,
             const struct client_auth_reply *reply, bool success)
{
    if (reply->proxy) {
        /* we want to proxy the connection to another server.
           don't do this unless authentication succeeded. with
           master user proxying we can get FAIL with proxy still set.

           proxy host=.. [port=..] [destuser=..] pass=.. */
        if (!success)
            return FALSE;
        if (proxy_start(client, reply) < 0)
            client_auth_failed(client);
        return TRUE;
    }

    if (reply->host != NULL) {
        const char *reason;

        if (reply->reason != NULL)
            reason = reply->reason;
        else if (reply->nologin)
            reason = "Try this server instead.";
        else
            reason = "Logged in, but you should use this server instead.";

        if (reply->nologin) {
            client_auth_result(client,
                CLIENT_AUTH_RESULT_REFERRAL_NOLOGIN,
                reply, reason);
        } else {
            client_auth_result(client,
                CLIENT_AUTH_RESULT_REFERRAL_SUCCESS,
                reply, reason);
            return TRUE;
        }
    } else if (reply->nologin) {
        /* Authentication went ok, but for some reason user isn't
           allowed to log in. Shouldn't probably happen. */
        if (reply->reason != NULL) {
            client_auth_result(client,
                CLIENT_AUTH_RESULT_AUTHFAILED_REASON,
                reply, reply->reason);
        } else if (reply->temp) {
            const char *timestamp, *msg;

            timestamp = t_strflocaltime("%Y-%m-%d %H:%M:%S", ioloop_time);
            msg = t_strdup_printf(AUTH_TEMP_FAILED_MSG" [%s:%s]",
                      my_hostname, timestamp);
            client_auth_result(client, CLIENT_AUTH_RESULT_TEMPFAIL,
                       reply, msg);
        } else if (reply->authz_failure) {
            client_auth_result(client,
                CLIENT_AUTH_RESULT_AUTHZFAILED, reply,
                "Authorization failed");
        } else {
            client_auth_result(client,
                CLIENT_AUTH_RESULT_AUTHFAILED, reply,
                AUTH_FAILED_MSG);
        }
    } else {
        /* normal login/failure */
        return FALSE;
    }

    i_assert(reply->nologin);

    if (!client->destroyed)
        client_auth_failed(client);
    return TRUE;
}

void client_auth_respond(struct client *client, const char *response)
{
    client->auth_waiting = FALSE;
    client_set_auth_waiting(client);
    auth_client_request_continue(client->auth_request, response);
    io_remove(&client->io);
}

void client_auth_abort(struct client *client)
{
    sasl_server_auth_abort(client);
}

void client_auth_fail(struct client *client, const char *text)
{
    sasl_server_auth_failed(client, text);
}

int client_auth_read_line(struct client *client)
{
    const unsigned char *data;
    size_t i, size;
    unsigned int len;

    if (i_stream_read_data(client->input, &data, &size, 0) == -1) {
        client_destroy(client, "Disconnected");
        return -1;
    }

    /* see if we have a full line */
    for (i = 0; i < size; i++) {
        if (data[i] == '\n')
            break;
    }
    if (client->auth_response == NULL)
        client->auth_response = str_new(default_pool, I_MAX(i+1, 256));
    if (str_len(client->auth_response) + i > LOGIN_MAX_AUTH_BUF_SIZE) {
        client_destroy(client, "Authentication response too large");
        return -1;
    }
    str_append_n(client->auth_response, data, i);
    i_stream_skip(client->input, i == size ? size : i+1);

    /* drop trailing \r */
    len = str_len(client->auth_response);
    if (len > 0 && str_c(client->auth_response)[len-1] == '\r')
        str_truncate(client->auth_response, len-1);

    return i < size;
}

void client_auth_parse_response(struct client *client)
{
    if (client_auth_read_line(client) <= 0)
        return;

    if (strcmp(str_c(client->auth_response), "*") == 0) {
        sasl_server_auth_abort(client);
        return;
    }

    client_auth_respond(client, str_c(client->auth_response));
    memset(str_c_modifiable(client->auth_response), 0,
           str_len(client->auth_response));
}

static void client_auth_input(struct client *client)
{
    client->v.auth_parse_response(client);
}

void client_auth_send_challenge(struct client *client, const char *data)
{
    struct const_iovec iov[3];

    iov[0].iov_base = "+ ";
    iov[0].iov_len = 2;
    iov[1].iov_base = data;
    iov[1].iov_len = strlen(data);
    iov[2].iov_base = "\r\n";
    iov[2].iov_len = 2;

    o_stream_nsendv(client->output, iov, 3);
}

static void
sasl_callback(struct client *client, enum sasl_server_reply sasl_reply,
          const char *data, const char *const *args)
{
    struct client_auth_reply reply;

    i_assert(!client->destroyed ||
         sasl_reply == SASL_SERVER_REPLY_AUTH_ABORTED ||
         sasl_reply == SASL_SERVER_REPLY_MASTER_FAILED);

    switch (sasl_reply) {
    case SASL_SERVER_REPLY_SUCCESS:
        if (client->to_auth_waiting != NULL)
            timeout_remove(&client->to_auth_waiting);
        if (args != NULL) {
            client_auth_parse_args(client, args, &reply);
            if (client_auth_handle_reply(client, &reply, TRUE))
                break;
        }
        client_auth_result(client, CLIENT_AUTH_RESULT_SUCCESS,
                   NULL, NULL);
        client_destroy_success(client, "Login");
        break;
    case SASL_SERVER_REPLY_AUTH_FAILED:
    case SASL_SERVER_REPLY_AUTH_ABORTED:
        if (client->to_auth_waiting != NULL)
            timeout_remove(&client->to_auth_waiting);
        if (args != NULL) {
            client_auth_parse_args(client, args, &reply);
            reply.nologin = TRUE;
            if (client_auth_handle_reply(client, &reply, FALSE))
                break;
        }

        if (sasl_reply == SASL_SERVER_REPLY_AUTH_ABORTED) {
            client_auth_result(client, CLIENT_AUTH_RESULT_ABORTED,
                NULL, "Authentication aborted by client.");
        } else if (data == NULL) {
            client_auth_result(client,
                CLIENT_AUTH_RESULT_AUTHFAILED, NULL,
                AUTH_FAILED_MSG);
        } else {
            client_auth_result(client,
                CLIENT_AUTH_RESULT_AUTHFAILED_REASON, NULL,
                AUTH_FAILED_MSG);
        }

        if (!client->destroyed)
            client_auth_failed(client);
        break;
    case SASL_SERVER_REPLY_MASTER_FAILED:
        if (data != NULL) {
            /* authentication itself succeeded, we just hit some
               internal failure. */
            client_auth_result(client, CLIENT_AUTH_RESULT_TEMPFAIL,
                       NULL, data);
        }

        /* the fd may still be hanging somewhere in kernel or another
           process. make sure the client gets disconnected. */
        if (shutdown(client->fd, SHUT_RDWR) < 0 && errno != ENOTCONN)
            i_error("shutdown() failed: %m");

        if (data == NULL)
            client_destroy_internal_failure(client);
        else
            client_destroy_success(client, data);
        break;
    case SASL_SERVER_REPLY_CONTINUE:
        client->v.auth_send_challenge(client, data);

        if (client->to_auth_waiting != NULL)
            timeout_remove(&client->to_auth_waiting);

        if (client->auth_response != NULL)
            str_truncate(client->auth_response, 0);

        i_assert(client->io == NULL);
        client->auth_waiting = TRUE;
        client->io = io_add(client->fd, IO_READ,
                    client_auth_input, client);
        client_auth_input(client);
        return;
    }

    client_unref(&client);
}

int client_auth_begin(struct client *client, const char *mech_name,
              const char *init_resp)
{
    if (!client->secured && strcmp(client->set->ssl, "required") == 0) {
        if (client->set->auth_verbose) {
            client_log(client, "Login failed: "
                   "SSL required for authentication");
        }
        client->auth_attempts++;
        client_auth_result(client, CLIENT_AUTH_RESULT_SSL_REQUIRED, NULL,
            "Authentication not allowed until SSL/TLS is enabled.");
        return 1;
    }


    client_ref(client);
    client->auth_initializing = TRUE;
    sasl_server_auth_begin(client, login_binary->protocol, mech_name,
                   init_resp, sasl_callback);
    client->auth_initializing = FALSE;
    if (!client->authenticating)
        return 1;

    /* don't handle input until we get the initial auth reply */
    if (client->io != NULL)
        io_remove(&client->io);
    client_set_auth_waiting(client);
    return 0;
}

bool client_check_plaintext_auth(struct client *client, bool pass_sent)
{
    if (client->secured || !client->set->disable_plaintext_auth)
        return TRUE;

    if (client->set->auth_verbose) {
        client_log(client, "Login failed: "
               "Plaintext authentication disabled");
    }
    if (pass_sent) {
        client_notify_status(client, TRUE,
             "Plaintext authentication not allowed "
             "without SSL/TLS, but your client did it anyway. "
             "If anyone was listening, the password was exposed.");
    }
    client_auth_result(client, CLIENT_AUTH_RESULT_SSL_REQUIRED, NULL,
               AUTH_PLAINTEXT_DISABLED_MSG);
    client->auth_tried_disabled_plaintext = TRUE;
    client->auth_attempts++;
    return FALSE;
}

void clients_notify_auth_connected(void)
{
    struct client *client, *next;

    for (client = clients; client != NULL; client = next) {
        next = client->next;

        if (client->to_auth_waiting != NULL)
            timeout_remove(&client->to_auth_waiting);

        client_notify_auth_ready(client);

        if (client->input_blocked) {
            client->input_blocked = FALSE;
            client_input(client);
        }
    }
}