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
98
99
100
101
|
#pragma once
//creates uncompressed ZIP archives
#include <nall/string.hpp>
#include <nall/hash/crc32.hpp>
namespace nall::Encode {
struct ZIP {
ZIP(const string& filename) {
fp.open(filename, file::mode::write);
timestamp = time(nullptr);
}
//append path: append("path/");
//append file: append("path/file", data, size);
auto append(string filename, const u8* data = nullptr, u32 size = 0u, time_t timestamp = 0) -> void {
filename.transform("\\", "/");
if(!timestamp) timestamp = this->timestamp;
u32 checksum = Hash::CRC32({data, size}).digest().hex();
directory.append({filename, timestamp, checksum, size, (u32)fp.offset()});
fp.writel(0x04034b50, 4); //signature
fp.writel(0x0014, 2); //minimum version (2.0)
fp.writel(0x0000, 2); //general purpose bit flags
fp.writel(0x0000, 2); //compression method (0 = uncompressed)
fp.writel(makeTime(timestamp), 2);
fp.writel(makeDate(timestamp), 2);
fp.writel(checksum, 4);
fp.writel(size, 4); //compressed size
fp.writel(size, 4); //uncompressed size
fp.writel(filename.length(), 2); //file name length
fp.writel(0x0000, 2); //extra field length
fp.print(filename); //file name
fp.write({data, size}); //file data
}
~ZIP() {
//central directory
u32 baseOffset = fp.offset();
for(auto& entry : directory) {
fp.writel(0x02014b50, 4); //signature
fp.writel(0x0014, 2); //version made by (2.0)
fp.writel(0x0014, 2); //version needed to extract (2.0)
fp.writel(0x0000, 2); //general purpose bit flags
fp.writel(0x0000, 2); //compression method (0 = uncompressed)
fp.writel(makeTime(entry.timestamp), 2);
fp.writel(makeDate(entry.timestamp), 2);
fp.writel(entry.checksum, 4);
fp.writel(entry.size, 4); //compressed size
fp.writel(entry.size, 4); //uncompressed size
fp.writel(entry.filename.length(), 2); //file name length
fp.writel(0x0000, 2); //extra field length
fp.writel(0x0000, 2); //file comment length
fp.writel(0x0000, 2); //disk number start
fp.writel(0x0000, 2); //internal file attributes
fp.writel(0x00000000, 4); //external file attributes
fp.writel(entry.offset, 4); //relative offset of file header
fp.print(entry.filename);
}
u32 finishOffset = fp.offset();
//end of central directory
fp.writel(0x06054b50, 4); //signature
fp.writel(0x0000, 2); //number of this disk
fp.writel(0x0000, 2); //disk where central directory starts
fp.writel(directory.size(), 2); //number of central directory records on this disk
fp.writel(directory.size(), 2); //total number of central directory records
fp.writel(finishOffset - baseOffset, 4); //size of central directory
fp.writel(baseOffset, 4); //offset of central directory
fp.writel(0x0000, 2); //comment length
fp.close();
}
protected:
auto makeTime(time_t timestamp) -> u16 {
tm* info = localtime(×tamp);
return (info->tm_hour << 11) | (info->tm_min << 5) | (info->tm_sec >> 1);
}
auto makeDate(time_t timestamp) -> u16 {
tm* info = localtime(×tamp);
return ((info->tm_year - 80) << 9) | ((1 + info->tm_mon) << 5) + (info->tm_mday);
}
file_buffer fp;
time_t timestamp;
struct entry_t {
string filename;
time_t timestamp;
u32 checksum;
u32 size;
u32 offset;
};
vector<entry_t> directory;
};
}
|
