當前位置：首頁 > 运维知识 > windows >内容正文

windows

C/C++ Zlib库封装MyZip压缩类

發布時間：2023/11/30 windows 49 coder

生活随笔收集整理的這篇文章主要介紹了 C/C++ Zlib库封装MyZip压缩类小編覺得挺不錯的,現在分享給大家,幫大家做個參考.

Zlib是一個開源的數據壓縮庫，提供了一種通用的數據壓縮和解壓縮算法。它最初由Jean-Loup Gailly和Mark Adler開發，旨在成為一個高效、輕量級的壓縮庫，其被廣泛應用于許多領域，包括網絡通信、文件壓縮、數據庫系統等。其壓縮算法是基于DEFLATE算法，這是一種無損數據壓縮算法，通常能夠提供相當高的壓縮比。

在軟件開發中，文件的壓縮和解壓縮是一項常見的任務，而ZIP是一種被廣泛應用的壓縮格式。為了方便地處理ZIP壓縮和解壓縮操作，開發者通常使用各種編程語言和庫來實現這些功能。本文將聚焦于一個簡化的C++實現，通過分析代碼，我們將深入了解其設計和實現細節。

類的功能實現

MyZip類旨在提供簡單易用的ZIP壓縮和解壓縮功能。通過成員函數Compress和UnCompress，該類使得對目錄的ZIP壓縮和ZIP文件的解壓變得相對容易。

ZIP壓縮函數 Compress

Compress函數通過zlib庫提供的ZIP壓縮功能，遞歸地將目錄下的文件添加到ZIP文件中。其中，nyCollectfileInDirtoZip函數負責遍歷目錄，而nyAddfiletoZip函數則用于添加文件到ZIP中。這種設計使得代碼模塊化，易于理解。

ZIP解壓函數 UnCompress

UnCompress函數通過zlib庫提供的ZIP解壓功能，將ZIP文件解壓到指定目錄。函數中使用了unz系列函數來遍歷ZIP文件中的文件信息，并根據文件類型進行相應的處理。這包括創建目錄和寫入文件，使得解壓后的目錄結構與ZIP文件一致。

將如上的壓縮與解壓方法封裝成MyZip類，調用zip.Compress()實現壓縮目錄，調用zip.UnCompress()則實現解壓縮目錄。這些函數使用了zlib庫的ZIP壓縮和解壓縮功能，并可以在項目中被應用，該類代碼如下所示；

#define ZLIB_WINAPI
#include <string>
#include <iostream>
#include <vector>
#include <Shlwapi.h> 
#include <zip.h>
#include <unzip.h>
#include <zlib.h>

using namespace std;

#pragma comment(lib, "Shlwapi.lib")
#pragma comment(lib, "zlibstat.lib")

class MyZip
{
private:
  // 向ZIP文件中添加文件
  bool nyAddfiletoZip(zipFile zfile, const std::string& fileNameinZip, const std::string& srcfile)
  {
    if (NULL == zfile || fileNameinZip.empty())
    {
      return false;
    }

    int nErr = 0;
    zip_fileinfo zinfo = { 0 };
    tm_zip tmz = { 0 };
    zinfo.tmz_date = tmz;
    zinfo.dosDate = 0;
    zinfo.internal_fa = 0;
    zinfo.external_fa = 0;

    // 構建新文件名
    char sznewfileName[MAX_PATH] = { 0 };
    memset(sznewfileName, 0x00, sizeof(sznewfileName));
    strcat_s(sznewfileName, fileNameinZip.c_str());
    if (srcfile.empty())
    {
      strcat_s(sznewfileName, "\\");
    }

    // 在ZIP中打開新文件
    nErr = zipOpenNewFileInZip(zfile, sznewfileName, &zinfo, NULL, 0, NULL, 0, NULL, Z_DEFLATED, Z_DEFAULT_COMPRESSION);
    if (nErr != ZIP_OK)
    {
      return false;
    }

    // 如果有源文件，讀取并寫入ZIP文件
    if (!srcfile.empty())
    {
      FILE* srcfp = _fsopen(srcfile.c_str(), "rb", _SH_DENYNO);
      if (NULL == srcfp)
      {
        return false;
      }

      int numBytes = 0;
      char* pBuf = new char[1024 * 100];
      if (NULL == pBuf)
      {
        return false;
      }

      // 逐塊讀取源文件并寫入ZIP
      while (!feof(srcfp))
      {
        memset(pBuf, 0x00, sizeof(pBuf));
        numBytes = fread(pBuf, 1, sizeof(pBuf), srcfp);
        nErr = zipWriteInFileInZip(zfile, pBuf, numBytes);
        if (ferror(srcfp))
        {
          break;
        }
      }

      delete[] pBuf;
      fclose(srcfp);
    }

    // 關閉ZIP文件中的當前文件
    zipCloseFileInZip(zfile);

    return true;
  }

  // 遞歸地將目錄下的文件添加到ZIP
  bool nyCollectfileInDirtoZip(zipFile zfile, const std::string& filepath, const std::string& parentdirName)
  {
    if (NULL == zfile || filepath.empty())
    {
      return false;
    }

    bool bFile = false;
    std::string relativepath = "";
    WIN32_FIND_DATAA findFileData;

    char szpath[MAX_PATH] = { 0 };
    if (::PathIsDirectoryA(filepath.c_str()))
    {
      strcpy_s(szpath, sizeof(szpath) / sizeof(szpath[0]), filepath.c_str());
      int len = strlen(szpath) + strlen("\\*.*") + 1;
      strcat_s(szpath, len, "\\*.*");
    }
    else
    {
      bFile = true;
      strcpy_s(szpath, sizeof(szpath) / sizeof(szpath[0]), filepath.c_str());
    }

    HANDLE hFile = ::FindFirstFileA(szpath, &findFileData);
    if (NULL == hFile)
    {
      return false;
    }

    do
    {
      // 構建相對路徑
      if (parentdirName.empty())
        relativepath = findFileData.cFileName;
      else
        relativepath = parentdirName + "\\" + findFileData.cFileName;

      // 如果是目錄，遞歸處理子目錄
      if (findFileData.dwFileAttributes == FILE_ATTRIBUTE_DIRECTORY)
      {
        if (strcmp(findFileData.cFileName, ".") != 0 && strcmp(findFileData.cFileName, "..") != 0)
        {
          nyAddfiletoZip(zfile, relativepath, "");

          char szTemp[MAX_PATH] = { 0 };
          strcpy_s(szTemp, filepath.c_str());
          strcat_s(szTemp, "\\");
          strcat_s(szTemp, findFileData.cFileName);
          nyCollectfileInDirtoZip(zfile, szTemp, relativepath);
        }
        continue;
      }

      char szTemp[MAX_PATH] = { 0 };
      if (bFile)
      {
        strcpy_s(szTemp, filepath.c_str());
      }
      else
      {
        strcpy_s(szTemp, filepath.c_str());
        strcat_s(szTemp, "\\");
        strcat_s(szTemp, findFileData.cFileName);
      }

      // 將文件添加到ZIP
      nyAddfiletoZip(zfile, relativepath, szTemp);

    } while (::FindNextFileA(hFile, &findFileData));

    FindClose(hFile);

    return true;
  }

  // 替換字符串中的所有指定子串
  std::string& replace_all(std::string& str, const std::string& old_value, const std::string& new_value)
  {
    while (true)
    {
      std::string::size_type pos(0);
      if ((pos = str.find(old_value)) != std::string::npos)
        str.replace(pos, old_value.length(), new_value);
      else
        break;
    }
    return str;
  }

  // 創建多級目錄
  BOOL CreatedMultipleDirectory(const std::string& direct)
  {
    std::string Directoryname = direct;
    if (Directoryname[Directoryname.length() - 1] != '\\')
    {
      Directoryname.append(1, '\\');
    }

    std::vector< std::string> vpath;
    std::string strtemp;
    BOOL  bSuccess = FALSE;

    // 遍歷目錄字符串，逐級創建目錄
    for (int i = 0; i < Directoryname.length(); i++)
    {
      if (Directoryname[i] != '\\')
      {
        strtemp.append(1, Directoryname[i]);
      }
      else
      {
        vpath.push_back(strtemp);
        strtemp.append(1, '\\');
      }
    }

    std::vector< std::string>::iterator vIter = vpath.begin();
    for (; vIter != vpath.end(); vIter++)
    {
      bSuccess = CreateDirectoryA(vIter->c_str(), NULL) ? TRUE : FALSE;
    }

    return bSuccess;
  }

public:
  // 壓縮目錄
  bool Compress(const std::string& dirpathName, const std::string& zipfileName, const std::string& parentdirName)
  {
    bool bRet = false;
    zipFile zFile = NULL;

    // 根據ZIP文件是否存在選擇打開方式
    if (!::PathFileExistsA(zipfileName.c_str()))
    {
      zFile = zipOpen(zipfileName.c_str(), APPEND_STATUS_CREATE);
    }
    else
    {
      zFile = zipOpen(zipfileName.c_str(), APPEND_STATUS_ADDINZIP);
    }

    if (NULL == zFile)
    {
      return bRet;
    }

    // 將目錄下的文件添加到ZIP
    if (nyCollectfileInDirtoZip(zFile, dirpathName, parentdirName))
    {
      bRet = true;
    }

    zipClose(zFile, NULL);

    return bRet;
  }

  // 解壓目錄
  bool UnCompress(const std::string& strFilePath, const std::string& strTempPath)
  {
    int nReturnValue;
    string tempFilePath;
    string srcFilePath(strFilePath);
    string destFilePath;

    // 打開ZIP文件
    unzFile unzfile = unzOpen(srcFilePath.c_str());
    if (unzfile == NULL)
    {
      return false;
    }

    unz_global_info* pGlobalInfo = new unz_global_info;
    nReturnValue = unzGetGlobalInfo(unzfile, pGlobalInfo);
    if (nReturnValue != UNZ_OK)
    {
      return false;
    }

    unz_file_info* pFileInfo = new unz_file_info;
    char szZipFName[MAX_PATH] = { 0 };
    char szExtraName[MAX_PATH] = { 0 };
    char szCommName[MAX_PATH] = { 0 };

    for (int i = 0; i < pGlobalInfo->number_entry; i++)
    {
      nReturnValue = unzGetCurrentFileInfo(unzfile, pFileInfo, szZipFName, MAX_PATH, szExtraName, MAX_PATH, szCommName, MAX_PATH);
      if (nReturnValue != UNZ_OK)
        return false;

      string strZipFName = szZipFName;

      // 如果是目錄，創建相應目錄
      if (pFileInfo->external_fa == FILE_ATTRIBUTE_DIRECTORY || (strZipFName.rfind('/') == strZipFName.length() - 1))
      {
        destFilePath = strTempPath + "http://" + szZipFName;
        CreateDirectoryA(destFilePath.c_str(), NULL);
      }
      else
      {
        string strFullFilePath;
        tempFilePath = strTempPath + "/" + szZipFName;
        strFullFilePath = tempFilePath;

        int nPos = tempFilePath.rfind("/");
        int nPosRev = tempFilePath.rfind("\\");
        if (nPosRev == string::npos && nPos == string::npos)
          continue;

        size_t nSplitPos = nPos > nPosRev ? nPos : nPosRev;
        destFilePath = tempFilePath.substr(0, nSplitPos + 1);

        // 創建多級目錄
        if (!PathIsDirectoryA(destFilePath.c_str()))
        {
          destFilePath = replace_all(destFilePath, "/", "\\");
          int bRet = CreatedMultipleDirectory(destFilePath);
        }

        strFullFilePath = replace_all(strFullFilePath, "/", "\\");

        // 創建文件并寫入數據
        HANDLE hFile = CreateFileA(strFullFilePath.c_str(), GENERIC_WRITE, 0, NULL, OPEN_ALWAYS, FILE_FLAG_WRITE_THROUGH, NULL);
        if (hFile == INVALID_HANDLE_VALUE)
        {
          return false;
        }

        nReturnValue = unzOpenCurrentFile(unzfile);
        if (nReturnValue != UNZ_OK)
        {
          CloseHandle(hFile);
          return false;
        }

        uLong BUFFER_SIZE = pFileInfo->uncompressed_size;
        void* szReadBuffer = NULL;
        szReadBuffer = (char*)malloc(BUFFER_SIZE);
        if (NULL == szReadBuffer)
        {
          break;
        }

        // 逐塊讀取ZIP文件并寫入目標文件
        while (TRUE)
        {
          memset(szReadBuffer, 0, BUFFER_SIZE);
          int nReadFileSize = 0;

          nReadFileSize = unzReadCurrentFile(unzfile, szReadBuffer, BUFFER_SIZE);

          if (nReadFileSize < 0)
          {
            unzCloseCurrentFile(unzfile);
            CloseHandle(hFile);
            return false;
          }
          else if (nReadFileSize == 0)
          {
            unzCloseCurrentFile(unzfile);
            CloseHandle(hFile);
            break;
          }
          else
          {
            DWORD dWrite = 0;
            BOOL bWriteSuccessed = WriteFile(hFile, szReadBuffer, BUFFER_SIZE, &dWrite, NULL);
            if (!bWriteSuccessed)
            {
              unzCloseCurrentFile(unzfile);
              CloseHandle(hFile);
              return false;
            }
          }
        }

        free(szReadBuffer);
      }
      unzGoToNextFile(unzfile);
    }

    delete pFileInfo;
    delete pGlobalInfo;

    if (unzfile)
    {
      unzClose(unzfile);
    }

    return true;
  }
};

如何使用類

壓縮文件時可以通過調用zip.Compress()函數實現，該函數接受3個參數，第一個參數是需要壓縮的目錄名，第二個參數是壓縮后保存的文件名，第三個參數則是壓縮后主目錄的名字，我們以壓縮D:\\csdn目錄下的所有文件為例，代碼如下所示；

int main(int argc, char* argv[])
{
	MyZip zip;

	// 壓縮目錄
	std::string compress_src = "D:\\csdn";                               // 壓縮目錄
	std::string compress_dst = "D:\\test.zip";                           // 壓縮后

	bool compress_flag = zip.Compress(compress_src, compress_dst, "lyshark");
	std::cout << "壓縮狀態: " << compress_flag << std::endl;

	system("pause");
	return 0;
}

壓縮后可以看到對應的壓縮包內容，如下所示；

解壓縮與壓縮類似，通過調用zip.UnCompress實現，該方法需要傳入兩個參數，被壓縮的文件名和解壓到的目錄名，如果目錄不存在則會創建并解壓。

int main(int argc, char* argv[])
{
	MyZip zip;

	// 解壓縮目錄
	std::string uncompress_src = "D:\\test.zip";                      // 被解壓文件
	std::string uncompress_dst = "D:\\dst";                           // 解壓到

	bool compress_flag = zip.UnCompress(uncompress_src, uncompress_dst);
	std::cout << "解壓縮狀態: " << compress_flag << std::endl;

	system("pause");
	return 0;
}

輸出效果如下所示；

總結

以上是生活随笔為你收集整理的C/C++ Zlib库封装MyZip压缩类的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。

zlib
MyZip

上一篇：图文剖析 big.js 四则运算源码
下一篇：回归算法全解析！一文读懂机器学习中的回归

日韩av黄I国产麻豆传媒I国产91av视频在线观看I日韩一区二区三区在线看I美女国产在线I麻豆视频国产在线观看I成人黄色短片

windows

C/C++ Zlib库封装MyZip压缩类

類的功能實現

如何使用類

總結