Compression.cpp

Go to the documentation of this file.

#include "botcraft/Network/Compression.hpp"
 
#ifdef USE_COMPRESSION
#include <zlib.h>
#include <string>
#include <cstring>
#include <stdexcept>
 
namespace Botcraft
{
    std::vector<unsigned char> Compress(const std::vector<unsigned char>& raw)
    {
        unsigned long size_to_compress = static_cast<unsigned long>(raw.size());
        unsigned long compressed_size = compressBound(size_to_compress);
 
        std::vector<unsigned char> compressed_data(compressed_size);
        int status = compress2(compressed_data.data(), &compressed_size, raw.data(), size_to_compress, Z_DEFAULT_COMPRESSION);
 
        if (status != Z_OK)
        {
            throw std::runtime_error("Error compressing packet");
        }
 
        compressed_data.resize(compressed_size);
        return compressed_data;
    }
 
    std::vector<unsigned char> Decompress(const std::vector<unsigned char>& compressed, const int start)
    {
        unsigned long size_to_decompress = static_cast<unsigned long>(compressed.size() - start);
 
        std::vector<unsigned char> decompressed_data;
        decompressed_data.reserve(size_to_decompress);
 
        std::vector<unsigned char> buffer(64 * 1024);
 
        z_stream strm;
        memset(&strm, 0, sizeof(strm));
        strm.next_in = const_cast<unsigned char*>(compressed.data() + start);
        strm.avail_in = size_to_decompress;
        strm.next_out = buffer.data();
        strm.avail_out = static_cast<unsigned int>(buffer.size());
 
        int res = inflateInit(&strm);
        if (res != Z_OK)
        {
            throw std::runtime_error("inflateInit failed: " + std::string(strm.msg));
        }
 
        for (;;)
        {
            res = inflate(&strm, Z_NO_FLUSH);
            switch (res)
            {
            case Z_OK:
                decompressed_data.insert(decompressed_data.end(), buffer.begin(), buffer.end() - strm.avail_out);
                strm.next_out = buffer.data();
                strm.avail_out = static_cast<unsigned int>(buffer.size());
                if (strm.avail_in == 0)
                {
                    inflateEnd(&strm);
                    return decompressed_data;
                }
                break;
            case Z_STREAM_END:
                decompressed_data.insert(decompressed_data.end(), buffer.begin(), buffer.end() - strm.avail_out);
                inflateEnd(&strm);
                return decompressed_data;
                break;
            default:
                inflateEnd(&strm);
                throw std::runtime_error("Inflate decompression failed: " + std::string(strm.msg));
                break;
            }
        }
    }
} //Botcraft
#endif