src/log/doveadm-connection.c

	doveadm-connection.c revision e93184a9055c2530366dfe617e07199603c399dd
/* Copyright (c) 2013-2016 Dovecot authors, see the included COPYING file */

#include "lib.h"
#include "ostream.h"
#include "str.h"
#include "strescape.h"
#include "master-service.h"
#include "log-error-buffer.h"
#include "doveadm-connection.h"

#include <unistd.h>

struct doveadm_connection {
    struct log_error_buffer *errorbuf;

    int fd;
    struct ostream *output;
};

static void doveadm_connection_destroy(struct doveadm_connection **_conn);

static int doveadm_connection_send_errors(struct doveadm_connection *conn)
{
    struct log_error_buffer_iter *iter;
    const struct log_error *error;
    string_t *str = t_str_new(256);
    int ret = 0;

    iter = log_error_buffer_iter_init(conn->errorbuf);
    while ((error = log_error_buffer_iter_next(iter)) != NULL) {
        str_truncate(str, 0);
        str_printfa(str, "%s\t%ld\t",
                failure_log_type_names[error->type],
                (long)error->timestamp);
        str_append_tabescaped(str, error->prefix);
        str_append_c(str, '\t');
        str_append_tabescaped(str, error->text);
        str_append_c(str, '\n');
        if (o_stream_send(conn->output,
                  str_data(str), str_len(str)) < 0) {
            ret = -1;
            break;
        }
    }
    log_error_buffer_iter_deinit(&iter);
    return ret;
}

static int doveadm_output(struct doveadm_connection *conn)
{
    int ret;

    if ((ret = o_stream_flush(conn->output)) != 0) {
        /* error / finished */
        doveadm_connection_destroy(&conn);
    }
    return 1;
}

void doveadm_connection_create(struct log_error_buffer *errorbuf, int fd)
{
    struct doveadm_connection *conn;

    conn = i_new(struct doveadm_connection, 1);
    conn->errorbuf = errorbuf;
    conn->fd = fd;
    conn->output = o_stream_create_fd(conn->fd, (size_t)-1);
    if (doveadm_connection_send_errors(conn) < 0)
        doveadm_connection_destroy(&conn);
    else {
        o_stream_set_flush_callback(conn->output, doveadm_output, conn);
        o_stream_set_flush_pending(conn->output, TRUE);
    }
}

static void doveadm_connection_destroy(struct doveadm_connection **_conn)
{
    struct doveadm_connection *conn = *_conn;

    *_conn = NULL;

    o_stream_destroy(&conn->output);
    if (close(conn->fd) < 0)
        i_error("close(doveadm connection) failed: %m");
    i_free(conn);

    master_service_client_connection_destroyed(master_service);
}