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
|
/*
* CZipLoader.h, part of VCMI engine
*
* Authors: listed in file AUTHORS in main folder
*
* License: GNU General Public License v2.0 or later
* Full text of license available in license.txt file, in main folder
*
*/
#pragma once
#include "ISimpleResourceLoader.h"
#include "CInputStream.h"
#include "ResourcePath.h"
#include "CCompressedStream.h"
#include "MinizipExtensions.h"
VCMI_LIB_NAMESPACE_BEGIN
class DLL_LINKAGE CZipStream : public CBufferedStream
{
unzFile file;
public:
/**
* @brief constructs zip stream from already opened file
* @param api virtual filesystem interface
* @param archive path to archive to open
* @param filepos position of file to open
*/
CZipStream(const std::shared_ptr<CIOApi> & api, const boost::filesystem::path & archive, unz64_file_pos filepos);
~CZipStream();
si64 getSize() override;
ui32 calculateCRC32() override;
protected:
si64 readMore(ui8 * data, si64 size) override;
};
class DLL_LINKAGE CZipLoader : public ISimpleResourceLoader
{
std::shared_ptr<CIOApi> ioApi;
zlib_filefunc64_def zlibApi;
boost::filesystem::path archiveName;
std::string mountPoint;
std::unordered_map<ResourcePath, unz64_file_pos> files;
std::unordered_map<ResourcePath, unz64_file_pos> listFiles(const std::string & mountPoint, const boost::filesystem::path &archive);
public:
CZipLoader(const std::string & mountPoint, const boost::filesystem::path & archive, std::shared_ptr<CIOApi> api = std::make_shared<CDefaultIOApi>());
/// Interface implementation
/// @see ISimpleResourceLoader
std::unique_ptr<CInputStream> load(const ResourcePath & resourceName) const override;
bool existsResource(const ResourcePath & resourceName) const override;
std::string getMountPoint() const override;
void updateFilteredFiles(std::function<bool(const std::string &)> filter) const override {}
std::unordered_set<ResourcePath> getFilteredFiles(std::function<bool(const ResourcePath &)> filter) const override;
};
class DLL_LINKAGE ZipArchive : boost::noncopyable
{
unzFile archive;
public:
ZipArchive(const boost::filesystem::path & from);
~ZipArchive();
std::vector<std::string> listFiles();
bool extract(const boost::filesystem::path & where, const std::vector<std::string> & what);
bool extract(const boost::filesystem::path & where, const std::string & what);
};
VCMI_LIB_NAMESPACE_END
|
