diff options
Diffstat (limited to 'stream')
-rw-r--r-- | stream/stream.c | 10 | ||||
-rw-r--r-- | stream/stream.h | 1 |
2 files changed, 10 insertions, 1 deletions
diff --git a/stream/stream.c b/stream/stream.c index 8361cd8748..3c728eacac 100644 --- a/stream/stream.c +++ b/stream/stream.c @@ -611,11 +611,19 @@ int stream_read(stream_t *s, void *mem, int total) return total; } +// Read ahead so that at least forward_size bytes are readable ahead. Returns +// the actual forward amount available (restricted by EOF or buffer limits). +int stream_peek(stream_t *s, int forward_size) +{ + while (stream_read_more(s, forward_size)) {} + return s->buf_end - s->buf_cur; +} + // Like stream_read(), but do not advance the current position. This may resize // the buffer to satisfy the read request. int stream_read_peek(stream_t *s, void *buf, int buf_size) { - while (stream_read_more(s, buf_size)) {} + stream_peek(s, buf_size); return ring_copy(s, buf, buf_size, s->buf_cur); } diff --git a/stream/stream.h b/stream/stream.h index c127860ceb..2116fdde27 100644 --- a/stream/stream.h +++ b/stream/stream.h @@ -212,6 +212,7 @@ bool stream_seek_skip(stream_t *s, int64_t pos); bool stream_seek(stream_t *s, int64_t pos); int stream_read(stream_t *s, void *mem, int total); int stream_read_partial(stream_t *s, void *buf, int buf_size); +int stream_peek(stream_t *s, int forward_size); int stream_read_peek(stream_t *s, void *buf, int buf_size); void stream_drop_buffers(stream_t *s); int64_t stream_get_size(stream_t *s); |