-
Notifications
You must be signed in to change notification settings - Fork 98
/
base64.cc
91 lines (82 loc) · 2.47 KB
/
base64.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
79
80
81
82
83
84
85
86
87
88
89
90
91
// compile with:
// g++ -std=c++20 -Wall -Os base64.cc -I ../src/utils/ ../src/utils/Base64.cc -lz -o encode-gz-base64
// g++ -std=c++20 -Wall -Os base64.cc -I ../src/utils/ ../src/utils/Base64.cc -lz -o decode-gz-base64
#include "Base64.hh"
#include <zlib.h>
#include <bit>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <memory>
#include <span>
#include <sys/stat.h>
#include <vector>
std::string encode(std::span<const char> src)
{
auto len = src.size();
uLongf dstLen = len + len / 1000 + 12 + 1; // worst-case
std::vector<unsigned char> buf(dstLen);
if (compress2(buf.data(), &dstLen,
std::bit_cast<const Bytef*>(src.data()), len, 9)
!= Z_OK) {
std::cerr << "Error while compressing blob.\n";
exit(1);
}
return Base64::encode(std::span(buf.data(), dstLen));
}
std::string decode(std::span<const char> src)
{
static const unsigned MAX_SIZE = 1024 * 1024; // 1MB
const auto& [decBuf, decBufSize] = Base64::decode(std::string_view(src.data(), src.size()));
std::vector<char> buf(MAX_SIZE);
uLongf dstLen = MAX_SIZE;
if (uncompress(std::bit_cast<Bytef*>(buf.data()), &dstLen,
std::bit_cast<const Bytef*>(decBuf.data()), uLong(decBufSize))
!= Z_OK) {
std::cerr << "Error while decompressing blob.\n";
exit(1);
}
return {buf.data(), dstLen};
}
int main(int argc, const char** argv)
{
std::span<const char*> arg(argv, argc);
if (arg.size() != 3) {
std::cerr << "Usage: " << arg[0] << " <input> <output>\n";
exit(1);
}
using FILE_t = std::unique_ptr<FILE, decltype([](FILE* f) { fclose(f); })>;
FILE_t inf(fopen(arg[1], "rb"));
if (!inf) {
std::cerr << "Error while opening " << arg[1] << '\n';
exit(1);
}
struct stat st;
fstat(fileno(inf.get()), &st);
size_t size = st.st_size;
std::vector<char> inBuf(size);
if (fread(inBuf.data(), size, 1, inf.get()) != 1) {
std::cerr << "Error while reading " << arg[1] << '\n';
exit(1);
}
std::string result;
if (strstr(arg[0], "encode-gz-base64")) {
result = encode(inBuf);
} else if (strstr(arg[0], "decode-gz-base64")) {
result = decode(inBuf);
} else {
std::cerr << "This executable should be named 'encode-gz-base64' or "
"'decode-gz-base64'.\n";
exit(1);
}
FILE_t outf(fopen(arg[2], "wb+"));
if (!outf) {
std::cerr << "Error while opening " << arg[2] << '\n';
exit(1);
}
if (fwrite(result.data(), result.size(), 1, outf.get()) != 1) {
std::cerr << "Error while writing " << arg[2] << '\n';
exit(1);
}
}