mshtml: Use IUri in IHTMLLocation::get_protocol implementation.

This commit is contained in:
Jacek Caban 2011-12-29 12:02:41 +01:00 committed by Alexandre Julliard
parent 723ae056b9
commit 98b8ebaff7
1 changed files with 20 additions and 13 deletions

View File

@ -276,7 +276,8 @@ static HRESULT WINAPI HTMLLocation_put_protocol(IHTMLLocation *iface, BSTR v)
static HRESULT WINAPI HTMLLocation_get_protocol(IHTMLLocation *iface, BSTR *p) static HRESULT WINAPI HTMLLocation_get_protocol(IHTMLLocation *iface, BSTR *p)
{ {
HTMLLocation *This = impl_from_IHTMLLocation(iface); HTMLLocation *This = impl_from_IHTMLLocation(iface);
URL_COMPONENTSW url = {sizeof(URL_COMPONENTSW)}; BSTR protocol, ret;
unsigned len;
HRESULT hres; HRESULT hres;
TRACE("(%p)->(%p)\n", This, p); TRACE("(%p)->(%p)\n", This, p);
@ -284,22 +285,28 @@ static HRESULT WINAPI HTMLLocation_get_protocol(IHTMLLocation *iface, BSTR *p)
if(!p) if(!p)
return E_POINTER; return E_POINTER;
url.dwSchemeLength = 1; if(!This->window || !This->window->uri) {
hres = get_url_components(This, &url); FIXME("No current URI\n");
return E_NOTIMPL;
}
hres = IUri_GetSchemeName(This->window->uri, &protocol);
if(FAILED(hres)) if(FAILED(hres))
return hres; return hres;
if(hres == S_FALSE) {
if(!url.dwSchemeLength) { SysFreeString(protocol);
FIXME("Unexpected blank protocol\n"); *p = NULL;
return E_NOTIMPL; return S_OK;
}else {
WCHAR *buf;
buf = *p = SysAllocStringLen(NULL, url.dwSchemeLength + 1);
memcpy(buf, url.lpszScheme, url.dwSchemeLength * sizeof(WCHAR));
buf[url.dwSchemeLength] = ':';
} }
if(!*p)
len = SysStringLen(protocol);
ret = SysAllocStringLen(protocol, len+1);
SysFreeString(protocol);
if(!ret)
return E_OUTOFMEMORY; return E_OUTOFMEMORY;
ret[len] = ':';
*p = ret;
return S_OK; return S_OK;
} }