1 files changed, 93 insertions, 0 deletions
diff --git a/stream/stm_buffer.c b/stream/stm_buffer.c
new file mode 100644
index 00000000..206137a6
--- /dev/null
+++ b/stream/stm_buffer.c
@@ -0,0 +1,93 @@
+#include <fitz.h>
+
+fz_error *
+fz_newbuffer(fz_buffer **bp, int size)
+{
+	fz_buffer *b;
+
+	b = *bp = fz_malloc(sizeof(fz_buffer));
+	if (!b) return fz_outofmem;
+
+	b->refs = 1;
+	b->ownsdata = 1;
+	b->bp = fz_malloc(size);
+	if (!b->bp) { fz_free(b); return fz_outofmem; }
+
+	b->rp = b->bp;
+	b->wp = b->bp;
+	b->ep = b->bp + size;
+	b->eof = 0;
+
+	return nil;
+}
+
+fz_error *
+fz_newbufferwithdata(fz_buffer **bp, unsigned char *data, int size)
+{
+	fz_buffer *b;
+
+	b = *bp = fz_malloc(sizeof(fz_buffer));
+	if (!b) return fz_outofmem;
+
+	b->refs = 1;
+	b->ownsdata = 0;
+	b->bp = data;
+
+	b->rp = b->bp;
+	b->wp = b->bp + size;
+	b->ep = b->bp + size;
+	b->eof = 0;
+
+	return nil;
+}
+
+fz_buffer *
+fz_keepbuffer(fz_buffer *buf)
+{
+	buf->refs ++;
+	return buf;
+}
+
+void
+fz_dropbuffer(fz_buffer *buf)
+{
+	if (--buf->refs == 0)
+	{
+		if (buf->ownsdata)
+			fz_free(buf->bp);
+		fz_free(buf);
+	}
+}
+
+fz_error *
+fz_growbuffer(fz_buffer *buf)
+{
+	unsigned char *newbp;
+
+	int rp = buf->rp - buf->bp;
+	int wp = buf->wp - buf->bp;
+	int ep = buf->ep - buf->bp;
+
+	assert(buf->ownsdata);
+
+	newbp = fz_realloc(buf->bp, ep * 2);
+	if (!newbp) return fz_outofmem;
+
+	buf->bp = newbp;
+	buf->rp = buf->bp + rp;
+	buf->wp = buf->bp + wp;
+	buf->ep = buf->bp + ep * 2;
+
+	return nil;
+}
+
+fz_error *
+fz_rewindbuffer(fz_buffer *buf)
+{
+	assert(buf->ownsdata);
+	memmove(buf->bp, buf->rp, buf->wp - buf->rp);
+	buf->wp = buf->bp + (buf->wp - buf->rp);
+	buf->rp = buf->bp;
+	return nil;
+}
+