mirror of
https://github.com/rakshasa/rtorrent.git
synced 2026-10-04 21:29:21 +00:00
noctuum
46 lines
1.2 KiB
C++
46 lines
1.2 KiB
C++
#include "config.h"
|
|
|
|
#include "utils/gzip.h"
|
|
|
|
#include <zlib.h>
|
|
#include <torrent/exceptions.h>
|
|
|
|
#include "utils/functional.h"
|
|
|
|
namespace utils {
|
|
|
|
void
|
|
gzip_compress_to_vector(const char* buffer, unsigned int length, std::vector<char>& output, unsigned int offset) {
|
|
z_stream zs{};
|
|
zs.zalloc = Z_NULL;
|
|
zs.zfree = Z_NULL;
|
|
zs.opaque = Z_NULL;
|
|
|
|
constexpr int window_bits = 15;
|
|
constexpr int gzip_encoding = 16;
|
|
constexpr int gzip_level = 6;
|
|
|
|
if (deflateInit2(&zs, Z_DEFAULT_COMPRESSION, Z_DEFLATED, window_bits | gzip_encoding, gzip_level, Z_DEFAULT_STRATEGY) != Z_OK)
|
|
throw torrent::internal_error("gzip_compress_to_vector(...) could not initialize gzip deflate.");
|
|
|
|
scope_guard guard([&zs]() { deflateEnd(&zs); });
|
|
|
|
auto max_response_size = deflateBound(&zs, length);
|
|
|
|
output.resize(offset + max_response_size);
|
|
|
|
zs.next_in = (Bytef*)buffer;
|
|
zs.avail_in = length;
|
|
zs.next_out = (Bytef*)(output.data() + offset);
|
|
zs.avail_out = max_response_size;
|
|
|
|
int ret = deflate(&zs, Z_FINISH);
|
|
|
|
if (ret != Z_STREAM_END)
|
|
throw torrent::internal_error("gzip_compress_to_vector(...) deflate did not return Z_STREAM_END: " + std::to_string(ret));
|
|
|
|
output.resize(offset + max_response_size - zs.avail_out);
|
|
}
|
|
|
|
} // namespace utils
|