94 lines
1.9 KiB
C
94 lines
1.9 KiB
C
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
|
||
|
|
#include "zstd_decompress.h"
|
||
|
|
|
||
|
|
typedef unsigned char u8;
|
||
|
|
|
||
|
|
// There's no good way to determine output size without decompressing
|
||
|
|
// For this example assume we'll never decompress at a ratio larger than 16
|
||
|
|
#define MAX_COMPRESSION_RATIO (16)
|
||
|
|
|
||
|
|
u8 *input;
|
||
|
|
u8 *output;
|
||
|
|
u8 *dict;
|
||
|
|
|
||
|
|
size_t read_file(const char *path, u8 **ptr) {
|
||
|
|
FILE *f = fopen(path, "rb");
|
||
|
|
if (!f) {
|
||
|
|
fprintf(stderr, "failed to open file %s\n", path);
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
|
||
|
|
fseek(f, 0L, SEEK_END);
|
||
|
|
size_t size = ftell(f);
|
||
|
|
rewind(f);
|
||
|
|
|
||
|
|
*ptr = malloc(size);
|
||
|
|
if (!ptr) {
|
||
|
|
fprintf(stderr, "failed to allocate memory to hold %s\n", path);
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t pos = 0;
|
||
|
|
while (!feof(f)) {
|
||
|
|
size_t read = fread(&(*ptr)[pos], 1, size, f);
|
||
|
|
if (ferror(f)) {
|
||
|
|
fprintf(stderr, "error while reading file %s\n", path);
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
pos += read;
|
||
|
|
}
|
||
|
|
|
||
|
|
fclose(f);
|
||
|
|
|
||
|
|
return pos;
|
||
|
|
}
|
||
|
|
|
||
|
|
void write_file(const char *path, const u8 *ptr, size_t size) {
|
||
|
|
FILE *f = fopen(path, "wb");
|
||
|
|
|
||
|
|
size_t written = 0;
|
||
|
|
while (written < size) {
|
||
|
|
written += fwrite(&ptr[written], 1, size, f);
|
||
|
|
if (ferror(f)) {
|
||
|
|
fprintf(stderr, "error while writing file %s\n", path);
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
fclose(f);
|
||
|
|
}
|
||
|
|
|
||
|
|
int main(int argc, char **argv) {
|
||
|
|
if (argc < 3) {
|
||
|
|
fprintf(stderr, "usage: %s <file.zst> <out_path> [dictionary]\n", argv[0]);
|
||
|
|
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t input_size = read_file(argv[1], &input);
|
||
|
|
size_t dict_size = 0;
|
||
|
|
if (argc >= 4) {
|
||
|
|
dict_size = read_file(argv[3], &dict);
|
||
|
|
}
|
||
|
|
|
||
|
|
output = malloc(MAX_COMPRESSION_RATIO * input_size);
|
||
|
|
if (!output) {
|
||
|
|
fprintf(stderr, "failed to allocate memory\n");
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t decompressed =
|
||
|
|
ZSTD_decompress_with_dict(output, input_size * MAX_COMPRESSION_RATIO,
|
||
|
|
input, input_size, dict, dict_size);
|
||
|
|
|
||
|
|
write_file(argv[2], output, decompressed);
|
||
|
|
|
||
|
|
free(input);
|
||
|
|
free(output);
|
||
|
|
free(dict);
|
||
|
|
input = output = dict = NULL;
|
||
|
|
}
|
||
|
|
|