myvtm/Module/mod_upload/ZipZilb.cpp

#include "ZipZilb.h"
#include "libtoolkit/path.h"


bool CreateZipFromFile(string srcFileName, string zipFileName)
{
	if (srcFileName.empty() || srcFileName.length() == 0 ||
		zipFileName.empty() || zipFileName.length() == 0) {
		Dbg("压缩文件参数为空：");
		return false;
	}
	int pos = srcFileName.find_last_of(SPLIT_SLASH);

/*
#ifdef RVC_OS_WIN
	int pos = srcFileName.find_last_of('\\');
#else
	int pos = srcFileName.find_last_of('/');
#endif 
*/

	if (pos < 0) {
		Dbg("文件路径不对：%s",srcFileName.c_str());
		return false;
	}
	string strName(srcFileName.substr(pos + 1));


	zipFile newZipFile = zipOpen(zipFileName.c_str(), APPEND_STATUS_CREATE); //创建zip文件
	if (newZipFile == NULL)
	{
		Dbg("无法创建zip文件!");
		return false;
	}
	//
	//压缩文件
	if (AddFileToZip(newZipFile, strName.c_str(), srcFileName.c_str()) != 0)
	{
		Dbg("压缩文件失败");
		zipClose(newZipFile, NULL); 
		if (remove(zipFileName.c_str()) != 0) {
			Dbg("删除临时压缩文件失败");
		}
		return false;
	}
	//释放文件
	zipClose(newZipFile, NULL); 
	Dbg("文件压缩成功: %s", zipFileName.c_str());
	return true;
	
}

bool AddFileToZip(zipFile zf, const char* fileNameInZip, const char* srcFile)
{
	FILE* srcfp = NULL;

	//初始化写入zip的文件信息
	zip_fileinfo zi;
	zi.tmz_date.tm_sec = zi.tmz_date.tm_min = zi.tmz_date.tm_hour =
		zi.tmz_date.tm_mday = zi.tmz_date.tm_mon = zi.tmz_date.tm_year = 0;
	zi.dosDate = 0;
	zi.internal_fa = 0;
	zi.external_fa = 0;

	//如果srcFile为空，加入空目录
	char new_file_name[256];
	memset(new_file_name, 0, sizeof(new_file_name));
	strcat(new_file_name, fileNameInZip);
	if (srcFile == NULL)
	{
		strcat(new_file_name, "/");
	}

	//在zip文件中创建新文件或文件夹
	int ret = zipOpenNewFileInZip(zf, new_file_name, &zi, NULL, 0, NULL, 0, NULL, Z_DEFLATED, Z_DEFAULT_COMPRESSION);

	if (ret != ZIP_OK)
	{
		if (srcFile != NULL) {
			Dbg("在zip文件中创建新文件失败: %s", srcFile);
		}
		else {
			Dbg("在zip文件中创建新目录失败: %s", new_file_name);
		}

		return false;
	}


	if (srcFile != NULL)
	{
		//打开源文件
		srcfp = fopen(srcFile, "rb+");
		if (srcfp == NULL)
		{
			Dbg("打开要往zip压缩的文件失败: %s", srcFile);
			zipCloseFileInZip(zf); //关闭zip中的新文件
			return false;
		}
		int fileLen = (int)GetFileLen(srcfp);
		//读入源文件并写入zip文件
		char buf[16 * 1024]; 
		int numBytes = 0;
		int sumWrite = 0;
		while (!feof(srcfp) && !ferror(srcfp))
		{
			numBytes = fread(buf, 1, sizeof(buf), srcfp);
			int ret = zipWriteInFileInZip(zf, buf, numBytes);
			if (ret != ZIP_OK) {
				Dbg("文件写入zip压缩包失败1: %s", srcFile);
				fclose(srcfp);
				zipCloseFileInZip(zf);
				return false;//写文件失败
			}
			sumWrite += numBytes;
		}
		//和原文件长度是否相同
		if (sumWrite != fileLen) {
			Dbg("文件写入zip压缩包失败,总长度不一致: %s", srcFile);
			fclose(srcfp);
			zipCloseFileInZip(zf);
			return false;//写文件失败
		}
		//关闭源文件
		fclose(srcfp);
	}

	//关闭zip文件
	zipCloseFileInZip(zf);
	return true;
	
}

long GetFileLen(FILE* _file)
{
	long curPosit = ftell(_file), fileLen;
	fseek(_file, 0, SEEK_END);
	fileLen = ftell(_file);
	fseek(_file, curPosit, SEEK_SET);
	return fileLen;
}