msvcrt: Stop at trylevel for ControlPc on target frame for non-consolidate unwinds.
[wine.git] / dlls / hlink / link.c
blobf5da0b49a7d9a6e6dec9d9cb2f18063110b59101
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 typedef struct
42 IHlink IHlink_iface;
43 LONG ref;
45 IPersistStream IPersistStream_iface;
46 IDataObject IDataObject_iface;
48 LPWSTR FriendlyName;
49 LPWSTR Location;
50 LPWSTR TargetFrameName;
51 IMoniker *Moniker;
52 IHlinkSite *Site;
53 DWORD SiteData;
54 BOOL absolute;
55 } HlinkImpl;
57 static inline HlinkImpl *impl_from_IHlink(IHlink *iface)
59 return CONTAINING_RECORD(iface, HlinkImpl, IHlink_iface);
63 static inline HlinkImpl* impl_from_IPersistStream( IPersistStream* iface)
65 return CONTAINING_RECORD(iface, HlinkImpl, IPersistStream_iface);
68 static inline HlinkImpl* impl_from_IDataObject( IDataObject* iface)
70 return CONTAINING_RECORD(iface, HlinkImpl, IDataObject_iface);
73 static HRESULT __GetMoniker(HlinkImpl* This, IMoniker** moniker,
74 DWORD ref_type)
76 HRESULT hres;
78 if (ref_type == HLINKGETREF_DEFAULT)
79 ref_type = HLINKGETREF_RELATIVE;
81 if (This->Moniker)
83 DWORD mktype = MKSYS_NONE;
85 hres = IMoniker_IsSystemMoniker(This->Moniker, &mktype);
86 if (hres == S_OK && mktype != MKSYS_NONE)
88 *moniker = This->Moniker;
89 IMoniker_AddRef(*moniker);
90 return S_OK;
94 if (ref_type == HLINKGETREF_ABSOLUTE && This->Site)
96 IMoniker *hls_moniker;
98 hres = IHlinkSite_GetMoniker(This->Site, This->SiteData,
99 OLEGETMONIKER_FORCEASSIGN, OLEWHICHMK_CONTAINER, &hls_moniker);
100 if (FAILED(hres))
101 return hres;
103 if (This->Moniker)
105 hres = IMoniker_ComposeWith(hls_moniker, This->Moniker, FALSE,
106 moniker);
107 IMoniker_Release(hls_moniker);
108 return hres;
111 *moniker = hls_moniker;
112 return S_OK;
115 *moniker = This->Moniker;
116 if (*moniker)
117 IMoniker_AddRef(*moniker);
119 return S_OK;
122 static HRESULT WINAPI IHlink_fnQueryInterface(IHlink* iface, REFIID riid,
123 LPVOID *ppvObj)
125 HlinkImpl *This = impl_from_IHlink(iface);
127 TRACE ("(%p)->(%s,%p)\n", This, debugstr_guid (riid), ppvObj);
129 *ppvObj = NULL;
131 if (IsEqualIID(riid, &IID_IUnknown) || (IsEqualIID(riid, &IID_IHlink)))
132 *ppvObj = &This->IHlink_iface;
133 else if (IsEqualIID(riid, &IID_IPersistStream))
134 *ppvObj = &This->IPersistStream_iface;
135 else if (IsEqualIID(riid, &IID_IDataObject))
136 *ppvObj = &This->IDataObject_iface;
138 if (*ppvObj)
140 IUnknown_AddRef((IUnknown*)(*ppvObj));
141 return S_OK;
143 return E_NOINTERFACE;
146 static ULONG WINAPI IHlink_fnAddRef (IHlink* iface)
148 HlinkImpl *This = impl_from_IHlink(iface);
149 ULONG refCount = InterlockedIncrement(&This->ref);
151 TRACE("(%p)->(count=%u)\n", This, refCount - 1);
153 return refCount;
156 static ULONG WINAPI IHlink_fnRelease (IHlink* iface)
158 HlinkImpl *This = impl_from_IHlink(iface);
159 ULONG refCount = InterlockedDecrement(&This->ref);
161 TRACE("(%p)->(count=%u)\n", This, refCount + 1);
162 if (refCount)
163 return refCount;
165 TRACE("-- destroying IHlink (%p)\n", This);
166 heap_free(This->FriendlyName);
167 heap_free(This->TargetFrameName);
168 heap_free(This->Location);
169 if (This->Moniker)
170 IMoniker_Release(This->Moniker);
171 if (This->Site)
172 IHlinkSite_Release(This->Site);
173 heap_free(This);
174 return 0;
177 static HRESULT WINAPI IHlink_fnSetHlinkSite( IHlink* iface,
178 IHlinkSite* pihlSite, DWORD dwSiteData)
180 HlinkImpl *This = impl_from_IHlink(iface);
182 TRACE("(%p)->(%p %i)\n", This, pihlSite, dwSiteData);
184 if (This->Site)
185 IHlinkSite_Release(This->Site);
187 This->Site = pihlSite;
188 if (This->Site)
189 IHlinkSite_AddRef(This->Site);
191 This->SiteData = dwSiteData;
193 return S_OK;
196 static HRESULT WINAPI IHlink_fnGetHlinkSite( IHlink* iface,
197 IHlinkSite** ppihlSite, DWORD *pdwSiteData)
199 HlinkImpl *This = impl_from_IHlink(iface);
201 TRACE("(%p)->(%p %p)\n", This, ppihlSite, pdwSiteData);
203 *ppihlSite = This->Site;
205 if (This->Site) {
206 IHlinkSite_AddRef(This->Site);
207 *pdwSiteData = This->SiteData;
210 return S_OK;
213 static HRESULT WINAPI IHlink_fnSetMonikerReference( IHlink* iface,
214 DWORD rfHLSETF, IMoniker *pmkTarget, LPCWSTR pwzLocation)
216 HlinkImpl *This = impl_from_IHlink(iface);
218 TRACE("(%p)->(%i %p %s)\n", This, rfHLSETF, pmkTarget,
219 debugstr_w(pwzLocation));
221 if(rfHLSETF == 0)
222 return E_INVALIDARG;
223 if(!(rfHLSETF & (HLINKSETF_TARGET | HLINKSETF_LOCATION)))
224 return rfHLSETF;
226 if(rfHLSETF & HLINKSETF_TARGET){
227 if (This->Moniker)
228 IMoniker_Release(This->Moniker);
230 This->Moniker = pmkTarget;
231 if (This->Moniker)
233 IBindCtx *pbc;
234 LPOLESTR display_name;
235 IMoniker_AddRef(This->Moniker);
236 CreateBindCtx( 0, &pbc);
237 IMoniker_GetDisplayName(This->Moniker, pbc, NULL, &display_name);
238 IBindCtx_Release(pbc);
239 This->absolute = display_name && strchrW(display_name, ':');
240 CoTaskMemFree(display_name);
244 if(rfHLSETF & HLINKSETF_LOCATION){
245 heap_free(This->Location);
246 This->Location = hlink_strdupW( pwzLocation );
249 return S_OK;
252 static HRESULT WINAPI IHlink_fnSetStringReference(IHlink* iface,
253 DWORD grfHLSETF, LPCWSTR pwzTarget, LPCWSTR pwzLocation)
255 HlinkImpl *This = impl_from_IHlink(iface);
257 TRACE("(%p)->(%i %s %s)\n", This, grfHLSETF, debugstr_w(pwzTarget),
258 debugstr_w(pwzLocation));
260 if(grfHLSETF > (HLINKSETF_TARGET | HLINKSETF_LOCATION) &&
261 grfHLSETF < -(HLINKSETF_TARGET | HLINKSETF_LOCATION))
262 return grfHLSETF;
264 if (grfHLSETF & HLINKSETF_TARGET)
266 if (This->Moniker)
268 IMoniker_Release(This->Moniker);
269 This->Moniker = NULL;
271 if (pwzTarget && *pwzTarget)
273 IMoniker *pMon;
274 IBindCtx *pbc = NULL;
275 ULONG eaten;
276 HRESULT r;
278 r = CreateBindCtx(0, &pbc);
279 if (FAILED(r))
280 return E_OUTOFMEMORY;
282 r = MkParseDisplayName(pbc, pwzTarget, &eaten, &pMon);
283 IBindCtx_Release(pbc);
285 if (FAILED(r))
287 LPCWSTR p = strchrW(pwzTarget, ':');
288 if (p && (p - pwzTarget > 1))
289 r = CreateURLMoniker(NULL, pwzTarget, &pMon);
290 else
291 r = CreateFileMoniker(pwzTarget, &pMon);
292 if (FAILED(r))
294 ERR("couldn't create moniker for %s, failed with error 0x%08x\n",
295 debugstr_w(pwzTarget), r);
296 return r;
300 IHlink_SetMonikerReference(iface, HLINKSETF_TARGET, pMon, NULL);
301 IMoniker_Release(pMon);
305 if (grfHLSETF & HLINKSETF_LOCATION)
307 heap_free(This->Location);
308 This->Location = NULL;
309 if (pwzLocation && *pwzLocation)
310 This->Location = hlink_strdupW( pwzLocation );
313 return S_OK;
316 static HRESULT WINAPI IHlink_fnGetMonikerReference(IHlink* iface,
317 DWORD dwWhichRef, IMoniker **ppimkTarget, LPWSTR *ppwzLocation)
319 HlinkImpl *This = impl_from_IHlink(iface);
321 TRACE("(%p) -> (%i %p %p)\n", This, dwWhichRef, ppimkTarget,
322 ppwzLocation);
324 if (ppimkTarget)
326 HRESULT hres = __GetMoniker(This, ppimkTarget, dwWhichRef);
327 if (FAILED(hres))
329 if (ppwzLocation)
330 *ppwzLocation = NULL;
331 return hres;
335 if (ppwzLocation)
336 IHlink_GetStringReference(iface, dwWhichRef, NULL, ppwzLocation);
338 return S_OK;
341 static HRESULT WINAPI IHlink_fnGetStringReference (IHlink* iface,
342 DWORD dwWhichRef, LPWSTR *ppwzTarget, LPWSTR *ppwzLocation)
344 HlinkImpl *This = impl_from_IHlink(iface);
346 TRACE("(%p) -> (%i %p %p)\n", This, dwWhichRef, ppwzTarget, ppwzLocation);
348 if(dwWhichRef != -1 && dwWhichRef & ~(HLINKGETREF_DEFAULT | HLINKGETREF_ABSOLUTE | HLINKGETREF_RELATIVE))
350 if(ppwzTarget)
351 *ppwzTarget = NULL;
352 if(ppwzLocation)
353 *ppwzLocation = NULL;
354 return E_INVALIDARG;
357 if (ppwzTarget)
359 IMoniker* mon;
360 HRESULT hres = __GetMoniker(This, &mon, dwWhichRef);
361 if (FAILED(hres))
363 if (ppwzLocation)
364 *ppwzLocation = NULL;
365 return hres;
367 if (mon)
369 IBindCtx *pbc;
371 CreateBindCtx( 0, &pbc);
372 IMoniker_GetDisplayName(mon, pbc, NULL, ppwzTarget);
373 IBindCtx_Release(pbc);
374 IMoniker_Release(mon);
376 else
377 *ppwzTarget = NULL;
379 if (ppwzLocation)
380 *ppwzLocation = hlink_co_strdupW( This->Location );
382 TRACE("(Target: %s Location: %s)\n",
383 (ppwzTarget)?debugstr_w(*ppwzTarget):"<NULL>",
384 (ppwzLocation)?debugstr_w(*ppwzLocation):"<NULL>");
386 return S_OK;
389 static HRESULT WINAPI IHlink_fnSetFriendlyName (IHlink *iface,
390 LPCWSTR pwzFriendlyName)
392 HlinkImpl *This = impl_from_IHlink(iface);
394 TRACE("(%p) -> (%s)\n", This, debugstr_w(pwzFriendlyName));
396 heap_free(This->FriendlyName);
397 This->FriendlyName = hlink_strdupW( pwzFriendlyName );
399 return S_OK;
402 static HRESULT WINAPI IHlink_fnGetFriendlyName (IHlink* iface,
403 DWORD grfHLFNAMEF, LPWSTR* ppwzFriendlyName)
405 HlinkImpl *This = impl_from_IHlink(iface);
407 TRACE("(%p) -> (%i %p)\n", This, grfHLFNAMEF, ppwzFriendlyName);
409 /* FIXME: Only using explicitly set and cached friendly names */
411 if (This->FriendlyName)
412 *ppwzFriendlyName = hlink_co_strdupW( This->FriendlyName );
413 else
415 IMoniker *moniker;
416 HRESULT hres = __GetMoniker(This, &moniker, HLINKGETREF_DEFAULT);
417 if (FAILED(hres))
419 *ppwzFriendlyName = NULL;
420 return hres;
422 if (moniker)
424 IBindCtx *bcxt;
425 CreateBindCtx(0, &bcxt);
427 IMoniker_GetDisplayName(moniker, bcxt, NULL, ppwzFriendlyName);
428 IBindCtx_Release(bcxt);
429 IMoniker_Release(moniker);
431 else
432 *ppwzFriendlyName = NULL;
435 return S_OK;
438 static HRESULT WINAPI IHlink_fnSetTargetFrameName(IHlink* iface,
439 LPCWSTR pwzTargetFramename)
441 HlinkImpl *This = impl_from_IHlink(iface);
442 TRACE("(%p)->(%s)\n", This, debugstr_w(pwzTargetFramename));
444 heap_free(This->TargetFrameName);
445 This->TargetFrameName = hlink_strdupW( pwzTargetFramename );
447 return S_OK;
450 static HRESULT WINAPI IHlink_fnGetTargetFrameName(IHlink* iface,
451 LPWSTR *ppwzTargetFrameName)
453 HlinkImpl *This = impl_from_IHlink(iface);
455 TRACE("(%p)->(%p)\n", This, ppwzTargetFrameName);
457 if(!This->TargetFrameName) {
458 *ppwzTargetFrameName = NULL;
459 return S_FALSE;
462 *ppwzTargetFrameName = hlink_co_strdupW( This->TargetFrameName );
463 if(!*ppwzTargetFrameName)
464 return E_OUTOFMEMORY;
466 return S_OK;
469 static HRESULT WINAPI IHlink_fnGetMiscStatus(IHlink* iface, DWORD* pdwStatus)
471 FIXME("\n");
472 return E_NOTIMPL;
475 static HRESULT WINAPI IHlink_fnNavigate(IHlink* iface, DWORD grfHLNF, LPBC pbc,
476 IBindStatusCallback *pbsc, IHlinkBrowseContext *phbc)
478 HlinkImpl *This = impl_from_IHlink(iface);
479 IMoniker *mon = NULL;
480 HRESULT r;
482 FIXME("Semi-Stub:(%p)->(%i %p %p %p)\n", This, grfHLNF, pbc, pbsc, phbc);
484 r = __GetMoniker(This, &mon, HLINKGETREF_ABSOLUTE);
485 TRACE("Moniker %p\n", mon);
487 if (SUCCEEDED(r))
489 IBindCtx *bcxt;
490 IUnknown *unk = NULL;
491 IHlinkTarget *target;
493 CreateBindCtx(0, &bcxt);
495 RegisterBindStatusCallback(bcxt, pbsc, NULL, 0);
497 r = IMoniker_BindToObject(mon, bcxt, NULL, &IID_IUnknown, (void**)&unk);
498 if (r == S_OK)
500 r = IUnknown_QueryInterface(unk, &IID_IHlinkTarget, (void**)&target);
501 IUnknown_Release(unk);
503 if (r == S_OK)
505 IHlinkTarget_SetBrowseContext(target, phbc);
506 r = 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_fnSetAdditionalParams(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_fnSetAdditionalParams,
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) != 0;
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,
946 HRESULT HLink_Constructor(IUnknown *pUnkOuter, REFIID riid, void **ppv)
948 HlinkImpl * hl;
950 TRACE("unkOut=%p riid=%s\n", pUnkOuter, debugstr_guid(riid));
951 *ppv = NULL;
953 if (pUnkOuter)
954 return CLASS_E_NOAGGREGATION;
956 hl = heap_alloc_zero(sizeof(HlinkImpl));
957 if (!hl)
958 return E_OUTOFMEMORY;
960 hl->ref = 1;
961 hl->IHlink_iface.lpVtbl = &hlvt;
962 hl->IPersistStream_iface.lpVtbl = &psvt;
963 hl->IDataObject_iface.lpVtbl = &dovt;
965 *ppv = hl;
966 return S_OK;