401 lines
7.4 KiB
C++
Executable File
401 lines
7.4 KiB
C++
Executable File
/*
|
|
* This file is a part of Winix
|
|
* and is not publicly distributed
|
|
*
|
|
* Copyright (c) 2008-2013, Tomasz Sowa
|
|
* All rights reserved.
|
|
*
|
|
*/
|
|
|
|
#include "compress.h"
|
|
#include "log.h"
|
|
|
|
|
|
|
|
|
|
Compress::Compress()
|
|
{
|
|
buffer_in = 0;
|
|
buffer_out = 0;
|
|
buffer_max_len = 65536; // 64KB
|
|
ready_for_compress = false;
|
|
compress_level = 6;
|
|
raw_deflate_inited = false;
|
|
deflate_inited = false;
|
|
gzip_inited = false;
|
|
}
|
|
|
|
|
|
Compress::~Compress()
|
|
{
|
|
delete [] buffer_in;
|
|
delete [] buffer_out;
|
|
|
|
if( raw_deflate_inited )
|
|
deflateEnd(&strm_raw_deflate);
|
|
|
|
if( deflate_inited )
|
|
deflateEnd(&strm_deflate);
|
|
|
|
if( gzip_inited )
|
|
deflateEnd(&strm_gzip);
|
|
}
|
|
|
|
|
|
bool Compress::AllocateMemory()
|
|
{
|
|
if( buffer_in )
|
|
delete [] buffer_in;
|
|
|
|
if( buffer_out )
|
|
delete [] buffer_out;
|
|
|
|
buffer_in = 0;
|
|
buffer_out = 0;
|
|
|
|
try
|
|
{
|
|
buffer_in = new char[buffer_max_len];
|
|
buffer_out = new char[buffer_max_len];
|
|
}
|
|
catch(const std::bad_alloc &)
|
|
{
|
|
log << log1 << "Compress: can't allocate memory" << logend;
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
|
|
bool Compress::InitRawDeflate()
|
|
{
|
|
raw_deflate_inited = false;
|
|
|
|
strm_raw_deflate.next_in = 0;
|
|
strm_raw_deflate.zalloc = Z_NULL;
|
|
strm_raw_deflate.zfree = Z_NULL;
|
|
strm_raw_deflate.opaque = Z_NULL;
|
|
|
|
int ret = deflateInit2(&strm_raw_deflate, compress_level, Z_DEFLATED, -15, 8, Z_DEFAULT_STRATEGY);
|
|
|
|
if( ret != Z_OK )
|
|
log << log1 << "Compress: problem with deflateInit2() for raw deflate" << logend;
|
|
else
|
|
raw_deflate_inited = true;
|
|
|
|
return ret == Z_OK;
|
|
}
|
|
|
|
|
|
bool Compress::InitDeflate()
|
|
{
|
|
deflate_inited = false;
|
|
|
|
strm_deflate.next_in = 0;
|
|
strm_deflate.zalloc = Z_NULL;
|
|
strm_deflate.zfree = Z_NULL;
|
|
strm_deflate.opaque = Z_NULL;
|
|
|
|
int ret = deflateInit2(&strm_deflate, compress_level, Z_DEFLATED, 15, 8, Z_DEFAULT_STRATEGY);
|
|
|
|
if( ret != Z_OK )
|
|
log << log1 << "Compress: problem with deflateInit2() for deflate" << logend;
|
|
else
|
|
deflate_inited = true;
|
|
|
|
return ret == Z_OK;
|
|
}
|
|
|
|
|
|
bool Compress::InitGzip()
|
|
{
|
|
gzip_inited = false;
|
|
|
|
strm_gzip.next_in = 0;
|
|
strm_gzip.zalloc = Z_NULL;
|
|
strm_gzip.zfree = Z_NULL;
|
|
strm_gzip.opaque = Z_NULL;
|
|
|
|
int ret = deflateInit2(&strm_gzip, compress_level, Z_DEFLATED, 15 + 16, 8, Z_DEFAULT_STRATEGY);
|
|
|
|
if( ret != Z_OK )
|
|
log << log1 << "Compress: problem with deflateInit2() for gzip" << logend;
|
|
else
|
|
gzip_inited = true;
|
|
|
|
return ret == Z_OK;
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
return:
|
|
0 - ok
|
|
1 - can't allocate memory
|
|
100 - unknown error
|
|
*/
|
|
int Compress::Init(int compress_level_)
|
|
{
|
|
compress_level = compress_level_;
|
|
|
|
if( !AllocateMemory() )
|
|
return 1;
|
|
|
|
if( InitRawDeflate() && InitDeflate() && InitGzip() )
|
|
ready_for_compress = true;
|
|
else
|
|
return 100;
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int Compress::MakeCompress(z_stream & strm, const char * source, size_t source_len, BinaryPage & out_stream, int encoding)
|
|
{
|
|
int ret, flush;
|
|
size_t have;
|
|
|
|
do
|
|
{
|
|
strm.avail_in = (source_len > buffer_max_len) ? buffer_max_len : source_len;
|
|
source_len -= strm.avail_in;
|
|
flush = (source_len == 0) ? Z_FINISH : Z_NO_FLUSH;
|
|
strm.next_in = (Bytef*)source;
|
|
source += strm.avail_in;
|
|
|
|
do
|
|
{
|
|
strm.avail_out = buffer_max_len;
|
|
strm.next_out = (Bytef*)buffer_out;
|
|
ret = deflate(&strm, flush);
|
|
|
|
if( ret == Z_STREAM_ERROR || ret == Z_BUF_ERROR )
|
|
{
|
|
log << log1 << "Compress: problem with deflate()" << logend;
|
|
return 2;
|
|
}
|
|
|
|
have = buffer_max_len - strm.avail_out;
|
|
last_out_size += have;
|
|
out_stream.write(buffer_out, have);
|
|
}
|
|
while( strm.avail_out == 0 );
|
|
|
|
if( strm.avail_in != 0 )
|
|
{
|
|
log << log1 << "Compress: problem with deflate() - not all input is used" << logend;
|
|
return 2;
|
|
}
|
|
}
|
|
while( flush != Z_FINISH );
|
|
|
|
|
|
if( ret != Z_STREAM_END )
|
|
{
|
|
log << log1 << "Compress: problem with deflate() - stream not complete" << logend;
|
|
return 2;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
void Compress::CopyToInputBuffer(BinaryPage::const_iterator & i, size_t len)
|
|
{
|
|
for(size_t a=0 ; a<len ; ++a, ++i)
|
|
buffer_in[a] = *i;
|
|
}
|
|
|
|
|
|
|
|
// new way
|
|
int Compress::MakeCompress(z_stream & strm, const BinaryPage & page, BinaryPage & out, int encoding)
|
|
{
|
|
int ret, flush;
|
|
size_t have;
|
|
BinaryPage::const_iterator i = page.begin();
|
|
|
|
size_t source_len = page.size();
|
|
|
|
do
|
|
{
|
|
strm.avail_in = (source_len > buffer_max_len) ? buffer_max_len : source_len;
|
|
source_len -= strm.avail_in;
|
|
flush = (source_len == 0) ? Z_FINISH : Z_NO_FLUSH;
|
|
strm.next_in = (Bytef*)buffer_in;
|
|
CopyToInputBuffer(i, strm.avail_in);
|
|
|
|
do
|
|
{
|
|
strm.avail_out = buffer_max_len;
|
|
strm.next_out = (Bytef*)buffer_out;
|
|
ret = deflate(&strm, flush);
|
|
|
|
if( ret == Z_STREAM_ERROR || ret == Z_BUF_ERROR )
|
|
{
|
|
log << log1 << "Compress: problem with deflate()" << logend;
|
|
return 2;
|
|
}
|
|
|
|
have = buffer_max_len - strm.avail_out;
|
|
last_out_size += have;
|
|
out.write(buffer_out, have);
|
|
}
|
|
while( strm.avail_out == 0 );
|
|
|
|
if( strm.avail_in != 0 )
|
|
{
|
|
log << log1 << "Compress: problem with deflate() - not all input is used" << logend;
|
|
return 2;
|
|
}
|
|
}
|
|
while( flush != Z_FINISH );
|
|
|
|
|
|
if( ret != Z_STREAM_END )
|
|
{
|
|
log << log1 << "Compress: problem with deflate() - stream not complete" << logend;
|
|
return 2;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
z_stream * Compress::SelectStream(int encoding)
|
|
{
|
|
z_stream * pstrm;
|
|
|
|
if( encoding == 0 )
|
|
pstrm = &strm_raw_deflate;
|
|
else
|
|
if( encoding == 1 )
|
|
pstrm = &strm_deflate;
|
|
else
|
|
pstrm = &strm_gzip;
|
|
|
|
return pstrm;
|
|
}
|
|
|
|
|
|
void Compress::ResetStream(z_stream * pstrm, int encoding)
|
|
{
|
|
if( deflateReset(pstrm) != Z_OK )
|
|
{
|
|
log << log1 << "Compress: problem with deflateReset()" << logend;
|
|
|
|
deflateEnd(pstrm);
|
|
|
|
if( encoding == 0 )
|
|
InitRawDeflate();
|
|
else
|
|
if( encoding == 1 )
|
|
InitDeflate();
|
|
else
|
|
InitGzip();
|
|
}
|
|
}
|
|
|
|
|
|
void Compress::PutLog(size_t source_len, int encoding)
|
|
{
|
|
double ratio = 100.0 - (double(last_out_size) / double(source_len) * 100.0);
|
|
char buffer[30];
|
|
sprintf(buffer, "%.1f", ratio);
|
|
|
|
log << log2 << "Compress: ";
|
|
|
|
if( encoding == 0 )
|
|
log << "raw deflate";
|
|
else
|
|
if( encoding == 1 )
|
|
log << "deflate";
|
|
else
|
|
log << "gzip";
|
|
|
|
log << ", original size: " << source_len << ", size after compressing: "
|
|
<< (int)last_out_size << ", ratio: " << buffer << "%" << logend;
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
return:
|
|
0 - ok;
|
|
1 - can't allocate memory
|
|
2 - error during compressing
|
|
3 - not inited (use Init() first)
|
|
100 - unknown
|
|
*/
|
|
int Compress::Compressing(const char * source, size_t source_len, BinaryPage & out_stream, int encoding)
|
|
{
|
|
int ret;
|
|
z_stream * pstrm;
|
|
|
|
last_out_size = 0;
|
|
out_stream.clear();
|
|
|
|
if( !ready_for_compress )
|
|
{
|
|
log << log1 << "Compress: not ready yet" << logend;
|
|
return 3;
|
|
}
|
|
|
|
// !! CHECK ME
|
|
// it is correct to immediately return? what about headers in the compressed page?
|
|
if( source_len == 0 )
|
|
return 0;
|
|
|
|
pstrm = SelectStream(encoding);
|
|
ret = MakeCompress(*pstrm, source, source_len, out_stream, encoding);
|
|
ResetStream(pstrm, encoding);
|
|
PutLog(source_len, encoding);
|
|
|
|
return ret;
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
return:
|
|
0 - ok;
|
|
1 - can't allocate memory
|
|
2 - error during compressing
|
|
3 - not inited (use Init() first)
|
|
100 - unknown
|
|
*/
|
|
int Compress::Compressing(const BinaryPage & in, BinaryPage & out, int encoding)
|
|
{
|
|
int ret;
|
|
z_stream * pstrm;
|
|
|
|
last_out_size = 0;
|
|
out.clear();
|
|
|
|
if( !ready_for_compress )
|
|
{
|
|
log << log1 << "Compress: not ready yet" << logend;
|
|
return 3;
|
|
}
|
|
|
|
// !! CHECK ME
|
|
// it is correct to immediately return? what about headers in the compressed page?
|
|
if( in.empty() )
|
|
return 0;
|
|
|
|
pstrm = SelectStream(encoding);
|
|
ret = MakeCompress(*pstrm, in, out, encoding);
|
|
ResetStream(pstrm, encoding);
|
|
PutLog(in.size(), encoding);
|
|
|
|
return ret;
|
|
}
|
|
|