237 lines
8.5 KiB
C++
237 lines
8.5 KiB
C++
#include "dynos.cpp.h"
|
|
#include <zlib.h>
|
|
|
|
static const u64 DYNOS_BIN_COMPRESS_MAGIC = 0x4E4942534F4E5944llu;
|
|
static FILE *sFile = NULL;
|
|
static u8 *sBufferUncompressed = NULL;
|
|
static u8 *sBufferCompressed = NULL;
|
|
static u64 sLengthUncompressed = 0;
|
|
static u64 sLengthCompressed = 0;
|
|
|
|
static inline void DynOS_Bin_Compress_Init() {
|
|
sFile = NULL;
|
|
sBufferUncompressed = NULL;
|
|
sBufferCompressed = NULL;
|
|
sLengthUncompressed = 0;
|
|
sLengthCompressed = 0;
|
|
}
|
|
|
|
static inline void DynOS_Bin_Compress_Close() {
|
|
if (sFile) f_close(sFile);
|
|
sFile = NULL;
|
|
}
|
|
|
|
static inline void DynOS_Bin_Compress_Free() {
|
|
if (sBufferCompressed) free(sBufferCompressed);
|
|
if (sBufferUncompressed) free(sBufferUncompressed);
|
|
DynOS_Bin_Compress_Close();
|
|
}
|
|
|
|
static inline bool DynOS_Bin_Compress_Check(bool condition, const char *function, const char *filename, const char *message) {
|
|
if (!condition) {
|
|
PrintError("ERROR: %s: File \"%s\": %s", function, filename, message);
|
|
DynOS_Bin_Compress_Free();
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool DynOS_Bin_IsCompressed(const SysPath &aFilename) {
|
|
DynOS_Bin_Compress_Init();
|
|
|
|
// Open input file
|
|
if (!DynOS_Bin_Compress_Check(
|
|
(sFile = fopen(aFilename.c_str(), "rb")) != NULL,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot open file"
|
|
)) return false;
|
|
|
|
// Read magic
|
|
u64 _Magic = 0;
|
|
if (!DynOS_Bin_Compress_Check(
|
|
fread(&_Magic, sizeof(u64), 1, sFile) == 1,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot read magic"
|
|
)) return false;
|
|
|
|
// Compare with magic constant
|
|
if (_Magic != DYNOS_BIN_COMPRESS_MAGIC) {
|
|
DynOS_Bin_Compress_Free();
|
|
return false;
|
|
}
|
|
|
|
// It is a compressed file
|
|
DynOS_Bin_Compress_Free();
|
|
return true;
|
|
}
|
|
|
|
bool DynOS_Bin_Compress(const SysPath &aFilename) {
|
|
DynOS_Bin_Compress_Init();
|
|
PrintNoNewLine("Compressing file \"%s\"...", aFilename.c_str());
|
|
|
|
// Open input file
|
|
if (!DynOS_Bin_Compress_Check(
|
|
(sFile = fopen(aFilename.c_str(), "rb")) != NULL,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot open file"
|
|
)) return false;
|
|
|
|
// Retrieve file length
|
|
if (!DynOS_Bin_Compress_Check(
|
|
fseek(sFile, 0, SEEK_END) == 0,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot retrieve file length"
|
|
)) return false;
|
|
|
|
// Check file length
|
|
if (!DynOS_Bin_Compress_Check(
|
|
(sLengthUncompressed = (u64) ftell(sFile)) != 0,
|
|
__FUNCTION__, aFilename.c_str(), "Empty file"
|
|
)) return false;
|
|
|
|
// Allocate memory for uncompressed buffer
|
|
if (!DynOS_Bin_Compress_Check(
|
|
(sBufferUncompressed = (u8 *) calloc(sLengthUncompressed, sizeof(u8))) != NULL,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot allocate memory for compression"
|
|
)) return false; else rewind(sFile);
|
|
|
|
// Read input data
|
|
if (!DynOS_Bin_Compress_Check(
|
|
fread(sBufferUncompressed, sizeof(u8), sLengthUncompressed, sFile) == sLengthUncompressed,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot read uncompressed data"
|
|
)) return false; else DynOS_Bin_Compress_Close();
|
|
|
|
// Compute maximum output file size
|
|
if (!DynOS_Bin_Compress_Check(
|
|
(sLengthCompressed = compressBound(sLengthUncompressed)) != 0,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot compute compressed size"
|
|
)) return false;
|
|
|
|
// Allocate memory for compressed buffer
|
|
if (!DynOS_Bin_Compress_Check(
|
|
(sBufferCompressed = (u8 *) calloc(sLengthCompressed, sizeof(u8))) != NULL,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot allocate memory for compression"
|
|
)) return false;
|
|
|
|
// Compress data
|
|
uLongf _LengthCompressed = (uLongf)sLengthCompressed;
|
|
if (!DynOS_Bin_Compress_Check(
|
|
compress2(sBufferCompressed, &_LengthCompressed, sBufferUncompressed, sLengthUncompressed, Z_BEST_COMPRESSION) == Z_OK,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot compress data"
|
|
)) return false;
|
|
sLengthCompressed = _LengthCompressed;
|
|
|
|
// Check output length
|
|
// If the compression generates a bigger file, skip the process, but don't return a failure
|
|
if (!DynOS_Bin_Compress_Check(
|
|
sLengthCompressed < sLengthUncompressed,
|
|
__FUNCTION__, aFilename.c_str(), "Compressed data is bigger than uncompressed; Skipping compression"
|
|
)) return true;
|
|
|
|
// Open output file
|
|
if (!DynOS_Bin_Compress_Check(
|
|
(sFile = fopen(aFilename.c_str(), "wb")) != NULL,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot open file"
|
|
)) return false;
|
|
|
|
// Write magic
|
|
if (!DynOS_Bin_Compress_Check(
|
|
fwrite(&DYNOS_BIN_COMPRESS_MAGIC, sizeof(u64), 1, sFile) == 1,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot write magic"
|
|
)) return false;
|
|
|
|
// Write uncompressed file size
|
|
if (!DynOS_Bin_Compress_Check(
|
|
fwrite(&sLengthUncompressed, sizeof(u64), 1, sFile) == 1,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot write uncompressed file size"
|
|
)) return false;
|
|
|
|
// Write compressed data
|
|
if (!DynOS_Bin_Compress_Check(
|
|
fwrite(sBufferCompressed, sizeof(u8), sLengthCompressed, sFile) == sLengthCompressed,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot write compressed data"
|
|
)) return false;
|
|
|
|
// Done, free buffers and files
|
|
DynOS_Bin_Compress_Free();
|
|
Print(" Done.");
|
|
return true;
|
|
}
|
|
|
|
BinFile *DynOS_Bin_Decompress(const SysPath &aFilename) {
|
|
DynOS_Bin_Compress_Init();
|
|
|
|
// Open input file
|
|
if (!DynOS_Bin_Compress_Check(
|
|
(sFile = f_open_r(aFilename.c_str())) != NULL,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot open file"
|
|
)) return NULL;
|
|
|
|
// Read magic
|
|
u64 _Magic = 0;
|
|
if (!DynOS_Bin_Compress_Check(
|
|
f_read(&_Magic, sizeof(u64), 1, sFile) == 1,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot read magic"
|
|
)) return NULL;
|
|
|
|
// Compare with magic constant
|
|
// If not equal, it's not a compressed file
|
|
if (_Magic != DYNOS_BIN_COMPRESS_MAGIC) {
|
|
DynOS_Bin_Compress_Free();
|
|
return BinFile::OpenR(aFilename.c_str());
|
|
}
|
|
PrintNoNewLine("Decompressing file \"%s\"...", aFilename.c_str());
|
|
|
|
// Read expected uncompressed file size
|
|
if (!DynOS_Bin_Compress_Check(
|
|
f_read(&sLengthUncompressed, sizeof(u64), 1, sFile) == 1,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot read uncompressed file size"
|
|
)) return NULL;
|
|
|
|
// Retrieve file length
|
|
if (!DynOS_Bin_Compress_Check(
|
|
f_seek(sFile, 0, SEEK_END) == 0,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot retrieve file length"
|
|
)) return NULL;
|
|
|
|
// Check file length
|
|
u64 _LengthHeader = (u64) (sizeof(u64) + sizeof(u64));
|
|
if (!DynOS_Bin_Compress_Check(
|
|
(sLengthCompressed = (u64) f_tell(sFile)) >= _LengthHeader,
|
|
__FUNCTION__, aFilename.c_str(), "Empty file"
|
|
)) return NULL;
|
|
|
|
// Allocate memory for compressed buffer
|
|
if (!DynOS_Bin_Compress_Check(
|
|
(sBufferCompressed = (u8 *) calloc(sLengthCompressed - _LengthHeader, sizeof(u8))) != NULL,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot allocate memory for decompression"
|
|
)) return NULL; else f_seek(sFile, _LengthHeader, SEEK_SET);
|
|
|
|
// Read input data
|
|
if (!DynOS_Bin_Compress_Check(
|
|
f_read(sBufferCompressed, sizeof(u8), sLengthCompressed - _LengthHeader, sFile) == sLengthCompressed - _LengthHeader,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot read compressed data"
|
|
)) return NULL; else DynOS_Bin_Compress_Close();
|
|
|
|
// Allocate memory for uncompressed buffer
|
|
if (!DynOS_Bin_Compress_Check(
|
|
(sBufferUncompressed = (u8 *) calloc(sLengthUncompressed, sizeof(u8))) != NULL,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot allocate memory for decompression"
|
|
)) return NULL;
|
|
|
|
// Uncompress data
|
|
uLongf _LengthUncompressed = (uLongf)sLengthUncompressed;
|
|
int uncompressRc = uncompress(sBufferUncompressed, &_LengthUncompressed, sBufferCompressed, sLengthCompressed);
|
|
sLengthUncompressed = _LengthUncompressed;
|
|
if (!DynOS_Bin_Compress_Check(
|
|
uncompressRc == Z_OK,
|
|
__FUNCTION__, aFilename.c_str(), "Cannot uncompress data"
|
|
)) {
|
|
PrintError("ERROR: uncompress rc: %d, length uncompressed: %lu, length compressed: %lu, length header: %lu", uncompressRc, sLengthUncompressed, sLengthCompressed, _LengthHeader);
|
|
return NULL;
|
|
}
|
|
Print("uncompress rc: %d, length uncompressed: %lu, length compressed: %lu, length header: %lu", uncompressRc, sLengthUncompressed, sLengthCompressed, _LengthHeader);
|
|
|
|
// Return uncompressed data as a BinFile
|
|
BinFile *_BinFile = BinFile::OpenB(sBufferUncompressed, sLengthUncompressed);
|
|
DynOS_Bin_Compress_Free();
|
|
Print(" Done.");
|
|
return _BinFile;
|
|
}
|