ostream-buffer.c revision b7eccb91e43ad9d973861dbe85e33837dc28f14e
bcb4e51a409d94ae670de96afb8483a4f7855294Stephan Bosch/* Copyright (c) 2002-2012 Dovecot authors, see the included COPYING file */
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch#include "lib.h"
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch#include "buffer.h"
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch#include "ostream-private.h"
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch
8620dc793885749c37000f11dd83b902b6844e20Stephan Boschstruct buffer_ostream {
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch struct ostream_private ostream;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch buffer_t *buf;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch};
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch
8620dc793885749c37000f11dd83b902b6844e20Stephan Boschstatic int o_stream_buffer_seek(struct ostream_private *stream, uoff_t offset)
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch{
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch stream->ostream.offset = offset;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch return 1;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch}
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch
8620dc793885749c37000f11dd83b902b6844e20Stephan Boschstatic int
8620dc793885749c37000f11dd83b902b6844e20Stephan Boscho_stream_buffer_write_at(struct ostream_private *stream,
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch const void *data, size_t size, uoff_t offset)
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch{
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch struct buffer_ostream *bstream = (struct buffer_ostream *)stream;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch buffer_write(bstream->buf, offset, data, size);
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch return 0;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch}
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch
8620dc793885749c37000f11dd83b902b6844e20Stephan Boschstatic ssize_t
8620dc793885749c37000f11dd83b902b6844e20Stephan Boscho_stream_buffer_sendv(struct ostream_private *stream,
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch const struct const_iovec *iov, unsigned int iov_count)
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch{
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch struct buffer_ostream *bstream = (struct buffer_ostream *)stream;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch size_t left, n;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch ssize_t ret = 0;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch unsigned int i;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch for (i = 0; i < iov_count; i++) {
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch left = bstream->ostream.max_buffer_size -
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch stream->ostream.offset;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch n = I_MIN(left, iov[i].iov_len);
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch buffer_write(bstream->buf, stream->ostream.offset,
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch iov[i].iov_base, n);
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch stream->ostream.offset += n;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch ret += n;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch if (n != iov[i].iov_len)
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch break;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch }
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch return ret;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch}
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch
8620dc793885749c37000f11dd83b902b6844e20Stephan Boschstruct ostream *o_stream_create_buffer(buffer_t *buf)
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch{
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch struct buffer_ostream *bstream;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch struct ostream *output;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch bstream = i_new(struct buffer_ostream, 1);
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch bstream->ostream.max_buffer_size = (size_t)-1;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch bstream->ostream.seek = o_stream_buffer_seek;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch bstream->ostream.sendv = o_stream_buffer_sendv;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch bstream->ostream.write_at = o_stream_buffer_write_at;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch bstream->buf = buf;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch output = o_stream_create(&bstream->ostream, NULL, -1);
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch o_stream_set_name(output, "(buffer)");
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch return output;
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch}
8620dc793885749c37000f11dd83b902b6844e20Stephan Bosch