aboutsummaryrefslogtreecommitdiff
path: root/include/io
diff options
context:
space:
mode:
Diffstat (limited to 'include/io')
-rw-r--r--include/io/istream.h63
1 files changed, 63 insertions, 0 deletions
diff --git a/include/io/istream.h b/include/io/istream.h
index 9a8257f..06e8a3b 100644
--- a/include/io/istream.h
+++ b/include/io/istream.h
@@ -116,6 +116,69 @@ SQFS_INLINE const char *istream_get_filename(istream_t *strm)
}
/**
+ * @brief Peek into the data buffered in an istream
+ *
+ * @memberof istream_t
+ *
+ * If the internal buffer is empty, the function tries to fetch more, which can
+ * block, and returns a positive return code if there is no more data to be
+ * read. Since this and other functions can alter the buffer pointer and
+ * contents, do not store the pointers returned here across function calls.
+ *
+ * Higher level functions like @ref istream_read, providing a Unix read() style
+ * API are built on top of this primitive.
+ *
+ * @param strm A pointer to an istream_t implementation.
+ * @param out Returns a pointer into an internal buffer on success.
+ * @param size Returns the number of bytes available in the buffer.
+ *
+ * @return Zero on success, a negative error code on failure,
+ * a postive number on EOF.
+ */
+SQFS_INLINE int istream_get_buffered_data(istream_t *strm, const sqfs_u8 **out,
+ size_t *size)
+{
+ if (strm->buffer_used == 0) {
+ int ret = strm->precache(strm);
+ if (ret)
+ return ret;
+ if (strm->buffer_used == 0)
+ return 1;
+ }
+
+ *out = strm->buffer;
+ *size = strm->buffer_used;
+ return 0;
+}
+
+/**
+ * @brief Mark a section of the internal buffer of an istream as used
+ *
+ * @memberof istream_t
+ *
+ * This marks the first `count` bytes of the internal buffer as used,
+ * forcing @ref istream_get_buffered_data to return data afterwards
+ * and potentially to load more data.
+ *
+ * @param strm A pointer to an istream_t implementation.
+ * @param count The number of bytes used up.
+ *
+ * @return Zero on success, a negative error code on failure.
+ */
+SQFS_INLINE int istream_advance_buffer(istream_t *strm, size_t count)
+{
+ if (count >= strm->buffer_used) {
+ strm->buffer += strm->buffer_used;
+ strm->buffer_used = 0;
+ return strm->precache(strm);
+ }
+
+ strm->buffer += count;
+ strm->buffer_used -= count;
+ return 0;
+}
+
+/**
* @brief Skip over a number of bytes in an input stream.
*
* @memberof istream_t