/* * Copyright 2006-2010 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 "config.h" #include #define COBJMACROS #define NONAMELESSUNION #define NONAMELESSSTRUCT #include "windef.h" #include "winbase.h" #include "winuser.h" #include "winreg.h" #include "ole2.h" #include "hlguids.h" #include "shlguid.h" #include "wininet.h" #include "shlwapi.h" #include "htiface.h" #include "shdeprecated.h" #include "wine/debug.h" #include "mshtml_private.h" #include "binding.h" WINE_DEFAULT_DEBUG_CHANNEL(mshtml); #define CONTENT_LENGTH "Content-Length" #define UTF8_STR "utf-8" #define UTF16_STR "utf-16" static const WCHAR emptyW[] = {0}; static const WCHAR text_htmlW[] = {'t','e','x','t','/','h','t','m','l',0}; enum { BOM_NONE, BOM_UTF8, BOM_UTF16 }; struct nsProtocolStream { nsIInputStream nsIInputStream_iface; LONG ref; char buf[1024]; DWORD buf_size; }; struct BSCallbackVtbl { void (*destroy)(BSCallback*); HRESULT (*init_bindinfo)(BSCallback*); HRESULT (*start_binding)(BSCallback*); HRESULT (*stop_binding)(BSCallback*,HRESULT); HRESULT (*read_data)(BSCallback*,IStream*); HRESULT (*on_progress)(BSCallback*,ULONG,LPCWSTR); HRESULT (*on_response)(BSCallback*,DWORD,LPCWSTR); HRESULT (*beginning_transaction)(BSCallback*,WCHAR**); }; static inline nsProtocolStream *impl_from_nsIInputStream(nsIInputStream *iface) { return CONTAINING_RECORD(iface, nsProtocolStream, nsIInputStream_iface); } static nsresult NSAPI nsInputStream_QueryInterface(nsIInputStream *iface, nsIIDRef riid, void **result) { nsProtocolStream *This = impl_from_nsIInputStream(iface); *result = NULL; if(IsEqualGUID(&IID_nsISupports, riid)) { TRACE("(%p)->(IID_nsISupports %p)\n", This, result); *result = &This->nsIInputStream_iface; }else if(IsEqualGUID(&IID_nsIInputStream, riid)) { TRACE("(%p)->(IID_nsIInputStream %p)\n", This, result); *result = &This->nsIInputStream_iface; } if(*result) { nsIInputStream_AddRef(&This->nsIInputStream_iface); return NS_OK; } WARN("unsupported interface %s\n", debugstr_guid(riid)); return NS_NOINTERFACE; } static nsrefcnt NSAPI nsInputStream_AddRef(nsIInputStream *iface) { nsProtocolStream *This = impl_from_nsIInputStream(iface); LONG ref = InterlockedIncrement(&This->ref); TRACE("(%p) ref=%d\n", This, ref); return ref; } static nsrefcnt NSAPI nsInputStream_Release(nsIInputStream *iface) { nsProtocolStream *This = impl_from_nsIInputStream(iface); LONG ref = InterlockedDecrement(&This->ref); TRACE("(%p) ref=%d\n", This, ref); if(!ref) heap_free(This); return ref; } static nsresult NSAPI nsInputStream_Close(nsIInputStream *iface) { nsProtocolStream *This = impl_from_nsIInputStream(iface); FIXME("(%p)\n", This); return NS_ERROR_NOT_IMPLEMENTED; } static nsresult NSAPI nsInputStream_Available(nsIInputStream *iface, PRUint32 *_retval) { nsProtocolStream *This = impl_from_nsIInputStream(iface); FIXME("(%p)->(%p)\n", This, _retval); return NS_ERROR_NOT_IMPLEMENTED; } static nsresult NSAPI nsInputStream_Read(nsIInputStream *iface, char *aBuf, PRUint32 aCount, PRUint32 *_retval) { nsProtocolStream *This = impl_from_nsIInputStream(iface); DWORD read = aCount; TRACE("(%p)->(%p %d %p)\n", This, aBuf, aCount, _retval); if(read > This->buf_size) read = This->buf_size; if(read) { memcpy(aBuf, This->buf, read); if(read < This->buf_size) memmove(This->buf, This->buf+read, This->buf_size-read); This->buf_size -= read; } *_retval = read; return NS_OK; } static nsresult NSAPI nsInputStream_ReadSegments(nsIInputStream *iface, nsresult (WINAPI *aWriter)(nsIInputStream*,void*,const char*,PRUint32,PRUint32,PRUint32*), void *aClousure, PRUint32 aCount, PRUint32 *_retval) { nsProtocolStream *This = impl_from_nsIInputStream(iface); PRUint32 written = 0; nsresult nsres; TRACE("(%p)->(%p %p %d %p)\n", This, aWriter, aClousure, aCount, _retval); if(!This->buf_size) return S_OK; if(aCount > This->buf_size) aCount = This->buf_size; nsres = aWriter(&This->nsIInputStream_iface, aClousure, This->buf, 0, aCount, &written); if(NS_FAILED(nsres)) TRACE("aWritter failed: %08x\n", nsres); else if(written != This->buf_size) FIXME("written %d != buf_size %d\n", written, This->buf_size); This->buf_size -= written; *_retval = written; return nsres; } static nsresult NSAPI nsInputStream_IsNonBlocking(nsIInputStream *iface, cpp_bool *_retval) { nsProtocolStream *This = impl_from_nsIInputStream(iface); FIXME("(%p)->(%p)\n", This, _retval); return NS_ERROR_NOT_IMPLEMENTED; } static const nsIInputStreamVtbl nsInputStreamVtbl = { nsInputStream_QueryInterface, nsInputStream_AddRef, nsInputStream_Release, nsInputStream_Close, nsInputStream_Available, nsInputStream_Read, nsInputStream_ReadSegments, nsInputStream_IsNonBlocking }; static nsProtocolStream *create_nsprotocol_stream(void) { nsProtocolStream *ret = heap_alloc(sizeof(nsProtocolStream)); ret->nsIInputStream_iface.lpVtbl = &nsInputStreamVtbl; ret->ref = 1; ret->buf_size = 0; return ret; } static inline BSCallback *impl_from_IBindStatusCallback(IBindStatusCallback *iface) { return CONTAINING_RECORD(iface, BSCallback, IBindStatusCallback_iface); } static HRESULT WINAPI BindStatusCallback_QueryInterface(IBindStatusCallback *iface, REFIID riid, void **ppv) { BSCallback *This = impl_from_IBindStatusCallback(iface); *ppv = NULL; if(IsEqualGUID(&IID_IUnknown, riid)) { TRACE("(%p)->(IID_IUnknown, %p)\n", This, ppv); *ppv = &This->IBindStatusCallback_iface; }else if(IsEqualGUID(&IID_IBindStatusCallback, riid)) { TRACE("(%p)->(IID_IBindStatusCallback, %p)\n", This, ppv); *ppv = &This->IBindStatusCallback_iface; }else if(IsEqualGUID(&IID_IServiceProvider, riid)) { TRACE("(%p)->(IID_IServiceProvider %p)\n", This, ppv); *ppv = &This->IServiceProvider_iface; }else if(IsEqualGUID(&IID_IHttpNegotiate, riid)) { TRACE("(%p)->(IID_IHttpNegotiate %p)\n", This, ppv); *ppv = &This->IHttpNegotiate2_iface; }else if(IsEqualGUID(&IID_IHttpNegotiate2, riid)) { TRACE("(%p)->(IID_IHttpNegotiate2 %p)\n", This, ppv); *ppv = &This->IHttpNegotiate2_iface; }else if(IsEqualGUID(&IID_IInternetBindInfo, riid)) { TRACE("(%p)->(IID_IInternetBindInfo %p)\n", This, ppv); *ppv = &This->IInternetBindInfo_iface; } if(*ppv) { IBindStatusCallback_AddRef(&This->IBindStatusCallback_iface); return S_OK; } TRACE("Unsupported riid = %s\n", debugstr_guid(riid)); return E_NOINTERFACE; } static ULONG WINAPI BindStatusCallback_AddRef(IBindStatusCallback *iface) { BSCallback *This = impl_from_IBindStatusCallback(iface); LONG ref = InterlockedIncrement(&This->ref); TRACE("(%p) ref = %d\n", This, ref); return ref; } static ULONG WINAPI BindStatusCallback_Release(IBindStatusCallback *iface) { BSCallback *This = impl_from_IBindStatusCallback(iface); LONG ref = InterlockedDecrement(&This->ref); TRACE("(%p) ref = %d\n", This, ref); if(!ref) { if(This->post_data) GlobalFree(This->post_data); if(This->mon) IMoniker_Release(This->mon); if(This->binding) IBinding_Release(This->binding); list_remove(&This->entry); list_init(&This->entry); heap_free(This->headers); This->vtbl->destroy(This); } return ref; } static HRESULT WINAPI BindStatusCallback_OnStartBinding(IBindStatusCallback *iface, DWORD dwReserved, IBinding *pbind) { BSCallback *This = impl_from_IBindStatusCallback(iface); TRACE("(%p)->(%d %p)\n", This, dwReserved, pbind); IBinding_AddRef(pbind); This->binding = pbind; if(This->window) list_add_head(&This->window->bindings, &This->entry); return This->vtbl->start_binding(This); } static HRESULT WINAPI BindStatusCallback_GetPriority(IBindStatusCallback *iface, LONG *pnPriority) { BSCallback *This = impl_from_IBindStatusCallback(iface); FIXME("(%p)->(%p)\n", This, pnPriority); return E_NOTIMPL; } static HRESULT WINAPI BindStatusCallback_OnLowResource(IBindStatusCallback *iface, DWORD reserved) { BSCallback *This = impl_from_IBindStatusCallback(iface); FIXME("(%p)->(%d)\n", This, reserved); return E_NOTIMPL; } static HRESULT WINAPI BindStatusCallback_OnProgress(IBindStatusCallback *iface, ULONG ulProgress, ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText) { BSCallback *This = impl_from_IBindStatusCallback(iface); TRACE("%p)->(%u %u %u %s)\n", This, ulProgress, ulProgressMax, ulStatusCode, debugstr_w(szStatusText)); return This->vtbl->on_progress(This, ulStatusCode, szStatusText); } static HRESULT WINAPI BindStatusCallback_OnStopBinding(IBindStatusCallback *iface, HRESULT hresult, LPCWSTR szError) { BSCallback *This = impl_from_IBindStatusCallback(iface); HRESULT hres; TRACE("(%p)->(%08x %s)\n", This, hresult, debugstr_w(szError)); /* NOTE: IE7 calls GetBindResult here */ hres = This->vtbl->stop_binding(This, hresult); if(This->binding) { IBinding_Release(This->binding); This->binding = NULL; } list_remove(&This->entry); list_init(&This->entry); This->window = NULL; return hres; } static HRESULT WINAPI BindStatusCallback_GetBindInfo(IBindStatusCallback *iface, DWORD *grfBINDF, BINDINFO *pbindinfo) { BSCallback *This = impl_from_IBindStatusCallback(iface); DWORD size; TRACE("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo); if(!This->bindinfo_ready) { HRESULT hres; hres = This->vtbl->init_bindinfo(This); if(FAILED(hres)) return hres; This->bindinfo_ready = TRUE; } *grfBINDF = This->bindf; size = pbindinfo->cbSize; memset(pbindinfo, 0, size); pbindinfo->cbSize = size; pbindinfo->cbstgmedData = This->post_data_len; pbindinfo->dwCodePage = CP_UTF8; pbindinfo->dwOptions = 0x80000; if(This->post_data) { pbindinfo->dwBindVerb = BINDVERB_POST; pbindinfo->stgmedData.tymed = TYMED_HGLOBAL; pbindinfo->stgmedData.u.hGlobal = This->post_data; pbindinfo->stgmedData.pUnkForRelease = (IUnknown*)&This->IBindStatusCallback_iface; IBindStatusCallback_AddRef(&This->IBindStatusCallback_iface); } return S_OK; } static HRESULT WINAPI BindStatusCallback_OnDataAvailable(IBindStatusCallback *iface, DWORD grfBSCF, DWORD dwSize, FORMATETC *pformatetc, STGMEDIUM *pstgmed) { BSCallback *This = impl_from_IBindStatusCallback(iface); TRACE("(%p)->(%08x %d %p %p)\n", This, grfBSCF, dwSize, pformatetc, pstgmed); return This->vtbl->read_data(This, pstgmed->u.pstm); } static HRESULT WINAPI BindStatusCallback_OnObjectAvailable(IBindStatusCallback *iface, REFIID riid, IUnknown *punk) { BSCallback *This = impl_from_IBindStatusCallback(iface); FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), punk); return E_NOTIMPL; } static const IBindStatusCallbackVtbl BindStatusCallbackVtbl = { BindStatusCallback_QueryInterface, BindStatusCallback_AddRef, BindStatusCallback_Release, BindStatusCallback_OnStartBinding, BindStatusCallback_GetPriority, BindStatusCallback_OnLowResource, BindStatusCallback_OnProgress, BindStatusCallback_OnStopBinding, BindStatusCallback_GetBindInfo, BindStatusCallback_OnDataAvailable, BindStatusCallback_OnObjectAvailable }; static inline BSCallback *impl_from_IHttpNegotiate2(IHttpNegotiate2 *iface) { return CONTAINING_RECORD(iface, BSCallback, IHttpNegotiate2_iface); } static HRESULT WINAPI HttpNegotiate_QueryInterface(IHttpNegotiate2 *iface, REFIID riid, void **ppv) { BSCallback *This = impl_from_IHttpNegotiate2(iface); return IBindStatusCallback_QueryInterface(&This->IBindStatusCallback_iface, riid, ppv); } static ULONG WINAPI HttpNegotiate_AddRef(IHttpNegotiate2 *iface) { BSCallback *This = impl_from_IHttpNegotiate2(iface); return IBindStatusCallback_AddRef(&This->IBindStatusCallback_iface); } static ULONG WINAPI HttpNegotiate_Release(IHttpNegotiate2 *iface) { BSCallback *This = impl_from_IHttpNegotiate2(iface); return IBindStatusCallback_Release(&This->IBindStatusCallback_iface); } static HRESULT WINAPI HttpNegotiate_BeginningTransaction(IHttpNegotiate2 *iface, LPCWSTR szURL, LPCWSTR szHeaders, DWORD dwReserved, LPWSTR *pszAdditionalHeaders) { BSCallback *This = impl_from_IHttpNegotiate2(iface); HRESULT hres; TRACE("(%p)->(%s %s %d %p)\n", This, debugstr_w(szURL), debugstr_w(szHeaders), dwReserved, pszAdditionalHeaders); *pszAdditionalHeaders = NULL; hres = This->vtbl->beginning_transaction(This, pszAdditionalHeaders); if(hres != S_FALSE) return hres; if(This->headers) { DWORD size; size = (strlenW(This->headers)+1)*sizeof(WCHAR); *pszAdditionalHeaders = CoTaskMemAlloc(size); if(!*pszAdditionalHeaders) return E_OUTOFMEMORY; memcpy(*pszAdditionalHeaders, This->headers, size); } return S_OK; } static HRESULT WINAPI HttpNegotiate_OnResponse(IHttpNegotiate2 *iface, DWORD dwResponseCode, LPCWSTR szResponseHeaders, LPCWSTR szRequestHeaders, LPWSTR *pszAdditionalRequestHeaders) { BSCallback *This = impl_from_IHttpNegotiate2(iface); TRACE("(%p)->(%d %s %s %p)\n", This, dwResponseCode, debugstr_w(szResponseHeaders), debugstr_w(szRequestHeaders), pszAdditionalRequestHeaders); return This->vtbl->on_response(This, dwResponseCode, szResponseHeaders); } static HRESULT WINAPI HttpNegotiate_GetRootSecurityId(IHttpNegotiate2 *iface, BYTE *pbSecurityId, DWORD *pcbSecurityId, DWORD_PTR dwReserved) { BSCallback *This = impl_from_IHttpNegotiate2(iface); FIXME("(%p)->(%p %p %ld)\n", This, pbSecurityId, pcbSecurityId, dwReserved); return E_NOTIMPL; } static const IHttpNegotiate2Vtbl HttpNegotiate2Vtbl = { HttpNegotiate_QueryInterface, HttpNegotiate_AddRef, HttpNegotiate_Release, HttpNegotiate_BeginningTransaction, HttpNegotiate_OnResponse, HttpNegotiate_GetRootSecurityId }; static inline BSCallback *impl_from_IInternetBindInfo(IInternetBindInfo *iface) { return CONTAINING_RECORD(iface, BSCallback, IInternetBindInfo_iface); } static HRESULT WINAPI InternetBindInfo_QueryInterface(IInternetBindInfo *iface, REFIID riid, void **ppv) { BSCallback *This = impl_from_IInternetBindInfo(iface); return IBindStatusCallback_QueryInterface(&This->IBindStatusCallback_iface, riid, ppv); } static ULONG WINAPI InternetBindInfo_AddRef(IInternetBindInfo *iface) { BSCallback *This = impl_from_IInternetBindInfo(iface); return IBindStatusCallback_AddRef(&This->IBindStatusCallback_iface); } static ULONG WINAPI InternetBindInfo_Release(IInternetBindInfo *iface) { BSCallback *This = impl_from_IInternetBindInfo(iface); return IBindStatusCallback_Release(&This->IBindStatusCallback_iface); } static HRESULT WINAPI InternetBindInfo_GetBindInfo(IInternetBindInfo *iface, DWORD *grfBINDF, BINDINFO *pbindinfo) { BSCallback *This = impl_from_IInternetBindInfo(iface); FIXME("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo); return E_NOTIMPL; } static HRESULT WINAPI InternetBindInfo_GetBindString(IInternetBindInfo *iface, ULONG ulStringType, LPOLESTR *ppwzStr, ULONG cEl, ULONG *pcElFetched) { BSCallback *This = impl_from_IInternetBindInfo(iface); FIXME("(%p)->(%u %p %u %p)\n", This, ulStringType, ppwzStr, cEl, pcElFetched); return E_NOTIMPL; } static const IInternetBindInfoVtbl InternetBindInfoVtbl = { InternetBindInfo_QueryInterface, InternetBindInfo_AddRef, InternetBindInfo_Release, InternetBindInfo_GetBindInfo, InternetBindInfo_GetBindString }; static inline BSCallback *impl_from_IServiceProvider(IServiceProvider *iface) { return CONTAINING_RECORD(iface, BSCallback, IServiceProvider_iface); } static HRESULT WINAPI BSCServiceProvider_QueryInterface(IServiceProvider *iface, REFIID riid, void **ppv) { BSCallback *This = impl_from_IServiceProvider(iface); return IBindStatusCallback_QueryInterface(&This->IBindStatusCallback_iface, riid, ppv); } static ULONG WINAPI BSCServiceProvider_AddRef(IServiceProvider *iface) { BSCallback *This = impl_from_IServiceProvider(iface); return IBindStatusCallback_AddRef(&This->IBindStatusCallback_iface); } static ULONG WINAPI BSCServiceProvider_Release(IServiceProvider *iface) { BSCallback *This = impl_from_IServiceProvider(iface); return IBindStatusCallback_Release(&This->IBindStatusCallback_iface); } static HRESULT WINAPI BSCServiceProvider_QueryService(IServiceProvider *iface, REFGUID guidService, REFIID riid, void **ppv) { BSCallback *This = impl_from_IServiceProvider(iface); TRACE("(%p)->(%s %s %p)\n", This, debugstr_guid(guidService), debugstr_guid(riid), ppv); if(This->window && IsEqualGUID(guidService, &IID_IWindowForBindingUI)) return IServiceProvider_QueryService(&This->window->base.IServiceProvider_iface, guidService, riid, ppv); return E_NOINTERFACE; } static const IServiceProviderVtbl ServiceProviderVtbl = { BSCServiceProvider_QueryInterface, BSCServiceProvider_AddRef, BSCServiceProvider_Release, BSCServiceProvider_QueryService }; static void init_bscallback(BSCallback *This, const BSCallbackVtbl *vtbl, IMoniker *mon, DWORD bindf) { This->IBindStatusCallback_iface.lpVtbl = &BindStatusCallbackVtbl; This->IServiceProvider_iface.lpVtbl = &ServiceProviderVtbl; This->IHttpNegotiate2_iface.lpVtbl = &HttpNegotiate2Vtbl; This->IInternetBindInfo_iface.lpVtbl = &InternetBindInfoVtbl; This->vtbl = vtbl; This->ref = 1; This->bindf = bindf; This->bom = BOM_NONE; list_init(&This->entry); if(mon) IMoniker_AddRef(mon); This->mon = mon; } static HRESULT read_stream(BSCallback *This, IStream *stream, void *buf, DWORD size, DWORD *ret_size) { DWORD read_size = 0, skip=0; BYTE *data = buf; HRESULT hres; hres = IStream_Read(stream, buf, size, &read_size); if(!This->readed && This->bom == BOM_NONE) { if(read_size >= 2 && data[0] == 0xff && data[1] == 0xfe) { This->bom = BOM_UTF16; skip = 2; }else if(read_size >= 3 && data[0] == 0xef && data[1] == 0xbb && data[2] == 0xbf) { This->bom = BOM_UTF8; skip = 3; } if(skip) { read_size -= skip; if(read_size) memmove(data, data+skip, read_size); } } This->readed += read_size; *ret_size = read_size; return hres; } static void parse_content_type(nsChannelBSC *This, const WCHAR *value) { const WCHAR *ptr; size_t len; static const WCHAR charsetW[] = {'c','h','a','r','s','e','t','='}; ptr = strchrW(value, ';'); if(!ptr) return; ptr++; while(*ptr && isspaceW(*ptr)) ptr++; len = strlenW(value); if(ptr + sizeof(charsetW)/sizeof(WCHAR) < value+len && !memicmpW(ptr, charsetW, sizeof(charsetW)/sizeof(WCHAR))) { size_t charset_len, lena; nsACString charset_str; const WCHAR *charset; char *charseta; ptr += sizeof(charsetW)/sizeof(WCHAR); if(*ptr == '\'') { FIXME("Quoted value\n"); return; }else { charset = ptr; while(*ptr && *ptr != ',') ptr++; charset_len = ptr-charset; } lena = WideCharToMultiByte(CP_ACP, 0, charset, charset_len, NULL, 0, NULL, NULL); charseta = heap_alloc(lena+1); if(!charseta) return; WideCharToMultiByte(CP_ACP, 0, charset, charset_len, charseta, lena, NULL, NULL); charseta[lena] = 0; nsACString_InitDepend(&charset_str, charseta); nsIHttpChannel_SetContentCharset(&This->nschannel->nsIHttpChannel_iface, &charset_str); nsACString_Finish(&charset_str); heap_free(charseta); }else { FIXME("unhandled: %s\n", debugstr_wn(ptr, len - (ptr-value))); } } static HRESULT parse_headers(const WCHAR *headers, struct list *headers_list) { const WCHAR *header, *header_end, *colon, *value; HRESULT hres; header = headers; while(*header) { if(header[0] == '\r' && header[1] == '\n' && !header[2]) break; for(colon = header; *colon && *colon != ':' && *colon != '\r'; colon++); if(*colon != ':') return E_FAIL; value = colon+1; while(*value == ' ') value++; if(!*value) return E_FAIL; for(header_end = value+1; *header_end && *header_end != '\r'; header_end++); hres = set_http_header(headers_list, header, colon-header, value, header_end-value); if(FAILED(hres)) return hres; header = header_end; if(header[0] == '\r' && header[1] == '\n') header += 2; } return S_OK; } static HRESULT process_response_headers(nsChannelBSC *This, const WCHAR *headers) { http_header_t *iter; HRESULT hres; static const WCHAR content_typeW[] = {'c','o','n','t','e','n','t','-','t','y','p','e',0}; hres = parse_headers(headers, &This->nschannel->response_headers); if(FAILED(hres)) return hres; LIST_FOR_EACH_ENTRY(iter, &This->nschannel->response_headers, http_header_t, entry) { if(!strcmpiW(iter->header, content_typeW)) parse_content_type(This, iter->data); } return S_OK; } HRESULT start_binding(HTMLInnerWindow *inner_window, BSCallback *bscallback, IBindCtx *bctx) { IStream *str = NULL; HRESULT hres; TRACE("(%p %p %p)\n", inner_window, bscallback, bctx); bscallback->window = inner_window; /* NOTE: IE7 calls IsSystemMoniker here*/ if(bctx) { RegisterBindStatusCallback(bctx, &bscallback->IBindStatusCallback_iface, NULL, 0); IBindCtx_AddRef(bctx); }else { hres = CreateAsyncBindCtx(0, &bscallback->IBindStatusCallback_iface, NULL, &bctx); if(FAILED(hres)) { WARN("CreateAsyncBindCtx failed: %08x\n", hres); bscallback->vtbl->stop_binding(bscallback, hres); return hres; } } hres = IMoniker_BindToStorage(bscallback->mon, bctx, NULL, &IID_IStream, (void**)&str); IBindCtx_Release(bctx); if(FAILED(hres)) { WARN("BindToStorage failed: %08x\n", hres); bscallback->vtbl->stop_binding(bscallback, hres); return hres; } if(str) IStream_Release(str); IMoniker_Release(bscallback->mon); bscallback->mon = NULL; return S_OK; } typedef struct { BSCallback bsc; DWORD size; char *buf; HRESULT hres; } BufferBSC; static inline BufferBSC *BufferBSC_from_BSCallback(BSCallback *iface) { return CONTAINING_RECORD(iface, BufferBSC, bsc); } static void BufferBSC_destroy(BSCallback *bsc) { BufferBSC *This = BufferBSC_from_BSCallback(bsc); heap_free(This->buf); heap_free(This); } static HRESULT BufferBSC_init_bindinfo(BSCallback *bsc) { return S_OK; } static HRESULT BufferBSC_start_binding(BSCallback *bsc) { return S_OK; } static HRESULT BufferBSC_stop_binding(BSCallback *bsc, HRESULT result) { BufferBSC *This = BufferBSC_from_BSCallback(bsc); This->hres = result; if(FAILED(result)) { heap_free(This->buf); This->buf = NULL; This->size = 0; } return S_OK; } static HRESULT BufferBSC_read_data(BSCallback *bsc, IStream *stream) { BufferBSC *This = BufferBSC_from_BSCallback(bsc); DWORD readed; HRESULT hres; if(!This->buf) { This->size = 128; This->buf = heap_alloc(This->size); } do { if(This->bsc.readed >= This->size) { This->size <<= 1; This->buf = heap_realloc(This->buf, This->size); } hres = read_stream(&This->bsc, stream, This->buf+This->bsc.readed, This->size-This->bsc.readed, &readed); }while(hres == S_OK); return S_OK; } static HRESULT BufferBSC_on_progress(BSCallback *bsc, ULONG status_code, LPCWSTR status_text) { return S_OK; } static HRESULT BufferBSC_on_response(BSCallback *bsc, DWORD response_code, LPCWSTR response_headers) { return S_OK; } static HRESULT BufferBSC_beginning_transaction(BSCallback *bsc, WCHAR **additional_headers) { return S_FALSE; } static const BSCallbackVtbl BufferBSCVtbl = { BufferBSC_destroy, BufferBSC_init_bindinfo, BufferBSC_start_binding, BufferBSC_stop_binding, BufferBSC_read_data, BufferBSC_on_progress, BufferBSC_on_response, BufferBSC_beginning_transaction }; static BufferBSC *create_bufferbsc(IMoniker *mon) { BufferBSC *ret = heap_alloc_zero(sizeof(*ret)); init_bscallback(&ret->bsc, &BufferBSCVtbl, mon, 0); ret->hres = E_FAIL; return ret; } HRESULT bind_mon_to_wstr(HTMLInnerWindow *window, IMoniker *mon, WCHAR **ret) { BufferBSC *bsc = create_bufferbsc(mon); int cp = CP_ACP; WCHAR *text; HRESULT hres; hres = start_binding(window, &bsc->bsc, NULL); if(SUCCEEDED(hres)) hres = bsc->hres; if(FAILED(hres)) { IBindStatusCallback_Release(&bsc->bsc.IBindStatusCallback_iface); return hres; } if(!bsc->bsc.readed) { *ret = NULL; return S_OK; } switch(bsc->bsc.bom) { case BOM_UTF16: if(bsc->bsc.readed % sizeof(WCHAR)) { FIXME("The buffer is not a valid utf16 string\n"); hres = E_FAIL; break; } text = heap_alloc(bsc->bsc.readed+sizeof(WCHAR)); if(!text) { hres = E_OUTOFMEMORY; break; } memcpy(text, bsc->buf, bsc->bsc.readed); text[bsc->bsc.readed/sizeof(WCHAR)] = 0; break; case BOM_UTF8: cp = CP_UTF8; /* fallthrough */ default: { DWORD len; len = MultiByteToWideChar(cp, 0, bsc->buf, bsc->bsc.readed, NULL, 0); text = heap_alloc((len+1)*sizeof(WCHAR)); if(!text) { hres = E_OUTOFMEMORY; break; } MultiByteToWideChar(CP_ACP, 0, bsc->buf, bsc->bsc.readed, text, len); text[len] = 0; } } IBindStatusCallback_Release(&bsc->bsc.IBindStatusCallback_iface); if(FAILED(hres)) return hres; *ret = text; return S_OK; } static HRESULT read_post_data_stream(nsChannelBSC *This, nsChannel *nschannel) { PRUint32 data_len = 0, available = 0; char *data, *post_data; nsresult nsres; HRESULT hres = S_OK; if(!nschannel->post_data_stream) return S_OK; nsres = nsIInputStream_Available(nschannel->post_data_stream, &available); if(NS_FAILED(nsres)) return E_FAIL; post_data = data = GlobalAlloc(0, available); if(!data) return E_OUTOFMEMORY; nsres = nsIInputStream_Read(nschannel->post_data_stream, data, available, &data_len); if(NS_FAILED(nsres)) { GlobalFree(data); return E_FAIL; } if(nschannel->post_data_contains_headers) { if(data_len >= 2 && data[0] == '\r' && data[1] == '\n') { post_data = data+2; data_len -= 2; }else { WCHAR *headers; DWORD size; char *ptr; post_data += data_len; for(ptr = data; ptr+4 < data+data_len; ptr++) { if(!memcmp(ptr, "\r\n\r\n", 4)) { post_data = ptr+4; break; } } data_len -= post_data-data; size = MultiByteToWideChar(CP_ACP, 0, data, post_data-data, NULL, 0); headers = heap_alloc((size+1)*sizeof(WCHAR)); if(headers) { MultiByteToWideChar(CP_ACP, 0, data, post_data-data, headers, size); headers[size] = 0; hres = parse_headers(headers , &nschannel->request_headers); if(SUCCEEDED(hres)) This->bsc.headers = headers; else heap_free(headers); }else { hres = E_OUTOFMEMORY; } } } if(FAILED(hres)) { GlobalFree(data); return hres; } if(!data_len) { GlobalFree(data); post_data = NULL; }else if(post_data != data) { char *new_data; new_data = GlobalAlloc(0, data_len); if(new_data) memcpy(new_data, post_data, data_len); GlobalFree(data); if(!new_data) return E_OUTOFMEMORY; post_data = new_data; } This->bsc.post_data = post_data; This->bsc.post_data_len = data_len; TRACE("post_data = %s\n", debugstr_a(This->bsc.post_data)); return S_OK; } static HRESULT on_start_nsrequest(nsChannelBSC *This) { nsresult nsres; /* FIXME: it's needed for http connections from BindToObject. */ if(!This->nschannel->response_status) This->nschannel->response_status = 200; nsres = nsIStreamListener_OnStartRequest(This->nslistener, (nsIRequest*)&This->nschannel->nsIHttpChannel_iface, This->nscontext); if(NS_FAILED(nsres)) { FIXME("OnStartRequest failed: %08x\n", nsres); return E_FAIL; } if(This->is_doc_channel) { update_window_doc(This->bsc.window); if(This->bsc.window->base.outer_window->readystate != READYSTATE_LOADING) set_ready_state(This->bsc.window->base.outer_window, READYSTATE_LOADING); } return S_OK; } static void on_stop_nsrequest(nsChannelBSC *This, HRESULT result) { nsresult nsres, request_result; switch(result) { case S_OK: request_result = NS_OK; break; case E_ABORT: request_result = NS_BINDING_ABORTED; break; default: request_result = NS_ERROR_FAILURE; } if(This->nslistener) { nsres = nsIStreamListener_OnStopRequest(This->nslistener, (nsIRequest*)&This->nschannel->nsIHttpChannel_iface, This->nscontext, request_result); if(NS_FAILED(nsres)) WARN("OnStopRequest failed: %08x\n", nsres); } if(This->nschannel->load_group) { nsres = nsILoadGroup_RemoveRequest(This->nschannel->load_group, (nsIRequest*)&This->nschannel->nsIHttpChannel_iface, NULL, request_result); if(NS_FAILED(nsres)) ERR("RemoveRequest failed: %08x\n", nsres); } } static HRESULT read_stream_data(nsChannelBSC *This, IStream *stream) { DWORD read; nsresult nsres; HRESULT hres; if(!This->nslistener) { BYTE buf[1024]; do { hres = read_stream(&This->bsc, stream, buf, sizeof(buf), &read); }while(hres == S_OK && read); return S_OK; } if(!This->nsstream) This->nsstream = create_nsprotocol_stream(); do { BOOL first_read = !This->bsc.readed; hres = read_stream(&This->bsc, stream, This->nsstream->buf+This->nsstream->buf_size, sizeof(This->nsstream->buf)-This->nsstream->buf_size, &read); if(!read) break; This->nsstream->buf_size += read; if(first_read) { switch(This->bsc.bom) { case BOM_UTF8: This->nschannel->charset = heap_strdupA(UTF8_STR); break; case BOM_UTF16: This->nschannel->charset = heap_strdupA(UTF16_STR); } if(!This->nschannel->content_type) { WCHAR *mime; hres = FindMimeFromData(NULL, NULL, This->nsstream->buf, This->nsstream->buf_size, This->is_doc_channel ? text_htmlW : NULL, 0, &mime, 0); if(FAILED(hres)) return hres; TRACE("Found MIME %s\n", debugstr_w(mime)); This->nschannel->content_type = heap_strdupWtoA(mime); CoTaskMemFree(mime); if(!This->nschannel->content_type) return E_OUTOFMEMORY; } on_start_nsrequest(This); } nsres = nsIStreamListener_OnDataAvailable(This->nslistener, (nsIRequest*)&This->nschannel->nsIHttpChannel_iface, This->nscontext, &This->nsstream->nsIInputStream_iface, This->bsc.readed-This->nsstream->buf_size, This->nsstream->buf_size); if(NS_FAILED(nsres)) ERR("OnDataAvailable failed: %08x\n", nsres); if(This->nsstream->buf_size == sizeof(This->nsstream->buf)) { ERR("buffer is full\n"); break; } }while(hres == S_OK); return S_OK; } typedef struct { nsIAsyncVerifyRedirectCallback nsIAsyncVerifyRedirectCallback_iface; LONG ref; nsChannel *nschannel; nsChannelBSC *bsc; } nsRedirectCallback; static nsRedirectCallback *impl_from_nsIAsyncVerifyRedirectCallback(nsIAsyncVerifyRedirectCallback *iface) { return CONTAINING_RECORD(iface, nsRedirectCallback, nsIAsyncVerifyRedirectCallback_iface); } static nsresult NSAPI nsAsyncVerifyRedirectCallback_QueryInterface(nsIAsyncVerifyRedirectCallback *iface, nsIIDRef riid, void **result) { nsRedirectCallback *This = impl_from_nsIAsyncVerifyRedirectCallback(iface); if(IsEqualGUID(&IID_nsISupports, riid)) { TRACE("(%p)->(IID_nsISuports %p)\n", This, result); *result = &This->nsIAsyncVerifyRedirectCallback_iface; }else if(IsEqualGUID(&IID_nsIAsyncVerifyRedirectCallback, riid)) { TRACE("(%p)->(IID_nsIAsyncVerifyRedirectCallback %p)\n", This, result); *result = &This->nsIAsyncVerifyRedirectCallback_iface; }else { *result = NULL; WARN("unimplemented iface %s\n", debugstr_guid(riid)); return NS_NOINTERFACE; } nsISupports_AddRef((nsISupports*)*result); return NS_OK; } static nsrefcnt NSAPI nsAsyncVerifyRedirectCallback_AddRef(nsIAsyncVerifyRedirectCallback *iface) { nsRedirectCallback *This = impl_from_nsIAsyncVerifyRedirectCallback(iface); LONG ref = InterlockedIncrement(&This->ref); TRACE("(%p) ref=%d\n", This, ref); return ref; } static nsrefcnt NSAPI nsAsyncVerifyRedirectCallback_Release(nsIAsyncVerifyRedirectCallback *iface) { nsRedirectCallback *This = impl_from_nsIAsyncVerifyRedirectCallback(iface); LONG ref = InterlockedDecrement(&This->ref); TRACE("(%p) ref=%d\n", This, ref); if(!ref) { IBindStatusCallback_Release(&This->bsc->bsc.IBindStatusCallback_iface); nsIHttpChannel_Release(&This->nschannel->nsIHttpChannel_iface); heap_free(This); } return ref; } static nsresult NSAPI nsAsyncVerifyRedirectCallback_AsyncOnChannelRedirect(nsIAsyncVerifyRedirectCallback *iface, nsresult result) { nsRedirectCallback *This = impl_from_nsIAsyncVerifyRedirectCallback(iface); nsChannel *old_nschannel; nsresult nsres; TRACE("(%p)->(%08x)\n", This, result); old_nschannel = This->bsc->nschannel; nsIHttpChannel_AddRef(&This->nschannel->nsIHttpChannel_iface); This->bsc->nschannel = This->nschannel; if(This->nschannel->load_group) { nsres = nsILoadGroup_AddRequest(This->nschannel->load_group, (nsIRequest*)&This->nschannel->nsIHttpChannel_iface, NULL); if(NS_FAILED(nsres)) ERR("AddRequest failed: %08x\n", nsres); } if(This->bsc->is_doc_channel) { IUri *uri = nsuri_get_uri(This->nschannel->uri); if(uri) { set_current_uri(This->bsc->bsc.window->base.outer_window, uri); IUri_Release(uri); }else { WARN("Could not get IUri from nsWineURI\n"); } } if(old_nschannel) { nsres = nsILoadGroup_RemoveRequest(old_nschannel->load_group, (nsIRequest*)&old_nschannel->nsIHttpChannel_iface, NULL, NS_OK); if(NS_FAILED(nsres)) ERR("RemoveRequest failed: %08x\n", nsres); nsIHttpChannel_Release(&old_nschannel->nsIHttpChannel_iface); } return NS_OK; } static const nsIAsyncVerifyRedirectCallbackVtbl nsAsyncVerifyRedirectCallbackVtbl = { nsAsyncVerifyRedirectCallback_QueryInterface, nsAsyncVerifyRedirectCallback_AddRef, nsAsyncVerifyRedirectCallback_Release, nsAsyncVerifyRedirectCallback_AsyncOnChannelRedirect }; static HRESULT create_redirect_callback(nsChannel *nschannel, nsChannelBSC *bsc, nsRedirectCallback **ret) { nsRedirectCallback *callback; callback = heap_alloc(sizeof(*callback)); if(!callback) return E_OUTOFMEMORY; callback->nsIAsyncVerifyRedirectCallback_iface.lpVtbl = &nsAsyncVerifyRedirectCallbackVtbl; callback->ref = 1; nsIHttpChannel_AddRef(&nschannel->nsIHttpChannel_iface); callback->nschannel = nschannel; IBindStatusCallback_AddRef(&bsc->bsc.IBindStatusCallback_iface); callback->bsc = bsc; *ret = callback; return S_OK; } static inline nsChannelBSC *nsChannelBSC_from_BSCallback(BSCallback *iface) { return CONTAINING_RECORD(iface, nsChannelBSC, bsc); } static void nsChannelBSC_destroy(BSCallback *bsc) { nsChannelBSC *This = nsChannelBSC_from_BSCallback(bsc); if(This->nschannel) nsIHttpChannel_Release(&This->nschannel->nsIHttpChannel_iface); if(This->nslistener) nsIStreamListener_Release(This->nslistener); if(This->nscontext) nsISupports_Release(This->nscontext); if(This->nsstream) nsIInputStream_Release(&This->nsstream->nsIInputStream_iface); heap_free(This); } static HRESULT nsChannelBSC_start_binding(BSCallback *bsc) { nsChannelBSC *This = nsChannelBSC_from_BSCallback(bsc); if(This->is_doc_channel) This->bsc.window->base.outer_window->base.inner_window->doc->skip_mutation_notif = FALSE; return S_OK; } static HRESULT nsChannelBSC_init_bindinfo(BSCallback *bsc) { nsChannelBSC *This = nsChannelBSC_from_BSCallback(bsc); HRESULT hres; if(This->nschannel && This->nschannel->post_data_stream) { hres = read_post_data_stream(This, This->nschannel); if(FAILED(hres)) return hres; } return S_OK; } typedef struct { task_t header; nsChannelBSC *bsc; } stop_request_task_t; static void stop_request_proc(task_t *_task) { stop_request_task_t *task = (stop_request_task_t*)_task; TRACE("(%p)\n", task->bsc); list_remove(&task->bsc->bsc.entry); list_init(&task->bsc->bsc.entry); on_stop_nsrequest(task->bsc, S_OK); } static void stop_request_task_destr(task_t *_task) { stop_request_task_t *task = (stop_request_task_t*)_task; IBindStatusCallback_Release(&task->bsc->bsc.IBindStatusCallback_iface); heap_free(task); } static HRESULT async_stop_request(nsChannelBSC *This) { stop_request_task_t *task; if(!This->bsc.readed) { TRACE("No data read, calling OnStartRequest\n"); on_start_nsrequest(This); } task = heap_alloc(sizeof(*task)); if(!task) return E_OUTOFMEMORY; IBindStatusCallback_AddRef(&This->bsc.IBindStatusCallback_iface); task->bsc = This; push_task(&task->header, stop_request_proc, stop_request_task_destr, This->bsc.window->task_magic); return S_OK; } static void handle_navigation_error(nsChannelBSC *This, DWORD result) { HTMLOuterWindow *outer_window; HTMLDocumentObj *doc; IOleCommandTarget *olecmd; BOOL is_error_url; SAFEARRAY *sa; SAFEARRAYBOUND bound; VARIANT var, varOut; LONG ind; BSTR unk; HRESULT hres; if(!This->is_doc_channel || !This->bsc.window) return; outer_window = This->bsc.window->base.outer_window; doc = outer_window->doc_obj; if(!doc || !doc->doc_object_service || !doc->client) return; hres = IDocObjectService_IsErrorUrl(doc->doc_object_service, outer_window->url, &is_error_url); if(FAILED(hres) || is_error_url) return; hres = IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&olecmd); if(FAILED(hres)) return; bound.lLbound = 0; bound.cElements = 8; sa = SafeArrayCreate(VT_VARIANT, 1, &bound); if(!sa) { IOleCommandTarget_Release(olecmd); return; } ind = 0; V_VT(&var) = VT_I4; V_I4(&var) = result; SafeArrayPutElement(sa, &ind, &var); ind = 1; V_VT(&var) = VT_BSTR; V_BSTR(&var) = outer_window->url; SafeArrayPutElement(sa, &ind, &var); ind = 3; V_VT(&var) = VT_UNKNOWN; V_UNKNOWN(&var) = (IUnknown*)&outer_window->base.IHTMLWindow2_iface; SafeArrayPutElement(sa, &ind, &var); /* FIXME: what are the following fields for? */ ind = 2; V_VT(&var) = VT_UNKNOWN; V_UNKNOWN(&var) = NULL; SafeArrayPutElement(sa, &ind, &var); ind = 4; V_VT(&var) = VT_BOOL; V_BOOL(&var) = FALSE; SafeArrayPutElement(sa, &ind, &var); ind = 5; V_VT(&var) = VT_BOOL; V_BOOL(&var) = FALSE; SafeArrayPutElement(sa, &ind, &var); ind = 6; V_VT(&var) = VT_BSTR; unk = SysAllocString(NULL); V_BSTR(&var) = unk; SafeArrayPutElement(sa, &ind, &var); ind = 7; V_VT(&var) = VT_UNKNOWN; V_UNKNOWN(&var) = NULL; SafeArrayPutElement(sa, &ind, &var); V_VT(&var) = VT_ARRAY; V_ARRAY(&var) = sa; V_VT(&varOut) = VT_BOOL; V_BOOL(&varOut) = VARIANT_TRUE; IOleCommandTarget_Exec(olecmd, &CGID_DocHostCmdPriv, 1, 0, &var, FAILED(hres)?NULL:&varOut); SysFreeString(unk); SafeArrayDestroy(sa); IOleCommandTarget_Release(olecmd); } static HRESULT nsChannelBSC_stop_binding(BSCallback *bsc, HRESULT result) { nsChannelBSC *This = nsChannelBSC_from_BSCallback(bsc); if(result != E_ABORT) { if(FAILED(result)) handle_navigation_error(This, result); else if(This->is_doc_channel) { result = async_stop_request(This); if(SUCCEEDED(result)) return S_OK; } } on_stop_nsrequest(This, result); return S_OK; } static HRESULT nsChannelBSC_read_data(BSCallback *bsc, IStream *stream) { nsChannelBSC *This = nsChannelBSC_from_BSCallback(bsc); return read_stream_data(This, stream); } static HRESULT handle_redirect(nsChannelBSC *This, const WCHAR *new_url) { nsRedirectCallback *callback; nsIChannelEventSink *sink; nsChannel *new_channel; nsresult nsres; HRESULT hres; TRACE("(%p)->(%s)\n", This, debugstr_w(new_url)); if(!This->nschannel || !This->nschannel->notif_callback) return S_OK; nsres = nsIInterfaceRequestor_GetInterface(This->nschannel->notif_callback, &IID_nsIChannelEventSink, (void**)&sink); if(NS_FAILED(nsres)) return S_OK; hres = create_redirect_nschannel(new_url, This->nschannel, &new_channel); if(SUCCEEDED(hres)) { TRACE("%p %p->%p\n", This, This->nschannel, new_channel); hres = create_redirect_callback(new_channel, This, &callback); nsIHttpChannel_Release(&new_channel->nsIHttpChannel_iface); } if(SUCCEEDED(hres)) { nsres = nsIChannelEventSink_AsyncOnChannelRedirect(sink, (nsIChannel*)&This->nschannel->nsIHttpChannel_iface, (nsIChannel*)&callback->nschannel->nsIHttpChannel_iface, REDIRECT_TEMPORARY, /* FIXME */ &callback->nsIAsyncVerifyRedirectCallback_iface); if(NS_FAILED(nsres)) FIXME("AsyncOnChannelRedirect failed: %08x\n", hres); else if(This->nschannel != callback->nschannel) FIXME("nschannel not updated\n"); nsIAsyncVerifyRedirectCallback_Release(&callback->nsIAsyncVerifyRedirectCallback_iface); } nsIChannelEventSink_Release(sink); return hres; } static HRESULT nsChannelBSC_on_progress(BSCallback *bsc, ULONG status_code, LPCWSTR status_text) { nsChannelBSC *This = nsChannelBSC_from_BSCallback(bsc); switch(status_code) { case BINDSTATUS_MIMETYPEAVAILABLE: if(!This->nschannel) return S_OK; heap_free(This->nschannel->content_type); This->nschannel->content_type = heap_strdupWtoA(status_text); break; case BINDSTATUS_REDIRECTING: return handle_redirect(This, status_text); case BINDSTATUS_BEGINDOWNLOADDATA: { IWinInetHttpInfo *http_info; DWORD status, size = sizeof(DWORD); HRESULT hres; if(!This->bsc.binding) break; hres = IBinding_QueryInterface(This->bsc.binding, &IID_IWinInetHttpInfo, (void**)&http_info); if(FAILED(hres)) break; hres = IWinInetHttpInfo_QueryInfo(http_info, HTTP_QUERY_STATUS_CODE|HTTP_QUERY_FLAG_NUMBER, &status, &size, NULL, NULL); IWinInetHttpInfo_Release(http_info); if(FAILED(hres) || status == HTTP_STATUS_OK) break; handle_navigation_error(This, status); } } return S_OK; } static HRESULT nsChannelBSC_on_response(BSCallback *bsc, DWORD response_code, LPCWSTR response_headers) { nsChannelBSC *This = nsChannelBSC_from_BSCallback(bsc); HRESULT hres; This->nschannel->response_status = response_code; if(response_headers) { const WCHAR *headers; headers = strchrW(response_headers, '\r'); if(headers && headers[1] == '\n') { headers += 2; hres = process_response_headers(This, headers); if(FAILED(hres)) { WARN("parsing headers failed: %08x\n", hres); return hres; } } } return S_OK; } static HRESULT nsChannelBSC_beginning_transaction(BSCallback *bsc, WCHAR **additional_headers) { nsChannelBSC *This = nsChannelBSC_from_BSCallback(bsc); http_header_t *iter; DWORD len = 0; WCHAR *ptr; static const WCHAR content_lengthW[] = {'C','o','n','t','e','n','t','-','L','e','n','g','t','h',0}; if(!This->nschannel) return S_FALSE; LIST_FOR_EACH_ENTRY(iter, &This->nschannel->request_headers, http_header_t, entry) { if(strcmpW(iter->header, content_lengthW)) len += strlenW(iter->header) + 2 /* ": " */ + strlenW(iter->data) + 2 /* "\r\n" */; } if(!len) return S_OK; *additional_headers = ptr = CoTaskMemAlloc((len+1)*sizeof(WCHAR)); if(!ptr) return E_OUTOFMEMORY; LIST_FOR_EACH_ENTRY(iter, &This->nschannel->request_headers, http_header_t, entry) { if(!strcmpW(iter->header, content_lengthW)) continue; len = strlenW(iter->header); memcpy(ptr, iter->header, len*sizeof(WCHAR)); ptr += len; *ptr++ = ':'; *ptr++ = ' '; len = strlenW(iter->data); memcpy(ptr, iter->data, len*sizeof(WCHAR)); ptr += len; *ptr++ = '\r'; *ptr++ = '\n'; } *ptr = 0; return S_OK; } static const BSCallbackVtbl nsChannelBSCVtbl = { nsChannelBSC_destroy, nsChannelBSC_init_bindinfo, nsChannelBSC_start_binding, nsChannelBSC_stop_binding, nsChannelBSC_read_data, nsChannelBSC_on_progress, nsChannelBSC_on_response, nsChannelBSC_beginning_transaction }; HRESULT create_channelbsc(IMoniker *mon, const WCHAR *headers, BYTE *post_data, DWORD post_data_size, BOOL is_doc_binding, nsChannelBSC **retval) { nsChannelBSC *ret; ret = heap_alloc_zero(sizeof(*ret)); if(!ret) return E_OUTOFMEMORY; init_bscallback(&ret->bsc, &nsChannelBSCVtbl, mon, BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA); ret->is_doc_channel = is_doc_binding; if(headers) { ret->bsc.headers = heap_strdupW(headers); if(!ret->bsc.headers) { IBindStatusCallback_Release(&ret->bsc.IBindStatusCallback_iface); return E_OUTOFMEMORY; } } if(post_data) { ret->bsc.post_data = GlobalAlloc(0, post_data_size); if(!ret->bsc.post_data) { heap_free(ret->bsc.headers); IBindStatusCallback_Release(&ret->bsc.IBindStatusCallback_iface); return E_OUTOFMEMORY; } memcpy(ret->bsc.post_data, post_data, post_data_size); ret->bsc.post_data_len = post_data_size; } *retval = ret; return S_OK; } typedef struct { task_t header; HTMLOuterWindow *window; HTMLInnerWindow *pending_window; } start_doc_binding_task_t; static void start_doc_binding_proc(task_t *_task) { start_doc_binding_task_t *task = (start_doc_binding_task_t*)_task; set_current_mon(task->window, task->pending_window->bscallback->bsc.mon); start_binding(task->pending_window, &task->pending_window->bscallback->bsc, NULL); } static void start_doc_binding_task_destr(task_t *_task) { start_doc_binding_task_t *task = (start_doc_binding_task_t*)_task; IHTMLWindow2_Release(&task->pending_window->base.IHTMLWindow2_iface); heap_free(task); } HRESULT async_start_doc_binding(HTMLOuterWindow *window, HTMLInnerWindow *pending_window) { start_doc_binding_task_t *task; TRACE("%p\n", pending_window); task = heap_alloc(sizeof(start_doc_binding_task_t)); if(!task) return E_OUTOFMEMORY; task->window = window; task->pending_window = pending_window; IHTMLWindow2_AddRef(&pending_window->base.IHTMLWindow2_iface); push_task(&task->header, start_doc_binding_proc, start_doc_binding_task_destr, pending_window->task_magic); return S_OK; } void abort_window_bindings(HTMLInnerWindow *window) { BSCallback *iter; remove_target_tasks(window->task_magic); while(!list_empty(&window->bindings)) { iter = LIST_ENTRY(window->bindings.next, BSCallback, entry); TRACE("Aborting %p\n", iter); IBindStatusCallback_AddRef(&iter->IBindStatusCallback_iface); if(iter->binding) IBinding_Abort(iter->binding); else iter->vtbl->stop_binding(iter, E_ABORT); iter->window = NULL; list_remove(&iter->entry); list_init(&iter->entry); IBindStatusCallback_Release(&iter->IBindStatusCallback_iface); } if(window->bscallback) { IBindStatusCallback_Release(&window->bscallback->bsc.IBindStatusCallback_iface); window->bscallback = NULL; } if(window->mon) { IMoniker_Release(window->mon); window->mon = NULL; } } HRESULT channelbsc_load_stream(HTMLInnerWindow *pending_window, IStream *stream) { nsChannelBSC *bscallback = pending_window->bscallback; HRESULT hres = S_OK; if(!bscallback->nschannel) { ERR("NULL nschannel\n"); return E_FAIL; } bscallback->nschannel->content_type = heap_strdupA("text/html"); if(!bscallback->nschannel->content_type) return E_OUTOFMEMORY; bscallback->bsc.window = pending_window; if(stream) hres = read_stream_data(bscallback, stream); if(SUCCEEDED(hres)) hres = async_stop_request(bscallback); if(FAILED(hres)) IBindStatusCallback_OnStopBinding(&bscallback->bsc.IBindStatusCallback_iface, hres, ERROR_SUCCESS); return hres; } void channelbsc_set_channel(nsChannelBSC *This, nsChannel *channel, nsIStreamListener *listener, nsISupports *context) { nsIHttpChannel_AddRef(&channel->nsIHttpChannel_iface); This->nschannel = channel; nsIStreamListener_AddRef(listener); This->nslistener = listener; if(context) { nsISupports_AddRef(context); This->nscontext = context; } if(This->bsc.headers) { HRESULT hres; hres = parse_headers(This->bsc.headers, &channel->request_headers); heap_free(This->bsc.headers); This->bsc.headers = NULL; if(FAILED(hres)) WARN("parse_headers failed: %08x\n", hres); } } typedef struct { task_t header; HTMLOuterWindow *window; IUri *uri; } navigate_javascript_task_t; static void navigate_javascript_proc(task_t *_task) { navigate_javascript_task_t *task = (navigate_javascript_task_t*)_task; HTMLOuterWindow *window = task->window; VARIANT v; BSTR code; HRESULT hres; static const WCHAR jscriptW[] = {'j','s','c','r','i','p','t',0}; task->window->readystate = READYSTATE_COMPLETE; hres = IUri_GetPath(task->uri, &code); if(FAILED(hres)) return; hres = UrlUnescapeW(code, NULL, NULL, URL_UNESCAPE_INPLACE); if(FAILED(hres)) { SysFreeString(code); return; } set_download_state(window->doc_obj, 1); V_VT(&v) = VT_EMPTY; hres = exec_script(window->base.inner_window, code, jscriptW, &v); SysFreeString(code); if(SUCCEEDED(hres) && V_VT(&v) != VT_EMPTY) { FIXME("javascirpt URL returned %s\n", debugstr_variant(&v)); VariantClear(&v); } if(window->doc_obj->view_sink) IAdviseSink_OnViewChange(window->doc_obj->view_sink, DVASPECT_CONTENT, -1); set_download_state(window->doc_obj, 0); } static void navigate_javascript_task_destr(task_t *_task) { navigate_javascript_task_t *task = (navigate_javascript_task_t*)_task; IUri_Release(task->uri); heap_free(task); } typedef struct { task_t header; HTMLOuterWindow *window; nsChannelBSC *bscallback; IMoniker *mon; } navigate_task_t; static void navigate_proc(task_t *_task) { navigate_task_t *task = (navigate_task_t*)_task; HRESULT hres; hres = set_moniker(&task->window->doc_obj->basedoc, task->mon, NULL, task->bscallback, TRUE); if(SUCCEEDED(hres)) { set_current_mon(task->window, task->bscallback->bsc.mon); start_binding(task->window->pending_window, &task->bscallback->bsc, NULL); } } static void navigate_task_destr(task_t *_task) { navigate_task_t *task = (navigate_task_t*)_task; IBindStatusCallback_Release(&task->bscallback->bsc.IBindStatusCallback_iface); IMoniker_Release(task->mon); heap_free(task); } static HRESULT navigate_fragment(HTMLOuterWindow *window, IUri *uri) { nsIDOMLocation *nslocation; nsAString nsfrag_str; BSTR frag; nsresult nsres; HRESULT hres; set_current_uri(window, uri); nsres = nsIDOMWindow_GetLocation(window->nswindow, &nslocation); if(FAILED(nsres) || !nslocation) return E_FAIL; hres = IUri_GetFragment(uri, &frag); if(FAILED(hres)) { nsIDOMLocation_Release(nslocation); return hres; } nsAString_InitDepend(&nsfrag_str, frag); nsres = nsIDOMLocation_SetHash(nslocation, &nsfrag_str); nsAString_Finish(&nsfrag_str); nsIDOMLocation_Release(nslocation); SysFreeString(frag); if(NS_FAILED(nsres)) { ERR("SetHash failed: %08x\n", nsres); return E_FAIL; } if(window->doc_obj->doc_object_service) { IDocObjectService_FireNavigateComplete2(window->doc_obj->doc_object_service, &window->base.IHTMLWindow2_iface, 0x10); IDocObjectService_FireDocumentComplete(window->doc_obj->doc_object_service, &window->base.IHTMLWindow2_iface, 0); } return S_OK; } HRESULT super_navigate(HTMLOuterWindow *window, IUri *uri, const WCHAR *headers, BYTE *post_data, DWORD post_data_size) { nsChannelBSC *bsc; IMoniker *mon; DWORD scheme; HRESULT hres; if(window->doc_obj->client) { IOleCommandTarget *cmdtrg; hres = IOleClientSite_QueryInterface(window->doc_obj->client, &IID_IOleCommandTarget, (void**)&cmdtrg); if(SUCCEEDED(hres)) { VARIANT in, out; BSTR url_str; hres = IUri_GetDisplayUri(uri, &url_str); if(SUCCEEDED(hres)) { V_VT(&in) = VT_BSTR; V_BSTR(&in) = url_str; V_VT(&out) = VT_BOOL; V_BOOL(&out) = VARIANT_TRUE; hres = IOleCommandTarget_Exec(cmdtrg, &CGID_ShellDocView, 67, 0, &in, &out); IOleCommandTarget_Release(cmdtrg); if(SUCCEEDED(hres)) VariantClear(&out); SysFreeString(url_str); } } } if(window->uri && !post_data_size && compare_ignoring_frag(window->uri, uri)) { TRACE("fragment navigate\n"); return navigate_fragment(window, uri); } hres = CreateURLMonikerEx2(NULL, uri, &mon, URL_MK_UNIFORM); if(FAILED(hres)) return hres; /* FIXME: Why not set_ready_state? */ window->readystate = READYSTATE_UNINITIALIZED; hres = create_channelbsc(mon, headers, post_data, post_data_size, TRUE, &bsc); if(FAILED(hres)) { IMoniker_Release(mon); return hres; } prepare_for_binding(&window->doc_obj->basedoc, mon, TRUE); hres = IUri_GetScheme(uri, &scheme); if(SUCCEEDED(hres) && scheme != URL_SCHEME_JAVASCRIPT) { navigate_task_t *task; task = heap_alloc(sizeof(*task)); if(!task) { IBindStatusCallback_Release(&bsc->bsc.IBindStatusCallback_iface); IMoniker_Release(mon); return E_OUTOFMEMORY; } /* Silently and repeated when real loading starts? */ window->readystate = READYSTATE_LOADING; call_docview_84(window->doc_obj); task->window = window; task->bscallback = bsc; task->mon = mon; push_task(&task->header, navigate_proc, navigate_task_destr, window->task_magic); }else { navigate_javascript_task_t *task; IBindStatusCallback_Release(&bsc->bsc.IBindStatusCallback_iface); IMoniker_Release(mon); task = heap_alloc(sizeof(*task)); if(!task) return E_OUTOFMEMORY; /* Why silently? */ window->readystate = READYSTATE_COMPLETE; call_docview_84(window->doc_obj); IUri_AddRef(uri); task->window = window; task->uri = uri; push_task(&task->header, navigate_javascript_proc, navigate_javascript_task_destr, window->task_magic); } return S_OK; } HRESULT navigate_new_window(HTMLOuterWindow *window, IUri *uri, const WCHAR *name, IHTMLWindow2 **ret) { IWebBrowser2 *web_browser; IHTMLWindow2 *new_window; IBindCtx *bind_ctx; nsChannelBSC *bsc; HRESULT hres; hres = create_channelbsc(NULL, NULL, NULL, 0, FALSE, &bsc); if(FAILED(hres)) return hres; hres = CreateAsyncBindCtx(0, &bsc->bsc.IBindStatusCallback_iface, NULL, &bind_ctx); if(FAILED(hres)) { IBindStatusCallback_Release(&bsc->bsc.IBindStatusCallback_iface); return hres; } hres = CoCreateInstance(&CLSID_InternetExplorer, NULL, CLSCTX_LOCAL_SERVER, &IID_IWebBrowser2, (void**)&web_browser); if(SUCCEEDED(hres)) { ITargetFramePriv2 *target_frame_priv; hres = IWebBrowser2_QueryInterface(web_browser, &IID_ITargetFramePriv2, (void**)&target_frame_priv); if(SUCCEEDED(hres)) { hres = ITargetFramePriv2_AggregatedNavigation2(target_frame_priv, HLNF_DISABLEWINDOWRESTRICTIONS|HLNF_OPENINNEWWINDOW, bind_ctx, &bsc->bsc.IBindStatusCallback_iface, name, uri, emptyW); ITargetFramePriv2_Release(target_frame_priv); if(SUCCEEDED(hres)) hres = do_query_service((IUnknown*)web_browser, &SID_SHTMLWindow, &IID_IHTMLWindow2, (void**)&new_window); } if(FAILED(hres)) { IWebBrowser2_Quit(web_browser); IWebBrowser2_Release(web_browser); } }else { WARN("Could not create InternetExplorer instance: %08x\n", hres); } IBindStatusCallback_Release(&bsc->bsc.IBindStatusCallback_iface); IBindCtx_Release(bind_ctx); if(FAILED(hres)) return hres; IWebBrowser2_put_Visible(web_browser, VARIANT_TRUE); IWebBrowser2_Release(web_browser); if(ret) *ret = new_window; else IHTMLWindow2_Release(new_window); return S_OK; } HRESULT hlink_frame_navigate(HTMLDocument *doc, LPCWSTR url, nsChannel *nschannel, DWORD hlnf, BOOL *cancel) { IHlinkFrame *hlink_frame; nsChannelBSC *callback; IBindCtx *bindctx; IMoniker *mon; IHlink *hlink; HRESULT hres; *cancel = FALSE; hres = do_query_service((IUnknown*)doc->doc_obj->client, &IID_IHlinkFrame, &IID_IHlinkFrame, (void**)&hlink_frame); if(FAILED(hres)) return S_OK; hres = create_channelbsc(NULL, NULL, NULL, 0, FALSE, &callback); if(FAILED(hres)) { IHlinkFrame_Release(hlink_frame); return hres; } if(nschannel) read_post_data_stream(callback, nschannel); hres = CreateAsyncBindCtx(0, &callback->bsc.IBindStatusCallback_iface, NULL, &bindctx); if(SUCCEEDED(hres)) hres = CoCreateInstance(&CLSID_StdHlink, NULL, CLSCTX_INPROC_SERVER, &IID_IHlink, (LPVOID*)&hlink); if(SUCCEEDED(hres)) hres = CreateURLMoniker(NULL, url, &mon); if(SUCCEEDED(hres)) { IHlink_SetMonikerReference(hlink, HLINKSETF_TARGET, mon, NULL); if(hlnf & HLNF_OPENINNEWWINDOW) { static const WCHAR wszBlank[] = {'_','b','l','a','n','k',0}; IHlink_SetTargetFrameName(hlink, wszBlank); /* FIXME */ } hres = IHlinkFrame_Navigate(hlink_frame, hlnf, bindctx, &callback->bsc.IBindStatusCallback_iface, hlink); IMoniker_Release(mon); *cancel = hres == S_OK; hres = S_OK; } IHlinkFrame_Release(hlink_frame); IBindCtx_Release(bindctx); IBindStatusCallback_Release(&callback->bsc.IBindStatusCallback_iface); return hres; } HRESULT navigate_url(HTMLOuterWindow *window, const WCHAR *new_url, const WCHAR *base_url) { WCHAR url[INTERNET_MAX_URL_LENGTH]; nsWineURI *uri; HRESULT hres; if(!new_url) { *url = 0; }else if(base_url) { DWORD len = 0; hres = CoInternetCombineUrl(base_url, new_url, URL_ESCAPE_SPACES_ONLY|URL_DONT_ESCAPE_EXTRA_INFO, url, sizeof(url)/sizeof(WCHAR), &len, 0); if(FAILED(hres)) return hres; }else { strcpyW(url, new_url); } if(window->doc_obj && window->doc_obj->hostui) { OLECHAR *translated_url = NULL; hres = IDocHostUIHandler_TranslateUrl(window->doc_obj->hostui, 0, url, &translated_url); if(hres == S_OK) { TRACE("%08x %s -> %s\n", hres, debugstr_w(url), debugstr_w(translated_url)); strcpyW(url, translated_url); CoTaskMemFree(translated_url); } } if(window->doc_obj && window->doc_obj->is_webbrowser && window == window->doc_obj->basedoc.window) { BOOL cancel = FALSE; IUri *uri; hres = IDocObjectService_FireBeforeNavigate2(window->doc_obj->doc_object_service, NULL, url, 0x40, NULL, NULL, 0, NULL, TRUE, &cancel); if(SUCCEEDED(hres) && cancel) { TRACE("Navigation canceled\n"); return S_OK; } hres = CreateUri(url, 0, 0, &uri); if(FAILED(hres)) return hres; hres = super_navigate(window, uri, NULL, NULL, 0); IUri_Release(uri); return hres; } if(window->doc_obj && window == window->doc_obj->basedoc.window) { BOOL cancel; hres = hlink_frame_navigate(&window->base.inner_window->doc->basedoc, url, NULL, 0, &cancel); if(FAILED(hres)) return hres; if(cancel) { TRACE("Navigation handled by hlink frame\n"); return S_OK; } } hres = create_doc_uri(window, url, &uri); if(FAILED(hres)) return hres; hres = load_nsuri(window, uri, NULL, LOAD_FLAGS_NONE); nsISupports_Release((nsISupports*)uri); return hres; }