使用quazip在内存中压缩和解压

quazip是一个QT下实现的压缩和解压库

下载地址：

Releases · stachenov/quazip · GitHub

quazip已经是带了内存中解压和压缩的功能，但是没有封装到JlCompress

在编译quazip时需要JlCompress类中的static bool compressFile(QuaZip* zip, QString fileName, QString fileDest);等几个private函数修改为protected，因为我是继承了JlCompress类，需要使用bool compressFile(QuaZip* zip, QString fileName, QString fileDest);函数。

下面是我实现的JlCompressEx类，

头文件内容

#pragma once


#include "quazip/JlCompress.h"
#include "quazip/quazip.h"

#include <QBuffer>
#include <QByteArray>

class JlCompressEx : public JlCompress
{
public:
    // 文件压缩到内存buffer，zipIoDevice可以使用QBuffer zipBuffer;
    static bool CompressToBuffer(QString file, const char *password, QIODevice& zipIoDevice);
    // 内存数据压缩到内存buffer
    static bool CompressToBuffer(QByteArray& sourceData, QString fileName, const char *password, QIODevice& zipIoDevice);
    // 文件压缩到内存buffer
    static bool CompressToBuffer(QStringList files, const char *password, QIODevice& zipIoDevice);
    /**
      \ 重写JlCompress::compressFile,添加密码参数
      \param zip Opened zip to compress the file to.
      \param fileName The full path to the source file.
      \param fileDest The full name of the file inside the archive.
      \return true if success, false otherwise.
    */
    static bool compressFileEx(QuaZip* zip, QString fileName, QString fileDest, const char *password);
public:
    // 压缩数据
    static bool CompressBuffer(QuaZip& zip, QByteArray& sourceData, QString fileName, const char *password);
    
public:
    // 解压，解压出来所有文件数据保存在filesList<文件名, 文件内容>
    static bool extractToBuffer(const QString& strZipFile, const char *password, QList<QPair<QString,QByteArray>>& filesList);
};

cpp文件内容 

#include "JlCompressEx.h"

#include <QDebug>


bool JlCompressEx::CompressToBuffer(QString file, const char *password, QIODevice& zipIoDevice)
{
    QuaZip zip(&zipIoDevice);
    //QDir().mkpath(QFileInfo(fileCompressed).absolutePath());
    if (!zip.open(QuaZip::mdCreate))
    {
        return false;
    }
    // Aggiungo il file
    if (!compressFileEx(&zip, file, QFileInfo(file).fileName(), password)) {
        return false;
    }

    // Chiudo il file zip
    zip.close();
    if (zip.getZipError() != UNZ_OK) {
        return false;
    }

    return true;
}

bool JlCompressEx::CompressToBuffer(QByteArray& sourceData, QString fileName, const char *password, QIODevice& zipIoDevice)
{
    QuaZip zip(&zipIoDevice);
    //QDir().mkpath(QFileInfo(fileCompressed).absolutePath());
    if (!zip.open(QuaZip::mdCreate))
    {
        return false;
    }

    if (!CompressBuffer(zip, sourceData, fileName, password))
    {
        return false;
    }

    // Chiudo il file zip
    zip.close();
    if (zip.getZipError() != UNZ_OK) {
        return false;
    }

    return true;
}

bool JlCompressEx::CompressToBuffer(QStringList files, const char *password, QIODevice& zipIoDevice)
{
    // Creo lo zip
    QuaZip zip(&zipIoDevice);
    //QDir().mkpath(QFileInfo(fileCompressed).absolutePath());
    if (!zip.open(QuaZip::mdCreate))
    {
        return false;
    }

    // Comprimo i file
    QFileInfo info;
    for (int index = 0; index < files.size(); ++index) {
        const QString & file(files.at(index));
        info.setFile(file);
        if (!info.exists() || !compressFileEx(&zip, file, info.fileName(), password)) {
            return false;
        }
    }

    // Chiudo il file zip
    zip.close();
    if (zip.getZipError() != 0)
    {
        return false;
    }

    return true;
}

bool JlCompressEx::compressFileEx(QuaZip *zip, QString fileName, QString fileDest, const char *password)
{
    // zip: oggetto dove aggiungere il file
    // fileName: nome del file reale
    // fileDest: nome del file all'interno del file compresso

    // Controllo l'apertura dello zip
    if (!zip) return false;
    if (zip->getMode()!=QuaZip::mdCreate &&
            zip->getMode()!=QuaZip::mdAppend &&
            zip->getMode()!=QuaZip::mdAdd) return false;

    // Apro il file originale
    QFile inFile;
    inFile.setFileName(fileName);
    if(!inFile.open(QIODevice::ReadOnly)) return false;

    // Apro il file risulato
    QuaZipFile outFile(zip);
    if(!outFile.open(QIODevice::WriteOnly, QuaZipNewInfo(fileDest, inFile.fileName()), password)) return false;

    // Copio i dati
    if (!copyData(inFile, outFile) || outFile.getZipError()!=UNZ_OK) {
        return false;
    }

    // Chiudo i file
    outFile.close();
    if (outFile.getZipError()!=UNZ_OK) return false;
    inFile.close();

    return true;
}

bool JlCompressEx::CompressBuffer(QuaZip& zip, QByteArray& sourceData, QString fileName, const char *password)
{
    // Controllo l'apertura dello zip
    if (zip.getMode() != QuaZip::mdCreate &&
            zip.getMode() != QuaZip::mdAppend &&
            zip.getMode() != QuaZip::mdAdd)
    {
        return false;
    }
    // Apro il file risulato
    QuaZipFile outFile(&zip);
    if (!outFile.open(QIODevice::WriteOnly, QuaZipNewInfo(fileName), password))
    {
        return false;
    }
    // Copio i dati
    if (outFile.write(sourceData) != sourceData.size())
    {
        return false;
    }
    if (outFile.getZipError() != UNZ_OK)
    {
        return false;
    }

    // Chiudo i file
    outFile.close();
    if (outFile.getZipError() != UNZ_OK)
    {
        return false;
    }

    return true;
}

bool JlCompressEx::extractToBuffer(const QString& strZipFile, const char *password, QList<QPair<QString,QByteArray>>& filesList)
{
    QuaZip zip(strZipFile);
    if (!zip.open(QuaZip::mdUnzip)) {
        qWarning() << "open zip file error:" << zip.getZipError() << ". file:" << strZipFile;
        return false;
    }
    for (bool more = zip.goToFirstFile(); more; more = zip.goToNextFile())
    {
        QuaZipFileInfo64 info;
        if (!zip.getCurrentFileInfo(&info))
        {
            continue;
        }
        qDebug() << "in zip: fileName:" << info.name << ", compressedSize:" << info.compressedSize  << ", uncompressedSize:" << info.uncompressedSize;
        QString strFileName = info.name;    // 如果为多层目录则包含目录路径

        if(0 == info.compressedSize || 0 == info.uncompressedSize)
        {// 文件夹也会遍历出来，大小为0
            continue;
        }

        // 读取文件数据
        QuaZipFile zipFile(&zip);
        if (!zipFile.open(QIODevice::ReadOnly, password))
        {
            qWarning() << "open file error:" << zipFile.getZipError() << ". file name:" << strFileName;
            continue;
        }
        if (zipFile.getZipError() != UNZ_OK)
        {
            qWarning() << "open file error:" << zipFile.getZipError();
            continue;
        }
        //qDebug() << "file size:" << zipFile.size() << " isOpen:" << zipFile.isOpen();
        //continue;
        QByteArray fileData;
        while (!zipFile.atEnd())
        {
            QByteArray dataTmp = zipFile.readAll();
            if(dataTmp.isEmpty())
            {// 当密码不正确时打开文件会成功，这里会读取失败
                zipFile.close();
                zip.close();
                return false;
            }
            if (zipFile.getZipError() != UNZ_OK)
            {
                zipFile.close();
                zip.close();
                return false;
            }
            fileData += dataTmp;
        }
        filesList.append(QPair<QString,QByteArray>(strFileName, fileData));

        zipFile.close();
    }
    //
    zip.close();
    if (zip.getZipError() != 0) {

    }
    return true;
}