CZipExtract Class Reference

Inheritance diagram for CZipExtract:

Collaboration diagram for CZipExtract:

Classes
class	CCompleteSettingsPage
class	CExtractSettingsPage

Public Member Functions
	CZipExtract (PCWSTR Filename)
	~CZipExtract ()
void	Close ()
STDMETHODIMP	QueryInterface (REFIID riid, void **ppvObject)
	STDMETHODIMP_ (ULONG) AddRef(void)
	STDMETHODIMP_ (ULONG) Release(void)
	STDMETHODIMP_ (unzFile) getZip()
void	runWizard ()
eZipExtractError	ExtractSingle (HWND hDlg, LPCSTR FullPath, bool is_dir, unz_file_info64 *Info, CStringA Name, CStringA Password, bool *bOverwriteAll, const bool *bCancel, int *ErrorCode)
bool	Extract (HWND hDlg, HWND hProgress, const bool *bCancel)
int	ShowExtractError (HWND hDlg, LPCSTR path, int Error, eZipExtractError ErrorType)
virtual	STDMETHODIMP_ (unzFile) getZip() PURE
Public Member Functions inherited from IUnknown
HRESULT	QueryInterface ([in] REFIID riid, [out, iid_is(riid)] void **ppvObject)
ULONG	AddRef ()
ULONG	Release ()

Static Public Member Functions
static INT CALLBACK	s_PropSheetCallbackProc (HWND hwndDlg, UINT uMsg, LPARAM lParam)

Private Attributes
CStringW	m_Filename
CStringW	m_Directory
CStringA	m_Password
bool	m_DirectoryChanged
unzFile	uf

Additional Inherited Members
Public Types inherited from IUnknown
typedef IUnknown *	LPUNKNOWN

Detailed Description

Definition at line 10 of file CZipExtract.cpp.

Constructor & Destructor Documentation

CZipExtract()

CZipExtract::CZipExtract ( PCWSTR  Filename )

inline

Definition at line 19 of file CZipExtract.cpp.

        :m_DirectoryChanged(false)
        ,uf(NULL)
    {
        m_Filename = Filename;
        m_Directory = m_Filename;
        PWSTR Dir = m_Directory.GetBuffer();
        PathRemoveExtensionW(Dir);
        m_Directory.ReleaseBuffer();
    }

~CZipExtract()

CZipExtract::~CZipExtract ( )

inline

Definition at line 30 of file CZipExtract.cpp.

    {
        if (uf)
        {
            DPRINT1("WARNING: uf not closed!\n");
            Close();
        }
    }

Member Function Documentation

Close()

void CZipExtract::Close ( )

inline

Definition at line 39 of file CZipExtract.cpp.

    {
        if (uf)
            unzClose(uf);
        uf = NULL;
    }

Referenced by Extract(), and ~CZipExtract().

Extract()

bool CZipExtract::Extract ( HWND  hDlg, HWND  hProgress, const bool *  bCancel  )

inline

Definition at line 554 of file CZipExtract.cpp.

    {
        unz_global_info64 gi;
        uf = unzOpen2_64(m_Filename.GetString(), &g_FFunc);
        int err = unzGetGlobalInfo64(uf, &gi);
        if (err != UNZ_OK)
        {
            DPRINT1("ERROR, unzGetGlobalInfo64: 0x%x\n", err);
            Close();
            return false;
        }
 
        CZipEnumerator zipEnum;
        if (!zipEnum.initialize(this))
        {
            DPRINT1("ERROR, zipEnum.initialize\n");
            Close();
            return false;
        }
 
        CWindow Progress(hProgress);
        Progress.SendMessage(PBM_SETRANGE32, 0, gi.number_entry);
        Progress.SendMessage(PBM_SETPOS, 0, 0);
 
        CStringA BaseDirectory = m_Directory;
        CStringA Name;
        CStringA Password = m_Password;
        unz_file_info64 Info;
        int CurrentFile = 0;
        bool bOverwriteAll = false;
        while (zipEnum.next(Name, Info))
        {
            if (*bCancel)
            {
                Close();
                return false;
            }
 
            bool is_dir = Name.GetLength() > 0 && Name[Name.GetLength()-1] == '/';
 
            char CombinedPath[MAX_PATH * 2] = { 0 };
            PathCombineA(CombinedPath, BaseDirectory, Name);
            CStringA FullPath = CombinedPath;
            FullPath.Replace('/', '\\');    /* SHPathPrepareForWriteA does not handle '/' */
        Retry:
            eZipExtractError Result = ExtractSingle(hDlg, FullPath, is_dir, &Info, Name, Password, &bOverwriteAll, bCancel, &err);
            if (Result != eDirectoryError)
                CurrentFile++;
            switch (Result)
            {
                case eNoError:
                    break;
 
                case eExtractAbort:
                case eUnpackError:
                {
                    Close();
                    return false;
                }
 
                case eDirectoryError:
                {
                    char StrippedPath[MAX_PATH] = { 0 };
 
                    StrCpyNA(StrippedPath, FullPath, _countof(StrippedPath));
                    if (!is_dir)
                        PathRemoveFileSpecA(StrippedPath);
                    PathStripPathA(StrippedPath);
                    if (ShowExtractError(hDlg, (LPCSTR)&StrippedPath, err, eDirectoryError) == IDRETRY)
                        goto Retry;
                    Close();
                    return false;
                }
 
                case eFileError:
                {
                    int Result = ShowExtractError(hDlg, FullPath, err, eFileError);
                    switch (Result)
                    {
                    case IDABORT:
                        Close();
                        return false;
                    case IDRETRY:
                        CurrentFile--;
                        goto Retry;
                    case IDIGNORE:
                        break;
                    }
                    break;
                }
 
                case eOpenError:
                {
                    if (err == UNZ_BADZIPFILE &&
                        Info.compression_method != 0 &&
                        Info.compression_method != Z_DEFLATED &&
                        Info.compression_method != Z_BZIP2ED)
                    {
                        if (ShowExtractError(hDlg, FullPath, Info.compression_method, eOpenError) == IDYES)
                            break;
                    }
                    Close();
                    return false;
                }
            }
            if (Result == eNoError && is_dir)
                continue;
            Progress.SendMessage(PBM_SETPOS, CurrentFile, 0);
        }
 
        Close();
        return true;
    }

Referenced by CZipExtract::CExtractSettingsPage::ExtractEntry().

ExtractSingle()

eZipExtractError CZipExtract::ExtractSingle ( HWND  hDlg, LPCSTR  FullPath, bool  is_dir, unz_file_info64 *  Info, CStringA  Name, CStringA  Password, bool *  bOverwriteAll, const bool *  bCancel, int *  ErrorCode  )

inline

Definition at line 371 of file CZipExtract.cpp.

    {
        int err;
        BYTE Buffer[2048];
        DWORD dwFlags = SHPPFW_DIRCREATE | (is_dir ? SHPPFW_NONE : SHPPFW_IGNOREFILENAME);
        HRESULT hr = SHPathPrepareForWriteA(hDlg, NULL, FullPath, dwFlags);
        if (FAILED_UNEXPECTEDLY(hr))
        {
            *ErrorCode = hr;
            return eDirectoryError;
        }
        if (is_dir)
            return eNoError;
 
        if (Info->flag & MINIZIP_PASSWORD_FLAG)
        {
            eZipPasswordResponse Response = eAccept;
            do
            {
                /* If there is a password set, try it */
                if (!Password.IsEmpty())
                {
                    err = unzOpenCurrentFilePassword(uf, Password);
                    if (err == UNZ_OK)
                    {
                        /* Try to read some bytes, because unzOpenCurrentFilePassword does not return failure */
                        char Buf[10];
                        err = unzReadCurrentFile(uf, Buf, sizeof(Buf));
                        unzCloseCurrentFile(uf);
                        if (err >= UNZ_OK)
                        {
                            /* 're'-open the file so that we can begin to extract */
                            err = unzOpenCurrentFilePassword(uf, Password);
                            break;
                        }
                    }
                }
                Response = _CZipAskPassword(hDlg, Name, Password);
            } while (Response == eAccept);
 
            if (Response == eSkip)
            {
                return eNoError;
            }
            else if (Response == eAbort)
            {
                return eExtractAbort;
            }
        }
        else
        {
            err = unzOpenCurrentFile(uf);
        }
 
        if (err != UNZ_OK)
        {
            DPRINT1("ERROR, unzOpenCurrentFilePassword: 0x%x\n", err);
            *ErrorCode = err;
            return eOpenError;
        }
 
        HANDLE hFile = CreateFileA(FullPath, GENERIC_WRITE, 0, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL);
        if (hFile == INVALID_HANDLE_VALUE)
        {
            DWORD dwErr = GetLastError();
            if (dwErr == ERROR_FILE_EXISTS)
            {
                bool bOverwrite = *bOverwriteAll;
                if (!*bOverwriteAll)
                {
                    eZipConfirmResponse Result = _CZipAskReplace(hDlg, FullPath);
                    switch (Result)
                    {
                    case eYesToAll:
                        *bOverwriteAll = true;
                        /* fall through */
                    case eYes:
                        bOverwrite = true;
                        break;
                    case eNo:
                        break;
                    case eCancel:
                        unzCloseCurrentFile(uf);
                        return eExtractAbort;
                    }
                }
 
                if (bOverwrite)
                {
                    hFile = CreateFileA(FullPath, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
                    if (hFile == INVALID_HANDLE_VALUE)
                    {
                        dwErr = GetLastError();
                    }
                }
                else
                {
                    unzCloseCurrentFile(uf);
                    return eNoError;
                }
            }
            if (hFile == INVALID_HANDLE_VALUE)
            {
                unzCloseCurrentFile(uf);
                DPRINT1("ERROR, CreateFileA: 0x%x (%s)\n", dwErr, *bOverwriteAll ? "Y" : "N");
                *ErrorCode = dwErr;
                return eFileError;
            }
        }
 
        do
        {
            if (*bCancel)
            {
                CloseHandle(hFile);
                BOOL deleteResult = DeleteFileA(FullPath);
                if (!deleteResult)
                    DPRINT1("ERROR, DeleteFileA: 0x%x\n", GetLastError());
                return eExtractAbort;
            }
 
            err = unzReadCurrentFile(uf, Buffer, sizeof(Buffer));
 
            if (err < 0)
            {
                DPRINT1("ERROR, unzReadCurrentFile: 0x%x\n", err);
                break;
            }
            else if (err > 0)
            {
                DWORD dwWritten;
                if (!WriteFile(hFile, Buffer, err, &dwWritten, NULL))
                {
                    DPRINT1("ERROR, WriteFile: 0x%x\n", GetLastError());
                    break;
                }
                if (dwWritten != (DWORD)err)
                {
                    DPRINT1("ERROR, WriteFile: dwWritten:%d err:%d\n", dwWritten, err);
                    break;
                }
            }
 
        } while (err > 0);
 
        /* Update Filetime */
        FILETIME LocalFileTime;
        DosDateTimeToFileTime((WORD)(Info->dosDate >> 16), (WORD)Info->dosDate, &LocalFileTime);
        FILETIME FileTime;
        LocalFileTimeToFileTime(&LocalFileTime, &FileTime);
        SetFileTime(hFile, &FileTime, &FileTime, &FileTime);
 
        /* Done */
        CloseHandle(hFile);
 
        if (err)
        {
            unzCloseCurrentFile(uf);
            DPRINT1("ERROR, unzReadCurrentFile2: 0x%x\n", err);
            *ErrorCode = err;
            return eUnpackError;
        }
        else
        {
            err = unzCloseCurrentFile(uf);
            if (err != UNZ_OK)
            {
                DPRINT1("ERROR(non-fatal), unzCloseCurrentFile: 0x%x\n", err);
            }
        }
        return eNoError;
    }

Referenced by Extract().

QueryInterface()

STDMETHODIMP CZipExtract::QueryInterface ( REFIID  riid, void **  ppvObject  )

inline

Implements IUnknown.

Definition at line 47 of file CZipExtract.cpp.

    {
        if (riid == IID_IUnknown)
        {
            *ppvObject = this;
            AddRef();
            return S_OK;
        }
        return E_NOINTERFACE;
    }

runWizard()

void CZipExtract::runWizard ( )

inline

Definition at line 347 of file CZipExtract.cpp.

    {
        PROPSHEETHEADERW psh = { sizeof(psh), 0 };
        psh.dwFlags = PSH_WIZARD97 | PSH_HEADER | PSH_USEICONID | PSH_USECALLBACK;
        psh.hInstance = _AtlBaseModule.GetResourceInstance();
 
        CExtractSettingsPage extractPage(this, &m_Password);
        CCompleteSettingsPage completePage(this);
        HPROPSHEETPAGE hpsp[] =
        {
            extractPage.Create(),
            completePage.Create()
        };
 
        psh.phpage = hpsp;
        psh.nPages = _countof(hpsp);
        psh.pszIcon = MAKEINTRESOURCE(IDI_ZIPFLDR);
        psh.pszbmWatermark = MAKEINTRESOURCE(IDB_WATERMARK);
        psh.pszbmHeader = MAKEINTRESOURCE(IDB_HEADER);
        psh.pfnCallback = s_PropSheetCallbackProc;
 
        PropertySheetW(&psh);
    }

Referenced by _CZipExtract_runWizard().

s_PropSheetCallbackProc()

static INT CALLBACK CZipExtract::s_PropSheetCallbackProc ( HWND  hwndDlg, UINT  uMsg, LPARAM  lParam  )

inlinestatic

Definition at line 335 of file CZipExtract.cpp.

    {
        if (uMsg == PSCB_INITIALIZED)
        {
            HICON hIcon = LoadIconW(_AtlBaseModule.GetResourceInstance(), MAKEINTRESOURCEW(IDI_ZIPFLDR));
            CWindow dlg(hwndDlg);
            dlg.SetIcon(hIcon, TRUE);
        }
 
        return 0;
    }

Referenced by runWizard().

ShowExtractError()

int CZipExtract::ShowExtractError ( HWND  hDlg, LPCSTR  path, int  Error, eZipExtractError  ErrorType  )

inline

Definition at line 668 of file CZipExtract.cpp.

    {
        CStringA strTitle(MAKEINTRESOURCEW(IDS_ERRORTITLE));
        CStringA strErr, strText;
        PSTR Win32ErrorString;
 
        if (ErrorType == eFileError || ErrorType == eOpenError)
            strText.LoadString(IDS_CANTEXTRACTFILE);
        else
            strText.LoadString(GetModuleHandleA("shell32.dll"), 128); // IDS_CREATEFOLDER_DENIED
 
        strText.FormatMessage(strText.GetString(), path);
 
        if (ErrorType == eFileError || HRESULT_FACILITY(Error) == FACILITY_WIN32)
        {
            if (FormatMessageA(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
                               NULL, ErrorType == eFileError ? Error : HRESULT_CODE(Error), 0,
                               (PSTR)&Win32ErrorString, 0, NULL) != 0)
            {
                strErr.SetString(Win32ErrorString);
                LocalFree(Win32ErrorString);
            }
        }
        if (ErrorType == eOpenError)
            strErr.Format(IDS_DECOMPRESSERROR, Error);
        else if (strErr.GetLength() == 0)
            strErr.Format(IDS_UNKNOWNERROR, Error);
 
        strText.Append("\r\n\r\n" + strErr);
 
        UINT mbFlags = MB_ICONWARNING;
        if (ErrorType == eDirectoryError)
            mbFlags |= MB_RETRYCANCEL;
        else if (ErrorType == eFileError)
            mbFlags |= MB_ABORTRETRYIGNORE;
        else if (ErrorType == eOpenError)
            mbFlags |= MB_YESNO;
 
        return MessageBoxA(hDlg, strText, strTitle, mbFlags);
    }

Referenced by Extract().

STDMETHODIMP_() [1/3]

CZipExtract::STDMETHODIMP_ ( ULONG  )

inline

Definition at line 57 of file CZipExtract.cpp.

    {
        return 2;
    }

STDMETHODIMP_() [2/3]

CZipExtract::STDMETHODIMP_ ( ULONG  )

inline

Definition at line 61 of file CZipExtract.cpp.

    {
        return 1;
    }

STDMETHODIMP_() [3/3]

CZipExtract::STDMETHODIMP_ ( unzFile  )

inlinevirtual

Reimplemented from IZip.

Definition at line 65 of file CZipExtract.cpp.

    {
        return uf;
    }

Member Data Documentation

m_Directory

CStringW CZipExtract::m_Directory

private

Definition at line 14 of file CZipExtract.cpp.

Referenced by CZipExtract(), Extract(), CZipExtract::CExtractSettingsPage::OnBrowse(), CZipExtract::CExtractSettingsPage::OnSetActive(), CZipExtract::CCompleteSettingsPage::OnSetActive(), CZipExtract::CCompleteSettingsPage::OnWizardFinish(), and CZipExtract::CExtractSettingsPage::UpdateDirectory().

m_DirectoryChanged

bool CZipExtract::m_DirectoryChanged

private

Definition at line 16 of file CZipExtract.cpp.

Referenced by CZipExtract::CExtractSettingsPage::OnBrowse(), CZipExtract::CExtractSettingsPage::OnEnChangeDirectory(), CZipExtract::CExtractSettingsPage::OnSetActive(), CZipExtract::CExtractSettingsPage::OnWizardNext(), and CZipExtract::CExtractSettingsPage::UpdateDirectory().

m_Filename

CStringW CZipExtract::m_Filename

private

Definition at line 13 of file CZipExtract.cpp.

Referenced by CZipExtract(), and Extract().

m_Password

CStringA CZipExtract::m_Password

private

Definition at line 15 of file CZipExtract.cpp.

Referenced by Extract(), and runWizard().

uf

unzFile CZipExtract::uf

private

Definition at line 17 of file CZipExtract.cpp.

Referenced by Close(), Extract(), ExtractSingle(), STDMETHODIMP_(), and ~CZipExtract().

---

The documentation for this class was generated from the following file:

• dll/shellext/zipfldr/CZipExtract.cpp