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 WCHAR swShell32Name[MAX_PATH];
  23
  24 DWORD NumIconOverlayHandlers = 0;
  25 IShellIconOverlayIdentifier ** Handlers = NULL;
  26
  27 static HRESULT getIconLocationForFolder(IShellFolder * psf, LPCITEMIDLIST pidl, UINT uFlags,
  28                                         LPWSTR szIconFile, UINT cchMax, int *piIndex, UINT *pwFlags)
  29 {
  30     static const WCHAR shellClassInfo[] = { '.', 'S', 'h', 'e', 'l', 'l', 'C', 'l', 'a', 's', 's', 'I', 'n', 'f', 'o', 0 };
  31     static const WCHAR iconFile[] = { 'I', 'c', 'o', 'n', 'F', 'i', 'l', 'e', 0 };
  32     static const WCHAR clsid[] = { 'C', 'L', 'S', 'I', 'D', 0 };
  33     static const WCHAR clsid2[] = { 'C', 'L', 'S', 'I', 'D', '2', 0 };
  34     static const WCHAR iconIndex[] = { 'I', 'c', 'o', 'n', 'I', 'n', 'd', 'e', 'x', 0 };
  35     static const WCHAR wszDesktopIni[] = { 'd','e','s','k','t','o','p','.','i','n','i',0 };
  36     int icon_idx;
  37
  38     if (!(uFlags & GIL_DEFAULTICON) && (_ILGetFileAttributes(ILFindLastID(pidl), NULL, 0) & (FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_READONLY)) != 0 )
  39     {
  40         WCHAR wszFolderPath[MAX_PATH];
  41
  42         if (!ILGetDisplayNameExW(psf, pidl, wszFolderPath, 0))
  43             return FALSE;
  44
  45         PathAppendW(wszFolderPath, wszDesktopIni);
  46
  47         if (PathFileExistsW(wszFolderPath))
  48         {
  49             WCHAR wszPath[MAX_PATH];
  50             WCHAR wszCLSIDValue[CHARS_IN_GUID];
  51
  52             if (GetPrivateProfileStringW(shellClassInfo, iconFile, NULL, wszPath, MAX_PATH, wszFolderPath))
  53             {
  54                 ExpandEnvironmentStringsW(wszPath, szIconFile, cchMax);
  55
  56                 *piIndex = GetPrivateProfileIntW(shellClassInfo, iconIndex, 0, wszFolderPath);
  57                 return S_OK;
  58             }
  59             else if (GetPrivateProfileStringW(shellClassInfo, clsid, NULL, wszCLSIDValue, CHARS_IN_GUID, wszFolderPath) &&
  60                 HCR_GetIconW(wszCLSIDValue, szIconFile, NULL, cchMax, &icon_idx))
  61             {
  62                 *piIndex = icon_idx;
  63                 return S_OK;
  64             }
  65             else if (GetPrivateProfileStringW(shellClassInfo, clsid2, NULL, wszCLSIDValue, CHARS_IN_GUID, wszFolderPath) &&
  66                 HCR_GetIconW(wszCLSIDValue, szIconFile, NULL, cchMax, &icon_idx))
  67             {
  68                 *piIndex = icon_idx;
  69                 return S_OK;
  70             }
  71         }
  72     }
  73
  74     static const WCHAR folder[] = { 'F', 'o', 'l', 'd', 'e', 'r', 0 };
  75
  76     if (!HCR_GetIconW(folder, szIconFile, NULL, cchMax, &icon_idx))
  77     {
  78         lstrcpynW(szIconFile, swShell32Name, cchMax);
  79         icon_idx = -IDI_SHELL_FOLDER;
  80     }
  81
  82     if (uFlags & GIL_OPENICON)
  83         *piIndex = icon_idx < 0 ? icon_idx - 1 : icon_idx + 1;
  84     else
  85         *piIndex = icon_idx;
  86
  87     return S_OK;
  88 }
  89
  90 void InitIconOverlays(void)
  91 {
  92     HKEY hKey;
  93     DWORD dwIndex, dwResult, dwSize;
  94     WCHAR szName[MAX_PATH];
  95     WCHAR szValue[100];
  96     CLSID clsid;
  97
  98     if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Explorer\\ShellIconOverlayIdentifiers", 0, KEY_READ, &hKey) != ERROR_SUCCESS)
  99         return;
 100
 101     if (RegQueryInfoKeyW(hKey, NULL, NULL, NULL, &dwResult, NULL, NULL, NULL, NULL, NULL, NULL, NULL) != ERROR_SUCCESS)
 102     {
 103         RegCloseKey(hKey);
 104         return;
 105     }
 106
 107     Handlers = (IShellIconOverlayIdentifier **)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, dwResult * sizeof(IShellIconOverlayIdentifier*));
 108     if (!Handlers)
 109     {
 110         RegCloseKey(hKey);
 111         return;
 112     }
 113
 114     dwIndex = 0;
 115
 116     CoInitialize(0);
 117
 118     do
 119     {
 120         dwSize = sizeof(szName) / sizeof(WCHAR);
 121         dwResult = RegEnumKeyExW(hKey, dwIndex, szName, &dwSize, NULL, NULL, NULL, NULL);
 122
 123         if (dwResult == ERROR_NO_MORE_ITEMS)
 124             break;
 125
 126         if (dwResult == ERROR_SUCCESS)
 127         {
 128             dwSize = sizeof(szValue) / sizeof(WCHAR);
 129             if (RegGetValueW(hKey, szName, NULL, RRF_RT_REG_SZ, NULL, szValue, &dwSize) == ERROR_SUCCESS)
 130             {
 131                 CComPtr<IShellIconOverlayIdentifier> Overlay;
 132
 133                 CLSIDFromString(szValue, &clsid);
 134                 dwResult = CoCreateInstance(clsid, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARG(IShellIconOverlayIdentifier, &Overlay));
 135                 if (dwResult == S_OK)
 136                 {
 137                     Handlers[NumIconOverlayHandlers] = Overlay.Detach();
 138                     NumIconOverlayHandlers++;
 139                 }
 140             }
 141         }
 142
 143         dwIndex++;
 144
 145     } while(1);
 146
 147     RegCloseKey(hKey);
 148 }
 149
 150 BOOL
 151 GetIconOverlay(LPCITEMIDLIST pidl, WCHAR * wTemp, int* pIndex)
 152 {
 153     DWORD Index;
 154     HRESULT hResult;
 155     int Priority;
 156     int HighestPriority;
 157     ULONG IconIndex;
 158     ULONG Flags;
 159     WCHAR szPath[MAX_PATH];
 160
 161     if(!SHGetPathFromIDListW(pidl, szPath))
 162         return FALSE;
 163
 164
 165     HighestPriority = 101;
 166     IconIndex = NumIconOverlayHandlers;
 167     for(Index = 0; Index < NumIconOverlayHandlers; Index++)
 168     {
 169         hResult = Handlers[Index]->IsMemberOf(szPath, SFGAO_FILESYSTEM);
 170         if (hResult == S_OK)
 171         {
 172             hResult = Handlers[Index]->GetPriority(&Priority);
 173             if (hResult == S_OK)
 174             {
 175                 if (Priority < HighestPriority)
 176                 {
 177                     HighestPriority = Priority;
 178                     IconIndex = Index;
 179                 }
 180             }
 181         }
 182     }
 183
 184     if (IconIndex == NumIconOverlayHandlers)
 185         return FALSE;
 186
 187     hResult = Handlers[IconIndex]->GetOverlayInfo(wTemp, MAX_PATH, pIndex, &Flags);
 188
 189     if (hResult == S_OK)
 190         return TRUE;
 191     else
 192         return FALSE;
 193 }
 194
 195 HRESULT CGuidItemExtractIcon_CreateInstance(IShellFolder * psf, LPCITEMIDLIST pidl, REFIID iid, LPVOID * ppvOut)
 196 {
 197     CComPtr<IDefaultExtractIconInit>    initIcon;
 198     HRESULT hr;
 199     GUID const * riid;
 200     int icon_idx;
 201     WCHAR wTemp[MAX_PATH];
 202
 203     hr = SHCreateDefaultExtractIcon(IID_PPV_ARG(IDefaultExtractIconInit,&initIcon));
 204     if (FAILED(hr))
 205         return hr;
 206
 207     if (_ILIsDesktop(pidl))
 208     {
 209         initIcon->SetNormalIcon(swShell32Name, -IDI_SHELL_DESKTOP);
 210         return initIcon->QueryInterface(iid, ppvOut);
 211     }
 212
 213     riid = _ILGetGUIDPointer(pidl);
 214     if (!riid)
 215         return E_FAIL;
 216
 217     /* my computer and other shell extensions */
 218     static const WCHAR fmt[] = { 'C', 'L', 'S', 'I', 'D', '\\',
 219                                  '{', '%', '0', '8', 'l', 'x', '-', '%', '0', '4', 'x', '-', '%', '0', '4', 'x', '-',
 220                                  '%', '0', '2', 'x', '%', '0', '2', 'x', '-', '%', '0', '2', 'x', '%', '0', '2', 'x',
 221                                  '%', '0', '2', 'x', '%', '0', '2', 'x', '%', '0', '2', 'x', '%', '0', '2', 'x', '}', 0
 222                                };
 223     WCHAR xriid[50];
 224
 225     swprintf(xriid, fmt,
 226              riid->Data1, riid->Data2, riid->Data3,
 227              riid->Data4[0], riid->Data4[1], riid->Data4[2], riid->Data4[3],
 228              riid->Data4[4], riid->Data4[5], riid->Data4[6], riid->Data4[7]);
 229
 230     const WCHAR* iconname = NULL;
 231     if (_ILIsBitBucket(pidl))
 232     {
 233         static const WCHAR szFull[] = {'F','u','l','l',0};
 234         static const WCHAR szEmpty[] = {'E','m','p','t','y',0};
 235         CComPtr<IEnumIDList> EnumIDList;
 236         CoInitialize(NULL);
 237
 238         CComPtr<IShellFolder2> psfRecycleBin;
 239         CComPtr<IShellFolder> psfDesktop;
 240         hr = SHGetDesktopFolder(&psfDesktop);
 241
 242         if (SUCCEEDED(hr))
 243             hr = psfDesktop->BindToObject(pidl, NULL, IID_PPV_ARG(IShellFolder2, &psfRecycleBin));
 244         if (SUCCEEDED(hr))
 245             hr = psfRecycleBin->EnumObjects(NULL, SHCONTF_FOLDERS | SHCONTF_NONFOLDERS, &EnumIDList);
 246
 247         ULONG itemcount;
 248         LPITEMIDLIST pidl = NULL;
 249         if (SUCCEEDED(hr) && (hr = EnumIDList->Next(1, &pidl, &itemcount)) == S_OK)
 250         {
 251             CoTaskMemFree(pidl);
 252             iconname = szFull;
 253         } else {
 254             iconname = szEmpty;
 255         }
 256     }
 257
 258     if (HCR_GetIconW(xriid, wTemp, iconname, MAX_PATH, &icon_idx))
 259     {
 260         initIcon->SetNormalIcon(wTemp, icon_idx);
 261     }
 262     else
 263     {
 264         if (IsEqualGUID(*riid, CLSID_MyComputer))
 265             initIcon->SetNormalIcon(swShell32Name, -IDI_SHELL_MY_COMPUTER);
 266         else if (IsEqualGUID(*riid, CLSID_MyDocuments))
 267             initIcon->SetNormalIcon(swShell32Name, -IDI_SHELL_MY_DOCUMENTS);
 268         else if (IsEqualGUID(*riid, CLSID_NetworkPlaces))
 269             initIcon->SetNormalIcon(swShell32Name, -IDI_SHELL_MY_NETWORK_PLACES);
 270         else
 271             initIcon->SetNormalIcon(swShell32Name, -IDI_SHELL_FOLDER);
 272     }
 273
 274     return initIcon->QueryInterface(iid, ppvOut);
 275 }
 276
 277 HRESULT CFSExtractIcon_CreateInstance(IShellFolder * psf, LPCITEMIDLIST pidl, REFIID iid, LPVOID * ppvOut)
 278 {
 279     CComPtr<IDefaultExtractIconInit>    initIcon;
 280     HRESULT hr;
 281     int icon_idx;
 282     UINT flags;
 283     CHAR sTemp[MAX_PATH];
 284     WCHAR wTemp[MAX_PATH];
 285
 286     hr = SHCreateDefaultExtractIcon(IID_PPV_ARG(IDefaultExtractIconInit,&initIcon));
 287     if (FAILED(hr))
 288         return hr;
 289
 290     if (_ILIsFolder (pidl))
 291     {
 292         if (SUCCEEDED(getIconLocationForFolder(psf,
 293                           pidl, 0, wTemp, MAX_PATH,
 294                           &icon_idx,
 295                           &flags)))
 296         {
 297             initIcon->SetNormalIcon(wTemp, icon_idx);
 298             // FIXME: if/when getIconLocationForFolder does something for
 299             //        GIL_FORSHORTCUT, code below should be uncommented. and
 300             //        the following line removed.
 301             initIcon->SetShortcutIcon(wTemp, icon_idx);
 302         }
 303         if (SUCCEEDED(getIconLocationForFolder(psf,
 304                           pidl, GIL_DEFAULTICON, wTemp, MAX_PATH,
 305                           &icon_idx,
 306                           &flags)))
 307         {
 308             initIcon->SetDefaultIcon(wTemp, icon_idx);
 309         }
 310         // if (SUCCEEDED(getIconLocationForFolder(psf,
 311         //                   pidl, GIL_FORSHORTCUT, wTemp, MAX_PATH,
 312         //                   &icon_idx,
 313         //                   &flags)))
 314         // {
 315         //     initIcon->SetShortcutIcon(wTemp, icon_idx);
 316         // }
 317         if (SUCCEEDED(getIconLocationForFolder(psf,
 318                           pidl, GIL_OPENICON, wTemp, MAX_PATH,
 319                           &icon_idx,
 320                           &flags)))
 321         {
 322             initIcon->SetOpenIcon(wTemp, icon_idx);
 323         }
 324     }
 325     else
 326     {
 327         BOOL found = FALSE;
 328
 329         if (_ILGetExtension(pidl, sTemp, MAX_PATH))
 330         {
 331             if (HCR_MapTypeToValueA(sTemp, sTemp, MAX_PATH, TRUE)
 332                     && HCR_GetIconA(sTemp, sTemp, NULL, MAX_PATH, &icon_idx))
 333             {
 334                 if (!lstrcmpA("%1", sTemp)) /* icon is in the file */
 335                 {
 336                     ILGetDisplayNameExW(psf, pidl, wTemp, 0);
 337                     icon_idx = 0;
 338                 }
 339                 else
 340                 {
 341                     MultiByteToWideChar(CP_ACP, 0, sTemp, -1, wTemp, MAX_PATH);
 342                 }
 343
 344                 found = TRUE;
 345             }
 346             else if (!lstrcmpiA(sTemp, "lnkfile"))
 347             {
 348                 /* extract icon from shell shortcut */
 349                 CComPtr<IShellLinkW>        psl;
 350
 351                 HRESULT hr = psf->GetUIObjectOf(NULL, 1, &pidl, IID_NULL_PPV_ARG(IShellLinkW, &psl));
 352
 353                 if (SUCCEEDED(hr))
 354                 {
 355                     hr = psl->GetIconLocation(wTemp, MAX_PATH, &icon_idx);
 356
 357                     if (SUCCEEDED(hr) && *sTemp)
 358                         found = TRUE;
 359
 360                 }
 361             }
 362         }
 363
 364         if (!found)
 365             /* default icon */
 366             initIcon->SetNormalIcon(swShell32Name, 0);
 367         else
 368             initIcon->SetNormalIcon(wTemp, icon_idx);
 369     }
 370
 371     return initIcon->QueryInterface(iid, ppvOut);
 372 }