[Zlib]_[初级]_[使用zlib库压缩目录]

时间：2022-09-17 23:19:26

场景 http://blog.csdn.net/infoworld/article/details/60480313

  1. 压缩目录为zip不用多说了, 很多场景都会用到, 比如打包文件, 打包目录下载等.

说明

  1. zlib里的minizip并不可以直接使用来压缩文件夹, 但并不妨碍我们借鉴它的代码. 关键其实就在这个变量里savefilenameinzip, 它支持目录结构的名字. 所以只需要在调用 zipOpenNewFileInZip3_64 时, 传递带相对目录结构的savefilenameinzip即可. 注: 可能最新版的zlib的minizip.c已支持, 我用的是1.2.5版本.

  2. 在之前的 使用zlib库压缩文件 里, 我们使用了ZipHelper 来压缩文件到直接目录, 这次就稍微改改就可以压缩目录. 接口使用utf8编码的字符串, 目的是以后能扩展支持macOS平台. 注意在编译时要加宏 ZLIB_WINAPI

例子

zip_helper.h


#ifndef __ZIP_HELPER 
#define __ZIP_HELPER 

#include <map> 
#include <string> 

//1.暂时不支持子目录 
//注意: 因为使用了zlib库,使用时加上预编译宏 ZLIB_WINAPI 
class ZipHelper  
{  
public:  
    ZipHelper(){}  
    ~ZipHelper(){}  

//path: utf8 path 
    ZipHelper& AddFile(const char* input_path,const char* inzip_path = "");
    ZipHelper& AddDir(const char* input_dir,const char* temp_dir = NULL);

//output_path :utf8 path 
bool ToZip(const char* output_path);

private:  
std::map<std::string,std::string> files_;  
};  


#endif

zip_helper.cpp


#include "zip_helper.h" 


#ifndef _WIN32 
#ifndef __USE_FILE_OFFSET64 
#define __USE_FILE_OFFSET64 
#endif 
#ifndef __USE_LARGEFILE64 
#define __USE_LARGEFILE64 
#endif 
#ifndef _LARGEFILE64_SOURCE 
#define _LARGEFILE64_SOURCE 
#endif 
#ifndef _FILE_OFFSET_BIT 
#define _FILE_OFFSET_BIT 64 
#endif 
#endif 

#include <stdio.h> 
#include <stdlib.h> 
#include <string.h> 
#include <time.h> 
#include <errno.h> 
#include <fcntl.h>

#include <direct.h> 
#include <io.h> 

#include "zlib.h" 
#include "zip.h" 

#ifdef _WIN32 
#define USEWIN32IOAPI 
#include "iowin32.h" 
#endif 

#define WRITEBUFFERSIZE (16384) 
#define MAXFILENAME (256) 




#ifdef _WIN32 

static char* Utf8ToAnsi(const char* utf8)    
{    
// 先转换为UNICODE 
int dwUnicodeLen = MultiByteToWideChar(CP_UTF8,0,utf8,-1,NULL,0);    
if(!dwUnicodeLen)    
    {    
return strdup(utf8);    
    }    
    size_t num = dwUnicodeLen*sizeof(wchar_t);    
wchar_t *pwText = (wchar_t*)malloc(num+2);    
memset(pwText,0,num+2);    
    MultiByteToWideChar(CP_UTF8,0,utf8,-1,pwText,dwUnicodeLen);  

// 再转换为ANSI 
int len;    
    len = WideCharToMultiByte(CP_ACP, 0, pwText, -1, NULL, 0, NULL, NULL);    
char *szANSI = (char*)malloc(len + 1);    
memset(szANSI, 0, len + 1);    
    WideCharToMultiByte(CP_ACP, 0, pwText, -1, szANSI, len, NULL,NULL);    

free(pwText);  
return (char*)szANSI;    
}   

static wchar_t* QXUtf82Unicode(const char* utf)    
{    
if(!utf || !strlen(utf))    
    {    
return NULL;    
    }    
int dwUnicodeLen = MultiByteToWideChar(CP_UTF8,0,utf,-1,NULL,0);    
    size_t num = dwUnicodeLen*sizeof(wchar_t);    
wchar_t *pwText = (wchar_t*)malloc(num);    
memset(pwText,0,num);    
    MultiByteToWideChar(CP_UTF8,0,utf,-1,pwText,dwUnicodeLen);    
return pwText;    
}

static char* QXUnicode2Utf8(const wchar_t* unicode)    
{    
if(unicode == NULL)
    {
return strdup("\0");
    }

int len;  
    len = WideCharToMultiByte(CP_UTF8, 0,unicode, -1, NULL, 0, NULL, NULL);  
char *szUtf8 = (char*)malloc(len + 1);  
memset(szUtf8, 0, len + 1);  
    WideCharToMultiByte(CP_UTF8, 0,unicode, -1, szUtf8, len, NULL,NULL);  
return szUtf8;   
}   

static FILE* ZipFopen(const char* path,const char* mode)  
{  
wchar_t* path_u = QXUtf82Unicode(path);  
wchar_t* mode_u = QXUtf82Unicode(mode);  
    FILE* file = _wfopen(path_u,mode_u);  
free(path_u);  
free(mode_u);  
return file;  
}  

/* name of file to get info on */  
/* return value: access, modific. and creation times */  
/* dostime */  
uLong filetime(const char* f, tm_zip *tmzip, uLong *dt)            
{  
int ret = 0;  
    {  
        FILETIME ftLocal;  
        HANDLE hFind;  
        WIN32_FIND_DATA ff32;  
wchar_t *unicode = QXUtf82Unicode(f);  
        hFind = FindFirstFile(unicode,&ff32);  
if (hFind != INVALID_HANDLE_VALUE)  
        {  
            FileTimeToLocalFileTime(&(ff32.ftLastWriteTime),&ftLocal);  
            FileTimeToDosDateTime(&ftLocal,((LPWORD)dt)+1,((LPWORD)dt)+0);  
            FindClose(hFind);  
            ret = 1;  
        }  
free(unicode);  
    }  
return ret;  
}  
#else 
#define ZipFopen fopen; 
#endif 

ZipHelper& ZipHelper::AddFile(const char* input_path,const char* inzip_path)  
{  
    files_[input_path] = inzip_path;  
return *this;  
}

ZipHelper& ZipHelper::AddDir(const char* input_dir,const char* temp_dir){
    WIN32_FIND_DATA findFileData;
auto input_dirw = QXUtf82Unicode(input_dir);

std::wstring path(input_dirw);
auto pos = path.at(path.size()-1);
if(pos != L'\\' && pos != L'/'){
        path.append(L"/");
    }
    HANDLE hFind  = ::FindFirstFile((path+L"*").c_str(),&findFileData);
    size_t file_name_len = 0;
if (hFind == INVALID_HANDLE_VALUE){  
free(input_dirw);
return *this;  
    }else{

if(!temp_dir){
            temp_dir = input_dir;
        }
std::string temp_dir2(temp_dir);
        pos = temp_dir2.at(temp_dir2.size()-1);
if(pos != '\\' && pos != '/'){
            temp_dir2.append("/");
        }

while (FindNextFile(hFind, &findFileData)){ 

if(!wcscmp(findFileData.cFileName,L".") || !wcscmp(findFileData.cFileName,L".."))
continue;

std::wstring path1(path);
            path1.append(findFileData.cFileName);
char* path_utf8 = QXUnicode2Utf8(path1.c_str());
if(findFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY){
                AddDir(path_utf8,temp_dir);
            }else{
                AddFile(path_utf8,path_utf8+strlen(temp_dir2.c_str()));
            }
free(path_utf8);
        }  
        FindClose(hFind);  
    }
free(input_dirw);
}

bool ZipHelper::ToZip(const char* output_path)  
{  
int err=0;  
    zipFile zf;  
int errclose;  
int opt_compress_level = Z_DEFAULT_COMPRESSION;  
#ifdef USEWIN32IOAPI 
    zlib_filefunc64_def ffunc;  
    fill_win32_filefunc64W(&ffunc);  
wchar_t* temp_path = QXUtf82Unicode(output_path);  
    zf = zipOpen2_64(temp_path,APPEND_STATUS_CREATE,NULL,&ffunc);  
free(temp_path);  
#else 
    zf = zipOpen64(output_path,APPEND_STATUS_CREATE);  
#endif 

if (zf == NULL)  
    {  
printf("error opening %s\n",output_path);  
        err= ZIP_ERRNO;  
return false;  
    }  

void* buf = NULL;  
int size_buf = WRITEBUFFERSIZE;  
    buf = (void*)malloc(size_buf);  
for (auto ite = files_.begin(); ite != files_.end(); ++ite)  
    {  
        FILE * fin;  
int size_read;  
const char* filenameinzip = ite->first.c_str();
const char *savefilenameinzip = ite->second.c_str();  
        zip_fileinfo zi;  
unsigned long crcFile=0;  
int zip64 = 0;  

        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;  
        filetime(filenameinzip,&zi.tmz_date,&zi.dosDate);  

const char* pos = NULL;  
if(!strlen(savefilenameinzip)){
            savefilenameinzip = filenameinzip;
if( (pos = strrchr(savefilenameinzip,'\\'))   
                || (pos = strrchr(savefilenameinzip,'/')) ){  
                    pos++;  
            }else{  
                pos = savefilenameinzip;  
            }
        }else{
            pos = savefilenameinzip;
        }

// 这个版本不支持UTF8字符串的正确存储,所以要转换为ANSI显示. 
char* pos_ansi = Utf8ToAnsi(pos);  
        err = zipOpenNewFileInZip3_64(zf,pos_ansi,&zi,  
            NULL,0,NULL,0,NULL,  
            (opt_compress_level != 0) ? Z_DEFLATED : 0,  
            opt_compress_level,0,  
            -MAX_WBITS, DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY,  
            NULL,crcFile, zip64);  
free(pos_ansi);  

if (err != ZIP_OK)  
        {  
printf("error in opening %s in zipfile\n",pos);  
        }  
else  
        {  
            fin = ZipFopen(filenameinzip,"rb");  
if (fin==NULL)  
            {  
                err=ZIP_ERRNO;  
printf("error in opening %s for reading\n",filenameinzip);  
            }  
        }  

if (err == ZIP_OK)  
do  
            {  
                err = ZIP_OK;  
                size_read = (int)fread(buf,1,size_buf,fin);  
if (size_read < size_buf)  
                {  
if (feof(fin)==0)  
                    {  
printf("error in reading %s\n",filenameinzip);  
                        err = ZIP_ERRNO;  
                    }  
                }  
if (size_read>0)  
                {  
                    err = zipWriteInFileInZip (zf,buf,size_read);  
if (err<0)  
                    {  
printf("error in writing %s in the zipfile\n",  
                            filenameinzip);  
                    }  
                }  
            } while ((err == ZIP_OK) && (size_read>0));  

if(fin)  
            {  
                fclose(fin);  
            }  

if (err<0)  
            {  
                err=ZIP_ERRNO;  
            }  
else  
            {  
                err = zipCloseFileInZip(zf);  
if (err!=ZIP_OK)  
                {  
printf("error in closing %s in the zipfile\n",filenameinzip);  
                }  

            }  
    }  
    errclose = zipClose(zf,NULL);  
if (errclose != ZIP_OK)  
    {  
printf("error in closing %s\n",output_path);  
return false;  
    }  
return true;  
}

main.cpp


#include "zip_helper.h"

int main(int argc,char* argv[]){
    ZipHelper zh;
zh.AddDir("Win32");
zh.AddFile("./main.cpp");
zh.ToZip("1.zip");
}

项目下载路径

参考

minizip
File Attribute Constants
FindFirstFile
使用zlib库压缩文件
标签：

相关文章