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 GenericExtractIcon_CreateInstance(IShellFolder * psf, LPCITEMIDLIST pidl, REFIID iid, LPVOID * ppvOut)
 196 {
 197     CComPtr<IDefaultExtractIconInit>    initIcon;
 198     GUID const * riid;
 199     int icon_idx;
 200     UINT flags;
 201     CHAR sTemp[MAX_PATH];
 202     WCHAR wTemp[MAX_PATH];
 203     LPCITEMIDLIST pSimplePidl = pidl;
 204     HRESULT hr;
 205
 206     hr = SHCreateDefaultExtractIcon(IID_PPV_ARG(IDefaultExtractIconInit,&initIcon));
 207     if (FAILED(hr))
 208         return hr;
 209
 210     if (_ILIsDesktop(pSimplePidl))
 211     {
 212         initIcon->SetNormalIcon(swShell32Name, -IDI_SHELL_DESKTOP);
 213     }
 214     else if ((riid = _ILGetGUIDPointer(pSimplePidl)))
 215     {
 216         /* my computer and other shell extensions */
 217         static const WCHAR fmt[] = { 'C', 'L', 'S', 'I', 'D', '\\',
 218                                      '{', '%', '0', '8', 'l', 'x', '-', '%', '0', '4', 'x', '-', '%', '0', '4', 'x', '-',
 219                                      '%', '0', '2', 'x', '%', '0', '2', 'x', '-', '%', '0', '2', 'x', '%', '0', '2', 'x',
 220                                      '%', '0', '2', 'x', '%', '0', '2', 'x', '%', '0', '2', 'x', '%', '0', '2', 'x', '}', 0
 221                                    };
 222         WCHAR xriid[50];
 223
 224         swprintf(xriid, fmt,
 225                  riid->Data1, riid->Data2, riid->Data3,
 226                  riid->Data4[0], riid->Data4[1], riid->Data4[2], riid->Data4[3],
 227                  riid->Data4[4], riid->Data4[5], riid->Data4[6], riid->Data4[7]);
 228
 229         const WCHAR* iconname = NULL;
 230         if (_ILIsBitBucket(pSimplePidl))
 231         {
 232             static const WCHAR szFull[] = {'F','u','l','l',0};
 233             static const WCHAR szEmpty[] = {'E','m','p','t','y',0};
 234             CComPtr<IEnumIDList> EnumIDList;
 235             CoInitialize(NULL);
 236
 237             CComPtr<IShellFolder2> psfRecycleBin;
 238             CComPtr<IShellFolder> psfDesktop;
 239             hr = SHGetDesktopFolder(&psfDesktop);
 240
 241             if (SUCCEEDED(hr))
 242                 hr = psfDesktop->BindToObject(pSimplePidl, NULL, IID_PPV_ARG(IShellFolder2, &psfRecycleBin));
 243             if (SUCCEEDED(hr))
 244                 hr = psfRecycleBin->EnumObjects(NULL, SHCONTF_FOLDERS | SHCONTF_NONFOLDERS, &EnumIDList);
 245
 246             ULONG itemcount;
 247             LPITEMIDLIST pidl = NULL;
 248             if (SUCCEEDED(hr) && (hr = EnumIDList->Next(1, &pidl, &itemcount)) == S_OK)
 249             {
 250                 CoTaskMemFree(pidl);
 251                 iconname = szFull;
 252             } else {
 253                 iconname = szEmpty;
 254             }
 255         }
 256
 257         if (HCR_GetIconW(xriid, wTemp, iconname, MAX_PATH, &icon_idx))
 258         {
 259             initIcon->SetNormalIcon(wTemp, icon_idx);
 260         }
 261         else
 262         {
 263             if (IsEqualGUID(*riid, CLSID_MyComputer))
 264                 initIcon->SetNormalIcon(swShell32Name, -IDI_SHELL_MY_COMPUTER);
 265             else if (IsEqualGUID(*riid, CLSID_MyDocuments))
 266                 initIcon->SetNormalIcon(swShell32Name, -IDI_SHELL_MY_DOCUMENTS);
 267             else if (IsEqualGUID(*riid, CLSID_NetworkPlaces))
 268                 initIcon->SetNormalIcon(swShell32Name, -IDI_SHELL_MY_NETWORK_PLACES);
 269             else
 270                 initIcon->SetNormalIcon(swShell32Name, -IDI_SHELL_FOLDER);
 271         }
 272     }
 273
 274     else if (_ILIsDrive (pSimplePidl))
 275     {
 276         static const WCHAR drive[] = { 'D', 'r', 'i', 'v', 'e', 0 };
 277         int icon_idx = -1;
 278
 279         if (_ILGetDrive(pSimplePidl, sTemp, MAX_PATH))
 280         {
 281             switch(GetDriveTypeA(sTemp))
 282             {
 283                 case DRIVE_REMOVABLE:
 284                     icon_idx = IDI_SHELL_3_14_FLOPPY;
 285                     break;
 286                 case DRIVE_CDROM:
 287                     icon_idx = IDI_SHELL_CDROM;
 288                     break;
 289                 case DRIVE_REMOTE:
 290                     icon_idx = IDI_SHELL_NETDRIVE;
 291                     break;
 292                 case DRIVE_RAMDISK:
 293                     icon_idx = IDI_SHELL_RAMDISK;
 294                     break;
 295                 case DRIVE_NO_ROOT_DIR:
 296                     icon_idx = IDI_SHELL_CDROM;
 297                     break;
 298             }
 299         }
 300
 301         if (icon_idx != -1)
 302         {
 303             initIcon->SetNormalIcon(swShell32Name, -icon_idx);
 304         }
 305         else
 306         {
 307             if (HCR_GetIconW(drive, wTemp, NULL, MAX_PATH, &icon_idx))
 308                 initIcon->SetNormalIcon(wTemp, icon_idx);
 309             else
 310                 initIcon->SetNormalIcon(swShell32Name, -IDI_SHELL_DRIVE);
 311         }
 312     }
 313
 314     else if (_ILIsFolder (pSimplePidl))
 315     {
 316         if (SUCCEEDED(getIconLocationForFolder(psf,
 317                           pidl, 0, wTemp, MAX_PATH,
 318                           &icon_idx,
 319                           &flags)))
 320         {
 321             initIcon->SetNormalIcon(wTemp, icon_idx);
 322             // FIXME: if/when getIconLocationForFolder does something for
 323             //        GIL_FORSHORTCUT, code below should be uncommented. and
 324             //        the following line removed.
 325             initIcon->SetShortcutIcon(wTemp, icon_idx);
 326         }
 327         if (SUCCEEDED(getIconLocationForFolder(psf,
 328                           pidl, GIL_DEFAULTICON, wTemp, MAX_PATH,
 329                           &icon_idx,
 330                           &flags)))
 331         {
 332             initIcon->SetDefaultIcon(wTemp, icon_idx);
 333         }
 334         // if (SUCCEEDED(getIconLocationForFolder(psf,
 335         //                   pidl, GIL_FORSHORTCUT, wTemp, MAX_PATH,
 336         //                   &icon_idx,
 337         //                   &flags)))
 338         // {
 339         //     initIcon->SetShortcutIcon(wTemp, icon_idx);
 340         // }
 341         if (SUCCEEDED(getIconLocationForFolder(psf,
 342                           pidl, GIL_OPENICON, wTemp, MAX_PATH,
 343                           &icon_idx,
 344                           &flags)))
 345         {
 346             initIcon->SetOpenIcon(wTemp, icon_idx);
 347         }
 348     }
 349     else
 350     {
 351         BOOL found = FALSE;
 352
 353         if (_ILGetExtension(pSimplePidl, sTemp, MAX_PATH))
 354         {
 355             if (HCR_MapTypeToValueA(sTemp, sTemp, MAX_PATH, TRUE)
 356                     && HCR_GetIconA(sTemp, sTemp, NULL, MAX_PATH, &icon_idx))
 357             {
 358                 if (!lstrcmpA("%1", sTemp)) /* icon is in the file */
 359                 {
 360                     ILGetDisplayNameExW(psf, pidl, wTemp, 0);
 361                     icon_idx = 0;
 362                 }
 363                 else
 364                 {
 365                     MultiByteToWideChar(CP_ACP, 0, sTemp, -1, wTemp, MAX_PATH);
 366                 }
 367
 368                 found = TRUE;
 369             }
 370             else if (!lstrcmpiA(sTemp, "lnkfile"))
 371             {
 372                 /* extract icon from shell shortcut */
 373                 CComPtr<IShellLinkW>        psl;
 374
 375                 HRESULT hr = psf->GetUIObjectOf(NULL, 1, &pidl, IID_NULL_PPV_ARG(IShellLinkW, &psl));
 376
 377                 if (SUCCEEDED(hr))
 378                 {
 379                     hr = psl->GetIconLocation(wTemp, MAX_PATH, &icon_idx);
 380
 381                     if (SUCCEEDED(hr) && *sTemp)
 382                         found = TRUE;
 383
 384                 }
 385             }
 386         }
 387
 388         if (!found)
 389             /* default icon */
 390             initIcon->SetNormalIcon(swShell32Name, 0);
 391         else
 392             initIcon->SetNormalIcon(wTemp, icon_idx);
 393     }
 394
 395     return initIcon->QueryInterface(iid, ppvOut);
 396 }