mirror of
https://github.com/vcmi/vcmi.git
synced 2024-11-30 08:57:00 +02:00
121 lines
2.9 KiB
C++
121 lines
2.9 KiB
C++
/*
|
|
* CZipSaver.cpp, 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
|
|
*
|
|
*/
|
|
|
|
#include "StdInc.h"
|
|
#include "CZipSaver.h"
|
|
|
|
///CZipOutputStream
|
|
CZipOutputStream::CZipOutputStream(CZipSaver * owner_, zipFile archive, const std::string & archiveFilename):
|
|
handle(archive),
|
|
owner(owner_)
|
|
{
|
|
zip_fileinfo fileInfo;
|
|
|
|
std::time_t t = time(nullptr);
|
|
fileInfo.dosDate = 0;
|
|
|
|
struct tm * localTime = std::localtime(&t);
|
|
fileInfo.tmz_date.tm_hour = localTime->tm_hour;
|
|
fileInfo.tmz_date.tm_mday = localTime->tm_mday;
|
|
fileInfo.tmz_date.tm_min = localTime->tm_min;
|
|
fileInfo.tmz_date.tm_mon = localTime->tm_mon;
|
|
fileInfo.tmz_date.tm_sec = localTime->tm_sec;
|
|
fileInfo.tmz_date.tm_year = localTime->tm_year;
|
|
|
|
fileInfo.external_fa = 0; //???
|
|
fileInfo.internal_fa = 0;
|
|
|
|
int status = zipOpenNewFileInZip4_64(
|
|
handle,
|
|
archiveFilename.c_str(),
|
|
&fileInfo,
|
|
nullptr,//extrafield_local
|
|
0,
|
|
nullptr,//extrafield_global
|
|
0,
|
|
nullptr,//comment
|
|
Z_DEFLATED,
|
|
Z_DEFAULT_COMPRESSION,
|
|
0,//raw
|
|
-15,//windowBits
|
|
9,//memLevel
|
|
Z_DEFAULT_STRATEGY,//strategy
|
|
nullptr,//password
|
|
0,//crcForCrypting
|
|
20,//versionMadeBy
|
|
0,//flagBase
|
|
0//zip64
|
|
);
|
|
|
|
if(status != ZIP_OK)
|
|
throw new std::runtime_error("CZipOutputStream: zipOpenNewFileInZip failed");
|
|
|
|
owner->activeStream = this;
|
|
}
|
|
|
|
CZipOutputStream::~CZipOutputStream()
|
|
{
|
|
int status = zipCloseFileInZip(handle);
|
|
if (status != ZIP_OK)
|
|
logGlobal->errorStream() << "CZipOutputStream: stream finalize failed: "<<status;
|
|
owner->activeStream = nullptr;
|
|
}
|
|
|
|
si64 CZipOutputStream::write(const ui8 * data, si64 size)
|
|
{
|
|
int ret = zipWriteInFileInZip(handle, (const void*)data, (unsigned)size);
|
|
|
|
if (ret == ZIP_OK)
|
|
return size;
|
|
else
|
|
return 0;
|
|
}
|
|
|
|
///CZipSaver
|
|
CZipSaver::CZipSaver(std::shared_ptr<CIOApi> api, const std::string & path):
|
|
ioApi(api),
|
|
zipApi(ioApi->getApiStructure()),
|
|
handle(nullptr),
|
|
activeStream(nullptr)
|
|
{
|
|
handle = zipOpen2_64(path.c_str(), APPEND_STATUS_CREATE, nullptr, &zipApi);
|
|
|
|
if (handle == nullptr)
|
|
throw new std::runtime_error("CZipSaver: Failed to create archive");
|
|
}
|
|
|
|
CZipSaver::~CZipSaver()
|
|
{
|
|
if(activeStream != nullptr)
|
|
{
|
|
logGlobal->error("CZipSaver::~CZipSaver: active stream found");
|
|
zipCloseFileInZip(handle);
|
|
}
|
|
|
|
|
|
if(handle != nullptr)
|
|
{
|
|
int status = zipClose(handle, nullptr);
|
|
if (status != ZIP_OK)
|
|
logGlobal->errorStream() << "CZipSaver: archive finalize failed: "<<status;
|
|
}
|
|
|
|
}
|
|
|
|
std::unique_ptr<COutputStream> CZipSaver::addFile(const std::string & archiveFilename)
|
|
{
|
|
if(activeStream != nullptr)
|
|
throw new std::runtime_error("CZipSaver::addFile: stream already opened");
|
|
|
|
std::unique_ptr<COutputStream> stream(new CZipOutputStream(this, handle, archiveFilename));
|
|
return std::move(stream);
|
|
}
|
|
|