dlls/mshtml/navigate.c

   1 /*
   2  * Copyright 2006-2007 Jacek Caban for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include "config.h"
  20
  21 #include <stdarg.h>
  22
  23 #define COBJMACROS
  24 #define NONAMELESSUNION
  25 #define NONAMELESSSTRUCT
  26
  27 #include "windef.h"
  28 #include "winbase.h"
  29 #include "winuser.h"
  30 #include "ole2.h"
  31 #include "hlguids.h"
  32 #include "shlguid.h"
  33
  34 #include "wine/debug.h"
  35 #include "wine/unicode.h"
  36
  37 #include "mshtml_private.h"
  38
  39 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  40
  41 #define CONTENT_LENGTH "Content-Length"
  42 #define UTF16_STR "utf-16"
  43
  44 typedef struct {
  45     const nsIInputStreamVtbl *lpInputStreamVtbl;
  46
  47     LONG ref;
  48
  49     char buf[1024];
  50     DWORD buf_size;
  51 } nsProtocolStream;
  52
  53 #define NSINSTREAM(x) ((nsIInputStream*) &(x)->lpInputStreamVtbl)
  54
  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;
  62
  63 struct BSCallback {
  64     const IBindStatusCallbackVtbl *lpBindStatusCallbackVtbl;
  65     const IServiceProviderVtbl    *lpServiceProviderVtbl;
  66     const IHttpNegotiate2Vtbl     *lpHttpNegotiate2Vtbl;
  67     const IInternetBindInfoVtbl   *lpInternetBindInfoVtbl;
  68
  69     const BSCallbackVtbl          *vtbl;
  70
  71     LONG ref;
  72
  73     LPWSTR headers;
  74     HGLOBAL post_data;
  75     ULONG post_data_len;
  76     ULONG readed;
  77     DWORD bindf;
  78
  79     IMoniker *mon;
  80     IBinding *binding;
  81
  82     HTMLDocument *doc;
  83
  84     struct list entry;
  85 };
  86
  87 #define NSINSTREAM_THIS(iface) DEFINE_THIS(nsProtocolStream, InputStream, iface)
  88
  89 static nsresult NSAPI nsInputStream_QueryInterface(nsIInputStream *iface, nsIIDRef riid,
  90                                                    nsQIResult result)
  91 {
  92     nsProtocolStream *This = NSINSTREAM_THIS(iface);
  93
  94     *result = NULL;
  95
  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);
 102     }
 103
 104     if(*result) {
 105         nsIInputStream_AddRef(NSINSTREAM(This));
 106         return NS_OK;
 107     }
 108
 109     WARN("unsupported interface %s\n", debugstr_guid(riid));
 110     return NS_NOINTERFACE;
 111 }
 112
 113 static nsrefcnt NSAPI nsInputStream_AddRef(nsIInputStream *iface)
 114 {
 115     nsProtocolStream *This = NSINSTREAM_THIS(iface);
 116     LONG ref = InterlockedIncrement(&This->ref);
 117
 118     TRACE("(%p) ref=%d\n", This, ref);
 119
 120     return ref;
 121 }
 122
 123
 124 static nsrefcnt NSAPI nsInputStream_Release(nsIInputStream *iface)
 125 {
 126     nsProtocolStream *This = NSINSTREAM_THIS(iface);
 127     LONG ref = InterlockedDecrement(&This->ref);
 128
 129     TRACE("(%p) ref=%d\n", This, ref);
 130
 131     if(!ref)
 132         heap_free(This);
 133
 134     return ref;
 135 }
 136
 137 static nsresult NSAPI nsInputStream_Close(nsIInputStream *iface)
 138 {
 139     nsProtocolStream *This = NSINSTREAM_THIS(iface);
 140     FIXME("(%p)\n", This);
 141     return NS_ERROR_NOT_IMPLEMENTED;
 142 }
 143
 144 static nsresult NSAPI nsInputStream_Available(nsIInputStream *iface, PRUint32 *_retval)
 145 {
 146     nsProtocolStream *This = NSINSTREAM_THIS(iface);
 147     FIXME("(%p)->(%p)\n", This, _retval);
 148     return NS_ERROR_NOT_IMPLEMENTED;
 149 }
 150
 151 static nsresult NSAPI nsInputStream_Read(nsIInputStream *iface, char *aBuf, PRUint32 aCount,
 152                                          PRUint32 *_retval)
 153 {
 154     nsProtocolStream *This = NSINSTREAM_THIS(iface);
 155     DWORD read = aCount;
 156
 157     TRACE("(%p)->(%p %d %p)\n", This, aBuf, aCount, _retval);
 158
 159     if(read > This->buf_size)
 160         read = This->buf_size;
 161
 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;
 167     }
 168
 169     *_retval = read;
 170     return NS_OK;
 171 }
 172
 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)
 176 {
 177     nsProtocolStream *This = NSINSTREAM_THIS(iface);
 178     PRUint32 written = 0;
 179     nsresult nsres;
 180
 181     TRACE("(%p)->(%p %p %d %p)\n", This, aWriter, aClousure, aCount, _retval);
 182
 183     if(!This->buf_size)
 184         return S_OK;
 185
 186     if(aCount > This->buf_size)
 187         aCount = This->buf_size;
 188
 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);
 194
 195     This->buf_size -= written;
 196
 197     *_retval = written;
 198     return nsres;
 199 }
 200
 201 static nsresult NSAPI nsInputStream_IsNonBlocking(nsIInputStream *iface, PRBool *_retval)
 202 {
 203     nsProtocolStream *This = NSINSTREAM_THIS(iface);
 204     FIXME("(%p)->(%p)\n", This, _retval);
 205     return NS_ERROR_NOT_IMPLEMENTED;
 206 }
 207
 208 #undef NSINSTREAM_THIS
 209
 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
 219 };
 220
 221 static nsProtocolStream *create_nsprotocol_stream(void)
 222 {
 223     nsProtocolStream *ret = heap_alloc(sizeof(nsProtocolStream));
 224
 225     ret->lpInputStreamVtbl = &nsInputStreamVtbl;
 226     ret->ref = 1;
 227     ret->buf_size = 0;
 228
 229     return ret;
 230 }
 231
 232 #define STATUSCLB_THIS(iface) DEFINE_THIS(BSCallback, BindStatusCallback, iface)
 233
 234 static HRESULT WINAPI BindStatusCallback_QueryInterface(IBindStatusCallback *iface,
 235         REFIID riid, void **ppv)
 236 {
 237     BSCallback *This = STATUSCLB_THIS(iface);
 238
 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);
 258     }
 259
 260     if(*ppv) {
 261         IBindStatusCallback_AddRef(STATUSCLB(This));
 262         return S_OK;
 263     }
 264
 265     TRACE("Unsupported riid = %s\n", debugstr_guid(riid));
 266     return E_NOINTERFACE;
 267 }
 268
 269 static ULONG WINAPI BindStatusCallback_AddRef(IBindStatusCallback *iface)
 270 {
 271     BSCallback *This = STATUSCLB_THIS(iface);
 272     LONG ref = InterlockedIncrement(&This->ref);
 273
 274     TRACE("(%p) ref = %d\n", This, ref);
 275
 276     return ref;
 277 }
 278
 279 static ULONG WINAPI BindStatusCallback_Release(IBindStatusCallback *iface)
 280 {
 281     BSCallback *This = STATUSCLB_THIS(iface);
 282     LONG ref = InterlockedDecrement(&This->ref);
 283
 284     TRACE("(%p) ref = %d\n", This, ref);
 285
 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);
 295
 296         This->vtbl->destroy(This);
 297     }
 298
 299     return ref;
 300 }
 301
 302 static HRESULT WINAPI BindStatusCallback_OnStartBinding(IBindStatusCallback *iface,
 303         DWORD dwReserved, IBinding *pbind)
 304 {
 305     BSCallback *This = STATUSCLB_THIS(iface);
 306
 307     TRACE("(%p)->(%d %p)\n", This, dwReserved, pbind);
 308
 309     IBinding_AddRef(pbind);
 310     This->binding = pbind;
 311
 312     if(This->doc)
 313         list_add_head(&This->doc->bindings, &This->entry);
 314
 315     return This->vtbl->start_binding(This);
 316 }
 317
 318 static HRESULT WINAPI BindStatusCallback_GetPriority(IBindStatusCallback *iface, LONG *pnPriority)
 319 {
 320     BSCallback *This = STATUSCLB_THIS(iface);
 321     FIXME("(%p)->(%p)\n", This, pnPriority);
 322     return E_NOTIMPL;
 323 }
 324
 325 static HRESULT WINAPI BindStatusCallback_OnLowResource(IBindStatusCallback *iface, DWORD reserved)
 326 {
 327     BSCallback *This = STATUSCLB_THIS(iface);
 328     FIXME("(%p)->(%d)\n", This, reserved);
 329     return E_NOTIMPL;
 330 }
 331
 332 static HRESULT WINAPI BindStatusCallback_OnProgress(IBindStatusCallback *iface, ULONG ulProgress,
 333         ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText)
 334 {
 335     BSCallback *This = STATUSCLB_THIS(iface);
 336
 337     TRACE("%p)->(%u %u %u %s)\n", This, ulProgress, ulProgressMax, ulStatusCode,
 338             debugstr_w(szStatusText));
 339
 340     return This->vtbl->on_progress(This, ulStatusCode, szStatusText);
 341 }
 342
 343 static HRESULT WINAPI BindStatusCallback_OnStopBinding(IBindStatusCallback *iface,
 344         HRESULT hresult, LPCWSTR szError)
 345 {
 346     BSCallback *This = STATUSCLB_THIS(iface);
 347
 348     TRACE("(%p)->(%08x %s)\n", This, hresult, debugstr_w(szError));
 349
 350     /* NOTE: IE7 calls GetBindResult here */
 351
 352     if(This->binding) {
 353         IBinding_Release(This->binding);
 354         This->binding = NULL;
 355     }
 356
 357     list_remove(&This->entry);
 358
 359     return This->vtbl->stop_binding(This, hresult);
 360 }
 361
 362 static HRESULT WINAPI BindStatusCallback_GetBindInfo(IBindStatusCallback *iface,
 363         DWORD *grfBINDF, BINDINFO *pbindinfo)
 364 {
 365     BSCallback *This = STATUSCLB_THIS(iface);
 366     DWORD size;
 367
 368     TRACE("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo);
 369
 370     *grfBINDF = This->bindf;
 371
 372     size = pbindinfo->cbSize;
 373     memset(pbindinfo, 0, size);
 374     pbindinfo->cbSize = size;
 375
 376     pbindinfo->cbstgmedData = This->post_data_len;
 377     pbindinfo->dwCodePage = CP_UTF8;
 378     pbindinfo->dwOptions = 0x80000;
 379
 380     if(This->post_data) {
 381         pbindinfo->dwBindVerb = BINDVERB_POST;
 382
 383         pbindinfo->stgmedData.tymed = TYMED_HGLOBAL;
 384         pbindinfo->stgmedData.u.hGlobal = This->post_data;
 385         pbindinfo->stgmedData.pUnkForRelease = (IUnknown*)STATUSCLB(This);
 386         IBindStatusCallback_AddRef(STATUSCLB(This));
 387     }
 388
 389     return S_OK;
 390 }
 391
 392 static HRESULT WINAPI BindStatusCallback_OnDataAvailable(IBindStatusCallback *iface,
 393         DWORD grfBSCF, DWORD dwSize, FORMATETC *pformatetc, STGMEDIUM *pstgmed)
 394 {
 395     BSCallback *This = STATUSCLB_THIS(iface);
 396
 397     TRACE("(%p)->(%08x %d %p %p)\n", This, grfBSCF, dwSize, pformatetc, pstgmed);
 398
 399     return This->vtbl->read_data(This, pstgmed->u.pstm);
 400 }
 401
 402 static HRESULT WINAPI BindStatusCallback_OnObjectAvailable(IBindStatusCallback *iface,
 403         REFIID riid, IUnknown *punk)
 404 {
 405     BSCallback *This = STATUSCLB_THIS(iface);
 406     FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), punk);
 407     return E_NOTIMPL;
 408 }
 409
 410 #undef STATUSCLB_THIS
 411
 412 static const IBindStatusCallbackVtbl BindStatusCallbackVtbl = {
 413     BindStatusCallback_QueryInterface,
 414     BindStatusCallback_AddRef,
 415     BindStatusCallback_Release,
 416     BindStatusCallback_OnStartBinding,
 417     BindStatusCallback_GetPriority,
 418     BindStatusCallback_OnLowResource,
 419     BindStatusCallback_OnProgress,
 420     BindStatusCallback_OnStopBinding,
 421     BindStatusCallback_GetBindInfo,
 422     BindStatusCallback_OnDataAvailable,
 423     BindStatusCallback_OnObjectAvailable
 424 };
 425
 426 #define HTTPNEG_THIS(iface) DEFINE_THIS(BSCallback, HttpNegotiate2, iface)
 427
 428 static HRESULT WINAPI HttpNegotiate_QueryInterface(IHttpNegotiate2 *iface,
 429                                                    REFIID riid, void **ppv)
 430 {
 431     BSCallback *This = HTTPNEG_THIS(iface);
 432     return IBindStatusCallback_QueryInterface(STATUSCLB(This), riid, ppv);
 433 }
 434
 435 static ULONG WINAPI HttpNegotiate_AddRef(IHttpNegotiate2 *iface)
 436 {
 437     BSCallback *This = HTTPNEG_THIS(iface);
 438     return IBindStatusCallback_AddRef(STATUSCLB(This));
 439 }
 440
 441 static ULONG WINAPI HttpNegotiate_Release(IHttpNegotiate2 *iface)
 442 {
 443     BSCallback *This = HTTPNEG_THIS(iface);
 444     return IBindStatusCallback_Release(STATUSCLB(This));
 445 }
 446
 447 static HRESULT WINAPI HttpNegotiate_BeginningTransaction(IHttpNegotiate2 *iface,
 448         LPCWSTR szURL, LPCWSTR szHeaders, DWORD dwReserved, LPWSTR *pszAdditionalHeaders)
 449 {
 450     BSCallback *This = HTTPNEG_THIS(iface);
 451     DWORD size;
 452
 453     TRACE("(%p)->(%s %s %d %p)\n", This, debugstr_w(szURL), debugstr_w(szHeaders),
 454           dwReserved, pszAdditionalHeaders);
 455
 456     if(!This->headers) {
 457         *pszAdditionalHeaders = NULL;
 458         return S_OK;
 459     }
 460
 461     size = (strlenW(This->headers)+1)*sizeof(WCHAR);
 462     *pszAdditionalHeaders = CoTaskMemAlloc(size);
 463     memcpy(*pszAdditionalHeaders, This->headers, size);
 464
 465     return S_OK;
 466 }
 467
 468 static HRESULT WINAPI HttpNegotiate_OnResponse(IHttpNegotiate2 *iface, DWORD dwResponseCode,
 469         LPCWSTR szResponseHeaders, LPCWSTR szRequestHeaders, LPWSTR *pszAdditionalRequestHeaders)
 470 {
 471     BSCallback *This = HTTPNEG_THIS(iface);
 472     FIXME("(%p)->(%d %s %s %p)\n", This, dwResponseCode, debugstr_w(szResponseHeaders),
 473           debugstr_w(szRequestHeaders), pszAdditionalRequestHeaders);
 474     return E_NOTIMPL;
 475 }
 476
 477 static HRESULT WINAPI HttpNegotiate_GetRootSecurityId(IHttpNegotiate2 *iface,
 478         BYTE *pbSecurityId, DWORD *pcbSecurityId, DWORD_PTR dwReserved)
 479 {
 480     BSCallback *This = HTTPNEG_THIS(iface);
 481     FIXME("(%p)->(%p %p %ld)\n", This, pbSecurityId, pcbSecurityId, dwReserved);
 482     return E_NOTIMPL;
 483 }
 484
 485 #undef HTTPNEG
 486
 487 static const IHttpNegotiate2Vtbl HttpNegotiate2Vtbl = {
 488     HttpNegotiate_QueryInterface,
 489     HttpNegotiate_AddRef,
 490     HttpNegotiate_Release,
 491     HttpNegotiate_BeginningTransaction,
 492     HttpNegotiate_OnResponse,
 493     HttpNegotiate_GetRootSecurityId
 494 };
 495
 496 #define BINDINFO_THIS(iface) DEFINE_THIS(BSCallback, InternetBindInfo, iface)
 497
 498 static HRESULT WINAPI InternetBindInfo_QueryInterface(IInternetBindInfo *iface,
 499                                                       REFIID riid, void **ppv)
 500 {
 501     BSCallback *This = BINDINFO_THIS(iface);
 502     return IBindStatusCallback_QueryInterface(STATUSCLB(This), riid, ppv);
 503 }
 504
 505 static ULONG WINAPI InternetBindInfo_AddRef(IInternetBindInfo *iface)
 506 {
 507     BSCallback *This = BINDINFO_THIS(iface);
 508     return IBindStatusCallback_AddRef(STATUSCLB(This));
 509 }
 510
 511 static ULONG WINAPI InternetBindInfo_Release(IInternetBindInfo *iface)
 512 {
 513     BSCallback *This = BINDINFO_THIS(iface);
 514     return IBindStatusCallback_Release(STATUSCLB(This));
 515 }
 516
 517 static HRESULT WINAPI InternetBindInfo_GetBindInfo(IInternetBindInfo *iface,
 518                                                    DWORD *grfBINDF, BINDINFO *pbindinfo)
 519 {
 520     BSCallback *This = BINDINFO_THIS(iface);
 521     FIXME("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo);
 522     return E_NOTIMPL;
 523 }
 524
 525 static HRESULT WINAPI InternetBindInfo_GetBindString(IInternetBindInfo *iface,
 526         ULONG ulStringType, LPOLESTR *ppwzStr, ULONG cEl, ULONG *pcElFetched)
 527 {
 528     BSCallback *This = BINDINFO_THIS(iface);
 529     FIXME("(%p)->(%u %p %u %p)\n", This, ulStringType, ppwzStr, cEl, pcElFetched);
 530     return E_NOTIMPL;
 531 }
 532
 533 #undef BINDINFO_THIS
 534
 535 static const IInternetBindInfoVtbl InternetBindInfoVtbl = {
 536     InternetBindInfo_QueryInterface,
 537     InternetBindInfo_AddRef,
 538     InternetBindInfo_Release,
 539     InternetBindInfo_GetBindInfo,
 540     InternetBindInfo_GetBindString
 541 };
 542
 543 #define SERVPROV_THIS(iface) DEFINE_THIS(BSCallback, ServiceProvider, iface)
 544
 545 static HRESULT WINAPI BSCServiceProvider_QueryInterface(IServiceProvider *iface,
 546                                                         REFIID riid, void **ppv)
 547 {
 548     BSCallback *This = SERVPROV_THIS(iface);
 549     return IBindStatusCallback_QueryInterface(STATUSCLB(This), riid, ppv);
 550 }
 551
 552 static ULONG WINAPI BSCServiceProvider_AddRef(IServiceProvider *iface)
 553 {
 554     BSCallback *This = SERVPROV_THIS(iface);
 555     return IBindStatusCallback_AddRef(STATUSCLB(This));
 556 }
 557
 558 static ULONG WINAPI BSCServiceProvider_Release(IServiceProvider *iface)
 559 {
 560     BSCallback *This = SERVPROV_THIS(iface);
 561     return IBindStatusCallback_Release(STATUSCLB(This));
 562 }
 563
 564 static HRESULT WINAPI BSCServiceProvider_QueryService(IServiceProvider *iface,
 565         REFGUID guidService, REFIID riid, void **ppv)
 566 {
 567     BSCallback *This = SERVPROV_THIS(iface);
 568     TRACE("(%p)->(%s %s %p)\n", This, debugstr_guid(guidService), debugstr_guid(riid), ppv);
 569     return E_NOINTERFACE;
 570 }
 571
 572 #undef SERVPROV_THIS
 573
 574 static const IServiceProviderVtbl ServiceProviderVtbl = {
 575     BSCServiceProvider_QueryInterface,
 576     BSCServiceProvider_AddRef,
 577     BSCServiceProvider_Release,
 578     BSCServiceProvider_QueryService
 579 };
 580
 581 static void init_bscallback(BSCallback *This, const BSCallbackVtbl *vtbl, IMoniker *mon, DWORD bindf)
 582 {
 583     This->lpBindStatusCallbackVtbl = &BindStatusCallbackVtbl;
 584     This->lpServiceProviderVtbl    = &ServiceProviderVtbl;
 585     This->lpHttpNegotiate2Vtbl     = &HttpNegotiate2Vtbl;
 586     This->lpInternetBindInfoVtbl   = &InternetBindInfoVtbl;
 587     This->vtbl = vtbl;
 588     This->ref = 1;
 589     This->bindf = bindf;
 590
 591     list_init(&This->entry);
 592
 593     if(mon)
 594         IMoniker_AddRef(mon);
 595     This->mon = mon;
 596 }
 597
 598 /* Calls undocumented 84 cmd of CGID_ShellDocView */
 599 static void call_docview_84(HTMLDocument *doc)
 600 {
 601     IOleCommandTarget *olecmd;
 602     VARIANT var;
 603     HRESULT hres;
 604
 605     if(!doc->client)
 606         return;
 607
 608     hres = IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&olecmd);
 609     if(FAILED(hres))
 610         return;
 611
 612     VariantInit(&var);
 613     hres = IOleCommandTarget_Exec(olecmd, &CGID_ShellDocView, 84, 0, NULL, &var);
 614     IOleCommandTarget_Release(olecmd);
 615     if(SUCCEEDED(hres) && V_VT(&var) != VT_NULL)
 616         FIXME("handle result\n");
 617 }
 618
 619 static void parse_post_data(nsIInputStream *post_data_stream, LPWSTR *headers_ret,
 620                             HGLOBAL *post_data_ret, ULONG *post_data_len_ret)
 621 {
 622     PRUint32 post_data_len = 0, available = 0;
 623     HGLOBAL post_data = NULL;
 624     LPWSTR headers = NULL;
 625     DWORD headers_len = 0, len;
 626     const char *ptr, *ptr2, *post_data_end;
 627
 628     nsIInputStream_Available(post_data_stream, &available);
 629     post_data = GlobalAlloc(0, available+1);
 630     nsIInputStream_Read(post_data_stream, post_data, available, &post_data_len);
 631
 632     TRACE("post_data = %s\n", debugstr_an(post_data, post_data_len));
 633
 634     ptr = ptr2 = post_data;
 635     post_data_end = (const char*)post_data+post_data_len;
 636
 637     while(ptr < post_data_end && (*ptr != '\r' || ptr[1] != '\n')) {
 638         while(ptr < post_data_end && (*ptr != '\r' || ptr[1] != '\n'))
 639             ptr++;
 640
 641         if(!*ptr) {
 642             FIXME("*ptr = 0\n");
 643             return;
 644         }
 645
 646         ptr += 2;
 647
 648         if(ptr-ptr2 >= sizeof(CONTENT_LENGTH)
 649            && CompareStringA(LOCALE_SYSTEM_DEFAULT, NORM_IGNORECASE,
 650                              CONTENT_LENGTH, sizeof(CONTENT_LENGTH)-1,
 651                              ptr2, sizeof(CONTENT_LENGTH)-1) == CSTR_EQUAL) {
 652             ptr2 = ptr;
 653             continue;
 654         }
 655
 656         len = MultiByteToWideChar(CP_ACP, 0, ptr2, ptr-ptr2, NULL, 0);
 657
 658         if(headers)
 659             headers = heap_realloc(headers,(headers_len+len+1)*sizeof(WCHAR));
 660         else
 661             headers = heap_alloc((len+1)*sizeof(WCHAR));
 662
 663         len = MultiByteToWideChar(CP_ACP, 0, ptr2, ptr-ptr2, headers+headers_len, len);
 664         headers_len += len;
 665
 666         ptr2 = ptr;
 667     }
 668
 669     headers[headers_len] = 0;
 670     *headers_ret = headers;
 671
 672     if(ptr >= post_data_end-2) {
 673         GlobalFree(post_data);
 674         return;
 675     }
 676
 677     ptr += 2;
 678
 679     if(headers_len) {
 680         post_data_len -= ptr-(const char*)post_data;
 681         memmove(post_data, ptr, post_data_len);
 682         post_data = GlobalReAlloc(post_data, post_data_len+1, 0);
 683     }
 684
 685     *post_data_ret = post_data;
 686     *post_data_len_ret = post_data_len;
 687 }
 688
 689 HRESULT start_binding(HTMLDocument *doc, BSCallback *bscallback, IBindCtx *bctx)
 690 {
 691     IStream *str = NULL;
 692     HRESULT hres;
 693
 694     bscallback->doc = doc;
 695
 696     /* NOTE: IE7 calls IsSystemMoniker here*/
 697
 698     call_docview_84(doc);
 699
 700     if(bctx) {
 701         RegisterBindStatusCallback(bctx, STATUSCLB(bscallback), NULL, 0);
 702         IBindCtx_AddRef(bctx);
 703     }else {
 704         hres = CreateAsyncBindCtx(0, STATUSCLB(bscallback), NULL, &bctx);
 705         if(FAILED(hres)) {
 706             WARN("CreateAsyncBindCtx failed: %08x\n", hres);
 707             bscallback->vtbl->stop_binding(bscallback, hres);
 708             return hres;
 709         }
 710     }
 711
 712     hres = IMoniker_BindToStorage(bscallback->mon, bctx, NULL, &IID_IStream, (void**)&str);
 713     IBindCtx_Release(bctx);
 714     if(FAILED(hres)) {
 715         WARN("BindToStorage failed: %08x\n", hres);
 716         bscallback->vtbl->stop_binding(bscallback, hres);
 717         return hres;
 718     }
 719
 720     if(str)
 721         IStream_Release(str);
 722
 723     IMoniker_Release(bscallback->mon);
 724     bscallback->mon = NULL;
 725
 726     return S_OK;
 727 }
 728
 729 typedef struct {
 730     BSCallback bsc;
 731
 732     DWORD size;
 733     BYTE *buf;
 734     HRESULT hres;
 735 } BufferBSC;
 736
 737 #define BUFFERBSC_THIS(bsc) ((BufferBSC*) bsc)
 738
 739 static void BufferBSC_destroy(BSCallback *bsc)
 740 {
 741     BufferBSC *This = BUFFERBSC_THIS(bsc);
 742
 743     heap_free(This->buf);
 744     heap_free(This);
 745 }
 746
 747 static HRESULT BufferBSC_start_binding(BSCallback *bsc)
 748 {
 749     return S_OK;
 750 }
 751
 752 static HRESULT BufferBSC_stop_binding(BSCallback *bsc, HRESULT result)
 753 {
 754     BufferBSC *This = BUFFERBSC_THIS(bsc);
 755
 756     This->hres = result;
 757
 758     if(FAILED(result)) {
 759         heap_free(This->buf);
 760         This->buf = NULL;
 761         This->size = 0;
 762     }
 763
 764     return S_OK;
 765 }
 766
 767 static HRESULT BufferBSC_read_data(BSCallback *bsc, IStream *stream)
 768 {
 769     BufferBSC *This = BUFFERBSC_THIS(bsc);
 770     DWORD readed;
 771     HRESULT hres;
 772
 773     if(!This->buf) {
 774         This->size = 128;
 775         This->buf = heap_alloc(This->size);
 776     }
 777
 778     do {
 779         if(This->bsc.readed == This->size) {
 780             This->size <<= 1;
 781             This->buf = heap_realloc(This->buf, This->size);
 782         }
 783
 784         readed = 0;
 785         hres = IStream_Read(stream, This->buf+This->bsc.readed, This->size-This->bsc.readed, &readed);
 786         This->bsc.readed += readed;
 787     }while(hres == S_OK);
 788
 789     return S_OK;
 790 }
 791
 792 static HRESULT BufferBSC_on_progress(BSCallback *bsc, ULONG status_code, LPCWSTR status_text)
 793 {
 794     return S_OK;
 795 }
 796
 797 #undef BUFFERBSC_THIS
 798
 799 static const BSCallbackVtbl BufferBSCVtbl = {
 800     BufferBSC_destroy,
 801     BufferBSC_start_binding,
 802     BufferBSC_stop_binding,
 803     BufferBSC_read_data,
 804     BufferBSC_on_progress,
 805 };
 806
 807
 808 static BufferBSC *create_bufferbsc(IMoniker *mon)
 809 {
 810     BufferBSC *ret = heap_alloc_zero(sizeof(*ret));
 811
 812     init_bscallback(&ret->bsc, &BufferBSCVtbl, mon, 0);
 813     ret->hres = E_FAIL;
 814
 815     return ret;
 816 }
 817
 818 HRESULT bind_mon_to_buffer(HTMLDocument *doc, IMoniker *mon, void **buf, DWORD *size)
 819 {
 820     BufferBSC *bsc = create_bufferbsc(mon);
 821     HRESULT hres;
 822
 823     *buf = NULL;
 824
 825     hres = start_binding(doc, &bsc->bsc, NULL);
 826     if(SUCCEEDED(hres)) {
 827         hres = bsc->hres;
 828         if(SUCCEEDED(hres)) {
 829             *buf = bsc->buf;
 830             bsc->buf = NULL;
 831             *size = bsc->bsc.readed;
 832             bsc->size = 0;
 833         }
 834     }
 835
 836     IBindStatusCallback_Release(STATUSCLB(&bsc->bsc));
 837
 838     return hres;
 839 }
 840
 841 struct nsChannelBSC {
 842     BSCallback bsc;
 843
 844     nsChannel *nschannel;
 845     nsIStreamListener *nslistener;
 846     nsISupports *nscontext;
 847
 848     nsProtocolStream *nsstream;
 849 };
 850
 851 static HRESULT read_stream_data(nsChannelBSC *This, IStream *stream)
 852 {
 853     nsresult nsres;
 854     HRESULT hres;
 855
 856     if(!This->nslistener) {
 857         BYTE buf[1024];
 858         DWORD read;
 859
 860         do {
 861             read = 0;
 862             hres = IStream_Read(stream, buf, sizeof(buf), &read);
 863         }while(hres == S_OK && read);
 864
 865         return S_OK;
 866     }
 867
 868     if(!This->nsstream)
 869         This->nsstream = create_nsprotocol_stream();
 870
 871     do {
 872         hres = IStream_Read(stream, This->nsstream->buf, sizeof(This->nsstream->buf),
 873                 &This->nsstream->buf_size);
 874         if(!This->nsstream->buf_size)
 875             break;
 876
 877         if(!This->bsc.readed && This->nsstream->buf_size >= 2 && *(WORD*)This->nsstream->buf == 0xfeff) {
 878             This->nschannel->charset = heap_alloc(sizeof(UTF16_STR));
 879             memcpy(This->nschannel->charset, UTF16_STR, sizeof(UTF16_STR));
 880         }
 881
 882         if(!This->bsc.readed) {
 883             nsres = nsIStreamListener_OnStartRequest(This->nslistener,
 884                     (nsIRequest*)NSCHANNEL(This->nschannel), This->nscontext);
 885             if(NS_FAILED(nsres))
 886                 FIXME("OnStartRequest failed: %08x\n", nsres);
 887
 888             /* events are reset when a new document URI is loaded, so re-initialise them here */
 889             if(This->bsc.doc && This->bsc.doc->bscallback == This && This->bsc.doc->nscontainer) {
 890                 update_nsdocument(This->bsc.doc);
 891                 init_nsevents(This->bsc.doc->nscontainer);
 892             }
 893         }
 894
 895         This->bsc.readed += This->nsstream->buf_size;
 896
 897         nsres = nsIStreamListener_OnDataAvailable(This->nslistener,
 898                 (nsIRequest*)NSCHANNEL(This->nschannel), This->nscontext,
 899                 NSINSTREAM(This->nsstream), This->bsc.readed-This->nsstream->buf_size,
 900                 This->nsstream->buf_size);
 901         if(NS_FAILED(nsres))
 902             ERR("OnDataAvailable failed: %08x\n", nsres);
 903
 904         if(This->nsstream->buf_size)
 905             FIXME("buffer is not empty!\n");
 906     }while(hres == S_OK);
 907
 908     return S_OK;
 909 }
 910
 911 static void on_stop_nsrequest(nsChannelBSC *This)
 912 {
 913     if(!This->nslistener)
 914         return;
 915
 916     nsIStreamListener_OnStopRequest(This->nslistener, (nsIRequest*)NSCHANNEL(This->nschannel),
 917             This->nscontext, NS_OK);
 918 }
 919
 920 static void add_nsrequest(nsChannelBSC *This)
 921 {
 922     nsresult nsres;
 923
 924     if(!This->nschannel || !This->nschannel->load_group)
 925         return;
 926
 927     nsres = nsILoadGroup_AddRequest(This->nschannel->load_group,
 928             (nsIRequest*)NSCHANNEL(This->nschannel), This->nscontext);
 929
 930     if(NS_FAILED(nsres))
 931         ERR("AddRequest failed:%08x\n", nsres);
 932 }
 933
 934 #define NSCHANNELBSC_THIS(bsc) ((nsChannelBSC*) bsc)
 935
 936 static void nsChannelBSC_destroy(BSCallback *bsc)
 937 {
 938     nsChannelBSC *This = NSCHANNELBSC_THIS(bsc);
 939
 940     if(This->nschannel)
 941         nsIChannel_Release(NSCHANNEL(This->nschannel));
 942     if(This->nslistener)
 943         nsIStreamListener_Release(This->nslistener);
 944     if(This->nscontext)
 945         nsISupports_Release(This->nscontext);
 946     if(This->nsstream)
 947         nsIInputStream_Release(NSINSTREAM(This->nsstream));
 948     heap_free(This);
 949 }
 950
 951 static HRESULT nsChannelBSC_start_binding(BSCallback *bsc)
 952 {
 953     nsChannelBSC *This = NSCHANNELBSC_THIS(bsc);
 954
 955     add_nsrequest(This);
 956
 957     return S_OK;
 958 }
 959
 960 static HRESULT nsChannelBSC_stop_binding(BSCallback *bsc, HRESULT result)
 961 {
 962     nsChannelBSC *This = NSCHANNELBSC_THIS(bsc);
 963
 964     on_stop_nsrequest(This);
 965
 966     if(This->nslistener) {
 967         if(This->nschannel->load_group) {
 968             nsresult nsres;
 969
 970             nsres = nsILoadGroup_RemoveRequest(This->nschannel->load_group,
 971                     (nsIRequest*)NSCHANNEL(This->nschannel), NULL, NS_OK);
 972             if(NS_FAILED(nsres))
 973                 ERR("RemoveRequest failed: %08x\n", nsres);
 974         }
 975     }
 976
 977     if(FAILED(result))
 978         return S_OK;
 979
 980     if(This->bsc.doc && This->bsc.doc->bscallback == This && !This->bsc.doc->nscontainer) {
 981         task_t *task = heap_alloc(sizeof(task_t));
 982
 983         task->doc = This->bsc.doc;
 984         task->task_id = TASK_PARSECOMPLETE;
 985         task->next = NULL;
 986
 987         /*
 988          * This should be done in the worker thread that parses HTML,
 989          * but we don't have such thread.
 990          */
 991         push_task(task);
 992     }
 993
 994     return S_OK;
 995 }
 996
 997 static HRESULT nsChannelBSC_read_data(BSCallback *bsc, IStream *stream)
 998 {
 999     nsChannelBSC *This = NSCHANNELBSC_THIS(bsc);
1000
1001     return read_stream_data(This, stream);
1002 }
1003
1004 static HRESULT nsChannelBSC_on_progress(BSCallback *bsc, ULONG status_code, LPCWSTR status_text)
1005 {
1006     nsChannelBSC *This = NSCHANNELBSC_THIS(bsc);
1007
1008     switch(status_code) {
1009     case BINDSTATUS_MIMETYPEAVAILABLE:
1010         if(!This->nschannel)
1011             return S_OK;
1012
1013         heap_free(This->nschannel->content_type);
1014         This->nschannel->content_type = heap_strdupWtoA(status_text);
1015     }
1016
1017     return S_OK;
1018 }
1019
1020 #undef NSCHANNELBSC_THIS
1021
1022 static const BSCallbackVtbl nsChannelBSCVtbl = {
1023     nsChannelBSC_destroy,
1024     nsChannelBSC_start_binding,
1025     nsChannelBSC_stop_binding,
1026     nsChannelBSC_read_data,
1027     nsChannelBSC_on_progress,
1028 };
1029
1030 nsChannelBSC *create_channelbsc(IMoniker *mon)
1031 {
1032     nsChannelBSC *ret = heap_alloc_zero(sizeof(*ret));
1033
1034     init_bscallback(&ret->bsc, &nsChannelBSCVtbl, mon, BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA);
1035
1036     return ret;
1037 }
1038
1039 IMoniker *get_channelbsc_mon(nsChannelBSC *This)
1040 {
1041     if(This->bsc.mon)
1042         IMoniker_AddRef(This->bsc.mon);
1043     return This->bsc.mon;
1044 }
1045
1046 void set_document_bscallback(HTMLDocument *doc, nsChannelBSC *callback)
1047 {
1048     BSCallback *iter;
1049
1050     if(doc->bscallback) {
1051         if(doc->bscallback->bsc.binding)
1052             IBinding_Abort(doc->bscallback->bsc.binding);
1053         doc->bscallback->bsc.doc = NULL;
1054         IBindStatusCallback_Release(STATUSCLB(&doc->bscallback->bsc));
1055     }
1056
1057     LIST_FOR_EACH_ENTRY(iter, &doc->bindings, BSCallback, entry) {
1058         iter->doc = NULL;
1059         list_remove(&iter->entry);
1060     }
1061
1062     doc->bscallback = callback;
1063
1064     if(callback) {
1065         IBindStatusCallback_AddRef(STATUSCLB(&callback->bsc));
1066         callback->bsc.doc = doc;
1067     }
1068 }
1069
1070 HRESULT channelbsc_load_stream(nsChannelBSC *bscallback, IStream *stream)
1071 {
1072     HRESULT hres;
1073
1074     const char text_html[] = "text/html";
1075
1076     add_nsrequest(bscallback);
1077
1078     if(bscallback->nschannel)
1079         bscallback->nschannel->content_type = heap_strdupA(text_html);
1080
1081     hres = read_stream_data(bscallback, stream);
1082     IBindStatusCallback_OnStopBinding(STATUSCLB(&bscallback->bsc), hres, ERROR_SUCCESS);
1083
1084     return hres;
1085 }
1086
1087 void channelbsc_set_channel(nsChannelBSC *This, nsChannel *channel, nsIStreamListener *listener, nsISupports *context)
1088 {
1089     nsIChannel_AddRef(NSCHANNEL(channel));
1090     This->nschannel = channel;
1091
1092     nsIStreamListener_AddRef(listener);
1093     This->nslistener = listener;
1094
1095     if(context) {
1096         nsISupports_AddRef(context);
1097         This->nscontext = context;
1098     }
1099 }
1100
1101 void hlink_frame_navigate(HTMLDocument *doc, IHlinkFrame *hlink_frame,
1102                           LPCWSTR uri, nsIInputStream *post_data_stream, DWORD hlnf)
1103 {
1104     BSCallback *callback;
1105     IBindCtx *bindctx;
1106     IMoniker *mon;
1107     IHlink *hlink;
1108     HRESULT hr;
1109
1110     callback = &create_channelbsc(NULL)->bsc;
1111
1112     if(post_data_stream) {
1113         parse_post_data(post_data_stream, &callback->headers, &callback->post_data,
1114                         &callback->post_data_len);
1115         TRACE("headers = %s post_data = %s\n", debugstr_w(callback->headers),
1116               debugstr_an(callback->post_data, callback->post_data_len));
1117     }
1118
1119     hr = CreateAsyncBindCtx(0, STATUSCLB(callback), NULL, &bindctx);
1120     if (FAILED(hr)) {
1121         IBindStatusCallback_Release(STATUSCLB(callback));
1122         return;
1123     }
1124
1125     hr = CoCreateInstance(&CLSID_StdHlink, NULL, CLSCTX_INPROC_SERVER, &IID_IHlink, (LPVOID*)&hlink);
1126     if (FAILED(hr)) {
1127         IBindCtx_Release(bindctx);
1128         IBindStatusCallback_Release(STATUSCLB(callback));
1129         return;
1130     }
1131
1132     hr = CreateURLMoniker(NULL, uri, &mon);
1133     if (SUCCEEDED(hr)) {
1134         IHlink_SetMonikerReference(hlink, 0, mon, NULL);
1135
1136         if(hlnf & HLNF_OPENINNEWWINDOW) {
1137             static const WCHAR wszBlank[] = {'_','b','l','a','n','k',0};
1138             IHlink_SetTargetFrameName(hlink, wszBlank); /* FIXME */
1139         }
1140
1141         IHlinkFrame_Navigate(hlink_frame, hlnf, bindctx, STATUSCLB(callback), hlink);
1142
1143         IMoniker_Release(mon);
1144     }
1145
1146     IBindCtx_Release(bindctx);
1147     IBindStatusCallback_Release(STATUSCLB(callback));
1148 }