iostream-temp: Support o_stream_pwrite()

diff --git a/src/lib/iostream-temp.c b/src/lib/iostream-temp.c
index 83681a679fb67d304ace32fb1dd138200b8c7b20..06e26a5cb50efd782933369674e6c36181880808 100644 (file)
--- a/src/lib/iostream-temp.c
+++ b/src/lib/iostream-temp.c
@@ -188,6 +188,24 @@ static off_t o_stream_temp_send_istream(struct ostream_private *_outstream,
                              instream, IO_BLOCK_SIZE);
 }
 
+static int
+o_stream_temp_write_at(struct ostream_private *stream,
+                      const void *data, size_t size, uoff_t offset)
+{
+       struct temp_ostream *tstream = (struct temp_ostream *)stream;
+
+       if (tstream->fd == -1) {
+               i_assert(stream->ostream.offset == tstream->buf->used);
+               buffer_write(tstream->buf, offset, data, size);
+               stream->ostream.offset = tstream->buf->used;
+       } else if (pwrite_full(tstream->fd, data, size, offset) < 0) {
+               stream->ostream.stream_errno = errno;
+               i_close_fd(&tstream->fd);
+               return -1;
+       }
+       return 0;
+}
+
 struct ostream *iostream_temp_create(const char *temp_path_prefix,
                                     enum iostream_temp_flags flags)
 {
@@ -197,6 +215,7 @@ struct ostream *iostream_temp_create(const char *temp_path_prefix,
        tstream = i_new(struct temp_ostream, 1);
        tstream->ostream.sendv = o_stream_temp_sendv;
        tstream->ostream.send_istream = o_stream_temp_send_istream;
+       tstream->ostream.write_at = o_stream_temp_write_at;
        tstream->ostream.iostream.close = o_stream_temp_close;
        tstream->temp_path_prefix = i_strdup(temp_path_prefix);
        tstream->flags = flags;