src/lib/ostream.c

	ostream.c revision 904ca86f5d3f27505f50f5342214aabb78629cc8
76b43e4417bab52e913da39b5f5bc2a130d3f149Timo Sirainen/* Copyright (c) 2002-2013 Dovecot authors, see the included COPYING file */
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen#include "lib.h"
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen#include "istream.h"
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen#include "ostream-private.h"
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainenvoid o_stream_set_name(struct ostream *stream, const char *name)
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen{
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    i_free(stream->real_stream->iostream.name);
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    stream->real_stream->iostream.name = i_strdup(name);
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen}
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainenconst char *o_stream_get_name(struct ostream *stream)
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen{
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen    while (stream->real_stream->iostream.name == NULL) {
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen        stream = stream->real_stream->parent;
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen        if (stream == NULL)
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen            return "";
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen    }
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    return stream->real_stream->iostream.name;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen}
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainenint o_stream_get_fd(struct ostream *stream)
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen{
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    return stream->real_stream->fd;
ebfbf5d78dcf95e8b176429f4b5b0694eb4e17d5Timo Sirainen}
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
ebfbf5d78dcf95e8b176429f4b5b0694eb4e17d5Timo Sirainenconst char *o_stream_get_error(struct ostream *stream)
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen{
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    struct ostream *s;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    /* we'll only return errors for streams that have stream_errno set.
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen       we might be returning unintended error otherwise. */
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    if (stream->stream_errno == 0)
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen        return "<no error>";
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    for (s = stream; s != NULL; s = s->real_stream->parent) {
88187ee880b4829443e0d55ea7d145d9d5880217Timo Sirainen        if (s->stream_errno == 0)
1777c974563740daac427d3ef738903d8f6ad7d0Timo Sirainen            break;
1777c974563740daac427d3ef738903d8f6ad7d0Timo Sirainen        if (s->real_stream->iostream.error != NULL)
1777c974563740daac427d3ef738903d8f6ad7d0Timo Sirainen            return s->real_stream->iostream.error;
ebfbf5d78dcf95e8b176429f4b5b0694eb4e17d5Timo Sirainen    }
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    return strerror(stream->stream_errno);
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen}
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainenstatic void o_stream_close_full(struct ostream *stream, bool close_parents)
ebfbf5d78dcf95e8b176429f4b5b0694eb4e17d5Timo Sirainen{
ebfbf5d78dcf95e8b176429f4b5b0694eb4e17d5Timo Sirainen    io_stream_close(&stream->real_stream->iostream, close_parents);
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    stream->closed = TRUE;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    if (stream->stream_errno == 0)
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen        stream->stream_errno = EPIPE;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen}
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainenvoid o_stream_destroy(struct ostream **stream)
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen{
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen    o_stream_close_full(*stream, FALSE);
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    o_stream_unref(stream);
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen}
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainenvoid o_stream_ref(struct ostream *stream)
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen{
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    io_stream_ref(&stream->real_stream->iostream);
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen}
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainenvoid o_stream_unref(struct ostream **_stream)
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen{
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    struct ostream *stream = *_stream;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen    if (stream->real_stream->last_errors_not_checked &&
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen        !stream->real_stream->error_handling_disabled &&
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen        stream->real_stream->iostream.refcount == 1) {
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen        i_panic("output stream %s is missing error handling",
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen            o_stream_get_name(stream));
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    }
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen    io_stream_unref(&stream->real_stream->iostream);
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen    *_stream = NULL;
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen}
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainenvoid o_stream_close(struct ostream *stream)
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen{
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen    o_stream_close_full(stream, TRUE);
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen}
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen#undef o_stream_set_flush_callback
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainenvoid o_stream_set_flush_callback(struct ostream *stream,
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen                 stream_flush_callback_t *callback,
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen                 void *context)
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen{
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen    struct ostream_private *_stream = stream->real_stream;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen    _stream->set_flush_callback(_stream, callback, context);
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen}
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainenvoid o_stream_unset_flush_callback(struct ostream *stream)
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen{
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    struct ostream_private *_stream = stream->real_stream;
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen    _stream->set_flush_callback(_stream, NULL, NULL);
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen}
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainenvoid o_stream_set_max_buffer_size(struct ostream *stream, size_t max_size)
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen{
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen    io_stream_set_max_buffer_size(&stream->real_stream->iostream, max_size);
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen}
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainenvoid o_stream_cork(struct ostream *stream)
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen{
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen    struct ostream_private *_stream = stream->real_stream;
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen    if (unlikely(stream->closed))
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen        return;
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen    _stream->cork(_stream, TRUE);
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen}
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainenvoid o_stream_uncork(struct ostream *stream)
fcde781c3ceb470c8dff34a68df19c69f93bcec9Timo Sirainen{
fcde781c3ceb470c8dff34a68df19c69f93bcec9Timo Sirainen    struct ostream_private *_stream = stream->real_stream;
fcde781c3ceb470c8dff34a68df19c69f93bcec9Timo Sirainen
fcde781c3ceb470c8dff34a68df19c69f93bcec9Timo Sirainen    if (unlikely(stream->closed))
fcde781c3ceb470c8dff34a68df19c69f93bcec9Timo Sirainen        return;
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen    _stream->cork(_stream, FALSE);
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen}
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainenstatic void o_stream_clear_error(struct ostream *stream)
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen{
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen    stream->stream_errno = 0;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    i_free_and_null(stream->real_stream->iostream.error);
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen}
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainenint o_stream_flush(struct ostream *stream)
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen{
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen    struct ostream_private *_stream = stream->real_stream;
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen    int ret = 1;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen    if (unlikely(stream->closed)) {
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen        errno = stream->stream_errno;
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen        return -1;
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen    }
fcde781c3ceb470c8dff34a68df19c69f93bcec9Timo Sirainen
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen    o_stream_clear_error(stream);
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen    if (unlikely((ret = _stream->flush(_stream)) < 0)) {
1caf757864e7734345660e7d190f84e42668a6f8Timo Sirainen        i_assert(stream->stream_errno != 0);
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen        stream->last_failed_errno = stream->stream_errno;
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen        errno = stream->stream_errno;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    }
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    return ret;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen}
43d32cbe60fdaef2699d99f1ca259053e9350411Timo Sirainen
43d32cbe60fdaef2699d99f1ca259053e9350411Timo Sirainenvoid o_stream_set_flush_pending(struct ostream *stream, bool set)
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen{
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    struct ostream_private *_stream = stream->real_stream;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    if (unlikely(stream->closed))
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen        return;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    _stream->flush_pending(_stream, set);
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen}
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainensize_t o_stream_get_buffer_used_size(const struct ostream *stream)
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen{
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen    const struct ostream_private *_stream = stream->real_stream;
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen    return _stream->get_used_size(_stream);
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen}
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainensize_t o_stream_get_buffer_avail_size(const struct ostream *stream)
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen{
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen    size_t used = o_stream_get_buffer_used_size(stream);
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen    return stream->real_stream->max_buffer_size <= used ? 0 :
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen        stream->real_stream->max_buffer_size - used;
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen}
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainenint o_stream_seek(struct ostream *stream, uoff_t offset)
916221f976af0ed8b397f06f4f381c0ac0be3b86Timo Sirainen{
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen    struct ostream_private *_stream = stream->real_stream;
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    if (unlikely(stream->closed)) {
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen        errno = stream->stream_errno;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen        return -1;
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen    }
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
916221f976af0ed8b397f06f4f381c0ac0be3b86Timo Sirainen    o_stream_clear_error(stream);
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    if (unlikely(_stream->seek(_stream, offset) < 0)) {
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen        i_assert(stream->stream_errno != 0);
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen        stream->last_failed_errno = stream->stream_errno;
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen        errno = stream->stream_errno;
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen        return -1;
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen    }
916221f976af0ed8b397f06f4f381c0ac0be3b86Timo Sirainen    return 1;
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen}
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainenssize_t o_stream_send(struct ostream *stream, const void *data, size_t size)
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen{
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    struct const_iovec iov;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen    memset(&iov, 0, sizeof(iov));
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    iov.iov_base = data;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    iov.iov_len = size;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    return o_stream_sendv(stream, &iov, 1);
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen}
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainenssize_t o_stream_sendv(struct ostream *stream, const struct const_iovec *iov,
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen               unsigned int iov_count)
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen{
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    struct ostream_private *_stream = stream->real_stream;
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen    unsigned int i;
e62f6437a4ff01d692a5a61369fe4168d69191edTimo Sirainen    size_t total_size;
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    ssize_t ret;
1b0cfbf3cc77a670b92fff5c30f7b1eb17a63ab1Timo Sirainen
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen    if (unlikely(stream->closed)) {
74ab5ea66c0c4b388f1c774ae6a47ab94f1b4f18Timo Sirainen        errno = stream->stream_errno;
        return -1;
    }

    o_stream_clear_error(stream);
    for (i = 0, total_size = 0; i < iov_count; i++)
        total_size += iov[i].iov_len;
    if (total_size == 0)
        return 0;

    ret = _stream->sendv(_stream, iov, iov_count);
    if (unlikely(ret != (ssize_t)total_size)) {
        if (ret < 0) {
            i_assert(stream->stream_errno != 0);
            stream->last_failed_errno = stream->stream_errno;
            errno = stream->stream_errno;
        } else {
            stream->overflow = TRUE;
        }
    }
    return ret;
}

ssize_t o_stream_send_str(struct ostream *stream, const char *str)
{
    return o_stream_send(stream, str, strlen(str));
}

void o_stream_nsend(struct ostream *stream, const void *data, size_t size)
{
    struct const_iovec iov;

    memset(&iov, 0, sizeof(iov));
    iov.iov_base = data;
    iov.iov_len = size;

    o_stream_nsendv(stream, &iov, 1);
}

void o_stream_nsendv(struct ostream *stream, const struct const_iovec *iov,
             unsigned int iov_count)
{
    if (unlikely(stream->closed))
        return;
    (void)o_stream_sendv(stream, iov, iov_count);
    stream->real_stream->last_errors_not_checked = TRUE;
}

void o_stream_nsend_str(struct ostream *stream, const char *str)
{
    o_stream_nsend(stream, str, strlen(str));
}

void o_stream_nflush(struct ostream *stream)
{
    if (unlikely(stream->closed))
        return;
    (void)o_stream_flush(stream);
    stream->real_stream->last_errors_not_checked = TRUE;
}

int o_stream_nfinish(struct ostream *stream)
{
    o_stream_nflush(stream);
    o_stream_ignore_last_errors(stream);
    errno = stream->last_failed_errno;
    return stream->last_failed_errno != 0 ? -1 : 0;
}

void o_stream_ignore_last_errors(struct ostream *stream)
{
    while (stream != NULL) {
        stream->real_stream->last_errors_not_checked = FALSE;
        stream = stream->real_stream->parent;
    }
}

void o_stream_set_no_error_handling(struct ostream *stream, bool set)
{
    stream->real_stream->error_handling_disabled = set;
}

off_t o_stream_send_istream(struct ostream *outstream,
                struct istream *instream)
{
    struct ostream_private *_outstream = outstream->real_stream;
    off_t ret;

    if (unlikely(outstream->closed || instream->closed)) {
        errno = outstream->stream_errno;
        return -1;
    }

    o_stream_clear_error(outstream);
    ret = _outstream->send_istream(_outstream, instream);
    if (unlikely(ret < 0)) {
        i_assert(outstream->stream_errno != 0);
        outstream->last_failed_errno = outstream->stream_errno;
        errno = outstream->stream_errno;
    }
    return ret;
}

int o_stream_pwrite(struct ostream *stream, const void *data, size_t size,
            uoff_t offset)
{
    int ret;

    if (unlikely(stream->closed)) {
        errno = stream->stream_errno;
        return -1;
    }

    o_stream_clear_error(stream);
    ret = stream->real_stream->write_at(stream->real_stream,
                        data, size, offset);
    if (unlikely(ret < 0)) {
        i_assert(stream->stream_errno != 0);
        stream->last_failed_errno = stream->stream_errno;
        errno = stream->stream_errno;
    }
    return ret;
}

off_t io_stream_copy(struct ostream *outstream, struct istream *instream,
             size_t block_size)
{
    uoff_t start_offset;
    struct const_iovec iov;
    const unsigned char *data;
    ssize_t ret;

    start_offset = instream->v_offset;
    for (;;) {
        (void)i_stream_read_data(instream, &data, &iov.iov_len,
                     block_size-1);
        if (iov.iov_len == 0) {
            /* all sent */
            break;
        }

        iov.iov_base = data;
        ret = o_stream_sendv(outstream, &iov, 1);
        if (ret <= 0) {
            if (ret == 0)
                break;
            return -1;
        }
        i_stream_skip(instream, ret);

        if ((size_t)ret != iov.iov_len)
            break;
    }

    return (off_t)(instream->v_offset - start_offset);
}

void o_stream_switch_ioloop(struct ostream *stream)
{
    struct ostream_private *_stream = stream->real_stream;

    _stream->switch_ioloop(_stream);
}

static void o_stream_default_close(struct iostream_private *stream,
                   bool close_parent)
{
    struct ostream_private *_stream = (struct ostream_private *)stream;

    (void)o_stream_flush(&_stream->ostream);
    if (close_parent && _stream->parent != NULL)
        o_stream_close(_stream->parent);
}

static void o_stream_default_destroy(struct iostream_private *stream)
{
    struct ostream_private *_stream = (struct ostream_private *)stream;

    if (_stream->parent != NULL)
        o_stream_unref(&_stream->parent);
}

static void
o_stream_default_set_max_buffer_size(struct iostream_private *stream,
                     size_t max_size)
{
    struct ostream_private *_stream = (struct ostream_private *)stream;

    if (_stream->parent != NULL)
        o_stream_set_max_buffer_size(_stream->parent, max_size);
    _stream->max_buffer_size = max_size;
}

static void o_stream_default_cork(struct ostream_private *_stream, bool set)
{
    _stream->corked = set;
    if (set) {
        if (_stream->parent != NULL)
            o_stream_cork(_stream->parent);
    } else {
        (void)o_stream_flush(&_stream->ostream);
        if (_stream->parent != NULL)
            o_stream_uncork(_stream->parent);
    }
}

void o_stream_copy_error_from_parent(struct ostream_private *_stream)
{
    struct ostream *src = _stream->parent;
    struct ostream *dest = &_stream->ostream;

    dest->stream_errno = src->stream_errno;
    dest->last_failed_errno = src->last_failed_errno;
    dest->overflow = src->overflow;
}

static int o_stream_default_flush(struct ostream_private *_stream)
{
    int ret;

    if (_stream->parent == NULL)
        return 1;

    if ((ret = o_stream_flush(_stream->parent)) < 0)
        o_stream_copy_error_from_parent(_stream);
    return ret;
}

static void
o_stream_default_set_flush_callback(struct ostream_private *_stream,
                    stream_flush_callback_t *callback,
                    void *context)
{
    if (_stream->parent != NULL)
        o_stream_set_flush_callback(_stream->parent, callback, context);

    _stream->callback = callback;
    _stream->context = context;
}

static void
o_stream_default_set_flush_pending(struct ostream_private *_stream, bool set)
{
    if (_stream->parent != NULL)
        o_stream_set_flush_pending(_stream->parent, set);
}

static size_t
o_stream_default_get_used_size(const struct ostream_private *_stream)
{
    if (_stream->parent == NULL)
        return 0;
    else
        return o_stream_get_buffer_used_size(_stream->parent);
}

static int
o_stream_default_seek(struct ostream_private *_stream,
              uoff_t offset ATTR_UNUSED)
{
    _stream->ostream.stream_errno = ESPIPE;
    return -1;
}

static int
o_stream_default_write_at(struct ostream_private *_stream,
              const void *data ATTR_UNUSED,
              size_t size ATTR_UNUSED, uoff_t offset ATTR_UNUSED)
{
    _stream->ostream.stream_errno = ESPIPE;
    return -1;
}

static off_t o_stream_default_send_istream(struct ostream_private *outstream,
                       struct istream *instream)
{
    return io_stream_copy(&outstream->ostream, instream, IO_BLOCK_SIZE);
}

static void o_stream_default_switch_ioloop(struct ostream_private *_stream)
{
    if (_stream->parent != NULL)
        o_stream_switch_ioloop(_stream->parent);
}

struct ostream *
o_stream_create(struct ostream_private *_stream, struct ostream *parent, int fd)
{
    _stream->fd = fd;
    _stream->ostream.real_stream = _stream;
    if (parent != NULL) {
        _stream->parent = parent;
        o_stream_ref(parent);

        _stream->callback = parent->real_stream->callback;
        _stream->context = parent->real_stream->context;
        _stream->max_buffer_size = parent->real_stream->max_buffer_size;
        _stream->error_handling_disabled =
            parent->real_stream->error_handling_disabled;
    }

    if (_stream->iostream.close == NULL)
        _stream->iostream.close = o_stream_default_close;
    if (_stream->iostream.destroy == NULL)
        _stream->iostream.destroy = o_stream_default_destroy;
    if (_stream->iostream.set_max_buffer_size == NULL) {
        _stream->iostream.set_max_buffer_size =
            o_stream_default_set_max_buffer_size;
    }

    if (_stream->cork == NULL)
        _stream->cork = o_stream_default_cork;
    if (_stream->flush == NULL)
        _stream->flush = o_stream_default_flush;
    if (_stream->set_flush_callback == NULL) {
        _stream->set_flush_callback =
            o_stream_default_set_flush_callback;
    }
    if (_stream->flush_pending == NULL)
        _stream->flush_pending = o_stream_default_set_flush_pending;
    if (_stream->get_used_size == NULL)
        _stream->get_used_size = o_stream_default_get_used_size;
    if (_stream->seek == NULL)
        _stream->seek = o_stream_default_seek;
    if (_stream->write_at == NULL)
        _stream->write_at = o_stream_default_write_at;
    if (_stream->send_istream == NULL)
        _stream->send_istream = o_stream_default_send_istream;
    if (_stream->switch_ioloop == NULL)
        _stream->switch_ioloop = o_stream_default_switch_ioloop;

    io_stream_init(&_stream->iostream);
    return &_stream->ostream;
}

struct ostream *o_stream_create_error(int stream_errno)
{
    struct ostream_private *stream;
    struct ostream *output;

    stream = i_new(struct ostream_private, 1);
    stream->ostream.closed = TRUE;
    stream->ostream.stream_errno = stream_errno;
    stream->ostream.last_failed_errno = stream_errno;

    output = o_stream_create(stream, NULL, -1);
    o_stream_set_name(output, "(error)");
    return output;
}