dlls/shell32/shfldr_mycomp.c

   1 /*
   2  *    Virtual Workplace folder
   3  *
   4  *    Copyright 1997            Marcus Meissner
   5  *    Copyright 1998, 1999, 2002    Juergen Schmied
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21
  22 #include <stdlib.h>
  23 #include <string.h>
  24 #include <stdarg.h>
  25 #include <stdio.h>
  26
  27 #define COBJMACROS
  28 #define NONAMELESSUNION
  29
  30 #include "winerror.h"
  31 #include "windef.h"
  32 #include "winbase.h"
  33 #include "winreg.h"
  34
  35 #include "wingdi.h"
  36 #include "pidl.h"
  37 #include "shlguid.h"
  38 #include "shell32_main.h"
  39 #include "shresdef.h"
  40 #include "shlwapi.h"
  41 #include "wine/debug.h"
  42 #include "debughlp.h"
  43 #include "shfldr.h"
  44
  45 WINE_DEFAULT_DEBUG_CHANNEL (shell);
  46
  47 /***********************************************************************
  48 *   IShellFolder implementation
  49 */
  50
  51 typedef struct {
  52     IShellFolder2   IShellFolder2_iface;
  53     IPersistFolder2 IPersistFolder2_iface;
  54     LONG            ref;
  55
  56     /* both paths are parsible from the desktop */
  57     LPITEMIDLIST pidlRoot;    /* absolute pidl */
  58 } IMyComputerFolderImpl;
  59
  60 static const IShellFolder2Vtbl vt_ShellFolder2;
  61 static const IPersistFolder2Vtbl vt_PersistFolder2;
  62
  63 static inline IMyComputerFolderImpl *impl_from_IShellFolder2(IShellFolder2 *iface)
  64 {
  65     return CONTAINING_RECORD(iface, IMyComputerFolderImpl, IShellFolder2_iface);
  66 }
  67
  68 static inline IMyComputerFolderImpl *impl_from_IPersistFolder2(IPersistFolder2 *iface)
  69 {
  70     return CONTAINING_RECORD(iface, IMyComputerFolderImpl, IPersistFolder2_iface);
  71 }
  72
  73
  74 /***********************************************************************
  75 *   IShellFolder [MyComputer] implementation
  76 */
  77
  78 static const shvheader mycomputer_header[] =
  79 {
  80     { &FMTID_Storage, PID_STG_NAME, IDS_SHV_COLUMN1, SHCOLSTATE_TYPE_STR | SHCOLSTATE_ONBYDEFAULT, LVCFMT_RIGHT, 15 },
  81     { &FMTID_Storage, PID_STG_STORAGETYPE, IDS_SHV_COLUMN3, SHCOLSTATE_TYPE_STR | SHCOLSTATE_ONBYDEFAULT, LVCFMT_RIGHT, 10 },
  82     { NULL, 0, IDS_SHV_COLUMN6, SHCOLSTATE_TYPE_STR | SHCOLSTATE_ONBYDEFAULT, LVCFMT_RIGHT, 10 },
  83     { NULL, 0, IDS_SHV_COLUMN7, SHCOLSTATE_TYPE_STR | SHCOLSTATE_ONBYDEFAULT, LVCFMT_RIGHT, 10 },
  84 };
  85
  86 /**************************************************************************
  87 *    ISF_MyComputer_Constructor
  88 */
  89 HRESULT WINAPI ISF_MyComputer_Constructor (IUnknown * pUnkOuter, REFIID riid, LPVOID * ppv)
  90 {
  91     IMyComputerFolderImpl *sf;
  92
  93     TRACE ("unkOut=%p %s\n", pUnkOuter, shdebugstr_guid (riid));
  94
  95     if (!ppv)
  96         return E_POINTER;
  97     if (pUnkOuter)
  98         return CLASS_E_NOAGGREGATION;
  99
 100     sf = LocalAlloc (LMEM_ZEROINIT, sizeof (IMyComputerFolderImpl));
 101     if (!sf)
 102         return E_OUTOFMEMORY;
 103
 104     sf->ref = 0;
 105     sf->IShellFolder2_iface.lpVtbl = &vt_ShellFolder2;
 106     sf->IPersistFolder2_iface.lpVtbl = &vt_PersistFolder2;
 107     sf->pidlRoot = _ILCreateMyComputer ();    /* my qualified pidl */
 108
 109     if (FAILED (IShellFolder2_QueryInterface (&sf->IShellFolder2_iface, riid, ppv)))
 110     {
 111         IShellFolder2_Release (&sf->IShellFolder2_iface);
 112         return E_NOINTERFACE;
 113     }
 114
 115     TRACE ("--(%p)\n", sf);
 116     return S_OK;
 117 }
 118
 119 /**************************************************************************
 120  *    ISF_MyComputer_fnQueryInterface
 121  *
 122  * NOTES supports not IPersist/IPersistFolder
 123  */
 124 static HRESULT WINAPI ISF_MyComputer_fnQueryInterface (IShellFolder2 *iface,
 125                REFIID riid, LPVOID *ppvObj)
 126 {
 127     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 128
 129     TRACE ("(%p)->(%s,%p)\n", This, shdebugstr_guid (riid), ppvObj);
 130
 131     *ppvObj = NULL;
 132
 133     if (IsEqualIID (riid, &IID_IUnknown) ||
 134         IsEqualIID (riid, &IID_IShellFolder) ||
 135         IsEqualIID (riid, &IID_IShellFolder2))
 136     {
 137         *ppvObj = &This->IShellFolder2_iface;
 138     }
 139     else if (IsEqualIID (riid, &IID_IPersist) ||
 140              IsEqualIID (riid, &IID_IPersistFolder) ||
 141              IsEqualIID (riid, &IID_IPersistFolder2))
 142     {
 143         *ppvObj = &This->IPersistFolder2_iface;
 144     }
 145
 146     if (*ppvObj)
 147     {
 148         IUnknown_AddRef ((IUnknown *) (*ppvObj));
 149         TRACE ("-- Interface: (%p)->(%p)\n", ppvObj, *ppvObj);
 150         return S_OK;
 151     }
 152     TRACE ("-- Interface: E_NOINTERFACE\n");
 153     return E_NOINTERFACE;
 154 }
 155
 156 static ULONG WINAPI ISF_MyComputer_fnAddRef (IShellFolder2 * iface)
 157 {
 158     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 159     ULONG refCount = InterlockedIncrement(&This->ref);
 160
 161     TRACE ("(%p)->(count=%lu)\n", This, refCount - 1);
 162
 163     return refCount;
 164 }
 165
 166 static ULONG WINAPI ISF_MyComputer_fnRelease (IShellFolder2 * iface)
 167 {
 168     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 169     ULONG refCount = InterlockedDecrement(&This->ref);
 170
 171     TRACE ("(%p)->(count=%lu)\n", This, refCount + 1);
 172
 173     if (!refCount)
 174     {
 175         TRACE ("-- destroying IShellFolder(%p)\n", This);
 176         SHFree (This->pidlRoot);
 177         LocalFree (This);
 178     }
 179     return refCount;
 180 }
 181
 182 /**************************************************************************
 183 *    ISF_MyComputer_fnParseDisplayName
 184 */
 185 static HRESULT WINAPI ISF_MyComputer_fnParseDisplayName (IShellFolder2 *iface,
 186                HWND hwndOwner, LPBC pbc, LPOLESTR lpszDisplayName,
 187                DWORD * pchEaten, LPITEMIDLIST * ppidl, DWORD * pdwAttributes)
 188 {
 189     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 190     HRESULT hr = E_INVALIDARG;
 191     LPCWSTR szNext = NULL;
 192     WCHAR szElement[MAX_PATH];
 193     LPITEMIDLIST pidlTemp = NULL;
 194     CLSID clsid;
 195
 196     TRACE("(%p)->(HWND=%p,%p,%p=%s,%p,pidl=%p,%p)\n", This,
 197           hwndOwner, pbc, lpszDisplayName, debugstr_w (lpszDisplayName),
 198           pchEaten, ppidl, pdwAttributes);
 199
 200     *ppidl = 0;
 201     if (pchEaten)
 202         *pchEaten = 0;        /* strange but like the original */
 203
 204     /* handle CLSID paths */
 205     if (lpszDisplayName[0] == ':' && lpszDisplayName[1] == ':')
 206     {
 207         szNext = GetNextElementW (lpszDisplayName, szElement, MAX_PATH);
 208         TRACE ("-- element: %s\n", debugstr_w (szElement));
 209         SHCLSIDFromStringW (szElement + 2, &clsid);
 210         pidlTemp = _ILCreateGuid (PT_GUID, &clsid);
 211     }
 212     /* do we have an absolute path name ? */
 213     else if (PathGetDriveNumberW (lpszDisplayName) >= 0 &&
 214               lpszDisplayName[2] == (WCHAR) '\\')
 215     {
 216         szNext = GetNextElementW (lpszDisplayName, szElement, MAX_PATH);
 217         /* make drive letter uppercase to enable PIDL comparison */
 218         szElement[0] = toupper(szElement[0]);
 219         pidlTemp = _ILCreateDrive (szElement);
 220     }
 221
 222     if (szNext && *szNext)
 223     {
 224         hr = SHELL32_ParseNextElement (iface, hwndOwner, pbc, &pidlTemp,
 225                               (LPOLESTR) szNext, pchEaten, pdwAttributes);
 226     }
 227     else
 228     {
 229         if (pdwAttributes && *pdwAttributes)
 230             SHELL32_GetItemAttributes (&This->IShellFolder2_iface, pidlTemp, pdwAttributes);
 231         hr = S_OK;
 232     }
 233
 234     *ppidl = pidlTemp;
 235
 236     TRACE ("(%p)->(-- ret=0x%08lx)\n", This, hr);
 237
 238     return hr;
 239 }
 240
 241 /* retrieve a map of drives that should be displayed */
 242 static DWORD get_drive_map(void)
 243 {
 244     static DWORD drive_mask;
 245     static BOOL init_done = FALSE;
 246
 247     if (!init_done)
 248     {
 249         DWORD type, size, data, mask = 0;
 250         HKEY hkey;
 251
 252         if (!RegOpenKeyW( HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\Windows\\CurrentVersion\\Policies\\Explorer", &hkey ))
 253         {
 254             size = sizeof(data);
 255             if (!RegQueryValueExW( hkey, L"NoDrives", NULL, &type, (LPBYTE)&data, &size ) && type == REG_DWORD)
 256                 mask |= data;
 257             RegCloseKey( hkey );
 258         }
 259         if (!RegOpenKeyW( HKEY_CURRENT_USER, L"Software\\Microsoft\\Windows\\CurrentVersion\\Policies\\Explorer", &hkey ))
 260         {
 261             size = sizeof(data);
 262             if (!RegQueryValueExW( hkey, L"NoDrives", NULL, &type, (LPBYTE)&data, &size ) && type == REG_DWORD)
 263                 mask |= data;
 264             RegCloseKey( hkey );
 265         }
 266         drive_mask = mask;
 267         init_done = TRUE;
 268     }
 269
 270     return GetLogicalDrives() & ~drive_mask;
 271 }
 272
 273 /**************************************************************************
 274  *  CreateMyCompEnumList()
 275  */
 276 static BOOL CreateMyCompEnumList(IEnumIDListImpl *list, DWORD dwFlags)
 277 {
 278     BOOL ret = TRUE;
 279
 280     TRACE("(%p)->(flags=0x%08lx)\n", list, dwFlags);
 281
 282     /* enumerate the folders */
 283     if (dwFlags & SHCONTF_FOLDERS)
 284     {
 285         WCHAR wszDriveName[] = L"A:\\";
 286         DWORD dwDrivemap = get_drive_map();
 287         HKEY hkey;
 288         UINT i;
 289
 290         while (ret && wszDriveName[0]<='Z')
 291         {
 292             if(dwDrivemap & 0x00000001L)
 293                 ret = AddToEnumList(list, _ILCreateDrive(wszDriveName));
 294             wszDriveName[0]++;
 295             dwDrivemap = dwDrivemap >> 1;
 296         }
 297
 298         TRACE("-- (%p)-> enumerate (mycomputer shell extensions)\n",list);
 299         for (i=0; i<2; i++) {
 300             if (ret && !RegOpenKeyExW(i == 0 ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER,
 301                                       L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Explorer\\MyComputer\\Namespace",
 302                                       0, KEY_READ, &hkey))
 303             {
 304                 WCHAR iid[50];
 305                 int i=0;
 306
 307                 while (ret)
 308                 {
 309                     DWORD size;
 310                     LONG r;
 311
 312                     size = ARRAY_SIZE(iid);
 313                     r = RegEnumKeyExW(hkey, i, iid, &size, 0, NULL, NULL, NULL);
 314                     if (ERROR_SUCCESS == r)
 315                     {
 316                         /* FIXME: shell extensions, shouldn't the type be
 317                          * PT_SHELLEXT? */
 318                         ret = AddToEnumList(list, _ILCreateGuidFromStrW(iid));
 319                         i++;
 320                     }
 321                     else if (ERROR_NO_MORE_ITEMS == r)
 322                         break;
 323                     else
 324                         ret = FALSE;
 325                 }
 326                 RegCloseKey(hkey);
 327             }
 328         }
 329     }
 330     return ret;
 331 }
 332
 333 /**************************************************************************
 334 *        ISF_MyComputer_fnEnumObjects
 335 */
 336 static HRESULT WINAPI ISF_MyComputer_fnEnumObjects (IShellFolder2 *iface,
 337                HWND hwndOwner, DWORD dwFlags, LPENUMIDLIST *ppEnumIDList)
 338 {
 339     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 340     IEnumIDListImpl *list;
 341
 342     TRACE("(%p)->(HWND=%p flags=0x%08lx pplist=%p)\n", This,
 343           hwndOwner, dwFlags, ppEnumIDList);
 344
 345     if (!(list = IEnumIDList_Constructor()))
 346         return E_OUTOFMEMORY;
 347     CreateMyCompEnumList(list, dwFlags);
 348     *ppEnumIDList = &list->IEnumIDList_iface;
 349
 350     TRACE ("-- (%p)->(new ID List: %p)\n", This, *ppEnumIDList);
 351
 352     return S_OK;
 353 }
 354
 355 /**************************************************************************
 356 *        ISF_MyComputer_fnBindToObject
 357 */
 358 static HRESULT WINAPI ISF_MyComputer_fnBindToObject (IShellFolder2 *iface,
 359                LPCITEMIDLIST pidl, LPBC pbcReserved, REFIID riid, LPVOID *ppvOut)
 360 {
 361     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 362
 363     TRACE("(%p)->(pidl=%p,%p,%s,%p)\n", This,
 364           pidl, pbcReserved, shdebugstr_guid (riid), ppvOut);
 365
 366     return SHELL32_BindToChild (This->pidlRoot, &CLSID_ShellFSFolder, NULL, pidl, riid, ppvOut);
 367 }
 368
 369 /**************************************************************************
 370 *    ISF_MyComputer_fnBindToStorage
 371 */
 372 static HRESULT WINAPI ISF_MyComputer_fnBindToStorage (IShellFolder2 * iface,
 373                LPCITEMIDLIST pidl, LPBC pbcReserved, REFIID riid, LPVOID *ppvOut)
 374 {
 375     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 376
 377     FIXME("(%p)->(pidl=%p,%p,%s,%p) stub\n", This,
 378           pidl, pbcReserved, shdebugstr_guid (riid), ppvOut);
 379
 380     *ppvOut = NULL;
 381     return E_NOTIMPL;
 382 }
 383
 384 /**************************************************************************
 385 *     ISF_MyComputer_fnCompareIDs
 386 */
 387
 388 static HRESULT WINAPI ISF_MyComputer_fnCompareIDs (IShellFolder2 *iface,
 389                LPARAM lParam, LPCITEMIDLIST pidl1, LPCITEMIDLIST pidl2)
 390 {
 391     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 392     HRESULT hr;
 393
 394     TRACE ("(%p)->(0x%08Ix,pidl1=%p,pidl2=%p)\n", This, lParam, pidl1, pidl2);
 395     hr = SHELL32_CompareIDs(&This->IShellFolder2_iface, lParam, pidl1, pidl2);
 396     TRACE ("-- 0x%08lx\n", hr);
 397     return hr;
 398 }
 399
 400 /**************************************************************************
 401 *    ISF_MyComputer_fnCreateViewObject
 402 */
 403 static HRESULT WINAPI ISF_MyComputer_fnCreateViewObject (IShellFolder2 *iface,
 404                HWND hwndOwner, REFIID riid, LPVOID * ppvOut)
 405 {
 406     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 407     LPSHELLVIEW pShellView;
 408     HRESULT hr = E_INVALIDARG;
 409
 410     TRACE("(%p)->(hwnd=%p,%s,%p)\n", This,
 411           hwndOwner, shdebugstr_guid (riid), ppvOut);
 412
 413     if (!ppvOut)
 414         return hr;
 415
 416     *ppvOut = NULL;
 417
 418     if (IsEqualIID (riid, &IID_IDropTarget))
 419     {
 420         WARN ("IDropTarget not implemented\n");
 421         hr = E_NOTIMPL;
 422     }
 423     else if (IsEqualIID (riid, &IID_IContextMenu))
 424     {
 425         hr = BackgroundMenu_Constructor((IShellFolder*)iface, FALSE, riid, ppvOut);
 426     }
 427     else if (IsEqualIID (riid, &IID_IShellView))
 428     {
 429         pShellView = IShellView_Constructor ((IShellFolder *) iface);
 430         if (pShellView)
 431         {
 432             hr = IShellView_QueryInterface (pShellView, riid, ppvOut);
 433             IShellView_Release (pShellView);
 434         }
 435     }
 436     TRACE ("-- (%p)->(interface=%p)\n", This, ppvOut);
 437     return hr;
 438 }
 439
 440 /**************************************************************************
 441 *  ISF_MyComputer_fnGetAttributesOf
 442 */
 443 static HRESULT WINAPI ISF_MyComputer_fnGetAttributesOf (IShellFolder2 * iface,
 444                 UINT cidl, LPCITEMIDLIST * apidl, DWORD * rgfInOut)
 445 {
 446     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 447     HRESULT hr = S_OK;
 448
 449     TRACE ("(%p)->(cidl=%d apidl=%p mask=%p (0x%08lx))\n",
 450            This, cidl, apidl, rgfInOut, rgfInOut ? *rgfInOut : 0);
 451
 452     if (!rgfInOut)
 453         return E_INVALIDARG;
 454     if (cidl && !apidl)
 455         return E_INVALIDARG;
 456
 457     if (*rgfInOut == 0)
 458         *rgfInOut = ~0;
 459
 460     if(cidl == 0){
 461         IShellFolder2 *parent = NULL;
 462         LPCITEMIDLIST rpidl = NULL;
 463
 464         hr = SHBindToParent(This->pidlRoot, &IID_IShellFolder2, (void **)&parent, &rpidl);
 465         if(SUCCEEDED(hr)) {
 466             SHELL32_GetItemAttributes(parent, rpidl, rgfInOut);
 467             IShellFolder2_Release(parent);
 468         }
 469     } else {
 470         while (cidl > 0 && *apidl) {
 471             pdump (*apidl);
 472             SHELL32_GetItemAttributes(&This->IShellFolder2_iface, *apidl, rgfInOut);
 473             apidl++;
 474             cidl--;
 475         }
 476     }
 477     /* make sure SFGAO_VALIDATE is cleared, some apps depend on that */
 478     *rgfInOut &= ~SFGAO_VALIDATE;
 479
 480     TRACE ("-- result=0x%08lx\n", *rgfInOut);
 481     return hr;
 482 }
 483
 484 /**************************************************************************
 485 *    ISF_MyComputer_fnGetUIObjectOf
 486 *
 487 * PARAMETERS
 488 *  hwndOwner [in]  Parent window for any output
 489 *  cidl      [in]  array size
 490 *  apidl     [in]  simple pidl array
 491 *  riid      [in]  Requested Interface
 492 *  prgfInOut [   ] reserved
 493 *  ppvObject [out] Resulting Interface
 494 *
 495 */
 496 static HRESULT WINAPI ISF_MyComputer_fnGetUIObjectOf (IShellFolder2 * iface,
 497                 HWND hwndOwner, UINT cidl, LPCITEMIDLIST * apidl, REFIID riid,
 498                 UINT * prgfInOut, LPVOID * ppvOut)
 499 {
 500     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 501
 502     LPITEMIDLIST pidl;
 503     IUnknown *pObj = NULL;
 504     HRESULT hr = E_INVALIDARG;
 505
 506     TRACE("(%p)->(%p,%u,apidl=%p,%s,%p,%p)\n", This,
 507           hwndOwner, cidl, apidl, shdebugstr_guid (riid), prgfInOut, ppvOut);
 508
 509     if (!ppvOut)
 510         return hr;
 511
 512     *ppvOut = NULL;
 513
 514     if (IsEqualIID (riid, &IID_IContextMenu) && (cidl >= 1))
 515     {
 516         return ItemMenu_Constructor((IShellFolder*) iface, This->pidlRoot, apidl, cidl, riid, ppvOut);
 517     }
 518     else if (IsEqualIID (riid, &IID_IDataObject) && (cidl >= 1))
 519     {
 520         pObj = (LPUNKNOWN) IDataObject_Constructor (hwndOwner,
 521                                               This->pidlRoot, apidl, cidl);
 522         hr = S_OK;
 523     }
 524     else if (IsEqualIID (riid, &IID_IExtractIconA) && (cidl == 1))
 525     {
 526         pidl = ILCombine (This->pidlRoot, apidl[0]);
 527         pObj = (LPUNKNOWN) IExtractIconA_Constructor (pidl);
 528         SHFree (pidl);
 529         hr = S_OK;
 530     }
 531     else if (IsEqualIID (riid, &IID_IExtractIconW) && (cidl == 1))
 532     {
 533         pidl = ILCombine (This->pidlRoot, apidl[0]);
 534         pObj = (LPUNKNOWN) IExtractIconW_Constructor (pidl);
 535         SHFree (pidl);
 536         hr = S_OK;
 537     }
 538     else if (IsEqualIID (riid, &IID_IDropTarget) && (cidl >= 1))
 539     {
 540         hr = IShellFolder2_QueryInterface (iface, &IID_IDropTarget,
 541                                           (LPVOID *) &pObj);
 542     }
 543     else if ((IsEqualIID(riid,&IID_IShellLinkW) ||
 544               IsEqualIID(riid,&IID_IShellLinkA)) && (cidl == 1))
 545     {
 546         pidl = ILCombine (This->pidlRoot, apidl[0]);
 547         hr = IShellLink_ConstructFromFile(NULL, riid, pidl, &pObj);
 548         SHFree (pidl);
 549     }
 550     else
 551         hr = E_NOINTERFACE;
 552
 553     if (SUCCEEDED(hr) && !pObj)
 554         hr = E_OUTOFMEMORY;
 555
 556     *ppvOut = pObj;
 557     TRACE ("(%p)->hr=0x%08lx\n", This, hr);
 558     return hr;
 559 }
 560
 561 /**************************************************************************
 562 *    ISF_MyComputer_fnGetDisplayNameOf
 563 */
 564 static HRESULT WINAPI ISF_MyComputer_fnGetDisplayNameOf (IShellFolder2 *iface,
 565                LPCITEMIDLIST pidl, DWORD dwFlags, LPSTRRET strRet)
 566 {
 567     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 568
 569     LPWSTR pszPath;
 570     HRESULT hr = S_OK;
 571
 572     TRACE ("(%p)->(pidl=%p,0x%08lx,%p)\n", This, pidl, dwFlags, strRet);
 573     pdump (pidl);
 574
 575     if (!strRet)
 576         return E_INVALIDARG;
 577
 578     pszPath = CoTaskMemAlloc((MAX_PATH +1) * sizeof(WCHAR));
 579     if (!pszPath)
 580         return E_OUTOFMEMORY;
 581
 582     pszPath[0] = 0;
 583
 584     if (!pidl->mkid.cb)
 585     {
 586         /* parsing name like ::{...} */
 587         pszPath[0] = ':';
 588         pszPath[1] = ':';
 589         SHELL32_GUIDToStringW(&CLSID_MyComputer, &pszPath[2]);
 590     }
 591     else if (_ILIsPidlSimple(pidl))
 592     {
 593         /* take names of special folders only if its only this folder */
 594         if (_ILIsSpecialFolder(pidl))
 595         {
 596             GUID const *clsid;
 597
 598             clsid = _ILGetGUIDPointer (pidl);
 599             if (clsid)
 600             {
 601                 if ((GET_SHGDN_FOR (dwFlags) & (SHGDN_FORPARSING | SHGDN_FORADDRESSBAR)) == SHGDN_FORPARSING)
 602                 {
 603                     BOOL bWantsForParsing = FALSE;
 604                     WCHAR szRegPath[100];
 605                     LONG r;
 606
 607                     /*
 608                      * We can only get a filesystem path from a shellfolder
 609                      * if the value WantsFORPARSING exists in
 610                      *      CLSID\\{...}\\shellfolder
 611                      * exception: the MyComputer folder has this keys not
 612                      *            but like any filesystem backed
 613                      *            folder it needs these behaviour
 614                      *
 615                      * Get the "WantsFORPARSING" flag from the registry
 616                      */
 617
 618                     lstrcpyW (szRegPath, L"CLSID\\");
 619                     SHELL32_GUIDToStringW (clsid, &szRegPath[6]);
 620                     lstrcatW (szRegPath, L"\\shellfolder");
 621                     r = SHGetValueW (HKEY_CLASSES_ROOT, szRegPath, L"WantsForParsing", NULL, NULL, NULL);
 622                     if (r == ERROR_SUCCESS)
 623                         bWantsForParsing = TRUE;
 624
 625                     if ((GET_SHGDN_RELATION (dwFlags) == SHGDN_NORMAL) &&
 626                         bWantsForParsing)
 627                     {
 628                         /*
 629                          * We need the filesystem path to the destination folder
 630                          * Only the folder itself can know it
 631                          */
 632                         hr = SHELL32_GetDisplayNameOfChild (iface, pidl,
 633                                                 dwFlags, pszPath, MAX_PATH);
 634                     }
 635                     else
 636                     {
 637                         LPWSTR p = pszPath;
 638
 639                         /* parsing name like ::{...} */
 640                         p[0] = ':';
 641                         p[1] = ':';
 642                         p += 2;
 643                         p += SHELL32_GUIDToStringW(&CLSID_MyComputer, p);
 644
 645                         /* \:: */
 646                         p[0] = '\\';
 647                         p[1] = ':';
 648                         p[2] = ':';
 649                         p += 3;
 650                         SHELL32_GUIDToStringW(clsid, p);
 651                     }
 652                 }
 653                 else
 654                 {
 655                     /* user friendly name */
 656                     HCR_GetClassNameW (clsid, pszPath, MAX_PATH);
 657                 }
 658             }
 659             else
 660             {
 661                 /* append my own path */
 662                 _ILSimpleGetTextW (pidl, pszPath, MAX_PATH);
 663             }
 664         }
 665         else if (_ILIsDrive(pidl))
 666         {
 667             _ILSimpleGetTextW (pidl, pszPath, MAX_PATH);    /* append my own path */
 668
 669             /* long view "lw_name (C:)" */
 670             if (!(dwFlags & SHGDN_FORPARSING))
 671             {
 672                 WCHAR wszDrive[32 /* label */ + 6 /* ' (C:)'\0 */] = {0};
 673
 674                 GetVolumeInformationW (pszPath, wszDrive, ARRAY_SIZE(wszDrive) - 5, NULL, NULL,
 675                         NULL, NULL, 0);
 676                 lstrcatW (wszDrive, L" (");
 677                 lstrcpynW (wszDrive + lstrlenW(wszDrive), pszPath, 3);
 678                 lstrcatW (wszDrive, L")");
 679                 lstrcpyW (pszPath, wszDrive);
 680             }
 681         }
 682         else
 683         {
 684             /* Neither a shell namespace extension nor a drive letter. */
 685             ERR("Wrong pidl type\n");
 686             CoTaskMemFree(pszPath);
 687             return E_INVALIDARG;
 688         }
 689     }
 690     else
 691     {
 692         /* Complex pidl. Let the child folder do the work */
 693         hr = SHELL32_GetDisplayNameOfChild(iface, pidl, dwFlags, pszPath, MAX_PATH);
 694     }
 695
 696     if (SUCCEEDED (hr))
 697     {
 698         /* Win9x always returns ANSI strings, NT always returns Unicode strings */
 699         if (GetVersion() & 0x80000000)
 700         {
 701             strRet->uType = STRRET_CSTR;
 702             if (!WideCharToMultiByte(CP_ACP, 0, pszPath, -1, strRet->u.cStr, MAX_PATH,
 703                     NULL, NULL))
 704                 strRet->u.cStr[0] = '\0';
 705             CoTaskMemFree(pszPath);
 706         }
 707         else
 708         {
 709             strRet->uType = STRRET_WSTR;
 710             strRet->u.pOleStr = pszPath;
 711         }
 712     }
 713     else
 714         CoTaskMemFree(pszPath);
 715
 716     TRACE ("-- (%p)->(%s)\n", This, strRet->uType == STRRET_CSTR ? strRet->u.cStr : debugstr_w(strRet->u.pOleStr));
 717     return hr;
 718 }
 719
 720 /**************************************************************************
 721 *  ISF_MyComputer_fnSetNameOf
 722 *  Changes the name of a file object or subfolder, possibly changing its item
 723 *  identifier in the process.
 724 *
 725 * PARAMETERS
 726 *  hwndOwner  [in]   Owner window for output
 727 *  pidl       [in]   simple pidl of item to change
 728 *  lpszName   [in]   the items new display name
 729 *  dwFlags    [in]   SHGNO formatting flags
 730 *  ppidlOut   [out]  simple pidl returned
 731 */
 732 static HRESULT WINAPI ISF_MyComputer_fnSetNameOf (
 733                IShellFolder2 * iface, HWND hwndOwner, LPCITEMIDLIST pidl,
 734                LPCOLESTR lpName, DWORD dwFlags, LPITEMIDLIST * pPidlOut)
 735 {
 736     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 737     FIXME ("(%p)->(%p,pidl=%p,%s,%lu,%p)\n", This,
 738            hwndOwner, pidl, debugstr_w (lpName), dwFlags, pPidlOut);
 739     return E_FAIL;
 740 }
 741
 742 static HRESULT WINAPI ISF_MyComputer_fnGetDefaultSearchGUID(IShellFolder2 *iface, GUID *guid)
 743 {
 744     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 745     TRACE("(%p)->(%p)\n", This, guid);
 746     return E_NOTIMPL;
 747 }
 748 static HRESULT WINAPI ISF_MyComputer_fnEnumSearches (
 749                IShellFolder2 * iface, IEnumExtraSearch ** ppenum)
 750 {
 751     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 752     FIXME ("(%p)\n", This);
 753     return E_NOTIMPL;
 754 }
 755
 756 static HRESULT WINAPI ISF_MyComputer_fnGetDefaultColumn(IShellFolder2 *iface, DWORD reserved,
 757         ULONG *sort, ULONG *display)
 758 {
 759     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 760
 761     TRACE("(%p)->(%#lx, %p, %p)\n", This, reserved, sort, display);
 762
 763     return E_NOTIMPL;
 764 }
 765
 766 static HRESULT WINAPI ISF_MyComputer_fnGetDefaultColumnState (
 767                IShellFolder2 * iface, UINT iColumn, DWORD * pcsFlags)
 768 {
 769     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 770
 771     TRACE ("(%p)->(%d %p)\n", This, iColumn, pcsFlags);
 772
 773     if (!pcsFlags || iColumn >= ARRAY_SIZE(mycomputer_header))
 774         return E_INVALIDARG;
 775
 776     *pcsFlags = mycomputer_header[iColumn].pcsFlags;
 777
 778     return S_OK;
 779 }
 780
 781 static HRESULT WINAPI ISF_MyComputer_fnGetDetailsEx (IShellFolder2 * iface,
 782                LPCITEMIDLIST pidl, const SHCOLUMNID * pscid, VARIANT * pv)
 783 {
 784     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 785     FIXME ("(%p)\n", This);
 786     return E_NOTIMPL;
 787 }
 788
 789 /* FIXME: drive size >4GB is rolling over */
 790 static HRESULT WINAPI ISF_MyComputer_fnGetDetailsOf (IShellFolder2 *iface,
 791                LPCITEMIDLIST pidl, UINT iColumn, SHELLDETAILS *psd)
 792 {
 793     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 794     char szPath[MAX_PATH];
 795     ULARGE_INTEGER ulBytes;
 796     HRESULT hr = S_OK;
 797
 798     TRACE ("(%p)->(%p %i %p)\n", This, pidl, iColumn, psd);
 799
 800     if (!psd || iColumn >= ARRAY_SIZE(mycomputer_header))
 801         return E_INVALIDARG;
 802
 803     if (!pidl)
 804         return SHELL32_GetColumnDetails(mycomputer_header, iColumn, psd);
 805
 806     psd->str.u.cStr[0] = 0;
 807     psd->str.uType = STRRET_CSTR;
 808
 809     switch (iColumn)
 810     {
 811         case 2:        /* total size */
 812             if (_ILIsDrive (pidl))
 813             {
 814                 _ILSimpleGetText (pidl, szPath, MAX_PATH);
 815                 GetDiskFreeSpaceExA (szPath, NULL, &ulBytes, NULL);
 816                 StrFormatByteSizeA (ulBytes.u.LowPart, psd->str.u.cStr, MAX_PATH);
 817             }
 818             break;
 819         case 3:        /* free size */
 820             if (_ILIsDrive (pidl))
 821             {
 822                 _ILSimpleGetText (pidl, szPath, MAX_PATH);
 823                 GetDiskFreeSpaceExA (szPath, &ulBytes, NULL, NULL);
 824                 StrFormatByteSizeA (ulBytes.u.LowPart, psd->str.u.cStr, MAX_PATH);
 825             }
 826             break;
 827
 828         default:
 829             return shellfolder_get_file_details( iface, pidl, mycomputer_header, iColumn, psd );
 830     }
 831
 832     return hr;
 833 }
 834
 835 static HRESULT WINAPI ISF_MyComputer_fnMapColumnToSCID (IShellFolder2 *iface, UINT column, SHCOLUMNID *scid)
 836 {
 837     IMyComputerFolderImpl *This = impl_from_IShellFolder2(iface);
 838
 839     TRACE("(%p)->(%u %p)\n", This, column, scid);
 840
 841     if (column >= ARRAY_SIZE(mycomputer_header))
 842         return E_INVALIDARG;
 843
 844     return shellfolder_map_column_to_scid(mycomputer_header, column, scid);
 845 }
 846
 847 static const IShellFolder2Vtbl vt_ShellFolder2 =
 848 {
 849     ISF_MyComputer_fnQueryInterface,
 850     ISF_MyComputer_fnAddRef,
 851     ISF_MyComputer_fnRelease,
 852     ISF_MyComputer_fnParseDisplayName,
 853     ISF_MyComputer_fnEnumObjects,
 854     ISF_MyComputer_fnBindToObject,
 855     ISF_MyComputer_fnBindToStorage,
 856     ISF_MyComputer_fnCompareIDs,
 857     ISF_MyComputer_fnCreateViewObject,
 858     ISF_MyComputer_fnGetAttributesOf,
 859     ISF_MyComputer_fnGetUIObjectOf,
 860     ISF_MyComputer_fnGetDisplayNameOf,
 861     ISF_MyComputer_fnSetNameOf,
 862     /* ShellFolder2 */
 863     ISF_MyComputer_fnGetDefaultSearchGUID,
 864     ISF_MyComputer_fnEnumSearches,
 865     ISF_MyComputer_fnGetDefaultColumn,
 866     ISF_MyComputer_fnGetDefaultColumnState,
 867     ISF_MyComputer_fnGetDetailsEx,
 868     ISF_MyComputer_fnGetDetailsOf,
 869     ISF_MyComputer_fnMapColumnToSCID
 870 };
 871
 872 /************************************************************************
 873  *    IMCFldr_PersistFolder2_QueryInterface
 874  */
 875 static HRESULT WINAPI IMCFldr_PersistFolder2_QueryInterface (
 876                IPersistFolder2 * iface, REFIID iid, LPVOID * ppvObj)
 877 {
 878     IMyComputerFolderImpl *This = impl_from_IPersistFolder2(iface);
 879     TRACE ("(%p)\n", This);
 880     return IShellFolder2_QueryInterface (&This->IShellFolder2_iface, iid, ppvObj);
 881 }
 882
 883 /************************************************************************
 884  *    IMCFldr_PersistFolder2_AddRef
 885  */
 886 static ULONG WINAPI IMCFldr_PersistFolder2_AddRef (IPersistFolder2 * iface)
 887 {
 888     IMyComputerFolderImpl *This = impl_from_IPersistFolder2(iface);
 889     TRACE ("(%p)->(count=%lu)\n", This, This->ref);
 890     return IShellFolder2_AddRef (&This->IShellFolder2_iface);
 891 }
 892
 893 /************************************************************************
 894  *    ISFPersistFolder_Release
 895  */
 896 static ULONG WINAPI IMCFldr_PersistFolder2_Release (IPersistFolder2 * iface)
 897 {
 898     IMyComputerFolderImpl *This = impl_from_IPersistFolder2(iface);
 899     TRACE ("(%p)->(count=%lu)\n", This, This->ref);
 900     return IShellFolder2_Release (&This->IShellFolder2_iface);
 901 }
 902
 903 /************************************************************************
 904  *    IMCFldr_PersistFolder2_GetClassID
 905  */
 906 static HRESULT WINAPI IMCFldr_PersistFolder2_GetClassID (
 907                IPersistFolder2 * iface, CLSID * lpClassId)
 908 {
 909     IMyComputerFolderImpl *This = impl_from_IPersistFolder2(iface);
 910
 911     TRACE ("(%p)\n", This);
 912
 913     if (!lpClassId)
 914     return E_POINTER;
 915     *lpClassId = CLSID_MyComputer;
 916
 917     return S_OK;
 918 }
 919
 920 /************************************************************************
 921  *    IMCFldr_PersistFolder2_Initialize
 922  *
 923  * NOTES: it makes no sense to change the pidl
 924  */
 925 static HRESULT WINAPI IMCFldr_PersistFolder2_Initialize (
 926                IPersistFolder2 * iface, LPCITEMIDLIST pidl)
 927 {
 928     IMyComputerFolderImpl *This = impl_from_IPersistFolder2(iface);
 929     TRACE ("(%p)->(%p)\n", This, pidl);
 930     return E_NOTIMPL;
 931 }
 932
 933 /**************************************************************************
 934  *    IPersistFolder2_fnGetCurFolder
 935  */
 936 static HRESULT WINAPI IMCFldr_PersistFolder2_GetCurFolder (
 937                IPersistFolder2 * iface, LPITEMIDLIST * pidl)
 938 {
 939     IMyComputerFolderImpl *This = impl_from_IPersistFolder2(iface);
 940
 941     TRACE ("(%p)->(%p)\n", This, pidl);
 942
 943     if (!pidl)
 944         return E_POINTER;
 945     *pidl = ILClone (This->pidlRoot);
 946     return S_OK;
 947 }
 948
 949 static const IPersistFolder2Vtbl vt_PersistFolder2 =
 950 {
 951     IMCFldr_PersistFolder2_QueryInterface,
 952     IMCFldr_PersistFolder2_AddRef,
 953     IMCFldr_PersistFolder2_Release,
 954     IMCFldr_PersistFolder2_GetClassID,
 955     IMCFldr_PersistFolder2_Initialize,
 956     IMCFldr_PersistFolder2_GetCurFolder
 957 };