使用POCO库进行ZIP压缩和解压
POCO C++ Libraries提供了一个ZIP模块，可以方便地进行文件和数据流的压缩与解压操作。下面我将介绍如何使用POCO的ZIP模块进行这些操作。
1. 基本ZIP文件操作
压缩文件/目录到ZIP
#include <Poco/Zip/Compress.h>
#include <Poco/File.h>
#include <Poco/Path.h>
#include <iostream>
#include <fstream>
void compressToZip(const std::string& source, const std::string& zipFile)
{
try {
std::ofstream out(zipFile, std::ios::binary);
Poco::Zip::Compress c(out, true);
Poco::File f(source);
if (f.isDirectory()) {
c.addRecursive(f, Poco::Zip::ZipCommon::CL_MAXIMUM, f.path());
} else {
c.addFile(f, f.path());
}
c.close(); // 必须调用close()完成ZIP文件
out.close();
std::cout << "Successfully created ZIP file: " << zipFile << std::endl;
} catch (Poco::Exception& e) {
std::cerr << "Error: " << e.displayText() << std::endl;
}
}
// 使用示例
// compressToZip("my_folder", "output.zip");
// compressToZip("file.txt", "file.zip");
解压ZIP文件
#include <Poco/Zip/Decompress.h>
#include <Poco/Path.h>
#include <iostream>
#include <fstream>
void decompressZip(const std::string& zipFile, const std::string& destDir)
{
try {
std::ifstream in(zipFile, std::ios::binary);
Poco::Zip::Decompress d(in, Poco::Path(destDir));
d.decompressAllFiles();
in.close();
std::cout << "Successfully extracted ZIP file to: " << destDir << std::endl;
} catch (Poco::Exception& e) {
std::cerr << "Error: " << e.displayText() << std::endl;
}
}
// 使用示例
// decompressZip("archive.zip", "extracted_files");
2. 流数据压缩和解压
POCO还提供了对数据流进行压缩和解压的功能，使用的是Poco::Zip::ZipStream。
压缩数据流
#include <Poco/Zip/ZipStream.h>
#include <Poco/Zip/ZipArchive.h>
#include <Poco/Zip/ZipLocalFileHeader.h>
#include <iostream>
#include <sstream>
#include <fstream>
void compressStreamToZip(const std::string& outputFile)
{
try {
std::ofstream out(outputFile, std::ios::binary);
Poco::Zip::ZipOutputStream zipOut(out);
// 添加第一个文件到ZIP
zipOut.putNextEntry("file1.txt");
std::string content1 = "This is the content of file1.txt";
zipOut << content1;
// 添加第二个文件到ZIP
zipOut.putNextEntry("subdir/file2.txt");
std::string content2 = "This is file2.txt in a subdirectory";
zipOut << content2;
zipOut.close();
out.close();
std::cout << "Successfully created ZIP from streams: " << outputFile << std::endl;
} catch (Poco::Exception& e) {
std::cerr << "Error: " << e.displayText() << std::endl;
}
}
// 使用示例
// compressStreamToZip("stream_output.zip");
解压数据流
#include <Poco/Zip/ZipStream.h>
#include <Poco/Zip/ZipArchive.h>
#include <iostream>
#include <fstream>
#include <sstream>
void decompressZipToStreams(const std::string& zipFile)
{
try {
std::ifstream in(zipFile, std::ios::binary);
Poco::Zip::ZipArchive archive(in);
for (auto it = archive.headerBegin(); it != archive.headerEnd(); ++it) {
const Poco::Zip::ZipLocalFileHeader& header = *it;
if (header.isFile()) {
std::cout << "Extracting: " << header.getFileName() << std::endl;
std::istream* pStream = archive.findStream(header.getFileName());
if (pStream) {
std::stringstream content;
Poco::StreamCopier::copyStream(*pStream, content);
std::cout << "Content:\n" << content.str() << "\n" << std::endl;
}
}
}
in.close();
} catch (Poco::Exception& e) {
std::cerr << "Error: " << e.displayText() << std::endl;
}
}
// 使用示例
// decompressZipToStreams("stream_output.zip");
3. 内存中的ZIP操作
在内存中创建ZIP
#include <Poco/Zip/ZipStream.h>
#include <Poco/Zip/ZipArchive.h>
#include <iostream>
#include <sstream>
void createZipInMemory()
{
try {
std::stringstream outStream(std::ios::in | std::ios::out | std::ios::binary);
Poco::Zip::ZipOutputStream zipOut(outStream);
// 添加文件到内存中的ZIP
zipOut.putNextEntry("memory_file.txt");
std::string content = "This file was created in memory";
zipOut << content;
zipOut.close();
// 获取ZIP数据
std::string zipData = outStream.str();
std::cout << "Created ZIP in memory, size: " << zipData.size() << " bytes" << std::endl;
// 可以将zipData保存到文件或通过网络发送
// std::ofstream("memory.zip", std::ios::binary) << zipData;
} catch (Poco::Exception& e) {
std::cerr << "Error: " << e.displayText() << std::endl;
}
}
// 使用示例
// createZipInMemory();
从内存中读取ZIP
#include <Poco/Zip/ZipArchive.h>
#include <iostream>
#include <sstream>
#include <vector>
void readZipFromMemory(const std::vector<char>& zipData)
{
try {
std::stringstream inStream(std::ios::in | std::ios::out | std::ios::binary);
inStream.write(zipData.data(), zipData.size());
Poco::Zip::ZipArchive archive(inStream);
for (auto it = archive.headerBegin(); it != archive.headerEnd(); ++it) {
const Poco::Zip::ZipLocalFileHeader& header = *it;
if (header.isFile()) {
std::cout << "Found file in memory ZIP: " << header.getFileName() << std::endl;
std::istream* pStream = archive.findStream(header.getFileName());
if (pStream) {
std::stringstream content;
Poco::StreamCopier::copyStream(*pStream, content);
std::cout << "Content:\n" << content.str() << std::endl;
}
}
}
} catch (Poco::Exception& e) {
std::cerr << "Error: " << e.displayText() << std::endl;
}
}
// 使用示例
/*
std::vector<char> zipData = ...; // 从文件或网络获取的ZIP数据
readZipFromMemory(zipData);
*/
注意事项
- 确保在项目中正确链接POCO的Foundation和Zip库。
- 压缩或解压大文件时，考虑使用缓冲区以提高性能。
- 处理ZIP文件时要注意路径安全问题，特别是解压来自不可信源的ZIP文件时。
- 所有示例代码都需要包含相应的POCO头文件，并处理可能的异常。
以上示例展示了POCO库中ZIP模块的基本用法，你可以根据实际需求进行调整和扩展。