118 lines
3.8 KiB
C++
118 lines
3.8 KiB
C++
#include "CCZip.h"
|
|
#include <cstring>
|
|
#include "filesystem"
|
|
#include "vector"
|
|
#include "fstream"
|
|
|
|
namespace fs_t = std::filesystem;
|
|
|
|
bool CTL::Zip::UnzipFile(const std::string &zipPath, const std::string &outputDir) {
|
|
// 打开 ZIP 文件
|
|
const unzFile zipfile = unzOpen(zipPath.c_str());
|
|
if (!zipfile) return false;
|
|
|
|
// 获取全局信息
|
|
unz_global_info global_info;
|
|
if (unzGetGlobalInfo(zipfile, &global_info) != UNZ_OK) {
|
|
unzClose(zipfile);
|
|
return false;
|
|
}
|
|
|
|
// 遍历所有文件
|
|
char filename[256]{};
|
|
|
|
for (uLong i = 0; i < global_info.number_entry; ++i) {
|
|
unz_file_info file_info;
|
|
if (unzGetCurrentFileInfo(zipfile, &file_info, filename, sizeof(filename),
|
|
nullptr, 0, nullptr, 0) != UNZ_OK) {
|
|
break;
|
|
}
|
|
|
|
// 构建输出路径
|
|
std::string fullPath = outputDir + "/" + filename;
|
|
|
|
// 检查是否为目录
|
|
if (filename[strlen(filename) - 1] == '/') {
|
|
// 创建目录（需要平台相关实现）
|
|
fs_t::create_directories(fullPath);
|
|
}
|
|
else {
|
|
// 打开当前文件
|
|
if (unzOpenCurrentFile(zipfile) != UNZ_OK) break;
|
|
|
|
// 创建输出文件
|
|
FILE* out = fopen(fullPath.c_str(), "wb");
|
|
if (!out) break;
|
|
|
|
// 读取并写入文件内容
|
|
int bytesRead;
|
|
do {
|
|
unsigned char buffer[FILE_BUFFER_SIZE];
|
|
bytesRead = unzReadCurrentFile(zipfile, buffer, FILE_BUFFER_SIZE);
|
|
if (bytesRead > 0) {
|
|
fwrite(buffer, 1, bytesRead, out);
|
|
}
|
|
} while (bytesRead > 0);
|
|
|
|
fclose(out);
|
|
unzCloseCurrentFile(zipfile);
|
|
}
|
|
|
|
// 移动到下一个文件
|
|
if (i + 1 < global_info.number_entry) {
|
|
if (unzGoToNextFile(zipfile) != UNZ_OK) break;
|
|
}
|
|
}
|
|
|
|
unzClose(zipfile);
|
|
return true;
|
|
}
|
|
|
|
bool CTL::Zip::ZipFolder(const std::string &folderPath, const std::string &zipPath) {
|
|
zipFile zip = zipOpen64(zipPath.c_str(), APPEND_STATUS_CREATE);
|
|
if (!zip) return false;
|
|
|
|
const fs_t::path baseDir(folderPath);
|
|
for (const auto& entry : fs_t::recursive_directory_iterator(baseDir, fs_t::directory_options::follow_directory_symlink)) {
|
|
const fs_t::path& filePath = entry.path();
|
|
const std::string relativePath = fs_t::relative(filePath, baseDir).string();
|
|
|
|
if (fs_t::is_directory(filePath)) {
|
|
// 添加目录条目
|
|
zip_fileinfo zipInfo = {0};
|
|
std::string dirEntry = relativePath + "/";
|
|
zipOpenNewFileInZip(zip, dirEntry.c_str(), &zipInfo,
|
|
nullptr, 0, nullptr, 0, nullptr,
|
|
Z_DEFLATED, Z_DEFAULT_COMPRESSION);
|
|
zipCloseFileInZip(zip);
|
|
} else {
|
|
// 添加文件
|
|
std::ifstream file(filePath, std::ios::binary);
|
|
if (!file.is_open()) continue;
|
|
|
|
// 获取文件大小
|
|
file.seekg(0, std::ios::end);
|
|
const size_t size = file.tellg();
|
|
file.seekg(0, std::ios::beg);
|
|
|
|
// 读取文件内容
|
|
std::vector<char> buffer(size);
|
|
file.read(buffer.data(), size);
|
|
file.close();
|
|
|
|
// 准备 ZIP 文件信息
|
|
zip_fileinfo zipInfo = {0};
|
|
zipOpenNewFileInZip(zip, relativePath.c_str(), &zipInfo,
|
|
nullptr, 0, nullptr, 0, nullptr,
|
|
Z_DEFLATED, Z_DEFAULT_COMPRESSION);
|
|
|
|
// 写入文件内容
|
|
zipWriteInFileInZip(zip, buffer.data(), size);
|
|
zipCloseFileInZip(zip);
|
|
}
|
|
}
|
|
|
|
zipClose(zip, nullptr);
|
|
return true;
|
|
}
|