1 files changed, 123 insertions, 0 deletions

diff --git a/src/bzip2.cpp b/src/bzip2.cpp
new file mode 100644
index 0000000..b8c0f74
--- /dev/null
+++ b/src/bzip2.cpp
@@ -0,0 +1,123 @@
+#include "bu/bzip2.h"
+#include "bu/exceptions.h"
+
+using namespace Bu;
+
+Bu::BZip2::BZip2( Bu::Stream &rNext, int nCompression ) :
+        Bu::Filter( rNext ),
+        nCompression( nCompression )
+{
+        start();
+}
+
+Bu::BZip2::~BZip2()
+{
+        printf("-> Bu::BZip2::~BZip2()\n");     
+        stop();
+}
+
+void Bu::BZip2::start()
+{
+        printf("-> Bu::BZip2::start()\n");      
+        printf("Hey, it's starting...\n");
+        bzState.state = NULL;
+        bzState.bzalloc = NULL;
+        bzState.bzfree = NULL;
+        bzState.opaque = NULL;
+
+        nBufSize = 50000;
+        pBuf = new char[nBufSize];
+}
+
+void Bu::BZip2::stop()
+{
+        printf("-> Bu::BZip2::stop()\n");
+        if( bzState.state )
+        {
+                if( bReading )
+                {
+                }
+                else
+                {
+                        for(;;)
+                        {
+                                bzState.next_in = NULL;
+                                bzState.avail_in = 0;
+                                bzState.avail_out = nBufSize;
+                                bzState.next_out = pBuf;
+                                int res = BZ2_bzCompress( &bzState, BZ_FINISH );
+                                if( bzState.avail_out < nBufSize )
+                                {
+                                        rNext.write( pBuf, nBufSize-bzState.avail_out );
+                                }
+                                if( res == BZ_STREAM_END )
+                                        break;
+                        }
+                        BZ2_bzCompressEnd( &bzState );
+                }
+        }
+}
+
+void Bu::BZip2::bzError( int code )
+{
+        switch( code )
+        {
+                case BZ_OK:
+                        return;
+
+                case BZ_CONFIG_ERROR:
+                        throw ExceptionBase("The bzip2 library has been miscompiled.");
+
+                case BZ_PARAM_ERROR:
+                        throw ExceptionBase("bzip2 parameter error.");
+
+                case BZ_MEM_ERROR:
+                        throw ExceptionBase("Not enough memory available for bzip2.");
+        }
+}
+
+size_t Bu::BZip2::read( void *pData, size_t nBytes )
+{
+        if( !bzState.state )
+        {
+                bReading = true;
+        }
+        if( bReading == false )
+                throw ExceptionBase("This bzip2 filter is in writing mode, you can't read.");
+        //bzState.next_in = pData;
+        //bzState.avail_in = nSizeIn;
+        
+        //printf("%db at [%08X] (%db)\n", bzState.avail_in, (uint32_t)bzState.next_in, bzState.total_in_lo32 );
+        return 0;
+}
+
+size_t Bu::BZip2::write( const void *pData, size_t nBytes )
+{
+        if( !bzState.state )
+        {
+                bReading = false;
+                BZ2_bzCompressInit( &bzState, nCompression, 0, 30 );
+        }
+        if( bReading == true )
+                throw ExceptionBase("This bzip2 filter is in reading mode, you can't write.");
+
+        bzState.next_in = (char *)pData;
+        bzState.avail_in = nBytes;
+        for(;;)
+        {
+                bzState.avail_out = nBufSize;
+                bzState.next_out = pBuf;
+
+                BZ2_bzCompress( &bzState, BZ_RUN );
+
+                if( bzState.avail_out < nBufSize )
+                {
+                        rNext.write( pBuf, nBufSize-bzState.avail_out );
+                }
+                if( bzState.avail_in == 0 )
+                        break;
+        }
+
+        return 0;
+}
+