LiteSetup 中的压缩解压代码。
zlib 只是对内存段进行压缩解压。对于文件,可以分段读取到内存后再操作。
compressBound 的 BUG
注意:compressBound 存在 BUG。当压缩后的大小大于原大小时,该函数返回的是压缩后的最大值,会导致内存溢出。
这个 BUG 在 LiteSetup 项目中被发现——例如压缩一个已经经过 rar 压缩的文件,就可能触发溢出。
UnCompressBlock* Core::UnCompressBuffer(char*buffer, unsigned int len, unsigned int len_orign)
{
UnCompressBlock *ret = new UnCompressBlock;
char *dest = new char[len_orign + 10];
if (0 == uncompress((Bytef*)dest, (uLongf*)&len_orign, (Bytef*)buffer, len))
{
ret->buffer = dest;
ret->_size = len_orign;
return ret;
}
delete[] dest;
delete ret;
return nullptr;
}
CompressBlock* Core::CompressBuffer(char * buffer, unsigned int _size)
{
CompressBlock *ret = new CompressBlock;
unsigned int len = compressBound((uLong)_size);
char * dest = new char[len+1024*1024];
if (compress((Bytef*)dest, (uLongf*)&len, (Bytef*)buffer, _size) == 0)
{
ret->buffer = dest;
ret->_size = len;
ret->_size_orign = _size;
return ret;
}
delete ret;
delete[]dest;
return nullptr;
}