gzip.hpp

// Copyright Take Vos 2020.
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at https://www.boost.org/LICENSE_1_0.txt)
 
#pragma once
 
#include "../container/container.hpp"
#include "../file/file.hpp"
#include "../parser/parser.hpp"
#include "../macros.hpp"
#include "inflate.hpp"
#include <cstddef>
#include <filesystem>
 
hi_export_module(hikogui.codec.gzip);
 
hi_export namespace hi { inline namespace v1 {
namespace detail {
 
struct gzip_member_header {
    uint8_t ID1;
    uint8_t ID2;
    uint8_t CM;
    uint8_t FLG;
    little_uint32_buf_t MTIME;
    uint8_t XFL;
    uint8_t OS;
};
 
[[nodiscard]] hi_inline bstring gzip_decompress_member(std::span<std::byte const> bytes, std::size_t &offset, std::size_t max_size)
{
    auto const header = make_placement_ptr<gzip_member_header>(bytes, offset);
 
    hi_check(header->ID1 == 31, "GZIP Member header ID1 must be 31");
    hi_check(header->ID2 == 139, "GZIP Member header ID2 must be 139");
    hi_check(header->CM == 8, "GZIP Member header CM must be 8");
    hi_check((header->FLG & 0xe0) == 0, "GZIP Member header FLG reserved bits must be 0");
    hi_check(header->XFL == 2 or header->XFL == 4, "GZIP Member header XFL must be 2 or 4");
    [[maybe_unused]] auto const FTEXT = to_bool(header->FLG & 1);
    auto const FHCRC = to_bool(header->FLG & 2);
    auto const FEXTRA = to_bool(header->FLG & 4);
    auto const FNAME = to_bool(header->FLG & 8);
    auto const FCOMMENT = to_bool(header->FLG & 16);
 
    if (FEXTRA) {
        auto const XLEN = **make_placement_ptr<little_uint16_buf_t>(bytes, offset);
        offset += XLEN;
    }
 
    if (FNAME) {
        auto c = std::byte{};
        do {
            hi_check(offset < bytes.size(), "GZIP Member header FNAME reading beyond end of buffer");
            c = bytes[offset++];
        } while (c != std::byte{0});
    }
 
    if (FCOMMENT) {
        auto c = std::byte{};
        do {
            hi_check(offset < bytes.size(), "GZIP Member header FCOMMENT reading beyond end of buffer");
            c = bytes[offset++];
        } while (c != std::byte{0});
    }
 
    if (FHCRC) {
        [[maybe_unused]] auto const CRC16 = make_placement_ptr<little_uint16_buf_t>(bytes, offset);
    }
 
    auto r = inflate(bytes, offset, max_size);
 
    [[maybe_unused]] auto CRC32 = **make_placement_ptr<little_uint32_buf_t>(bytes, offset);
    [[maybe_unused]] auto ISIZE = **make_placement_ptr<little_uint32_buf_t>(bytes, offset);
 
    hi_check(
        ISIZE == (size(r) & 0xffffffff),
        "GZIP Member header ISIZE must be same as the lower 32 bits of the inflated size.");
    return r;
}
 
}
 
hi_export [[nodiscard]] hi_inline bstring gzip_decompress(std::span<std::byte const> bytes, std::size_t max_size)
{
    auto r = bstring{};
 
    auto offset = 0_uz;
    while (offset < bytes.size()) {
        auto member = detail::gzip_decompress_member(bytes, offset, max_size);
        max_size -= member.size();
        r.append(member);
    }
 
    return r;
}
 
hi_export [[nodiscard]] hi_inline bstring gzip_decompress(std::filesystem::path const &path, std::size_t max_size = 0x01000000)
{
    return gzip_decompress(as_span<std::byte const>(file_view{path}), max_size);
}
 
}} // namespace hi::inline v1