mshtml: Call stop_binding before releasing binding object.
[wine/testsucceed.git] / dlls / mshtml / navigate.c
blobeee0b34f417491975f535d912caabe13b07cea80
1 /*
2 * Copyright 2006-2007 Jacek Caban for CodeWeavers
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.
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.
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
19 #include "config.h"
21 #include <stdarg.h>
23 #define COBJMACROS
24 #define NONAMELESSUNION
25 #define NONAMELESSSTRUCT
27 #include "windef.h"
28 #include "winbase.h"
29 #include "winuser.h"
30 #include "ole2.h"
31 #include "hlguids.h"
32 #include "shlguid.h"
34 #include "wine/debug.h"
35 #include "wine/unicode.h"
37 #include "mshtml_private.h"
39 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
41 #define CONTENT_LENGTH "Content-Length"
42 #define UTF16_STR "utf-16"
44 typedef struct {
45 const nsIInputStreamVtbl *lpInputStreamVtbl;
47 LONG ref;
49 char buf[1024];
50 DWORD buf_size;
51 } nsProtocolStream;
53 #define NSINSTREAM(x) ((nsIInputStream*) &(x)->lpInputStreamVtbl)
55 typedef struct {
56 void (*destroy)(BSCallback*);
57 HRESULT (*start_binding)(BSCallback*);
58 HRESULT (*stop_binding)(BSCallback*,HRESULT);
59 HRESULT (*read_data)(BSCallback*,IStream*);
60 HRESULT (*on_progress)(BSCallback*,ULONG,LPCWSTR);
61 } BSCallbackVtbl;
63 struct BSCallback {
64 const IBindStatusCallbackVtbl *lpBindStatusCallbackVtbl;
65 const IServiceProviderVtbl *lpServiceProviderVtbl;
66 const IHttpNegotiate2Vtbl *lpHttpNegotiate2Vtbl;
67 const IInternetBindInfoVtbl *lpInternetBindInfoVtbl;
69 const BSCallbackVtbl *vtbl;
71 LONG ref;
73 LPWSTR headers;
74 HGLOBAL post_data;
75 ULONG post_data_len;
76 ULONG readed;
77 DWORD bindf;
79 IMoniker *mon;
80 IBinding *binding;
82 HTMLDocument *doc;
84 struct list entry;
87 #define NSINSTREAM_THIS(iface) DEFINE_THIS(nsProtocolStream, InputStream, iface)
89 static nsresult NSAPI nsInputStream_QueryInterface(nsIInputStream *iface, nsIIDRef riid,
90 nsQIResult result)
92 nsProtocolStream *This = NSINSTREAM_THIS(iface);
94 *result = NULL;
96 if(IsEqualGUID(&IID_nsISupports, riid)) {
97 TRACE("(%p)->(IID_nsISupports %p)\n", This, result);
98 *result = NSINSTREAM(This);
99 }else if(IsEqualGUID(&IID_nsIInputStream, riid)) {
100 TRACE("(%p)->(IID_nsIInputStream %p)\n", This, result);
101 *result = NSINSTREAM(This);
104 if(*result) {
105 nsIInputStream_AddRef(NSINSTREAM(This));
106 return NS_OK;
109 WARN("unsupported interface %s\n", debugstr_guid(riid));
110 return NS_NOINTERFACE;
113 static nsrefcnt NSAPI nsInputStream_AddRef(nsIInputStream *iface)
115 nsProtocolStream *This = NSINSTREAM_THIS(iface);
116 LONG ref = InterlockedIncrement(&This->ref);
118 TRACE("(%p) ref=%d\n", This, ref);
120 return ref;
124 static nsrefcnt NSAPI nsInputStream_Release(nsIInputStream *iface)
126 nsProtocolStream *This = NSINSTREAM_THIS(iface);
127 LONG ref = InterlockedDecrement(&This->ref);
129 TRACE("(%p) ref=%d\n", This, ref);
131 if(!ref)
132 heap_free(This);
134 return ref;
137 static nsresult NSAPI nsInputStream_Close(nsIInputStream *iface)
139 nsProtocolStream *This = NSINSTREAM_THIS(iface);
140 FIXME("(%p)\n", This);
141 return NS_ERROR_NOT_IMPLEMENTED;
144 static nsresult NSAPI nsInputStream_Available(nsIInputStream *iface, PRUint32 *_retval)
146 nsProtocolStream *This = NSINSTREAM_THIS(iface);
147 FIXME("(%p)->(%p)\n", This, _retval);
148 return NS_ERROR_NOT_IMPLEMENTED;
151 static nsresult NSAPI nsInputStream_Read(nsIInputStream *iface, char *aBuf, PRUint32 aCount,
152 PRUint32 *_retval)
154 nsProtocolStream *This = NSINSTREAM_THIS(iface);
155 DWORD read = aCount;
157 TRACE("(%p)->(%p %d %p)\n", This, aBuf, aCount, _retval);
159 if(read > This->buf_size)
160 read = This->buf_size;
162 if(read) {
163 memcpy(aBuf, This->buf, read);
164 if(read < This->buf_size)
165 memmove(This->buf, This->buf+read, This->buf_size-read);
166 This->buf_size -= read;
169 *_retval = read;
170 return NS_OK;
173 static nsresult NSAPI nsInputStream_ReadSegments(nsIInputStream *iface,
174 nsresult (WINAPI *aWriter)(nsIInputStream*,void*,const char*,PRUint32,PRUint32,PRUint32*),
175 void *aClousure, PRUint32 aCount, PRUint32 *_retval)
177 nsProtocolStream *This = NSINSTREAM_THIS(iface);
178 PRUint32 written = 0;
179 nsresult nsres;
181 TRACE("(%p)->(%p %p %d %p)\n", This, aWriter, aClousure, aCount, _retval);
183 if(!This->buf_size)
184 return S_OK;
186 if(aCount > This->buf_size)
187 aCount = This->buf_size;
189 nsres = aWriter(NSINSTREAM(This), aClousure, This->buf, 0, aCount, &written);
190 if(NS_FAILED(nsres))
191 TRACE("aWritter failed: %08x\n", nsres);
192 else if(written != This->buf_size)
193 FIXME("written %d != buf_size %d\n", written, This->buf_size);
195 This->buf_size -= written;
197 *_retval = written;
198 return nsres;
201 static nsresult NSAPI nsInputStream_IsNonBlocking(nsIInputStream *iface, PRBool *_retval)
203 nsProtocolStream *This = NSINSTREAM_THIS(iface);
204 FIXME("(%p)->(%p)\n", This, _retval);
205 return NS_ERROR_NOT_IMPLEMENTED;
208 #undef NSINSTREAM_THIS
210 static const nsIInputStreamVtbl nsInputStreamVtbl = {
211 nsInputStream_QueryInterface,
212 nsInputStream_AddRef,
213 nsInputStream_Release,
214 nsInputStream_Close,
215 nsInputStream_Available,
216 nsInputStream_Read,
217 nsInputStream_ReadSegments,
218 nsInputStream_IsNonBlocking
221 static nsProtocolStream *create_nsprotocol_stream(void)
223 nsProtocolStream *ret = heap_alloc(sizeof(nsProtocolStream));
225 ret->lpInputStreamVtbl = &nsInputStreamVtbl;
226 ret->ref = 1;
227 ret->buf_size = 0;
229 return ret;
232 #define STATUSCLB_THIS(iface) DEFINE_THIS(BSCallback, BindStatusCallback, iface)
234 static HRESULT WINAPI BindStatusCallback_QueryInterface(IBindStatusCallback *iface,
235 REFIID riid, void **ppv)
237 BSCallback *This = STATUSCLB_THIS(iface);
239 *ppv = NULL;
240 if(IsEqualGUID(&IID_IUnknown, riid)) {
241 TRACE("(%p)->(IID_IUnknown, %p)\n", This, ppv);
242 *ppv = STATUSCLB(This);
243 }else if(IsEqualGUID(&IID_IBindStatusCallback, riid)) {
244 TRACE("(%p)->(IID_IBindStatusCallback, %p)\n", This, ppv);
245 *ppv = STATUSCLB(This);
246 }else if(IsEqualGUID(&IID_IServiceProvider, riid)) {
247 TRACE("(%p)->(IID_IServiceProvider %p)\n", This, ppv);
248 *ppv = SERVPROV(This);
249 }else if(IsEqualGUID(&IID_IHttpNegotiate, riid)) {
250 TRACE("(%p)->(IID_IHttpNegotiate %p)\n", This, ppv);
251 *ppv = HTTPNEG(This);
252 }else if(IsEqualGUID(&IID_IHttpNegotiate2, riid)) {
253 TRACE("(%p)->(IID_IHttpNegotiate2 %p)\n", This, ppv);
254 *ppv = HTTPNEG(This);
255 }else if(IsEqualGUID(&IID_IInternetBindInfo, riid)) {
256 TRACE("(%p)->(IID_IInternetBindInfo %p)\n", This, ppv);
257 *ppv = BINDINFO(This);
260 if(*ppv) {
261 IBindStatusCallback_AddRef(STATUSCLB(This));
262 return S_OK;
265 TRACE("Unsupported riid = %s\n", debugstr_guid(riid));
266 return E_NOINTERFACE;
269 static ULONG WINAPI BindStatusCallback_AddRef(IBindStatusCallback *iface)
271 BSCallback *This = STATUSCLB_THIS(iface);
272 LONG ref = InterlockedIncrement(&This->ref);
274 TRACE("(%p) ref = %d\n", This, ref);
276 return ref;
279 static ULONG WINAPI BindStatusCallback_Release(IBindStatusCallback *iface)
281 BSCallback *This = STATUSCLB_THIS(iface);
282 LONG ref = InterlockedDecrement(&This->ref);
284 TRACE("(%p) ref = %d\n", This, ref);
286 if(!ref) {
287 if(This->post_data)
288 GlobalFree(This->post_data);
289 if(This->mon)
290 IMoniker_Release(This->mon);
291 if(This->binding)
292 IBinding_Release(This->binding);
293 list_remove(&This->entry);
294 heap_free(This->headers);
296 This->vtbl->destroy(This);
299 return ref;
302 static HRESULT WINAPI BindStatusCallback_OnStartBinding(IBindStatusCallback *iface,
303 DWORD dwReserved, IBinding *pbind)
305 BSCallback *This = STATUSCLB_THIS(iface);
307 TRACE("(%p)->(%d %p)\n", This, dwReserved, pbind);
309 IBinding_AddRef(pbind);
310 This->binding = pbind;
312 if(This->doc)
313 list_add_head(&This->doc->bindings, &This->entry);
315 return This->vtbl->start_binding(This);
318 static HRESULT WINAPI BindStatusCallback_GetPriority(IBindStatusCallback *iface, LONG *pnPriority)
320 BSCallback *This = STATUSCLB_THIS(iface);
321 FIXME("(%p)->(%p)\n", This, pnPriority);
322 return E_NOTIMPL;
325 static HRESULT WINAPI BindStatusCallback_OnLowResource(IBindStatusCallback *iface, DWORD reserved)
327 BSCallback *This = STATUSCLB_THIS(iface);
328 FIXME("(%p)->(%d)\n", This, reserved);
329 return E_NOTIMPL;
332 static HRESULT WINAPI BindStatusCallback_OnProgress(IBindStatusCallback *iface, ULONG ulProgress,
333 ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText)
335 BSCallback *This = STATUSCLB_THIS(iface);
337 TRACE("%p)->(%u %u %u %s)\n", This, ulProgress, ulProgressMax, ulStatusCode,
338 debugstr_w(szStatusText));
340 return This->vtbl->on_progress(This, ulStatusCode, szStatusText);
343 static HRESULT WINAPI BindStatusCallback_OnStopBinding(IBindStatusCallback *iface,
344 HRESULT hresult, LPCWSTR szError)
346 BSCallback *This = STATUSCLB_THIS(iface);
347 HRESULT hres;
349 TRACE("(%p)->(%08x %s)\n", This, hresult, debugstr_w(szError));
351 /* NOTE: IE7 calls GetBindResult here */
353 hres = This->vtbl->stop_binding(This, hresult);
355 if(This->binding) {
356 IBinding_Release(This->binding);
357 This->binding = NULL;
360 list_remove(&This->entry);
361 This->doc = NULL;
363 return hres;
366 static HRESULT WINAPI BindStatusCallback_GetBindInfo(IBindStatusCallback *iface,
367 DWORD *grfBINDF, BINDINFO *pbindinfo)
369 BSCallback *This = STATUSCLB_THIS(iface);
370 DWORD size;
372 TRACE("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo);
374 *grfBINDF = This->bindf;
376 size = pbindinfo->cbSize;
377 memset(pbindinfo, 0, size);
378 pbindinfo->cbSize = size;
380 pbindinfo->cbstgmedData = This->post_data_len;
381 pbindinfo->dwCodePage = CP_UTF8;
382 pbindinfo->dwOptions = 0x80000;
384 if(This->post_data) {
385 pbindinfo->dwBindVerb = BINDVERB_POST;
387 pbindinfo->stgmedData.tymed = TYMED_HGLOBAL;
388 pbindinfo->stgmedData.u.hGlobal = This->post_data;
389 pbindinfo->stgmedData.pUnkForRelease = (IUnknown*)STATUSCLB(This);
390 IBindStatusCallback_AddRef(STATUSCLB(This));
393 return S_OK;
396 static HRESULT WINAPI BindStatusCallback_OnDataAvailable(IBindStatusCallback *iface,
397 DWORD grfBSCF, DWORD dwSize, FORMATETC *pformatetc, STGMEDIUM *pstgmed)
399 BSCallback *This = STATUSCLB_THIS(iface);
401 TRACE("(%p)->(%08x %d %p %p)\n", This, grfBSCF, dwSize, pformatetc, pstgmed);
403 return This->vtbl->read_data(This, pstgmed->u.pstm);
406 static HRESULT WINAPI BindStatusCallback_OnObjectAvailable(IBindStatusCallback *iface,
407 REFIID riid, IUnknown *punk)
409 BSCallback *This = STATUSCLB_THIS(iface);
410 FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), punk);
411 return E_NOTIMPL;
414 #undef STATUSCLB_THIS
416 static const IBindStatusCallbackVtbl BindStatusCallbackVtbl = {
417 BindStatusCallback_QueryInterface,
418 BindStatusCallback_AddRef,
419 BindStatusCallback_Release,
420 BindStatusCallback_OnStartBinding,
421 BindStatusCallback_GetPriority,
422 BindStatusCallback_OnLowResource,
423 BindStatusCallback_OnProgress,
424 BindStatusCallback_OnStopBinding,
425 BindStatusCallback_GetBindInfo,
426 BindStatusCallback_OnDataAvailable,
427 BindStatusCallback_OnObjectAvailable
430 #define HTTPNEG_THIS(iface) DEFINE_THIS(BSCallback, HttpNegotiate2, iface)
432 static HRESULT WINAPI HttpNegotiate_QueryInterface(IHttpNegotiate2 *iface,
433 REFIID riid, void **ppv)
435 BSCallback *This = HTTPNEG_THIS(iface);
436 return IBindStatusCallback_QueryInterface(STATUSCLB(This), riid, ppv);
439 static ULONG WINAPI HttpNegotiate_AddRef(IHttpNegotiate2 *iface)
441 BSCallback *This = HTTPNEG_THIS(iface);
442 return IBindStatusCallback_AddRef(STATUSCLB(This));
445 static ULONG WINAPI HttpNegotiate_Release(IHttpNegotiate2 *iface)
447 BSCallback *This = HTTPNEG_THIS(iface);
448 return IBindStatusCallback_Release(STATUSCLB(This));
451 static HRESULT WINAPI HttpNegotiate_BeginningTransaction(IHttpNegotiate2 *iface,
452 LPCWSTR szURL, LPCWSTR szHeaders, DWORD dwReserved, LPWSTR *pszAdditionalHeaders)
454 BSCallback *This = HTTPNEG_THIS(iface);
455 DWORD size;
457 TRACE("(%p)->(%s %s %d %p)\n", This, debugstr_w(szURL), debugstr_w(szHeaders),
458 dwReserved, pszAdditionalHeaders);
460 if(!This->headers) {
461 *pszAdditionalHeaders = NULL;
462 return S_OK;
465 size = (strlenW(This->headers)+1)*sizeof(WCHAR);
466 *pszAdditionalHeaders = CoTaskMemAlloc(size);
467 memcpy(*pszAdditionalHeaders, This->headers, size);
469 return S_OK;
472 static HRESULT WINAPI HttpNegotiate_OnResponse(IHttpNegotiate2 *iface, DWORD dwResponseCode,
473 LPCWSTR szResponseHeaders, LPCWSTR szRequestHeaders, LPWSTR *pszAdditionalRequestHeaders)
475 BSCallback *This = HTTPNEG_THIS(iface);
476 FIXME("(%p)->(%d %s %s %p)\n", This, dwResponseCode, debugstr_w(szResponseHeaders),
477 debugstr_w(szRequestHeaders), pszAdditionalRequestHeaders);
478 return E_NOTIMPL;
481 static HRESULT WINAPI HttpNegotiate_GetRootSecurityId(IHttpNegotiate2 *iface,
482 BYTE *pbSecurityId, DWORD *pcbSecurityId, DWORD_PTR dwReserved)
484 BSCallback *This = HTTPNEG_THIS(iface);
485 FIXME("(%p)->(%p %p %ld)\n", This, pbSecurityId, pcbSecurityId, dwReserved);
486 return E_NOTIMPL;
489 #undef HTTPNEG
491 static const IHttpNegotiate2Vtbl HttpNegotiate2Vtbl = {
492 HttpNegotiate_QueryInterface,
493 HttpNegotiate_AddRef,
494 HttpNegotiate_Release,
495 HttpNegotiate_BeginningTransaction,
496 HttpNegotiate_OnResponse,
497 HttpNegotiate_GetRootSecurityId
500 #define BINDINFO_THIS(iface) DEFINE_THIS(BSCallback, InternetBindInfo, iface)
502 static HRESULT WINAPI InternetBindInfo_QueryInterface(IInternetBindInfo *iface,
503 REFIID riid, void **ppv)
505 BSCallback *This = BINDINFO_THIS(iface);
506 return IBindStatusCallback_QueryInterface(STATUSCLB(This), riid, ppv);
509 static ULONG WINAPI InternetBindInfo_AddRef(IInternetBindInfo *iface)
511 BSCallback *This = BINDINFO_THIS(iface);
512 return IBindStatusCallback_AddRef(STATUSCLB(This));
515 static ULONG WINAPI InternetBindInfo_Release(IInternetBindInfo *iface)
517 BSCallback *This = BINDINFO_THIS(iface);
518 return IBindStatusCallback_Release(STATUSCLB(This));
521 static HRESULT WINAPI InternetBindInfo_GetBindInfo(IInternetBindInfo *iface,
522 DWORD *grfBINDF, BINDINFO *pbindinfo)
524 BSCallback *This = BINDINFO_THIS(iface);
525 FIXME("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo);
526 return E_NOTIMPL;
529 static HRESULT WINAPI InternetBindInfo_GetBindString(IInternetBindInfo *iface,
530 ULONG ulStringType, LPOLESTR *ppwzStr, ULONG cEl, ULONG *pcElFetched)
532 BSCallback *This = BINDINFO_THIS(iface);
533 FIXME("(%p)->(%u %p %u %p)\n", This, ulStringType, ppwzStr, cEl, pcElFetched);
534 return E_NOTIMPL;
537 #undef BINDINFO_THIS
539 static const IInternetBindInfoVtbl InternetBindInfoVtbl = {
540 InternetBindInfo_QueryInterface,
541 InternetBindInfo_AddRef,
542 InternetBindInfo_Release,
543 InternetBindInfo_GetBindInfo,
544 InternetBindInfo_GetBindString
547 #define SERVPROV_THIS(iface) DEFINE_THIS(BSCallback, ServiceProvider, iface)
549 static HRESULT WINAPI BSCServiceProvider_QueryInterface(IServiceProvider *iface,
550 REFIID riid, void **ppv)
552 BSCallback *This = SERVPROV_THIS(iface);
553 return IBindStatusCallback_QueryInterface(STATUSCLB(This), riid, ppv);
556 static ULONG WINAPI BSCServiceProvider_AddRef(IServiceProvider *iface)
558 BSCallback *This = SERVPROV_THIS(iface);
559 return IBindStatusCallback_AddRef(STATUSCLB(This));
562 static ULONG WINAPI BSCServiceProvider_Release(IServiceProvider *iface)
564 BSCallback *This = SERVPROV_THIS(iface);
565 return IBindStatusCallback_Release(STATUSCLB(This));
568 static HRESULT WINAPI BSCServiceProvider_QueryService(IServiceProvider *iface,
569 REFGUID guidService, REFIID riid, void **ppv)
571 BSCallback *This = SERVPROV_THIS(iface);
572 TRACE("(%p)->(%s %s %p)\n", This, debugstr_guid(guidService), debugstr_guid(riid), ppv);
573 return E_NOINTERFACE;
576 #undef SERVPROV_THIS
578 static const IServiceProviderVtbl ServiceProviderVtbl = {
579 BSCServiceProvider_QueryInterface,
580 BSCServiceProvider_AddRef,
581 BSCServiceProvider_Release,
582 BSCServiceProvider_QueryService
585 static void init_bscallback(BSCallback *This, const BSCallbackVtbl *vtbl, IMoniker *mon, DWORD bindf)
587 This->lpBindStatusCallbackVtbl = &BindStatusCallbackVtbl;
588 This->lpServiceProviderVtbl = &ServiceProviderVtbl;
589 This->lpHttpNegotiate2Vtbl = &HttpNegotiate2Vtbl;
590 This->lpInternetBindInfoVtbl = &InternetBindInfoVtbl;
591 This->vtbl = vtbl;
592 This->ref = 1;
593 This->bindf = bindf;
595 list_init(&This->entry);
597 if(mon)
598 IMoniker_AddRef(mon);
599 This->mon = mon;
602 /* Calls undocumented 84 cmd of CGID_ShellDocView */
603 static void call_docview_84(HTMLDocument *doc)
605 IOleCommandTarget *olecmd;
606 VARIANT var;
607 HRESULT hres;
609 if(!doc->client)
610 return;
612 hres = IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&olecmd);
613 if(FAILED(hres))
614 return;
616 VariantInit(&var);
617 hres = IOleCommandTarget_Exec(olecmd, &CGID_ShellDocView, 84, 0, NULL, &var);
618 IOleCommandTarget_Release(olecmd);
619 if(SUCCEEDED(hres) && V_VT(&var) != VT_NULL)
620 FIXME("handle result\n");
623 static void parse_post_data(nsIInputStream *post_data_stream, LPWSTR *headers_ret,
624 HGLOBAL *post_data_ret, ULONG *post_data_len_ret)
626 PRUint32 post_data_len = 0, available = 0;
627 HGLOBAL post_data = NULL;
628 LPWSTR headers = NULL;
629 DWORD headers_len = 0, len;
630 const char *ptr, *ptr2, *post_data_end;
632 nsIInputStream_Available(post_data_stream, &available);
633 post_data = GlobalAlloc(0, available+1);
634 nsIInputStream_Read(post_data_stream, post_data, available, &post_data_len);
636 TRACE("post_data = %s\n", debugstr_an(post_data, post_data_len));
638 ptr = ptr2 = post_data;
639 post_data_end = (const char*)post_data+post_data_len;
641 while(ptr < post_data_end && (*ptr != '\r' || ptr[1] != '\n')) {
642 while(ptr < post_data_end && (*ptr != '\r' || ptr[1] != '\n'))
643 ptr++;
645 if(!*ptr) {
646 FIXME("*ptr = 0\n");
647 return;
650 ptr += 2;
652 if(ptr-ptr2 >= sizeof(CONTENT_LENGTH)
653 && CompareStringA(LOCALE_SYSTEM_DEFAULT, NORM_IGNORECASE,
654 CONTENT_LENGTH, sizeof(CONTENT_LENGTH)-1,
655 ptr2, sizeof(CONTENT_LENGTH)-1) == CSTR_EQUAL) {
656 ptr2 = ptr;
657 continue;
660 len = MultiByteToWideChar(CP_ACP, 0, ptr2, ptr-ptr2, NULL, 0);
662 if(headers)
663 headers = heap_realloc(headers,(headers_len+len+1)*sizeof(WCHAR));
664 else
665 headers = heap_alloc((len+1)*sizeof(WCHAR));
667 len = MultiByteToWideChar(CP_ACP, 0, ptr2, ptr-ptr2, headers+headers_len, len);
668 headers_len += len;
670 ptr2 = ptr;
673 headers[headers_len] = 0;
674 *headers_ret = headers;
676 if(ptr >= post_data_end-2) {
677 GlobalFree(post_data);
678 return;
681 ptr += 2;
683 if(headers_len) {
684 post_data_len -= ptr-(const char*)post_data;
685 memmove(post_data, ptr, post_data_len);
686 post_data = GlobalReAlloc(post_data, post_data_len+1, 0);
689 *post_data_ret = post_data;
690 *post_data_len_ret = post_data_len;
693 HRESULT start_binding(HTMLDocument *doc, BSCallback *bscallback, IBindCtx *bctx)
695 IStream *str = NULL;
696 HRESULT hres;
698 bscallback->doc = doc;
700 /* NOTE: IE7 calls IsSystemMoniker here*/
702 call_docview_84(doc);
704 if(bctx) {
705 RegisterBindStatusCallback(bctx, STATUSCLB(bscallback), NULL, 0);
706 IBindCtx_AddRef(bctx);
707 }else {
708 hres = CreateAsyncBindCtx(0, STATUSCLB(bscallback), NULL, &bctx);
709 if(FAILED(hres)) {
710 WARN("CreateAsyncBindCtx failed: %08x\n", hres);
711 bscallback->vtbl->stop_binding(bscallback, hres);
712 return hres;
716 hres = IMoniker_BindToStorage(bscallback->mon, bctx, NULL, &IID_IStream, (void**)&str);
717 IBindCtx_Release(bctx);
718 if(FAILED(hres)) {
719 WARN("BindToStorage failed: %08x\n", hres);
720 bscallback->vtbl->stop_binding(bscallback, hres);
721 return hres;
724 if(str)
725 IStream_Release(str);
727 IMoniker_Release(bscallback->mon);
728 bscallback->mon = NULL;
730 return S_OK;
733 typedef struct {
734 BSCallback bsc;
736 DWORD size;
737 BYTE *buf;
738 HRESULT hres;
739 } BufferBSC;
741 #define BUFFERBSC_THIS(bsc) ((BufferBSC*) bsc)
743 static void BufferBSC_destroy(BSCallback *bsc)
745 BufferBSC *This = BUFFERBSC_THIS(bsc);
747 heap_free(This->buf);
748 heap_free(This);
751 static HRESULT BufferBSC_start_binding(BSCallback *bsc)
753 return S_OK;
756 static HRESULT BufferBSC_stop_binding(BSCallback *bsc, HRESULT result)
758 BufferBSC *This = BUFFERBSC_THIS(bsc);
760 This->hres = result;
762 if(FAILED(result)) {
763 heap_free(This->buf);
764 This->buf = NULL;
765 This->size = 0;
768 return S_OK;
771 static HRESULT BufferBSC_read_data(BSCallback *bsc, IStream *stream)
773 BufferBSC *This = BUFFERBSC_THIS(bsc);
774 DWORD readed;
775 HRESULT hres;
777 if(!This->buf) {
778 This->size = 128;
779 This->buf = heap_alloc(This->size);
782 do {
783 if(This->bsc.readed == This->size) {
784 This->size <<= 1;
785 This->buf = heap_realloc(This->buf, This->size);
788 readed = 0;
789 hres = IStream_Read(stream, This->buf+This->bsc.readed, This->size-This->bsc.readed, &readed);
790 This->bsc.readed += readed;
791 }while(hres == S_OK);
793 return S_OK;
796 static HRESULT BufferBSC_on_progress(BSCallback *bsc, ULONG status_code, LPCWSTR status_text)
798 return S_OK;
801 #undef BUFFERBSC_THIS
803 static const BSCallbackVtbl BufferBSCVtbl = {
804 BufferBSC_destroy,
805 BufferBSC_start_binding,
806 BufferBSC_stop_binding,
807 BufferBSC_read_data,
808 BufferBSC_on_progress,
812 static BufferBSC *create_bufferbsc(IMoniker *mon)
814 BufferBSC *ret = heap_alloc_zero(sizeof(*ret));
816 init_bscallback(&ret->bsc, &BufferBSCVtbl, mon, 0);
817 ret->hres = E_FAIL;
819 return ret;
822 HRESULT bind_mon_to_buffer(HTMLDocument *doc, IMoniker *mon, void **buf, DWORD *size)
824 BufferBSC *bsc = create_bufferbsc(mon);
825 HRESULT hres;
827 *buf = NULL;
829 hres = start_binding(doc, &bsc->bsc, NULL);
830 if(SUCCEEDED(hres)) {
831 hres = bsc->hres;
832 if(SUCCEEDED(hres)) {
833 *buf = bsc->buf;
834 bsc->buf = NULL;
835 *size = bsc->bsc.readed;
836 bsc->size = 0;
840 IBindStatusCallback_Release(STATUSCLB(&bsc->bsc));
842 return hres;
845 struct nsChannelBSC {
846 BSCallback bsc;
848 nsChannel *nschannel;
849 nsIStreamListener *nslistener;
850 nsISupports *nscontext;
852 nsProtocolStream *nsstream;
855 static HRESULT read_stream_data(nsChannelBSC *This, IStream *stream)
857 nsresult nsres;
858 HRESULT hres;
860 if(!This->nslistener) {
861 BYTE buf[1024];
862 DWORD read;
864 do {
865 read = 0;
866 hres = IStream_Read(stream, buf, sizeof(buf), &read);
867 }while(hres == S_OK && read);
869 return S_OK;
872 if(!This->nsstream)
873 This->nsstream = create_nsprotocol_stream();
875 do {
876 hres = IStream_Read(stream, This->nsstream->buf, sizeof(This->nsstream->buf),
877 &This->nsstream->buf_size);
878 if(!This->nsstream->buf_size)
879 break;
881 if(!This->bsc.readed && This->nsstream->buf_size >= 2 && *(WORD*)This->nsstream->buf == 0xfeff) {
882 This->nschannel->charset = heap_alloc(sizeof(UTF16_STR));
883 memcpy(This->nschannel->charset, UTF16_STR, sizeof(UTF16_STR));
886 if(!This->bsc.readed) {
887 nsres = nsIStreamListener_OnStartRequest(This->nslistener,
888 (nsIRequest*)NSCHANNEL(This->nschannel), This->nscontext);
889 if(NS_FAILED(nsres))
890 FIXME("OnStartRequest failed: %08x\n", nsres);
892 /* events are reset when a new document URI is loaded, so re-initialise them here */
893 if(This->bsc.doc && This->bsc.doc->bscallback == This && This->bsc.doc->nscontainer) {
894 update_nsdocument(This->bsc.doc);
895 init_nsevents(This->bsc.doc->nscontainer);
899 This->bsc.readed += This->nsstream->buf_size;
901 nsres = nsIStreamListener_OnDataAvailable(This->nslistener,
902 (nsIRequest*)NSCHANNEL(This->nschannel), This->nscontext,
903 NSINSTREAM(This->nsstream), This->bsc.readed-This->nsstream->buf_size,
904 This->nsstream->buf_size);
905 if(NS_FAILED(nsres))
906 ERR("OnDataAvailable failed: %08x\n", nsres);
908 if(This->nsstream->buf_size)
909 FIXME("buffer is not empty!\n");
910 }while(hres == S_OK);
912 return S_OK;
915 static void on_stop_nsrequest(nsChannelBSC *This)
917 if(!This->nslistener)
918 return;
920 nsIStreamListener_OnStopRequest(This->nslistener, (nsIRequest*)NSCHANNEL(This->nschannel),
921 This->nscontext, NS_OK);
924 static void add_nsrequest(nsChannelBSC *This)
926 nsresult nsres;
928 if(!This->nschannel || !This->nschannel->load_group)
929 return;
931 nsres = nsILoadGroup_AddRequest(This->nschannel->load_group,
932 (nsIRequest*)NSCHANNEL(This->nschannel), This->nscontext);
934 if(NS_FAILED(nsres))
935 ERR("AddRequest failed:%08x\n", nsres);
938 #define NSCHANNELBSC_THIS(bsc) ((nsChannelBSC*) bsc)
940 static void nsChannelBSC_destroy(BSCallback *bsc)
942 nsChannelBSC *This = NSCHANNELBSC_THIS(bsc);
944 if(This->nschannel)
945 nsIChannel_Release(NSCHANNEL(This->nschannel));
946 if(This->nslistener)
947 nsIStreamListener_Release(This->nslistener);
948 if(This->nscontext)
949 nsISupports_Release(This->nscontext);
950 if(This->nsstream)
951 nsIInputStream_Release(NSINSTREAM(This->nsstream));
952 heap_free(This);
955 static HRESULT nsChannelBSC_start_binding(BSCallback *bsc)
957 nsChannelBSC *This = NSCHANNELBSC_THIS(bsc);
959 add_nsrequest(This);
961 return S_OK;
964 static HRESULT nsChannelBSC_stop_binding(BSCallback *bsc, HRESULT result)
966 nsChannelBSC *This = NSCHANNELBSC_THIS(bsc);
968 on_stop_nsrequest(This);
970 if(This->nslistener) {
971 if(This->nschannel->load_group) {
972 nsresult nsres;
974 nsres = nsILoadGroup_RemoveRequest(This->nschannel->load_group,
975 (nsIRequest*)NSCHANNEL(This->nschannel), NULL, NS_OK);
976 if(NS_FAILED(nsres))
977 ERR("RemoveRequest failed: %08x\n", nsres);
981 if(FAILED(result))
982 return S_OK;
984 if(This->bsc.doc && This->bsc.doc->bscallback == This && !This->bsc.doc->nscontainer) {
985 task_t *task = heap_alloc(sizeof(task_t));
987 task->doc = This->bsc.doc;
988 task->task_id = TASK_PARSECOMPLETE;
989 task->next = NULL;
992 * This should be done in the worker thread that parses HTML,
993 * but we don't have such thread.
995 push_task(task);
998 return S_OK;
1001 static HRESULT nsChannelBSC_read_data(BSCallback *bsc, IStream *stream)
1003 nsChannelBSC *This = NSCHANNELBSC_THIS(bsc);
1005 return read_stream_data(This, stream);
1008 static HRESULT nsChannelBSC_on_progress(BSCallback *bsc, ULONG status_code, LPCWSTR status_text)
1010 nsChannelBSC *This = NSCHANNELBSC_THIS(bsc);
1012 switch(status_code) {
1013 case BINDSTATUS_MIMETYPEAVAILABLE:
1014 if(!This->nschannel)
1015 return S_OK;
1017 heap_free(This->nschannel->content_type);
1018 This->nschannel->content_type = heap_strdupWtoA(status_text);
1021 return S_OK;
1024 #undef NSCHANNELBSC_THIS
1026 static const BSCallbackVtbl nsChannelBSCVtbl = {
1027 nsChannelBSC_destroy,
1028 nsChannelBSC_start_binding,
1029 nsChannelBSC_stop_binding,
1030 nsChannelBSC_read_data,
1031 nsChannelBSC_on_progress,
1034 nsChannelBSC *create_channelbsc(IMoniker *mon)
1036 nsChannelBSC *ret = heap_alloc_zero(sizeof(*ret));
1038 init_bscallback(&ret->bsc, &nsChannelBSCVtbl, mon, BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA);
1040 return ret;
1043 IMoniker *get_channelbsc_mon(nsChannelBSC *This)
1045 if(This->bsc.mon)
1046 IMoniker_AddRef(This->bsc.mon);
1047 return This->bsc.mon;
1050 void set_document_bscallback(HTMLDocument *doc, nsChannelBSC *callback)
1052 BSCallback *iter;
1054 if(doc->bscallback) {
1055 if(doc->bscallback->bsc.binding)
1056 IBinding_Abort(doc->bscallback->bsc.binding);
1057 doc->bscallback->bsc.doc = NULL;
1058 IBindStatusCallback_Release(STATUSCLB(&doc->bscallback->bsc));
1061 LIST_FOR_EACH_ENTRY(iter, &doc->bindings, BSCallback, entry) {
1062 iter->doc = NULL;
1063 list_remove(&iter->entry);
1066 doc->bscallback = callback;
1068 if(callback) {
1069 IBindStatusCallback_AddRef(STATUSCLB(&callback->bsc));
1070 callback->bsc.doc = doc;
1074 HRESULT channelbsc_load_stream(nsChannelBSC *bscallback, IStream *stream)
1076 HRESULT hres;
1078 const char text_html[] = "text/html";
1080 add_nsrequest(bscallback);
1082 if(bscallback->nschannel)
1083 bscallback->nschannel->content_type = heap_strdupA(text_html);
1085 hres = read_stream_data(bscallback, stream);
1086 IBindStatusCallback_OnStopBinding(STATUSCLB(&bscallback->bsc), hres, ERROR_SUCCESS);
1088 return hres;
1091 void channelbsc_set_channel(nsChannelBSC *This, nsChannel *channel, nsIStreamListener *listener, nsISupports *context)
1093 nsIChannel_AddRef(NSCHANNEL(channel));
1094 This->nschannel = channel;
1096 nsIStreamListener_AddRef(listener);
1097 This->nslistener = listener;
1099 if(context) {
1100 nsISupports_AddRef(context);
1101 This->nscontext = context;
1105 void hlink_frame_navigate(HTMLDocument *doc, IHlinkFrame *hlink_frame,
1106 LPCWSTR uri, nsIInputStream *post_data_stream, DWORD hlnf)
1108 BSCallback *callback;
1109 IBindCtx *bindctx;
1110 IMoniker *mon;
1111 IHlink *hlink;
1112 HRESULT hr;
1114 callback = &create_channelbsc(NULL)->bsc;
1116 if(post_data_stream) {
1117 parse_post_data(post_data_stream, &callback->headers, &callback->post_data,
1118 &callback->post_data_len);
1119 TRACE("headers = %s post_data = %s\n", debugstr_w(callback->headers),
1120 debugstr_an(callback->post_data, callback->post_data_len));
1123 hr = CreateAsyncBindCtx(0, STATUSCLB(callback), NULL, &bindctx);
1124 if (FAILED(hr)) {
1125 IBindStatusCallback_Release(STATUSCLB(callback));
1126 return;
1129 hr = CoCreateInstance(&CLSID_StdHlink, NULL, CLSCTX_INPROC_SERVER, &IID_IHlink, (LPVOID*)&hlink);
1130 if (FAILED(hr)) {
1131 IBindCtx_Release(bindctx);
1132 IBindStatusCallback_Release(STATUSCLB(callback));
1133 return;
1136 hr = CreateURLMoniker(NULL, uri, &mon);
1137 if (SUCCEEDED(hr)) {
1138 IHlink_SetMonikerReference(hlink, 0, mon, NULL);
1140 if(hlnf & HLNF_OPENINNEWWINDOW) {
1141 static const WCHAR wszBlank[] = {'_','b','l','a','n','k',0};
1142 IHlink_SetTargetFrameName(hlink, wszBlank); /* FIXME */
1145 IHlinkFrame_Navigate(hlink_frame, hlnf, bindctx, STATUSCLB(callback), hlink);
1147 IMoniker_Release(mon);
1150 IBindCtx_Release(bindctx);
1151 IBindStatusCallback_Release(STATUSCLB(callback));