[NTOS:PS] Use KD routine to safely read memory from thread stack
[reactos.git] / dll / shellext / zipfldr / CZipEnumerator.hpp
1 /*
2 * PROJECT: ReactOS Zip Shell Extension
3 * LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
4 * PURPOSE: CZipEnumerator
5 * COPYRIGHT: Copyright 2017 Mark Jansen (mark.jansen@reactos.org)
6 */
7
8 struct CZipEnumerator
9 {
10 private:
11 CComPtr<IZip> m_Zip;
12 bool m_First;
13 CAtlList<CStringA> m_Returned;
14 public:
15 CZipEnumerator()
16 :m_First(true)
17 {
18 }
19
20 bool initialize(IZip* zip)
21 {
22 m_Zip = zip;
23 return reset();
24 }
25
26 bool reset()
27 {
28 unzFile uf = m_Zip->getZip();
29 m_First = true;
30 if (unzGoToFirstFile(uf) != UNZ_OK)
31 return false;
32 m_Returned.RemoveAll();
33 return true;
34 }
35
36 bool next_unique(const char* prefix, CStringA& name, bool& folder, unz_file_info64& info)
37 {
38 size_t len = strlen(prefix);
39 CStringA tmp;
40 while (next(tmp, info))
41 {
42 if (!_strnicmp(tmp, prefix, len))
43 {
44 int pos = tmp.Find('/', len);
45 if (pos < 0)
46 {
47 name = tmp.Mid(len);
48 folder = false;
49 }
50 else
51 {
52 name = tmp.Mid(len, pos - len);
53 folder = true;
54 }
55 tmp = name;
56 tmp.MakeLower();
57
58 POSITION it = m_Returned.Find(tmp);
59 if (!name.IsEmpty() && !it)
60 {
61 m_Returned.AddTail(tmp);
62 return true;
63 }
64 }
65 }
66 return false;
67 }
68
69 bool next(CStringA& name, unz_file_info64& info)
70 {
71 int err;
72
73 unzFile uf = m_Zip->getZip();
74 if (!m_First)
75 {
76 err = unzGoToNextFile(uf);
77 if (err == UNZ_END_OF_LIST_OF_FILE)
78 {
79 return false;
80 }
81 }
82 m_First = false;
83
84 err = unzGetCurrentFileInfo64(uf, &info, NULL, 0, NULL, 0, NULL, 0);
85 if (err == UNZ_OK)
86 {
87 PSTR buf = name.GetBuffer(info.size_filename);
88 err = unzGetCurrentFileInfo64(uf, NULL, buf, name.GetAllocLength(), NULL, 0, NULL, 0);
89 name.ReleaseBuffer(info.size_filename);
90 name.Replace('\\', '/');
91 }
92 return err == UNZ_OK;
93 }
94 };
95