GzipFileReader.hpp

Go to the documentation of this file.

#ifndef BYTEME_GZIP_FILE_READER_HPP
#define BYTEME_GZIP_FILE_READER_HPP
 
#include <cstddef>
#include <stdexcept>
#include <vector>
#include <optional>
 
#include "zlib.h"
 
#include "SelfClosingGzFile.hpp"
#include "Reader.hpp"
#include "utils.hpp"
 
namespace byteme {
 
struct GzipFileReaderOptions {
    std::optional<unsigned> buffer_size;
};
 
class GzipFileReader final : public Reader {
public:
    GzipFileReader(const char* path, const GzipFileReaderOptions& options) : my_gzfile(path, "rb") {
        set_optional_gzbuffer_size(my_gzfile, options.buffer_size);
    }
 
public:
    std::size_t read(unsigned char* buffer, std::size_t n) {
        // While gzbuffer and gzread accepts 'unsigned', the output number of bytes is actually an 'int'!
        // Moreover, if the requested 'len' can't fit in an 'int', gzread will throw an error -
        // see comments on the return value for gzread at https://www.zlib.net/manual.html.
        // So, we forcibly cap the length to 'int'. 
        return safe_read<int>(buffer, n, [this](unsigned char* buffer, int n) -> int {
            const auto ret = gzread(this->my_gzfile.handle, buffer, n);
            if (ret < n) {
                int status;
                const auto msg = gzerror(this->my_gzfile.handle, &status);
                if (status != Z_OK) {
                    throw std::runtime_error(msg);
                }
            }
            return ret;
        });
    }
 
private:
    SelfClosingGzFile my_gzfile;
};
 
}
 
#endif