4 * Copyright 2016 Giannis Adamopoulos
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 WINE_DEFAULT_DEBUG_CHANNEL (shell
);
25 HRESULT
CGuidItemExtractIcon_CreateInstance(LPCITEMIDLIST pidl
, REFIID iid
, LPVOID
* ppvOut
)
27 CComPtr
<IDefaultExtractIconInit
> initIcon
;
31 WCHAR wTemp
[MAX_PATH
];
33 hr
= SHCreateDefaultExtractIcon(IID_PPV_ARG(IDefaultExtractIconInit
,&initIcon
));
37 if (_ILIsDesktop(pidl
))
39 initIcon
->SetNormalIcon(swShell32Name
, -IDI_SHELL_DESKTOP
);
40 return initIcon
->QueryInterface(iid
, ppvOut
);
43 riid
= _ILGetGUIDPointer(pidl
);
47 /* my computer and other shell extensions */
48 static const WCHAR fmt
[] = { 'C', 'L', 'S', 'I', 'D', '\\',
49 '{', '%', '0', '8', 'l', 'x', '-', '%', '0', '4', 'x', '-', '%', '0', '4', 'x', '-',
50 '%', '0', '2', 'x', '%', '0', '2', 'x', '-', '%', '0', '2', 'x', '%', '0', '2', 'x',
51 '%', '0', '2', 'x', '%', '0', '2', 'x', '%', '0', '2', 'x', '%', '0', '2', 'x', '}', 0
56 riid
->Data1
, riid
->Data2
, riid
->Data3
,
57 riid
->Data4
[0], riid
->Data4
[1], riid
->Data4
[2], riid
->Data4
[3],
58 riid
->Data4
[4], riid
->Data4
[5], riid
->Data4
[6], riid
->Data4
[7]);
60 const WCHAR
* iconname
= NULL
;
61 if (_ILIsBitBucket(pidl
))
63 static const WCHAR szFull
[] = {'F','u','l','l',0};
64 static const WCHAR szEmpty
[] = {'E','m','p','t','y',0};
65 CComPtr
<IEnumIDList
> EnumIDList
;
68 CComPtr
<IShellFolder2
> psfRecycleBin
;
69 CComPtr
<IShellFolder
> psfDesktop
;
70 hr
= SHGetDesktopFolder(&psfDesktop
);
73 hr
= psfDesktop
->BindToObject(pidl
, NULL
, IID_PPV_ARG(IShellFolder2
, &psfRecycleBin
));
75 hr
= psfRecycleBin
->EnumObjects(NULL
, SHCONTF_FOLDERS
| SHCONTF_NONFOLDERS
, &EnumIDList
);
78 LPITEMIDLIST pidl
= NULL
;
79 if (SUCCEEDED(hr
) && (hr
= EnumIDList
->Next(1, &pidl
, &itemcount
)) == S_OK
)
88 if (HCR_GetIconW(xriid
, wTemp
, iconname
, MAX_PATH
, &icon_idx
))
90 initIcon
->SetNormalIcon(wTemp
, icon_idx
);
94 if (IsEqualGUID(*riid
, CLSID_MyComputer
))
95 initIcon
->SetNormalIcon(swShell32Name
, -IDI_SHELL_MY_COMPUTER
);
96 else if (IsEqualGUID(*riid
, CLSID_MyDocuments
))
97 initIcon
->SetNormalIcon(swShell32Name
, -IDI_SHELL_MY_DOCUMENTS
);
98 else if (IsEqualGUID(*riid
, CLSID_NetworkPlaces
))
99 initIcon
->SetNormalIcon(swShell32Name
, -IDI_SHELL_MY_NETWORK_PLACES
);
101 initIcon
->SetNormalIcon(swShell32Name
, -IDI_SHELL_FOLDER
);
104 return initIcon
->QueryInterface(iid
, ppvOut
);
108 public CComObjectRootEx
<CComMultiThreadModelNoCS
>,
113 CAtlStringW m_rootPath
;
114 CComHeapPtr
<ITEMIDLIST
> m_pidlRoot
;
116 HRESULT
GetGuidItemAttributes (LPCITEMIDLIST pidl
, LPDWORD pdwAttributes
);
120 HRESULT WINAPI
Initialize(const GUID
*pGuid
, LPCITEMIDLIST pidlRoot
, LPCWSTR lpszPath
);
123 virtual HRESULT WINAPI
ParseDisplayName(HWND hwndOwner
, LPBC pbc
, LPOLESTR lpszDisplayName
, ULONG
*pchEaten
, PIDLIST_RELATIVE
*ppidl
, ULONG
*pdwAttributes
);
124 virtual HRESULT WINAPI
EnumObjects(HWND hwndOwner
, DWORD dwFlags
, LPENUMIDLIST
*ppEnumIDList
);
125 virtual HRESULT WINAPI
BindToObject(PCUIDLIST_RELATIVE pidl
, LPBC pbcReserved
, REFIID riid
, LPVOID
*ppvOut
);
126 virtual HRESULT WINAPI
BindToStorage(PCUIDLIST_RELATIVE pidl
, LPBC pbcReserved
, REFIID riid
, LPVOID
*ppvOut
);
127 virtual HRESULT WINAPI
CompareIDs(LPARAM lParam
, PCUIDLIST_RELATIVE pidl1
, PCUIDLIST_RELATIVE pidl2
);
128 virtual HRESULT WINAPI
CreateViewObject(HWND hwndOwner
, REFIID riid
, LPVOID
*ppvOut
);
129 virtual HRESULT WINAPI
GetAttributesOf(UINT cidl
, PCUITEMID_CHILD_ARRAY apidl
, DWORD
*rgfInOut
);
130 virtual HRESULT WINAPI
GetUIObjectOf(HWND hwndOwner
, UINT cidl
, PCUITEMID_CHILD_ARRAY apidl
, REFIID riid
, UINT
* prgfInOut
, LPVOID
* ppvOut
);
131 virtual HRESULT WINAPI
GetDisplayNameOf(PCUITEMID_CHILD pidl
, DWORD dwFlags
, LPSTRRET strRet
);
132 virtual HRESULT WINAPI
SetNameOf(HWND hwndOwner
, PCUITEMID_CHILD pidl
, LPCOLESTR lpName
, DWORD dwFlags
, PITEMID_CHILD
*pPidlOut
);
135 virtual HRESULT WINAPI
GetDefaultSearchGUID(GUID
*pguid
);
136 virtual HRESULT WINAPI
EnumSearches(IEnumExtraSearch
**ppenum
);
137 virtual HRESULT WINAPI
GetDefaultColumn(DWORD dwRes
, ULONG
*pSort
, ULONG
*pDisplay
);
138 virtual HRESULT WINAPI
GetDefaultColumnState(UINT iColumn
, DWORD
*pcsFlags
);
139 virtual HRESULT WINAPI
GetDetailsEx(PCUITEMID_CHILD pidl
, const SHCOLUMNID
*pscid
, VARIANT
*pv
);
140 virtual HRESULT WINAPI
GetDetailsOf(PCUITEMID_CHILD pidl
, UINT iColumn
, SHELLDETAILS
*psd
);
141 virtual HRESULT WINAPI
MapColumnToSCID(UINT column
, SHCOLUMNID
*pscid
);
143 DECLARE_NOT_AGGREGATABLE(CRegFolder
)
145 DECLARE_PROTECT_FINAL_CONSTRUCT()
147 BEGIN_COM_MAP(CRegFolder
)
148 COM_INTERFACE_ENTRY_IID(IID_IShellFolder2
, IShellFolder2
)
149 COM_INTERFACE_ENTRY_IID(IID_IShellFolder
, IShellFolder
)
153 CRegFolder::CRegFolder()
157 CRegFolder::~CRegFolder()
161 HRESULT WINAPI
CRegFolder::Initialize(const GUID
*pGuid
, LPCITEMIDLIST pidlRoot
, LPCWSTR lpszPath
)
163 memcpy(&m_guid
, pGuid
, sizeof(m_guid
));
165 m_rootPath
= lpszPath
;
167 return E_OUTOFMEMORY
;
169 m_pidlRoot
.Attach(ILClone(pidlRoot
));
171 return E_OUTOFMEMORY
;
176 HRESULT
CRegFolder::GetGuidItemAttributes (LPCITEMIDLIST pidl
, LPDWORD pdwAttributes
)
178 /* First try to get them from the registry */
179 if (HCR_GetFolderAttributes(pidl
, pdwAttributes
) && *pdwAttributes
)
185 /* If we can't get it from the registry we have to query the child */
186 CComPtr
<IShellFolder
> psf2
;
187 if (SUCCEEDED(BindToObject(pidl
, 0, IID_PPV_ARG(IShellFolder
, &psf2
))))
189 return psf2
->GetAttributesOf(0, NULL
, pdwAttributes
);
193 *pdwAttributes
&= SFGAO_CANLINK
;
197 HRESULT WINAPI
CRegFolder::ParseDisplayName(HWND hwndOwner
, LPBC pbc
, LPOLESTR lpszDisplayName
,
198 ULONG
*pchEaten
, PIDLIST_RELATIVE
*ppidl
, ULONG
*pdwAttributes
)
202 if (!lpszDisplayName
|| !ppidl
)
210 UINT cch
= wcslen(lpszDisplayName
);
211 if (cch
< 39 || lpszDisplayName
[0] != L
':' || lpszDisplayName
[1] != L
':')
214 pidl
= _ILCreateGuidFromStrW(lpszDisplayName
+ 2);
221 if (pdwAttributes
&& *pdwAttributes
)
223 GetGuidItemAttributes(*ppidl
, pdwAttributes
);
228 HRESULT hr
= SHELL32_ParseNextElement(this, hwndOwner
, pbc
, &pidl
, lpszDisplayName
+ 41, pchEaten
, pdwAttributes
);
239 HRESULT WINAPI
CRegFolder::EnumObjects(HWND hwndOwner
, DWORD dwFlags
, LPENUMIDLIST
*ppEnumIDList
)
244 HRESULT WINAPI
CRegFolder::BindToObject(PCUIDLIST_RELATIVE pidl
, LPBC pbcReserved
, REFIID riid
, LPVOID
*ppvOut
)
246 CComPtr
<IPersistFolder
> pFolder
;
249 if (!ppvOut
|| !pidl
|| !pidl
->mkid
.cb
)
254 GUID
*pGUID
= _ILGetGUIDPointer(pidl
);
257 ERR("CRegFolder::BindToObject called for non guid item!\n");
261 LPITEMIDLIST pidlChild
= ILCloneFirst (pidl
);
263 return E_OUTOFMEMORY
;
265 CComPtr
<IShellFolder
> psf
;
266 hr
= SHELL32_CoCreateInitSF(m_pidlRoot
, NULL
, pidlChild
, pGUID
, -1, IID_PPV_ARG(IShellFolder
, &psf
));
271 if (_ILIsPidlSimple (pidl
))
273 return psf
->QueryInterface(riid
, ppvOut
);
277 return psf
->BindToObject(ILGetNext (pidl
), pbcReserved
, riid
, ppvOut
);
281 HRESULT WINAPI
CRegFolder::BindToStorage(PCUIDLIST_RELATIVE pidl
, LPBC pbcReserved
, REFIID riid
, LPVOID
*ppvOut
)
286 HRESULT WINAPI
CRegFolder::CompareIDs(LPARAM lParam
, PCUIDLIST_RELATIVE pidl1
, PCUIDLIST_RELATIVE pidl2
)
288 if (!pidl1
|| !pidl2
|| pidl1
->mkid
.cb
== 0 || pidl2
->mkid
.cb
== 0)
290 ERR("Got an empty pidl!\n");
294 BOOL bIsGuidFolder1
= _ILIsSpecialFolder(pidl1
);
295 BOOL bIsGuidFolder2
= _ILIsSpecialFolder(pidl2
);
297 if (!bIsGuidFolder1
&& !bIsGuidFolder2
)
299 ERR("Got no guid pidl!\n");
302 else if (bIsGuidFolder1
&& bIsGuidFolder2
)
304 return SHELL32_CompareDetails(this, lParam
, pidl1
, pidl2
);
307 /* Guid folders come first compared to everything else */
308 return MAKE_COMPARE_HRESULT(bIsGuidFolder1
? -1 : 1);
311 HRESULT WINAPI
CRegFolder::CreateViewObject(HWND hwndOwner
, REFIID riid
, LPVOID
*ppvOut
)
316 HRESULT WINAPI
CRegFolder::GetAttributesOf(UINT cidl
, PCUITEMID_CHILD_ARRAY apidl
, DWORD
*rgfInOut
)
318 if (!rgfInOut
|| !cidl
|| !apidl
)
324 while(cidl
> 0 && *apidl
)
326 if (_ILIsSpecialFolder(*apidl
))
327 GetGuidItemAttributes(*apidl
, rgfInOut
);
329 ERR("Got an unkown pidl here!\n");
334 /* make sure SFGAO_VALIDATE is cleared, some apps depend on that */
335 *rgfInOut
&= ~SFGAO_VALIDATE
;
340 HRESULT WINAPI
CRegFolder::GetUIObjectOf(HWND hwndOwner
, UINT cidl
, PCUITEMID_CHILD_ARRAY apidl
,
341 REFIID riid
, UINT
* prgfInOut
, LPVOID
* ppvOut
)
344 HRESULT hr
= E_INVALIDARG
;
351 if ((IsEqualIID (riid
, IID_IExtractIconA
) || IsEqualIID (riid
, IID_IExtractIconW
)) && (cidl
== 1))
353 hr
= CGuidItemExtractIcon_CreateInstance(apidl
[0], riid
, &pObj
);
363 HRESULT WINAPI
CRegFolder::GetDisplayNameOf(PCUITEMID_CHILD pidl
, DWORD dwFlags
, LPSTRRET strRet
)
365 if (!strRet
|| !_ILIsSpecialFolder(pidl
))
370 LPWSTR pszPath
= (LPWSTR
)CoTaskMemAlloc((MAX_PATH
+ 1) * sizeof(WCHAR
));
372 return E_OUTOFMEMORY
;
374 /* parsing name like ::{...} */
377 SHELL32_GUIDToStringW(m_guid
, &pszPath
[2]);
379 strRet
->uType
= STRRET_WSTR
;
380 strRet
->pOleStr
= pszPath
;
386 GUID
const *clsid
= _ILGetGUIDPointer (pidl
);
388 /* First of all check if we need to query the name from the child item */
389 if (GET_SHGDN_FOR (dwFlags
) == SHGDN_FORPARSING
&&
390 GET_SHGDN_RELATION (dwFlags
) == SHGDN_NORMAL
)
392 int bWantsForParsing
= FALSE
;
395 * We can only get a filesystem path from a shellfolder if the
396 * value WantsFORPARSING in CLSID\\{...}\\shellfolder exists.
398 * Exception: The MyComputer folder doesn't have this key,
399 * but any other filesystem backed folder it needs it.
401 if (IsEqualIID (*clsid
, CLSID_MyComputer
))
403 bWantsForParsing
= TRUE
;
408 if (HCR_RegOpenClassIDKey(*clsid
, &hkeyClass
))
410 LONG res
= SHGetValueW(hkeyClass
, L
"Shellfolder", L
"WantsForParsing", NULL
, NULL
, NULL
);
411 bWantsForParsing
= (res
== ERROR_SUCCESS
);
412 RegCloseKey(hkeyClass
);
416 if (bWantsForParsing
)
419 * we need the filesystem path to the destination folder.
420 * Only the folder itself can know it
422 return SHELL32_GetDisplayNameOfChild (this, pidl
, dwFlags
, strRet
);
426 /* Allocate the buffer for the result */
427 LPWSTR pszPath
= (LPWSTR
)CoTaskMemAlloc((MAX_PATH
+ 1) * sizeof(WCHAR
));
429 return E_OUTOFMEMORY
;
433 if (GET_SHGDN_FOR (dwFlags
) == SHGDN_FORPARSING
)
435 wcscpy(pszPath
, m_rootPath
);
436 PWCHAR pItemName
= &pszPath
[wcslen(pszPath
)];
438 /* parsing name like ::{...} */
441 SHELL32_GUIDToStringW (*clsid
, &pItemName
[2]);
445 /* user friendly name */
446 if (!HCR_GetClassNameW (*clsid
, pszPath
, MAX_PATH
))
452 strRet
->uType
= STRRET_WSTR
;
453 strRet
->pOleStr
= pszPath
;
457 CoTaskMemFree(pszPath
);
463 HRESULT WINAPI
CRegFolder::SetNameOf(HWND hwndOwner
, PCUITEMID_CHILD pidl
, /* simple pidl */
464 LPCOLESTR lpName
, DWORD dwFlags
, PITEMID_CHILD
*pPidlOut
)
466 GUID
const *clsid
= _ILGetGUIDPointer (pidl
);
473 ERR("Pidl is not reg item!\n");
477 hr
= StringFromCLSID(*clsid
, &pStr
);
478 if (FAILED_UNEXPECTEDLY(hr
))
481 swprintf(szName
, L
"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\CLSID\\%s", pStr
);
483 DWORD cbData
= (wcslen(lpName
) + 1) * sizeof(WCHAR
);
484 LONG res
= SHSetValueW(HKEY_CURRENT_USER
, szName
, NULL
, RRF_RT_REG_SZ
, lpName
, cbData
);
488 if (res
== ERROR_SUCCESS
)
490 *pPidlOut
= ILClone(pidl
);
498 HRESULT WINAPI
CRegFolder::GetDefaultSearchGUID(GUID
*pguid
)
503 HRESULT WINAPI
CRegFolder::EnumSearches(IEnumExtraSearch
** ppenum
)
508 HRESULT WINAPI
CRegFolder::GetDefaultColumn(DWORD dwRes
, ULONG
*pSort
, ULONG
*pDisplay
)
518 HRESULT WINAPI
CRegFolder::GetDefaultColumnState(UINT iColumn
, DWORD
*pcsFlags
)
522 *pcsFlags
= SHCOLSTATE_TYPE_STR
| SHCOLSTATE_ONBYDEFAULT
;
526 HRESULT WINAPI
CRegFolder::GetDetailsEx(PCUITEMID_CHILD pidl
, const SHCOLUMNID
*pscid
, VARIANT
*pv
)
531 HRESULT WINAPI
CRegFolder::GetDetailsOf(PCUITEMID_CHILD pidl
, UINT iColumn
, SHELLDETAILS
*psd
)
533 if (!psd
|| iColumn
>= 2)
536 GUID
const *clsid
= _ILGetGUIDPointer (pidl
);
540 ERR("Pidl is not reg item!\n");
547 return GetDisplayNameOf(pidl
, SHGDN_NORMAL
| SHGDN_INFOLDER
, &psd
->str
);
548 case 1: /* comment */
550 if (HCR_RegOpenClassIDKey(*clsid
, &hKey
))
552 psd
->str
.cStr
[0] = 0x00;
553 psd
->str
.uType
= STRRET_CSTR
;
554 RegLoadMUIStringA(hKey
, "InfoTip", psd
->str
.cStr
, MAX_PATH
, NULL
, 0, NULL
);
562 HRESULT WINAPI
CRegFolder::MapColumnToSCID(UINT column
, SHCOLUMNID
*pscid
)
567 /* In latest windows version this is exported but it takes different arguments! */
568 HRESULT
CRegFolder_CreateInstance(const GUID
*pGuid
, LPCITEMIDLIST pidlRoot
, LPCWSTR lpszPath
, REFIID riid
, void **ppv
)
570 return ShellObjectCreatorInit
<CRegFolder
>(pGuid
, pidlRoot
, lpszPath
, riid
, ppv
);