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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
/* This file is part of the Spring engine (GPL v2 or later), see LICENSE.html */
#include "ZipArchive.h"
#include <algorithm>
#include <stdexcept>
#include <assert.h>
#include "System/Util.h"
#include "System/Log/ILog.h"
CZipArchiveFactory::CZipArchiveFactory()
: IArchiveFactory("sdz")
{
}
IArchive* CZipArchiveFactory::DoCreateArchive(const std::string& filePath) const
{
return new CZipArchive(filePath);
}
CZipArchive::CZipArchive(const std::string& archiveName)
: CBufferedArchive(archiveName)
{
zip = unzOpen(archiveName.c_str());
if (!zip) {
LOG_L(L_ERROR, "Error opening \"%s\"", archiveName.c_str());
return;
}
// We need to map file positions to speed up opening later
for (int ret = unzGoToFirstFile(zip); ret == UNZ_OK; ret = unzGoToNextFile(zip))
{
unz_file_info info;
char fName[512];
unzGetCurrentFileInfo(zip, &info, fName, 512, NULL, 0, NULL, 0);
const std::string fLowerName = StringToLower(fName);
if (fLowerName.empty()) {
continue;
}
const char last = fLowerName[fLowerName.length() - 1];
if ((last == '/') || (last == '\\')) {
continue; // exclude directory names
}
FileData fd;
unzGetFilePos(zip, &fd.fp);
fd.size = info.uncompressed_size;
fd.origName = fName;
fd.crc = info.crc;
fileData.push_back(fd);
lcNameIndex[fLowerName] = fileData.size() - 1;
}
}
CZipArchive::~CZipArchive()
{
if (zip) {
unzClose(zip);
}
}
bool CZipArchive::IsOpen()
{
return (zip != NULL);
}
unsigned int CZipArchive::NumFiles() const
{
return fileData.size();
}
void CZipArchive::FileInfo(unsigned int fid, std::string& name, int& size) const
{
assert(IsFileId(fid));
name = fileData[fid].origName;
size = fileData[fid].size;
}
unsigned int CZipArchive::GetCrc32(unsigned int fid)
{
assert(IsFileId(fid));
return fileData[fid].crc;
}
// To simplify things, files are always read completely into memory from
// the zip-file, since zlib does not provide any way of reading more
// than one file at a time
bool CZipArchive::GetFileImpl(unsigned int fid, std::vector<boost::uint8_t>& buffer)
{
// Prevent opening files on missing/invalid archives
if (!zip) {
return false;
}
assert(IsFileId(fid));
unzGoToFilePos(zip, &fileData[fid].fp);
unz_file_info fi;
unzGetCurrentFileInfo(zip, &fi, NULL, 0, NULL, 0, NULL, 0);
if (unzOpenCurrentFile(zip) != UNZ_OK) {
return false;
}
buffer.resize(fi.uncompressed_size);
bool ret = true;
if (!buffer.empty() && unzReadCurrentFile(zip, &buffer[0], fi.uncompressed_size) != fi.uncompressed_size) {
ret = false;
}
if (unzCloseCurrentFile(zip) == UNZ_CRCERROR) {
ret = false;
}
if (!ret) {
buffer.clear();
}
return ret;
}
|
