git.oblomov.eu Git - wine/blob - dlls/mshtml/navigate.c

   1 /*
   2  * Copyright 2006-2010 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 "winreg.h"
  31 #include "ole2.h"
  32 #include "hlguids.h"
  33 #include "shlguid.h"
  34 #include "wininet.h"
  35 #include "shlwapi.h"
  36
  37 #include "wine/debug.h"
  38 #include "wine/unicode.h"
  39
  40 #include "mshtml_private.h"
  41
  42 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  43
  44 #define CONTENT_LENGTH "Content-Length"
  45 #define UTF16_STR "utf-16"
  46
  47 typedef struct {
  48     nsIInputStream nsIInputStream_iface;
  49
  50     LONG ref;
  51
  52     char buf[1024];
  53     DWORD buf_size;
  54 } nsProtocolStream;
  55
  56 typedef struct {
  57     void (*destroy)(BSCallback*);
  58     HRESULT (*init_bindinfo)(BSCallback*);
  59     HRESULT (*start_binding)(BSCallback*);
  60     HRESULT (*stop_binding)(BSCallback*,HRESULT);
  61     HRESULT (*read_data)(BSCallback*,IStream*);
  62     HRESULT (*on_progress)(BSCallback*,ULONG,LPCWSTR);
  63     HRESULT (*on_response)(BSCallback*,DWORD,LPCWSTR);
  64     HRESULT (*beginning_transaction)(BSCallback*,WCHAR**);
  65 } BSCallbackVtbl;
  66
  67 struct BSCallback {
  68     IBindStatusCallback IBindStatusCallback_iface;
  69     IServiceProvider    IServiceProvider_iface;
  70     IHttpNegotiate2     IHttpNegotiate2_iface;
  71     IInternetBindInfo   IInternetBindInfo_iface;
  72
  73     const BSCallbackVtbl          *vtbl;
  74
  75     LONG ref;
  76
  77     LPWSTR headers;
  78     HGLOBAL post_data;
  79     ULONG post_data_len;
  80     ULONG readed;
  81     DWORD bindf;
  82     BOOL bindinfo_ready;
  83
  84     IMoniker *mon;
  85     IBinding *binding;
  86
  87     HTMLDocumentNode *doc;
  88
  89     struct list entry;
  90 };
  91
  92 static inline nsProtocolStream *impl_from_nsIInputStream(nsIInputStream *iface)
  93 {
  94     return CONTAINING_RECORD(iface, nsProtocolStream, nsIInputStream_iface);
  95 }
  96
  97 static nsresult NSAPI nsInputStream_QueryInterface(nsIInputStream *iface, nsIIDRef riid,
  98         void **result)
  99 {
 100     nsProtocolStream *This = impl_from_nsIInputStream(iface);
 101
 102     *result = NULL;
 103
 104     if(IsEqualGUID(&IID_nsISupports, riid)) {
 105         TRACE("(%p)->(IID_nsISupports %p)\n", This, result);
 106         *result  = &This->nsIInputStream_iface;
 107     }else if(IsEqualGUID(&IID_nsIInputStream, riid)) {
 108         TRACE("(%p)->(IID_nsIInputStream %p)\n", This, result);
 109         *result  = &This->nsIInputStream_iface;
 110     }
 111
 112     if(*result) {
 113         nsIInputStream_AddRef(&This->nsIInputStream_iface);
 114         return NS_OK;
 115     }
 116
 117     WARN("unsupported interface %s\n", debugstr_guid(riid));
 118     return NS_NOINTERFACE;
 119 }
 120
 121 static nsrefcnt NSAPI nsInputStream_AddRef(nsIInputStream *iface)
 122 {
 123     nsProtocolStream *This = impl_from_nsIInputStream(iface);
 124     LONG ref = InterlockedIncrement(&This->ref);
 125
 126     TRACE("(%p) ref=%d\n", This, ref);
 127
 128     return ref;
 129 }
 130
 131
 132 static nsrefcnt NSAPI nsInputStream_Release(nsIInputStream *iface)
 133 {
 134     nsProtocolStream *This = impl_from_nsIInputStream(iface);
 135     LONG ref = InterlockedDecrement(&This->ref);
 136
 137     TRACE("(%p) ref=%d\n", This, ref);
 138
 139     if(!ref)
 140         heap_free(This);
 141
 142     return ref;
 143 }
 144
 145 static nsresult NSAPI nsInputStream_Close(nsIInputStream *iface)
 146 {
 147     nsProtocolStream *This = impl_from_nsIInputStream(iface);
 148     FIXME("(%p)\n", This);
 149     return NS_ERROR_NOT_IMPLEMENTED;
 150 }
 151
 152 static nsresult NSAPI nsInputStream_Available(nsIInputStream *iface, PRUint32 *_retval)
 153 {
 154     nsProtocolStream *This = impl_from_nsIInputStream(iface);
 155     FIXME("(%p)->(%p)\n", This, _retval);
 156     return NS_ERROR_NOT_IMPLEMENTED;
 157 }
 158
 159 static nsresult NSAPI nsInputStream_Read(nsIInputStream *iface, char *aBuf, PRUint32 aCount,
 160                                          PRUint32 *_retval)
 161 {
 162     nsProtocolStream *This = impl_from_nsIInputStream(iface);
 163     DWORD read = aCount;
 164
 165     TRACE("(%p)->(%p %d %p)\n", This, aBuf, aCount, _retval);
 166
 167     if(read > This->buf_size)
 168         read = This->buf_size;
 169
 170     if(read) {
 171         memcpy(aBuf, This->buf, read);
 172         if(read < This->buf_size)
 173             memmove(This->buf, This->buf+read, This->buf_size-read);
 174         This->buf_size -= read;
 175     }
 176
 177     *_retval = read;
 178     return NS_OK;
 179 }
 180
 181 static nsresult NSAPI nsInputStream_ReadSegments(nsIInputStream *iface,
 182         nsresult (WINAPI *aWriter)(nsIInputStream*,void*,const char*,PRUint32,PRUint32,PRUint32*),
 183         void *aClousure, PRUint32 aCount, PRUint32 *_retval)
 184 {
 185     nsProtocolStream *This = impl_from_nsIInputStream(iface);
 186     PRUint32 written = 0;
 187     nsresult nsres;
 188
 189     TRACE("(%p)->(%p %p %d %p)\n", This, aWriter, aClousure, aCount, _retval);
 190
 191     if(!This->buf_size)
 192         return S_OK;
 193
 194     if(aCount > This->buf_size)
 195         aCount = This->buf_size;
 196
 197     nsres = aWriter(&This->nsIInputStream_iface, aClousure, This->buf, 0, aCount, &written);
 198     if(NS_FAILED(nsres))
 199         TRACE("aWritter failed: %08x\n", nsres);
 200     else if(written != This->buf_size)
 201         FIXME("written %d != buf_size %d\n", written, This->buf_size);
 202
 203     This->buf_size -= written;
 204
 205     *_retval = written;
 206     return nsres;
 207 }
 208
 209 static nsresult NSAPI nsInputStream_IsNonBlocking(nsIInputStream *iface, PRBool *_retval)
 210 {
 211     nsProtocolStream *This = impl_from_nsIInputStream(iface);
 212     FIXME("(%p)->(%p)\n", This, _retval);
 213     return NS_ERROR_NOT_IMPLEMENTED;
 214 }
 215
 216 static const nsIInputStreamVtbl nsInputStreamVtbl = {
 217     nsInputStream_QueryInterface,
 218     nsInputStream_AddRef,
 219     nsInputStream_Release,
 220     nsInputStream_Close,
 221     nsInputStream_Available,
 222     nsInputStream_Read,
 223     nsInputStream_ReadSegments,
 224     nsInputStream_IsNonBlocking
 225 };
 226
 227 static nsProtocolStream *create_nsprotocol_stream(void)
 228 {
 229     nsProtocolStream *ret = heap_alloc(sizeof(nsProtocolStream));
 230
 231     ret->nsIInputStream_iface.lpVtbl = &nsInputStreamVtbl;
 232     ret->ref = 1;
 233     ret->buf_size = 0;
 234
 235     return ret;
 236 }
 237
 238 static inline BSCallback *impl_from_IBindStatusCallback(IBindStatusCallback *iface)
 239 {
 240     return CONTAINING_RECORD(iface, BSCallback, IBindStatusCallback_iface);
 241 }
 242
 243 static HRESULT WINAPI BindStatusCallback_QueryInterface(IBindStatusCallback *iface,
 244         REFIID riid, void **ppv)
 245 {
 246     BSCallback *This = impl_from_IBindStatusCallback(iface);
 247
 248     *ppv = NULL;
 249     if(IsEqualGUID(&IID_IUnknown, riid)) {
 250         TRACE("(%p)->(IID_IUnknown, %p)\n", This, ppv);
 251         *ppv = &This->IBindStatusCallback_iface;
 252     }else if(IsEqualGUID(&IID_IBindStatusCallback, riid)) {
 253         TRACE("(%p)->(IID_IBindStatusCallback, %p)\n", This, ppv);
 254         *ppv = &This->IBindStatusCallback_iface;
 255     }else if(IsEqualGUID(&IID_IServiceProvider, riid)) {
 256         TRACE("(%p)->(IID_IServiceProvider %p)\n", This, ppv);
 257         *ppv = &This->IServiceProvider_iface;
 258     }else if(IsEqualGUID(&IID_IHttpNegotiate, riid)) {
 259         TRACE("(%p)->(IID_IHttpNegotiate %p)\n", This, ppv);
 260         *ppv = &This->IHttpNegotiate2_iface;
 261     }else if(IsEqualGUID(&IID_IHttpNegotiate2, riid)) {
 262         TRACE("(%p)->(IID_IHttpNegotiate2 %p)\n", This, ppv);
 263         *ppv = &This->IHttpNegotiate2_iface;
 264     }else if(IsEqualGUID(&IID_IInternetBindInfo, riid)) {
 265         TRACE("(%p)->(IID_IInternetBindInfo %p)\n", This, ppv);
 266         *ppv = &This->IInternetBindInfo_iface;
 267     }
 268
 269     if(*ppv) {
 270         IBindStatusCallback_AddRef(&This->IBindStatusCallback_iface);
 271         return S_OK;
 272     }
 273
 274     TRACE("Unsupported riid = %s\n", debugstr_guid(riid));
 275     return E_NOINTERFACE;
 276 }
 277
 278 static ULONG WINAPI BindStatusCallback_AddRef(IBindStatusCallback *iface)
 279 {
 280     BSCallback *This = impl_from_IBindStatusCallback(iface);
 281     LONG ref = InterlockedIncrement(&This->ref);
 282
 283     TRACE("(%p) ref = %d\n", This, ref);
 284
 285     return ref;
 286 }
 287
 288 static ULONG WINAPI BindStatusCallback_Release(IBindStatusCallback *iface)
 289 {
 290     BSCallback *This = impl_from_IBindStatusCallback(iface);
 291     LONG ref = InterlockedDecrement(&This->ref);
 292
 293     TRACE("(%p) ref = %d\n", This, ref);
 294
 295     if(!ref) {
 296         if(This->post_data)
 297             GlobalFree(This->post_data);
 298         if(This->mon)
 299             IMoniker_Release(This->mon);
 300         if(This->binding)
 301             IBinding_Release(This->binding);
 302         list_remove(&This->entry);
 303         list_init(&This->entry);
 304         heap_free(This->headers);
 305
 306         This->vtbl->destroy(This);
 307     }
 308
 309     return ref;
 310 }
 311
 312 static HRESULT WINAPI BindStatusCallback_OnStartBinding(IBindStatusCallback *iface,
 313         DWORD dwReserved, IBinding *pbind)
 314 {
 315     BSCallback *This = impl_from_IBindStatusCallback(iface);
 316
 317     TRACE("(%p)->(%d %p)\n", This, dwReserved, pbind);
 318
 319     IBinding_AddRef(pbind);
 320     This->binding = pbind;
 321
 322     if(This->doc)
 323         list_add_head(&This->doc->bindings, &This->entry);
 324
 325     return This->vtbl->start_binding(This);
 326 }
 327
 328 static HRESULT WINAPI BindStatusCallback_GetPriority(IBindStatusCallback *iface, LONG *pnPriority)
 329 {
 330     BSCallback *This = impl_from_IBindStatusCallback(iface);
 331     FIXME("(%p)->(%p)\n", This, pnPriority);
 332     return E_NOTIMPL;
 333 }
 334
 335 static HRESULT WINAPI BindStatusCallback_OnLowResource(IBindStatusCallback *iface, DWORD reserved)
 336 {
 337     BSCallback *This = impl_from_IBindStatusCallback(iface);
 338     FIXME("(%p)->(%d)\n", This, reserved);
 339     return E_NOTIMPL;
 340 }
 341
 342 static HRESULT WINAPI BindStatusCallback_OnProgress(IBindStatusCallback *iface, ULONG ulProgress,
 343         ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText)
 344 {
 345     BSCallback *This = impl_from_IBindStatusCallback(iface);
 346
 347     TRACE("%p)->(%u %u %u %s)\n", This, ulProgress, ulProgressMax, ulStatusCode,
 348             debugstr_w(szStatusText));
 349
 350     return This->vtbl->on_progress(This, ulStatusCode, szStatusText);
 351 }
 352
 353 static HRESULT WINAPI BindStatusCallback_OnStopBinding(IBindStatusCallback *iface,
 354         HRESULT hresult, LPCWSTR szError)
 355 {
 356     BSCallback *This = impl_from_IBindStatusCallback(iface);
 357     HRESULT hres;
 358
 359     TRACE("(%p)->(%08x %s)\n", This, hresult, debugstr_w(szError));
 360
 361     /* NOTE: IE7 calls GetBindResult here */
 362
 363     hres = This->vtbl->stop_binding(This, hresult);
 364
 365     if(This->binding) {
 366         IBinding_Release(This->binding);
 367         This->binding = NULL;
 368     }
 369
 370     list_remove(&This->entry);
 371     list_init(&This->entry);
 372     This->doc = NULL;
 373
 374     return hres;
 375 }
 376
 377 static HRESULT WINAPI BindStatusCallback_GetBindInfo(IBindStatusCallback *iface,
 378         DWORD *grfBINDF, BINDINFO *pbindinfo)
 379 {
 380     BSCallback *This = impl_from_IBindStatusCallback(iface);
 381     DWORD size;
 382
 383     TRACE("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo);
 384
 385     if(!This->bindinfo_ready) {
 386         HRESULT hres;
 387
 388         hres = This->vtbl->init_bindinfo(This);
 389         if(FAILED(hres))
 390             return hres;
 391
 392         This->bindinfo_ready = TRUE;
 393     }
 394
 395     *grfBINDF = This->bindf;
 396
 397     size = pbindinfo->cbSize;
 398     memset(pbindinfo, 0, size);
 399     pbindinfo->cbSize = size;
 400
 401     pbindinfo->cbstgmedData = This->post_data_len;
 402     pbindinfo->dwCodePage = CP_UTF8;
 403     pbindinfo->dwOptions = 0x80000;
 404
 405     if(This->post_data) {
 406         pbindinfo->dwBindVerb = BINDVERB_POST;
 407
 408         pbindinfo->stgmedData.tymed = TYMED_HGLOBAL;
 409         pbindinfo->stgmedData.u.hGlobal = This->post_data;
 410         pbindinfo->stgmedData.pUnkForRelease = (IUnknown*)&This->IBindStatusCallback_iface;
 411         IBindStatusCallback_AddRef(&This->IBindStatusCallback_iface);
 412     }
 413
 414     return S_OK;
 415 }
 416
 417 static HRESULT WINAPI BindStatusCallback_OnDataAvailable(IBindStatusCallback *iface,
 418         DWORD grfBSCF, DWORD dwSize, FORMATETC *pformatetc, STGMEDIUM *pstgmed)
 419 {
 420     BSCallback *This = impl_from_IBindStatusCallback(iface);
 421
 422     TRACE("(%p)->(%08x %d %p %p)\n", This, grfBSCF, dwSize, pformatetc, pstgmed);
 423
 424     return This->vtbl->read_data(This, pstgmed->u.pstm);
 425 }
 426
 427 static HRESULT WINAPI BindStatusCallback_OnObjectAvailable(IBindStatusCallback *iface,
 428         REFIID riid, IUnknown *punk)
 429 {
 430     BSCallback *This = impl_from_IBindStatusCallback(iface);
 431     FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), punk);
 432     return E_NOTIMPL;
 433 }
 434
 435 static const IBindStatusCallbackVtbl BindStatusCallbackVtbl = {
 436     BindStatusCallback_QueryInterface,
 437     BindStatusCallback_AddRef,
 438     BindStatusCallback_Release,
 439     BindStatusCallback_OnStartBinding,
 440     BindStatusCallback_GetPriority,
 441     BindStatusCallback_OnLowResource,
 442     BindStatusCallback_OnProgress,
 443     BindStatusCallback_OnStopBinding,
 444     BindStatusCallback_GetBindInfo,
 445     BindStatusCallback_OnDataAvailable,
 446     BindStatusCallback_OnObjectAvailable
 447 };
 448
 449 static inline BSCallback *impl_from_IHttpNegotiate2(IHttpNegotiate2 *iface)
 450 {
 451     return CONTAINING_RECORD(iface, BSCallback, IHttpNegotiate2_iface);
 452 }
 453
 454 static HRESULT WINAPI HttpNegotiate_QueryInterface(IHttpNegotiate2 *iface,
 455                                                    REFIID riid, void **ppv)
 456 {
 457     BSCallback *This = impl_from_IHttpNegotiate2(iface);
 458     return IBindStatusCallback_QueryInterface(&This->IBindStatusCallback_iface, riid, ppv);
 459 }
 460
 461 static ULONG WINAPI HttpNegotiate_AddRef(IHttpNegotiate2 *iface)
 462 {
 463     BSCallback *This = impl_from_IHttpNegotiate2(iface);
 464     return IBindStatusCallback_AddRef(&This->IBindStatusCallback_iface);
 465 }
 466
 467 static ULONG WINAPI HttpNegotiate_Release(IHttpNegotiate2 *iface)
 468 {
 469     BSCallback *This = impl_from_IHttpNegotiate2(iface);
 470     return IBindStatusCallback_Release(&This->IBindStatusCallback_iface);
 471 }
 472
 473 static HRESULT WINAPI HttpNegotiate_BeginningTransaction(IHttpNegotiate2 *iface,
 474         LPCWSTR szURL, LPCWSTR szHeaders, DWORD dwReserved, LPWSTR *pszAdditionalHeaders)
 475 {
 476     BSCallback *This = impl_from_IHttpNegotiate2(iface);
 477     HRESULT hres;
 478
 479     TRACE("(%p)->(%s %s %d %p)\n", This, debugstr_w(szURL), debugstr_w(szHeaders),
 480           dwReserved, pszAdditionalHeaders);
 481
 482     *pszAdditionalHeaders = NULL;
 483
 484     hres = This->vtbl->beginning_transaction(This, pszAdditionalHeaders);
 485     if(hres != S_FALSE)
 486         return hres;
 487
 488     if(This->headers) {
 489         DWORD size;
 490
 491         size = (strlenW(This->headers)+1)*sizeof(WCHAR);
 492         *pszAdditionalHeaders = CoTaskMemAlloc(size);
 493         if(!*pszAdditionalHeaders)
 494             return E_OUTOFMEMORY;
 495         memcpy(*pszAdditionalHeaders, This->headers, size);
 496     }
 497
 498     return S_OK;
 499 }
 500
 501 static HRESULT WINAPI HttpNegotiate_OnResponse(IHttpNegotiate2 *iface, DWORD dwResponseCode,
 502         LPCWSTR szResponseHeaders, LPCWSTR szRequestHeaders, LPWSTR *pszAdditionalRequestHeaders)
 503 {
 504     BSCallback *This = impl_from_IHttpNegotiate2(iface);
 505
 506     TRACE("(%p)->(%d %s %s %p)\n", This, dwResponseCode, debugstr_w(szResponseHeaders),
 507           debugstr_w(szRequestHeaders), pszAdditionalRequestHeaders);
 508
 509     return This->vtbl->on_response(This, dwResponseCode, szResponseHeaders);
 510 }
 511
 512 static HRESULT WINAPI HttpNegotiate_GetRootSecurityId(IHttpNegotiate2 *iface,
 513         BYTE *pbSecurityId, DWORD *pcbSecurityId, DWORD_PTR dwReserved)
 514 {
 515     BSCallback *This = impl_from_IHttpNegotiate2(iface);
 516     FIXME("(%p)->(%p %p %ld)\n", This, pbSecurityId, pcbSecurityId, dwReserved);
 517     return E_NOTIMPL;
 518 }
 519
 520 static const IHttpNegotiate2Vtbl HttpNegotiate2Vtbl = {
 521     HttpNegotiate_QueryInterface,
 522     HttpNegotiate_AddRef,
 523     HttpNegotiate_Release,
 524     HttpNegotiate_BeginningTransaction,
 525     HttpNegotiate_OnResponse,
 526     HttpNegotiate_GetRootSecurityId
 527 };
 528
 529 static inline BSCallback *impl_from_IInternetBindInfo(IInternetBindInfo *iface)
 530 {
 531     return CONTAINING_RECORD(iface, BSCallback, IInternetBindInfo_iface);
 532 }
 533
 534 static HRESULT WINAPI InternetBindInfo_QueryInterface(IInternetBindInfo *iface,
 535                                                       REFIID riid, void **ppv)
 536 {
 537     BSCallback *This = impl_from_IInternetBindInfo(iface);
 538     return IBindStatusCallback_QueryInterface(&This->IBindStatusCallback_iface, riid, ppv);
 539 }
 540
 541 static ULONG WINAPI InternetBindInfo_AddRef(IInternetBindInfo *iface)
 542 {
 543     BSCallback *This = impl_from_IInternetBindInfo(iface);
 544     return IBindStatusCallback_AddRef(&This->IBindStatusCallback_iface);
 545 }
 546
 547 static ULONG WINAPI InternetBindInfo_Release(IInternetBindInfo *iface)
 548 {
 549     BSCallback *This = impl_from_IInternetBindInfo(iface);
 550     return IBindStatusCallback_Release(&This->IBindStatusCallback_iface);
 551 }
 552
 553 static HRESULT WINAPI InternetBindInfo_GetBindInfo(IInternetBindInfo *iface,
 554                                                    DWORD *grfBINDF, BINDINFO *pbindinfo)
 555 {
 556     BSCallback *This = impl_from_IInternetBindInfo(iface);
 557     FIXME("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo);
 558     return E_NOTIMPL;
 559 }
 560
 561 static HRESULT WINAPI InternetBindInfo_GetBindString(IInternetBindInfo *iface,
 562         ULONG ulStringType, LPOLESTR *ppwzStr, ULONG cEl, ULONG *pcElFetched)
 563 {
 564     BSCallback *This = impl_from_IInternetBindInfo(iface);
 565     FIXME("(%p)->(%u %p %u %p)\n", This, ulStringType, ppwzStr, cEl, pcElFetched);
 566     return E_NOTIMPL;
 567 }
 568
 569 static const IInternetBindInfoVtbl InternetBindInfoVtbl = {
 570     InternetBindInfo_QueryInterface,
 571     InternetBindInfo_AddRef,
 572     InternetBindInfo_Release,
 573     InternetBindInfo_GetBindInfo,
 574     InternetBindInfo_GetBindString
 575 };
 576
 577 static inline BSCallback *impl_from_IServiceProvider(IServiceProvider *iface)
 578 {
 579     return CONTAINING_RECORD(iface, BSCallback, IServiceProvider_iface);
 580 }
 581
 582 static HRESULT WINAPI BSCServiceProvider_QueryInterface(IServiceProvider *iface,
 583                                                         REFIID riid, void **ppv)
 584 {
 585     BSCallback *This = impl_from_IServiceProvider(iface);
 586     return IBindStatusCallback_QueryInterface(&This->IBindStatusCallback_iface, riid, ppv);
 587 }
 588
 589 static ULONG WINAPI BSCServiceProvider_AddRef(IServiceProvider *iface)
 590 {
 591     BSCallback *This = impl_from_IServiceProvider(iface);
 592     return IBindStatusCallback_AddRef(&This->IBindStatusCallback_iface);
 593 }
 594
 595 static ULONG WINAPI BSCServiceProvider_Release(IServiceProvider *iface)
 596 {
 597     BSCallback *This = impl_from_IServiceProvider(iface);
 598     return IBindStatusCallback_Release(&This->IBindStatusCallback_iface);
 599 }
 600
 601 static HRESULT WINAPI BSCServiceProvider_QueryService(IServiceProvider *iface,
 602         REFGUID guidService, REFIID riid, void **ppv)
 603 {
 604     BSCallback *This = impl_from_IServiceProvider(iface);
 605     TRACE("(%p)->(%s %s %p)\n", This, debugstr_guid(guidService), debugstr_guid(riid), ppv);
 606     return E_NOINTERFACE;
 607 }
 608
 609 static const IServiceProviderVtbl ServiceProviderVtbl = {
 610     BSCServiceProvider_QueryInterface,
 611     BSCServiceProvider_AddRef,
 612     BSCServiceProvider_Release,
 613     BSCServiceProvider_QueryService
 614 };
 615
 616 static void init_bscallback(BSCallback *This, const BSCallbackVtbl *vtbl, IMoniker *mon, DWORD bindf)
 617 {
 618     This->IBindStatusCallback_iface.lpVtbl = &BindStatusCallbackVtbl;
 619     This->IServiceProvider_iface.lpVtbl = &ServiceProviderVtbl;
 620     This->IHttpNegotiate2_iface.lpVtbl = &HttpNegotiate2Vtbl;
 621     This->IInternetBindInfo_iface.lpVtbl = &InternetBindInfoVtbl;
 622     This->vtbl = vtbl;
 623     This->ref = 1;
 624     This->bindf = bindf;
 625
 626     list_init(&This->entry);
 627
 628     if(mon)
 629         IMoniker_AddRef(mon);
 630     This->mon = mon;
 631 }
 632
 633 /* Calls undocumented 84 cmd of CGID_ShellDocView */
 634 static void call_docview_84(HTMLDocumentObj *doc)
 635 {
 636     IOleCommandTarget *olecmd;
 637     VARIANT var;
 638     HRESULT hres;
 639
 640     if(!doc->client)
 641         return;
 642
 643     hres = IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&olecmd);
 644     if(FAILED(hres))
 645         return;
 646
 647     VariantInit(&var);
 648     hres = IOleCommandTarget_Exec(olecmd, &CGID_ShellDocView, 84, 0, NULL, &var);
 649     IOleCommandTarget_Release(olecmd);
 650     if(SUCCEEDED(hres) && V_VT(&var) != VT_NULL)
 651         FIXME("handle result\n");
 652 }
 653
 654 static HRESULT parse_headers(const WCHAR *headers, struct list *headers_list)
 655 {
 656     const WCHAR *header, *header_end, *colon, *value;
 657     HRESULT hres;
 658
 659     header = headers;
 660     while(*header) {
 661         if(header[0] == '\r' && header[1] == '\n' && !header[2])
 662             break;
 663         for(colon = header; *colon && *colon != ':' && *colon != '\r'; colon++);
 664         if(*colon != ':')
 665             return E_FAIL;
 666
 667         value = colon+1;
 668         while(*value == ' ')
 669             value++;
 670         if(!*value)
 671             return E_FAIL;
 672
 673         for(header_end = value+1; *header_end && *header_end != '\r'; header_end++);
 674
 675         hres = set_http_header(headers_list, header, colon-header, value, header_end-value);
 676         if(FAILED(hres))
 677             return hres;
 678
 679         header = header_end;
 680         if(header[0] == '\r' && header[1] == '\n')
 681             header += 2;
 682     }
 683
 684     return S_OK;
 685 }
 686
 687 static HRESULT read_post_data_stream(nsIInputStream *stream, HGLOBAL *post_data,
 688         ULONG *post_data_len)
 689 {
 690     PRUint32 data_len = 0, available = 0;
 691     char *data;
 692     nsresult nsres;
 693
 694     nsres =  nsIInputStream_Available(stream, &available);
 695     if(NS_FAILED(nsres))
 696         return E_FAIL;
 697
 698     data = GlobalAlloc(0, available+1);
 699     if(!data)
 700         return E_OUTOFMEMORY;
 701
 702     nsres = nsIInputStream_Read(stream, data, available, &data_len);
 703     if(NS_FAILED(nsres)) {
 704         GlobalFree(data);
 705         return E_FAIL;
 706     }
 707
 708     data[data_len] = 0;
 709     *post_data = data;
 710     *post_data_len = data_len;
 711     return S_OK;
 712 }
 713
 714 HRESULT start_binding(HTMLWindow *window, HTMLDocumentNode *doc, BSCallback *bscallback, IBindCtx *bctx)
 715 {
 716     IStream *str = NULL;
 717     HRESULT hres;
 718
 719     bscallback->doc = doc;
 720
 721     /* NOTE: IE7 calls IsSystemMoniker here*/
 722
 723     if(window) {
 724         if(bscallback->mon != window->mon)
 725             set_current_mon(window, bscallback->mon);
 726         call_docview_84(window->doc_obj);
 727     }
 728
 729     if(bctx) {
 730         RegisterBindStatusCallback(bctx, &bscallback->IBindStatusCallback_iface, NULL, 0);
 731         IBindCtx_AddRef(bctx);
 732     }else {
 733         hres = CreateAsyncBindCtx(0, &bscallback->IBindStatusCallback_iface, NULL, &bctx);
 734         if(FAILED(hres)) {
 735             WARN("CreateAsyncBindCtx failed: %08x\n", hres);
 736             bscallback->vtbl->stop_binding(bscallback, hres);
 737             return hres;
 738         }
 739     }
 740
 741     hres = IMoniker_BindToStorage(bscallback->mon, bctx, NULL, &IID_IStream, (void**)&str);
 742     IBindCtx_Release(bctx);
 743     if(FAILED(hres)) {
 744         WARN("BindToStorage failed: %08x\n", hres);
 745         bscallback->vtbl->stop_binding(bscallback, hres);
 746         return hres;
 747     }
 748
 749     if(str)
 750         IStream_Release(str);
 751
 752     IMoniker_Release(bscallback->mon);
 753     bscallback->mon = NULL;
 754
 755     return S_OK;
 756 }
 757
 758 typedef struct {
 759     BSCallback bsc;
 760
 761     DWORD size;
 762     BYTE *buf;
 763     HRESULT hres;
 764 } BufferBSC;
 765
 766 static inline BufferBSC *BufferBSC_from_BSCallback(BSCallback *iface)
 767 {
 768     return CONTAINING_RECORD(iface, BufferBSC, bsc);
 769 }
 770
 771 static void BufferBSC_destroy(BSCallback *bsc)
 772 {
 773     BufferBSC *This = BufferBSC_from_BSCallback(bsc);
 774
 775     heap_free(This->buf);
 776     heap_free(This);
 777 }
 778
 779 static HRESULT BufferBSC_init_bindinfo(BSCallback *bsc)
 780 {
 781     return S_OK;
 782 }
 783
 784 static HRESULT BufferBSC_start_binding(BSCallback *bsc)
 785 {
 786     return S_OK;
 787 }
 788
 789 static HRESULT BufferBSC_stop_binding(BSCallback *bsc, HRESULT result)
 790 {
 791     BufferBSC *This = BufferBSC_from_BSCallback(bsc);
 792
 793     This->hres = result;
 794
 795     if(FAILED(result)) {
 796         heap_free(This->buf);
 797         This->buf = NULL;
 798         This->size = 0;
 799     }
 800
 801     return S_OK;
 802 }
 803
 804 static HRESULT BufferBSC_read_data(BSCallback *bsc, IStream *stream)
 805 {
 806     BufferBSC *This = BufferBSC_from_BSCallback(bsc);
 807     DWORD readed;
 808     HRESULT hres;
 809
 810     if(!This->buf) {
 811         This->size = 128;
 812         This->buf = heap_alloc(This->size);
 813     }
 814
 815     do {
 816         if(This->bsc.readed == This->size) {
 817             This->size <<= 1;
 818             This->buf = heap_realloc(This->buf, This->size);
 819         }
 820
 821         readed = 0;
 822         hres = IStream_Read(stream, This->buf+This->bsc.readed, This->size-This->bsc.readed, &readed);
 823         This->bsc.readed += readed;
 824     }while(hres == S_OK);
 825
 826     return S_OK;
 827 }
 828
 829 static HRESULT BufferBSC_on_progress(BSCallback *bsc, ULONG status_code, LPCWSTR status_text)
 830 {
 831     return S_OK;
 832 }
 833
 834 static HRESULT BufferBSC_on_response(BSCallback *bsc, DWORD response_code,
 835         LPCWSTR response_headers)
 836 {
 837     return S_OK;
 838 }
 839
 840 static HRESULT BufferBSC_beginning_transaction(BSCallback *bsc, WCHAR **additional_headers)
 841 {
 842     return S_FALSE;
 843 }
 844
 845 static const BSCallbackVtbl BufferBSCVtbl = {
 846     BufferBSC_destroy,
 847     BufferBSC_init_bindinfo,
 848     BufferBSC_start_binding,
 849     BufferBSC_stop_binding,
 850     BufferBSC_read_data,
 851     BufferBSC_on_progress,
 852     BufferBSC_on_response,
 853     BufferBSC_beginning_transaction
 854 };
 855
 856
 857 static BufferBSC *create_bufferbsc(IMoniker *mon)
 858 {
 859     BufferBSC *ret = heap_alloc_zero(sizeof(*ret));
 860
 861     init_bscallback(&ret->bsc, &BufferBSCVtbl, mon, 0);
 862     ret->hres = E_FAIL;
 863
 864     return ret;
 865 }
 866
 867 HRESULT bind_mon_to_buffer(HTMLDocumentNode *doc, IMoniker *mon, void **buf, DWORD *size)
 868 {
 869     BufferBSC *bsc = create_bufferbsc(mon);
 870     HRESULT hres;
 871
 872     *buf = NULL;
 873
 874     hres = start_binding(NULL, doc, &bsc->bsc, NULL);
 875     if(SUCCEEDED(hres)) {
 876         hres = bsc->hres;
 877         if(SUCCEEDED(hres)) {
 878             *buf = bsc->buf;
 879             bsc->buf = NULL;
 880             *size = bsc->bsc.readed;
 881             bsc->size = 0;
 882         }
 883     }
 884
 885     IBindStatusCallback_Release(&bsc->bsc.IBindStatusCallback_iface);
 886
 887     return hres;
 888 }
 889
 890 struct nsChannelBSC {
 891     BSCallback bsc;
 892
 893     HTMLWindow *window;
 894
 895     nsChannel *nschannel;
 896     nsIStreamListener *nslistener;
 897     nsISupports *nscontext;
 898
 899     nsProtocolStream *nsstream;
 900 };
 901
 902 static HRESULT on_start_nsrequest(nsChannelBSC *This)
 903 {
 904     nsresult nsres;
 905
 906     /* FIXME: it's needed for http connections from BindToObject. */
 907     if(!This->nschannel->response_status)
 908         This->nschannel->response_status = 200;
 909
 910     nsres = nsIStreamListener_OnStartRequest(This->nslistener,
 911             (nsIRequest*)&This->nschannel->nsIHttpChannel_iface, This->nscontext);
 912     if(NS_FAILED(nsres)) {
 913         FIXME("OnStartRequest failed: %08x\n", nsres);
 914         return E_FAIL;
 915     }
 916
 917     if(This->window) {
 918         update_window_doc(This->window);
 919         if(This->window->doc != This->bsc.doc)
 920             This->bsc.doc = This->window->doc;
 921         if(This->window->readystate != READYSTATE_LOADING)
 922             set_ready_state(This->window, READYSTATE_LOADING);
 923     }
 924
 925     return S_OK;
 926 }
 927
 928 static void on_stop_nsrequest(nsChannelBSC *This, HRESULT result)
 929 {
 930     nsresult nsres, request_result;
 931
 932     switch(result) {
 933     case S_OK:
 934         request_result = NS_OK;
 935         break;
 936     case E_ABORT:
 937         request_result = NS_BINDING_ABORTED;
 938         break;
 939     default:
 940         request_result = NS_ERROR_FAILURE;
 941     }
 942
 943     if(!This->bsc.readed && SUCCEEDED(result)) {
 944         TRACE("No data read! Calling OnStartRequest\n");
 945         on_start_nsrequest(This);
 946     }
 947
 948     if(This->nslistener) {
 949         nsres = nsIStreamListener_OnStopRequest(This->nslistener,
 950                  (nsIRequest*)&This->nschannel->nsIHttpChannel_iface, This->nscontext,
 951                  request_result);
 952         if(NS_FAILED(nsres))
 953             WARN("OnStopRequet failed: %08x\n", nsres);
 954     }
 955
 956     if(This->nschannel->load_group) {
 957         nsres = nsILoadGroup_RemoveRequest(This->nschannel->load_group,
 958                 (nsIRequest*)&This->nschannel->nsIHttpChannel_iface, NULL, request_result);
 959         if(NS_FAILED(nsres))
 960             ERR("RemoveRequest failed: %08x\n", nsres);
 961     }
 962 }
 963
 964 static HRESULT read_stream_data(nsChannelBSC *This, IStream *stream)
 965 {
 966     DWORD read;
 967     nsresult nsres;
 968     HRESULT hres;
 969
 970     if(!This->nslistener) {
 971         BYTE buf[1024];
 972
 973         do {
 974             read = 0;
 975             hres = IStream_Read(stream, buf, sizeof(buf), &read);
 976         }while(hres == S_OK && read);
 977
 978         return S_OK;
 979     }
 980
 981     if(!This->nsstream)
 982         This->nsstream = create_nsprotocol_stream();
 983
 984     do {
 985         read = 0;
 986         hres = IStream_Read(stream, This->nsstream->buf+This->nsstream->buf_size,
 987                 sizeof(This->nsstream->buf)-This->nsstream->buf_size, &read);
 988         if(!read)
 989             break;
 990
 991         This->nsstream->buf_size += read;
 992
 993         if(!This->bsc.readed) {
 994             if(This->nsstream->buf_size >= 2
 995                && (BYTE)This->nsstream->buf[0] == 0xff
 996                && (BYTE)This->nsstream->buf[1] == 0xfe)
 997                 This->nschannel->charset = heap_strdupA(UTF16_STR);
 998
 999             if(!This->nschannel->content_type) {
1000                 WCHAR *mime;
1001
1002                 hres = FindMimeFromData(NULL, NULL, This->nsstream->buf, This->nsstream->buf_size, NULL, 0, &mime, 0);
1003                 if(FAILED(hres))
1004                     return hres;
1005
1006                 TRACE("Found MIME %s\n", debugstr_w(mime));
1007
1008                 This->nschannel->content_type = heap_strdupWtoA(mime);
1009                 CoTaskMemFree(mime);
1010                 if(!This->nschannel->content_type)
1011                     return E_OUTOFMEMORY;
1012             }
1013
1014             on_start_nsrequest(This);
1015         }
1016
1017         This->bsc.readed += This->nsstream->buf_size;
1018
1019         nsres = nsIStreamListener_OnDataAvailable(This->nslistener,
1020                 (nsIRequest*)&This->nschannel->nsIHttpChannel_iface, This->nscontext,
1021                 &This->nsstream->nsIInputStream_iface, This->bsc.readed-This->nsstream->buf_size,
1022                 This->nsstream->buf_size);
1023         if(NS_FAILED(nsres))
1024             ERR("OnDataAvailable failed: %08x\n", nsres);
1025
1026         if(This->nsstream->buf_size == sizeof(This->nsstream->buf)) {
1027             ERR("buffer is full\n");
1028             break;
1029         }
1030     }while(hres == S_OK);
1031
1032     return S_OK;
1033 }
1034
1035 static inline nsChannelBSC *nsChannelBSC_from_BSCallback(BSCallback *iface)
1036 {
1037     return CONTAINING_RECORD(iface, nsChannelBSC, bsc);
1038 }
1039
1040 static void nsChannelBSC_destroy(BSCallback *bsc)
1041 {
1042     nsChannelBSC *This = nsChannelBSC_from_BSCallback(bsc);
1043
1044     if(This->nschannel)
1045         nsIChannel_Release(&This->nschannel->nsIHttpChannel_iface);
1046     if(This->nslistener)
1047         nsIStreamListener_Release(This->nslistener);
1048     if(This->nscontext)
1049         nsISupports_Release(This->nscontext);
1050     if(This->nsstream)
1051         nsIInputStream_Release(&This->nsstream->nsIInputStream_iface);
1052     heap_free(This);
1053 }
1054
1055 static HRESULT nsChannelBSC_start_binding(BSCallback *bsc)
1056 {
1057     nsChannelBSC *This = nsChannelBSC_from_BSCallback(bsc);
1058
1059     if(This->window)
1060         This->window->doc->skip_mutation_notif = FALSE;
1061
1062     return S_OK;
1063 }
1064
1065 static HRESULT nsChannelBSC_init_bindinfo(BSCallback *bsc)
1066 {
1067     nsChannelBSC *This = nsChannelBSC_from_BSCallback(bsc);
1068     HRESULT hres;
1069
1070     if(This->nschannel && This->nschannel->post_data_stream) {
1071         hres = read_post_data_stream(This->nschannel->post_data_stream,
1072                 &This->bsc.post_data, &This->bsc.post_data_len);
1073         if(FAILED(hres))
1074             return hres;
1075
1076         TRACE("post_data = %s\n", debugstr_an(This->bsc.post_data, This->bsc.post_data_len));
1077     }
1078
1079     return S_OK;
1080 }
1081
1082 typedef struct {
1083     task_t header;
1084     nsChannelBSC *bsc;
1085 } stop_request_task_t;
1086
1087 static void stop_request_proc(task_t *_task)
1088 {
1089     stop_request_task_t *task = (stop_request_task_t*)_task;
1090
1091     TRACE("(%p)\n", task->bsc);
1092
1093     list_remove(&task->bsc->bsc.entry);
1094     list_init(&task->bsc->bsc.entry);
1095     on_stop_nsrequest(task->bsc, S_OK);
1096     IBindStatusCallback_Release(&task->bsc->bsc.IBindStatusCallback_iface);
1097 }
1098
1099 static HRESULT async_stop_request(nsChannelBSC *This)
1100 {
1101     stop_request_task_t *task;
1102
1103     task = heap_alloc(sizeof(*task));
1104     if(!task)
1105         return E_OUTOFMEMORY;
1106
1107     IBindStatusCallback_AddRef(&This->bsc.IBindStatusCallback_iface);
1108     task->bsc = This;
1109     push_task(&task->header, stop_request_proc, This->bsc.doc->basedoc.doc_obj->basedoc.task_magic);
1110     return S_OK;
1111 }
1112
1113 static void handle_navigation_error(nsChannelBSC *This, DWORD result)
1114 {
1115     HTMLDocumentObj *doc;
1116     IOleCommandTarget *olecmd;
1117     BOOL is_error_url;
1118     SAFEARRAY *sa;
1119     SAFEARRAYBOUND bound;
1120     VARIANT var, varOut;
1121     LONG ind;
1122     BSTR url, unk;
1123     HRESULT hres;
1124
1125     if(!This->window)
1126         return;
1127
1128     doc = This->window->doc_obj;
1129     if(!doc || !doc->doc_object_service || !doc->client)
1130         return;
1131
1132     hres = IDocObjectService_IsErrorUrl(doc->doc_object_service,
1133             This->window->url, &is_error_url);
1134     if(FAILED(hres) || is_error_url)
1135         return;
1136
1137     hres = IOleClientSite_QueryInterface(doc->client,
1138             &IID_IOleCommandTarget, (void**)&olecmd);
1139     if(FAILED(hres))
1140         return;
1141
1142     bound.lLbound = 0;
1143     bound.cElements = 8;
1144     sa = SafeArrayCreate(VT_VARIANT, 1, &bound);
1145     if(!sa) {
1146         IOleCommandTarget_Release(olecmd);
1147         return;
1148     }
1149
1150     ind = 0;
1151     V_VT(&var) = VT_I4;
1152     V_I4(&var) = result;
1153     SafeArrayPutElement(sa, &ind, &var);
1154
1155     ind = 1;
1156     V_VT(&var) = VT_BSTR;
1157     url = SysAllocString(This->window->url);
1158     V_BSTR(&var) = url;
1159     SafeArrayPutElement(sa, &ind, &var);
1160
1161     ind = 3;
1162     V_VT(&var) = VT_UNKNOWN;
1163     V_UNKNOWN(&var) = (IUnknown*)&This->window->IHTMLWindow2_iface;
1164     SafeArrayPutElement(sa, &ind, &var);
1165
1166     /* FIXME: what are the following fields for? */
1167     ind = 2;
1168     V_VT(&var) = VT_UNKNOWN;
1169     V_UNKNOWN(&var) = NULL;
1170     SafeArrayPutElement(sa, &ind, &var);
1171
1172     ind = 4;
1173     V_VT(&var) = VT_BOOL;
1174     V_BOOL(&var) = FALSE;
1175     SafeArrayPutElement(sa, &ind, &var);
1176
1177     ind = 5;
1178     V_VT(&var) = VT_BOOL;
1179     V_BOOL(&var) = FALSE;
1180     SafeArrayPutElement(sa, &ind, &var);
1181
1182     ind = 6;
1183     V_VT(&var) = VT_BSTR;
1184     unk = SysAllocString(NULL);
1185     V_BSTR(&var) = unk;
1186     SafeArrayPutElement(sa, &ind, &var);
1187
1188     ind = 7;
1189     V_VT(&var) = VT_UNKNOWN;
1190     V_UNKNOWN(&var) = NULL;
1191     SafeArrayPutElement(sa, &ind, &var);
1192
1193     V_VT(&var) = VT_ARRAY;
1194     V_ARRAY(&var) = sa;
1195     V_VT(&varOut) = VT_BOOL;
1196     V_BOOL(&varOut) = VARIANT_TRUE;
1197     IOleCommandTarget_Exec(olecmd, &CGID_DocHostCmdPriv, 1, 0, &var, FAILED(hres)?NULL:&varOut);
1198
1199     SysFreeString(url);
1200     SysFreeString(unk);
1201     SafeArrayDestroy(sa);
1202     IOleCommandTarget_Release(olecmd);
1203 }
1204
1205 static HRESULT nsChannelBSC_stop_binding(BSCallback *bsc, HRESULT result)
1206 {
1207     nsChannelBSC *This = nsChannelBSC_from_BSCallback(bsc);
1208
1209     if(FAILED(result))
1210         handle_navigation_error(This, result);
1211     else if(This->window) {
1212         result = async_stop_request(This);
1213         if(SUCCEEDED(result))
1214            return S_OK;
1215     }
1216
1217     on_stop_nsrequest(This, result);
1218     return S_OK;
1219 }
1220
1221 static HRESULT nsChannelBSC_read_data(BSCallback *bsc, IStream *stream)
1222 {
1223     nsChannelBSC *This = nsChannelBSC_from_BSCallback(bsc);
1224
1225     return read_stream_data(This, stream);
1226 }
1227
1228 static HRESULT nsChannelBSC_on_progress(BSCallback *bsc, ULONG status_code, LPCWSTR status_text)
1229 {
1230     nsChannelBSC *This = nsChannelBSC_from_BSCallback(bsc);
1231
1232     switch(status_code) {
1233     case BINDSTATUS_MIMETYPEAVAILABLE:
1234         if(!This->nschannel)
1235             return S_OK;
1236
1237         heap_free(This->nschannel->content_type);
1238         This->nschannel->content_type = heap_strdupWtoA(status_text);
1239         break;
1240     case BINDSTATUS_REDIRECTING:
1241         TRACE("redirect to %s\n", debugstr_w(status_text));
1242
1243         /* FIXME: We should find a better way to handle this */
1244         set_wine_url(This->nschannel->uri, status_text);
1245         break;
1246     case BINDSTATUS_BEGINDOWNLOADDATA: {
1247         IWinInetHttpInfo *http_info;
1248         DWORD status, size = sizeof(DWORD);
1249         HRESULT hres;
1250
1251         if(!This->bsc.binding)
1252             break;
1253
1254         hres = IBinding_QueryInterface(This->bsc.binding, &IID_IWinInetHttpInfo, (void**)&http_info);
1255         if(FAILED(hres))
1256             break;
1257
1258         hres = IWinInetHttpInfo_QueryInfo(http_info,
1259                 HTTP_QUERY_STATUS_CODE|HTTP_QUERY_FLAG_NUMBER, &status, &size, NULL, NULL);
1260         IWinInetHttpInfo_Release(http_info);
1261         if(FAILED(hres) || status == HTTP_STATUS_OK)
1262             break;
1263
1264         handle_navigation_error(This, status);
1265     }
1266     }
1267
1268     return S_OK;
1269 }
1270
1271 static HRESULT nsChannelBSC_on_response(BSCallback *bsc, DWORD response_code,
1272         LPCWSTR response_headers)
1273 {
1274     nsChannelBSC *This = nsChannelBSC_from_BSCallback(bsc);
1275     HRESULT hres;
1276
1277     This->nschannel->response_status = response_code;
1278
1279     if(response_headers) {
1280         const WCHAR *headers;
1281
1282         headers = strchrW(response_headers, '\r');
1283         if(headers && headers[1] == '\n') {
1284             headers += 2;
1285             hres = parse_headers(headers, &This->nschannel->response_headers);
1286             if(FAILED(hres)) {
1287                 WARN("parsing headers failed: %08x\n", hres);
1288                 return hres;
1289             }
1290         }
1291     }
1292
1293     return S_OK;
1294 }
1295
1296 static HRESULT nsChannelBSC_beginning_transaction(BSCallback *bsc, WCHAR **additional_headers)
1297 {
1298     nsChannelBSC *This = nsChannelBSC_from_BSCallback(bsc);
1299     http_header_t *iter;
1300     DWORD len = 0;
1301     WCHAR *ptr;
1302
1303     static const WCHAR content_lengthW[] =
1304         {'C','o','n','t','e','n','t','-','L','e','n','g','t','h',0};
1305
1306     if(!This->nschannel)
1307         return S_FALSE;
1308
1309     LIST_FOR_EACH_ENTRY(iter, &This->nschannel->request_headers, http_header_t, entry) {
1310         if(strcmpW(iter->header, content_lengthW))
1311             len += strlenW(iter->header) + 2 /* ": " */ + strlenW(iter->data) + 2 /* "\r\n" */;
1312     }
1313
1314     if(!len)
1315         return S_OK;
1316
1317     *additional_headers = ptr = CoTaskMemAlloc((len+1)*sizeof(WCHAR));
1318     if(!ptr)
1319         return E_OUTOFMEMORY;
1320
1321     LIST_FOR_EACH_ENTRY(iter, &This->nschannel->request_headers, http_header_t, entry) {
1322         if(!strcmpW(iter->header, content_lengthW))
1323             continue;
1324
1325         len = strlenW(iter->header);
1326         memcpy(ptr, iter->header, len*sizeof(WCHAR));
1327         ptr += len;
1328
1329         *ptr++ = ':';
1330         *ptr++ = ' ';
1331
1332         len = strlenW(iter->data);
1333         memcpy(ptr, iter->data, len*sizeof(WCHAR));
1334         ptr += len;
1335
1336         *ptr++ = '\r';
1337         *ptr++ = '\n';
1338     }
1339
1340     *ptr = 0;
1341
1342     return S_OK;
1343 }
1344
1345 static const BSCallbackVtbl nsChannelBSCVtbl = {
1346     nsChannelBSC_destroy,
1347     nsChannelBSC_init_bindinfo,
1348     nsChannelBSC_start_binding,
1349     nsChannelBSC_stop_binding,
1350     nsChannelBSC_read_data,
1351     nsChannelBSC_on_progress,
1352     nsChannelBSC_on_response,
1353     nsChannelBSC_beginning_transaction
1354 };
1355
1356 HRESULT create_channelbsc(IMoniker *mon, WCHAR *headers, BYTE *post_data, DWORD post_data_size, nsChannelBSC **retval)
1357 {
1358     nsChannelBSC *ret;
1359
1360     ret = heap_alloc_zero(sizeof(*ret));
1361     if(!ret)
1362         return E_OUTOFMEMORY;
1363
1364     init_bscallback(&ret->bsc, &nsChannelBSCVtbl, mon, BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA);
1365
1366     if(headers) {
1367         ret->bsc.headers = heap_strdupW(headers);
1368         if(!ret->bsc.headers) {
1369             IBindStatusCallback_Release(&ret->bsc.IBindStatusCallback_iface);
1370             return E_OUTOFMEMORY;
1371         }
1372     }
1373
1374     if(post_data) {
1375         ret->bsc.post_data = GlobalAlloc(0, post_data_size);
1376         if(!ret->bsc.post_data) {
1377             heap_free(ret->bsc.headers);
1378             IBindStatusCallback_Release(&ret->bsc.IBindStatusCallback_iface);
1379             return E_OUTOFMEMORY;
1380         }
1381
1382         memcpy(ret->bsc.post_data, post_data, post_data_size);
1383         ret->bsc.post_data_len = post_data_size;
1384     }
1385
1386     *retval = ret;
1387     return S_OK;
1388 }
1389
1390 IMoniker *get_channelbsc_mon(nsChannelBSC *This)
1391 {
1392     if(This->bsc.mon)
1393         IMoniker_AddRef(This->bsc.mon);
1394     return This->bsc.mon;
1395 }
1396
1397 void set_window_bscallback(HTMLWindow *window, nsChannelBSC *callback)
1398 {
1399     if(window->bscallback) {
1400         if(window->bscallback->bsc.binding)
1401             IBinding_Abort(window->bscallback->bsc.binding);
1402         window->bscallback->bsc.doc = NULL;
1403         window->bscallback->window = NULL;
1404         IBindStatusCallback_Release(&window->bscallback->bsc.IBindStatusCallback_iface);
1405     }
1406
1407     window->bscallback = callback;
1408
1409     if(callback) {
1410         callback->window = window;
1411         IBindStatusCallback_AddRef(&callback->bsc.IBindStatusCallback_iface);
1412         callback->bsc.doc = window->doc;
1413     }
1414 }
1415
1416 typedef struct {
1417     task_t header;
1418     HTMLWindow *window;
1419     nsChannelBSC *bscallback;
1420 } start_doc_binding_task_t;
1421
1422 static void start_doc_binding_proc(task_t *_task)
1423 {
1424     start_doc_binding_task_t *task = (start_doc_binding_task_t*)_task;
1425
1426     start_binding(task->window, NULL, (BSCallback*)task->bscallback, NULL);
1427     IBindStatusCallback_Release(&task->bscallback->bsc.IBindStatusCallback_iface);
1428 }
1429
1430 HRESULT async_start_doc_binding(HTMLWindow *window, nsChannelBSC *bscallback)
1431 {
1432     start_doc_binding_task_t *task;
1433
1434     task = heap_alloc(sizeof(start_doc_binding_task_t));
1435     if(!task)
1436         return E_OUTOFMEMORY;
1437
1438     task->window = window;
1439     task->bscallback = bscallback;
1440     IBindStatusCallback_AddRef(&bscallback->bsc.IBindStatusCallback_iface);
1441
1442     push_task(&task->header, start_doc_binding_proc, window->task_magic);
1443     return S_OK;
1444 }
1445
1446 void abort_document_bindings(HTMLDocumentNode *doc)
1447 {
1448     BSCallback *iter, *next;
1449
1450     LIST_FOR_EACH_ENTRY_SAFE(iter, next, &doc->bindings, BSCallback, entry) {
1451         if(iter->doc)
1452             remove_target_tasks(iter->doc->basedoc.task_magic);
1453
1454         if(iter->binding)
1455             IBinding_Abort(iter->binding);
1456         else {
1457             list_remove(&iter->entry);
1458             list_init(&iter->entry);
1459             iter->vtbl->stop_binding(iter, S_OK);
1460         }
1461
1462         iter->doc = NULL;
1463     }
1464 }
1465
1466 HRESULT channelbsc_load_stream(nsChannelBSC *bscallback, IStream *stream)
1467 {
1468     HRESULT hres = S_OK;
1469
1470     if(!bscallback->nschannel) {
1471         ERR("NULL nschannel\n");
1472         return E_FAIL;
1473     }
1474
1475     bscallback->nschannel->content_type = heap_strdupA("text/html");
1476     if(!bscallback->nschannel->content_type)
1477         return E_OUTOFMEMORY;
1478
1479     list_add_head(&bscallback->bsc.doc->bindings, &bscallback->bsc.entry);
1480     if(stream)
1481         hres = read_stream_data(bscallback, stream);
1482     if(SUCCEEDED(hres))
1483         hres = async_stop_request(bscallback);
1484     if(FAILED(hres))
1485         IBindStatusCallback_OnStopBinding(&bscallback->bsc.IBindStatusCallback_iface, hres,
1486                 ERROR_SUCCESS);
1487
1488     return hres;
1489 }
1490
1491 void channelbsc_set_channel(nsChannelBSC *This, nsChannel *channel, nsIStreamListener *listener, nsISupports *context)
1492 {
1493     nsIChannel_AddRef(&channel->nsIHttpChannel_iface);
1494     This->nschannel = channel;
1495
1496     nsIStreamListener_AddRef(listener);
1497     This->nslistener = listener;
1498
1499     if(context) {
1500         nsISupports_AddRef(context);
1501         This->nscontext = context;
1502     }
1503
1504     if(This->bsc.headers) {
1505         HRESULT hres;
1506
1507         hres = parse_headers(This->bsc.headers, &channel->request_headers);
1508         heap_free(This->bsc.headers);
1509         This->bsc.headers = NULL;
1510         if(FAILED(hres))
1511             WARN("parse_headers failed: %08x\n", hres);
1512     }
1513 }
1514
1515 HRESULT hlink_frame_navigate(HTMLDocument *doc, LPCWSTR url,
1516         nsIInputStream *post_data_stream, DWORD hlnf, BOOL *cancel)
1517 {
1518     IHlinkFrame *hlink_frame;
1519     nsChannelBSC *callback;
1520     IServiceProvider *sp;
1521     IBindCtx *bindctx;
1522     IMoniker *mon;
1523     IHlink *hlink;
1524     HRESULT hres;
1525
1526     *cancel = FALSE;
1527
1528     hres = IOleClientSite_QueryInterface(doc->doc_obj->client, &IID_IServiceProvider,
1529             (void**)&sp);
1530     if(FAILED(hres))
1531         return S_OK;
1532
1533     hres = IServiceProvider_QueryService(sp, &IID_IHlinkFrame, &IID_IHlinkFrame,
1534             (void**)&hlink_frame);
1535     IServiceProvider_Release(sp);
1536     if(FAILED(hres))
1537         return S_OK;
1538
1539     hres = create_channelbsc(NULL, NULL, NULL, 0, &callback);
1540     if(FAILED(hres)) {
1541         IHlinkFrame_Release(hlink_frame);
1542         return hres;
1543     }
1544
1545     if(post_data_stream) {
1546         read_post_data_stream(post_data_stream, &callback->bsc.post_data, &callback->bsc.post_data_len);
1547         TRACE("post_data = %s\n", debugstr_an(callback->bsc.post_data, callback->bsc.post_data_len));
1548     }
1549
1550     hres = CreateAsyncBindCtx(0, &callback->bsc.IBindStatusCallback_iface, NULL, &bindctx);
1551     if(SUCCEEDED(hres))
1552         hres = CoCreateInstance(&CLSID_StdHlink, NULL, CLSCTX_INPROC_SERVER,
1553                 &IID_IHlink, (LPVOID*)&hlink);
1554
1555     if(SUCCEEDED(hres))
1556         hres = CreateURLMoniker(NULL, url, &mon);
1557
1558     if(SUCCEEDED(hres)) {
1559         IHlink_SetMonikerReference(hlink, HLINKSETF_TARGET, mon, NULL);
1560
1561         if(hlnf & HLNF_OPENINNEWWINDOW) {
1562             static const WCHAR wszBlank[] = {'_','b','l','a','n','k',0};
1563             IHlink_SetTargetFrameName(hlink, wszBlank); /* FIXME */
1564         }
1565
1566         hres = IHlinkFrame_Navigate(hlink_frame, hlnf, bindctx,
1567                 &callback->bsc.IBindStatusCallback_iface, hlink);
1568         IMoniker_Release(mon);
1569         *cancel = hres == S_OK;
1570         hres = S_OK;
1571     }
1572
1573     IHlinkFrame_Release(hlink_frame);
1574     IBindCtx_Release(bindctx);
1575     IBindStatusCallback_Release(&callback->bsc.IBindStatusCallback_iface);
1576     return hres;
1577 }
1578
1579 HRESULT navigate_url(HTMLWindow *window, const WCHAR *new_url, const WCHAR *base_url)
1580 {
1581     WCHAR url[INTERNET_MAX_URL_LENGTH];
1582     nsWineURI *uri;
1583     HRESULT hres;
1584
1585     if(!new_url) {
1586         *url = 0;
1587     }else if(base_url) {
1588         DWORD len = 0;
1589
1590         hres = CoInternetCombineUrl(base_url, new_url, URL_ESCAPE_SPACES_ONLY|URL_DONT_ESCAPE_EXTRA_INFO,
1591                 url, sizeof(url)/sizeof(WCHAR), &len, 0);
1592         if(FAILED(hres))
1593             return hres;
1594     }else {
1595         strcpyW(url, new_url);
1596     }
1597
1598     if(window->doc_obj && window->doc_obj->hostui) {
1599         OLECHAR *translated_url = NULL;
1600
1601         hres = IDocHostUIHandler_TranslateUrl(window->doc_obj->hostui, 0, url,
1602                 &translated_url);
1603         if(hres == S_OK) {
1604             TRACE("%08x %s -> %s\n", hres, debugstr_w(url), debugstr_w(translated_url));
1605             strcpyW(url, translated_url);
1606             CoTaskMemFree(translated_url);
1607         }
1608     }
1609
1610     if(window->doc_obj && window == window->doc_obj->basedoc.window) {
1611         BOOL cancel;
1612
1613         hres = hlink_frame_navigate(&window->doc->basedoc, url, NULL, 0, &cancel);
1614         if(FAILED(hres))
1615             return hres;
1616
1617         if(cancel) {
1618             TRACE("Navigation handled by hlink frame\n");
1619             return S_OK;
1620         }
1621     }
1622
1623     hres = create_doc_uri(window, url, &uri);
1624     if(FAILED(hres))
1625         return hres;
1626
1627     hres = load_nsuri(window, uri, NULL, LOAD_FLAGS_NONE);
1628     nsISupports_Release((nsISupports*)uri);
1629     return hres;
1630 }