CZipCreator.cpp File Reference

#include "precomp.h"
#include "atlsimpcoll.h"
#include "minizip/zip.h"
#include "minizip/iowin32.h"
#include <process.h>

Include dependency graph for CZipCreator.cpp:

Go to the source code of this file.

Classes
struct	CZipCreatorImpl

Enumerations
enum	CZC_ERROR { CZCERR_ZEROITEMS = 1, CZCERR_NOFILES, CZCERR_CREATE, CZCERR_READ }

Functions
static CStringW	DoGetZipName (LPCWSTR filename)
static CStringA	DoGetAnsiName (LPCWSTR filename)
static CStringW	DoGetBaseName (LPCWSTR filename)
static CStringA	DoGetNameInZip (const CStringW &basename, const CStringW &filename)
static BOOL	DoReadAllOfFile (LPCWSTR filename, CSimpleArray< BYTE > &contents, zip_fileinfo *pzi)
static void	DoAddFilesFromItem (CSimpleArray< CStringW > &files, LPCWSTR item)
static unsigned __stdcall	create_zip_function (void *arg)

Enumeration Type Documentation

CZC_ERROR

enum CZC_ERROR

Enumerator
CZCERR_ZEROITEMS
CZCERR_NOFILES
CZCERR_CREATE
CZCERR_READ

Definition at line 220 of file CZipCreator.cpp.

{
    CZCERR_ZEROITEMS = 1,
    CZCERR_NOFILES,
    CZCERR_CREATE,
    CZCERR_READ
};

Function Documentation

create_zip_function()

static unsigned __stdcall create_zip_function ( void *  arg )

static

Definition at line 183 of file CZipCreator.cpp.

{
    CZipCreator *pCreator = reinterpret_cast<CZipCreator *>(arg);
    return pCreator->m_pimpl->JustDoIt();
}

Referenced by CZipCreator::runThread().

DoAddFilesFromItem()

static void DoAddFilesFromItem ( CSimpleArray< CStringW > &  files, LPCWSTR  item  )

static

Definition at line 127 of file CZipCreator.cpp.

{
    if (!PathIsDirectoryW(item))
    {
        files.Add(item);
        return;
    }

    WCHAR szPath[MAX_PATH];
    StringCbCopyW(szPath, sizeof(szPath), item);
    PathAppendW(szPath, L"*");

    WIN32_FIND_DATAW find;
    HANDLE hFind = FindFirstFileW(szPath, &find);
    if (hFind == INVALID_HANDLE_VALUE)
        return;

    do
    {
        if (wcscmp(find.cFileName, L".") == 0 ||
            wcscmp(find.cFileName, L"..") == 0)
        {
            continue;
        }

        StringCbCopyW(szPath, sizeof(szPath), item);
        PathAppendW(szPath, find.cFileName);

        if (find.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
            DoAddFilesFromItem(files, szPath);
        else
            files.Add(szPath);
    } while (FindNextFileW(hFind, &find));

    FindClose(hFind);
}

Referenced by CZipCreatorImpl::JustDoIt().

DoGetAnsiName()

static CStringA DoGetAnsiName ( LPCWSTR  filename )

static

Definition at line 37 of file CZipCreator.cpp.

{
    CHAR buf[MAX_PATH];
    WideCharToMultiByte(CP_ACP, 0, filename, -1, buf, _countof(buf), NULL, NULL);
    return buf;
}

Referenced by DoGetNameInZip().

DoGetBaseName()

static CStringW DoGetBaseName ( LPCWSTR  filename )

static

Definition at line 44 of file CZipCreator.cpp.

{
    WCHAR szBaseName[MAX_PATH];
    StringCbCopyW(szBaseName, sizeof(szBaseName), filename);
    PathRemoveFileSpecW(szBaseName);
    PathAddBackslashW(szBaseName);
    return szBaseName;
}

Referenced by CZipCreatorImpl::JustDoIt().

DoGetNameInZip()

static CStringA DoGetNameInZip ( const CStringW &  basename, const CStringW &  filename  )

static

Definition at line 54 of file CZipCreator.cpp.

{
    CStringW basenameI = basename, filenameI = filename;
    basenameI.MakeUpper();
    filenameI.MakeUpper();

    CStringW ret;
    if (filenameI.Find(basenameI) == 0)
        ret = filename.Mid(basename.GetLength());
    else
        ret = filename;

    ret.Replace(L'\\', L'/');

    return DoGetAnsiName(ret);
}

Referenced by CZipCreatorImpl::JustDoIt().

DoGetZipName()

static CStringW DoGetZipName ( LPCWSTR  filename )

static

Definition at line 15 of file CZipCreator.cpp.

{
    WCHAR szPath[MAX_PATH];
    StringCbCopyW(szPath, sizeof(szPath), filename);
    PathRemoveExtensionW(szPath);

    CStringW ret = szPath;
    ret += L".zip";

    UINT i = 2;
    while (PathFileExistsW(ret))
    {
        CStringW str;
        str.Format(L" (%u).zip", i++);

        ret = szPath;
        ret += str;
    }

    return ret;
}

Referenced by CZipCreatorImpl::JustDoIt().

DoReadAllOfFile()

static BOOL DoReadAllOfFile ( LPCWSTR  filename, CSimpleArray< BYTE > &  contents, zip_fileinfo *  pzi  )

static

Definition at line 72 of file CZipCreator.cpp.

{
    contents.RemoveAll();

    HANDLE hFile = CreateFileW(filename, GENERIC_READ, FILE_SHARE_READ,
                               NULL, OPEN_EXISTING,
                               FILE_FLAG_SEQUENTIAL_SCAN, NULL);
    if (hFile == INVALID_HANDLE_VALUE)
    {
        DPRINT1("%S: cannot open\n", filename);
        return FALSE;
    }

    FILETIME ft, ftLocal;
    ZeroMemory(pzi, sizeof(*pzi));
    if (GetFileTime(hFile, NULL, NULL, &ft))
    {
        SYSTEMTIME st;
        FileTimeToLocalFileTime(&ft, &ftLocal);
        FileTimeToSystemTime(&ftLocal, &st);
        pzi->tmz_date.tm_sec = st.wSecond;
        pzi->tmz_date.tm_min = st.wMinute;
        pzi->tmz_date.tm_hour = st.wHour;
        pzi->tmz_date.tm_mday = st.wDay;
        pzi->tmz_date.tm_mon = st.wMonth - 1;
        pzi->tmz_date.tm_year = st.wYear;
    }

    const DWORD cbBuff = 0x7FFF;
    LPBYTE pbBuff = reinterpret_cast<LPBYTE>(CoTaskMemAlloc(cbBuff));
    if (!pbBuff)
    {
        DPRINT1("Out of memory\n");
        CloseHandle(hFile);
        return FALSE;
    }

    for (;;)
    {
        DWORD cbRead;
        if (!ReadFile(hFile, pbBuff, cbBuff, &cbRead, NULL) || !cbRead)
            break;

        for (DWORD i = 0; i < cbRead; ++i)
            contents.Add(pbBuff[i]);
    }

    CoTaskMemFree(pbBuff);
    CloseHandle(hFile);

    return TRUE;
}

Referenced by CZipCreatorImpl::JustDoIt().