C++ 压缩文件及文件夹方法 使用zlib开源库

#include "stdafx.h"
#include <string>
#include <iostream>
#include <vector>
#include <Shlwapi.h> 
#include "zip.h"
#include "unzip.h"
#include "zlib.h"
#include <stdio.h>
#include <stdlib.h>
#include <fstream>
#include <sstream>
using namespace std;
//部分头文件不需要（自行去掉）
#pragma comment(lib,"Shlwapi.lib")
bool nyAddfiletoZip(zipFile zfile,const std::string& fileNameinZip,const std::string& srcfile);
bool nyCollectfileInDirtoZip(zipFile zfile,const std::string& filepath,const std::string& parentdirName);
bool nyCreateZipfromDir(const std::string& dirpathName,const std::string& zipfileName,const std::string& parentdirName);
int _tmain(int argc,_TCHAR* argv[])
{
	std::string dirpath = "D:RecycleBinwei";			//源文件/文件夹
	std::string zipfileName = "D:RecycleBinlango.rar";	//目的压缩包
	nyCreateZipfromDir(dirpath,zipfileName,"wei");
	system("pause");
	return 0;
}
/*
* 函数功能 :解压zip文件
* 备 注 :参数strFilePath表示zip压缩文件的路径
*			参数strTempPath表示要解压到的文件目录
*/
bool nyAddfiletoZip(zipFile zfile,const std::string& srcfile)
{
	if (NULL == zfile || fileNameinZip.empty()/* || srcfile.empty()为空代表空目录*/)
	{
		return 0;
	}
	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,"");
	}
	nErr = zipOpenNewFileInZip(zfile,sznewfileName,&zinfo,NULL,Z_DEFLATED,Z_DEFAULT_COMPRESSION);
	if (nErr != ZIP_OK)
	{
		return false;
	}
	if (!srcfile.empty())
	{
		//打开源文件
		FILE* srcfp = _fsopen(srcfile.c_str(),"rb",_SH_DENYNO);
		if (NULL == srcfp)
		{
			std::cout << "Open source file failed." << std::endl;
			return false;
		}
		//读入源文件写入zip文件
		int numBytes = 0;
		char* pBuf = new char[1024 * 100];
		if (NULL == pBuf)
		{
			std::cout << "new buffer failed." << std::endl;
			return 0;
		}
		while (!feof(srcfp))
		{
			memset(pBuf,sizeof(pBuf));
			numBytes = fread(pBuf,1,sizeof(pBuf),srcfp);
			nErr = zipWriteInFileInZip(zfile,pBuf,numBytes);
			if (ferror(srcfp))
			{
				break;
			}
		}
		delete[] pBuf;
		fclose(srcfp);
	}
	zipCloseFileInZip(zfile);
	return true;
}
bool nyCollectfileInDirtoZip(zipFile zfile,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,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;//生成zip文件中的相对路径
		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);
		}
		nyAddfiletoZip(zfile,szTemp);
	} while (::FindNextFileA(hFile,&findFileData));
	FindClose(hFile);
	return true;
}
bool nyCreateZipfromDir(const std::string& dirpathName,const std::string& parentdirName)
{
	bool bRet = false;
	
	/***********参数注释*********/
	/*APPEND_STATUS_CREATE		创建追加
	APPEND_STATUS_CREATEAFTER	创建后追加（覆盖方式）
	APPEND_STATUS_ADDINZIP		直接追加*/
	/****************************/
	zipFile zFile = NULL;
	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)
	{
		std::cout << "create zip file failed." << std::endl;
		return bRet;
	}
	if (nyCollectfileInDirtoZip(zFile,dirpathName,parentdirName))
	{
		bRet = true;
	}
	zipClose(zFile,NULL);
	return bRet;
}