Added the SBuffer class, a memory buffer that can be used like a stream. Yay!

3 files changed, 123 insertions, 0 deletions

diff --git a/src/sbuffer.cpp b/src/sbuffer.cpp
new file mode 100644
index 0000000..00324b5
--- /dev/null
+++ b/src/sbuffer.cpp
@@ -0,0 +1,62 @@
+#include <string.h>
+#include "sbuffer.h"
+
+SBuffer::SBuffer() :
+        bOpen( true ),
+        nPos( 0 )
+{
+}
+
+SBuffer::~SBuffer()
+{
+}
+
+void SBuffer::close()
+{
+        bOpen = false;
+        fbData.clearData();
+}
+
+size_t SBuffer::read( char *pBuf, size_t nBytes )
+{
+        long nLeft = fbData.getLength() - nPos;
+        if( nBytes > nLeft )
+                nBytes = nLeft;
+
+        if( nLeft == 0 )
+                return 0;
+
+        memcpy( pBuf, fbData.getData()+nPos, nBytes );
+        nPos += nBytes;
+
+        return nBytes;
+}
+
+size_t SBuffer::write( const char *pBuf, size_t nBytes )
+{
+        fbData.appendData( pBuf, nBytes );
+        nPos += nBytes;
+
+        return nBytes;
+}
+
+long SBuffer::tell()
+{
+        return nPos;
+}
+
+void SBuffer::seek( long offset )
+{
+        nPos += offset;
+}
+
+void SBuffer::setPos( long pos )
+{
+        nPos = pos;
+}
+
+void SBuffer::setPosEnd( long pos )
+{
+        nPos = fbData.getLength() - pos;
+}
+
diff --git a/src/sbuffer.h b/src/sbuffer.h
new file mode 100644
index 0000000..1dda33a
--- /dev/null
+++ b/src/sbuffer.h
@@ -0,0 +1,34 @@
+#ifndef S_BUFFER_H
+#define S_BUFFER_H
+
+#include <stdint.h>
+
+#include "stream.h"
+#include "flexbuf.h"
+
+class SBuffer : public Stream
+{
+public:
+        SBuffer();
+        virtual ~SBuffer();
+
+        virtual void close();
+        virtual size_t read( char *pBuf, size_t nBytes );
+
+        /**
+         *@todo Update this to write at nPos, not just append data.
+         */
+        virtual size_t write( const char *pBuf, size_t nBytes );
+
+        virtual long tell();
+        virtual void seek( long offset );
+        virtual void setPos( long pos );
+        virtual void setPosEnd( long pos );
+
+private:
+        long nPos;
+        bool bOpen;
+        FlexBuf fbData;
+};
+
+#endif
diff --git a/src/tests/sbuffer.cpp b/src/tests/sbuffer.cpp
new file mode 100644
index 0000000..d8bbe4d
--- /dev/null
+++ b/src/tests/sbuffer.cpp
@@ -0,0 +1,27 @@
+#include "sbuffer.h"
+
+int main()
+{
+        SBuffer buf;
+
+        buf.write("abcdefg", 7 );
+
+        printf("tell: %d\n", buf.tell() );
+
+        char abuf[6];
+        int nRead;
+        nRead = buf.read( abuf, 5 );
+        abuf[nRead] = '\0';
+        printf("Read %d bytes \"%s\"\n", nRead, abuf );
+
+        buf.setPos( 0 );
+        nRead = buf.read( abuf, 5 );
+        abuf[nRead] = '\0';
+        printf("Read %d bytes \"%s\"\n", nRead, abuf );
+
+        nRead = buf.read( abuf, 5 );
+        abuf[nRead] = '\0';
+        printf("Read %d bytes \"%s\"\n", nRead, abuf );
+
+}
+