goldendict-ng/decompress.cc
2012-02-13 19:43:08 +04:00

73 lines
1.6 KiB
C++

#include <string.h>
#include "decompress.hh"
#include "zlib.h"
#include "bzlib.h"
string decompressZlib( char * bufptr, unsigned length )
{
z_stream zs;
char buf[2048];
string str;
int res;
memset( &zs, 0, sizeof(zs) );
zs.next_in = (Bytef *)bufptr;
zs.avail_in = length;
while( 1 )
{
res = inflateInit( &zs );
if( res != Z_OK )
break;
while( res != Z_STREAM_END )
{
memset( buf, 0, sizeof(buf) );
zs.next_out = (Bytef *)buf;
zs.avail_out = 2047;
res = inflate( &zs, Z_SYNC_FLUSH );
str += buf;
if( res != Z_OK && res != Z_STREAM_END )
break;
}
break;
}
inflateEnd( &zs );
if( res != Z_STREAM_END )
str.clear();
return str;
}
string decompressBzip2( char * bufptr, unsigned length )
{
bz_stream zs;
char buf[2048];
string str;
int res;
memset( &zs, 0, sizeof(zs) );
zs.next_in = bufptr;
zs.avail_in = length;
zs.total_in_lo32 = length;
while( 1 )
{
res = BZ2_bzDecompressInit( &zs, 0, 0 );
if( res != BZ_OK )
break;
while( res != BZ_STREAM_END )
{
memset( buf, 0, sizeof(buf) );
zs.next_out = buf;
zs.avail_out = 2047;
zs.total_out_lo32 = length;
res = BZ2_bzDecompress( &zs );
str += buf;
if( res != BZ_OK && res != BZ_STREAM_END )
break;
}
break;
}
BZ2_bzDecompressEnd( &zs );
if( res != BZ_STREAM_END )
str.clear();
return str;
}