mirror of https://github.com/BOINC/boinc.git
337 lines
11 KiB
C++
337 lines
11 KiB
C++
// this is the "wrapper" file for the zip/unzip functions to expose to BOINC clients
|
|
|
|
#pragma warning( disable : 4786 ) // Disable warning messages for vector
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#else
|
|
extern {
|
|
#endif
|
|
int unzip_main(int argc, char** argv);
|
|
int zip_main(int argc, char** argv);
|
|
#include "./unzip/unzip.h"
|
|
}
|
|
|
|
#include "./zip/zip.h"
|
|
|
|
#ifdef _WIN32
|
|
#include <windows.h>
|
|
#else
|
|
#include "config.h"
|
|
#include <string>
|
|
#include <string.h>
|
|
using std::string;
|
|
#endif
|
|
|
|
#include "boinc_zip.h"
|
|
#include "filesys.h" // from BOINC for DirScan
|
|
#include <algorithm>
|
|
|
|
// send in an output filename, advanced options (usually NULL), and numFileIn, szfileIn
|
|
|
|
#ifndef _MAX_PATH
|
|
#define _MAX_PATH 255
|
|
#endif
|
|
|
|
unsigned char g_ucSort;
|
|
|
|
// a "binary predicate" for use by the std::sort algorithm
|
|
// return true if "first > second" according to the g_ucSort type
|
|
|
|
bool StringVectorSort(const std::string& first, const std::string& second)
|
|
{
|
|
|
|
bool bRet = false;
|
|
if (g_ucSort & SORT_NAME
|
|
&& g_ucSort & SORT_ASCENDING
|
|
&& strcmp(first.c_str(), second.c_str())<0)
|
|
bRet = true;
|
|
else if (g_ucSort & SORT_NAME
|
|
&& g_ucSort & SORT_DESCENDING
|
|
&& strcmp(first.c_str(), second.c_str())>0)
|
|
bRet = true;
|
|
else if (g_ucSort & SORT_TIME)
|
|
{
|
|
struct stat st[2];
|
|
stat(first.c_str(), &st[0]);
|
|
stat(second.c_str(), &st[1]);
|
|
if (g_ucSort & SORT_ASCENDING) {
|
|
bRet = st[0].st_mtime < st[1].st_mtime;
|
|
}
|
|
else {
|
|
bRet = st[0].st_mtime > st[1].st_mtime;
|
|
}
|
|
}
|
|
|
|
return bRet;
|
|
}
|
|
|
|
int boinc_zip(int bZipType, const std::string szFileZip, const std::string szFileIn)
|
|
{
|
|
ZipFileList tempvec;
|
|
tempvec.push_back(szFileIn);
|
|
return boinc_zip(bZipType, szFileZip, &tempvec);
|
|
}
|
|
|
|
int boinc_zip(int bZipType, const char* szFileZip, const char* szFileIn)
|
|
{ // just use the regular boinc_zip
|
|
std::string strFileZip, strFileIn;
|
|
strFileZip.assign(szFileZip);
|
|
strFileIn.assign(szFileIn);
|
|
ZipFileList tempvec;
|
|
tempvec.push_back(strFileIn);
|
|
return boinc_zip(bZipType, strFileZip, &tempvec);
|
|
}
|
|
|
|
int boinc_zip(int bZipType, const std::string szFileZip,
|
|
const ZipFileList* pvectszFileIn)
|
|
{
|
|
int carg;
|
|
char** av;
|
|
int iRet = 0, i = 0, nVecSize = 0;
|
|
|
|
if (pvectszFileIn) nVecSize = pvectszFileIn->size();
|
|
|
|
// if unzipping but no file out, so it just uses cwd, only 3 args
|
|
//if (bZipType == UNZIP_IT)
|
|
// carg = 3 + nVecSize;
|
|
//else
|
|
carg = 3 + nVecSize;
|
|
|
|
// make a dynamic array
|
|
av = (char**) calloc(carg, sizeof(char*));
|
|
for (i=0;i<carg;i++)
|
|
av[i] = (char*) calloc(_MAX_PATH,sizeof(char));
|
|
|
|
// just form an argc/argv to spoof the "main"
|
|
// default options are to recurse into directories
|
|
//if (options && strlen(options))
|
|
// strcpy(av[1], options);
|
|
|
|
if (bZipType == ZIP_IT)
|
|
{
|
|
strcpy(av[0], "zip");
|
|
// default zip options -- no dir names, no subdirs, highest compression, quiet mode
|
|
if (strlen(av[1])==0)
|
|
strcpy(av[1], "-j9");
|
|
//strcpy(av[1], "-9jq");
|
|
strcpy(av[2], szFileZip.c_str());
|
|
|
|
//sz 3 onward will be each vector
|
|
int jj;
|
|
for (jj=0; jj<nVecSize; jj++)
|
|
strcpy(av[3+jj], pvectszFileIn->at(jj).c_str());
|
|
}
|
|
else
|
|
{
|
|
strcpy(av[0], "unzip");
|
|
// default unzip options -- preserve subdirs, overwrite
|
|
if (strlen(av[1])==0)
|
|
strcpy(av[1], "-o");
|
|
strcpy(av[2], szFileZip.c_str());
|
|
|
|
// if they passed in a directory unzip there
|
|
if (carg == 4)
|
|
sprintf(av[3], "-d%s", pvectszFileIn->at(0).c_str());
|
|
}
|
|
// strcpy(av[carg-1], ""); // null arg
|
|
// printf("args: %s %s %s %s\n", av[0], av[1], av[2], av[3]);
|
|
|
|
if (bZipType == ZIP_IT)
|
|
{
|
|
if (access(szFileZip.c_str(), 0) == 0)
|
|
{ // old zip file exists so unlink (otherwise zip will reuse, doesn't seem to be a flag to
|
|
// bypass zip reusing it
|
|
unlink(szFileZip.c_str());
|
|
}
|
|
iRet = zip_main(carg, av);
|
|
}
|
|
else {
|
|
// make sure zip file exists
|
|
if (access(szFileZip.c_str(), 0) == 0)
|
|
iRet = unzip_main(carg, av);
|
|
else
|
|
iRet = 2;
|
|
}
|
|
|
|
for (i=0;i<carg;i++)
|
|
free(av[i]);
|
|
free(av);
|
|
return iRet;
|
|
}
|
|
|
|
|
|
// -------------------------------------------------------------------
|
|
//
|
|
// Function: bool boinc_filelist(const std::string directory
|
|
// const std::string pattern,
|
|
// ZipFileList* pList)
|
|
//
|
|
// Description: Supply a directory and a pattern as arguments, along
|
|
// with a FileList variable to hold the result list; sorted by name.
|
|
// Returns a vector of files in the directory which match the
|
|
// pattern. Returns true for success, or false if there was a problem.
|
|
//
|
|
// CMC Note: this is a 'BOINC-friendly' implementation of "old" CPDN code
|
|
// the "wildcard matching"/regexp is a bit crude, it matches substrings in
|
|
// order in a file; to match all files such as *.pc.*.x4.*.nc" you would send in
|
|
// ".pc|.x4.|.nc" for 'pattern'
|
|
//
|
|
// --------------------------------------------------------------------
|
|
|
|
bool boinc_filelist(const std::string directory,
|
|
const std::string pattern,
|
|
ZipFileList* pList,
|
|
const unsigned char ucSort, const bool bClear)
|
|
{
|
|
g_ucSort = ucSort; // set the global sort type right off the bat
|
|
std::string strFile;
|
|
// at most three |'s may be passed in pattern match
|
|
int iPos[3], iFnd, iCtr, i, lastPos;
|
|
std::string strFullPath;
|
|
char strPart[3][32];
|
|
std::string spattern = pattern;
|
|
std::string strDir = directory;
|
|
std::string strUserDir = directory;
|
|
int iLen = strUserDir.size();
|
|
|
|
if (!pList) return false;
|
|
|
|
// wildcards are blank!
|
|
if (pattern == "*" || pattern == "*.*") spattern.assign("");
|
|
|
|
if (bClear)
|
|
pList->clear(); // removes old entries that may be in pList
|
|
|
|
// first tack on a final slash on user dir if required
|
|
if (strUserDir[iLen-1] != '\\'
|
|
&& strUserDir[iLen] != '/')
|
|
{
|
|
// need a final slash, but what type?
|
|
// / is safe on all OS's for CPDN at least
|
|
// but if they already used \ use that
|
|
// well they didn't use a backslash so just use a slash
|
|
if (strUserDir.find("\\") == string::npos)
|
|
strUserDir += "/";
|
|
else
|
|
strUserDir += "\\";
|
|
}
|
|
|
|
// transform strDir to either all \\ or all /
|
|
int j;
|
|
for (j=0; j<(int)directory.size(); j++) {
|
|
// take off final / or backslash
|
|
if (j == ((int)directory.size()-1)
|
|
&& (strDir[j] == '/' || strDir[j]=='\\'))
|
|
strDir.resize(directory.size()-1);
|
|
else {
|
|
#ifdef _WIN32 // transform paths appropriate for OS
|
|
if (directory[j] == '/')
|
|
strDir[j] = '\\';
|
|
#else
|
|
if (directory[j] == '\\')
|
|
strDir[j] = '/';
|
|
#endif
|
|
}
|
|
}
|
|
|
|
DirScanner dirscan(strDir);
|
|
memset(strPart, 0x00, 3*32);
|
|
while (dirscan.scan(strFile))
|
|
{
|
|
iCtr = 0;
|
|
lastPos = 0;
|
|
iPos[0] = -1;
|
|
iPos[1] = -1;
|
|
iPos[2] = -1;
|
|
// match the whole filename returned against the regexp to see if it's a hit
|
|
// first get all the |'s to get the pieces to verify
|
|
while (iCtr<3 && (iPos[iCtr] = (int) spattern.find('|', lastPos)) > -1)
|
|
{
|
|
if (iCtr==0) {
|
|
strncpy(strPart[0], spattern.c_str(), iPos[iCtr]);
|
|
}
|
|
else {
|
|
strncpy(strPart[iCtr], spattern.c_str()+lastPos, iPos[iCtr]-lastPos);
|
|
}
|
|
lastPos = iPos[iCtr]+1;
|
|
|
|
iCtr++;
|
|
}
|
|
if (iCtr>0) // found a | so need to get the part from lastpos onward
|
|
{
|
|
strncpy(strPart[iCtr], spattern.c_str()+lastPos, spattern.length() - lastPos);
|
|
}
|
|
|
|
// check no | were found at all
|
|
if (iCtr == 0)
|
|
{
|
|
strcpy(strPart[0], spattern.c_str());
|
|
iCtr++; // fake iCtr up 1 to get in the loop below
|
|
}
|
|
|
|
bool bFound = true;
|
|
for (i = 0; i <= iCtr && bFound; i++)
|
|
{
|
|
if (i==0) {
|
|
iFnd = (int) strFile.find(strPart[0]);
|
|
bFound = (bool) (iFnd > -1);
|
|
}
|
|
else {
|
|
// search forward of the old part found
|
|
iFnd = (int) strFile.find(strPart[i], iFnd+1);
|
|
bFound = bFound && (bool) (iFnd > -1);
|
|
}
|
|
}
|
|
|
|
if (bFound)
|
|
{
|
|
// this pattern matched the file, add to vector
|
|
// NB: first get stat to make sure it really is a file
|
|
strFullPath = strUserDir + strFile;
|
|
// only add if the file really exists (i.e. not a directory)
|
|
if (is_file(strFullPath.c_str())) {
|
|
pList->push_back(strFullPath);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
// sort by file creation time
|
|
if (pList->size()>1) { // sort if list is greather than 1
|
|
std::sort(pList->begin(), pList->end(), StringVectorSort); // may as well sort it?
|
|
}
|
|
return true;
|
|
}
|
|
|
|
int
|
|
boinc_UnzipToMemory ( char *zip, char *file, string &retstr )
|
|
{
|
|
UzpOpts opts = {}; /* options for UzpUnzipToMemory() */
|
|
UzpCB funcs = {}; /* function pointers for UzpUnzipToMemory() */
|
|
UzpBuffer buf;
|
|
int ret;
|
|
|
|
/* Read compressed file to memory. */
|
|
funcs.structlen = sizeof(UzpCB);
|
|
funcs.msgfn = (MsgFn *)printf;
|
|
funcs.inputfn = (InputFn *)scanf;
|
|
funcs.pausefn = (PauseFn *)( 0x01 );
|
|
funcs.passwdfn = (PasswdFn *)( NULL );
|
|
|
|
memset( &buf, 0, sizeof(buf) ); // data-blocks needs to be empty
|
|
ret = UzpUnzipToMemory( zip, file, &opts, &funcs, &buf );
|
|
|
|
if ( ret )
|
|
retstr = (string) buf.strptr;
|
|
|
|
if ( buf.strptr) free ( buf.strptr );
|
|
|
|
return ret;
|
|
|
|
} // boinc_UnzipToMemory()
|
|
|
|
|
|
const char *BOINC_RCSID_bdf38b2dfb = "$Id$";
|
|
|