[USBPORT] Fix incorrect transfer direction for URBs. Patch by Vadim Galyant. CORE...
[reactos.git] / reactos / dll / win32 / shell32 / folders.cpp
1 /*
2 * Copyright 1997 Marcus Meissner
3 * Copyright 1998 Juergen Schmied
4 *
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2.1 of the License, or (at your option) any later version.
9 *
10 * This library is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
14 *
15 * You should have received a copy of the GNU Lesser General Public
16 * License along with this library; if not, write to the Free Software
17 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
18 */
19
20 #include "precomp.h"
21
22 WINE_DEFAULT_DEBUG_CHANNEL(shell);
23
24 WCHAR swShell32Name[MAX_PATH];
25
26 DWORD NumIconOverlayHandlers = 0;
27 IShellIconOverlayIdentifier ** Handlers = NULL;
28
29 static HRESULT getIconLocationForFolder(IShellFolder * psf, LPCITEMIDLIST pidl, UINT uFlags,
30 LPWSTR szIconFile, UINT cchMax, int *piIndex, UINT *pwFlags)
31 {
32 static const WCHAR shellClassInfo[] = { '.', 'S', 'h', 'e', 'l', 'l', 'C', 'l', 'a', 's', 's', 'I', 'n', 'f', 'o', 0 };
33 static const WCHAR iconFile[] = { 'I', 'c', 'o', 'n', 'F', 'i', 'l', 'e', 0 };
34 static const WCHAR clsid[] = { 'C', 'L', 'S', 'I', 'D', 0 };
35 static const WCHAR clsid2[] = { 'C', 'L', 'S', 'I', 'D', '2', 0 };
36 static const WCHAR iconIndex[] = { 'I', 'c', 'o', 'n', 'I', 'n', 'd', 'e', 'x', 0 };
37 static const WCHAR wszDesktopIni[] = { 'd','e','s','k','t','o','p','.','i','n','i',0 };
38 int icon_idx;
39
40 if (!(uFlags & GIL_DEFAULTICON) && (_ILGetFileAttributes(ILFindLastID(pidl), NULL, 0) & (FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_READONLY)) != 0 )
41 {
42 WCHAR wszFolderPath[MAX_PATH];
43
44 if (!ILGetDisplayNameExW(psf, pidl, wszFolderPath, 0))
45 return FALSE;
46
47 PathAppendW(wszFolderPath, wszDesktopIni);
48
49 if (PathFileExistsW(wszFolderPath))
50 {
51 WCHAR wszPath[MAX_PATH];
52 WCHAR wszCLSIDValue[CHARS_IN_GUID];
53
54 if (GetPrivateProfileStringW(shellClassInfo, iconFile, NULL, wszPath, MAX_PATH, wszFolderPath))
55 {
56 ExpandEnvironmentStringsW(wszPath, szIconFile, cchMax);
57
58 *piIndex = GetPrivateProfileIntW(shellClassInfo, iconIndex, 0, wszFolderPath);
59 return S_OK;
60 }
61 else if (GetPrivateProfileStringW(shellClassInfo, clsid, NULL, wszCLSIDValue, CHARS_IN_GUID, wszFolderPath) &&
62 HCR_GetIconW(wszCLSIDValue, szIconFile, NULL, cchMax, &icon_idx))
63 {
64 *piIndex = icon_idx;
65 return S_OK;
66 }
67 else if (GetPrivateProfileStringW(shellClassInfo, clsid2, NULL, wszCLSIDValue, CHARS_IN_GUID, wszFolderPath) &&
68 HCR_GetIconW(wszCLSIDValue, szIconFile, NULL, cchMax, &icon_idx))
69 {
70 *piIndex = icon_idx;
71 return S_OK;
72 }
73 }
74 }
75
76 static const WCHAR folder[] = { 'F', 'o', 'l', 'd', 'e', 'r', 0 };
77
78 if (!HCR_GetIconW(folder, szIconFile, NULL, cchMax, &icon_idx))
79 {
80 lstrcpynW(szIconFile, swShell32Name, cchMax);
81 icon_idx = -IDI_SHELL_FOLDER;
82 }
83
84 if (uFlags & GIL_OPENICON)
85 *piIndex = icon_idx < 0 ? icon_idx - 1 : icon_idx + 1;
86 else
87 *piIndex = icon_idx;
88
89 return S_OK;
90 }
91
92 static void InitIconOverlays(void)
93 {
94 HKEY hKey;
95 DWORD dwIndex, dwResult, dwSize;
96 WCHAR szName[MAX_PATH];
97 WCHAR szValue[100];
98 CLSID clsid;
99
100 if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Explorer\\ShellIconOverlayIdentifiers", 0, KEY_READ, &hKey) != ERROR_SUCCESS)
101 return;
102
103 if (RegQueryInfoKeyW(hKey, NULL, NULL, NULL, &dwResult, NULL, NULL, NULL, NULL, NULL, NULL, NULL) != ERROR_SUCCESS)
104 {
105 RegCloseKey(hKey);
106 return;
107 }
108
109 Handlers = (IShellIconOverlayIdentifier **)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, dwResult * sizeof(IShellIconOverlayIdentifier*));
110 if (!Handlers)
111 {
112 RegCloseKey(hKey);
113 return;
114 }
115
116 dwIndex = 0;
117
118 CoInitialize(0);
119
120 do
121 {
122 dwSize = sizeof(szName) / sizeof(WCHAR);
123 dwResult = RegEnumKeyExW(hKey, dwIndex, szName, &dwSize, NULL, NULL, NULL, NULL);
124
125 if (dwResult == ERROR_NO_MORE_ITEMS)
126 break;
127
128 if (dwResult == ERROR_SUCCESS)
129 {
130 dwSize = sizeof(szValue) / sizeof(WCHAR);
131 if (RegGetValueW(hKey, szName, NULL, RRF_RT_REG_SZ, NULL, szValue, &dwSize) == ERROR_SUCCESS)
132 {
133 CComPtr<IShellIconOverlayIdentifier> Overlay;
134
135 CLSIDFromString(szValue, &clsid);
136 dwResult = CoCreateInstance(clsid, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARG(IShellIconOverlayIdentifier, &Overlay));
137 if (dwResult == S_OK)
138 {
139 Handlers[NumIconOverlayHandlers] = Overlay.Detach();
140 NumIconOverlayHandlers++;
141 }
142 }
143 }
144
145 dwIndex++;
146
147 } while(1);
148
149 RegCloseKey(hKey);
150 }
151
152 BOOL
153 GetIconOverlay(LPCITEMIDLIST pidl, WCHAR * wTemp, int* pIndex)
154 {
155 DWORD Index;
156 HRESULT hResult;
157 int Priority;
158 int HighestPriority;
159 ULONG IconIndex;
160 ULONG Flags;
161 WCHAR szPath[MAX_PATH];
162
163 if(!SHGetPathFromIDListW(pidl, szPath))
164 return FALSE;
165
166 if (!Handlers)
167 InitIconOverlays();
168
169 HighestPriority = 101;
170 IconIndex = NumIconOverlayHandlers;
171 for(Index = 0; Index < NumIconOverlayHandlers; Index++)
172 {
173 hResult = Handlers[Index]->IsMemberOf(szPath, SFGAO_FILESYSTEM);
174 if (hResult == S_OK)
175 {
176 hResult = Handlers[Index]->GetPriority(&Priority);
177 if (hResult == S_OK)
178 {
179 if (Priority < HighestPriority)
180 {
181 HighestPriority = Priority;
182 IconIndex = Index;
183 }
184 }
185 }
186 }
187
188 if (IconIndex == NumIconOverlayHandlers)
189 return FALSE;
190
191 hResult = Handlers[IconIndex]->GetOverlayInfo(wTemp, MAX_PATH, pIndex, &Flags);
192
193 if (hResult == S_OK)
194 return TRUE;
195 else
196 return FALSE;
197 }
198
199 HRESULT CFSExtractIcon_CreateInstance(IShellFolder * psf, LPCITEMIDLIST pidl, REFIID iid, LPVOID * ppvOut)
200 {
201 CComPtr<IDefaultExtractIconInit> initIcon;
202 HRESULT hr;
203 int icon_idx = 0;
204 UINT flags = 0; // FIXME: Use it!
205 CHAR sTemp[MAX_PATH] = "";
206 WCHAR wTemp[MAX_PATH] = L"";
207
208 hr = SHCreateDefaultExtractIcon(IID_PPV_ARG(IDefaultExtractIconInit,&initIcon));
209 if (FAILED(hr))
210 return hr;
211
212 if (_ILIsFolder (pidl))
213 {
214 if (SUCCEEDED(getIconLocationForFolder(psf,
215 pidl, 0, wTemp, _countof(wTemp),
216 &icon_idx,
217 &flags)))
218 {
219 initIcon->SetNormalIcon(wTemp, icon_idx);
220 // FIXME: if/when getIconLocationForFolder does something for
221 // GIL_FORSHORTCUT, code below should be uncommented. and
222 // the following line removed.
223 initIcon->SetShortcutIcon(wTemp, icon_idx);
224 }
225 if (SUCCEEDED(getIconLocationForFolder(psf,
226 pidl, GIL_DEFAULTICON, wTemp, _countof(wTemp),
227 &icon_idx,
228 &flags)))
229 {
230 initIcon->SetDefaultIcon(wTemp, icon_idx);
231 }
232 // if (SUCCEEDED(getIconLocationForFolder(psf,
233 // pidl, GIL_FORSHORTCUT, wTemp, _countof(wTemp),
234 // &icon_idx,
235 // &flags)))
236 // {
237 // initIcon->SetShortcutIcon(wTemp, icon_idx);
238 // }
239 if (SUCCEEDED(getIconLocationForFolder(psf,
240 pidl, GIL_OPENICON, wTemp, _countof(wTemp),
241 &icon_idx,
242 &flags)))
243 {
244 initIcon->SetOpenIcon(wTemp, icon_idx);
245 }
246 }
247 else
248 {
249 BOOL found = FALSE;
250
251 if (_ILGetExtension(pidl, sTemp, _countof(sTemp)))
252 {
253 if (HCR_MapTypeToValueA(sTemp, sTemp, _countof(sTemp), TRUE)
254 && HCR_GetIconA(sTemp, sTemp, NULL, _countof(sTemp), &icon_idx))
255 {
256 if (!lstrcmpA("%1", sTemp)) /* icon is in the file */
257 {
258 ILGetDisplayNameExW(psf, pidl, wTemp, 0);
259 icon_idx = 0;
260 }
261 else
262 {
263 MultiByteToWideChar(CP_ACP, 0, sTemp, -1, wTemp, _countof(wTemp));
264 }
265
266 found = TRUE;
267 }
268 else if (!lstrcmpiA(sTemp, "lnkfile"))
269 {
270 /* extract icon from shell shortcut */
271 CComPtr<IShellLinkW> psl;
272 CComPtr<IExtractIconW> pei;
273
274 HRESULT hr = psf->GetUIObjectOf(NULL, 1, &pidl, IID_NULL_PPV_ARG(IShellLinkW, &psl));
275 if (SUCCEEDED(hr))
276 {
277 hr = psl->GetIconLocation(wTemp, _countof(wTemp), &icon_idx);
278 if (FAILED(hr) || !*wTemp)
279 {
280 /* The icon was not found directly, try to retrieve it from the shell link target */
281 hr = psl->QueryInterface(IID_PPV_ARG(IExtractIconW, &pei));
282 if (FAILED(hr) || !pei)
283 TRACE("No IExtractIconW interface!\n");
284 else
285 hr = pei->GetIconLocation(GIL_FORSHELL, wTemp, _countof(wTemp), &icon_idx, &flags);
286 }
287
288 if (SUCCEEDED(hr) && *wTemp)
289 found = TRUE;
290 }
291 }
292 }
293
294 /* FIXME: We should normally use the correct icon format according to 'flags' */
295 if (!found)
296 /* default icon */
297 initIcon->SetNormalIcon(swShell32Name, 0);
298 else
299 initIcon->SetNormalIcon(wTemp, icon_idx);
300 }
301
302 return initIcon->QueryInterface(iid, ppvOut);
303 }