/* * Copyright 2008 Jacek Caban for CodeWeavers * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ #include <stdarg.h> #define COBJMACROS #include "windef.h" #include "winbase.h" #include "winuser.h" #include "winreg.h" #include "ole2.h" #include "wininet.h" #include "shlwapi.h" #include "wine/debug.h" #include "mshtml_private.h" #include "binding.h" #include "resource.h" WINE_DEFAULT_DEBUG_CHANNEL(mshtml); static HRESULT get_url(HTMLLocation *This, const WCHAR **ret) { if(!This->window || !This->window->base.outer_window || !This->window->base.outer_window->url) { FIXME("No current URL\n"); return E_NOTIMPL; } *ret = This->window->base.outer_window->url; return S_OK; } static IUri *get_uri(HTMLLocation *This) { if(!This->window || !This->window->base.outer_window) return NULL; return This->window->base.outer_window->uri; } static HRESULT get_url_components(HTMLLocation *This, URL_COMPONENTSW *url) { const WCHAR *doc_url; HRESULT hres; hres = get_url(This, &doc_url); if(FAILED(hres)) return hres; if(!InternetCrackUrlW(doc_url, 0, 0, url)) { FIXME("InternetCrackUrlW failed: 0x%08x\n", GetLastError()); SetLastError(0); return E_FAIL; } return S_OK; } static inline HTMLLocation *impl_from_IHTMLLocation(IHTMLLocation *iface) { return CONTAINING_RECORD(iface, HTMLLocation, IHTMLLocation_iface); } static HRESULT WINAPI HTMLLocation_QueryInterface(IHTMLLocation *iface, REFIID riid, void **ppv) { HTMLLocation *This = impl_from_IHTMLLocation(iface); TRACE("(%p)->(%s %p)\n", This, debugstr_mshtml_guid(riid), ppv); if(IsEqualGUID(&IID_IUnknown, riid)) { *ppv = &This->IHTMLLocation_iface; }else if(IsEqualGUID(&IID_IHTMLLocation, riid)) { *ppv = &This->IHTMLLocation_iface; }else if(IsEqualGUID(&IID_IMarshal, riid)) { *ppv = NULL; FIXME("(%p)->(IID_IMarshal %p)\n", This, ppv); return E_NOINTERFACE; }else if(dispex_query_interface(&This->dispex, riid, ppv)) { return *ppv ? S_OK : E_NOINTERFACE; }else { *ppv = NULL; WARN("(%p)->(%s %p)\n", This, debugstr_mshtml_guid(riid), ppv); return E_NOINTERFACE; } IUnknown_AddRef((IUnknown*)*ppv); return S_OK; } static ULONG WINAPI HTMLLocation_AddRef(IHTMLLocation *iface) { HTMLLocation *This = impl_from_IHTMLLocation(iface); LONG ref = InterlockedIncrement(&This->ref); TRACE("(%p) ref=%d\n", This, ref); return ref; } static ULONG WINAPI HTMLLocation_Release(IHTMLLocation *iface) { HTMLLocation *This = impl_from_IHTMLLocation(iface); LONG ref = InterlockedDecrement(&This->ref); TRACE("(%p) ref=%d\n", This, ref); if(!ref) { if(This->window) This->window->location = NULL; release_dispex(&This->dispex); heap_free(This); } return ref; } static HRESULT WINAPI HTMLLocation_GetTypeInfoCount(IHTMLLocation *iface, UINT *pctinfo) { HTMLLocation *This = impl_from_IHTMLLocation(iface); return IDispatchEx_GetTypeInfoCount(&This->dispex.IDispatchEx_iface, pctinfo); } static HRESULT WINAPI HTMLLocation_GetTypeInfo(IHTMLLocation *iface, UINT iTInfo, LCID lcid, ITypeInfo **ppTInfo) { HTMLLocation *This = impl_from_IHTMLLocation(iface); return IDispatchEx_GetTypeInfo(&This->dispex.IDispatchEx_iface, iTInfo, lcid, ppTInfo); } static HRESULT WINAPI HTMLLocation_GetIDsOfNames(IHTMLLocation *iface, REFIID riid, LPOLESTR *rgszNames, UINT cNames, LCID lcid, DISPID *rgDispId) { HTMLLocation *This = impl_from_IHTMLLocation(iface); return IDispatchEx_GetIDsOfNames(&This->dispex.IDispatchEx_iface, riid, rgszNames, cNames, lcid, rgDispId); } static HRESULT WINAPI HTMLLocation_Invoke(IHTMLLocation *iface, DISPID dispIdMember, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr) { HTMLLocation *This = impl_from_IHTMLLocation(iface); return IDispatchEx_Invoke(&This->dispex.IDispatchEx_iface, dispIdMember, riid, lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr); } static HRESULT WINAPI HTMLLocation_put_href(IHTMLLocation *iface, BSTR v) { HTMLLocation *This = impl_from_IHTMLLocation(iface); TRACE("(%p)->(%s)\n", This, debugstr_w(v)); if(!This->window || !This->window->base.outer_window) { FIXME("No window available\n"); return E_FAIL; } return navigate_url(This->window->base.outer_window, v, This->window->base.outer_window->uri, BINDING_NAVIGATED); } static HRESULT WINAPI HTMLLocation_get_href(IHTMLLocation *iface, BSTR *p) { HTMLLocation *This = impl_from_IHTMLLocation(iface); URL_COMPONENTSW url = {sizeof(URL_COMPONENTSW)}; WCHAR *buf = NULL, *url_path = NULL; HRESULT hres, ret; DWORD len = 0; int i; TRACE("(%p)->(%p)\n", This, p); if(!p) return E_POINTER; url.dwSchemeLength = 1; url.dwHostNameLength = 1; url.dwUrlPathLength = 1; url.dwExtraInfoLength = 1; hres = get_url_components(This, &url); if(FAILED(hres)) return hres; switch(url.nScheme) { case INTERNET_SCHEME_FILE: { /* prepend a slash */ url_path = HeapAlloc(GetProcessHeap(), 0, (url.dwUrlPathLength + 1) * sizeof(WCHAR)); if(!url_path) return E_OUTOFMEMORY; url_path[0] = '/'; memcpy(url_path + 1, url.lpszUrlPath, url.dwUrlPathLength * sizeof(WCHAR)); url.lpszUrlPath = url_path; url.dwUrlPathLength = url.dwUrlPathLength + 1; } break; case INTERNET_SCHEME_HTTP: case INTERNET_SCHEME_HTTPS: case INTERNET_SCHEME_FTP: if(!url.dwUrlPathLength) { /* add a slash if it's blank */ url_path = url.lpszUrlPath = HeapAlloc(GetProcessHeap(), 0, 1 * sizeof(WCHAR)); if(!url.lpszUrlPath) return E_OUTOFMEMORY; url.lpszUrlPath[0] = '/'; url.dwUrlPathLength = 1; } break; default: break; } /* replace \ with / */ for(i = 0; i < url.dwUrlPathLength; ++i) if(url.lpszUrlPath[i] == '\\') url.lpszUrlPath[i] = '/'; if(InternetCreateUrlW(&url, ICU_ESCAPE, NULL, &len)) { FIXME("InternetCreateUrl succeeded with NULL buffer?\n"); ret = E_FAIL; goto cleanup; } if(GetLastError() != ERROR_INSUFFICIENT_BUFFER) { FIXME("InternetCreateUrl failed with error: %08x\n", GetLastError()); SetLastError(0); ret = E_FAIL; goto cleanup; } SetLastError(0); buf = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR)); if(!buf) { ret = E_OUTOFMEMORY; goto cleanup; } if(!InternetCreateUrlW(&url, ICU_ESCAPE, buf, &len)) { FIXME("InternetCreateUrl failed with error: %08x\n", GetLastError()); SetLastError(0); ret = E_FAIL; goto cleanup; } *p = SysAllocStringLen(buf, len); if(!*p) { ret = E_OUTOFMEMORY; goto cleanup; } ret = S_OK; cleanup: HeapFree(GetProcessHeap(), 0, buf); HeapFree(GetProcessHeap(), 0, url_path); return ret; } static HRESULT WINAPI HTMLLocation_put_protocol(IHTMLLocation *iface, BSTR v) { HTMLLocation *This = impl_from_IHTMLLocation(iface); FIXME("(%p)->(%s)\n", This, debugstr_w(v)); return E_NOTIMPL; } static HRESULT WINAPI HTMLLocation_get_protocol(IHTMLLocation *iface, BSTR *p) { HTMLLocation *This = impl_from_IHTMLLocation(iface); BSTR protocol, ret; unsigned len; IUri *uri; HRESULT hres; TRACE("(%p)->(%p)\n", This, p); if(!p) return E_POINTER; if(!(uri = get_uri(This))) { FIXME("No current URI\n"); return E_NOTIMPL; } hres = IUri_GetSchemeName(uri, &protocol); if(FAILED(hres)) return hres; if(hres == S_FALSE) { SysFreeString(protocol); *p = NULL; return S_OK; } len = SysStringLen(protocol); ret = SysAllocStringLen(protocol, len+1); SysFreeString(protocol); if(!ret) return E_OUTOFMEMORY; ret[len] = ':'; *p = ret; return S_OK; } static HRESULT WINAPI HTMLLocation_put_host(IHTMLLocation *iface, BSTR v) { HTMLLocation *This = impl_from_IHTMLLocation(iface); FIXME("(%p)->(%s)\n", This, debugstr_w(v)); return E_NOTIMPL; } static HRESULT WINAPI HTMLLocation_get_host(IHTMLLocation *iface, BSTR *p) { HTMLLocation *This = impl_from_IHTMLLocation(iface); URL_COMPONENTSW url = {sizeof(URL_COMPONENTSW)}; HRESULT hres; TRACE("(%p)->(%p)\n", This, p); if(!p) return E_POINTER; url.dwHostNameLength = 1; hres = get_url_components(This, &url); if(FAILED(hres)) return hres; if(!url.dwHostNameLength){ *p = NULL; return S_OK; } if(url.nPort) { /* <hostname>:<port> */ const WCHAR format[] = {'%','u',0}; DWORD len, port_len; WCHAR portW[6]; WCHAR *buf; port_len = snprintfW(portW, sizeof(portW)/sizeof(portW[0]), format, url.nPort); len = url.dwHostNameLength + 1 /* ':' */ + port_len; buf = *p = SysAllocStringLen(NULL, len); memcpy(buf, url.lpszHostName, url.dwHostNameLength * sizeof(WCHAR)); buf[url.dwHostNameLength] = ':'; memcpy(buf + url.dwHostNameLength + 1, portW, port_len * sizeof(WCHAR)); }else *p = SysAllocStringLen(url.lpszHostName, url.dwHostNameLength); if(!*p) return E_OUTOFMEMORY; return S_OK; } static HRESULT WINAPI HTMLLocation_put_hostname(IHTMLLocation *iface, BSTR v) { HTMLLocation *This = impl_from_IHTMLLocation(iface); FIXME("(%p)->(%s)\n", This, debugstr_w(v)); return E_NOTIMPL; } static HRESULT WINAPI HTMLLocation_get_hostname(IHTMLLocation *iface, BSTR *p) { HTMLLocation *This = impl_from_IHTMLLocation(iface); BSTR hostname; IUri *uri; HRESULT hres; TRACE("(%p)->(%p)\n", This, p); if(!p) return E_POINTER; if(!(uri = get_uri(This))) { FIXME("No current URI\n"); return E_NOTIMPL; } hres = IUri_GetHost(uri, &hostname); if(hres == S_OK) { *p = hostname; }else if(hres == S_FALSE) { SysFreeString(hostname); *p = NULL; }else { return hres; } return S_OK; } static HRESULT WINAPI HTMLLocation_put_port(IHTMLLocation *iface, BSTR v) { HTMLLocation *This = impl_from_IHTMLLocation(iface); FIXME("(%p)->(%s)\n", This, debugstr_w(v)); return E_NOTIMPL; } static HRESULT WINAPI HTMLLocation_get_port(IHTMLLocation *iface, BSTR *p) { HTMLLocation *This = impl_from_IHTMLLocation(iface); DWORD port; IUri *uri; HRESULT hres; TRACE("(%p)->(%p)\n", This, p); if(!p) return E_POINTER; if(!(uri = get_uri(This))) { FIXME("No current URI\n"); return E_NOTIMPL; } hres = IUri_GetPort(uri, &port); if(FAILED(hres)) return hres; if(hres == S_OK) { static const WCHAR formatW[] = {'%','u',0}; WCHAR buf[12]; sprintfW(buf, formatW, port); *p = SysAllocString(buf); }else { *p = SysAllocStringLen(NULL, 0); } if(!*p) return E_OUTOFMEMORY; return S_OK; } static HRESULT WINAPI HTMLLocation_put_pathname(IHTMLLocation *iface, BSTR v) { HTMLLocation *This = impl_from_IHTMLLocation(iface); FIXME("(%p)->(%s)\n", This, debugstr_w(v)); return E_NOTIMPL; } static HRESULT WINAPI HTMLLocation_get_pathname(IHTMLLocation *iface, BSTR *p) { HTMLLocation *This = impl_from_IHTMLLocation(iface); URL_COMPONENTSW url = {sizeof(URL_COMPONENTSW)}; HRESULT hres; TRACE("(%p)->(%p)\n", This, p); if(!p) return E_POINTER; url.dwUrlPathLength = 1; url.dwExtraInfoLength = 1; hres = get_url_components(This, &url); if(FAILED(hres)) return hres; if(url.dwUrlPathLength && url.lpszUrlPath[0] == '/') *p = SysAllocStringLen(url.lpszUrlPath + 1, url.dwUrlPathLength - 1); else *p = SysAllocStringLen(url.lpszUrlPath, url.dwUrlPathLength); if(!*p) return E_OUTOFMEMORY; return S_OK; } static HRESULT WINAPI HTMLLocation_put_search(IHTMLLocation *iface, BSTR v) { HTMLLocation *This = impl_from_IHTMLLocation(iface); FIXME("(%p)->(%s)\n", This, debugstr_w(v)); return E_NOTIMPL; } static HRESULT WINAPI HTMLLocation_get_search(IHTMLLocation *iface, BSTR *p) { HTMLLocation *This = impl_from_IHTMLLocation(iface); BSTR query; IUri *uri; HRESULT hres; TRACE("(%p)->(%p)\n", This, p); if(!p) return E_POINTER; if(!(uri = get_uri(This))) { FIXME("No current URI\n"); return E_NOTIMPL; } hres = IUri_GetQuery(uri, &query); if(hres == S_OK) { *p = query; }else if(hres == S_FALSE) { SysFreeString(query); *p = NULL; }else { return hres; } return S_OK; } static HRESULT WINAPI HTMLLocation_put_hash(IHTMLLocation *iface, BSTR v) { HTMLLocation *This = impl_from_IHTMLLocation(iface); FIXME("(%p)->(%s)\n", This, debugstr_w(v)); return E_NOTIMPL; } static HRESULT WINAPI HTMLLocation_get_hash(IHTMLLocation *iface, BSTR *p) { HTMLLocation *This = impl_from_IHTMLLocation(iface); BSTR hash; IUri *uri; HRESULT hres; TRACE("(%p)->(%p)\n", This, p); if(!p) return E_POINTER; if(!(uri = get_uri(This))) { FIXME("No current URI\n"); return E_NOTIMPL; } hres = IUri_GetFragment(uri, &hash); if(hres == S_OK) { *p = hash; }else if(hres == S_FALSE) { SysFreeString(hash); *p = NULL; }else { return hres; } return S_OK; } static HRESULT WINAPI HTMLLocation_reload(IHTMLLocation *iface, VARIANT_BOOL flag) { HTMLLocation *This = impl_from_IHTMLLocation(iface); FIXME("(%p)->(%x)\n", This, flag); return E_NOTIMPL; } static HRESULT WINAPI HTMLLocation_replace(IHTMLLocation *iface, BSTR bstr) { HTMLLocation *This = impl_from_IHTMLLocation(iface); TRACE("(%p)->(%s)\n", This, debugstr_w(bstr)); if(!This->window || !This->window->base.outer_window) { FIXME("No window available\n"); return E_FAIL; } return navigate_url(This->window->base.outer_window, bstr, This->window->base.outer_window->uri, BINDING_NAVIGATED|BINDING_REPLACE); } static HRESULT WINAPI HTMLLocation_assign(IHTMLLocation *iface, BSTR bstr) { HTMLLocation *This = impl_from_IHTMLLocation(iface); TRACE("(%p)->(%s)\n", This, debugstr_w(bstr)); return IHTMLLocation_put_href(iface, bstr); } static HRESULT WINAPI HTMLLocation_toString(IHTMLLocation *iface, BSTR *String) { HTMLLocation *This = impl_from_IHTMLLocation(iface); TRACE("(%p)->(%p)\n", This, String); return IHTMLLocation_get_href(&This->IHTMLLocation_iface, String); } static const IHTMLLocationVtbl HTMLLocationVtbl = { HTMLLocation_QueryInterface, HTMLLocation_AddRef, HTMLLocation_Release, HTMLLocation_GetTypeInfoCount, HTMLLocation_GetTypeInfo, HTMLLocation_GetIDsOfNames, HTMLLocation_Invoke, HTMLLocation_put_href, HTMLLocation_get_href, HTMLLocation_put_protocol, HTMLLocation_get_protocol, HTMLLocation_put_host, HTMLLocation_get_host, HTMLLocation_put_hostname, HTMLLocation_get_hostname, HTMLLocation_put_port, HTMLLocation_get_port, HTMLLocation_put_pathname, HTMLLocation_get_pathname, HTMLLocation_put_search, HTMLLocation_get_search, HTMLLocation_put_hash, HTMLLocation_get_hash, HTMLLocation_reload, HTMLLocation_replace, HTMLLocation_assign, HTMLLocation_toString }; static const tid_t HTMLLocation_iface_tids[] = { IHTMLLocation_tid, 0 }; static dispex_static_data_t HTMLLocation_dispex = { NULL, DispHTMLLocation_tid, HTMLLocation_iface_tids }; HRESULT HTMLLocation_Create(HTMLInnerWindow *window, HTMLLocation **ret) { HTMLLocation *location; location = heap_alloc(sizeof(*location)); if(!location) return E_OUTOFMEMORY; location->IHTMLLocation_iface.lpVtbl = &HTMLLocationVtbl; location->ref = 1; location->window = window; init_dispex(&location->dispex, (IUnknown*)&location->IHTMLLocation_iface, &HTMLLocation_dispex); *ret = location; return S_OK; }