-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_nanozdec.cc
78 lines (59 loc) · 1.86 KB
/
test_nanozdec.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#define NANOZLIB_IMPLEMENTATION
#include "nanozlib.h"
#include <cstdio>
#include <cstdlib>
#include <vector>
#include <iostream>
#include <fstream>
int main(int argc, char **argv) {
if (argc < 3) {
std::cout << "test compressed_data.bin uncompressed_size (dump_uncompressed_data)\n";
exit(-1);
}
bool dump = false;
// TODO: Use string -> uint64
std::string filename = argv[1];
uint64_t uncompressed_size = atoi(argv[2]);
std::vector<char> uncompressed_data;
uncompressed_data.resize(uncompressed_size);
if (argc > 3) {
dump = true;
}
std::ifstream f(filename.c_str(), std::ifstream::binary);
if (!f) {
std::cerr << "File not found or invalid.\n";
return EXIT_FAILURE;
}
f.peek();
if (!f) {
std::cerr << "Invalid file.\n";
return EXIT_FAILURE;
}
f.seekg(0, f.end);
size_t sz = static_cast<size_t>(f.tellg());
f.seekg(0, f.beg);
//std::cout << "file sz = " << sz << "\n";
std::vector<unsigned char> src_buf;
src_buf.resize(sz);
f.read(reinterpret_cast<char *>(src_buf.data()), static_cast<std::streamsize>(sz));
if (!f) {
std::cerr << "Failed to read file.\n";
return EXIT_FAILURE;
}
uint64_t uncompressed_size_in_zlib{0};
nanoz_status_t ret = nanoz_uncompress(src_buf.data(), sz, uint64_t(uncompressed_data.size()),
reinterpret_cast<unsigned char *>(uncompressed_data.data()), &uncompressed_size_in_zlib);
if (ret != NANOZ_SUCCESS) {
std::cerr << "nanoz error: " << int(ret) << "\n";
return EXIT_FAILURE;
}
if (uncompressed_size_in_zlib != uncompressed_size) {
std::cerr << "Expected uncompressed size " << uncompressed_size << ", but actual uncompressed size is " << uncompressed_size_in_zlib << "\n";
return EXIT_FAILURE;
}
if (dump) {
std::string s(uncompressed_data.begin(), uncompressed_data.end());
std::cout << s << "\n";
}
return EXIT_SUCCESS;
}