使用POCO库进行ZIP压缩和解压

news2025/4/13 2:28:12

使用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);
*/