CZipEnumerator Struct Reference

#include <CZipEnumerator.hpp>

Public Member Functions
	CZipEnumerator ()
BOOL	Initialize (IZip *zip)
BOOL	Reset ()
BOOL	Next (CStringW &name, unz_file_info64 &info)
BOOL	NextUnique (PCWSTR prefix, CStringW &name, bool &folder, unz_file_info64 &info)

Static Public Member Functions
static DWORD	CalculateCRC32 (PCSTR filename)
static CStringA	GetUtf8Name (PCSTR originalName, const BYTE *extraField, DWORD extraFieldLen)

Public Attributes
CComPtr< IZip >	m_Zip
BOOL	m_First = TRUE
CAtlList< CStringW >	m_Returned
UINT	m_nCodePage = GetZipCodePage(TRUE)

Detailed Description

Definition at line 9 of file CZipEnumerator.hpp.

Constructor & Destructor Documentation

CZipEnumerator()

CZipEnumerator::CZipEnumerator

(

)

Definition at line 14 of file CZipEnumerator.cpp.

{
}

Member Function Documentation

CalculateCRC32()

DWORD CZipEnumerator::CalculateCRC32 ( PCSTR  filename )

static

Definition at line 35 of file CZipEnumerator.cpp.

{
    ATLASSERT(filename);
    DWORD crc = crc32(0, Z_NULL, 0);
    crc = crc32(crc, (const Bytef*)filename, strlen(filename));
    return crc;
}

Referenced by GetUtf8Name().

GetUtf8Name()

CStringA CZipEnumerator::GetUtf8Name ( PCSTR  originalName, const BYTE *  extraField, DWORD  extraFieldLen  )

static

Definition at line 44 of file CZipEnumerator.cpp.

{
    ATLASSERT(originalName);
    ATLASSERT(extraField);
 
    if (extraFieldLen < EF_HEADER_SIZE)
        return ""; // Failure
 
    const BYTE* pbField = extraField;
    const BYTE* pbEnd = extraField + extraFieldLen;
 
    while (pbField + EF_HEADER_SIZE <= pbEnd)
    {
        // Beware of alignment exception
        WORD fieldId, fieldSize;
        CopyMemory(&fieldId, pbField, sizeof(fieldId));
        CopyMemory(&fieldSize, pbField + 2, sizeof(fieldSize));
 
        if (fieldId != EF_UNIPATH)
        {
            pbField += EF_HEADER_SIZE + fieldSize; // Next field
            continue;
        }
 
        if (fieldSize < 5 || pbField + EF_HEADER_SIZE + fieldSize > pbEnd)
            return "";
 
        const BYTE* fieldData = pbField + EF_HEADER_SIZE;
        BYTE version = fieldData[0];
        if (version != EF_UNIPATH_VERSION)
            return ""; // Failure
 
        DWORD storedCRC, calculatedCRC = CalculateCRC32(originalName);
        CopyMemory(&storedCRC, fieldData + 1, sizeof(storedCRC));
        if (storedCRC != calculatedCRC)
            return ""; // Failure
 
        DWORD utf8NameLen = fieldSize - 5;
        if (utf8NameLen > 0)
            return CStringA((LPCSTR)(fieldData + 5), utf8NameLen); // Success
 
        pbField += EF_HEADER_SIZE + fieldSize; // Next field
    }
 
    return ""; // Failure
}

Referenced by Next().

Initialize()

BOOL CZipEnumerator::Initialize

(

IZip *

zip

)

Definition at line 18 of file CZipEnumerator.cpp.

{
    ATLASSERT(zip);
    m_Zip = zip;
    return Reset();
}

Referenced by CZipExtract::Extract(), and CEnumZipContents::Initialize().

Next()

BOOL CZipEnumerator::Next	(	CStringW &	name,
		unz_file_info64 &	info
	)

Definition at line 94 of file CZipEnumerator.cpp.

{
    INT err;
    unzFile uf = m_Zip->getZip();
 
    if (!m_First)
    {
        err = unzGoToNextFile(uf);
        if (err == UNZ_END_OF_LIST_OF_FILE)
            return FALSE;
    }
    m_First = FALSE;
 
    err = unzGetCurrentFileInfo64(uf, &info, NULL, 0, NULL, 0, NULL, 0);
    if (err != UNZ_OK)
        return FALSE;
 
    CAtlArray<BYTE> extra;
    extra.SetCount(info.size_file_extra);
 
    CStringA nameA;
    PSTR buf = nameA.GetBuffer(info.size_filename);
    err = unzGetCurrentFileInfo64(uf, NULL, buf, nameA.GetAllocLength(),
        (info.size_file_extra > 0) ? extra.GetData() : NULL, info.size_file_extra,
        NULL, 0);
    nameA.ReleaseBuffer(info.size_filename);
 
    if (err != UNZ_OK)
        return FALSE;
 
    CStringA utf8Name;
    if (info.size_file_extra > 0)
        utf8Name = GetUtf8Name(nameA, extra.GetData(), info.size_file_extra);
 
    if (utf8Name.GetLength() > 0)
        name = CA2WEX<MAX_PATH>(utf8Name, CP_UTF8);
    else if (info.flag & MINIZIP_UTF8_FLAG)
        name = CA2WEX<MAX_PATH>(nameA, CP_UTF8);
    else
        name = CA2WEX<MAX_PATH>(nameA, m_nCodePage);
 
    name.Replace('\\', '/');
 
    return TRUE;
}

Referenced by CZipExtract::Extract(), and NextUnique().

NextUnique()

BOOL CZipEnumerator::NextUnique	(	PCWSTR	prefix,
		CStringW &	name,
		bool &	folder,
		unz_file_info64 &	info
	)

Definition at line 140 of file CZipEnumerator.cpp.

{
    ATLASSERT(prefix);
    CStringW tmp;
    SIZE_T cchPrefix = wcslen(prefix);
    while (Next(tmp, info))
    {
        if (StrCmpNIW(tmp, prefix, cchPrefix) != 0)
            continue;
 
        INT ichSlash = tmp.Find(L'/', cchPrefix);
        folder = (ichSlash >= 0);
        tmp = name = (folder ? tmp.Mid(cchPrefix, ichSlash - cchPrefix) : tmp.Mid(cchPrefix));
        tmp.MakeLower();
 
        POSITION it = m_Returned.Find(tmp);
        if (!name.IsEmpty() && !it)
        {
            m_Returned.AddTail(tmp);
            return TRUE;
        }
    }
    return FALSE;
}

Referenced by CEnumZipContents::Next(), and CEnumZipContents::Skip().

Reset()

BOOL CZipEnumerator::Reset

(

)

Definition at line 25 of file CZipEnumerator.cpp.

{
    unzFile uf = m_Zip->getZip();
    m_First = TRUE;
    if (unzGoToFirstFile(uf) != UNZ_OK)
        return FALSE;
    m_Returned.RemoveAll();
    return TRUE;
}

Referenced by Initialize(), and CEnumZipContents::Reset().

Member Data Documentation

m_First

BOOL CZipEnumerator::m_First = TRUE

Definition at line 12 of file CZipEnumerator.hpp.

Referenced by Next(), and Reset().

m_nCodePage

UINT CZipEnumerator::m_nCodePage = GetZipCodePage(TRUE)

Definition at line 14 of file CZipEnumerator.hpp.

Referenced by Next().

m_Returned

CAtlList<CStringW> CZipEnumerator::m_Returned

Definition at line 13 of file CZipEnumerator.hpp.

Referenced by NextUnique(), and Reset().

m_Zip

CComPtr<IZip> CZipEnumerator::m_Zip

Definition at line 11 of file CZipEnumerator.hpp.

Referenced by Initialize(), Next(), and Reset().

---

The documentation for this struct was generated from the following files:

• dll/shellext/zipfldr/CZipEnumerator.hpp
• dll/shellext/zipfldr/CZipEnumerator.cpp