dlls/shell32/folders.c

   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 "config.h"
  21 #include "wine/port.h"
  22
  23 #include <stdlib.h>
  24 #include <stdarg.h>
  25 #include <stdio.h>
  26 #include <string.h>
  27
  28 #define COBJMACROS
  29
  30 #include "windef.h"
  31 #include "winbase.h"
  32 #include "winerror.h"
  33 #include "objbase.h"
  34 #include "undocshell.h"
  35 #include "shlguid.h"
  36
  37 #include "wine/debug.h"
  38
  39 #include "pidl.h"
  40 #include "shell32_main.h"
  41 #include "shfldr.h"
  42 #include "shresdef.h"
  43
  44 WINE_DEFAULT_DEBUG_CHANNEL(shell);
  45
  46 /***********************************************************************
  47 *   IExtractIconW implementation
  48 */
  49 typedef struct
  50 {
  51         IExtractIconW      IExtractIconW_iface;
  52         IExtractIconA      IExtractIconA_iface;
  53         IPersistFile       IPersistFile_iface;
  54         LONG               ref;
  55         LPITEMIDLIST       pidl;
  56 } IExtractIconWImpl;
  57
  58 static inline IExtractIconWImpl *impl_from_IExtractIconW(IExtractIconW *iface)
  59 {
  60     return CONTAINING_RECORD(iface, IExtractIconWImpl, IExtractIconW_iface);
  61 }
  62
  63 static inline IExtractIconWImpl *impl_from_IExtractIconA(IExtractIconA *iface)
  64 {
  65     return CONTAINING_RECORD(iface, IExtractIconWImpl, IExtractIconA_iface);
  66 }
  67
  68 static inline IExtractIconWImpl *impl_from_IPersistFile(IPersistFile *iface)
  69 {
  70     return CONTAINING_RECORD(iface, IExtractIconWImpl, IPersistFile_iface);
  71 }
  72
  73
  74 /**************************************************************************
  75  *  IExtractIconW::QueryInterface
  76  */
  77 static HRESULT WINAPI IExtractIconW_fnQueryInterface(IExtractIconW *iface, REFIID riid,
  78         void **ppv)
  79 {
  80     IExtractIconWImpl *This = impl_from_IExtractIconW(iface);
  81
  82     TRACE("(%p)->(\n\tIID:\t%s,%p)\n", This, debugstr_guid(riid), ppv);
  83
  84     *ppv = NULL;
  85     if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IExtractIconW))
  86         *ppv = iface;
  87     else if (IsEqualIID(riid, &IID_IPersistFile))
  88         *ppv = &This->IPersistFile_iface;
  89     else if (IsEqualIID(riid, &IID_IExtractIconA))
  90         *ppv = &This->IExtractIconA_iface;
  91
  92     if(*ppv)
  93     {
  94         IUnknown_AddRef((IUnknown*)*ppv);
  95         TRACE("-- Interface: (%p)->(%p)\n", ppv, *ppv);
  96         return S_OK;
  97     }
  98     TRACE("-- Interface: E_NOINTERFACE\n");
  99     return E_NOINTERFACE;
 100 }
 101
 102 /**************************************************************************
 103 *  IExtractIconW::AddRef
 104 */
 105 static ULONG WINAPI IExtractIconW_fnAddRef(IExtractIconW * iface)
 106 {
 107         IExtractIconWImpl *This = impl_from_IExtractIconW(iface);
 108         ULONG refCount = InterlockedIncrement(&This->ref);
 109
 110         TRACE("(%p)->(count=%u)\n", This, refCount - 1);
 111
 112         return refCount;
 113 }
 114 /**************************************************************************
 115 *  IExtractIconW::Release
 116 */
 117 static ULONG WINAPI IExtractIconW_fnRelease(IExtractIconW * iface)
 118 {
 119         IExtractIconWImpl *This = impl_from_IExtractIconW(iface);
 120         ULONG refCount = InterlockedDecrement(&This->ref);
 121
 122         TRACE("(%p)->(count=%u)\n", This, refCount + 1);
 123
 124         if (!refCount)
 125         {
 126           TRACE(" destroying IExtractIcon(%p)\n",This);
 127           SHFree(This->pidl);
 128           HeapFree(GetProcessHeap(),0,This);
 129           return 0;
 130         }
 131         return refCount;
 132 }
 133
 134 static HRESULT getIconLocationForFolder(IExtractIconWImpl *This, UINT uFlags, LPWSTR szIconFile,
 135         UINT cchMax, int *piIndex, UINT *pwFlags)
 136 {
 137     int icon_idx;
 138     WCHAR wszPath[MAX_PATH];
 139     WCHAR wszCLSIDValue[CHARS_IN_GUID];
 140     static const WCHAR shellClassInfo[] = { '.','S','h','e','l','l','C','l','a','s','s','I','n','f','o',0 };
 141     static const WCHAR iconFile[] = { 'I','c','o','n','F','i','l','e',0 };
 142     static const WCHAR clsid[] = { 'C','L','S','I','D',0 };
 143     static const WCHAR clsid2[] = { 'C','L','S','I','D','2',0 };
 144     static const WCHAR iconIndex[] = { 'I','c','o','n','I','n','d','e','x',0 };
 145
 146     if (SHELL32_GetCustomFolderAttribute(This->pidl, shellClassInfo, iconFile,
 147         wszPath, MAX_PATH))
 148     {
 149         WCHAR wszIconIndex[10];
 150         SHELL32_GetCustomFolderAttribute(This->pidl, shellClassInfo, iconIndex,
 151             wszIconIndex, 10);
 152         *piIndex = atoiW(wszIconIndex);
 153     }
 154     else if (SHELL32_GetCustomFolderAttribute(This->pidl, shellClassInfo, clsid,
 155         wszCLSIDValue, CHARS_IN_GUID) &&
 156         HCR_GetDefaultIconW(wszCLSIDValue, szIconFile, cchMax, &icon_idx))
 157     {
 158        *piIndex = icon_idx;
 159     }
 160     else if (SHELL32_GetCustomFolderAttribute(This->pidl, shellClassInfo, clsid2,
 161         wszCLSIDValue, CHARS_IN_GUID) &&
 162         HCR_GetDefaultIconW(wszCLSIDValue, szIconFile, cchMax, &icon_idx))
 163     {
 164        *piIndex = icon_idx;
 165     }
 166     else
 167     {
 168         static const WCHAR folder[] = { 'F','o','l','d','e','r',0 };
 169
 170         if (!HCR_GetDefaultIconW(folder, szIconFile, cchMax, &icon_idx))
 171         {
 172             lstrcpynW(szIconFile, swShell32Name, cchMax);
 173             icon_idx = -IDI_SHELL_FOLDER;
 174         }
 175
 176         if (uFlags & GIL_OPENICON)
 177             *piIndex = icon_idx<0? icon_idx-1: icon_idx+1;
 178         else
 179             *piIndex = icon_idx;
 180     }
 181
 182     return S_OK;
 183 }
 184
 185 WCHAR swShell32Name[MAX_PATH];
 186
 187 /**************************************************************************
 188 *  IExtractIconW::GetIconLocation
 189 *
 190 * mapping filetype to icon
 191 */
 192 static HRESULT WINAPI IExtractIconW_fnGetIconLocation(IExtractIconW * iface, UINT uFlags,
 193         LPWSTR szIconFile, UINT cchMax, int * piIndex, UINT * pwFlags)
 194 {
 195         IExtractIconWImpl *This = impl_from_IExtractIconW(iface);
 196         char    sTemp[MAX_PATH];
 197         int             icon_idx;
 198         GUID const * riid;
 199         LPITEMIDLIST    pSimplePidl = ILFindLastID(This->pidl);
 200
 201         TRACE("(%p) (flags=%u, %s, %u, %p, %p)\n", This, uFlags, debugstr_w(szIconFile),
 202                 cchMax, piIndex, pwFlags);
 203
 204         if (pwFlags)
 205           *pwFlags = 0;
 206
 207         if (_ILIsDesktop(pSimplePidl))
 208         {
 209           lstrcpynW(szIconFile, swShell32Name, cchMax);
 210           *piIndex = -IDI_SHELL_DESKTOP;
 211         }
 212
 213         /* my computer and other shell extensions */
 214         else if ((riid = _ILGetGUIDPointer(pSimplePidl)))
 215         {
 216           static const WCHAR fmt[] = { 'C','L','S','I','D','\\',
 217        '{','%','0','8','l','x','-','%','0','4','x','-','%','0','4','x','-',
 218        '%','0','2','x','%','0','2','x','-','%','0','2','x', '%','0','2','x',
 219        '%','0','2','x','%','0','2','x','%','0','2','x','%','0','2','x','}',0 };
 220           WCHAR xriid[50];
 221
 222           sprintfW(xriid, fmt,
 223                   riid->Data1, riid->Data2, riid->Data3,
 224                   riid->Data4[0], riid->Data4[1], riid->Data4[2], riid->Data4[3],
 225                   riid->Data4[4], riid->Data4[5], riid->Data4[6], riid->Data4[7]);
 226
 227           if (HCR_GetDefaultIconW(xriid, szIconFile, cchMax, &icon_idx))
 228           {
 229             *piIndex = icon_idx;
 230           }
 231           else
 232           {
 233             lstrcpynW(szIconFile, swShell32Name, cchMax);
 234             if(IsEqualGUID(riid, &CLSID_MyComputer))
 235                 *piIndex = -IDI_SHELL_MY_COMPUTER;
 236             else if(IsEqualGUID(riid, &CLSID_MyDocuments))
 237                 *piIndex = -IDI_SHELL_MY_DOCUMENTS;
 238             else if(IsEqualGUID(riid, &CLSID_NetworkPlaces))
 239                 *piIndex = -IDI_SHELL_MY_NETWORK_PLACES;
 240             else if(IsEqualGUID(riid, &CLSID_UnixFolder) ||
 241                     IsEqualGUID(riid, &CLSID_UnixDosFolder))
 242                 *piIndex = -IDI_SHELL_DRIVE;
 243             else
 244                 *piIndex = -IDI_SHELL_FOLDER;
 245           }
 246         }
 247
 248         else if (_ILIsDrive (pSimplePidl))
 249         {
 250           static const WCHAR drive[] = { 'D','r','i','v','e',0 };
 251
 252           int icon_idx = -1;
 253
 254           if (_ILGetDrive(pSimplePidl, sTemp, MAX_PATH))
 255           {
 256                 switch(GetDriveTypeA(sTemp))
 257                 {
 258                   case DRIVE_REMOVABLE:   icon_idx = IDI_SHELL_FLOPPY;        break;
 259                   case DRIVE_CDROM:       icon_idx = IDI_SHELL_CDROM;         break;
 260                   case DRIVE_REMOTE:      icon_idx = IDI_SHELL_NETDRIVE;      break;
 261                   case DRIVE_RAMDISK:     icon_idx = IDI_SHELL_RAMDISK;       break;
 262                 }
 263           }
 264
 265           if (icon_idx != -1)
 266           {
 267                 lstrcpynW(szIconFile, swShell32Name, cchMax);
 268                 *piIndex = -icon_idx;
 269           }
 270           else
 271           {
 272                 if (HCR_GetDefaultIconW(drive, szIconFile, cchMax, &icon_idx))
 273                 {
 274                   *piIndex = icon_idx;
 275                 }
 276                 else
 277                 {
 278                   lstrcpynW(szIconFile, swShell32Name, cchMax);
 279                   *piIndex = -IDI_SHELL_DRIVE;
 280                 }
 281           }
 282         }
 283         else if (_ILIsFolder (pSimplePidl))
 284         {
 285             getIconLocationForFolder(This, uFlags, szIconFile, cchMax, piIndex, pwFlags);
 286         }
 287         else
 288         {
 289           BOOL found = FALSE;
 290
 291           if (_ILIsCPanelStruct(pSimplePidl))
 292           {
 293             if (SUCCEEDED(CPanel_GetIconLocationW(pSimplePidl, szIconFile, cchMax, piIndex)))
 294                 found = TRUE;
 295           }
 296           else if (_ILGetExtension(pSimplePidl, sTemp, MAX_PATH))
 297           {
 298             if (HCR_MapTypeToValueA(sTemp, sTemp, MAX_PATH, TRUE)
 299                 && HCR_GetDefaultIconA(sTemp, sTemp, MAX_PATH, &icon_idx))
 300             {
 301               if (!lstrcmpA("%1", sTemp))               /* icon is in the file */
 302               {
 303                 SHGetPathFromIDListW(This->pidl, szIconFile);
 304                 *piIndex = 0;
 305               }
 306               else
 307               {
 308                 MultiByteToWideChar(CP_ACP, 0, sTemp, -1, szIconFile, cchMax);
 309                 *piIndex = icon_idx;
 310               }
 311
 312               found = TRUE;
 313             }
 314             else if (!lstrcmpiA(sTemp, "lnkfile"))
 315             {
 316               /* extract icon from shell shortcut */
 317               IShellFolder* dsf;
 318               IShellLinkW* psl;
 319
 320               if (SUCCEEDED(SHGetDesktopFolder(&dsf)))
 321               {
 322                 HRESULT hr = IShellFolder_GetUIObjectOf(dsf, NULL, 1, (LPCITEMIDLIST*)&This->pidl, &IID_IShellLinkW, NULL, (LPVOID*)&psl);
 323
 324                 if (SUCCEEDED(hr))
 325                 {
 326                   hr = IShellLinkW_GetIconLocation(psl, szIconFile, MAX_PATH, piIndex);
 327
 328                   if (SUCCEEDED(hr) && *szIconFile)
 329                     found = TRUE;
 330
 331                   IShellLinkW_Release(psl);
 332                 }
 333
 334                 IShellFolder_Release(dsf);
 335               }
 336             }
 337           }
 338
 339           if (!found)                                   /* default icon */
 340           {
 341             lstrcpynW(szIconFile, swShell32Name, cchMax);
 342             *piIndex = 0;
 343           }
 344         }
 345
 346         TRACE("-- %s %x\n", debugstr_w(szIconFile), *piIndex);
 347         return S_OK;
 348 }
 349
 350 /**************************************************************************
 351 *  IExtractIconW::Extract
 352 */
 353 static HRESULT WINAPI IExtractIconW_fnExtract(IExtractIconW * iface, LPCWSTR pszFile,
 354         UINT nIconIndex, HICON *phiconLarge, HICON *phiconSmall, UINT nIconSize)
 355 {
 356         IExtractIconWImpl *This = impl_from_IExtractIconW(iface);
 357         int index;
 358         HIMAGELIST big_icons, small_icons;
 359
 360         FIXME("(%p) (file=%s index=%d %p %p size=%08x) semi-stub\n", This, debugstr_w(pszFile),
 361                 (signed)nIconIndex, phiconLarge, phiconSmall, nIconSize);
 362
 363         index = SIC_GetIconIndex(pszFile, nIconIndex, 0);
 364         Shell_GetImageLists( &big_icons, &small_icons );
 365         if (phiconLarge)
 366           *phiconLarge = ImageList_GetIcon(big_icons, index, ILD_TRANSPARENT);
 367
 368         if (phiconSmall)
 369           *phiconSmall = ImageList_GetIcon(small_icons, index, ILD_TRANSPARENT);
 370
 371         return S_OK;
 372 }
 373
 374 static const IExtractIconWVtbl eivt =
 375 {
 376         IExtractIconW_fnQueryInterface,
 377         IExtractIconW_fnAddRef,
 378         IExtractIconW_fnRelease,
 379         IExtractIconW_fnGetIconLocation,
 380         IExtractIconW_fnExtract
 381 };
 382
 383 /**************************************************************************
 384  *  IExtractIconA::QueryInterface
 385  */
 386 static HRESULT WINAPI IExtractIconA_fnQueryInterface(IExtractIconA * iface, REFIID riid,
 387         void **ppv)
 388 {
 389     IExtractIconWImpl *This = impl_from_IExtractIconA(iface);
 390
 391     return IExtractIconW_QueryInterface(&This->IExtractIconW_iface, riid, ppv);
 392 }
 393
 394 /**************************************************************************
 395 *  IExtractIconA::AddRef
 396 */
 397 static ULONG WINAPI IExtractIconA_fnAddRef(IExtractIconA * iface)
 398 {
 399     IExtractIconWImpl *This = impl_from_IExtractIconA(iface);
 400
 401     return IExtractIconW_AddRef(&This->IExtractIconW_iface);
 402 }
 403 /**************************************************************************
 404 *  IExtractIconA::Release
 405 */
 406 static ULONG WINAPI IExtractIconA_fnRelease(IExtractIconA * iface)
 407 {
 408     IExtractIconWImpl *This = impl_from_IExtractIconA(iface);
 409
 410     return IExtractIconW_Release(&This->IExtractIconW_iface);
 411 }
 412 /**************************************************************************
 413 *  IExtractIconA::GetIconLocation
 414 *
 415 * mapping filetype to icon
 416 */
 417 static HRESULT WINAPI IExtractIconA_fnGetIconLocation(IExtractIconA * iface, UINT uFlags,
 418         LPSTR szIconFile, UINT cchMax, int * piIndex, UINT * pwFlags)
 419 {
 420         IExtractIconWImpl *This = impl_from_IExtractIconA(iface);
 421         HRESULT ret;
 422         LPWSTR lpwstrFile = HeapAlloc(GetProcessHeap(), 0, cchMax * sizeof(WCHAR));
 423
 424         TRACE("(%p) (flags=%u %p %u %p %p)\n", This, uFlags, szIconFile, cchMax, piIndex, pwFlags);
 425
 426         ret = IExtractIconW_GetIconLocation(&This->IExtractIconW_iface, uFlags, lpwstrFile, cchMax,
 427                 piIndex, pwFlags);
 428         WideCharToMultiByte(CP_ACP, 0, lpwstrFile, -1, szIconFile, cchMax, NULL, NULL);
 429         HeapFree(GetProcessHeap(), 0, lpwstrFile);
 430
 431         TRACE("-- %s %x\n", szIconFile, *piIndex);
 432         return ret;
 433 }
 434 /**************************************************************************
 435 *  IExtractIconA::Extract
 436 */
 437 static HRESULT WINAPI IExtractIconA_fnExtract(IExtractIconA * iface, LPCSTR pszFile,
 438         UINT nIconIndex, HICON *phiconLarge, HICON *phiconSmall, UINT nIconSize)
 439 {
 440         IExtractIconWImpl *This = impl_from_IExtractIconA(iface);
 441         HRESULT ret;
 442         INT len = MultiByteToWideChar(CP_ACP, 0, pszFile, -1, NULL, 0);
 443         LPWSTR lpwstrFile = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
 444
 445         TRACE("(%p) (file=%p index=%u %p %p size=%u)\n", This, pszFile, nIconIndex, phiconLarge, phiconSmall, nIconSize);
 446
 447         MultiByteToWideChar(CP_ACP, 0, pszFile, -1, lpwstrFile, len);
 448         ret = IExtractIconW_Extract(&This->IExtractIconW_iface, lpwstrFile, nIconIndex, phiconLarge,
 449                 phiconSmall, nIconSize);
 450         HeapFree(GetProcessHeap(), 0, lpwstrFile);
 451         return ret;
 452 }
 453
 454 static const IExtractIconAVtbl eiavt =
 455 {
 456         IExtractIconA_fnQueryInterface,
 457         IExtractIconA_fnAddRef,
 458         IExtractIconA_fnRelease,
 459         IExtractIconA_fnGetIconLocation,
 460         IExtractIconA_fnExtract
 461 };
 462
 463 /************************************************************************
 464  * IEIPersistFile::QueryInterface
 465  */
 466 static HRESULT WINAPI IEIPersistFile_fnQueryInterface(IPersistFile *iface, REFIID iid,
 467         void **ppv)
 468 {
 469     IExtractIconWImpl *This = impl_from_IPersistFile(iface);
 470
 471     return IExtractIconW_QueryInterface(&This->IExtractIconW_iface, iid, ppv);
 472 }
 473
 474 /************************************************************************
 475  * IEIPersistFile::AddRef
 476  */
 477 static ULONG WINAPI IEIPersistFile_fnAddRef(IPersistFile *iface)
 478 {
 479     IExtractIconWImpl *This = impl_from_IPersistFile(iface);
 480
 481     return IExtractIconW_AddRef(&This->IExtractIconW_iface);
 482 }
 483
 484 /************************************************************************
 485  * IEIPersistFile::Release
 486  */
 487 static ULONG WINAPI IEIPersistFile_fnRelease(IPersistFile *iface)
 488 {
 489     IExtractIconWImpl *This = impl_from_IPersistFile(iface);
 490
 491     return IExtractIconW_Release(&This->IExtractIconW_iface);
 492 }
 493
 494 /************************************************************************
 495  * IEIPersistFile::GetClassID
 496  */
 497 static HRESULT WINAPI IEIPersistFile_fnGetClassID(IPersistFile *iface, LPCLSID lpClassId)
 498 {
 499         CLSID StdFolderID = { 0x00000000, 0x0000, 0x0000, {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00} };
 500
 501         if (lpClassId==NULL)
 502           return E_POINTER;
 503
 504         *lpClassId = StdFolderID;
 505
 506         return S_OK;
 507 }
 508
 509 /************************************************************************
 510  * IEIPersistFile_Load
 511  */
 512 static HRESULT WINAPI IEIPersistFile_fnLoad(IPersistFile* iface, LPCOLESTR pszFileName,
 513         DWORD dwMode)
 514 {
 515     IExtractIconWImpl *This = impl_from_IPersistFile(iface);
 516
 517     FIXME("%p\n", This);
 518     return E_NOTIMPL;
 519 }
 520
 521 static const IPersistFileVtbl pfvt =
 522 {
 523         IEIPersistFile_fnQueryInterface,
 524         IEIPersistFile_fnAddRef,
 525         IEIPersistFile_fnRelease,
 526         IEIPersistFile_fnGetClassID,
 527         (void *) 0xdeadbeef /* IEIPersistFile_fnIsDirty */,
 528         IEIPersistFile_fnLoad,
 529         (void *) 0xdeadbeef /* IEIPersistFile_fnSave */,
 530         (void *) 0xdeadbeef /* IEIPersistFile_fnSaveCompleted */,
 531         (void *) 0xdeadbeef /* IEIPersistFile_fnGetCurFile */
 532 };
 533
 534 static IExtractIconWImpl *extracticon_create(LPCITEMIDLIST pidl)
 535 {
 536     IExtractIconWImpl *ei;
 537
 538     TRACE("%p\n", pidl);
 539
 540     ei = HeapAlloc(GetProcessHeap(), 0, sizeof(*ei));
 541     ei->ref=1;
 542     ei->IExtractIconW_iface.lpVtbl = &eivt;
 543     ei->IExtractIconA_iface.lpVtbl = &eiavt;
 544     ei->IPersistFile_iface.lpVtbl = &pfvt;
 545     ei->pidl=ILClone(pidl);
 546
 547     pdump(pidl);
 548
 549     TRACE("(%p)\n", ei);
 550     return ei;
 551 }
 552
 553 IExtractIconW *IExtractIconW_Constructor(LPCITEMIDLIST pidl)
 554 {
 555     IExtractIconWImpl *ei = extracticon_create(pidl);
 556
 557     return &ei->IExtractIconW_iface;
 558 }
 559
 560 IExtractIconA *IExtractIconA_Constructor(LPCITEMIDLIST pidl)
 561 {
 562     IExtractIconWImpl *ei = extracticon_create(pidl);
 563
 564     return &ei->IExtractIconA_iface;
 565 }