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
|
/* This file is part of the Spring engine (GPL v2 or later), see LICENSE.html */
#include "ZipArchive.h"
#include <algorithm>
#include <stdexcept>
#include <cassert>
#include "System/StringUtil.h"
#include "System/Log/ILog.h"
IArchive* CZipArchiveFactory::DoCreateArchive(const std::string& filePath) const
{
return new CZipArchive(filePath);
}
CZipArchive::CZipArchive(const std::string& archiveName): CBufferedArchive(archiveName)
{
std::lock_guard<spring::mutex> lck(archiveLock);
if ((zip = unzOpen(archiveName.c_str())) == nullptr) {
LOG_L(L_ERROR, "[%s] error opening \"%s\"", __func__, archiveName.c_str());
return;
}
unz_global_info64 globalZipInfo;
memset(&globalZipInfo, 0, sizeof(globalZipInfo));
unzGetGlobalInfo64(zip, &globalZipInfo);
// We need to map file positions to speed up opening later
fileEntries.reserve(globalZipInfo.number_entry);
for (int ret = unzGoToFirstFile(zip); ret == UNZ_OK; ret = unzGoToNextFile(zip)) {
unz_file_info info;
char fName[512];
unzGetCurrentFileInfo(zip, &info, fName, sizeof(fName), nullptr, 0, nullptr, 0);
if (fName[0] == 0)
continue;
const size_t fNameLen = strlen(fName);
// exclude directory names
if ((fName[fNameLen - 1] == '/') || (fName[fNameLen - 1] == '\\'))
continue;
FileEntry fd;
unzGetFilePos(zip, &fd.fp);
fd.size = info.uncompressed_size;
fd.origName = fName;
fd.crc = info.crc;
lcNameIndex.emplace(StringToLower(fd.origName), fileEntries.size());
fileEntries.emplace_back(std::move(fd));
}
}
CZipArchive::~CZipArchive()
{
std::lock_guard<spring::mutex> lck(archiveLock);
if (zip != nullptr) {
unzClose(zip);
zip = nullptr;
}
}
void CZipArchive::FileInfo(unsigned int fid, std::string& name, int& size) const
{
assert(IsFileId(fid));
name = fileEntries[fid].origName;
size = fileEntries[fid].size;
}
// 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
int CZipArchive::GetFileImpl(unsigned int fid, std::vector<std::uint8_t>& buffer)
{
// Prevent opening files on missing/invalid archives
if (zip == nullptr)
return -4;
// assert(archiveLock.locked());
assert(IsFileId(fid));
unzGoToFilePos(zip, &fileEntries[fid].fp);
unz_file_info fi;
unzGetCurrentFileInfo(zip, &fi, nullptr, 0, nullptr, 0, nullptr, 0);
if (unzOpenCurrentFile(zip) != UNZ_OK)
return -3;
buffer.clear();
buffer.resize(fi.uncompressed_size);
int ret = 1;
if (!buffer.empty() && unzReadCurrentFile(zip, buffer.data(), buffer.size()) != buffer.size())
ret -= 2;
if (unzCloseCurrentFile(zip) == UNZ_CRCERROR)
ret -= 1;
if (ret != 1)
buffer.clear();
return ret;
}
|
