msvcrt: Mark some fall-throughs in a switch statement.
[wine] / dlls / shell32 / shellitem.c
1 /*
2  * IShellItem and IShellItemArray implementations
3  *
4  * Copyright 2008 Vincent Povirk for CodeWeavers
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #include "config.h"
22 #include "wine/port.h"
23
24 #include <stdio.h>
25 #include <stdarg.h>
26
27 #define COBJMACROS
28 #define NONAMELESSUNION
29 #define NONAMELESSSTRUCT
30
31 #include "windef.h"
32 #include "winbase.h"
33 #include "wine/debug.h"
34
35 #include "pidl.h"
36 #include "shell32_main.h"
37 #include "debughlp.h"
38
39 WINE_DEFAULT_DEBUG_CHANNEL(shell);
40
41 typedef struct _ShellItem {
42     IShellItem2             IShellItem2_iface;
43     LONG                    ref;
44     LPITEMIDLIST            pidl;
45     IPersistIDList          IPersistIDList_iface;
46 } ShellItem;
47
48 static inline ShellItem *impl_from_IShellItem2(IShellItem2 *iface)
49 {
50     return CONTAINING_RECORD(iface, ShellItem, IShellItem2_iface);
51 }
52
53
54 static inline ShellItem *impl_from_IPersistIDList( IPersistIDList *iface )
55 {
56     return CONTAINING_RECORD(iface, ShellItem, IPersistIDList_iface);
57 }
58
59
60 static HRESULT WINAPI ShellItem_QueryInterface(IShellItem2 *iface, REFIID riid,
61     void **ppv)
62 {
63     ShellItem *This = impl_from_IShellItem2(iface);
64
65     TRACE("(%p,%p,%p)\n", iface, riid, ppv);
66
67     if (!ppv) return E_INVALIDARG;
68
69     if (IsEqualIID(&IID_IUnknown, riid) || IsEqualIID(&IID_IShellItem, riid) ||
70         IsEqualIID(&IID_IShellItem2, riid))
71     {
72         *ppv = This;
73     }
74     else if (IsEqualIID(&IID_IPersist, riid) || IsEqualIID(&IID_IPersistIDList, riid))
75     {
76         *ppv = &This->IPersistIDList_iface;
77     }
78     else {
79         FIXME("not implemented for %s\n", shdebugstr_guid(riid));
80         *ppv = NULL;
81         return E_NOINTERFACE;
82     }
83
84     IUnknown_AddRef((IUnknown*)*ppv);
85     return S_OK;
86 }
87
88 static ULONG WINAPI ShellItem_AddRef(IShellItem2 *iface)
89 {
90     ShellItem *This = impl_from_IShellItem2(iface);
91     ULONG ref = InterlockedIncrement(&This->ref);
92
93     TRACE("(%p), new refcount=%i\n", iface, ref);
94
95     return ref;
96 }
97
98 static ULONG WINAPI ShellItem_Release(IShellItem2 *iface)
99 {
100     ShellItem *This = impl_from_IShellItem2(iface);
101     ULONG ref = InterlockedDecrement(&This->ref);
102
103     TRACE("(%p), new refcount=%i\n", iface, ref);
104
105     if (ref == 0)
106     {
107         ILFree(This->pidl);
108         HeapFree(GetProcessHeap(), 0, This);
109     }
110
111     return ref;
112 }
113
114 static HRESULT ShellItem_get_parent_pidl(ShellItem *This, LPITEMIDLIST *parent_pidl)
115 {
116     *parent_pidl = ILClone(This->pidl);
117     if (*parent_pidl)
118     {
119         if (ILRemoveLastID(*parent_pidl))
120             return S_OK;
121         else
122         {
123             ILFree(*parent_pidl);
124             *parent_pidl = NULL;
125             return E_INVALIDARG;
126         }
127     }
128     else
129     {
130         *parent_pidl = NULL;
131         return E_OUTOFMEMORY;
132     }
133 }
134
135 static HRESULT ShellItem_get_parent_shellfolder(ShellItem *This, IShellFolder **ppsf)
136 {
137     LPITEMIDLIST parent_pidl;
138     IShellFolder *desktop;
139     HRESULT ret;
140
141     ret = ShellItem_get_parent_pidl(This, &parent_pidl);
142     if (SUCCEEDED(ret))
143     {
144         ret = SHGetDesktopFolder(&desktop);
145         if (SUCCEEDED(ret))
146         {
147             ret = IShellFolder_BindToObject(desktop, parent_pidl, NULL, &IID_IShellFolder, (void**)ppsf);
148             IShellFolder_Release(desktop);
149         }
150         ILFree(parent_pidl);
151     }
152
153     return ret;
154 }
155
156 static HRESULT ShellItem_get_shellfolder(ShellItem *This, IBindCtx *pbc, IShellFolder **ppsf)
157 {
158     IShellFolder *desktop;
159     HRESULT ret;
160
161     ret = SHGetDesktopFolder(&desktop);
162     if (SUCCEEDED(ret))
163     {
164         if (_ILIsDesktop(This->pidl))
165         {
166             *ppsf = desktop;
167             IShellFolder_AddRef(*ppsf);
168         }
169         else
170         {
171             ret = IShellFolder_BindToObject(desktop, This->pidl, pbc, &IID_IShellFolder, (void**)ppsf);
172         }
173
174         IShellFolder_Release(desktop);
175     }
176
177     return ret;
178 }
179
180 static HRESULT WINAPI ShellItem_BindToHandler(IShellItem2 *iface, IBindCtx *pbc,
181     REFGUID rbhid, REFIID riid, void **ppvOut)
182 {
183     ShellItem *This = impl_from_IShellItem2(iface);
184     HRESULT ret;
185     TRACE("(%p,%p,%s,%p,%p)\n", iface, pbc, shdebugstr_guid(rbhid), riid, ppvOut);
186
187     *ppvOut = NULL;
188     if (IsEqualGUID(rbhid, &BHID_SFObject))
189     {
190         IShellFolder *psf;
191         ret = ShellItem_get_shellfolder(This, pbc, &psf);
192         if (SUCCEEDED(ret))
193         {
194             ret = IShellFolder_QueryInterface(psf, riid, ppvOut);
195             IShellFolder_Release(psf);
196         }
197         return ret;
198     }
199     else if (IsEqualGUID(rbhid, &BHID_SFUIObject))
200     {
201         IShellFolder *psf_parent;
202         if (_ILIsDesktop(This->pidl))
203             ret = SHGetDesktopFolder(&psf_parent);
204         else
205             ret = ShellItem_get_parent_shellfolder(This, &psf_parent);
206
207         if (SUCCEEDED(ret))
208         {
209             LPCITEMIDLIST pidl = ILFindLastID(This->pidl);
210             ret = IShellFolder_GetUIObjectOf(psf_parent, NULL, 1, &pidl, riid, NULL, ppvOut);
211             IShellFolder_Release(psf_parent);
212         }
213         return ret;
214     }
215     else if (IsEqualGUID(rbhid, &BHID_DataObject))
216     {
217         return ShellItem_BindToHandler(&This->IShellItem2_iface, pbc, &BHID_SFUIObject,
218                                        &IID_IDataObject, ppvOut);
219     }
220
221     FIXME("Unsupported BHID %s.\n", debugstr_guid(rbhid));
222
223     return MK_E_NOOBJECT;
224 }
225
226 static HRESULT WINAPI ShellItem_GetParent(IShellItem2 *iface, IShellItem **ppsi)
227 {
228     ShellItem *This = impl_from_IShellItem2(iface);
229     LPITEMIDLIST parent_pidl;
230     HRESULT ret;
231
232     TRACE("(%p,%p)\n", iface, ppsi);
233
234     ret = ShellItem_get_parent_pidl(This, &parent_pidl);
235     if (SUCCEEDED(ret))
236     {
237         ret = SHCreateShellItem(NULL, NULL, parent_pidl, ppsi);
238         ILFree(parent_pidl);
239     }
240
241     return ret;
242 }
243
244 static HRESULT WINAPI ShellItem_GetDisplayName(IShellItem2 *iface, SIGDN sigdnName,
245     LPWSTR *ppszName)
246 {
247     ShellItem *This = impl_from_IShellItem2(iface);
248     TRACE("(%p,%x,%p)\n", iface, sigdnName, ppszName);
249
250     return SHGetNameFromIDList(This->pidl, sigdnName, ppszName);
251 }
252
253 static HRESULT WINAPI ShellItem_GetAttributes(IShellItem2 *iface, SFGAOF sfgaoMask,
254     SFGAOF *psfgaoAttribs)
255 {
256     ShellItem *This = impl_from_IShellItem2(iface);
257     IShellFolder *parent_folder;
258     LPITEMIDLIST child_pidl;
259     HRESULT ret;
260
261     TRACE("(%p,%x,%p)\n", iface, sfgaoMask, psfgaoAttribs);
262
263     if (_ILIsDesktop(This->pidl))
264         ret = SHGetDesktopFolder(&parent_folder);
265     else
266         ret = ShellItem_get_parent_shellfolder(This, &parent_folder);
267     if (SUCCEEDED(ret))
268     {
269         child_pidl = ILFindLastID(This->pidl);
270         *psfgaoAttribs = sfgaoMask;
271         ret = IShellFolder_GetAttributesOf(parent_folder, 1, (LPCITEMIDLIST*)&child_pidl, psfgaoAttribs);
272         IShellFolder_Release(parent_folder);
273     }
274
275     return ret;
276 }
277
278 static HRESULT WINAPI ShellItem_Compare(IShellItem2 *iface, IShellItem *oth,
279     SICHINTF hint, int *piOrder)
280 {
281     LPWSTR dispname, dispname_oth;
282     HRESULT ret;
283     TRACE("(%p,%p,%x,%p)\n", iface, oth, hint, piOrder);
284
285     if(hint & (SICHINT_CANONICAL | SICHINT_ALLFIELDS))
286         FIXME("Unsupported flags 0x%08x\n", hint);
287
288     ret = IShellItem_GetDisplayName(iface, SIGDN_DESKTOPABSOLUTEEDITING, &dispname);
289     if(SUCCEEDED(ret))
290     {
291         ret = IShellItem_GetDisplayName(oth, SIGDN_DESKTOPABSOLUTEEDITING, &dispname_oth);
292         if(SUCCEEDED(ret))
293         {
294             *piOrder = lstrcmpiW(dispname, dispname_oth);
295             CoTaskMemFree(dispname_oth);
296         }
297         CoTaskMemFree(dispname);
298     }
299
300     if(SUCCEEDED(ret) && *piOrder &&
301        (hint & SICHINT_TEST_FILESYSPATH_IF_NOT_EQUAL))
302     {
303         LPWSTR dispname, dispname_oth;
304
305         TRACE("Testing filesystem path.\n");
306         ret = IShellItem_GetDisplayName(iface, SIGDN_FILESYSPATH, &dispname);
307         if(SUCCEEDED(ret))
308         {
309             ret = IShellItem_GetDisplayName(oth, SIGDN_FILESYSPATH, &dispname_oth);
310             if(SUCCEEDED(ret))
311             {
312                 *piOrder = lstrcmpiW(dispname, dispname_oth);
313                 CoTaskMemFree(dispname_oth);
314             }
315             CoTaskMemFree(dispname);
316         }
317     }
318
319     if(FAILED(ret))
320         return ret;
321
322     if(*piOrder)
323         return S_FALSE;
324     else
325         return S_OK;
326 }
327
328 static HRESULT WINAPI ShellItem2_GetPropertyStore(IShellItem2 *iface, GETPROPERTYSTOREFLAGS flags,
329     REFIID riid, void **ppv)
330 {
331     ShellItem *This = impl_from_IShellItem2(iface);
332     FIXME("Stub: %p (%d, %s, %p)\n", This, flags, shdebugstr_guid(riid), ppv);
333     return E_NOTIMPL;
334 }
335
336 static HRESULT WINAPI ShellItem2_GetPropertyStoreWithCreateObject(IShellItem2 *iface,
337     GETPROPERTYSTOREFLAGS flags, IUnknown *punkCreateObject, REFIID riid, void **ppv)
338 {
339     ShellItem *This = impl_from_IShellItem2(iface);
340     FIXME("Stub: %p (%08x, %p, %s, %p)\n",
341           This, flags, punkCreateObject, shdebugstr_guid(riid), ppv);
342     return E_NOTIMPL;
343 }
344
345 static HRESULT WINAPI ShellItem2_GetPropertyStoreForKeys(IShellItem2 *iface, const PROPERTYKEY *rgKeys,
346     UINT cKeys, GETPROPERTYSTOREFLAGS flags, REFIID riid, void **ppv)
347 {
348     ShellItem *This = impl_from_IShellItem2(iface);
349     FIXME("Stub: %p (%p, %d, %08x, %s, %p)\n",
350           This, rgKeys, cKeys, flags, shdebugstr_guid(riid), ppv);
351     return E_NOTIMPL;
352 }
353
354 static HRESULT WINAPI ShellItem2_GetPropertyDescriptionList(IShellItem2 *iface,
355     REFPROPERTYKEY keyType, REFIID riid, void **ppv)
356 {
357     ShellItem *This = impl_from_IShellItem2(iface);
358     FIXME("Stub: %p (%p, %s, %p)\n", This, keyType, debugstr_guid(riid), ppv);
359     return E_NOTIMPL;
360 }
361
362 static HRESULT WINAPI ShellItem2_Update(IShellItem2 *iface, IBindCtx *pbc)
363 {
364     ShellItem *This = impl_from_IShellItem2(iface);
365     FIXME("Stub: %p (%p)\n", This, pbc);
366     return E_NOTIMPL;
367 }
368
369 static HRESULT WINAPI ShellItem2_GetProperty(IShellItem2 *iface, REFPROPERTYKEY key, PROPVARIANT *ppropvar)
370 {
371     ShellItem *This = impl_from_IShellItem2(iface);
372     FIXME("Stub: %p (%p, %p)\n", This, key, ppropvar);
373     return E_NOTIMPL;
374 }
375
376 static HRESULT WINAPI ShellItem2_GetCLSID(IShellItem2 *iface, REFPROPERTYKEY key, CLSID *pclsid)
377 {
378     ShellItem *This = impl_from_IShellItem2(iface);
379     FIXME("Stub: %p (%p, %p)\n", This, key, pclsid);
380     return E_NOTIMPL;
381 }
382
383 static HRESULT WINAPI ShellItem2_GetFileTime(IShellItem2 *iface, REFPROPERTYKEY key, FILETIME *pft)
384 {
385     ShellItem *This = impl_from_IShellItem2(iface);
386     FIXME("Stub: %p (%p, %p)\n", This, key, pft);
387     return E_NOTIMPL;
388 }
389
390 static HRESULT WINAPI ShellItem2_GetInt32(IShellItem2 *iface, REFPROPERTYKEY key, int *pi)
391 {
392     ShellItem *This = impl_from_IShellItem2(iface);
393     FIXME("Stub: %p (%p, %p)\n", This, key, pi);
394     return E_NOTIMPL;
395 }
396
397 static HRESULT WINAPI ShellItem2_GetString(IShellItem2 *iface, REFPROPERTYKEY key, LPWSTR *ppsz)
398 {
399     ShellItem *This = impl_from_IShellItem2(iface);
400     FIXME("Stub: %p (%p, %p)\n", This, key, ppsz);
401     return E_NOTIMPL;
402 }
403
404 static HRESULT WINAPI ShellItem2_GetUInt32(IShellItem2 *iface, REFPROPERTYKEY key, ULONG *pui)
405 {
406     ShellItem *This = impl_from_IShellItem2(iface);
407     FIXME("Stub: %p (%p, %p)\n", This, key, pui);
408     return E_NOTIMPL;
409 }
410
411 static HRESULT WINAPI ShellItem2_GetUInt64(IShellItem2 *iface, REFPROPERTYKEY key, ULONGLONG *pull)
412 {
413     ShellItem *This = impl_from_IShellItem2(iface);
414     FIXME("Stub: %p (%p, %p)\n", This, key, pull);
415     return E_NOTIMPL;
416 }
417
418 static HRESULT WINAPI ShellItem2_GetBool(IShellItem2 *iface, REFPROPERTYKEY key, BOOL *pf)
419 {
420     ShellItem *This = impl_from_IShellItem2(iface);
421     FIXME("Stub: %p (%p, %p)\n", This, key, pf);
422     return E_NOTIMPL;
423 }
424
425
426 static const IShellItem2Vtbl ShellItem2_Vtbl = {
427     ShellItem_QueryInterface,
428     ShellItem_AddRef,
429     ShellItem_Release,
430     ShellItem_BindToHandler,
431     ShellItem_GetParent,
432     ShellItem_GetDisplayName,
433     ShellItem_GetAttributes,
434     ShellItem_Compare,
435     ShellItem2_GetPropertyStore,
436     ShellItem2_GetPropertyStoreWithCreateObject,
437     ShellItem2_GetPropertyStoreForKeys,
438     ShellItem2_GetPropertyDescriptionList,
439     ShellItem2_Update,
440     ShellItem2_GetProperty,
441     ShellItem2_GetCLSID,
442     ShellItem2_GetFileTime,
443     ShellItem2_GetInt32,
444     ShellItem2_GetString,
445     ShellItem2_GetUInt32,
446     ShellItem2_GetUInt64,
447     ShellItem2_GetBool
448 };
449
450
451 static HRESULT ShellItem_GetClassID(ShellItem* This, CLSID *pClassID)
452 {
453     TRACE("(%p,%p)\n", This, pClassID);
454
455     *pClassID = CLSID_ShellItem;
456     return S_OK;
457 }
458
459
460 static HRESULT WINAPI ShellItem_IPersistIDList_QueryInterface(IPersistIDList *iface,
461     REFIID riid, void **ppv)
462 {
463     ShellItem *This = impl_from_IPersistIDList(iface);
464     return ShellItem_QueryInterface(&This->IShellItem2_iface, riid, ppv);
465 }
466
467 static ULONG WINAPI ShellItem_IPersistIDList_AddRef(IPersistIDList *iface)
468 {
469     ShellItem *This = impl_from_IPersistIDList(iface);
470     return ShellItem_AddRef(&This->IShellItem2_iface);
471 }
472
473 static ULONG WINAPI ShellItem_IPersistIDList_Release(IPersistIDList *iface)
474 {
475     ShellItem *This = impl_from_IPersistIDList(iface);
476     return ShellItem_Release(&This->IShellItem2_iface);
477 }
478
479 static HRESULT WINAPI ShellItem_IPersistIDList_GetClassID(IPersistIDList* iface,
480     CLSID *pClassID)
481 {
482     ShellItem *This = impl_from_IPersistIDList(iface);
483
484     return ShellItem_GetClassID(This, pClassID);
485 }
486
487 static HRESULT WINAPI ShellItem_IPersistIDList_SetIDList(IPersistIDList* iface,
488     LPCITEMIDLIST pidl)
489 {
490     ShellItem *This = impl_from_IPersistIDList(iface);
491     LPITEMIDLIST new_pidl;
492
493     TRACE("(%p,%p)\n", This, pidl);
494
495     new_pidl = ILClone(pidl);
496
497     if (new_pidl)
498     {
499         ILFree(This->pidl);
500         This->pidl = new_pidl;
501         return S_OK;
502     }
503     else
504         return E_OUTOFMEMORY;
505 }
506
507 static HRESULT WINAPI ShellItem_IPersistIDList_GetIDList(IPersistIDList* iface,
508     LPITEMIDLIST *ppidl)
509 {
510     ShellItem *This = impl_from_IPersistIDList(iface);
511
512     TRACE("(%p,%p)\n", This, ppidl);
513
514     *ppidl = ILClone(This->pidl);
515     if (*ppidl)
516         return S_OK;
517     else
518         return E_OUTOFMEMORY;
519 }
520
521 static const IPersistIDListVtbl ShellItem_IPersistIDList_Vtbl = {
522     ShellItem_IPersistIDList_QueryInterface,
523     ShellItem_IPersistIDList_AddRef,
524     ShellItem_IPersistIDList_Release,
525     ShellItem_IPersistIDList_GetClassID,
526     ShellItem_IPersistIDList_SetIDList,
527     ShellItem_IPersistIDList_GetIDList
528 };
529
530
531 HRESULT WINAPI IShellItem_Constructor(IUnknown *pUnkOuter, REFIID riid, void **ppv)
532 {
533     ShellItem *This;
534     HRESULT ret;
535
536     TRACE("(%p,%s)\n",pUnkOuter, debugstr_guid(riid));
537
538     *ppv = NULL;
539
540     if (pUnkOuter) return CLASS_E_NOAGGREGATION;
541
542     This = HeapAlloc(GetProcessHeap(), 0, sizeof(ShellItem));
543     This->IShellItem2_iface.lpVtbl = &ShellItem2_Vtbl;
544     This->ref = 1;
545     This->pidl = NULL;
546     This->IPersistIDList_iface.lpVtbl = &ShellItem_IPersistIDList_Vtbl;
547
548     ret = ShellItem_QueryInterface(&This->IShellItem2_iface, riid, ppv);
549     ShellItem_Release(&This->IShellItem2_iface);
550
551     return ret;
552 }
553
554 HRESULT WINAPI SHCreateShellItem(LPCITEMIDLIST pidlParent,
555     IShellFolder *psfParent, LPCITEMIDLIST pidl, IShellItem **ppsi)
556 {
557     ShellItem *This;
558     LPITEMIDLIST new_pidl;
559     HRESULT ret;
560
561     TRACE("(%p,%p,%p,%p)\n", pidlParent, psfParent, pidl, ppsi);
562
563     if (!pidl)
564     {
565         return E_INVALIDARG;
566     }
567     else if (pidlParent || psfParent)
568     {
569         LPITEMIDLIST temp_parent=NULL;
570         if (!pidlParent)
571         {
572             IPersistFolder2* ppf2Parent;
573
574             if (FAILED(IPersistFolder2_QueryInterface(psfParent, &IID_IPersistFolder2, (void**)&ppf2Parent)))
575             {
576                 FIXME("couldn't get IPersistFolder2 interface of parent\n");
577                 return E_NOINTERFACE;
578             }
579
580             if (FAILED(IPersistFolder2_GetCurFolder(ppf2Parent, &temp_parent)))
581             {
582                 FIXME("couldn't get parent PIDL\n");
583                 IPersistFolder2_Release(ppf2Parent);
584                 return E_NOINTERFACE;
585             }
586
587             pidlParent = temp_parent;
588             IPersistFolder2_Release(ppf2Parent);
589         }
590
591         new_pidl = ILCombine(pidlParent, pidl);
592         ILFree(temp_parent);
593
594         if (!new_pidl)
595             return E_OUTOFMEMORY;
596     }
597     else
598     {
599         new_pidl = ILClone(pidl);
600         if (!new_pidl)
601             return E_OUTOFMEMORY;
602     }
603
604     ret = IShellItem_Constructor(NULL, &IID_IShellItem, (void**)&This);
605     if (This)
606     {
607         *ppsi = (IShellItem*)&This->IShellItem2_iface;
608         This->pidl = new_pidl;
609     }
610     else
611     {
612         *ppsi = NULL;
613         ILFree(new_pidl);
614     }
615     return ret;
616 }
617
618 HRESULT WINAPI SHCreateItemFromParsingName(PCWSTR pszPath,
619     IBindCtx *pbc, REFIID riid, void **ppv)
620 {
621     LPITEMIDLIST pidl;
622     HRESULT ret;
623
624     *ppv = NULL;
625
626     ret = SHParseDisplayName(pszPath, pbc, &pidl, 0, NULL);
627     if(SUCCEEDED(ret))
628     {
629         ShellItem *This;
630         ret = IShellItem_Constructor(NULL, riid, (void**)&This);
631
632         if(SUCCEEDED(ret))
633         {
634             This->pidl = pidl;
635             *ppv = (void*)This;
636         }
637         else
638         {
639             ILFree(pidl);
640         }
641     }
642     return ret;
643 }
644
645 HRESULT WINAPI SHCreateItemFromIDList(PCIDLIST_ABSOLUTE pidl, REFIID riid, void **ppv)
646 {
647     ShellItem *psiimpl;
648     HRESULT ret;
649
650     if(!pidl)
651         return E_INVALIDARG;
652
653     ret = IShellItem_Constructor(NULL, riid, ppv);
654     if(SUCCEEDED(ret))
655     {
656         psiimpl = (ShellItem*)*ppv;
657         psiimpl->pidl = ILClone(pidl);
658     }
659
660     return ret;
661 }
662
663 HRESULT WINAPI SHGetItemFromDataObject(IDataObject *pdtobj,
664     DATAOBJ_GET_ITEM_FLAGS dwFlags, REFIID riid, void **ppv)
665 {
666     FORMATETC fmt;
667     STGMEDIUM medium;
668     HRESULT ret;
669
670     TRACE("%p, %x, %s, %p\n", pdtobj, dwFlags, debugstr_guid(riid), ppv);
671
672     if(!pdtobj)
673         return E_INVALIDARG;
674
675     fmt.cfFormat = RegisterClipboardFormatW(CFSTR_SHELLIDLISTW);
676     fmt.ptd = NULL;
677     fmt.dwAspect = DVASPECT_CONTENT;
678     fmt.lindex = -1;
679     fmt.tymed = TYMED_HGLOBAL;
680
681     ret = IDataObject_GetData(pdtobj, &fmt, &medium);
682     if(SUCCEEDED(ret))
683     {
684         LPIDA pida = GlobalLock(medium.u.hGlobal);
685
686         if((pida->cidl > 1 && !(dwFlags & DOGIF_ONLY_IF_ONE)) ||
687            pida->cidl == 1)
688         {
689             LPITEMIDLIST pidl;
690
691             /* Get the first pidl (parent + child1) */
692             pidl = ILCombine((LPCITEMIDLIST) ((LPBYTE)pida+pida->aoffset[0]),
693                              (LPCITEMIDLIST) ((LPBYTE)pida+pida->aoffset[1]));
694
695             ret = SHCreateItemFromIDList(pidl, riid, ppv);
696             ILFree(pidl);
697         }
698         else
699         {
700             ret = E_FAIL;
701         }
702
703         GlobalUnlock(medium.u.hGlobal);
704         GlobalFree(medium.u.hGlobal);
705     }
706
707     if(FAILED(ret) && !(dwFlags & DOGIF_NO_HDROP))
708     {
709         TRACE("Attempting to fall back on CF_HDROP.\n");
710
711         fmt.cfFormat = CF_HDROP;
712         fmt.ptd = NULL;
713         fmt.dwAspect = DVASPECT_CONTENT;
714         fmt.lindex = -1;
715         fmt.tymed = TYMED_HGLOBAL;
716
717         ret = IDataObject_GetData(pdtobj, &fmt, &medium);
718         if(SUCCEEDED(ret))
719         {
720             DROPFILES *df = GlobalLock(medium.u.hGlobal);
721             LPBYTE files = (LPBYTE)df + df->pFiles;
722             BOOL multiple_files = FALSE;
723
724             ret = E_FAIL;
725             if(!df->fWide)
726             {
727                 WCHAR filename[MAX_PATH];
728                 PCSTR first_file = (PCSTR)files;
729                 if(*(files + lstrlenA(first_file) + 1) != 0)
730                     multiple_files = TRUE;
731
732                 if( !(multiple_files && (dwFlags & DOGIF_ONLY_IF_ONE)) )
733                 {
734                     MultiByteToWideChar(CP_ACP, 0, first_file, -1, filename, MAX_PATH);
735                     ret = SHCreateItemFromParsingName(filename, NULL, riid, ppv);
736                 }
737             }
738             else
739             {
740                 PCWSTR first_file = (PCWSTR)files;
741                 if(*((PCWSTR)files + lstrlenW(first_file) + 1) != 0)
742                     multiple_files = TRUE;
743
744                 if( !(multiple_files && (dwFlags & DOGIF_ONLY_IF_ONE)) )
745                     ret = SHCreateItemFromParsingName(first_file, NULL, riid, ppv);
746             }
747
748             GlobalUnlock(medium.u.hGlobal);
749             GlobalFree(medium.u.hGlobal);
750         }
751     }
752
753     if(FAILED(ret) && !(dwFlags & DOGIF_NO_URL))
754     {
755         FIXME("Failed to create item, should try CF_URL.\n");
756     }
757
758     return ret;
759 }
760
761 HRESULT WINAPI SHGetItemFromObject(IUnknown *punk, REFIID riid, void **ppv)
762 {
763     LPITEMIDLIST pidl;
764     HRESULT ret;
765
766     ret = SHGetIDListFromObject(punk, &pidl);
767     if(SUCCEEDED(ret))
768     {
769         ret = SHCreateItemFromIDList(pidl, riid, ppv);
770         ILFree(pidl);
771     }
772
773     return ret;
774 }
775
776 /*************************************************************************
777  * IShellItemArray implementation
778  */
779 typedef struct {
780     IShellItemArray IShellItemArray_iface;
781     LONG ref;
782
783     IShellItem **array;
784     DWORD item_count;
785 } IShellItemArrayImpl;
786
787 static inline IShellItemArrayImpl *impl_from_IShellItemArray(IShellItemArray *iface)
788 {
789     return CONTAINING_RECORD(iface, IShellItemArrayImpl, IShellItemArray_iface);
790 }
791
792 static HRESULT WINAPI IShellItemArray_fnQueryInterface(IShellItemArray *iface,
793                                                        REFIID riid,
794                                                        void **ppvObject)
795 {
796     IShellItemArrayImpl *This = impl_from_IShellItemArray(iface);
797     TRACE("%p (%s, %p)\n", This, shdebugstr_guid(riid), ppvObject);
798
799     *ppvObject = NULL;
800     if(IsEqualIID(riid, &IID_IShellItemArray) ||
801        IsEqualIID(riid, &IID_IUnknown))
802     {
803         *ppvObject = This;
804     }
805
806     if(*ppvObject)
807     {
808         IUnknown_AddRef((IUnknown*)*ppvObject);
809         return S_OK;
810     }
811
812     return E_NOINTERFACE;
813 }
814
815 static ULONG WINAPI IShellItemArray_fnAddRef(IShellItemArray *iface)
816 {
817     IShellItemArrayImpl *This = impl_from_IShellItemArray(iface);
818     LONG ref = InterlockedIncrement(&This->ref);
819     TRACE("%p - ref %d\n", This, ref);
820
821     return ref;
822 }
823
824 static ULONG WINAPI IShellItemArray_fnRelease(IShellItemArray *iface)
825 {
826     IShellItemArrayImpl *This = impl_from_IShellItemArray(iface);
827     LONG ref = InterlockedDecrement(&This->ref);
828     TRACE("%p - ref %d\n", This, ref);
829
830     if(!ref)
831     {
832         UINT i;
833         TRACE("Freeing.\n");
834
835         for(i = 0; i < This->item_count; i++)
836             IShellItem_Release(This->array[i]);
837
838         HeapFree(GetProcessHeap(), 0, This->array);
839         HeapFree(GetProcessHeap(), 0, This);
840         return 0;
841     }
842
843     return ref;
844 }
845
846 static HRESULT WINAPI IShellItemArray_fnBindToHandler(IShellItemArray *iface,
847                                                       IBindCtx *pbc,
848                                                       REFGUID bhid,
849                                                       REFIID riid,
850                                                       void **ppvOut)
851 {
852     IShellItemArrayImpl *This = impl_from_IShellItemArray(iface);
853     FIXME("Stub: %p (%p, %s, %s, %p)\n",
854           This, pbc, shdebugstr_guid(bhid), shdebugstr_guid(riid), ppvOut);
855
856     return E_NOTIMPL;
857 }
858
859 static HRESULT WINAPI IShellItemArray_fnGetPropertyStore(IShellItemArray *iface,
860                                                          GETPROPERTYSTOREFLAGS flags,
861                                                          REFIID riid,
862                                                          void **ppv)
863 {
864     IShellItemArrayImpl *This = impl_from_IShellItemArray(iface);
865     FIXME("Stub: %p (%x, %s, %p)\n", This, flags, shdebugstr_guid(riid), ppv);
866
867     return E_NOTIMPL;
868 }
869
870 static HRESULT WINAPI IShellItemArray_fnGetPropertyDescriptionList(IShellItemArray *iface,
871                                                                    REFPROPERTYKEY keyType,
872                                                                    REFIID riid,
873                                                                    void **ppv)
874 {
875     IShellItemArrayImpl *This = impl_from_IShellItemArray(iface);
876     FIXME("Stub: %p (%p, %s, %p)\n",
877           This, keyType, shdebugstr_guid(riid), ppv);
878
879     return E_NOTIMPL;
880 }
881
882 static HRESULT WINAPI IShellItemArray_fnGetAttributes(IShellItemArray *iface,
883                                                       SIATTRIBFLAGS AttribFlags,
884                                                       SFGAOF sfgaoMask,
885                                                       SFGAOF *psfgaoAttribs)
886 {
887     IShellItemArrayImpl *This = impl_from_IShellItemArray(iface);
888     FIXME("Stub: %p (%x, %x, %p)\n", This, AttribFlags, sfgaoMask, psfgaoAttribs);
889
890     return E_NOTIMPL;
891 }
892
893 static HRESULT WINAPI IShellItemArray_fnGetCount(IShellItemArray *iface,
894                                                  DWORD *pdwNumItems)
895 {
896     IShellItemArrayImpl *This = impl_from_IShellItemArray(iface);
897     TRACE("%p (%p)\n", This, pdwNumItems);
898
899     *pdwNumItems = This->item_count;
900
901     return S_OK;
902 }
903
904 static HRESULT WINAPI IShellItemArray_fnGetItemAt(IShellItemArray *iface,
905                                                   DWORD dwIndex,
906                                                   IShellItem **ppsi)
907 {
908     IShellItemArrayImpl *This = impl_from_IShellItemArray(iface);
909     TRACE("%p (%x, %p)\n", This, dwIndex, ppsi);
910
911     /* zero indexed */
912     if(dwIndex + 1 > This->item_count)
913         return E_FAIL;
914
915     *ppsi = This->array[dwIndex];
916     IShellItem_AddRef(*ppsi);
917
918     return S_OK;
919 }
920
921 static HRESULT WINAPI IShellItemArray_fnEnumItems(IShellItemArray *iface,
922                                                   IEnumShellItems **ppenumShellItems)
923 {
924     IShellItemArrayImpl *This = impl_from_IShellItemArray(iface);
925     FIXME("Stub: %p (%p)\n", This, ppenumShellItems);
926
927     return E_NOTIMPL;
928 }
929
930 static const IShellItemArrayVtbl vt_IShellItemArray = {
931     IShellItemArray_fnQueryInterface,
932     IShellItemArray_fnAddRef,
933     IShellItemArray_fnRelease,
934     IShellItemArray_fnBindToHandler,
935     IShellItemArray_fnGetPropertyStore,
936     IShellItemArray_fnGetPropertyDescriptionList,
937     IShellItemArray_fnGetAttributes,
938     IShellItemArray_fnGetCount,
939     IShellItemArray_fnGetItemAt,
940     IShellItemArray_fnEnumItems
941 };
942
943 static HRESULT IShellItemArray_Constructor(IUnknown *pUnkOuter, REFIID riid, void **ppv)
944 {
945     IShellItemArrayImpl *This;
946     HRESULT ret;
947
948     TRACE("(%p, %s, %p)\n",pUnkOuter, debugstr_guid(riid), ppv);
949
950     if(pUnkOuter)
951         return CLASS_E_NOAGGREGATION;
952
953     This = HeapAlloc(GetProcessHeap(), 0, sizeof(IShellItemArrayImpl));
954     if(!This)
955         return E_OUTOFMEMORY;
956
957     This->ref = 1;
958     This->IShellItemArray_iface.lpVtbl = &vt_IShellItemArray;
959     This->array = NULL;
960     This->item_count = 0;
961
962     ret = IShellItemArray_QueryInterface(&This->IShellItemArray_iface, riid, ppv);
963     IShellItemArray_Release(&This->IShellItemArray_iface);
964
965     return ret;
966 }
967
968 HRESULT WINAPI SHCreateShellItemArray(PCIDLIST_ABSOLUTE pidlParent,
969                                       IShellFolder *psf,
970                                       UINT cidl,
971                                       PCUITEMID_CHILD_ARRAY ppidl,
972                                       IShellItemArray **ppsiItemArray)
973 {
974     IShellItemArrayImpl *This;
975     IShellItem **array;
976     HRESULT ret = E_FAIL;
977     UINT i;
978
979     TRACE("%p, %p, %d, %p, %p\n", pidlParent, psf, cidl, ppidl, ppsiItemArray);
980
981     if(!pidlParent && !psf)
982         return E_POINTER;
983
984     if(!ppidl)
985         return E_INVALIDARG;
986
987     array = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, cidl*sizeof(IShellItem*));
988     if(!array)
989         return E_OUTOFMEMORY;
990
991     for(i = 0; i < cidl; i++)
992     {
993         ret = SHCreateShellItem(pidlParent, psf, ppidl[i], &array[i]);
994         if(FAILED(ret)) break;
995     }
996
997     if(SUCCEEDED(ret))
998     {
999         ret = IShellItemArray_Constructor(NULL, &IID_IShellItemArray, (void**)&This);
1000         if(SUCCEEDED(ret))
1001         {
1002             This->array = array;
1003             This->item_count = cidl;
1004             *ppsiItemArray = &This->IShellItemArray_iface;
1005
1006             return ret;
1007         }
1008     }
1009
1010     /* Something failed, clean up. */
1011     for(i = 0; i < cidl; i++)
1012         if(array[i]) IShellItem_Release(array[i]);
1013     HeapFree(GetProcessHeap(), 0, array);
1014     *ppsiItemArray = NULL;
1015     return ret;
1016 }
1017
1018 HRESULT WINAPI SHCreateShellItemArrayFromShellItem(IShellItem *psi, REFIID riid, void **ppv)
1019 {
1020     IShellItemArrayImpl *This;
1021     IShellItem **array;
1022     HRESULT ret;
1023
1024     TRACE("%p, %s, %p\n", psi, shdebugstr_guid(riid), ppv);
1025
1026     array = HeapAlloc(GetProcessHeap(), 0, sizeof(IShellItem*));
1027     if(!array)
1028         return E_OUTOFMEMORY;
1029
1030     ret = IShellItemArray_Constructor(NULL, riid, (void**)&This);
1031     if(SUCCEEDED(ret))
1032     {
1033         array[0] = psi;
1034         IShellItem_AddRef(psi);
1035         This->array = array;
1036         This->item_count = 1;
1037         *ppv = This;
1038     }
1039     else
1040     {
1041         HeapFree(GetProcessHeap(), 0, array);
1042         *ppv = NULL;
1043     }
1044
1045     return ret;
1046 }
1047
1048 HRESULT WINAPI SHCreateShellItemArrayFromDataObject(IDataObject *pdo, REFIID riid, void **ppv)
1049 {
1050     IShellItemArray *psia;
1051     FORMATETC fmt;
1052     STGMEDIUM medium;
1053     HRESULT ret;
1054
1055     TRACE("%p, %s, %p\n", pdo, shdebugstr_guid(riid), ppv);
1056
1057     if(!pdo)
1058         return E_INVALIDARG;
1059
1060     *ppv = NULL;
1061
1062     fmt.cfFormat = RegisterClipboardFormatW(CFSTR_SHELLIDLISTW);
1063     fmt.ptd = NULL;
1064     fmt.dwAspect = DVASPECT_CONTENT;
1065     fmt.lindex = -1;
1066     fmt.tymed = TYMED_HGLOBAL;
1067
1068     ret = IDataObject_GetData(pdo, &fmt, &medium);
1069     if(SUCCEEDED(ret))
1070     {
1071         LPIDA pida = GlobalLock(medium.u.hGlobal);
1072         LPCITEMIDLIST parent_pidl;
1073         LPCITEMIDLIST *children;
1074         UINT i;
1075         TRACE("Converting %d objects.\n", pida->cidl);
1076
1077         parent_pidl = (LPCITEMIDLIST) ((LPBYTE)pida+pida->aoffset[0]);
1078
1079         children = HeapAlloc(GetProcessHeap(), 0, sizeof(LPCITEMIDLIST)*pida->cidl);
1080         for(i = 0; i < pida->cidl; i++)
1081             children[i] = (LPCITEMIDLIST) ((LPBYTE)pida+pida->aoffset[i+1]);
1082
1083         ret = SHCreateShellItemArray(parent_pidl, NULL, pida->cidl, children, &psia);
1084
1085         HeapFree(GetProcessHeap(), 0, children);
1086
1087         GlobalUnlock(medium.u.hGlobal);
1088         GlobalFree(medium.u.hGlobal);
1089     }
1090
1091     if(SUCCEEDED(ret))
1092     {
1093         ret = IShellItemArray_QueryInterface(psia, riid, ppv);
1094         IShellItemArray_Release(psia);
1095     }
1096
1097     return ret;
1098 }