-
Notifications
You must be signed in to change notification settings - Fork 3
/
dec.cpp
98 lines (77 loc) · 1.77 KB
/
dec.cpp
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
#include <cstdio>
#include <cstdlib>
#include <sys/stat.h>
#include "libbrotli/brotli/dec/decode.h"
#define KXVER 3
#include "k.h"
namespace {
static char err_type[] = "type";
static char err_open[] = "open";
static char err_stat[] = "stat";
static char err_mem[] = "mem";
static char err_read[] = "read";
static char err_parse[] = "dec_parse";
static char err_dec[] = "dec";
};
extern "C" K broqdec(K x)
{
const char* path = nullptr;
uint8_t* enc_data = nullptr;
if (x->t == -(KS))
{
path = x->s;
if (path[0] == ':')
{
path++;
}
}
else
{
return krr(err_type);
}
FILE* f = fopen(path, "rb");
if (f == nullptr)
{
return krr(err_open);
}
struct stat f_info;
if (fstat(fileno(f), &f_info) != 0)
{
fclose(f);
return krr(err_stat);
}
size_t enc_size = f_info.st_size;
enc_data = (uint8_t*)malloc(enc_size);
if (enc_data == nullptr)
{
fclose(f);
return krr(err_mem);
}
if (fread(enc_data, 1, enc_size, f) != enc_size)
{
fclose(f);
free(enc_data);
return krr(err_read);
}
fclose(f);
size_t dec_size = 0;
if (BrotliDecompressedSize(enc_size, enc_data, &dec_size) != 1)
{
free(enc_data);
return krr(err_parse);
}
K dec = ktn(KC, dec_size);
if (dec_size) {
uint8_t *dec_buf = &kC(dec)[0];
size_t dec_actual_size = dec_size;
auto r = BrotliDecompressBuffer(enc_size, enc_data, &dec_actual_size, dec_buf);
if (r != BROTLI_RESULT_SUCCESS)
{
free(enc_data);
r0(dec);
return krr(err_dec);
}
}
free(enc_data);
return dec;
}