Added o_stream_get_buffer_avail_size().

--HG--
branch : HEAD

diff --git a/src/lib/ostream.c b/src/lib/ostream.c
index bffda76843672323fbbbe6b794ef96f10ac2f570..fe92a7b9ce625b107109d78c8690582de3dd3628 100644 (file)
--- a/src/lib/ostream.c
+++ b/src/lib/ostream.c
@@ -115,6 +115,14 @@ size_t o_stream_get_buffer_used_size(const struct ostream *stream)
                _stream->get_used_size(_stream);
 }
 
+size_t o_stream_get_buffer_avail_size(const struct ostream *stream)
+{
+       size_t used = o_stream_get_buffer_used_size(stream);
+
+       return stream->real_stream->max_buffer_size <= used ? 0 :
+               stream->real_stream->max_buffer_size - used;
+}
+
 int o_stream_seek(struct ostream *stream, uoff_t offset)
 {
        struct ostream_private *_stream = stream->real_stream;

diff --git a/src/lib/ostream.h b/src/lib/ostream.h
index 92b18c87c6619bc22f467f1176f9a9c49b98e578..d28e8a0c2d77a5533916456b8c0b889a5f94595c 100644 (file)
--- a/src/lib/ostream.h
+++ b/src/lib/ostream.h
@@ -72,6 +72,8 @@ int o_stream_flush(struct ostream *stream);
 void o_stream_set_flush_pending(struct ostream *stream, bool set);
 /* Returns number of bytes currently in buffer. */
 size_t o_stream_get_buffer_used_size(const struct ostream *stream) ATTR_PURE;
+/* Returns number of bytes we can still write without failing. */
+size_t o_stream_get_buffer_avail_size(const struct ostream *stream) ATTR_PURE;
 
 /* Seek to specified position from beginning of file. This works only for
    files. Returns 1 if successful, -1 if error. */