magic_numbers.hpp

Go to the documentation of this file.

#ifndef BYTEME_MAGIC_NUMBERS_HPP
#define BYTEME_MAGIC_NUMBERS_HPP
 
#include <cstddef>
 
#include "SelfClosingFILE.hpp"
 
namespace byteme {
 
inline bool is_zlib(const unsigned char* buffer, std::size_t n) {
    if (n >= 2 && buffer[0] == 0x78) {
        auto level = buffer[1];
        // From https://en.wikipedia.org/wiki/List_of_file_signatures
        if (level == 0x01 || level == 0x5e || level == 0x9c || level == 0xda || level == 0x20 || level == 0x7d || level == 0xbb || level == 0xf9) {
            return true;
        }
    }
    return false;
}
 
inline bool is_gzip(const unsigned char* buffer, std::size_t n) {
    // ommitting the 0x08 requirement as Gzip could be used with non-DEFLATE algorithms.
    return (n >= 2 && buffer[0] == 0x1f && buffer[1] == 0x8b);
}
 
inline bool is_zlib_or_gzip(const unsigned char* buffer, std::size_t n) {
    return is_zlib(buffer, n) || is_gzip(buffer, n);
}
 
inline bool is_gzip(const char* path) {
    SelfClosingFILE file(path, "rb");
    unsigned char header[3];
    auto read = std::fread(header, sizeof(unsigned char), 3, file.handle);
    return is_gzip(header, read);
}
 
enum class ZlibCompressionMode : char { DEFLATE, ZLIB, GZIP };
 
}
 
#endif