您好,登录后才能下订单哦!
密码登录
登录注册
点击 登录注册 即表示同意《亿速云用户服务条款》
#include <fcntl.h> #include <sys/stat.h> #include <stdio.h> #include <dirent.h> #include <string.h> #include <stdlib.h> #include <vector> #include <string> #include <iostream> #include <fstream> #include "minizip/zip.h" #include "minizip/unzip.h" using namespace std; void EnumDirFiles(const string& dirPrefix,const string& dirName,vector<string>& vFiles) { if (dirPrefix.empty() || dirName.empty()) return; string dirNameTmp = dirName; string dirPre = dirPrefix; if (dirNameTmp.find_last_of("/") != dirNameTmp.length() - 1) dirNameTmp += "/"; if (dirNameTmp[0] == '/') dirNameTmp = dirNameTmp.substr(1); if (dirPre.find_last_of("/") != dirPre.length() - 1) dirPre += "/"; string path; path = dirPre + dirNameTmp; struct stat fileStat; DIR* pDir = opendir(path.c_str()); if (!pDir) return; struct dirent* pDirEnt = NULL; while ( (pDirEnt = readdir(pDir)) != NULL ) { if (strcmp(pDirEnt->d_name,".") == 0 || strcmp(pDirEnt->d_name,"..") == 0) continue; string tmpDir = dirPre + dirNameTmp + pDirEnt->d_name; if (stat(tmpDir.c_str(),&fileStat) != 0) continue; string innerDir = dirNameTmp + pDirEnt->d_name; if (fileStat.st_mode & S_IFDIR == S_IFDIR) { EnumDirFiles(dirPrefix,innerDir,vFiles); continue; } vFiles.push_back(innerDir); } if (pDir) closedir(pDir); } int writeInZipFile(zipFile zFile,const string& file) { fstream f(file.c_str(),std::ios::binary | std::ios::in); f.seekg(0, std::ios::end); long size = f.tellg(); f.seekg(0, std::ios::beg); if ( size <= 0 ) { return zipWriteInFileInZip(zFile,NULL,0); } char* buf = new char[size]; f.read(buf,size); int ret = zipWriteInFileInZip(zFile,buf,size); delete[] buf; return ret; } int main(int argc, char *argv[]) { if (argc < 3) { cout<<"usage: mini from to"<<endl; return -1; } string dest = string(argv[1]); string src = string(argv[2]); if (src.find_last_of("/") == src.length() - 1) src = src.substr(0,src.length()-1); struct stat fileInfo; stat(src.c_str(), &fileInfo); if (S_ISREG(fileInfo.st_mode)) { zipFile zFile = zipOpen(dest.c_str(),APPEND_STATUS_CREATE); if (zFile == NULL) { cout<<"openfile failed"<<endl; return -1; } zip_fileinfo zFileInfo = { 0 }; int ret = zipOpenNewFileInZip(zFile,src.c_str(),&zFileInfo,NULL,0,NULL,0,NULL,0,Z_DEFAULT_COMPRESSION); if (ret != ZIP_OK) { cout<<"openfile in zip failed"<<endl; zipClose(zFile,NULL); return -1; } ret = writeInZipFile(zFile,src); if (ret != ZIP_OK) { cout<<"write in zip failed"<<endl; zipClose(zFile,NULL); return -1; } zipClose(zFile,NULL); cout<<"zip ok"<<endl; } else if (S_ISDIR(fileInfo.st_mode)) { size_t pos = src.find_last_of("/"); string dirName = src.substr(pos + 1); string dirPrefix = src.substr(0,pos); zipFile zFile = zipOpen(dest.c_str(),APPEND_STATUS_CREATE); if (zFile == NULL) { cout<<"openfile failed"<<endl; return -1; } vector<string> vFiles; EnumDirFiles(dirPrefix,dirName,vFiles); vector<string>::iterator itF = vFiles.begin(); for(;itF != vFiles.end(); ++itF) { zip_fileinfo zFileInfo = { 0 }; int ret = zipOpenNewFileInZip(zFile,itF->c_str(),&zFileInfo,NULL,0,NULL,0,NULL,0,Z_DEFAULT_COMPRESSION); if (ret != ZIP_OK) { cout<<"openfile in zip failed"<<endl; zipClose(zFile,NULL); return -1; } ret = writeInZipFile(zFile,*itF); if (ret != ZIP_OK) { cout<<"write in zip failed"<<endl; zipClose(zFile,NULL); return -1; } } zipClose(zFile,NULL); cout<<"zip ok"<<endl; } return 0; }
target=mini lib=-lminizip -laes -lz libpath=-L/usr/local/lib incpath=-I/usr/local/include ${target}: g++ main.cpp -g -o $@ ${lib} ${libpath} ${incpath} clean: -rm ${target}
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。