hlink: Use an iface instead of a vtbl pointer in HlinkBCImpl.
[wine] / dlls / hlink / hlink_main.c
1 /*
2  * Implementation of hyperlinking (hlink.dll)
3  *
4  * Copyright 2005 Aric Stewart 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 "hlink_private.h"
22
23 #include "winreg.h"
24 #include "rpcproxy.h"
25 #include "hlguids.h"
26
27 #include "wine/debug.h"
28
29 WINE_DEFAULT_DEBUG_CHANNEL(hlink);
30
31 static HINSTANCE instance;
32
33 typedef HRESULT (CALLBACK *LPFNCREATEINSTANCE)(IUnknown*, REFIID, LPVOID*);
34
35 typedef struct
36 {
37     const IClassFactoryVtbl *lpVtbl;
38     LPFNCREATEINSTANCE      lpfnCI;
39 } CFImpl;
40
41 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
42 {
43     TRACE("%p %d %p\n", hinstDLL, fdwReason, lpvReserved);
44
45     switch (fdwReason)
46     {
47     case DLL_PROCESS_ATTACH:
48         instance = hinstDLL;
49         DisableThreadLibraryCalls(hinstDLL);
50         break;
51     case DLL_PROCESS_DETACH:
52         break;
53     }
54     return TRUE;
55 }
56
57 /***********************************************************************
58  *             DllCanUnloadNow (HLINK.@)
59  */
60 HRESULT WINAPI DllCanUnloadNow( void )
61 {
62     return S_OK;
63 }
64
65 /***********************************************************************
66  *             HlinkCreateFromMoniker (HLINK.@)
67  */
68 HRESULT WINAPI HlinkCreateFromMoniker( IMoniker *pimkTrgt, LPCWSTR pwzLocation,
69         LPCWSTR pwzFriendlyName, IHlinkSite* pihlsite, DWORD dwSiteData,
70         IUnknown* piunkOuter, REFIID riid, void** ppvObj)
71 {
72     IHlink *hl = NULL;
73     HRESULT r = S_OK;
74
75     TRACE("%p %s %s %p %i %p %s %p\n", pimkTrgt, debugstr_w(pwzLocation),
76             debugstr_w(pwzFriendlyName), pihlsite, dwSiteData, piunkOuter,
77             debugstr_guid(riid), ppvObj);
78
79     r = CoCreateInstance(&CLSID_StdHlink, piunkOuter, CLSCTX_INPROC_SERVER, riid, (LPVOID*)&hl);
80     if (FAILED(r))
81         return r;
82
83     IHlink_SetMonikerReference(hl, HLINKSETF_LOCATION | HLINKSETF_TARGET, pimkTrgt, pwzLocation);
84
85     if (pwzFriendlyName)
86         IHlink_SetFriendlyName(hl, pwzFriendlyName);
87     if (pihlsite)
88         IHlink_SetHlinkSite(hl, pihlsite, dwSiteData);
89
90     *ppvObj = hl;
91
92     TRACE("Returning %i\n",r);
93
94     return r;
95 }
96
97 /***********************************************************************
98  *             HlinkCreateFromString (HLINK.@)
99  */
100 HRESULT WINAPI HlinkCreateFromString( LPCWSTR pwzTarget, LPCWSTR pwzLocation,
101         LPCWSTR pwzFriendlyName, IHlinkSite* pihlsite, DWORD dwSiteData,
102         IUnknown* piunkOuter, REFIID riid, void** ppvObj)
103 {
104     IHlink *hl = NULL;
105     HRESULT r = S_OK;
106     WCHAR *hash, *tgt;
107     const WCHAR *loc;
108
109     TRACE("%s %s %s %p %i %p %s %p\n", debugstr_w(pwzTarget),
110             debugstr_w(pwzLocation), debugstr_w(pwzFriendlyName), pihlsite,
111             dwSiteData, piunkOuter, debugstr_guid(riid), ppvObj);
112
113     r = CoCreateInstance(&CLSID_StdHlink, piunkOuter, CLSCTX_INPROC_SERVER, riid, (LPVOID*)&hl);
114     if (FAILED(r))
115         return r;
116
117     if (pwzTarget)
118     {
119         hash = strchrW(pwzTarget, '#');
120         if (hash)
121         {
122             if (hash == pwzTarget)
123                 tgt = NULL;
124             else
125             {
126                 int tgt_len = hash - pwzTarget;
127                 tgt = heap_alloc((tgt_len + 1) * sizeof(WCHAR));
128                 if (!tgt)
129                     return E_OUTOFMEMORY;
130                 memcpy(tgt, pwzTarget, tgt_len * sizeof(WCHAR));
131                 tgt[tgt_len] = 0;
132             }
133             if (!pwzLocation)
134                 loc = hash + 1;
135             else
136                 loc = pwzLocation;
137         }
138         else
139         {
140             tgt = hlink_strdupW(pwzTarget);
141             if (!tgt)
142                 return E_OUTOFMEMORY;
143             loc = pwzLocation;
144         }
145     }
146     else
147     {
148         tgt = NULL;
149         loc = pwzLocation;
150     }
151
152     IHlink_SetStringReference(hl, HLINKSETF_TARGET | HLINKSETF_LOCATION, tgt, loc);
153
154     heap_free(tgt);
155
156     if (pwzFriendlyName)
157         IHlink_SetFriendlyName(hl, pwzFriendlyName);
158
159     if (pihlsite)
160         IHlink_SetHlinkSite(hl, pihlsite, dwSiteData);
161
162     TRACE("Returning %i\n",r);
163     *ppvObj = hl;
164
165     return r;
166 }
167
168
169 /***********************************************************************
170  *             HlinkCreateBrowseContext (HLINK.@)
171  */
172 HRESULT WINAPI HlinkCreateBrowseContext( IUnknown* piunkOuter, REFIID riid, void** ppvObj)
173 {
174     HRESULT r = S_OK;
175
176     TRACE("%p %s %p\n", piunkOuter, debugstr_guid(riid), ppvObj);
177
178     r = CoCreateInstance(&CLSID_StdHlinkBrowseContext, piunkOuter, CLSCTX_INPROC_SERVER, riid, ppvObj);
179
180     TRACE("returning %i\n",r);
181
182     return r;
183 }
184
185 /***********************************************************************
186  *             HlinkNavigate (HLINK.@)
187  */
188 HRESULT WINAPI HlinkNavigate(IHlink *phl, IHlinkFrame *phlFrame,
189         DWORD grfHLNF, LPBC pbc, IBindStatusCallback *pbsc,
190         IHlinkBrowseContext *phlbc)
191 {
192     HRESULT r = S_OK;
193
194     TRACE("%p %p %i %p %p %p\n", phl, phlFrame, grfHLNF, pbc, pbsc, phlbc);
195
196     if (phlFrame)
197         r = IHlinkFrame_Navigate(phlFrame, grfHLNF, pbc, pbsc, phl);
198     else if (phl)
199         r = IHlink_Navigate(phl, grfHLNF, pbc, pbsc, phlbc);
200
201     return r;
202 }
203
204 /***********************************************************************
205  *             HlinkOnNavigate (HLINK.@)
206  */
207 HRESULT WINAPI HlinkOnNavigate( IHlinkFrame *phlFrame,
208         IHlinkBrowseContext* phlbc, DWORD grfHLNF, IMoniker *pmkTarget,
209         LPCWSTR pwzLocation, LPCWSTR pwzFriendlyName, ULONG* puHLID)
210 {
211     HRESULT r = S_OK;
212
213     TRACE("%p %p %i %p %s %s %p\n",phlFrame, phlbc, grfHLNF, pmkTarget,
214             debugstr_w(pwzLocation), debugstr_w(pwzFriendlyName), puHLID);
215
216     r = IHlinkBrowseContext_OnNavigateHlink(phlbc, grfHLNF, pmkTarget,
217             pwzLocation, pwzFriendlyName, puHLID);
218
219     if (phlFrame)
220         r = IHlinkFrame_OnNavigate(phlFrame,grfHLNF,pmkTarget, pwzLocation,
221                 pwzFriendlyName, 0);
222
223     return r;
224 }
225
226 /***********************************************************************
227  *             HlinkCreateFromData (HLINK.@)
228  */
229 HRESULT WINAPI HlinkCreateFromData(IDataObject *piDataObj,
230         IHlinkSite *pihlsite, DWORD dwSiteData, IUnknown *piunkOuter,
231         REFIID riid, void **ppvObj)
232 {
233     FIXME("%p %p %d %p %p %p\n",
234           piDataObj, pihlsite, dwSiteData, piunkOuter, riid, ppvObj);
235     *ppvObj = NULL;
236     return E_NOTIMPL;
237 }
238
239 /***********************************************************************
240  *             HlinkQueryCreateFromData (HLINK.@)
241  */
242 HRESULT WINAPI HlinkQueryCreateFromData(IDataObject* piDataObj)
243 {
244     FIXME("%p\n", piDataObj);
245     return E_NOTIMPL;
246 }
247
248 /***********************************************************************
249  *             HlinkNavigateToStringReference (HLINK.@)
250  */
251 HRESULT WINAPI HlinkNavigateToStringReference( LPCWSTR pwzTarget,
252         LPCWSTR pwzLocation, IHlinkSite *pihlsite, DWORD dwSiteData,
253         IHlinkFrame *pihlframe, DWORD grfHLNF, LPBC pibc,
254         IBindStatusCallback *pibsc, IHlinkBrowseContext *pihlbc)
255 {
256     HRESULT r;
257     IHlink *hlink = NULL;
258
259     FIXME("%s %s %p %08x %p %08x %p %p %p\n",
260           debugstr_w(pwzTarget), debugstr_w(pwzLocation), pihlsite,
261           dwSiteData, pihlframe, grfHLNF, pibc, pibsc, pihlbc);
262
263     r = HlinkCreateFromString( pwzTarget, pwzLocation, NULL, pihlsite,
264                                dwSiteData, NULL, &IID_IHlink, (LPVOID*) &hlink );
265     if (SUCCEEDED(r))
266         r = HlinkNavigate(hlink, pihlframe, grfHLNF, pibc, pibsc, pihlbc);
267
268     return r;
269 }
270
271 /***********************************************************************
272  *             HlinkIsShortcut (HLINK.@)
273  */
274 HRESULT WINAPI HlinkIsShortcut(LPCWSTR pwzFileName)
275 {
276     int len;
277
278     static const WCHAR url_ext[] = {'.','u','r','l',0};
279
280     TRACE("(%s)\n", debugstr_w(pwzFileName));
281
282     if(!pwzFileName)
283         return E_INVALIDARG;
284
285     len = strlenW(pwzFileName)-4;
286     if(len < 0)
287         return S_FALSE;
288
289     return strcmpiW(pwzFileName+len, url_ext) ? S_FALSE : S_OK;
290 }
291
292 /***********************************************************************
293  *             HlinkGetSpecialReference (HLINK.@)
294  */
295 HRESULT WINAPI HlinkGetSpecialReference(ULONG uReference, LPWSTR *ppwzReference)
296 {
297     DWORD res, type, size = 100;
298     LPCWSTR value_name;
299     WCHAR *buf;
300     HKEY hkey;
301
302     static const WCHAR start_pageW[] = {'S','t','a','r','t',' ','P','a','g','e',0};
303     static const WCHAR search_pageW[] = {'S','e','a','r','c','h',' ','P','a','g','e',0};
304
305     static const WCHAR ie_main_keyW[] =
306         {'S','o','f','t','w','a','r','e',
307          '\\','M','i','c','r','o','s','o','f','t','\\',
308          'I','n','t','e','r','n','e','t',' ','E','x','p','l','o','r','e','r',
309          '\\','M','a','i','n',0};
310
311     TRACE("(%u %p)\n", uReference, ppwzReference);
312
313     *ppwzReference = NULL;
314
315     switch(uReference) {
316     case HLSR_HOME:
317         value_name = start_pageW;
318         break;
319     case HLSR_SEARCHPAGE:
320         value_name = search_pageW;
321         break;
322     case HLSR_HISTORYFOLDER:
323         return E_NOTIMPL;
324     default:
325         return E_INVALIDARG;
326     }
327
328     res = RegOpenKeyW(HKEY_CURRENT_USER, ie_main_keyW, &hkey);
329     if(res != ERROR_SUCCESS) {
330         WARN("Could not open key: %u\n", res);
331         return HRESULT_FROM_WIN32(res);
332     }
333
334     buf = CoTaskMemAlloc(size);
335     res = RegQueryValueExW(hkey, value_name, NULL, &type, (PBYTE)buf, &size);
336     buf = CoTaskMemRealloc(buf, size);
337     if(res == ERROR_MORE_DATA)
338         res = RegQueryValueExW(hkey, value_name, NULL, &type, (PBYTE)buf, &size);
339     RegCloseKey(hkey);
340     if(res != ERROR_SUCCESS) {
341         WARN("Could not query value %s: %u\n", debugstr_w(value_name), res);
342         CoTaskMemFree(buf);
343         return HRESULT_FROM_WIN32(res);
344     }
345
346     *ppwzReference = buf;
347     return S_OK;
348 }
349
350 /***********************************************************************
351  *             HlinkTranslateURL (HLINK.@)
352  */
353 HRESULT WINAPI HlinkTranslateURL(LPCWSTR pwzURL, DWORD grfFlags, LPWSTR *ppwzTranslatedURL)
354 {
355     FIXME("(%s %08x %p)\n", debugstr_w(pwzURL), grfFlags, ppwzTranslatedURL);
356     return E_NOTIMPL;
357 }
358
359 /***********************************************************************
360  *             HlinkUpdateStackItem (HLINK.@)
361  */
362 HRESULT WINAPI HlinkUpdateStackItem(IHlinkFrame *pihlframe, IHlinkBrowseContext *pihlbc,
363         ULONG uHLID, IMoniker *pimkTrgt, LPCWSTR pwzLocation, LPCWSTR pwzFriendlyName)
364 {
365     FIXME("(%p %p %u %p %s %s)\n", pihlframe, pihlbc, uHLID, pimkTrgt, debugstr_w(pwzLocation),
366           debugstr_w(pwzFriendlyName));
367     return E_NOTIMPL;
368 }
369
370 /***********************************************************************
371  *             HlinkParseDisplayName (HLINK.@)
372  */
373 HRESULT WINAPI HlinkParseDisplayName(LPBC pibc, LPCWSTR pwzDisplayName, BOOL fNoForceAbs,
374         ULONG *pcchEaten, IMoniker **ppimk)
375 {
376     HRESULT hres;
377
378     TRACE("(%p %s %x %p %p)\n", pibc, debugstr_w(pwzDisplayName), fNoForceAbs, pcchEaten, ppimk);
379
380     if(fNoForceAbs)
381         FIXME("Unsupported fNoForceAbs\n");
382
383     hres = MkParseDisplayNameEx(pibc, pwzDisplayName, pcchEaten, ppimk);
384     if(SUCCEEDED(hres))
385         return hres;
386
387     hres = MkParseDisplayName(pibc, pwzDisplayName, pcchEaten, ppimk);
388     if(SUCCEEDED(hres))
389         return hres;
390
391     hres = CreateFileMoniker(pwzDisplayName, ppimk);
392     if(SUCCEEDED(hres))
393         *pcchEaten = strlenW(pwzDisplayName);
394
395     return hres;
396 }
397
398 /***********************************************************************
399  *             HlinkResolveMonikerForData (HLINK.@)
400  */
401 HRESULT WINAPI HlinkResolveMonikerForData(LPMONIKER pimkReference, DWORD reserved, LPBC pibc,
402         ULONG cFmtetc, FORMATETC *rgFmtetc, IBindStatusCallback *pibsc, LPMONIKER pimkBase)
403 {
404     LPOLESTR name = NULL;
405     IBindCtx *bctx;
406     DWORD mksys = 0;
407     void *obj = NULL;
408     HRESULT hres;
409
410     TRACE("(%p %x %p %d %p %p %p)\n", pimkReference, reserved, pibc, cFmtetc, rgFmtetc, pibsc, pimkBase);
411
412     if(cFmtetc || rgFmtetc || pimkBase)
413         FIXME("Unsupported args\n");
414
415     hres = RegisterBindStatusCallback(pibc, pibsc, NULL /* FIXME */, 0);
416     if(FAILED(hres))
417         return hres;
418
419     hres = IMoniker_IsSystemMoniker(pimkReference, &mksys);
420     if(SUCCEEDED(hres) && mksys != MKSYS_URLMONIKER)
421         WARN("sysmk = %x\n", mksys);
422
423     /* FIXME: What is it for? */
424     CreateBindCtx(0, &bctx);
425     hres = IMoniker_GetDisplayName(pimkReference, bctx, NULL, &name);
426     IBindCtx_Release(bctx);
427     if(SUCCEEDED(hres)) {
428         TRACE("got display name %s\n", debugstr_w(name));
429         CoTaskMemFree(name);
430     }
431
432     return IMoniker_BindToStorage(pimkReference, pibc, NULL, &IID_IUnknown, &obj);
433 }
434
435 /***********************************************************************
436  *             HlinkClone (HLINK.@)
437  */
438 HRESULT WINAPI HlinkClone(IHlink *hlink, REFIID riid, IHlinkSite *hls,
439         DWORD site_data, void **obj)
440 {
441     IMoniker *mk, *clone_mk = NULL;
442     WCHAR *loc, *name = NULL;
443     HRESULT hres;
444
445     if(!hlink || !riid || !obj)
446         return E_INVALIDARG;
447
448     *obj = NULL;
449
450     hres = IHlink_GetMonikerReference(hlink, HLINKGETREF_DEFAULT, &mk, &loc);
451     if(FAILED(hres))
452         return hres;
453
454     if(mk) {
455         IStream *strm;
456         LARGE_INTEGER lgint;
457
458         hres = CreateStreamOnHGlobal(NULL, TRUE, &strm);
459         if(FAILED(hres)) {
460             IMoniker_Release(mk);
461             goto cleanup;
462         }
463
464         hres = OleSaveToStream((IPersistStream*)mk, strm);
465         if(FAILED(hres)) {
466             IStream_Release(strm);
467             IMoniker_Release(mk);
468             goto cleanup;
469         }
470         IMoniker_Release(mk);
471
472         lgint.QuadPart = 0;
473         hres = IStream_Seek(strm, lgint, STREAM_SEEK_SET, NULL);
474         if(FAILED(hres)) {
475             IStream_Release(strm);
476             goto cleanup;
477         }
478
479         hres = OleLoadFromStream(strm, &IID_IMoniker, (void**)&clone_mk);
480         IStream_Release(strm);
481         if(FAILED(hres))
482             goto cleanup;
483     }
484
485     hres = IHlink_GetFriendlyName(hlink, HLFNAMEF_DEFAULT, &name);
486     if(FAILED(hres))
487         goto cleanup;
488
489     hres = HlinkCreateFromMoniker(clone_mk, loc, name, hls, site_data, NULL,
490             &IID_IHlink, obj);
491
492 cleanup:
493     if(clone_mk)
494         IMoniker_Release(clone_mk);
495     CoTaskMemFree(loc);
496     CoTaskMemFree(name);
497     return hres;
498 }
499
500 static HRESULT WINAPI HLinkCF_fnQueryInterface ( LPCLASSFACTORY iface,
501         REFIID riid, LPVOID *ppvObj)
502 {
503     CFImpl *This = (CFImpl *)iface;
504
505     TRACE("(%p)->(%s)\n",This,debugstr_guid(riid));
506
507     *ppvObj = NULL;
508
509     if (IsEqualIID(riid, &IID_IUnknown) ||
510         IsEqualIID(riid, &IID_IClassFactory))
511     {
512         *ppvObj = This;
513         return S_OK;
514     }
515
516     TRACE("-- E_NOINTERFACE\n");
517     return E_NOINTERFACE;
518 }
519
520 static ULONG WINAPI HLinkCF_fnAddRef (LPCLASSFACTORY iface)
521 {
522     return 2;
523 }
524
525 static ULONG WINAPI HLinkCF_fnRelease(LPCLASSFACTORY iface)
526 {
527     return 1;
528 }
529
530 static HRESULT WINAPI HLinkCF_fnCreateInstance( LPCLASSFACTORY iface,
531         LPUNKNOWN pUnkOuter, REFIID riid, LPVOID *ppvObject)
532 {
533     CFImpl *This = (CFImpl *)iface;
534
535     TRACE("%p->(%p,%s,%p)\n", This, pUnkOuter, debugstr_guid(riid), ppvObject);
536
537     *ppvObject = NULL;
538
539     return This->lpfnCI(pUnkOuter, riid, ppvObject);
540 }
541
542 static HRESULT WINAPI HLinkCF_fnLockServer(LPCLASSFACTORY iface, BOOL fLock)
543 {
544     FIXME("%p %d\n", iface, fLock);
545     return E_NOTIMPL;
546 }
547
548 static const IClassFactoryVtbl hlcfvt =
549 {
550     HLinkCF_fnQueryInterface,
551     HLinkCF_fnAddRef,
552     HLinkCF_fnRelease,
553     HLinkCF_fnCreateInstance,
554     HLinkCF_fnLockServer
555 };
556
557 static CFImpl HLink_cf = { &hlcfvt, HLink_Constructor };
558 static CFImpl HLinkBrowseContext_cf = { &hlcfvt, HLinkBrowseContext_Constructor };
559
560 /***********************************************************************
561  *             DllGetClassObject (HLINK.@)
562  */
563 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID iid, LPVOID *ppv)
564 {
565     IClassFactory   *pcf = NULL;
566
567     TRACE("%s %s %p\n", debugstr_guid(rclsid), debugstr_guid(iid), ppv);
568
569     if (!ppv)
570         return E_INVALIDARG;
571     *ppv = NULL;
572
573     if (IsEqualIID(rclsid, &CLSID_StdHlink))
574         pcf = (IClassFactory*) &HLink_cf;
575     else if (IsEqualIID(rclsid, &CLSID_StdHlinkBrowseContext))
576         pcf = (IClassFactory*) &HLinkBrowseContext_cf;
577     else
578         return CLASS_E_CLASSNOTAVAILABLE;
579
580     return IClassFactory_QueryInterface(pcf, iid, ppv);
581 }
582
583 /***********************************************************************
584  *              DllRegisterServer (HLINK.@)
585  */
586 HRESULT WINAPI DllRegisterServer(void)
587 {
588     return __wine_register_resources( instance, NULL );
589 }
590
591 /***********************************************************************
592  *              DllUnregisterServer (HLINK.@)
593  */
594 HRESULT WINAPI DllUnregisterServer(void)
595 {
596     return __wine_unregister_resources( instance, NULL );
597 }