netstat: Initial implementation.
[wine] / dlls / shell32 / shfldr_desktop.c
1
2 /*
3  *    Virtual Desktop Folder
4  *
5  *    Copyright 1997            Marcus Meissner
6  *    Copyright 1998, 1999, 2002    Juergen Schmied
7  *
8  * This library is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * This library is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with this library; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21  */
22
23 #include "config.h"
24 #include "wine/port.h"
25
26 #include <stdlib.h>
27 #include <string.h>
28 #include <stdarg.h>
29 #include <stdio.h>
30
31 #define COBJMACROS
32 #define NONAMELESSUNION
33 #define NONAMELESSSTRUCT
34
35 #include "winerror.h"
36 #include "windef.h"
37 #include "winbase.h"
38 #include "winreg.h"
39 #include "wingdi.h"
40 #include "winuser.h"
41
42 #include "ole2.h"
43 #include "shlguid.h"
44
45 #include "pidl.h"
46 #include "undocshell.h"
47 #include "shell32_main.h"
48 #include "shresdef.h"
49 #include "shlwapi.h"
50 #include "shellfolder.h"
51 #include "wine/debug.h"
52 #include "debughlp.h"
53 #include "shfldr.h"
54
55 WINE_DEFAULT_DEBUG_CHANNEL (shell);
56
57 /* Undocumented functions from shdocvw */
58 extern HRESULT WINAPI IEParseDisplayNameWithBCW(DWORD codepage, LPCWSTR lpszDisplayName, LPBC pbc, LPITEMIDLIST *ppidl);
59
60
61 /***********************************************************************
62 *     Desktopfolder implementation
63 */
64
65 typedef struct {
66     IShellFolder2 IShellFolder2_iface;
67     IPersistFolder2 IPersistFolder2_iface;
68     LONG ref;
69
70     /* both paths are parsible from the desktop */
71     LPWSTR sPathTarget;     /* complete path to target used for enumeration and ChangeNotify */
72     LPITEMIDLIST pidlRoot;  /* absolute pidl */
73
74     UINT cfShellIDList;        /* clipboardformat for IDropTarget */
75     BOOL fAcceptFmt;        /* flag for pending Drop */
76 } IDesktopFolderImpl;
77
78 static inline IDesktopFolderImpl *impl_from_IShellFolder2(IShellFolder2 *iface)
79 {
80     return CONTAINING_RECORD(iface, IDesktopFolderImpl, IShellFolder2_iface);
81 }
82
83 static inline IDesktopFolderImpl *impl_from_IPersistFolder2( IPersistFolder2 *iface )
84 {
85     return CONTAINING_RECORD(iface, IDesktopFolderImpl, IPersistFolder2_iface);
86 }
87
88 static const shvheader desktop_header[] = {
89     {IDS_SHV_COLUMN1, SHCOLSTATE_TYPE_STR | SHCOLSTATE_ONBYDEFAULT, LVCFMT_RIGHT, 15},
90     {IDS_SHV_COLUMN2, SHCOLSTATE_TYPE_STR | SHCOLSTATE_ONBYDEFAULT, LVCFMT_RIGHT, 10},
91     {IDS_SHV_COLUMN3, SHCOLSTATE_TYPE_STR | SHCOLSTATE_ONBYDEFAULT, LVCFMT_RIGHT, 10},
92     {IDS_SHV_COLUMN4, SHCOLSTATE_TYPE_DATE | SHCOLSTATE_ONBYDEFAULT, LVCFMT_RIGHT, 12},
93     {IDS_SHV_COLUMN5, SHCOLSTATE_TYPE_STR | SHCOLSTATE_ONBYDEFAULT, LVCFMT_RIGHT, 5}
94 };
95
96 #define DESKTOPSHELLVIEWCOLUMNS sizeof(desktop_header)/sizeof(shvheader)
97
98 /**************************************************************************
99  *    ISF_Desktop_fnQueryInterface
100  *
101  */
102 static HRESULT WINAPI ISF_Desktop_fnQueryInterface(
103                 IShellFolder2 * iface, REFIID riid, LPVOID * ppvObj)
104 {
105     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
106
107     TRACE ("(%p)->(%s,%p)\n", This, shdebugstr_guid (riid), ppvObj);
108
109     *ppvObj = NULL;
110
111     if (IsEqualIID (riid, &IID_IUnknown) ||
112         IsEqualIID (riid, &IID_IShellFolder) ||
113         IsEqualIID (riid, &IID_IShellFolder2))
114     {
115         *ppvObj = This;
116     }
117     else if (IsEqualIID (riid, &IID_IPersist) ||
118              IsEqualIID (riid, &IID_IPersistFolder) ||
119              IsEqualIID (riid, &IID_IPersistFolder2))
120     {
121         *ppvObj = &This->IPersistFolder2_iface;
122     }
123
124     if (*ppvObj)
125     {
126         IUnknown_AddRef ((IUnknown *) (*ppvObj));
127         TRACE ("-- Interface: (%p)->(%p)\n", ppvObj, *ppvObj);
128         return S_OK;
129     }
130     TRACE ("-- Interface: E_NOINTERFACE\n");
131     return E_NOINTERFACE;
132 }
133
134 static ULONG WINAPI ISF_Desktop_fnAddRef (IShellFolder2 * iface)
135 {
136     return 2; /* non-heap based object */
137 }
138
139 static ULONG WINAPI ISF_Desktop_fnRelease (IShellFolder2 * iface)
140 {
141     return 1; /* non-heap based object */
142 }
143
144 /**************************************************************************
145  *    ISF_Desktop_fnParseDisplayName
146  *
147  * NOTES
148  *    "::{20D04FE0-3AEA-1069-A2D8-08002B30309D}" and "" binds
149  *    to MyComputer
150  */
151 static HRESULT WINAPI ISF_Desktop_fnParseDisplayName (IShellFolder2 * iface,
152                 HWND hwndOwner, LPBC pbc, LPOLESTR lpszDisplayName,
153                 DWORD * pchEaten, LPITEMIDLIST * ppidl, DWORD * pdwAttributes)
154 {
155     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
156     IShellFolder *shell_folder = (IShellFolder*)iface;
157     WCHAR szElement[MAX_PATH];
158     LPCWSTR szNext = NULL;
159     LPITEMIDLIST pidlTemp = NULL;
160     PARSEDURLW urldata;
161     HRESULT hr = S_OK;
162     CLSID clsid;
163
164     TRACE ("(%p)->(HWND=%p,%p,%p=%s,%p,pidl=%p,%p)\n",
165            This, hwndOwner, pbc, lpszDisplayName, debugstr_w(lpszDisplayName),
166            pchEaten, ppidl, pdwAttributes);
167
168     if (!ppidl) return E_INVALIDARG;
169     *ppidl = 0;
170
171     if (!lpszDisplayName) return E_INVALIDARG;
172
173     if (pchEaten)
174         *pchEaten = 0;        /* strange but like the original */
175
176     urldata.cbSize = sizeof(urldata);
177
178     if (lpszDisplayName[0] == ':' && lpszDisplayName[1] == ':')
179     {
180         szNext = GetNextElementW (lpszDisplayName, szElement, MAX_PATH);
181         TRACE ("-- element: %s\n", debugstr_w (szElement));
182         SHCLSIDFromStringW (szElement + 2, &clsid);
183         pidlTemp = _ILCreateGuid (PT_GUID, &clsid);
184     }
185     else if (PathGetDriveNumberW (lpszDisplayName) >= 0)
186     {
187         /* it's a filesystem path with a drive. Let MyComputer/UnixDosFolder parse it */
188         if (UNIXFS_is_rooted_at_desktop()) 
189             pidlTemp = _ILCreateGuid(PT_GUID, &CLSID_UnixDosFolder);
190         else
191             pidlTemp = _ILCreateMyComputer ();
192         szNext = lpszDisplayName;
193     }
194     else if (PathIsUNCW(lpszDisplayName))
195     {
196         pidlTemp = _ILCreateNetwork();
197         szNext = lpszDisplayName;
198     }
199     else if( (pidlTemp = SHELL32_CreatePidlFromBindCtx(pbc, lpszDisplayName)) )
200     {
201         *ppidl = pidlTemp;
202         return S_OK;
203     }
204     else if (SUCCEEDED(ParseURLW(lpszDisplayName, &urldata)))
205     {
206         if (urldata.nScheme == URL_SCHEME_SHELL) /* handle shell: urls */
207         {
208             TRACE ("-- shell url: %s\n", debugstr_w(urldata.pszSuffix));
209             SHCLSIDFromStringW (urldata.pszSuffix+2, &clsid);
210             pidlTemp = _ILCreateGuid (PT_GUID, &clsid);
211         }
212         else
213             return IEParseDisplayNameWithBCW(CP_ACP,lpszDisplayName,pbc,ppidl);
214     }
215     else
216     {
217         /* it's a filesystem path on the desktop. Let a FSFolder parse it */
218
219         if (*lpszDisplayName)
220         {
221             if (*lpszDisplayName == '/')
222             {
223                 /* UNIX paths should be parsed by unixfs */
224                 IShellFolder *unixFS;
225                 hr = UnixFolder_Constructor(NULL, &IID_IShellFolder, (LPVOID*)&unixFS);
226                 if (SUCCEEDED(hr))
227                 {
228                     hr = IShellFolder_ParseDisplayName(unixFS, NULL, NULL,
229                             lpszDisplayName, NULL, &pidlTemp, NULL);
230                     IShellFolder_Release(unixFS);
231                 }
232             }
233             else
234             {
235                 /* build a complete path to create a simple pidl */
236                 WCHAR szPath[MAX_PATH];
237                 LPWSTR pathPtr;
238
239                 lstrcpynW(szPath, This->sPathTarget, MAX_PATH);
240                 pathPtr = PathAddBackslashW(szPath);
241                 if (pathPtr)
242                 {
243                     lstrcpynW(pathPtr, lpszDisplayName, MAX_PATH - (pathPtr - szPath));
244                     hr = _ILCreateFromPathW(szPath, &pidlTemp);
245                 }
246                 else
247                 {
248                     /* should never reach here, but for completeness */
249                     hr = HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER);
250                 }
251             }
252         }
253         else
254             pidlTemp = _ILCreateMyComputer();
255
256         szNext = NULL;
257     }
258
259     if (SUCCEEDED(hr) && pidlTemp)
260     {
261         if (szNext && *szNext)
262         {
263             hr = SHELL32_ParseNextElement(iface, hwndOwner, pbc,
264                     &pidlTemp, (LPOLESTR) szNext, pchEaten, pdwAttributes);
265         }
266         else
267         {
268             if (pdwAttributes && *pdwAttributes)
269                 hr = SHELL32_GetItemAttributes(shell_folder, pidlTemp, pdwAttributes);
270         }
271     }
272
273     *ppidl = pidlTemp;
274
275     TRACE ("(%p)->(-- ret=0x%08x)\n", This, hr);
276
277     return hr;
278 }
279
280 /**************************************************************************
281  *  CreateDesktopEnumList()
282  */
283 static const WCHAR Desktop_NameSpaceW[] = { 'S','O','F','T','W','A','R','E',
284  '\\','M','i','c','r','o','s','o','f','t','\\','W','i','n','d','o','w','s','\\',
285  'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\','E','x','p','l',
286  'o','r','e','r','\\','D','e','s','k','t','o','p','\\','N','a','m','e','s','p',
287  'a','c','e','\0' };
288
289 static BOOL CreateDesktopEnumList(IEnumIDListImpl *list, DWORD dwFlags)
290 {
291     BOOL ret = TRUE;
292     WCHAR szPath[MAX_PATH];
293
294     TRACE("(%p)->(flags=0x%08x)\n", list, dwFlags);
295
296     /* enumerate the root folders */
297     if (dwFlags & SHCONTF_FOLDERS)
298     {
299         HKEY hkey;
300         UINT i;
301
302         /* create the pidl for This item */
303         ret = AddToEnumList(list, _ILCreateMyComputer());
304
305         for (i=0; i<2; i++) {
306             if (ret && !RegOpenKeyExW(i == 0 ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER,
307                                       Desktop_NameSpaceW, 0, KEY_READ, &hkey))
308             {
309                 WCHAR iid[50];
310                 int i=0;
311
312                 while (ret)
313                 {
314                     DWORD size;
315                     LONG r;
316
317                     size = sizeof (iid) / sizeof (iid[0]);
318                     r = RegEnumKeyExW(hkey, i, iid, &size, 0, NULL, NULL, NULL);
319                     if (ERROR_SUCCESS == r)
320                     {
321                         ret = AddToEnumList(list, _ILCreateGuidFromStrW(iid));
322                         i++;
323                     }
324                     else if (ERROR_NO_MORE_ITEMS == r)
325                         break;
326                     else
327                         ret = FALSE;
328                 }
329                 RegCloseKey(hkey);
330             }
331         }
332     }
333
334     /* enumerate the elements in %windir%\desktop */
335     ret = ret && SHGetSpecialFolderPathW(0, szPath, CSIDL_DESKTOPDIRECTORY, FALSE);
336     ret = ret && CreateFolderEnumList(list, szPath, dwFlags);
337
338     return ret;
339 }
340
341 /**************************************************************************
342  *        ISF_Desktop_fnEnumObjects
343  */
344 static HRESULT WINAPI ISF_Desktop_fnEnumObjects (IShellFolder2 * iface,
345                 HWND hwndOwner, DWORD dwFlags, LPENUMIDLIST * ppEnumIDList)
346 {
347     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
348     IEnumIDListImpl *list;
349
350     TRACE ("(%p)->(HWND=%p flags=0x%08x pplist=%p)\n",
351            This, hwndOwner, dwFlags, ppEnumIDList);
352
353     if (!(list = IEnumIDList_Constructor()))
354         return E_OUTOFMEMORY;
355     CreateDesktopEnumList(list, dwFlags);
356     *ppEnumIDList = &list->IEnumIDList_iface;
357
358     TRACE ("-- (%p)->(new ID List: %p)\n", This, *ppEnumIDList);
359
360     return S_OK;
361 }
362
363 /**************************************************************************
364  *        ISF_Desktop_fnBindToObject
365  */
366 static HRESULT WINAPI ISF_Desktop_fnBindToObject (IShellFolder2 * iface,
367                 LPCITEMIDLIST pidl, LPBC pbcReserved, REFIID riid, LPVOID * ppvOut)
368 {
369     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
370
371     TRACE ("(%p)->(pidl=%p,%p,%s,%p)\n",
372            This, pidl, pbcReserved, shdebugstr_guid (riid), ppvOut);
373
374     return SHELL32_BindToChild( This->pidlRoot, This->sPathTarget, pidl, riid, ppvOut );
375 }
376
377 /**************************************************************************
378  *    ISF_Desktop_fnBindToStorage
379  */
380 static HRESULT WINAPI ISF_Desktop_fnBindToStorage (IShellFolder2 * iface,
381                 LPCITEMIDLIST pidl, LPBC pbcReserved, REFIID riid, LPVOID * ppvOut)
382 {
383     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
384
385     FIXME ("(%p)->(pidl=%p,%p,%s,%p) stub\n",
386            This, pidl, pbcReserved, shdebugstr_guid (riid), ppvOut);
387
388     *ppvOut = NULL;
389     return E_NOTIMPL;
390 }
391
392 /**************************************************************************
393  *     ISF_Desktop_fnCompareIDs
394  */
395 static HRESULT WINAPI ISF_Desktop_fnCompareIDs (IShellFolder2 *iface,
396                         LPARAM lParam, LPCITEMIDLIST pidl1, LPCITEMIDLIST pidl2)
397 {
398     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
399     IShellFolder *shell_folder = (IShellFolder*)iface;
400     HRESULT hr;
401
402     TRACE ("(%p)->(0x%08lx,pidl1=%p,pidl2=%p)\n", This, lParam, pidl1, pidl2);
403     hr = SHELL32_CompareIDs ( shell_folder, lParam, pidl1, pidl2);
404     TRACE ("-- 0x%08x\n", hr);
405     return hr;
406 }
407
408 /**************************************************************************
409  *    ISF_Desktop_fnCreateViewObject
410  */
411 static HRESULT WINAPI ISF_Desktop_fnCreateViewObject (IShellFolder2 * iface,
412                               HWND hwndOwner, REFIID riid, LPVOID * ppvOut)
413 {
414     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
415     LPSHELLVIEW pShellView;
416     HRESULT hr = E_INVALIDARG;
417
418     TRACE ("(%p)->(hwnd=%p,%s,%p)\n",
419            This, hwndOwner, shdebugstr_guid (riid), ppvOut);
420
421     if (!ppvOut)
422         return E_INVALIDARG;
423
424     *ppvOut = NULL;
425
426     if (IsEqualIID (riid, &IID_IDropTarget))
427     {
428         WARN ("IDropTarget not implemented\n");
429         hr = E_NOTIMPL;
430     }
431     else if (IsEqualIID (riid, &IID_IContextMenu))
432     {
433         WARN ("IContextMenu not implemented\n");
434         hr = E_NOTIMPL;
435     }
436     else if (IsEqualIID (riid, &IID_IShellView))
437     {
438         pShellView = IShellView_Constructor ((IShellFolder *) iface);
439         if (pShellView)
440         {
441             hr = IShellView_QueryInterface (pShellView, riid, ppvOut);
442             IShellView_Release (pShellView);
443         }
444     }
445     TRACE ("-- (%p)->(interface=%p)\n", This, ppvOut);
446     return hr;
447 }
448
449 /**************************************************************************
450  *  ISF_Desktop_fnGetAttributesOf
451  */
452 static HRESULT WINAPI ISF_Desktop_fnGetAttributesOf (IShellFolder2 * iface,
453                 UINT cidl, LPCITEMIDLIST * apidl, DWORD * rgfInOut)
454 {
455     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
456     IShellFolder *shell_folder = (IShellFolder*)iface;
457
458     static const DWORD dwDesktopAttributes = 
459         SFGAO_STORAGE | SFGAO_HASPROPSHEET | SFGAO_STORAGEANCESTOR |
460         SFGAO_FILESYSANCESTOR | SFGAO_FOLDER | SFGAO_FILESYSTEM | SFGAO_HASSUBFOLDER;
461     static const DWORD dwMyComputerAttributes = 
462         SFGAO_CANRENAME | SFGAO_CANDELETE | SFGAO_HASPROPSHEET |
463         SFGAO_DROPTARGET | SFGAO_FILESYSANCESTOR | SFGAO_FOLDER | SFGAO_HASSUBFOLDER;
464
465     TRACE ("(%p)->(cidl=%d apidl=%p mask=%p (0x%08x))\n",
466            This, cidl, apidl, rgfInOut, rgfInOut ? *rgfInOut : 0);
467
468     if (!rgfInOut)
469         return E_INVALIDARG;
470     if (cidl && !apidl)
471         return E_INVALIDARG;
472
473     if (*rgfInOut == 0)
474         *rgfInOut = ~0;
475     
476     if(cidl == 0) {
477         *rgfInOut &= dwDesktopAttributes; 
478     } else {
479         while (cidl > 0 && *apidl) {
480             pdump (*apidl);
481             if (_ILIsDesktop(*apidl)) { 
482                 *rgfInOut &= dwDesktopAttributes;
483             } else if (_ILIsMyComputer(*apidl)) {
484                 *rgfInOut &= dwMyComputerAttributes;
485             } else {
486                 SHELL32_GetItemAttributes ( shell_folder, *apidl, rgfInOut);
487             }
488             apidl++;
489             cidl--;
490         }
491     }
492     /* make sure SFGAO_VALIDATE is cleared, some apps depend on that */
493     *rgfInOut &= ~SFGAO_VALIDATE;
494
495     TRACE ("-- result=0x%08x\n", *rgfInOut);
496
497     return S_OK;
498 }
499
500 /**************************************************************************
501  *    ISF_Desktop_fnGetUIObjectOf
502  *
503  * PARAMETERS
504  *  HWND           hwndOwner, //[in ] Parent window for any output
505  *  UINT           cidl,      //[in ] array size
506  *  LPCITEMIDLIST* apidl,     //[in ] simple pidl array
507  *  REFIID         riid,      //[in ] Requested Interface
508  *  UINT*          prgfInOut, //[   ] reserved
509  *  LPVOID*        ppvObject) //[out] Resulting Interface
510  *
511  */
512 static HRESULT WINAPI ISF_Desktop_fnGetUIObjectOf (IShellFolder2 * iface,
513                 HWND hwndOwner, UINT cidl, LPCITEMIDLIST * apidl,
514                 REFIID riid, UINT * prgfInOut, LPVOID * ppvOut)
515 {
516     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
517
518     LPITEMIDLIST pidl;
519     IUnknown *pObj = NULL;
520     HRESULT hr = E_INVALIDARG;
521
522     TRACE ("(%p)->(%p,%u,apidl=%p,%s,%p,%p)\n",
523        This, hwndOwner, cidl, apidl, shdebugstr_guid (riid), prgfInOut, ppvOut);
524
525     if (!ppvOut)
526         return E_INVALIDARG;
527
528     *ppvOut = NULL;
529
530     if (IsEqualIID (riid, &IID_IContextMenu))
531     {
532         if (cidl > 0)
533             return ItemMenu_Constructor((IShellFolder*)iface, This->pidlRoot, apidl, cidl, riid, ppvOut);
534         else
535             return BackgroundMenu_Constructor((IShellFolder*)iface, TRUE, riid, ppvOut);
536     }
537     else if (IsEqualIID (riid, &IID_IDataObject) && (cidl >= 1))
538     {
539         pObj = (LPUNKNOWN) IDataObject_Constructor( hwndOwner,
540                                                   This->pidlRoot, apidl, cidl);
541         hr = S_OK;
542     }
543     else if (IsEqualIID (riid, &IID_IExtractIconA) && (cidl == 1))
544     {
545         pidl = ILCombine (This->pidlRoot, apidl[0]);
546         pObj = (LPUNKNOWN) IExtractIconA_Constructor (pidl);
547         SHFree (pidl);
548         hr = S_OK;
549     }
550     else if (IsEqualIID (riid, &IID_IExtractIconW) && (cidl == 1))
551     {
552         pidl = ILCombine (This->pidlRoot, apidl[0]);
553         pObj = (LPUNKNOWN) IExtractIconW_Constructor (pidl);
554         SHFree (pidl);
555         hr = S_OK;
556     }
557     else if (IsEqualIID (riid, &IID_IDropTarget) && (cidl >= 1))
558     {
559         hr = IShellFolder2_QueryInterface (iface,
560                                           &IID_IDropTarget, (LPVOID *) & pObj);
561     }
562     else if ((IsEqualIID(riid,&IID_IShellLinkW) ||
563               IsEqualIID(riid,&IID_IShellLinkA)) && (cidl == 1))
564     {
565         pidl = ILCombine (This->pidlRoot, apidl[0]);
566         hr = IShellLink_ConstructFromFile(NULL, riid, pidl, (LPVOID*)&pObj);
567         SHFree (pidl);
568     }
569     else
570         hr = E_NOINTERFACE;
571
572     if (SUCCEEDED(hr) && !pObj)
573         hr = E_OUTOFMEMORY;
574
575     *ppvOut = pObj;
576     TRACE ("(%p)->hr=0x%08x\n", This, hr);
577     return hr;
578 }
579
580 /**************************************************************************
581  *    ISF_Desktop_fnGetDisplayNameOf
582  *
583  * NOTES
584  *    special case: pidl = null gives desktop-name back
585  */
586 static HRESULT WINAPI ISF_Desktop_fnGetDisplayNameOf (IShellFolder2 * iface,
587                 LPCITEMIDLIST pidl, DWORD dwFlags, LPSTRRET strRet)
588 {
589     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
590     HRESULT hr = S_OK;
591     LPWSTR pszPath;
592
593     TRACE ("(%p)->(pidl=%p,0x%08x,%p)\n", This, pidl, dwFlags, strRet);
594     pdump (pidl);
595
596     if (!strRet)
597         return E_INVALIDARG;
598
599     pszPath = CoTaskMemAlloc((MAX_PATH +1) * sizeof(WCHAR));
600     if (!pszPath)
601         return E_OUTOFMEMORY;
602
603     if (_ILIsDesktop (pidl))
604     {
605         if ((GET_SHGDN_RELATION (dwFlags) == SHGDN_NORMAL) &&
606             (GET_SHGDN_FOR (dwFlags) & SHGDN_FORPARSING))
607             strcpyW(pszPath, This->sPathTarget);
608         else
609             HCR_GetClassNameW(&CLSID_ShellDesktop, pszPath, MAX_PATH);
610     }
611     else if (_ILIsPidlSimple (pidl))
612     {
613         GUID const *clsid;
614
615         if ((clsid = _ILGetGUIDPointer (pidl)))
616         {
617             if (GET_SHGDN_FOR (dwFlags) & SHGDN_FORPARSING)
618             {
619                 int bWantsForParsing;
620
621                 /*
622                  * We can only get a filesystem path from a shellfolder if the
623                  *  value WantsFORPARSING in CLSID\\{...}\\shellfolder exists.
624                  *
625                  * Exception: The MyComputer folder doesn't have this key,
626                  *   but any other filesystem backed folder it needs it.
627                  */
628                 if (IsEqualIID (clsid, &CLSID_MyComputer))
629                 {
630                     bWantsForParsing = TRUE;
631                 }
632                 else
633                 {
634                     /* get the "WantsFORPARSING" flag from the registry */
635                     static const WCHAR clsidW[] =
636                      { 'C','L','S','I','D','\\',0 };
637                     static const WCHAR shellfolderW[] =
638                      { '\\','s','h','e','l','l','f','o','l','d','e','r',0 };
639                     static const WCHAR wantsForParsingW[] =
640                      { 'W','a','n','t','s','F','o','r','P','a','r','s','i','n',
641                      'g',0 };
642                     WCHAR szRegPath[100];
643                     LONG r;
644
645                     lstrcpyW (szRegPath, clsidW);
646                     SHELL32_GUIDToStringW (clsid, &szRegPath[6]);
647                     lstrcatW (szRegPath, shellfolderW);
648                     r = SHGetValueW(HKEY_CLASSES_ROOT, szRegPath,
649                                     wantsForParsingW, NULL, NULL, NULL);
650                     if (r == ERROR_SUCCESS)
651                         bWantsForParsing = TRUE;
652                     else
653                         bWantsForParsing = FALSE;
654                 }
655
656                 if ((GET_SHGDN_RELATION (dwFlags) == SHGDN_NORMAL) &&
657                      bWantsForParsing)
658                 {
659                     /*
660                      * we need the filesystem path to the destination folder.
661                      * Only the folder itself can know it
662                      */
663                     hr = SHELL32_GetDisplayNameOfChild (iface, pidl, dwFlags,
664                                                         pszPath,
665                                                         MAX_PATH);
666                 }
667                 else
668                 {
669                     /* parsing name like ::{...} */
670                     pszPath[0] = ':';
671                     pszPath[1] = ':';
672                     SHELL32_GUIDToStringW (clsid, &pszPath[2]);
673                 }
674             }
675             else
676             {
677                 /* user friendly name */
678                 HCR_GetClassNameW (clsid, pszPath, MAX_PATH);
679             }
680         }
681         else
682         {
683             int cLen = 0;
684
685             /* file system folder or file rooted at the desktop */
686             if ((GET_SHGDN_FOR(dwFlags) == SHGDN_FORPARSING) &&
687                 (GET_SHGDN_RELATION(dwFlags) != SHGDN_INFOLDER))
688             {
689                 lstrcpynW(pszPath, This->sPathTarget, MAX_PATH - 1);
690                 PathAddBackslashW(pszPath);
691                 cLen = lstrlenW(pszPath);
692             }
693
694             _ILSimpleGetTextW(pidl, pszPath + cLen, MAX_PATH - cLen);
695
696             if (!_ILIsFolder(pidl))
697                 SHELL_FS_ProcessDisplayFilename(pszPath, dwFlags);
698         }
699     }
700     else
701     {
702         /* a complex pidl, let the subfolder do the work */
703         hr = SHELL32_GetDisplayNameOfChild (iface, pidl, dwFlags,
704                                             pszPath, MAX_PATH);
705     }
706
707     if (SUCCEEDED(hr))
708     {
709         /* Win9x always returns ANSI strings, NT always returns Unicode strings */
710         if (GetVersion() & 0x80000000)
711         {
712             strRet->uType = STRRET_CSTR;
713             if (!WideCharToMultiByte(CP_ACP, 0, pszPath, -1, strRet->u.cStr, MAX_PATH,
714                                      NULL, NULL))
715                 strRet->u.cStr[0] = '\0';
716             CoTaskMemFree(pszPath);
717         }
718         else
719         {
720             strRet->uType = STRRET_WSTR;
721             strRet->u.pOleStr = pszPath;
722         }
723     }
724     else
725         CoTaskMemFree(pszPath);
726
727     TRACE ("-- (%p)->(%s,0x%08x)\n", This,
728     strRet->uType == STRRET_CSTR ? strRet->u.cStr :
729     debugstr_w(strRet->u.pOleStr), hr);
730     return hr;
731 }
732
733 /**************************************************************************
734  *  ISF_Desktop_fnSetNameOf
735  *  Changes the name of a file object or subfolder, possibly changing its item
736  *  identifier in the process.
737  *
738  * PARAMETERS
739  *  HWND          hwndOwner,  //[in ] Owner window for output
740  *  LPCITEMIDLIST pidl,       //[in ] simple pidl of item to change
741  *  LPCOLESTR     lpszName,   //[in ] the items new display name
742  *  DWORD         dwFlags,    //[in ] SHGNO formatting flags
743  *  LPITEMIDLIST* ppidlOut)   //[out] simple pidl returned
744  */
745 static HRESULT WINAPI ISF_Desktop_fnSetNameOf (IShellFolder2 * iface,
746                 HWND hwndOwner, LPCITEMIDLIST pidl,    /* simple pidl */
747                 LPCOLESTR lpName, DWORD dwFlags, LPITEMIDLIST * pPidlOut)
748 {
749     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
750
751     FIXME ("(%p)->(%p,pidl=%p,%s,%u,%p) stub\n", This, hwndOwner, pidl,
752            debugstr_w (lpName), dwFlags, pPidlOut);
753
754     return E_FAIL;
755 }
756
757 static HRESULT WINAPI ISF_Desktop_fnGetDefaultSearchGUID(IShellFolder2 *iface,
758                 GUID * pguid)
759 {
760     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
761     FIXME ("(%p)->(%p) stub\n", This, pguid);
762     return E_NOTIMPL;
763 }
764
765 static HRESULT WINAPI ISF_Desktop_fnEnumSearches (IShellFolder2 *iface,
766                 IEnumExtraSearch ** ppenum)
767 {
768     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
769     FIXME ("(%p)->(%p) stub\n", This, ppenum);
770     return E_NOTIMPL;
771 }
772
773 static HRESULT WINAPI ISF_Desktop_fnGetDefaultColumn (IShellFolder2 * iface,
774                 DWORD reserved, ULONG * pSort, ULONG * pDisplay)
775 {
776     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
777
778     TRACE ("(%p)->(%d %p %p)\n", This, reserved, pSort, pDisplay);
779
780     if (pSort)
781         *pSort = 0;
782     if (pDisplay)
783         *pDisplay = 0;
784
785     return S_OK;
786 }
787 static HRESULT WINAPI ISF_Desktop_fnGetDefaultColumnState (
788                 IShellFolder2 * iface, UINT iColumn, DWORD * pcsFlags)
789 {
790     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
791
792     TRACE ("(%p)->(%d %p)\n", This, iColumn, pcsFlags);
793
794     if (!pcsFlags || iColumn >= DESKTOPSHELLVIEWCOLUMNS)
795     return E_INVALIDARG;
796
797     *pcsFlags = desktop_header[iColumn].pcsFlags;
798
799     return S_OK;
800 }
801
802 static HRESULT WINAPI ISF_Desktop_fnGetDetailsEx (IShellFolder2 * iface,
803                 LPCITEMIDLIST pidl, const SHCOLUMNID * pscid, VARIANT * pv)
804 {
805     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
806     FIXME ("(%p)->(%p %p %p) stub\n", This, pidl, pscid, pv);
807     return E_NOTIMPL;
808 }
809
810 static HRESULT WINAPI ISF_Desktop_fnGetDetailsOf (IShellFolder2 * iface,
811                 LPCITEMIDLIST pidl, UINT iColumn, SHELLDETAILS * psd)
812 {
813     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
814
815     HRESULT hr = S_OK;
816
817     TRACE ("(%p)->(%p %i %p)\n", This, pidl, iColumn, psd);
818
819     if (!psd || iColumn >= DESKTOPSHELLVIEWCOLUMNS)
820         return E_INVALIDARG;
821
822     if (!pidl)
823         return SHELL32_GetColumnDetails(desktop_header, iColumn, psd);
824
825     /* the data from the pidl */
826     psd->str.uType = STRRET_CSTR;
827     switch (iColumn)
828     {
829     case 0:        /* name */
830         hr = IShellFolder2_GetDisplayNameOf(iface, pidl,
831                    SHGDN_NORMAL | SHGDN_INFOLDER, &psd->str);
832         break;
833     case 1:        /* size */
834         _ILGetFileSize (pidl, psd->str.u.cStr, MAX_PATH);
835         break;
836     case 2:        /* type */
837         _ILGetFileType (pidl, psd->str.u.cStr, MAX_PATH);
838         break;
839     case 3:        /* date */
840         _ILGetFileDate (pidl, psd->str.u.cStr, MAX_PATH);
841         break;
842     case 4:        /* attributes */
843         _ILGetFileAttributes (pidl, psd->str.u.cStr, MAX_PATH);
844         break;
845     }
846
847     return hr;
848 }
849
850 static HRESULT WINAPI ISF_Desktop_fnMapColumnToSCID (
851                 IShellFolder2 * iface, UINT column, SHCOLUMNID * pscid)
852 {
853     IDesktopFolderImpl *This = impl_from_IShellFolder2(iface);
854     FIXME ("(%p)->(%d %p) stub\n", This, column, pscid);
855     return E_NOTIMPL;
856 }
857
858 static const IShellFolder2Vtbl vt_MCFldr_ShellFolder2 =
859 {
860     ISF_Desktop_fnQueryInterface,
861     ISF_Desktop_fnAddRef,
862     ISF_Desktop_fnRelease,
863     ISF_Desktop_fnParseDisplayName,
864     ISF_Desktop_fnEnumObjects,
865     ISF_Desktop_fnBindToObject,
866     ISF_Desktop_fnBindToStorage,
867     ISF_Desktop_fnCompareIDs,
868     ISF_Desktop_fnCreateViewObject,
869     ISF_Desktop_fnGetAttributesOf,
870     ISF_Desktop_fnGetUIObjectOf,
871     ISF_Desktop_fnGetDisplayNameOf,
872     ISF_Desktop_fnSetNameOf,
873     /* ShellFolder2 */
874     ISF_Desktop_fnGetDefaultSearchGUID,
875     ISF_Desktop_fnEnumSearches,
876     ISF_Desktop_fnGetDefaultColumn,
877     ISF_Desktop_fnGetDefaultColumnState,
878     ISF_Desktop_fnGetDetailsEx,
879     ISF_Desktop_fnGetDetailsOf,
880     ISF_Desktop_fnMapColumnToSCID
881 };
882
883 /**************************************************************************
884  *    IPersist
885  */
886 static HRESULT WINAPI ISF_Desktop_IPersistFolder2_fnQueryInterface(
887     IPersistFolder2 *iface, REFIID riid, LPVOID *ppvObj)
888 {
889     IDesktopFolderImpl *This = impl_from_IPersistFolder2( iface );
890     return IShellFolder2_QueryInterface(&This->IShellFolder2_iface, riid, ppvObj);
891 }
892
893 static ULONG WINAPI ISF_Desktop_IPersistFolder2_fnAddRef(
894     IPersistFolder2 *iface)
895 {
896     IDesktopFolderImpl *This = impl_from_IPersistFolder2( iface );
897     return IShellFolder2_AddRef(&This->IShellFolder2_iface);
898 }
899
900 static ULONG WINAPI ISF_Desktop_IPersistFolder2_fnRelease(
901     IPersistFolder2 *iface)
902 {
903     IDesktopFolderImpl *This = impl_from_IPersistFolder2( iface );
904     return IShellFolder2_Release(&This->IShellFolder2_iface);
905 }
906
907 static HRESULT WINAPI ISF_Desktop_IPersistFolder2_fnGetClassID(
908     IPersistFolder2 *iface, CLSID *clsid)
909 {
910     *clsid = CLSID_ShellDesktop;
911     return S_OK;
912 }
913 static HRESULT WINAPI ISF_Desktop_IPersistFolder2_fnInitialize(
914     IPersistFolder2 *iface, LPCITEMIDLIST pidl)
915 {
916     IDesktopFolderImpl *This = impl_from_IPersistFolder2( iface );
917     FIXME ("(%p)->(%p) stub\n", This, pidl);
918     return E_NOTIMPL;
919 }
920 static HRESULT WINAPI ISF_Desktop_IPersistFolder2_fnGetCurFolder(
921     IPersistFolder2 *iface, LPITEMIDLIST *ppidl)
922 {
923     IDesktopFolderImpl *This = impl_from_IPersistFolder2( iface );
924     *ppidl = ILClone(This->pidlRoot);
925     return S_OK;
926 }
927
928 static const IPersistFolder2Vtbl vt_IPersistFolder2 =
929 {
930     ISF_Desktop_IPersistFolder2_fnQueryInterface,
931     ISF_Desktop_IPersistFolder2_fnAddRef,
932     ISF_Desktop_IPersistFolder2_fnRelease,
933     ISF_Desktop_IPersistFolder2_fnGetClassID,
934     ISF_Desktop_IPersistFolder2_fnInitialize,
935     ISF_Desktop_IPersistFolder2_fnGetCurFolder
936 };
937
938 /**************************************************************************
939  *    ISF_Desktop_Constructor
940  */
941 HRESULT WINAPI ISF_Desktop_Constructor (
942                 IUnknown * pUnkOuter, REFIID riid, LPVOID * ppv)
943 {
944     static IDesktopFolderImpl *cached_sf;
945     WCHAR szMyPath[MAX_PATH];
946
947     TRACE ("unkOut=%p %s\n", pUnkOuter, shdebugstr_guid (riid));
948
949     if (!ppv)
950         return E_POINTER;
951     if (pUnkOuter)
952         return CLASS_E_NOAGGREGATION;
953
954     if (!cached_sf)
955     {
956         IDesktopFolderImpl *sf;
957
958         if (!SHGetSpecialFolderPathW( 0, szMyPath, CSIDL_DESKTOPDIRECTORY, TRUE ))
959             return E_UNEXPECTED;
960
961         sf = LocalAlloc( LMEM_ZEROINIT, sizeof (IDesktopFolderImpl) );
962         if (!sf)
963             return E_OUTOFMEMORY;
964
965         sf->ref = 1;
966         sf->IShellFolder2_iface.lpVtbl = &vt_MCFldr_ShellFolder2;
967         sf->IPersistFolder2_iface.lpVtbl = &vt_IPersistFolder2;
968         sf->pidlRoot = _ILCreateDesktop();    /* my qualified pidl */
969         sf->sPathTarget = SHAlloc( (lstrlenW(szMyPath) + 1)*sizeof(WCHAR) );
970         lstrcpyW( sf->sPathTarget, szMyPath );
971
972         if (InterlockedCompareExchangePointer((void *)&cached_sf, sf, NULL) != NULL)
973         {
974             /* some other thread already been here */
975             SHFree( sf->pidlRoot );
976             SHFree( sf->sPathTarget );
977             LocalFree( sf );
978         }
979     }
980
981     return IShellFolder2_QueryInterface( &cached_sf->IShellFolder2_iface, riid, ppv );
982 }