msi: Use MsgWaitForMultipleObjectsEx to do waits.
[wine/multimedia.git] / dlls / hlink / link.c
blobeb6257e4a0a1a3bd41b925c56f31b6f7b3909cf3
1 /*
2 * Implementation of hyperlinking (hlink.dll)
4 * Copyright 2005 Aric Stewart for CodeWeavers
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.
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.
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
21 #include "hlink_private.h"
23 #include "shellapi.h"
24 #include "hlguids.h"
26 #include "wine/debug.h"
28 WINE_DEFAULT_DEBUG_CHANNEL(hlink);
30 #define HLINK_SAVE_MAGIC 0x00000002
31 #define HLINK_SAVE_MONIKER_PRESENT 0x01
32 #define HLINK_SAVE_MONIKER_IS_ABSOLUTE 0x02
33 #define HLINK_SAVE_LOCATION_PRESENT 0x08
34 #define HLINK_SAVE_FRIENDLY_PRESENT 0x10
35 /* 0x20, 0x40 unknown */
36 #define HLINK_SAVE_TARGET_FRAME_PRESENT 0x80
37 /* known flags */
38 #define HLINK_SAVE_ALL (HLINK_SAVE_TARGET_FRAME_PRESENT|HLINK_SAVE_FRIENDLY_PRESENT|HLINK_SAVE_LOCATION_PRESENT|0x04|HLINK_SAVE_MONIKER_IS_ABSOLUTE|HLINK_SAVE_MONIKER_PRESENT)
40 static const IHlinkVtbl hlvt;
41 static const IPersistStreamVtbl psvt;
42 static const IDataObjectVtbl dovt;
44 typedef struct
46 IHlink IHlink_iface;
47 LONG ref;
49 IPersistStream IPersistStream_iface;
50 IDataObject IDataObject_iface;
52 LPWSTR FriendlyName;
53 LPWSTR Location;
54 LPWSTR TargetFrameName;
55 IMoniker *Moniker;
56 IHlinkSite *Site;
57 DWORD SiteData;
58 BOOL absolute;
59 } HlinkImpl;
61 static inline HlinkImpl *impl_from_IHlink(IHlink *iface)
63 return CONTAINING_RECORD(iface, HlinkImpl, IHlink_iface);
67 static inline HlinkImpl* impl_from_IPersistStream( IPersistStream* iface)
69 return CONTAINING_RECORD(iface, HlinkImpl, IPersistStream_iface);
72 static inline HlinkImpl* impl_from_IDataObject( IDataObject* iface)
74 return CONTAINING_RECORD(iface, HlinkImpl, IDataObject_iface);
77 static HRESULT __GetMoniker(HlinkImpl* This, IMoniker** moniker,
78 DWORD ref_type)
80 HRESULT hres;
82 if (ref_type == HLINKGETREF_DEFAULT)
83 ref_type = HLINKGETREF_RELATIVE;
85 if (ref_type == HLINKGETREF_ABSOLUTE && This->Site)
87 IMoniker *hls_moniker;
89 hres = IHlinkSite_GetMoniker(This->Site, This->SiteData,
90 OLEGETMONIKER_FORCEASSIGN, OLEWHICHMK_CONTAINER, &hls_moniker);
91 if (FAILED(hres))
92 return hres;
94 if (This->Moniker)
96 hres = IMoniker_ComposeWith(hls_moniker, This->Moniker, FALSE,
97 moniker);
98 IMoniker_Release(hls_moniker);
99 return hres;
102 *moniker = hls_moniker;
103 return S_OK;
106 *moniker = This->Moniker;
107 if (*moniker)
108 IMoniker_AddRef(*moniker);
110 return S_OK;
113 HRESULT WINAPI HLink_Constructor(IUnknown *pUnkOuter, REFIID riid,
114 LPVOID *ppv)
116 HlinkImpl * hl;
118 TRACE("unkOut=%p riid=%s\n", pUnkOuter, debugstr_guid(riid));
119 *ppv = NULL;
121 if (pUnkOuter)
122 return CLASS_E_NOAGGREGATION;
124 hl = heap_alloc_zero(sizeof(HlinkImpl));
125 if (!hl)
126 return E_OUTOFMEMORY;
128 hl->ref = 1;
129 hl->IHlink_iface.lpVtbl = &hlvt;
130 hl->IPersistStream_iface.lpVtbl = &psvt;
131 hl->IDataObject_iface.lpVtbl = &dovt;
133 *ppv = hl;
134 return S_OK;
137 static HRESULT WINAPI IHlink_fnQueryInterface(IHlink* iface, REFIID riid,
138 LPVOID *ppvObj)
140 HlinkImpl *This = impl_from_IHlink(iface);
142 TRACE ("(%p)->(%s,%p)\n", This, debugstr_guid (riid), ppvObj);
144 *ppvObj = NULL;
146 if (IsEqualIID(riid, &IID_IUnknown) || (IsEqualIID(riid, &IID_IHlink)))
147 *ppvObj = This;
148 else if (IsEqualIID(riid, &IID_IPersistStream))
149 *ppvObj = &This->IPersistStream_iface;
150 else if (IsEqualIID(riid, &IID_IDataObject))
151 *ppvObj = &This->IDataObject_iface;
153 if (*ppvObj)
155 IUnknown_AddRef((IUnknown*)(*ppvObj));
156 return S_OK;
158 return E_NOINTERFACE;
161 static ULONG WINAPI IHlink_fnAddRef (IHlink* iface)
163 HlinkImpl *This = impl_from_IHlink(iface);
164 ULONG refCount = InterlockedIncrement(&This->ref);
166 TRACE("(%p)->(count=%u)\n", This, refCount - 1);
168 return refCount;
171 static ULONG WINAPI IHlink_fnRelease (IHlink* iface)
173 HlinkImpl *This = impl_from_IHlink(iface);
174 ULONG refCount = InterlockedDecrement(&This->ref);
176 TRACE("(%p)->(count=%u)\n", This, refCount + 1);
177 if (refCount)
178 return refCount;
180 TRACE("-- destroying IHlink (%p)\n", This);
181 heap_free(This->FriendlyName);
182 heap_free(This->TargetFrameName);
183 heap_free(This->Location);
184 if (This->Moniker)
185 IMoniker_Release(This->Moniker);
186 if (This->Site)
187 IHlinkSite_Release(This->Site);
188 heap_free(This);
189 return 0;
192 static HRESULT WINAPI IHlink_fnSetHlinkSite( IHlink* iface,
193 IHlinkSite* pihlSite, DWORD dwSiteData)
195 HlinkImpl *This = impl_from_IHlink(iface);
197 TRACE("(%p)->(%p %i)\n", This, pihlSite, dwSiteData);
199 if (This->Site)
200 IHlinkSite_Release(This->Site);
202 This->Site = pihlSite;
203 if (This->Site)
204 IHlinkSite_AddRef(This->Site);
206 This->SiteData = dwSiteData;
208 return S_OK;
211 static HRESULT WINAPI IHlink_fnGetHlinkSite( IHlink* iface,
212 IHlinkSite** ppihlSite, DWORD *pdwSiteData)
214 HlinkImpl *This = impl_from_IHlink(iface);
216 TRACE("(%p)->(%p %p)\n", This, ppihlSite, pdwSiteData);
218 *ppihlSite = This->Site;
220 if (This->Site) {
221 IHlinkSite_AddRef(This->Site);
222 *pdwSiteData = This->SiteData;
225 return S_OK;
228 static HRESULT WINAPI IHlink_fnSetMonikerReference( IHlink* iface,
229 DWORD rfHLSETF, IMoniker *pmkTarget, LPCWSTR pwzLocation)
231 HlinkImpl *This = impl_from_IHlink(iface);
233 TRACE("(%p)->(%i %p %s)\n", This, rfHLSETF, pmkTarget,
234 debugstr_w(pwzLocation));
236 if(rfHLSETF == 0)
237 return E_INVALIDARG;
238 if(!(rfHLSETF & (HLINKSETF_TARGET | HLINKSETF_LOCATION)))
239 return rfHLSETF;
241 if(rfHLSETF & HLINKSETF_TARGET){
242 if (This->Moniker)
243 IMoniker_Release(This->Moniker);
245 This->Moniker = pmkTarget;
246 if (This->Moniker)
248 LPOLESTR display_name;
249 IMoniker_AddRef(This->Moniker);
250 IMoniker_GetDisplayName(This->Moniker, NULL, NULL, &display_name);
251 This->absolute = display_name && strchrW(display_name, ':');
252 CoTaskMemFree(display_name);
256 if(rfHLSETF & HLINKSETF_LOCATION){
257 heap_free(This->Location);
258 This->Location = hlink_strdupW( pwzLocation );
261 return S_OK;
264 static HRESULT WINAPI IHlink_fnSetStringReference(IHlink* iface,
265 DWORD grfHLSETF, LPCWSTR pwzTarget, LPCWSTR pwzLocation)
267 HlinkImpl *This = impl_from_IHlink(iface);
269 TRACE("(%p)->(%i %s %s)\n", This, grfHLSETF, debugstr_w(pwzTarget),
270 debugstr_w(pwzLocation));
272 if(grfHLSETF > (HLINKSETF_TARGET | HLINKSETF_LOCATION) &&
273 grfHLSETF < -(HLINKSETF_TARGET | HLINKSETF_LOCATION))
274 return grfHLSETF;
276 if (grfHLSETF & HLINKSETF_TARGET)
278 if (This->Moniker)
280 IMoniker_Release(This->Moniker);
281 This->Moniker = NULL;
283 if (pwzTarget && *pwzTarget)
285 IMoniker *pMon;
286 IBindCtx *pbc = NULL;
287 ULONG eaten;
288 HRESULT r;
290 r = CreateBindCtx(0, &pbc);
291 if (FAILED(r))
292 return E_OUTOFMEMORY;
294 r = MkParseDisplayName(pbc, pwzTarget, &eaten, &pMon);
295 IBindCtx_Release(pbc);
297 if (FAILED(r))
299 LPCWSTR p = strchrW(pwzTarget, ':');
300 if (p && (p - pwzTarget > 1))
301 r = CreateURLMoniker(NULL, pwzTarget, &pMon);
302 else
303 r = CreateFileMoniker(pwzTarget, &pMon);
304 if (FAILED(r))
306 ERR("couldn't create moniker for %s, failed with error 0x%08x\n",
307 debugstr_w(pwzTarget), r);
308 return r;
312 IHlink_SetMonikerReference(iface, HLINKSETF_TARGET, pMon, NULL);
313 IMoniker_Release(pMon);
317 if (grfHLSETF & HLINKSETF_LOCATION)
319 heap_free(This->Location);
320 This->Location = NULL;
321 if (pwzLocation && *pwzLocation)
322 This->Location = hlink_strdupW( pwzLocation );
325 return S_OK;
328 static HRESULT WINAPI IHlink_fnGetMonikerReference(IHlink* iface,
329 DWORD dwWhichRef, IMoniker **ppimkTarget, LPWSTR *ppwzLocation)
331 HlinkImpl *This = impl_from_IHlink(iface);
333 TRACE("(%p) -> (%i %p %p)\n", This, dwWhichRef, ppimkTarget,
334 ppwzLocation);
336 if (ppimkTarget)
338 HRESULT hres = __GetMoniker(This, ppimkTarget, dwWhichRef);
339 if (FAILED(hres))
341 if (ppwzLocation)
342 *ppwzLocation = NULL;
343 return hres;
347 if (ppwzLocation)
348 IHlink_GetStringReference(iface, dwWhichRef, NULL, ppwzLocation);
350 return S_OK;
353 static HRESULT WINAPI IHlink_fnGetStringReference (IHlink* iface,
354 DWORD dwWhichRef, LPWSTR *ppwzTarget, LPWSTR *ppwzLocation)
356 HlinkImpl *This = impl_from_IHlink(iface);
358 TRACE("(%p) -> (%i %p %p)\n", This, dwWhichRef, ppwzTarget, ppwzLocation);
360 if(dwWhichRef != -1 && dwWhichRef & ~(HLINKGETREF_DEFAULT | HLINKGETREF_ABSOLUTE | HLINKGETREF_RELATIVE))
362 if(ppwzTarget)
363 *ppwzTarget = NULL;
364 if(ppwzLocation)
365 *ppwzLocation = NULL;
366 return E_INVALIDARG;
369 if (ppwzTarget)
371 IMoniker* mon;
372 HRESULT hres = __GetMoniker(This, &mon, dwWhichRef);
373 if (FAILED(hres))
375 if (ppwzLocation)
376 *ppwzLocation = NULL;
377 return hres;
379 if (mon)
381 IBindCtx *pbc;
383 CreateBindCtx( 0, &pbc);
384 IMoniker_GetDisplayName(mon, pbc, NULL, ppwzTarget);
385 IBindCtx_Release(pbc);
386 IMoniker_Release(mon);
388 else
389 *ppwzTarget = NULL;
391 if (ppwzLocation)
392 *ppwzLocation = hlink_co_strdupW( This->Location );
394 TRACE("(Target: %s Location: %s)\n",
395 (ppwzTarget)?debugstr_w(*ppwzTarget):"<NULL>",
396 (ppwzLocation)?debugstr_w(*ppwzLocation):"<NULL>");
398 return S_OK;
401 static HRESULT WINAPI IHlink_fnSetFriendlyName (IHlink *iface,
402 LPCWSTR pwzFriendlyName)
404 HlinkImpl *This = impl_from_IHlink(iface);
406 TRACE("(%p) -> (%s)\n", This, debugstr_w(pwzFriendlyName));
408 heap_free(This->FriendlyName);
409 This->FriendlyName = hlink_strdupW( pwzFriendlyName );
411 return S_OK;
414 static HRESULT WINAPI IHlink_fnGetFriendlyName (IHlink* iface,
415 DWORD grfHLFNAMEF, LPWSTR* ppwzFriendlyName)
417 HlinkImpl *This = impl_from_IHlink(iface);
419 TRACE("(%p) -> (%i %p)\n", This, grfHLFNAMEF, ppwzFriendlyName);
421 /* FIXME: Only using explicitly set and cached friendly names */
423 if (This->FriendlyName)
424 *ppwzFriendlyName = hlink_co_strdupW( This->FriendlyName );
425 else
427 IMoniker *moniker;
428 HRESULT hres = __GetMoniker(This, &moniker, HLINKGETREF_DEFAULT);
429 if (FAILED(hres))
431 *ppwzFriendlyName = NULL;
432 return hres;
434 if (moniker)
436 IBindCtx *bcxt;
437 CreateBindCtx(0, &bcxt);
439 IMoniker_GetDisplayName(moniker, bcxt, NULL, ppwzFriendlyName);
440 IBindCtx_Release(bcxt);
441 IMoniker_Release(moniker);
443 else
444 *ppwzFriendlyName = NULL;
447 return S_OK;
450 static HRESULT WINAPI IHlink_fnSetTargetFrameName(IHlink* iface,
451 LPCWSTR pwzTargetFramename)
453 HlinkImpl *This = impl_from_IHlink(iface);
454 TRACE("(%p)->(%s)\n", This, debugstr_w(pwzTargetFramename));
456 heap_free(This->TargetFrameName);
457 This->TargetFrameName = hlink_strdupW( pwzTargetFramename );
459 return S_OK;
462 static HRESULT WINAPI IHlink_fnGetTargetFrameName(IHlink* iface,
463 LPWSTR *ppwzTargetFrameName)
465 HlinkImpl *This = impl_from_IHlink(iface);
467 TRACE("(%p)->(%p)\n", This, ppwzTargetFrameName);
468 *ppwzTargetFrameName = hlink_co_strdupW( This->TargetFrameName );
470 return S_OK;
473 static HRESULT WINAPI IHlink_fnGetMiscStatus(IHlink* iface, DWORD* pdwStatus)
475 FIXME("\n");
476 return E_NOTIMPL;
479 static HRESULT WINAPI IHlink_fnNavigate(IHlink* iface, DWORD grfHLNF, LPBC pbc,
480 IBindStatusCallback *pbsc, IHlinkBrowseContext *phbc)
482 HlinkImpl *This = impl_from_IHlink(iface);
483 IMoniker *mon = NULL;
484 HRESULT r;
486 FIXME("Semi-Stub:(%p)->(%i %p %p %p)\n", This, grfHLNF, pbc, pbsc, phbc);
488 r = __GetMoniker(This, &mon, HLINKGETREF_ABSOLUTE);
489 TRACE("Moniker %p\n", mon);
491 if (SUCCEEDED(r))
493 IBindCtx *bcxt;
494 IHlinkTarget *target = NULL;
496 CreateBindCtx(0, &bcxt);
498 RegisterBindStatusCallback(bcxt, pbsc, NULL, 0);
500 r = IMoniker_BindToObject(mon, bcxt, NULL, &IID_IHlinkTarget,
501 (LPVOID*)&target);
502 TRACE("IHlinkTarget returned 0x%x\n", r);
503 if (r == S_OK)
505 IHlinkTarget_SetBrowseContext(target, phbc);
506 IHlinkTarget_Navigate(target, grfHLNF, This->Location);
507 IHlinkTarget_Release(target);
509 else
511 static const WCHAR szOpen[] = {'o','p','e','n',0};
512 LPWSTR target = NULL;
514 r = IHlink_GetStringReference(iface, HLINKGETREF_DEFAULT, &target, NULL);
515 if (SUCCEEDED(r) && target)
517 ShellExecuteW(NULL, szOpen, target, NULL, NULL, SW_SHOW);
518 CoTaskMemFree(target);
522 RevokeBindStatusCallback(bcxt, pbsc);
524 IBindCtx_Release(bcxt);
525 IMoniker_Release(mon);
528 if (This->Site)
529 IHlinkSite_OnNavigationComplete(This->Site, This->SiteData, 0, r, NULL);
531 TRACE("Finished Navigation\n");
532 return r;
535 static HRESULT WINAPI IHlink_fnSetAdditonalParams(IHlink* iface,
536 LPCWSTR pwzAdditionalParams)
538 TRACE("Not implemented in native IHlink\n");
539 return E_NOTIMPL;
542 static HRESULT WINAPI IHlink_fnGetAdditionalParams(IHlink* iface,
543 LPWSTR* ppwzAdditionalParams)
545 TRACE("Not implemented in native IHlink\n");
546 return E_NOTIMPL;
549 static const IHlinkVtbl hlvt =
551 IHlink_fnQueryInterface,
552 IHlink_fnAddRef,
553 IHlink_fnRelease,
554 IHlink_fnSetHlinkSite,
555 IHlink_fnGetHlinkSite,
556 IHlink_fnSetMonikerReference,
557 IHlink_fnGetMonikerReference,
558 IHlink_fnSetStringReference,
559 IHlink_fnGetStringReference,
560 IHlink_fnSetFriendlyName,
561 IHlink_fnGetFriendlyName,
562 IHlink_fnSetTargetFrameName,
563 IHlink_fnGetTargetFrameName,
564 IHlink_fnGetMiscStatus,
565 IHlink_fnNavigate,
566 IHlink_fnSetAdditonalParams,
567 IHlink_fnGetAdditionalParams
570 static HRESULT WINAPI IDataObject_fnQueryInterface(IDataObject* iface,
571 REFIID riid, LPVOID *ppvObj)
573 HlinkImpl *This = impl_from_IDataObject(iface);
574 TRACE("%p\n", This);
575 return IHlink_QueryInterface(&This->IHlink_iface, riid, ppvObj);
578 static ULONG WINAPI IDataObject_fnAddRef (IDataObject* iface)
580 HlinkImpl *This = impl_from_IDataObject(iface);
581 TRACE("%p\n", This);
582 return IHlink_AddRef(&This->IHlink_iface);
585 static ULONG WINAPI IDataObject_fnRelease (IDataObject* iface)
587 HlinkImpl *This = impl_from_IDataObject(iface);
588 TRACE("%p\n", This);
589 return IHlink_Release(&This->IHlink_iface);
592 static HRESULT WINAPI IDataObject_fnGetData(IDataObject* iface,
593 FORMATETC* pformatetcIn, STGMEDIUM* pmedium)
595 FIXME("\n");
596 return E_NOTIMPL;
599 static HRESULT WINAPI IDataObject_fnGetDataHere(IDataObject* iface,
600 FORMATETC* pformatetc, STGMEDIUM* pmedium)
602 FIXME("\n");
603 return E_NOTIMPL;
606 static HRESULT WINAPI IDataObject_fnQueryGetData(IDataObject* iface,
607 FORMATETC* pformatetc)
609 FIXME("\n");
610 return E_NOTIMPL;
613 static HRESULT WINAPI IDataObject_fnGetConicalFormatEtc(IDataObject* iface,
614 FORMATETC* pformatetcIn, FORMATETC* pformatetcOut)
616 FIXME("\n");
617 return E_NOTIMPL;
620 static HRESULT WINAPI IDataObject_fnSetData(IDataObject* iface,
621 FORMATETC* pformatetc, STGMEDIUM* pmedium, BOOL fRelease)
623 FIXME("\n");
624 return E_NOTIMPL;
627 static HRESULT WINAPI IDataObject_fnEnumFormatEtc(IDataObject* iface,
628 DWORD dwDirection, IEnumFORMATETC** ppenumFormatEtc)
630 FIXME("\n");
631 return E_NOTIMPL;
634 static HRESULT WINAPI IDataObject_fnDAdvise(IDataObject* iface,
635 FORMATETC* pformatetc, DWORD advf, IAdviseSink* pAdvSink,
636 DWORD* pdwConnection)
638 FIXME("\n");
639 return E_NOTIMPL;
642 static HRESULT WINAPI IDataObject_fnDUnadvise(IDataObject* iface,
643 DWORD dwConnection)
645 FIXME("\n");
646 return E_NOTIMPL;
649 static HRESULT WINAPI IDataObject_fnEnumDAdvise(IDataObject* iface,
650 IEnumSTATDATA** ppenumAdvise)
652 FIXME("\n");
653 return E_NOTIMPL;
656 static const IDataObjectVtbl dovt =
658 IDataObject_fnQueryInterface,
659 IDataObject_fnAddRef,
660 IDataObject_fnRelease,
661 IDataObject_fnGetData,
662 IDataObject_fnGetDataHere,
663 IDataObject_fnQueryGetData,
664 IDataObject_fnGetConicalFormatEtc,
665 IDataObject_fnSetData,
666 IDataObject_fnEnumFormatEtc,
667 IDataObject_fnDAdvise,
668 IDataObject_fnDUnadvise,
669 IDataObject_fnEnumDAdvise
672 static HRESULT WINAPI IPersistStream_fnQueryInterface(IPersistStream* iface,
673 REFIID riid, LPVOID *ppvObj)
675 HlinkImpl *This = impl_from_IPersistStream(iface);
676 TRACE("(%p)\n", This);
677 return IHlink_QueryInterface(&This->IHlink_iface, riid, ppvObj);
680 static ULONG WINAPI IPersistStream_fnAddRef (IPersistStream* iface)
682 HlinkImpl *This = impl_from_IPersistStream(iface);
683 TRACE("(%p)\n", This);
684 return IHlink_AddRef(&This->IHlink_iface);
687 static ULONG WINAPI IPersistStream_fnRelease (IPersistStream* iface)
689 HlinkImpl *This = impl_from_IPersistStream(iface);
690 TRACE("(%p)\n", This);
691 return IHlink_Release(&This->IHlink_iface);
694 static HRESULT WINAPI IPersistStream_fnGetClassID(IPersistStream* iface,
695 CLSID* pClassID)
697 HlinkImpl *This = impl_from_IPersistStream(iface);
698 TRACE("(%p)\n", This);
699 *pClassID = CLSID_StdHlink;
700 return S_OK;
703 static HRESULT WINAPI IPersistStream_fnIsDirty(IPersistStream* iface)
705 FIXME("\n");
706 return E_NOTIMPL;
709 static HRESULT write_hlink_string(IStream *pStm, LPCWSTR str)
711 DWORD len;
712 HRESULT hr;
714 TRACE("(%p, %s)\n", pStm, debugstr_w(str));
716 len = strlenW(str) + 1;
718 hr = IStream_Write(pStm, &len, sizeof(len), NULL);
719 if (FAILED(hr)) return hr;
721 hr = IStream_Write(pStm, str, len * sizeof(WCHAR), NULL);
722 if (FAILED(hr)) return hr;
724 return S_OK;
727 static inline ULONG size_hlink_string(LPCWSTR str)
729 return sizeof(DWORD) + (strlenW(str) + 1) * sizeof(WCHAR);
732 static HRESULT read_hlink_string(IStream *pStm, LPWSTR *out_str)
734 LPWSTR str;
735 DWORD len;
736 ULONG read;
737 HRESULT hr;
739 hr = IStream_Read(pStm, &len, sizeof(len), &read);
740 if (FAILED(hr)) return hr;
741 if (read != sizeof(len)) return STG_E_READFAULT;
743 TRACE("read len %d\n", len);
745 str = heap_alloc(len * sizeof(WCHAR));
746 if (!str) return E_OUTOFMEMORY;
748 hr = IStream_Read(pStm, str, len * sizeof(WCHAR), &read);
749 if (FAILED(hr))
751 heap_free(str);
752 return hr;
754 if (read != len * sizeof(WCHAR))
756 heap_free(str);
757 return STG_E_READFAULT;
759 TRACE("read string %s\n", debugstr_w(str));
761 *out_str = str;
762 return S_OK;
765 static HRESULT WINAPI IPersistStream_fnLoad(IPersistStream* iface,
766 IStream* pStm)
768 HRESULT r;
769 DWORD hdr[2];
770 DWORD read;
771 HlinkImpl *This = impl_from_IPersistStream(iface);
773 r = IStream_Read(pStm, hdr, sizeof(hdr), &read);
774 if (read != sizeof(hdr) || (hdr[0] != HLINK_SAVE_MAGIC))
776 r = E_FAIL;
777 goto end;
779 if (hdr[1] & ~HLINK_SAVE_ALL)
780 FIXME("unknown flag(s) 0x%x\n", hdr[1] & ~HLINK_SAVE_ALL);
782 if (hdr[1] & HLINK_SAVE_TARGET_FRAME_PRESENT)
784 TRACE("loading target frame name\n");
785 r = read_hlink_string(pStm, &This->TargetFrameName);
786 if (FAILED(r)) goto end;
789 if (hdr[1] & HLINK_SAVE_FRIENDLY_PRESENT)
791 TRACE("loading target friendly name\n");
792 if (!(hdr[1] & 0x4))
793 FIXME("0x4 flag not present with friendly name flag - not sure what this means\n");
794 r = read_hlink_string(pStm, &This->FriendlyName);
795 if (FAILED(r)) goto end;
798 if (hdr[1] & HLINK_SAVE_MONIKER_PRESENT)
800 TRACE("loading moniker\n");
801 r = OleLoadFromStream(pStm, &IID_IMoniker, (LPVOID*)&(This->Moniker));
802 if (FAILED(r))
803 goto end;
804 This->absolute = hdr[1] & HLINK_SAVE_MONIKER_IS_ABSOLUTE ? TRUE : FALSE;
807 if (hdr[1] & HLINK_SAVE_LOCATION_PRESENT)
809 TRACE("loading location\n");
810 r = read_hlink_string(pStm, &This->Location);
811 if (FAILED(r)) goto end;
814 end:
815 TRACE("Load Result 0x%x (%p)\n", r, This->Moniker);
817 return r;
820 static HRESULT WINAPI IPersistStream_fnSave(IPersistStream* iface,
821 IStream* pStm, BOOL fClearDirty)
823 HRESULT r;
824 HlinkImpl *This = impl_from_IPersistStream(iface);
825 DWORD hdr[2];
826 IMoniker *moniker;
828 TRACE("(%p) Moniker(%p)\n", This, This->Moniker);
830 r = __GetMoniker(This, &moniker, HLINKGETREF_DEFAULT);
831 if (FAILED(r))
832 return r;
833 r = E_FAIL;
835 hdr[0] = HLINK_SAVE_MAGIC;
836 hdr[1] = 0;
838 if (moniker)
839 hdr[1] |= HLINK_SAVE_MONIKER_PRESENT;
840 if (This->absolute)
841 hdr[1] |= HLINK_SAVE_MONIKER_IS_ABSOLUTE;
842 if (This->Location)
843 hdr[1] |= HLINK_SAVE_LOCATION_PRESENT;
844 if (This->FriendlyName)
845 hdr[1] |= HLINK_SAVE_FRIENDLY_PRESENT | 4 /* FIXME */;
846 if (This->TargetFrameName)
847 hdr[1] |= HLINK_SAVE_TARGET_FRAME_PRESENT;
849 IStream_Write(pStm, hdr, sizeof(hdr), NULL);
851 if (This->TargetFrameName)
853 r = write_hlink_string(pStm, This->TargetFrameName);
854 if (FAILED(r)) goto end;
857 if (This->FriendlyName)
859 r = write_hlink_string(pStm, This->FriendlyName);
860 if (FAILED(r)) goto end;
863 if (moniker)
865 IPersistStream* monstream;
867 monstream = NULL;
868 IMoniker_QueryInterface(moniker, &IID_IPersistStream,
869 (LPVOID*)&monstream);
870 if (monstream)
872 r = OleSaveToStream(monstream, pStm);
873 IPersistStream_Release(monstream);
875 if (FAILED(r)) goto end;
878 if (This->Location)
880 r = write_hlink_string(pStm, This->Location);
881 if (FAILED(r)) goto end;
884 end:
885 if (moniker) IMoniker_Release(moniker);
886 TRACE("Save Result 0x%x\n", r);
888 return r;
891 static HRESULT WINAPI IPersistStream_fnGetSizeMax(IPersistStream* iface,
892 ULARGE_INTEGER* pcbSize)
894 HRESULT r;
895 HlinkImpl *This = impl_from_IPersistStream(iface);
896 IMoniker *moniker;
898 TRACE("(%p) Moniker(%p)\n", This, This->Moniker);
900 pcbSize->QuadPart = sizeof(DWORD)*2;
902 if (This->TargetFrameName)
903 pcbSize->QuadPart += size_hlink_string(This->TargetFrameName);
905 if (This->FriendlyName)
906 pcbSize->QuadPart += size_hlink_string(This->FriendlyName);
908 r = __GetMoniker(This, &moniker, HLINKGETREF_DEFAULT);
909 if (FAILED(r))
910 return r;
911 r = E_FAIL;
913 if (moniker)
915 IPersistStream* monstream = NULL;
916 IMoniker_QueryInterface(moniker, &IID_IPersistStream,
917 (LPVOID*)&monstream);
918 if (monstream)
920 ULARGE_INTEGER mon_size;
921 r = IPersistStream_GetSizeMax(monstream, &mon_size);
922 pcbSize->QuadPart += mon_size.QuadPart;
923 IPersistStream_Release(monstream);
925 IMoniker_Release(moniker);
928 if (This->Location)
929 pcbSize->QuadPart += size_hlink_string(This->Location);
931 return r;
934 static const IPersistStreamVtbl psvt =
936 IPersistStream_fnQueryInterface,
937 IPersistStream_fnAddRef,
938 IPersistStream_fnRelease,
939 IPersistStream_fnGetClassID,
940 IPersistStream_fnIsDirty,
941 IPersistStream_fnLoad,
942 IPersistStream_fnSave,
943 IPersistStream_fnGetSizeMax,