reactos/dll/shellext/zipfldr/CZipEnumerator.hpp
Katayama Hirofumi MZ fcc4347898
[SHELLEXT][ZIPFLDR] Be case sensitive (#2292)
The filenames in a zip folder were all lowercase. Now the filenames are case sensitive.
2020-01-30 15:05:50 +09:00

96 lines
2.3 KiB
C++

/*
* PROJECT: ReactOS Zip Shell Extension
* LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
* PURPOSE: CZipEnumerator
* COPYRIGHT: Copyright 2017 Mark Jansen (mark.jansen@reactos.org)
*/
struct CZipEnumerator
{
private:
CComPtr<IZip> m_Zip;
bool m_First;
CAtlList<CStringA> m_Returned;
public:
CZipEnumerator()
:m_First(true)
{
}
bool initialize(IZip* zip)
{
m_Zip = zip;
return reset();
}
bool reset()
{
unzFile uf = m_Zip->getZip();
m_First = true;
if (unzGoToFirstFile(uf) != UNZ_OK)
return false;
m_Returned.RemoveAll();
return true;
}
bool next_unique(const char* prefix, CStringA& name, bool& folder, unz_file_info64& info)
{
size_t len = strlen(prefix);
CStringA tmp;
while (next(tmp, info))
{
if (!_strnicmp(tmp, prefix, len))
{
int pos = tmp.Find('/', len);
if (pos < 0)
{
name = tmp.Mid(len);
folder = false;
}
else
{
name = tmp.Mid(len, pos - len);
folder = true;
}
tmp = name;
tmp.MakeLower();
POSITION it = m_Returned.Find(tmp);
if (!name.IsEmpty() && !it)
{
m_Returned.AddTail(tmp);
return true;
}
}
}
return false;
}
bool next(CStringA& name, unz_file_info64& info)
{
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)
{
PSTR buf = name.GetBuffer(info.size_filename);
err = unzGetCurrentFileInfo64(uf, NULL, buf, name.GetAllocLength(), NULL, 0, NULL, 0);
name.ReleaseBuffer(info.size_filename);
name.Replace('\\', '/');
}
return err == UNZ_OK;
}
};