lib: Added ostream.blocking boolean

Similar to istream.blocking.

src/lib/ostream-buffer.c

@@ -59,6 +59,7 @@ struct ostream *o_stream_create_buffer(buffer_t *buf)
 	struct ostream *output;
 
 	bstream = i_new(struct buffer_ostream, 1);
+	bstream->ostream.ostream.blocking = TRUE;
 	bstream->ostream.max_buffer_size = (size_t)-1;
 	bstream->ostream.seek = o_stream_buffer_seek;
 	bstream->ostream.sendv = o_stream_buffer_sendv;

src/lib/ostream-file.c

@@ -1023,6 +1023,7 @@ o_stream_create_fd_file(int fd, uoff_t offset, bool autoclose_fd)
 	fstream_init_file(fstream);
 	fstream->real_offset = offset;
 	fstream->buffer_offset = offset;
+	ostream->blocking = fstream->file;
 	ostream->offset = offset;
 	return ostream;
 }

src/lib/ostream-null.c

@@ -23,6 +23,7 @@ struct ostream *o_stream_create_null(void)
 	struct ostream *output;
 
 	stream = i_new(struct ostream_private, 1);
+	stream->ostream.blocking = TRUE;
 	stream->sendv = o_stream_null_sendv;
 
 	output = o_stream_create(stream, NULL, -1);

src/lib/ostream.c

@@ -262,6 +262,7 @@ ssize_t o_stream_sendv(struct ostream *stream, const struct const_iovec *iov,
 			stream->last_failed_errno = stream->stream_errno;
 			errno = stream->stream_errno;
 		} else {
+			i_assert(!stream->blocking);
 			stream->overflow = TRUE;
 		}
 	}
@@ -570,6 +571,7 @@ o_stream_create(struct ostream_private *_stream, struct ostream *parent, int fd)
 	_stream->fd = fd;
 	_stream->ostream.real_stream = _stream;
 	if (parent != NULL) {
+		_stream->ostream.blocking = parent->blocking;
 		_stream->parent = parent;
 		o_stream_ref(parent);
 
@@ -622,6 +624,7 @@ struct ostream *o_stream_create_error(int stream_errno)
 	struct ostream *output;
 
 	stream = i_new(struct ostream_private, 1);
+	stream->ostream.blocking = TRUE;
 	stream->ostream.closed = TRUE;
 	stream->ostream.stream_errno = stream_errno;
 	stream->ostream.last_failed_errno = stream_errno;

src/lib/ostream.h

@@ -16,6 +16,8 @@ struct ostream {
 	   functions was neither sent nor buffered. It's never unset inside
 	   ostream code. */
 	unsigned int overflow:1;
+	/* o_stream_send() writes all the data or returns failure */
+	unsigned int blocking:1;
 	unsigned int closed:1;
 
 	struct ostream_private *real_stream;