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