mshtml: Mark function as hidden.
[wine] / dlls / mshtml / htmllocation.c
1 /*
2  * Copyright 2008 Jacek Caban for CodeWeavers
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17  */
18
19 #include <stdarg.h>
20
21 #define COBJMACROS
22
23 #include "windef.h"
24 #include "winbase.h"
25 #include "winuser.h"
26 #include "winreg.h"
27 #include "ole2.h"
28 #include "wininet.h"
29 #include "shlwapi.h"
30
31 #include "wine/debug.h"
32
33 #include "mshtml_private.h"
34 #include "resource.h"
35
36 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
37
38 static HRESULT get_url(HTMLLocation *This, const WCHAR **ret)
39 {
40     if(!This->window || !This->window->url) {
41         FIXME("No current URL\n");
42         return E_NOTIMPL;
43     }
44
45     *ret = This->window->url;
46     return S_OK;
47 }
48
49 static HRESULT get_url_components(HTMLLocation *This, URL_COMPONENTSW *url)
50 {
51     const WCHAR *doc_url;
52     HRESULT hres;
53
54     hres = get_url(This, &doc_url);
55     if(FAILED(hres))
56         return hres;
57
58     if(!InternetCrackUrlW(doc_url, 0, 0, url)) {
59         FIXME("InternetCrackUrlW failed: 0x%08x\n", GetLastError());
60         SetLastError(0);
61         return E_FAIL;
62     }
63
64     return S_OK;
65 }
66
67 static inline HTMLLocation *impl_from_IHTMLLocation(IHTMLLocation *iface)
68 {
69     return CONTAINING_RECORD(iface, HTMLLocation, IHTMLLocation_iface);
70 }
71
72 static HRESULT WINAPI HTMLLocation_QueryInterface(IHTMLLocation *iface, REFIID riid, void **ppv)
73 {
74     HTMLLocation *This = impl_from_IHTMLLocation(iface);
75
76     *ppv = NULL;
77
78     if(IsEqualGUID(&IID_IUnknown, riid)) {
79         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
80         *ppv = &This->IHTMLLocation_iface;
81     }else if(IsEqualGUID(&IID_IHTMLLocation, riid)) {
82         TRACE("(%p)->(IID_IHTMLLocation %p)\n", This, ppv);
83         *ppv = &This->IHTMLLocation_iface;
84     }else if(dispex_query_interface(&This->dispex, riid, ppv)) {
85         return *ppv ? S_OK : E_NOINTERFACE;
86     }
87
88     if(*ppv) {
89         IUnknown_AddRef((IUnknown*)*ppv);
90         return S_OK;
91     }
92
93     WARN("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
94     return E_NOINTERFACE;
95 }
96
97 static ULONG WINAPI HTMLLocation_AddRef(IHTMLLocation *iface)
98 {
99     HTMLLocation *This = impl_from_IHTMLLocation(iface);
100     LONG ref = InterlockedIncrement(&This->ref);
101
102     TRACE("(%p) ref=%d\n", This, ref);
103
104     return ref;
105 }
106
107 static ULONG WINAPI HTMLLocation_Release(IHTMLLocation *iface)
108 {
109     HTMLLocation *This = impl_from_IHTMLLocation(iface);
110     LONG ref = InterlockedDecrement(&This->ref);
111
112     TRACE("(%p) ref=%d\n", This, ref);
113
114     if(!ref) {
115         if(This->window)
116             This->window->location = NULL;
117         release_dispex(&This->dispex);
118         heap_free(This);
119     }
120
121     return ref;
122 }
123
124 static HRESULT WINAPI HTMLLocation_GetTypeInfoCount(IHTMLLocation *iface, UINT *pctinfo)
125 {
126     HTMLLocation *This = impl_from_IHTMLLocation(iface);
127     return IDispatchEx_GetTypeInfoCount(&This->dispex.IDispatchEx_iface, pctinfo);
128 }
129
130 static HRESULT WINAPI HTMLLocation_GetTypeInfo(IHTMLLocation *iface, UINT iTInfo,
131                                               LCID lcid, ITypeInfo **ppTInfo)
132 {
133     HTMLLocation *This = impl_from_IHTMLLocation(iface);
134     return IDispatchEx_GetTypeInfo(&This->dispex.IDispatchEx_iface, iTInfo, lcid, ppTInfo);
135 }
136
137 static HRESULT WINAPI HTMLLocation_GetIDsOfNames(IHTMLLocation *iface, REFIID riid,
138                                                 LPOLESTR *rgszNames, UINT cNames,
139                                                 LCID lcid, DISPID *rgDispId)
140 {
141     HTMLLocation *This = impl_from_IHTMLLocation(iface);
142     return IDispatchEx_GetIDsOfNames(&This->dispex.IDispatchEx_iface, riid, rgszNames, cNames,
143             lcid, rgDispId);
144 }
145
146 static HRESULT WINAPI HTMLLocation_Invoke(IHTMLLocation *iface, DISPID dispIdMember,
147                             REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
148                             VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
149 {
150     HTMLLocation *This = impl_from_IHTMLLocation(iface);
151     return IDispatchEx_Invoke(&This->dispex.IDispatchEx_iface, dispIdMember, riid, lcid,
152             wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
153 }
154
155 static HRESULT WINAPI HTMLLocation_put_href(IHTMLLocation *iface, BSTR v)
156 {
157     HTMLLocation *This = impl_from_IHTMLLocation(iface);
158
159     TRACE("(%p)->(%s)\n", This, debugstr_w(v));
160
161     if(!This->window) {
162         FIXME("No window available\n");
163         return E_FAIL;
164     }
165
166     return navigate_url(This->window, v, This->window->url);
167 }
168
169 static HRESULT WINAPI HTMLLocation_get_href(IHTMLLocation *iface, BSTR *p)
170 {
171     HTMLLocation *This = impl_from_IHTMLLocation(iface);
172     URL_COMPONENTSW url = {sizeof(URL_COMPONENTSW)};
173     WCHAR *buf = NULL, *url_path = NULL;
174     HRESULT hres, ret;
175     DWORD len = 0;
176     int i;
177
178     TRACE("(%p)->(%p)\n", This, p);
179
180     if(!p)
181         return E_POINTER;
182
183     url.dwSchemeLength = 1;
184     url.dwHostNameLength = 1;
185     url.dwUrlPathLength = 1;
186     url.dwExtraInfoLength = 1;
187     hres = get_url_components(This, &url);
188     if(FAILED(hres))
189         return hres;
190
191     switch(url.nScheme) {
192     case INTERNET_SCHEME_FILE:
193         {
194             /* prepend a slash */
195             url_path = HeapAlloc(GetProcessHeap(), 0, (url.dwUrlPathLength + 1) * sizeof(WCHAR));
196             if(!url_path)
197                 return E_OUTOFMEMORY;
198             url_path[0] = '/';
199             memcpy(url_path + 1, url.lpszUrlPath, url.dwUrlPathLength * sizeof(WCHAR));
200             url.lpszUrlPath = url_path;
201             url.dwUrlPathLength = url.dwUrlPathLength + 1;
202         }
203         break;
204
205     case INTERNET_SCHEME_HTTP:
206     case INTERNET_SCHEME_HTTPS:
207     case INTERNET_SCHEME_FTP:
208         if(!url.dwUrlPathLength) {
209             /* add a slash if it's blank */
210             url_path = url.lpszUrlPath = HeapAlloc(GetProcessHeap(), 0, 1 * sizeof(WCHAR));
211             if(!url.lpszUrlPath)
212                 return E_OUTOFMEMORY;
213             url.lpszUrlPath[0] = '/';
214             url.dwUrlPathLength = 1;
215         }
216         break;
217
218     default:
219         break;
220     }
221
222     /* replace \ with / */
223     for(i = 0; i < url.dwUrlPathLength; ++i)
224         if(url.lpszUrlPath[i] == '\\')
225             url.lpszUrlPath[i] = '/';
226
227     if(InternetCreateUrlW(&url, ICU_ESCAPE, NULL, &len)) {
228         FIXME("InternetCreateUrl succeeded with NULL buffer?\n");
229         ret = E_FAIL;
230         goto cleanup;
231     }
232
233     if(GetLastError() != ERROR_INSUFFICIENT_BUFFER) {
234         FIXME("InternetCreateUrl failed with error: %08x\n", GetLastError());
235         SetLastError(0);
236         ret = E_FAIL;
237         goto cleanup;
238     }
239     SetLastError(0);
240
241     buf = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
242     if(!buf) {
243         ret = E_OUTOFMEMORY;
244         goto cleanup;
245     }
246
247     if(!InternetCreateUrlW(&url, ICU_ESCAPE, buf, &len)) {
248         FIXME("InternetCreateUrl failed with error: %08x\n", GetLastError());
249         SetLastError(0);
250         ret = E_FAIL;
251         goto cleanup;
252     }
253
254     *p = SysAllocStringLen(buf, len);
255     if(!*p) {
256         ret = E_OUTOFMEMORY;
257         goto cleanup;
258     }
259
260     ret = S_OK;
261
262 cleanup:
263     HeapFree(GetProcessHeap(), 0, buf);
264     HeapFree(GetProcessHeap(), 0, url_path);
265
266     return ret;
267 }
268
269 static HRESULT WINAPI HTMLLocation_put_protocol(IHTMLLocation *iface, BSTR v)
270 {
271     HTMLLocation *This = impl_from_IHTMLLocation(iface);
272     FIXME("(%p)->(%s)\n", This, debugstr_w(v));
273     return E_NOTIMPL;
274 }
275
276 static HRESULT WINAPI HTMLLocation_get_protocol(IHTMLLocation *iface, BSTR *p)
277 {
278     HTMLLocation *This = impl_from_IHTMLLocation(iface);
279     BSTR protocol, ret;
280     unsigned len;
281     HRESULT hres;
282
283     TRACE("(%p)->(%p)\n", This, p);
284
285     if(!p)
286         return E_POINTER;
287
288     if(!This->window || !This->window->uri) {
289         FIXME("No current URI\n");
290         return E_NOTIMPL;
291     }
292
293     hres = IUri_GetSchemeName(This->window->uri, &protocol);
294     if(FAILED(hres))
295         return hres;
296     if(hres == S_FALSE) {
297         SysFreeString(protocol);
298         *p = NULL;
299         return S_OK;
300     }
301
302     len = SysStringLen(protocol);
303     ret = SysAllocStringLen(protocol, len+1);
304     SysFreeString(protocol);
305     if(!ret)
306         return E_OUTOFMEMORY;
307
308     ret[len] = ':';
309     *p = ret;
310     return S_OK;
311 }
312
313 static HRESULT WINAPI HTMLLocation_put_host(IHTMLLocation *iface, BSTR v)
314 {
315     HTMLLocation *This = impl_from_IHTMLLocation(iface);
316     FIXME("(%p)->(%s)\n", This, debugstr_w(v));
317     return E_NOTIMPL;
318 }
319
320 static HRESULT WINAPI HTMLLocation_get_host(IHTMLLocation *iface, BSTR *p)
321 {
322     HTMLLocation *This = impl_from_IHTMLLocation(iface);
323     URL_COMPONENTSW url = {sizeof(URL_COMPONENTSW)};
324     HRESULT hres;
325
326     TRACE("(%p)->(%p)\n", This, p);
327
328     if(!p)
329         return E_POINTER;
330
331     url.dwHostNameLength = 1;
332     hres = get_url_components(This, &url);
333     if(FAILED(hres))
334         return hres;
335
336     if(!url.dwHostNameLength){
337         *p = NULL;
338         return S_OK;
339     }
340
341     if(url.nPort) {
342         /* <hostname>:<port> */
343         const WCHAR format[] = {'%','u',0};
344         DWORD len = url.dwHostNameLength + 1 + 5;
345         WCHAR *buf;
346
347         buf = *p = SysAllocStringLen(NULL, len);
348         memcpy(buf, url.lpszHostName, url.dwHostNameLength * sizeof(WCHAR));
349         buf[url.dwHostNameLength] = ':';
350         snprintfW(buf + url.dwHostNameLength + 1, 6, format, url.nPort);
351     }else
352         *p = SysAllocStringLen(url.lpszHostName, url.dwHostNameLength);
353
354     if(!*p)
355         return E_OUTOFMEMORY;
356     return S_OK;
357 }
358
359 static HRESULT WINAPI HTMLLocation_put_hostname(IHTMLLocation *iface, BSTR v)
360 {
361     HTMLLocation *This = impl_from_IHTMLLocation(iface);
362     FIXME("(%p)->(%s)\n", This, debugstr_w(v));
363     return E_NOTIMPL;
364 }
365
366 static HRESULT WINAPI HTMLLocation_get_hostname(IHTMLLocation *iface, BSTR *p)
367 {
368     HTMLLocation *This = impl_from_IHTMLLocation(iface);
369     BSTR hostname;
370     HRESULT hres;
371
372     TRACE("(%p)->(%p)\n", This, p);
373
374     if(!p)
375         return E_POINTER;
376
377     if(!This->window || !This->window->uri) {
378         FIXME("No current URI\n");
379         return E_NOTIMPL;
380     }
381
382     hres = IUri_GetHost(This->window->uri, &hostname);
383     if(hres == S_OK) {
384         *p = hostname;
385     }else if(hres == S_FALSE) {
386         SysFreeString(hostname);
387         *p = NULL;
388     }else {
389         return hres;
390     }
391
392     return S_OK;
393 }
394
395 static HRESULT WINAPI HTMLLocation_put_port(IHTMLLocation *iface, BSTR v)
396 {
397     HTMLLocation *This = impl_from_IHTMLLocation(iface);
398     FIXME("(%p)->(%s)\n", This, debugstr_w(v));
399     return E_NOTIMPL;
400 }
401
402 static HRESULT WINAPI HTMLLocation_get_port(IHTMLLocation *iface, BSTR *p)
403 {
404     HTMLLocation *This = impl_from_IHTMLLocation(iface);
405     DWORD port;
406     HRESULT hres;
407
408     TRACE("(%p)->(%p)\n", This, p);
409
410     if(!p)
411         return E_POINTER;
412
413     if(!This->window || !This->window->uri) {
414         FIXME("No current URI\n");
415         return E_NOTIMPL;
416     }
417
418     hres = IUri_GetPort(This->window->uri, &port);
419     if(FAILED(hres))
420         return hres;
421
422     if(hres == S_OK) {
423         static const WCHAR formatW[] = {'%','u',0};
424         WCHAR buf[12];
425
426         sprintfW(buf, formatW, port);
427         *p = SysAllocString(buf);
428     }else {
429         *p = SysAllocStringLen(NULL, 0);
430     }
431
432     if(!*p)
433         return E_OUTOFMEMORY;
434     return S_OK;
435 }
436
437 static HRESULT WINAPI HTMLLocation_put_pathname(IHTMLLocation *iface, BSTR v)
438 {
439     HTMLLocation *This = impl_from_IHTMLLocation(iface);
440     FIXME("(%p)->(%s)\n", This, debugstr_w(v));
441     return E_NOTIMPL;
442 }
443
444 static HRESULT WINAPI HTMLLocation_get_pathname(IHTMLLocation *iface, BSTR *p)
445 {
446     HTMLLocation *This = impl_from_IHTMLLocation(iface);
447     URL_COMPONENTSW url = {sizeof(URL_COMPONENTSW)};
448     HRESULT hres;
449
450     TRACE("(%p)->(%p)\n", This, p);
451
452     if(!p)
453         return E_POINTER;
454
455     url.dwUrlPathLength = 1;
456     url.dwExtraInfoLength = 1;
457     hres = get_url_components(This, &url);
458     if(FAILED(hres))
459         return hres;
460
461     if(url.dwUrlPathLength && url.lpszUrlPath[0] == '/')
462         *p = SysAllocStringLen(url.lpszUrlPath + 1, url.dwUrlPathLength - 1);
463     else
464         *p = SysAllocStringLen(url.lpszUrlPath, url.dwUrlPathLength);
465
466     if(!*p)
467         return E_OUTOFMEMORY;
468     return S_OK;
469 }
470
471 static HRESULT WINAPI HTMLLocation_put_search(IHTMLLocation *iface, BSTR v)
472 {
473     HTMLLocation *This = impl_from_IHTMLLocation(iface);
474     FIXME("(%p)->(%s)\n", This, debugstr_w(v));
475     return E_NOTIMPL;
476 }
477
478 static HRESULT WINAPI HTMLLocation_get_search(IHTMLLocation *iface, BSTR *p)
479 {
480     HTMLLocation *This = impl_from_IHTMLLocation(iface);
481     BSTR query;
482     HRESULT hres;
483
484     TRACE("(%p)->(%p)\n", This, p);
485
486     if(!p)
487         return E_POINTER;
488
489     if(!This->window || !This->window->uri) {
490         FIXME("No current URI\n");
491         return E_NOTIMPL;
492     }
493
494     hres = IUri_GetQuery(This->window->uri, &query);
495     if(hres == S_OK) {
496         *p = query;
497     }else if(hres == S_FALSE) {
498         SysFreeString(query);
499         *p = NULL;
500     }else {
501         return hres;
502     }
503
504     return S_OK;
505 }
506
507 static HRESULT WINAPI HTMLLocation_put_hash(IHTMLLocation *iface, BSTR v)
508 {
509     HTMLLocation *This = impl_from_IHTMLLocation(iface);
510     FIXME("(%p)->(%s)\n", This, debugstr_w(v));
511     return E_NOTIMPL;
512 }
513
514 static HRESULT WINAPI HTMLLocation_get_hash(IHTMLLocation *iface, BSTR *p)
515 {
516     HTMLLocation *This = impl_from_IHTMLLocation(iface);
517     BSTR hash;
518     HRESULT hres;
519
520     TRACE("(%p)->(%p)\n", This, p);
521
522     if(!p)
523         return E_POINTER;
524
525     if(!This->window || !This->window->uri) {
526         FIXME("No current URI\n");
527         return E_NOTIMPL;
528     }
529
530     hres = IUri_GetFragment(This->window->uri, &hash);
531     if(hres == S_OK) {
532         *p = hash;
533     }else if(hres == S_FALSE) {
534         SysFreeString(hash);
535         *p = NULL;
536     }else {
537         return hres;
538     }
539
540     return S_OK;
541 }
542
543 static HRESULT WINAPI HTMLLocation_reload(IHTMLLocation *iface, VARIANT_BOOL flag)
544 {
545     HTMLLocation *This = impl_from_IHTMLLocation(iface);
546     FIXME("(%p)->(%x)\n", This, flag);
547     return E_NOTIMPL;
548 }
549
550 static HRESULT WINAPI HTMLLocation_replace(IHTMLLocation *iface, BSTR bstr)
551 {
552     HTMLLocation *This = impl_from_IHTMLLocation(iface);
553
554     TRACE("(%p)->(%s)\n", This, debugstr_w(bstr));
555
556     if(!This->window) {
557         FIXME("No window available\n");
558         return E_FAIL;
559     }
560
561     return navigate_url(This->window, bstr, This->window->url);
562 }
563
564 static HRESULT WINAPI HTMLLocation_assign(IHTMLLocation *iface, BSTR bstr)
565 {
566     HTMLLocation *This = impl_from_IHTMLLocation(iface);
567     FIXME("(%p)->(%s)\n", This, debugstr_w(bstr));
568     return E_NOTIMPL;
569 }
570
571 static HRESULT WINAPI HTMLLocation_toString(IHTMLLocation *iface, BSTR *String)
572 {
573     HTMLLocation *This = impl_from_IHTMLLocation(iface);
574     FIXME("(%p)->(%p)\n", This, String);
575     return E_NOTIMPL;
576 }
577
578 static const IHTMLLocationVtbl HTMLLocationVtbl = {
579     HTMLLocation_QueryInterface,
580     HTMLLocation_AddRef,
581     HTMLLocation_Release,
582     HTMLLocation_GetTypeInfoCount,
583     HTMLLocation_GetTypeInfo,
584     HTMLLocation_GetIDsOfNames,
585     HTMLLocation_Invoke,
586     HTMLLocation_put_href,
587     HTMLLocation_get_href,
588     HTMLLocation_put_protocol,
589     HTMLLocation_get_protocol,
590     HTMLLocation_put_host,
591     HTMLLocation_get_host,
592     HTMLLocation_put_hostname,
593     HTMLLocation_get_hostname,
594     HTMLLocation_put_port,
595     HTMLLocation_get_port,
596     HTMLLocation_put_pathname,
597     HTMLLocation_get_pathname,
598     HTMLLocation_put_search,
599     HTMLLocation_get_search,
600     HTMLLocation_put_hash,
601     HTMLLocation_get_hash,
602     HTMLLocation_reload,
603     HTMLLocation_replace,
604     HTMLLocation_assign,
605     HTMLLocation_toString
606 };
607
608 static const tid_t HTMLLocation_iface_tids[] = {
609     IHTMLLocation_tid,
610     0
611 };
612 static dispex_static_data_t HTMLLocation_dispex = {
613     NULL,
614     DispHTMLLocation_tid,
615     NULL,
616     HTMLLocation_iface_tids
617 };
618
619
620 HRESULT HTMLLocation_Create(HTMLWindow *window, HTMLLocation **ret)
621 {
622     HTMLLocation *location;
623
624     location = heap_alloc(sizeof(*location));
625     if(!location)
626         return E_OUTOFMEMORY;
627
628     location->IHTMLLocation_iface.lpVtbl = &HTMLLocationVtbl;
629     location->ref = 1;
630     location->window = window;
631
632     init_dispex(&location->dispex, (IUnknown*)&location->IHTMLLocation_iface, &HTMLLocation_dispex);
633
634     *ret = location;
635     return S_OK;
636 }