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