git.oblomov.eu Git - wine/blob - dlls/urlmon/binding.c

   1 /*
   2  * Copyright 2005-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 "urlmon_main.h"
  20 #include "winreg.h"
  21 #include "shlwapi.h"
  22
  23 #include "wine/debug.h"
  24
  25 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
  26
  27 static WCHAR cbinding_contextW[] = {'C','B','i','n','d','i','n','g',' ','C','o','n','t','e','x','t',0};
  28 static WCHAR bscb_holderW[] = { '_','B','S','C','B','_','H','o','l','d','e','r','_',0 };
  29
  30 typedef struct {
  31     IUnknown IUnknown_iface;
  32
  33     LONG ref;
  34
  35     IInternetProtocolEx *protocol;
  36
  37     HANDLE file;
  38     HRESULT hres;
  39
  40     LPWSTR cache_file;
  41 } stgmed_buf_t;
  42
  43 typedef struct _stgmed_obj_t stgmed_obj_t;
  44
  45 typedef struct {
  46     void (*release)(stgmed_obj_t*);
  47     HRESULT (*fill_stgmed)(stgmed_obj_t*,STGMEDIUM*);
  48     HRESULT (*get_result)(stgmed_obj_t*,DWORD,void**);
  49 } stgmed_obj_vtbl;
  50
  51 struct _stgmed_obj_t {
  52     const stgmed_obj_vtbl *vtbl;
  53 };
  54
  55 typedef enum {
  56     BEFORE_DOWNLOAD,
  57     DOWNLOADING,
  58     END_DOWNLOAD
  59 } download_state_t;
  60
  61 #define BINDING_LOCKED    0x0001
  62 #define BINDING_STOPPED   0x0002
  63 #define BINDING_OBJAVAIL  0x0004
  64 #define BINDING_ABORTED   0x0008
  65
  66 typedef struct {
  67     IBinding              IBinding_iface;
  68     IInternetProtocolSink IInternetProtocolSink_iface;
  69     IInternetBindInfo     IInternetBindInfo_iface;
  70     IWinInetHttpInfo      IWinInetHttpInfo_iface;
  71     IServiceProvider      IServiceProvider_iface;
  72
  73     LONG ref;
  74
  75     IBindStatusCallback *callback;
  76     IServiceProvider *service_provider;
  77
  78     BindProtocol *protocol;
  79
  80     stgmed_buf_t *stgmed_buf;
  81     stgmed_obj_t *stgmed_obj;
  82
  83     BINDINFO bindinfo;
  84     DWORD bindf;
  85     BOOL to_object;
  86     LPWSTR mime;
  87     UINT clipboard_format;
  88     LPWSTR url;
  89     LPWSTR redirect_url;
  90     IID iid;
  91     BOOL report_mime;
  92     BOOL use_cache_file;
  93     DWORD state;
  94     HRESULT hres;
  95     download_state_t download_state;
  96     IUnknown *obj;
  97     IMoniker *mon;
  98     IBindCtx *bctx;
  99     HWND notif_hwnd;
 100
 101     CRITICAL_SECTION section;
 102 } Binding;
 103
 104 static void read_protocol_data(stgmed_buf_t *stgmed_buf)
 105 {
 106     BYTE buf[8192];
 107     DWORD read;
 108     HRESULT hres;
 109
 110     do hres = IInternetProtocol_Read(stgmed_buf->protocol, buf, sizeof(buf), &read);
 111     while(hres == S_OK);
 112 }
 113
 114 static void dump_BINDINFO(BINDINFO *bi)
 115 {
 116     static const char * const BINDINFOF_str[] = {
 117         "#0",
 118         "BINDINFOF_URLENCODESTGMEDDATA",
 119         "BINDINFOF_URLENCODEDEXTRAINFO"
 120     };
 121
 122     static const char * const BINDVERB_str[] = {
 123         "BINDVERB_GET",
 124         "BINDVERB_POST",
 125         "BINDVERB_PUT",
 126         "BINDVERB_CUSTOM"
 127     };
 128
 129     TRACE("\n"
 130             "BINDINFO = {\n"
 131             "    %d, %s,\n"
 132             "    {%d, %p, %p},\n"
 133             "    %s,\n"
 134             "    %s,\n"
 135             "    %s,\n"
 136             "    %d, %08x, %d, %d\n"
 137             "    {%d %p %x},\n"
 138             "    %s\n"
 139             "    %p, %d\n"
 140             "}\n",
 141
 142             bi->cbSize, debugstr_w(bi->szExtraInfo),
 143             bi->stgmedData.tymed, bi->stgmedData.u.hGlobal, bi->stgmedData.pUnkForRelease,
 144             bi->grfBindInfoF > BINDINFOF_URLENCODEDEXTRAINFO
 145                 ? "unknown" : BINDINFOF_str[bi->grfBindInfoF],
 146             bi->dwBindVerb > BINDVERB_CUSTOM
 147                 ? "unknown" : BINDVERB_str[bi->dwBindVerb],
 148             debugstr_w(bi->szCustomVerb),
 149             bi->cbstgmedData, bi->dwOptions, bi->dwOptionsFlags, bi->dwCodePage,
 150             bi->securityAttributes.nLength,
 151             bi->securityAttributes.lpSecurityDescriptor,
 152             bi->securityAttributes.bInheritHandle,
 153             debugstr_guid(&bi->iid),
 154             bi->pUnk, bi->dwReserved
 155             );
 156 }
 157
 158 static void mime_available(Binding *This, LPCWSTR mime)
 159 {
 160     heap_free(This->mime);
 161     This->mime = heap_strdupW(mime);
 162
 163     if(!This->mime || !This->report_mime)
 164         return;
 165
 166     IBindStatusCallback_OnProgress(This->callback, 0, 0, BINDSTATUS_MIMETYPEAVAILABLE, This->mime);
 167
 168     This->clipboard_format = RegisterClipboardFormatW(This->mime);
 169 }
 170
 171 static void stop_binding(Binding *binding, HRESULT hres, LPCWSTR str)
 172 {
 173     if(binding->state & BINDING_LOCKED) {
 174         IInternetProtocolEx_UnlockRequest(&binding->protocol->IInternetProtocolEx_iface);
 175         binding->state &= ~BINDING_LOCKED;
 176     }
 177
 178     if(!(binding->state & BINDING_STOPPED)) {
 179         binding->state |= BINDING_STOPPED;
 180
 181         IBindStatusCallback_OnStopBinding(binding->callback, hres, str);
 182         binding->hres = hres;
 183     }
 184 }
 185
 186 static LPWSTR get_mime_clsid(LPCWSTR mime, CLSID *clsid)
 187 {
 188     LPWSTR key_name, ret;
 189     DWORD res, type, size;
 190     HKEY hkey;
 191     int len;
 192     HRESULT hres;
 193
 194     static const WCHAR mime_keyW[] =
 195         {'M','I','M','E','\\','D','a','t','a','b','a','s','e','\\',
 196          'C','o','n','t','e','n','t',' ','T','y','p','e','\\'};
 197     static const WCHAR clsidW[] = {'C','L','S','I','D',0};
 198
 199     len = strlenW(mime)+1;
 200     key_name = heap_alloc(sizeof(mime_keyW) + len*sizeof(WCHAR));
 201     memcpy(key_name, mime_keyW, sizeof(mime_keyW));
 202     strcpyW(key_name + sizeof(mime_keyW)/sizeof(WCHAR), mime);
 203
 204     res = RegOpenKeyW(HKEY_CLASSES_ROOT, key_name, &hkey);
 205     heap_free(key_name);
 206     if(res != ERROR_SUCCESS) {
 207         WARN("Could not open MIME key: %x\n", res);
 208         return NULL;
 209     }
 210
 211     size = 50*sizeof(WCHAR);
 212     ret = heap_alloc(size);
 213     res = RegQueryValueExW(hkey, clsidW, NULL, &type, (LPBYTE)ret, &size);
 214     RegCloseKey(hkey);
 215     if(res != ERROR_SUCCESS) {
 216         WARN("Could not get CLSID: %08x\n", res);
 217         heap_free(ret);
 218         return NULL;
 219     }
 220
 221     hres = CLSIDFromString(ret, clsid);
 222     if(FAILED(hres)) {
 223         WARN("Could not parse CLSID: %08x\n", hres);
 224         heap_free(ret);
 225         return NULL;
 226     }
 227
 228     return ret;
 229 }
 230
 231 static void load_doc_mon(Binding *binding, IPersistMoniker *persist)
 232 {
 233     IBindCtx *bctx;
 234     HRESULT hres;
 235
 236     hres = CreateAsyncBindCtxEx(binding->bctx, 0, NULL, NULL, &bctx, 0);
 237     if(FAILED(hres)) {
 238         WARN("CreateAsyncBindCtxEx failed: %08x\n", hres);
 239         return;
 240     }
 241
 242     IBindCtx_RevokeObjectParam(bctx, bscb_holderW);
 243     IBindCtx_RegisterObjectParam(bctx, cbinding_contextW, (IUnknown*)&binding->IBinding_iface);
 244
 245     hres = IPersistMoniker_Load(persist, binding->download_state == END_DOWNLOAD, binding->mon, bctx, 0x12);
 246     IBindCtx_RevokeObjectParam(bctx, cbinding_contextW);
 247     IBindCtx_Release(bctx);
 248     if(FAILED(hres))
 249         FIXME("Load failed: %08x\n", hres);
 250 }
 251
 252 static HRESULT create_mime_object(Binding *binding, const CLSID *clsid, LPCWSTR clsid_str)
 253 {
 254     IPersistMoniker *persist;
 255     HRESULT hres;
 256
 257     hres = CoCreateInstance(clsid, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 258                             &binding->iid, (void**)&binding->obj);
 259     if(FAILED(hres)) {
 260         WARN("CoCreateInstance failed: %08x\n", hres);
 261         return INET_E_CANNOT_INSTANTIATE_OBJECT;
 262     }
 263
 264     binding->state |= BINDING_OBJAVAIL;
 265
 266     hres = IUnknown_QueryInterface(binding->obj, &IID_IPersistMoniker, (void**)&persist);
 267     if(SUCCEEDED(hres)) {
 268         IMonikerProp *prop;
 269
 270         hres = IPersistMoniker_QueryInterface(persist, &IID_IMonikerProp, (void**)&prop);
 271         if(SUCCEEDED(hres)) {
 272             IMonikerProp_PutProperty(prop, MIMETYPEPROP, binding->mime);
 273             IMonikerProp_PutProperty(prop, CLASSIDPROP, clsid_str);
 274             IMonikerProp_Release(prop);
 275         }
 276
 277         load_doc_mon(binding, persist);
 278
 279         IPersistMoniker_Release(persist);
 280     }else {
 281         FIXME("Could not get IPersistMoniker: %08x\n", hres);
 282         /* FIXME: Try query IPersistFile */
 283     }
 284
 285     IBindStatusCallback_OnObjectAvailable(binding->callback, &binding->iid, binding->obj);
 286
 287     return S_OK;
 288 }
 289
 290 static void create_object(Binding *binding)
 291 {
 292     LPWSTR clsid_str;
 293     CLSID clsid;
 294     HRESULT hres;
 295
 296     if(!binding->mime) {
 297         FIXME("MIME not available\n");
 298         return;
 299     }
 300
 301     if(!(clsid_str = get_mime_clsid(binding->mime, &clsid))) {
 302         FIXME("Could not find object for MIME %s\n", debugstr_w(binding->mime));
 303         return;
 304     }
 305
 306     IBindStatusCallback_OnProgress(binding->callback, 0, 0, BINDSTATUS_CLASSIDAVAILABLE, clsid_str);
 307     IBindStatusCallback_OnProgress(binding->callback, 0, 0, BINDSTATUS_BEGINSYNCOPERATION, NULL);
 308
 309     hres = create_mime_object(binding, &clsid, clsid_str);
 310     heap_free(clsid_str);
 311
 312     IBindStatusCallback_OnProgress(binding->callback, 0, 0, BINDSTATUS_ENDSYNCOPERATION, NULL);
 313
 314     stop_binding(binding, hres, NULL);
 315     if(FAILED(hres))
 316         IInternetProtocolEx_Terminate(&binding->protocol->IInternetProtocolEx_iface, 0);
 317 }
 318
 319 static void cache_file_available(Binding *This, const WCHAR *file_name)
 320 {
 321     heap_free(This->stgmed_buf->cache_file);
 322     This->stgmed_buf->cache_file = heap_strdupW(file_name);
 323
 324     if(This->use_cache_file) {
 325         This->stgmed_buf->file = CreateFileW(file_name, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL,
 326                 OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
 327         if(This->stgmed_buf->file == INVALID_HANDLE_VALUE)
 328             WARN("CreateFile failed: %u\n", GetLastError());
 329     }
 330 }
 331
 332 static inline stgmed_buf_t *impl_from_IUnknown(IUnknown *iface)
 333 {
 334     return CONTAINING_RECORD(iface, stgmed_buf_t, IUnknown_iface);
 335 }
 336
 337 static HRESULT WINAPI StgMedUnk_QueryInterface(IUnknown *iface, REFIID riid, void **ppv)
 338 {
 339     stgmed_buf_t *This = impl_from_IUnknown(iface);
 340
 341     *ppv = NULL;
 342
 343     if(IsEqualGUID(riid, &IID_IUnknown)) {
 344         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
 345
 346         *ppv = &This->IUnknown_iface;
 347         IUnknown_AddRef(&This->IUnknown_iface);
 348         return S_OK;
 349     }
 350
 351     TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
 352     return E_NOINTERFACE;
 353 }
 354
 355 static ULONG WINAPI StgMedUnk_AddRef(IUnknown *iface)
 356 {
 357     stgmed_buf_t *This = impl_from_IUnknown(iface);
 358     LONG ref = InterlockedIncrement(&This->ref);
 359
 360     TRACE("(%p) ref=%d\n", This, ref);
 361
 362     return ref;
 363 }
 364
 365 static ULONG WINAPI StgMedUnk_Release(IUnknown *iface)
 366 {
 367     stgmed_buf_t *This = impl_from_IUnknown(iface);
 368     LONG ref = InterlockedDecrement(&This->ref);
 369
 370     TRACE("(%p) ref=%d\n", This, ref);
 371
 372     if(!ref) {
 373         if(This->file != INVALID_HANDLE_VALUE)
 374             CloseHandle(This->file);
 375         IInternetProtocol_Release(This->protocol);
 376         heap_free(This->cache_file);
 377         heap_free(This);
 378
 379         URLMON_UnlockModule();
 380     }
 381
 382     return ref;
 383 }
 384
 385 static const IUnknownVtbl StgMedUnkVtbl = {
 386     StgMedUnk_QueryInterface,
 387     StgMedUnk_AddRef,
 388     StgMedUnk_Release
 389 };
 390
 391 static stgmed_buf_t *create_stgmed_buf(IInternetProtocolEx *protocol)
 392 {
 393     stgmed_buf_t *ret = heap_alloc(sizeof(*ret));
 394
 395     ret->IUnknown_iface.lpVtbl = &StgMedUnkVtbl;
 396     ret->ref = 1;
 397     ret->file = INVALID_HANDLE_VALUE;
 398     ret->hres = S_OK;
 399     ret->cache_file = NULL;
 400
 401     IInternetProtocol_AddRef(protocol);
 402     ret->protocol = protocol;
 403
 404     URLMON_LockModule();
 405
 406     return ret;
 407 }
 408
 409 typedef struct {
 410     stgmed_obj_t stgmed_obj;
 411     IStream IStream_iface;
 412
 413     LONG ref;
 414
 415     stgmed_buf_t *buf;
 416 } ProtocolStream;
 417
 418 static inline ProtocolStream *impl_from_IStream(IStream *iface)
 419 {
 420     return CONTAINING_RECORD(iface, ProtocolStream, IStream_iface);
 421 }
 422
 423 static HRESULT WINAPI ProtocolStream_QueryInterface(IStream *iface,
 424                                                           REFIID riid, void **ppv)
 425 {
 426     ProtocolStream *This = impl_from_IStream(iface);
 427
 428     *ppv = NULL;
 429
 430     if(IsEqualGUID(&IID_IUnknown, riid)) {
 431         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
 432         *ppv = &This->IStream_iface;
 433     }else if(IsEqualGUID(&IID_ISequentialStream, riid)) {
 434         TRACE("(%p)->(IID_ISequentialStream %p)\n", This, ppv);
 435         *ppv = &This->IStream_iface;
 436     }else if(IsEqualGUID(&IID_IStream, riid)) {
 437         TRACE("(%p)->(IID_IStream %p)\n", This, ppv);
 438         *ppv = &This->IStream_iface;
 439     }
 440
 441     if(*ppv) {
 442         IStream_AddRef(&This->IStream_iface);
 443         return S_OK;
 444     }
 445
 446     WARN("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
 447     return E_NOINTERFACE;
 448 }
 449
 450 static ULONG WINAPI ProtocolStream_AddRef(IStream *iface)
 451 {
 452     ProtocolStream *This = impl_from_IStream(iface);
 453     LONG ref = InterlockedIncrement(&This->ref);
 454
 455     TRACE("(%p) ref=%d\n", This, ref);
 456
 457     return ref;
 458 }
 459
 460 static ULONG WINAPI ProtocolStream_Release(IStream *iface)
 461 {
 462     ProtocolStream *This = impl_from_IStream(iface);
 463     LONG ref = InterlockedDecrement(&This->ref);
 464
 465     TRACE("(%p) ref=%d\n", This, ref);
 466
 467     if(!ref) {
 468         IUnknown_Release(&This->buf->IUnknown_iface);
 469         heap_free(This);
 470
 471         URLMON_UnlockModule();
 472     }
 473
 474     return ref;
 475 }
 476
 477 static HRESULT WINAPI ProtocolStream_Read(IStream *iface, void *pv,
 478                                           ULONG cb, ULONG *pcbRead)
 479 {
 480     ProtocolStream *This = impl_from_IStream(iface);
 481     DWORD read = 0;
 482     HRESULT hres;
 483
 484     TRACE("(%p)->(%p %d %p)\n", This, pv, cb, pcbRead);
 485
 486     if(This->buf->file == INVALID_HANDLE_VALUE) {
 487         hres = This->buf->hres = IInternetProtocol_Read(This->buf->protocol, (PBYTE)pv, cb, &read);
 488     }else {
 489         hres = ReadFile(This->buf->file, pv, cb, &read, NULL) ? S_OK : INET_E_DOWNLOAD_FAILURE;
 490     }
 491
 492     if (pcbRead)
 493         *pcbRead = read;
 494
 495     if(hres == E_PENDING)
 496         return E_PENDING;
 497     else if(FAILED(hres))
 498         FIXME("Read failed: %08x\n", hres);
 499
 500     return read ? S_OK : S_FALSE;
 501 }
 502
 503 static HRESULT WINAPI ProtocolStream_Write(IStream *iface, const void *pv,
 504                                           ULONG cb, ULONG *pcbWritten)
 505 {
 506     ProtocolStream *This = impl_from_IStream(iface);
 507
 508     TRACE("(%p)->(%p %d %p)\n", This, pv, cb, pcbWritten);
 509
 510     return STG_E_ACCESSDENIED;
 511 }
 512
 513 static HRESULT WINAPI ProtocolStream_Seek(IStream *iface, LARGE_INTEGER dlibMove,
 514                                          DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
 515 {
 516     ProtocolStream *This = impl_from_IStream(iface);
 517     FIXME("(%p)->(%d %08x %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
 518     return E_NOTIMPL;
 519 }
 520
 521 static HRESULT WINAPI ProtocolStream_SetSize(IStream *iface, ULARGE_INTEGER libNewSize)
 522 {
 523     ProtocolStream *This = impl_from_IStream(iface);
 524     FIXME("(%p)->(%d)\n", This, libNewSize.u.LowPart);
 525     return E_NOTIMPL;
 526 }
 527
 528 static HRESULT WINAPI ProtocolStream_CopyTo(IStream *iface, IStream *pstm,
 529         ULARGE_INTEGER cb, ULARGE_INTEGER *pcbRead, ULARGE_INTEGER *pcbWritten)
 530 {
 531     ProtocolStream *This = impl_from_IStream(iface);
 532     FIXME("(%p)->(%p %d %p %p)\n", This, pstm, cb.u.LowPart, pcbRead, pcbWritten);
 533     return E_NOTIMPL;
 534 }
 535
 536 static HRESULT WINAPI ProtocolStream_Commit(IStream *iface, DWORD grfCommitFlags)
 537 {
 538     ProtocolStream *This = impl_from_IStream(iface);
 539
 540     TRACE("(%p)->(%08x)\n", This, grfCommitFlags);
 541
 542     return E_NOTIMPL;
 543 }
 544
 545 static HRESULT WINAPI ProtocolStream_Revert(IStream *iface)
 546 {
 547     ProtocolStream *This = impl_from_IStream(iface);
 548
 549     TRACE("(%p)\n", This);
 550
 551     return E_NOTIMPL;
 552 }
 553
 554 static HRESULT WINAPI ProtocolStream_LockRegion(IStream *iface, ULARGE_INTEGER libOffset,
 555                                                ULARGE_INTEGER cb, DWORD dwLockType)
 556 {
 557     ProtocolStream *This = impl_from_IStream(iface);
 558     FIXME("(%p)->(%d %d %d)\n", This, libOffset.u.LowPart, cb.u.LowPart, dwLockType);
 559     return E_NOTIMPL;
 560 }
 561
 562 static HRESULT WINAPI ProtocolStream_UnlockRegion(IStream *iface,
 563         ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
 564 {
 565     ProtocolStream *This = impl_from_IStream(iface);
 566     FIXME("(%p)->(%d %d %d)\n", This, libOffset.u.LowPart, cb.u.LowPart, dwLockType);
 567     return E_NOTIMPL;
 568 }
 569
 570 static HRESULT WINAPI ProtocolStream_Stat(IStream *iface, STATSTG *pstatstg,
 571                                          DWORD dwStatFlag)
 572 {
 573     ProtocolStream *This = impl_from_IStream(iface);
 574     TRACE("(%p)->(%p %08x)\n", This, pstatstg, dwStatFlag);
 575
 576     if(!pstatstg)
 577         return E_FAIL;
 578
 579     memset(pstatstg, 0, sizeof(STATSTG));
 580
 581     if(!(dwStatFlag&STATFLAG_NONAME) && This->buf->cache_file) {
 582         pstatstg->pwcsName = CoTaskMemAlloc((lstrlenW(This->buf->cache_file)+1)*sizeof(WCHAR));
 583         if(!pstatstg->pwcsName)
 584             return STG_E_INSUFFICIENTMEMORY;
 585
 586         lstrcpyW(pstatstg->pwcsName, This->buf->cache_file);
 587     }
 588
 589     pstatstg->type = STGTY_STREAM;
 590     if(This->buf->file != INVALID_HANDLE_VALUE) {
 591         GetFileSizeEx(This->buf->file, (PLARGE_INTEGER)&pstatstg->cbSize);
 592         GetFileTime(This->buf->file, &pstatstg->ctime, &pstatstg->atime, &pstatstg->mtime);
 593         if(pstatstg->cbSize.QuadPart)
 594             pstatstg->grfMode = GENERIC_READ;
 595     }
 596
 597     return S_OK;
 598 }
 599
 600 static HRESULT WINAPI ProtocolStream_Clone(IStream *iface, IStream **ppstm)
 601 {
 602     ProtocolStream *This = impl_from_IStream(iface);
 603     FIXME("(%p)->(%p)\n", This, ppstm);
 604     return E_NOTIMPL;
 605 }
 606
 607 static const IStreamVtbl ProtocolStreamVtbl = {
 608     ProtocolStream_QueryInterface,
 609     ProtocolStream_AddRef,
 610     ProtocolStream_Release,
 611     ProtocolStream_Read,
 612     ProtocolStream_Write,
 613     ProtocolStream_Seek,
 614     ProtocolStream_SetSize,
 615     ProtocolStream_CopyTo,
 616     ProtocolStream_Commit,
 617     ProtocolStream_Revert,
 618     ProtocolStream_LockRegion,
 619     ProtocolStream_UnlockRegion,
 620     ProtocolStream_Stat,
 621     ProtocolStream_Clone
 622 };
 623
 624 static void stgmed_stream_release(stgmed_obj_t *obj)
 625 {
 626     ProtocolStream *stream = (ProtocolStream*)obj;
 627     IStream_Release(&stream->IStream_iface);
 628 }
 629
 630 static HRESULT stgmed_stream_fill_stgmed(stgmed_obj_t *obj, STGMEDIUM *stgmed)
 631 {
 632     ProtocolStream *stream = (ProtocolStream*)obj;
 633
 634     stgmed->tymed = TYMED_ISTREAM;
 635     stgmed->u.pstm = &stream->IStream_iface;
 636     stgmed->pUnkForRelease = &stream->buf->IUnknown_iface;
 637
 638     return S_OK;
 639 }
 640
 641 static HRESULT stgmed_stream_get_result(stgmed_obj_t *obj, DWORD bindf, void **result)
 642 {
 643     ProtocolStream *stream = (ProtocolStream*)obj;
 644
 645     if(!(bindf & BINDF_ASYNCHRONOUS) && stream->buf->file == INVALID_HANDLE_VALUE
 646        && stream->buf->hres != S_FALSE)
 647         return INET_E_DATA_NOT_AVAILABLE;
 648
 649     IStream_AddRef(&stream->IStream_iface);
 650     *result = &stream->IStream_iface;
 651     return S_OK;
 652 }
 653
 654 static const stgmed_obj_vtbl stgmed_stream_vtbl = {
 655     stgmed_stream_release,
 656     stgmed_stream_fill_stgmed,
 657     stgmed_stream_get_result
 658 };
 659
 660 typedef struct {
 661     stgmed_obj_t stgmed_obj;
 662     stgmed_buf_t *buf;
 663 } stgmed_file_obj_t;
 664
 665 static stgmed_obj_t *create_stgmed_stream(stgmed_buf_t *buf)
 666 {
 667     ProtocolStream *ret = heap_alloc(sizeof(ProtocolStream));
 668
 669     ret->stgmed_obj.vtbl = &stgmed_stream_vtbl;
 670     ret->IStream_iface.lpVtbl = &ProtocolStreamVtbl;
 671     ret->ref = 1;
 672
 673     IUnknown_AddRef(&buf->IUnknown_iface);
 674     ret->buf = buf;
 675
 676     URLMON_LockModule();
 677
 678     return &ret->stgmed_obj;
 679 }
 680
 681 static void stgmed_file_release(stgmed_obj_t *obj)
 682 {
 683     stgmed_file_obj_t *file_obj = (stgmed_file_obj_t*)obj;
 684
 685     IUnknown_Release(&file_obj->buf->IUnknown_iface);
 686     heap_free(file_obj);
 687 }
 688
 689 static HRESULT stgmed_file_fill_stgmed(stgmed_obj_t *obj, STGMEDIUM *stgmed)
 690 {
 691     stgmed_file_obj_t *file_obj = (stgmed_file_obj_t*)obj;
 692
 693     if(!file_obj->buf->cache_file) {
 694         WARN("cache_file not set\n");
 695         return INET_E_DATA_NOT_AVAILABLE;
 696     }
 697
 698     read_protocol_data(file_obj->buf);
 699
 700     stgmed->tymed = TYMED_FILE;
 701     stgmed->u.lpszFileName = file_obj->buf->cache_file;
 702     stgmed->pUnkForRelease = &file_obj->buf->IUnknown_iface;
 703
 704     return S_OK;
 705 }
 706
 707 static HRESULT stgmed_file_get_result(stgmed_obj_t *obj, DWORD bindf, void **result)
 708 {
 709     return bindf & BINDF_ASYNCHRONOUS ? MK_S_ASYNCHRONOUS : S_OK;
 710 }
 711
 712 static const stgmed_obj_vtbl stgmed_file_vtbl = {
 713     stgmed_file_release,
 714     stgmed_file_fill_stgmed,
 715     stgmed_file_get_result
 716 };
 717
 718 static stgmed_obj_t *create_stgmed_file(stgmed_buf_t *buf)
 719 {
 720     stgmed_file_obj_t *ret = heap_alloc(sizeof(*ret));
 721
 722     ret->stgmed_obj.vtbl = &stgmed_file_vtbl;
 723
 724     IUnknown_AddRef(&buf->IUnknown_iface);
 725     ret->buf = buf;
 726
 727     return &ret->stgmed_obj;
 728 }
 729
 730 static inline Binding *impl_from_IBinding(IBinding *iface)
 731 {
 732     return CONTAINING_RECORD(iface, Binding, IBinding_iface);
 733 }
 734
 735 static HRESULT WINAPI Binding_QueryInterface(IBinding *iface, REFIID riid, void **ppv)
 736 {
 737     Binding *This = impl_from_IBinding(iface);
 738
 739     *ppv = NULL;
 740
 741     if(IsEqualGUID(&IID_IUnknown, riid)) {
 742         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
 743         *ppv = &This->IBinding_iface;
 744     }else if(IsEqualGUID(&IID_IBinding, riid)) {
 745         TRACE("(%p)->(IID_IBinding %p)\n", This, ppv);
 746         *ppv = &This->IBinding_iface;
 747     }else if(IsEqualGUID(&IID_IInternetProtocolSink, riid)) {
 748         TRACE("(%p)->(IID_IInternetProtocolSink %p)\n", This, ppv);
 749         *ppv = &This->IInternetProtocolSink_iface;
 750     }else if(IsEqualGUID(&IID_IInternetBindInfo, riid)) {
 751         TRACE("(%p)->(IID_IInternetBindInfo %p)\n", This, ppv);
 752         *ppv = &This->IInternetBindInfo_iface;
 753     }else if(IsEqualGUID(&IID_IServiceProvider, riid)) {
 754         TRACE("(%p)->(IID_IServiceProvider %p)\n", This, ppv);
 755         *ppv = &This->IServiceProvider_iface;
 756     }else if(IsEqualGUID(&IID_IWinInetInfo, riid)) {
 757         TRACE("(%p)->(IID_IWinInetInfo %p)\n", This, ppv);
 758
 759         /* NOTE: This violidates COM rules, but tests prove that we should do it */
 760         if(!This->protocol->wininet_info)
 761            return E_NOINTERFACE;
 762
 763         *ppv = &This->IWinInetHttpInfo_iface;
 764     }else if(IsEqualGUID(&IID_IWinInetHttpInfo, riid)) {
 765         TRACE("(%p)->(IID_IWinInetHttpInfo %p)\n", This, ppv);
 766
 767         if(!This->protocol->wininet_http_info)
 768             return E_NOINTERFACE;
 769
 770         *ppv = &This->IWinInetHttpInfo_iface;
 771     }
 772
 773     if(*ppv) {
 774         IBinding_AddRef(&This->IBinding_iface);
 775         return S_OK;
 776     }
 777
 778     WARN("Unsupported interface %s\n", debugstr_guid(riid));
 779     return E_NOINTERFACE;
 780 }
 781
 782 static ULONG WINAPI Binding_AddRef(IBinding *iface)
 783 {
 784     Binding *This = impl_from_IBinding(iface);
 785     LONG ref = InterlockedIncrement(&This->ref);
 786
 787     TRACE("(%p) ref=%d\n", This, ref);
 788
 789     return ref;
 790 }
 791
 792 static ULONG WINAPI Binding_Release(IBinding *iface)
 793 {
 794     Binding *This = impl_from_IBinding(iface);
 795     LONG ref = InterlockedDecrement(&This->ref);
 796
 797     TRACE("(%p) ref=%d\n", This, ref);
 798
 799     if(!ref) {
 800         if(This->notif_hwnd)
 801             release_notif_hwnd(This->notif_hwnd);
 802         if(This->mon)
 803             IMoniker_Release(This->mon);
 804         if(This->callback)
 805             IBindStatusCallback_Release(This->callback);
 806         if(This->protocol)
 807             IInternetProtocolEx_Release(&This->protocol->IInternetProtocolEx_iface);
 808         if(This->service_provider)
 809             IServiceProvider_Release(This->service_provider);
 810         if(This->stgmed_buf)
 811             IUnknown_Release(&This->stgmed_buf->IUnknown_iface);
 812         if(This->stgmed_obj)
 813             This->stgmed_obj->vtbl->release(This->stgmed_obj);
 814         if(This->obj)
 815             IUnknown_Release(This->obj);
 816         if(This->bctx)
 817             IBindCtx_Release(This->bctx);
 818
 819         ReleaseBindInfo(&This->bindinfo);
 820         This->section.DebugInfo->Spare[0] = 0;
 821         DeleteCriticalSection(&This->section);
 822         SysFreeString(This->url);
 823         heap_free(This->mime);
 824         heap_free(This->redirect_url);
 825         heap_free(This);
 826
 827         URLMON_UnlockModule();
 828     }
 829
 830     return ref;
 831 }
 832
 833 static HRESULT WINAPI Binding_Abort(IBinding *iface)
 834 {
 835     Binding *This = impl_from_IBinding(iface);
 836     HRESULT hres;
 837
 838     TRACE("(%p)\n", This);
 839
 840     if(This->state & BINDING_ABORTED)
 841         return E_FAIL;
 842
 843     hres = IInternetProtocolEx_Abort(&This->protocol->IInternetProtocolEx_iface, E_ABORT,
 844             ERROR_SUCCESS);
 845     if(FAILED(hres))
 846         return hres;
 847
 848     This->state |= BINDING_ABORTED;
 849     return S_OK;
 850 }
 851
 852 static HRESULT WINAPI Binding_Suspend(IBinding *iface)
 853 {
 854     Binding *This = impl_from_IBinding(iface);
 855     FIXME("(%p)\n", This);
 856     return E_NOTIMPL;
 857 }
 858
 859 static HRESULT WINAPI Binding_Resume(IBinding *iface)
 860 {
 861     Binding *This = impl_from_IBinding(iface);
 862     FIXME("(%p)\n", This);
 863     return E_NOTIMPL;
 864 }
 865
 866 static HRESULT WINAPI Binding_SetPriority(IBinding *iface, LONG nPriority)
 867 {
 868     Binding *This = impl_from_IBinding(iface);
 869     FIXME("(%p)->(%d)\n", This, nPriority);
 870     return E_NOTIMPL;
 871 }
 872
 873 static HRESULT WINAPI Binding_GetPriority(IBinding *iface, LONG *pnPriority)
 874 {
 875     Binding *This = impl_from_IBinding(iface);
 876     FIXME("(%p)->(%p)\n", This, pnPriority);
 877     return E_NOTIMPL;
 878 }
 879
 880 static HRESULT WINAPI Binding_GetBindResult(IBinding *iface, CLSID *pclsidProtocol,
 881         DWORD *pdwResult, LPOLESTR *pszResult, DWORD *pdwReserved)
 882 {
 883     Binding *This = impl_from_IBinding(iface);
 884     FIXME("(%p)->(%p %p %p %p)\n", This, pclsidProtocol, pdwResult, pszResult, pdwReserved);
 885     return E_NOTIMPL;
 886 }
 887
 888 static const IBindingVtbl BindingVtbl = {
 889     Binding_QueryInterface,
 890     Binding_AddRef,
 891     Binding_Release,
 892     Binding_Abort,
 893     Binding_Suspend,
 894     Binding_Resume,
 895     Binding_SetPriority,
 896     Binding_GetPriority,
 897     Binding_GetBindResult
 898 };
 899
 900 static Binding *get_bctx_binding(IBindCtx *bctx)
 901 {
 902     IBinding *binding;
 903     IUnknown *unk;
 904     HRESULT hres;
 905
 906     hres = IBindCtx_GetObjectParam(bctx, cbinding_contextW, &unk);
 907     if(FAILED(hres))
 908         return NULL;
 909
 910     hres = IUnknown_QueryInterface(unk, &IID_IBinding, (void**)&binding);
 911     IUnknown_Release(unk);
 912     if(FAILED(hres))
 913         return NULL;
 914
 915     if (binding->lpVtbl != &BindingVtbl)
 916         return NULL;
 917     return impl_from_IBinding(binding);
 918 }
 919
 920 static inline Binding *impl_from_IInternetProtocolSink(IInternetProtocolSink *iface)
 921 {
 922     return CONTAINING_RECORD(iface, Binding, IInternetProtocolSink_iface);
 923 }
 924
 925 static HRESULT WINAPI InternetProtocolSink_QueryInterface(IInternetProtocolSink *iface,
 926         REFIID riid, void **ppv)
 927 {
 928     Binding *This = impl_from_IInternetProtocolSink(iface);
 929     return IBinding_QueryInterface(&This->IBinding_iface, riid, ppv);
 930 }
 931
 932 static ULONG WINAPI InternetProtocolSink_AddRef(IInternetProtocolSink *iface)
 933 {
 934     Binding *This = impl_from_IInternetProtocolSink(iface);
 935     return IBinding_AddRef(&This->IBinding_iface);
 936 }
 937
 938 static ULONG WINAPI InternetProtocolSink_Release(IInternetProtocolSink *iface)
 939 {
 940     Binding *This = impl_from_IInternetProtocolSink(iface);
 941     return IBinding_Release(&This->IBinding_iface);
 942 }
 943
 944 static HRESULT WINAPI InternetProtocolSink_Switch(IInternetProtocolSink *iface,
 945         PROTOCOLDATA *pProtocolData)
 946 {
 947     Binding *This = impl_from_IInternetProtocolSink(iface);
 948
 949     WARN("(%p)->(%p)\n", This, pProtocolData);
 950
 951     return E_FAIL;
 952 }
 953
 954 static void on_progress(Binding *This, ULONG progress, ULONG progress_max,
 955                         ULONG status_code, LPCWSTR status_text)
 956 {
 957     IBindStatusCallback_OnProgress(This->callback, progress, progress_max,
 958             status_code, status_text);
 959 }
 960
 961 static HRESULT WINAPI InternetProtocolSink_ReportProgress(IInternetProtocolSink *iface,
 962         ULONG ulStatusCode, LPCWSTR szStatusText)
 963 {
 964     Binding *This = impl_from_IInternetProtocolSink(iface);
 965
 966     TRACE("(%p)->(%u %s)\n", This, ulStatusCode, debugstr_w(szStatusText));
 967
 968     switch(ulStatusCode) {
 969     case BINDSTATUS_FINDINGRESOURCE:
 970         on_progress(This, 0, 0, BINDSTATUS_FINDINGRESOURCE, szStatusText);
 971         break;
 972     case BINDSTATUS_CONNECTING:
 973         on_progress(This, 0, 0, BINDSTATUS_CONNECTING, szStatusText);
 974         break;
 975     case BINDSTATUS_REDIRECTING:
 976         heap_free(This->redirect_url);
 977         This->redirect_url = heap_strdupW(szStatusText);
 978         on_progress(This, 0, 0, BINDSTATUS_REDIRECTING, szStatusText);
 979         break;
 980     case BINDSTATUS_BEGINDOWNLOADDATA:
 981         break;
 982     case BINDSTATUS_SENDINGREQUEST:
 983         on_progress(This, 0, 0, BINDSTATUS_SENDINGREQUEST, szStatusText);
 984         break;
 985     case BINDSTATUS_PROTOCOLCLASSID:
 986         break;
 987     case BINDSTATUS_MIMETYPEAVAILABLE:
 988     case BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE:
 989         mime_available(This, szStatusText);
 990         break;
 991     case BINDSTATUS_CACHEFILENAMEAVAILABLE:
 992         cache_file_available(This, szStatusText);
 993         break;
 994     case BINDSTATUS_DECODING:
 995         IBindStatusCallback_OnProgress(This->callback, 0, 0, BINDSTATUS_DECODING, szStatusText);
 996         break;
 997     case BINDSTATUS_LOADINGMIMEHANDLER:
 998         on_progress(This, 0, 0, BINDSTATUS_LOADINGMIMEHANDLER, szStatusText);
 999         break;
1000     case BINDSTATUS_DIRECTBIND: /* FIXME: Handle BINDSTATUS_DIRECTBIND in BindProtocol */
1001         This->report_mime = FALSE;
1002         break;
1003     case BINDSTATUS_ACCEPTRANGES:
1004         break;
1005     default:
1006         FIXME("Unhandled status code %d\n", ulStatusCode);
1007         return E_NOTIMPL;
1008     };
1009
1010     return S_OK;
1011 }
1012
1013 static void report_data(Binding *This, DWORD bscf, ULONG progress, ULONG progress_max)
1014 {
1015     FORMATETC formatetc = {0, NULL, 1, -1, TYMED_ISTREAM};
1016     BOOL sent_begindownloaddata = FALSE;
1017
1018     TRACE("(%p)->(%d %u %u)\n", This, bscf, progress, progress_max);
1019
1020     if(This->download_state == END_DOWNLOAD || (This->state & BINDING_ABORTED)) {
1021         read_protocol_data(This->stgmed_buf);
1022         return;
1023     }
1024
1025     if(This->state & BINDING_STOPPED)
1026         return;
1027
1028     if(This->stgmed_buf->file != INVALID_HANDLE_VALUE)
1029         read_protocol_data(This->stgmed_buf);
1030
1031     if(This->download_state == BEFORE_DOWNLOAD) {
1032         This->download_state = DOWNLOADING;
1033         sent_begindownloaddata = TRUE;
1034         IBindStatusCallback_OnProgress(This->callback, progress, progress_max,
1035                 BINDSTATUS_BEGINDOWNLOADDATA, This->url);
1036
1037         if(This->stgmed_buf->cache_file)
1038             IBindStatusCallback_OnProgress(This->callback, progress, progress_max,
1039                     BINDSTATUS_CACHEFILENAMEAVAILABLE, This->stgmed_buf->cache_file);
1040     }
1041
1042     if(This->stgmed_buf->hres == S_FALSE || (bscf & BSCF_LASTDATANOTIFICATION)) {
1043         This->download_state = END_DOWNLOAD;
1044         IBindStatusCallback_OnProgress(This->callback, progress, progress_max,
1045                 BINDSTATUS_ENDDOWNLOADDATA, This->url);
1046     }else if(!sent_begindownloaddata) {
1047         IBindStatusCallback_OnProgress(This->callback, progress, progress_max,
1048                 BINDSTATUS_DOWNLOADINGDATA, This->url);
1049     }
1050
1051     if(This->state & (BINDING_STOPPED|BINDING_ABORTED))
1052         return;
1053
1054     if(This->to_object) {
1055         if(!(This->state & BINDING_OBJAVAIL)) {
1056             IBinding_AddRef(&This->IBinding_iface);
1057             create_object(This);
1058             IBinding_Release(&This->IBinding_iface);
1059         }
1060     }else {
1061         STGMEDIUM stgmed;
1062         HRESULT hres;
1063
1064         if(!(This->state & BINDING_LOCKED)) {
1065             HRESULT hres = IInternetProtocolEx_LockRequest(
1066                     &This->protocol->IInternetProtocolEx_iface, 0);
1067             if(SUCCEEDED(hres))
1068                 This->state |= BINDING_LOCKED;
1069         }
1070
1071         hres = This->stgmed_obj->vtbl->fill_stgmed(This->stgmed_obj, &stgmed);
1072         if(FAILED(hres)) {
1073             stop_binding(This, hres, NULL);
1074             return;
1075         }
1076
1077         formatetc.tymed = stgmed.tymed;
1078         formatetc.cfFormat = This->clipboard_format;
1079
1080         IBindStatusCallback_OnDataAvailable(This->callback, bscf, progress,
1081                 &formatetc, &stgmed);
1082
1083         if(This->download_state == END_DOWNLOAD)
1084             stop_binding(This, S_OK, NULL);
1085     }
1086 }
1087
1088 static HRESULT WINAPI InternetProtocolSink_ReportData(IInternetProtocolSink *iface,
1089         DWORD grfBSCF, ULONG ulProgress, ULONG ulProgressMax)
1090 {
1091     Binding *This = impl_from_IInternetProtocolSink(iface);
1092
1093     TRACE("(%p)->(%d %u %u)\n", This, grfBSCF, ulProgress, ulProgressMax);
1094
1095     report_data(This, grfBSCF, ulProgress, ulProgressMax);
1096     return S_OK;
1097 }
1098
1099 static HRESULT WINAPI InternetProtocolSink_ReportResult(IInternetProtocolSink *iface,
1100         HRESULT hrResult, DWORD dwError, LPCWSTR szResult)
1101 {
1102     Binding *This = impl_from_IInternetProtocolSink(iface);
1103
1104     TRACE("(%p)->(%08x %d %s)\n", This, hrResult, dwError, debugstr_w(szResult));
1105
1106     stop_binding(This, hrResult, szResult);
1107     IInternetProtocolEx_Terminate(&This->protocol->IInternetProtocolEx_iface, 0);
1108     return S_OK;
1109 }
1110
1111 static const IInternetProtocolSinkVtbl InternetProtocolSinkVtbl = {
1112     InternetProtocolSink_QueryInterface,
1113     InternetProtocolSink_AddRef,
1114     InternetProtocolSink_Release,
1115     InternetProtocolSink_Switch,
1116     InternetProtocolSink_ReportProgress,
1117     InternetProtocolSink_ReportData,
1118     InternetProtocolSink_ReportResult
1119 };
1120
1121 static inline Binding *impl_from_IInternetBindInfo(IInternetBindInfo *iface)
1122 {
1123     return CONTAINING_RECORD(iface, Binding, IInternetBindInfo_iface);
1124 }
1125
1126 static HRESULT WINAPI InternetBindInfo_QueryInterface(IInternetBindInfo *iface,
1127         REFIID riid, void **ppv)
1128 {
1129     Binding *This = impl_from_IInternetBindInfo(iface);
1130     return IBinding_QueryInterface(&This->IBinding_iface, riid, ppv);
1131 }
1132
1133 static ULONG WINAPI InternetBindInfo_AddRef(IInternetBindInfo *iface)
1134 {
1135     Binding *This = impl_from_IInternetBindInfo(iface);
1136     return IBinding_AddRef(&This->IBinding_iface);
1137 }
1138
1139 static ULONG WINAPI InternetBindInfo_Release(IInternetBindInfo *iface)
1140 {
1141     Binding *This = impl_from_IInternetBindInfo(iface);
1142     return IBinding_Release(&This->IBinding_iface);
1143 }
1144
1145 static HRESULT WINAPI InternetBindInfo_GetBindInfo(IInternetBindInfo *iface,
1146         DWORD *grfBINDF, BINDINFO *pbindinfo)
1147 {
1148     Binding *This = impl_from_IInternetBindInfo(iface);
1149
1150     TRACE("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo);
1151
1152     *grfBINDF = This->bindf;
1153     return CopyBindInfo(&This->bindinfo, pbindinfo);
1154 }
1155
1156 static HRESULT WINAPI InternetBindInfo_GetBindString(IInternetBindInfo *iface,
1157         ULONG ulStringType, LPOLESTR *ppwzStr, ULONG cEl, ULONG *pcElFetched)
1158 {
1159     Binding *This = impl_from_IInternetBindInfo(iface);
1160
1161     TRACE("(%p)->(%d %p %d %p)\n", This, ulStringType, ppwzStr, cEl, pcElFetched);
1162
1163     switch(ulStringType) {
1164     case BINDSTRING_ACCEPT_MIMES: {
1165         static const WCHAR wszMimes[] = {'*','/','*',0};
1166
1167         if(!ppwzStr || !pcElFetched)
1168             return E_INVALIDARG;
1169
1170         ppwzStr[0] = CoTaskMemAlloc(sizeof(wszMimes));
1171         memcpy(ppwzStr[0], wszMimes, sizeof(wszMimes));
1172         *pcElFetched = 1;
1173         return S_OK;
1174     }
1175     case BINDSTRING_USER_AGENT: {
1176         IInternetBindInfo *bindinfo = NULL;
1177         HRESULT hres;
1178
1179         hres = IBindStatusCallback_QueryInterface(This->callback, &IID_IInternetBindInfo,
1180                                                   (void**)&bindinfo);
1181         if(FAILED(hres))
1182             return hres;
1183
1184         hres = IInternetBindInfo_GetBindString(bindinfo, ulStringType, ppwzStr,
1185                                                cEl, pcElFetched);
1186         IInternetBindInfo_Release(bindinfo);
1187
1188         return hres;
1189     }
1190     case BINDSTRING_URL: {
1191         DWORD size = (SysStringLen(This->url)+1) * sizeof(WCHAR);
1192
1193         if(!ppwzStr || !pcElFetched)
1194             return E_INVALIDARG;
1195
1196         *ppwzStr = CoTaskMemAlloc(size);
1197         memcpy(*ppwzStr, This->url, size);
1198         *pcElFetched = 1;
1199         return S_OK;
1200     }
1201     }
1202
1203     FIXME("not supported string type %d\n", ulStringType);
1204     return E_NOTIMPL;
1205 }
1206
1207 static const IInternetBindInfoVtbl InternetBindInfoVtbl = {
1208     InternetBindInfo_QueryInterface,
1209     InternetBindInfo_AddRef,
1210     InternetBindInfo_Release,
1211     InternetBindInfo_GetBindInfo,
1212     InternetBindInfo_GetBindString
1213 };
1214
1215 static inline Binding *impl_from_IWinInetHttpInfo(IWinInetHttpInfo *iface)
1216 {
1217     return CONTAINING_RECORD(iface, Binding, IWinInetHttpInfo_iface);
1218 }
1219
1220 static HRESULT WINAPI WinInetHttpInfo_QueryInterface(IWinInetHttpInfo *iface, REFIID riid, void **ppv)
1221 {
1222     Binding *This = impl_from_IWinInetHttpInfo(iface);
1223     return IBinding_QueryInterface(&This->IBinding_iface, riid, ppv);
1224 }
1225
1226 static ULONG WINAPI WinInetHttpInfo_AddRef(IWinInetHttpInfo *iface)
1227 {
1228     Binding *This = impl_from_IWinInetHttpInfo(iface);
1229     return IBinding_AddRef(&This->IBinding_iface);
1230 }
1231
1232 static ULONG WINAPI WinInetHttpInfo_Release(IWinInetHttpInfo *iface)
1233 {
1234     Binding *This = impl_from_IWinInetHttpInfo(iface);
1235     return IBinding_Release(&This->IBinding_iface);
1236 }
1237
1238 static HRESULT WINAPI WinInetHttpInfo_QueryOption(IWinInetHttpInfo *iface, DWORD dwOption,
1239         void *pBuffer, DWORD *pcbBuffer)
1240 {
1241     Binding *This = impl_from_IWinInetHttpInfo(iface);
1242     TRACE("(%p)->(%x %p %p)\n", This, dwOption, pBuffer, pcbBuffer);
1243
1244     if(!This->protocol->wininet_info)
1245         return E_FAIL;
1246
1247     return IWinInetInfo_QueryOption(This->protocol->wininet_info,
1248             dwOption, pBuffer, pcbBuffer);
1249 }
1250
1251 static HRESULT WINAPI WinInetHttpInfo_QueryInfo(IWinInetHttpInfo *iface, DWORD dwOption,
1252         void *pBuffer, DWORD *pcbBuffer, DWORD *pdwFlags, DWORD *pdwReserved)
1253 {
1254     Binding *This = impl_from_IWinInetHttpInfo(iface);
1255     TRACE("(%p)->(%x %p %p %p %p)\n", This, dwOption, pBuffer, pcbBuffer, pdwFlags, pdwReserved);
1256
1257     if(!This->protocol->wininet_http_info)
1258         return E_FAIL;
1259
1260     return IWinInetHttpInfo_QueryInfo(This->protocol->wininet_http_info,
1261             dwOption, pBuffer, pcbBuffer, pdwFlags, pdwReserved);
1262 }
1263
1264 static const IWinInetHttpInfoVtbl WinInetHttpInfoVtbl = {
1265     WinInetHttpInfo_QueryInterface,
1266     WinInetHttpInfo_AddRef,
1267     WinInetHttpInfo_Release,
1268     WinInetHttpInfo_QueryOption,
1269     WinInetHttpInfo_QueryInfo
1270 };
1271
1272 static inline Binding *impl_from_IServiceProvider(IServiceProvider *iface)
1273 {
1274     return CONTAINING_RECORD(iface, Binding, IServiceProvider_iface);
1275 }
1276
1277 static HRESULT WINAPI ServiceProvider_QueryInterface(IServiceProvider *iface,
1278         REFIID riid, void **ppv)
1279 {
1280     Binding *This = impl_from_IServiceProvider(iface);
1281     return IBinding_QueryInterface(&This->IBinding_iface, riid, ppv);
1282 }
1283
1284 static ULONG WINAPI ServiceProvider_AddRef(IServiceProvider *iface)
1285 {
1286     Binding *This = impl_from_IServiceProvider(iface);
1287     return IBinding_AddRef(&This->IBinding_iface);
1288 }
1289
1290 static ULONG WINAPI ServiceProvider_Release(IServiceProvider *iface)
1291 {
1292     Binding *This = impl_from_IServiceProvider(iface);
1293     return IBinding_Release(&This->IBinding_iface);
1294 }
1295
1296 static HRESULT WINAPI ServiceProvider_QueryService(IServiceProvider *iface,
1297         REFGUID guidService, REFIID riid, void **ppv)
1298 {
1299     Binding *This = impl_from_IServiceProvider(iface);
1300     HRESULT hres;
1301
1302     TRACE("(%p)->(%s %s %p)\n", This, debugstr_guid(guidService), debugstr_guid(riid), ppv);
1303
1304     if(This->service_provider) {
1305         hres = IServiceProvider_QueryService(This->service_provider, guidService,
1306                                              riid, ppv);
1307         if(SUCCEEDED(hres))
1308             return hres;
1309     }
1310
1311     WARN("unknown service %s\n", debugstr_guid(guidService));
1312     return E_NOINTERFACE;
1313 }
1314
1315 static const IServiceProviderVtbl ServiceProviderVtbl = {
1316     ServiceProvider_QueryInterface,
1317     ServiceProvider_AddRef,
1318     ServiceProvider_Release,
1319     ServiceProvider_QueryService
1320 };
1321
1322 static HRESULT get_callback(IBindCtx *pbc, IBindStatusCallback **callback)
1323 {
1324     IUnknown *unk;
1325     HRESULT hres;
1326
1327     hres = IBindCtx_GetObjectParam(pbc, bscb_holderW, &unk);
1328     if(FAILED(hres))
1329         return create_default_callback(callback);
1330
1331     hres = IUnknown_QueryInterface(unk, &IID_IBindStatusCallback, (void**)callback);
1332     IUnknown_Release(unk);
1333     return hres;
1334 }
1335
1336 static BOOL is_urlmon_protocol(IUri *uri)
1337 {
1338     DWORD scheme;
1339     HRESULT hres;
1340
1341     hres = IUri_GetScheme(uri, &scheme);
1342     if(FAILED(hres))
1343         return FALSE;
1344
1345     switch(scheme) {
1346     case URL_SCHEME_FILE:
1347     case URL_SCHEME_FTP:
1348     case URL_SCHEME_GOPHER:
1349     case URL_SCHEME_HTTP:
1350     case URL_SCHEME_HTTPS:
1351     case URL_SCHEME_MK:
1352         return TRUE;
1353     }
1354
1355     return FALSE;
1356 }
1357
1358 static HRESULT Binding_Create(IMoniker *mon, Binding *binding_ctx, IUri *uri, IBindCtx *pbc,
1359         BOOL to_obj, REFIID riid, Binding **binding)
1360 {
1361     Binding *ret;
1362     HRESULT hres;
1363
1364     URLMON_LockModule();
1365
1366     ret = heap_alloc_zero(sizeof(Binding));
1367
1368     ret->IBinding_iface.lpVtbl = &BindingVtbl;
1369     ret->IInternetProtocolSink_iface.lpVtbl = &InternetProtocolSinkVtbl;
1370     ret->IInternetBindInfo_iface.lpVtbl = &InternetBindInfoVtbl;
1371     ret->IWinInetHttpInfo_iface.lpVtbl = &WinInetHttpInfoVtbl;
1372     ret->IServiceProvider_iface.lpVtbl = &ServiceProviderVtbl;
1373
1374     ret->ref = 1;
1375
1376     ret->to_object = to_obj;
1377     ret->iid = *riid;
1378     ret->notif_hwnd = get_notif_hwnd();
1379     ret->report_mime = !binding_ctx;
1380     ret->download_state = BEFORE_DOWNLOAD;
1381
1382     if(to_obj) {
1383         IBindCtx_AddRef(pbc);
1384         ret->bctx = pbc;
1385     }
1386
1387     if(mon) {
1388         IMoniker_AddRef(mon);
1389         ret->mon = mon;
1390     }
1391
1392     ret->bindinfo.cbSize = sizeof(BINDINFO);
1393
1394     InitializeCriticalSection(&ret->section);
1395     ret->section.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": Binding.section");
1396
1397     hres = get_callback(pbc, &ret->callback);
1398     if(FAILED(hres)) {
1399         WARN("Could not get IBindStatusCallback\n");
1400         IBinding_Release(&ret->IBinding_iface);
1401         return hres;
1402     }
1403
1404     IBindStatusCallback_QueryInterface(ret->callback, &IID_IServiceProvider,
1405                                        (void**)&ret->service_provider);
1406
1407     if(binding_ctx) {
1408         ret->protocol = binding_ctx->protocol;
1409         IInternetProtocolEx_AddRef(&ret->protocol->IInternetProtocolEx_iface);
1410     }else {
1411         hres = create_binding_protocol(TRUE, &ret->protocol);
1412         if(FAILED(hres)) {
1413             WARN("Could not get protocol handler\n");
1414             IBinding_Release(&ret->IBinding_iface);
1415             return hres;
1416         }
1417     }
1418
1419     hres = IBindStatusCallback_GetBindInfo(ret->callback, &ret->bindf, &ret->bindinfo);
1420     if(FAILED(hres)) {
1421         WARN("GetBindInfo failed: %08x\n", hres);
1422         IBinding_Release(&ret->IBinding_iface);
1423         return hres;
1424     }
1425
1426     TRACE("bindf %08x\n", ret->bindf);
1427     dump_BINDINFO(&ret->bindinfo);
1428
1429     ret->bindf |= BINDF_FROMURLMON;
1430     if(to_obj)
1431         ret->bindinfo.dwOptions |= 0x100000;
1432
1433     if(!(ret->bindf & BINDF_ASYNCHRONOUS) || !(ret->bindf & BINDF_PULLDATA)) {
1434         ret->bindf |= BINDF_NEEDFILE;
1435         ret->use_cache_file = TRUE;
1436     }else if(!is_urlmon_protocol(uri)) {
1437         ret->bindf |= BINDF_NEEDFILE;
1438     }
1439
1440     hres = IUri_GetDisplayUri(uri, &ret->url);
1441     if(FAILED(hres)) {
1442         IBinding_Release(&ret->IBinding_iface);
1443         return hres;
1444     }
1445
1446     if(binding_ctx) {
1447         ret->stgmed_buf = binding_ctx->stgmed_buf;
1448         IUnknown_AddRef(&ret->stgmed_buf->IUnknown_iface);
1449         ret->clipboard_format = binding_ctx->clipboard_format;
1450     }else {
1451         ret->stgmed_buf = create_stgmed_buf(&ret->protocol->IInternetProtocolEx_iface);
1452     }
1453
1454     if(to_obj) {
1455         ret->stgmed_obj = NULL;
1456     }else if(IsEqualGUID(&IID_IStream, riid)) {
1457         ret->stgmed_obj = create_stgmed_stream(ret->stgmed_buf);
1458     }else if(IsEqualGUID(&IID_IUnknown, riid)) {
1459         ret->bindf |= BINDF_NEEDFILE;
1460         ret->stgmed_obj = create_stgmed_file(ret->stgmed_buf);
1461     }else {
1462         FIXME("Unsupported riid %s\n", debugstr_guid(riid));
1463         IBinding_Release(&ret->IBinding_iface);
1464         return E_NOTIMPL;
1465     }
1466
1467     *binding = ret;
1468     return S_OK;
1469 }
1470
1471 static HRESULT start_binding(IMoniker *mon, Binding *binding_ctx, IUri *uri, IBindCtx *pbc,
1472                              BOOL to_obj, REFIID riid, Binding **ret)
1473 {
1474     Binding *binding = NULL;
1475     HRESULT hres;
1476     MSG msg;
1477
1478     hres = Binding_Create(mon, binding_ctx, uri, pbc, to_obj, riid, &binding);
1479     if(FAILED(hres))
1480         return hres;
1481
1482     hres = IBindStatusCallback_OnStartBinding(binding->callback, 0, &binding->IBinding_iface);
1483     if(FAILED(hres)) {
1484         WARN("OnStartBinding failed: %08x\n", hres);
1485         if(hres != E_ABORT && hres != E_NOTIMPL)
1486             hres = INET_E_DOWNLOAD_FAILURE;
1487
1488         stop_binding(binding, hres, NULL);
1489         IBinding_Release(&binding->IBinding_iface);
1490         return hres;
1491     }
1492
1493     if(binding_ctx) {
1494         set_binding_sink(binding->protocol, &binding->IInternetProtocolSink_iface,
1495                 &binding->IInternetBindInfo_iface);
1496         if(binding_ctx->redirect_url)
1497             IBindStatusCallback_OnProgress(binding->callback, 0, 0, BINDSTATUS_REDIRECTING, binding_ctx->redirect_url);
1498         report_data(binding, BSCF_FIRSTDATANOTIFICATION | (binding_ctx->download_state == END_DOWNLOAD ? BSCF_LASTDATANOTIFICATION : 0),
1499                 0, 0);
1500     }else {
1501         hres = IInternetProtocolEx_StartEx(&binding->protocol->IInternetProtocolEx_iface, uri,
1502                 &binding->IInternetProtocolSink_iface, &binding->IInternetBindInfo_iface,
1503                 PI_APARTMENTTHREADED|PI_MIMEVERIFICATION, 0);
1504
1505         TRACE("start ret %08x\n", hres);
1506
1507         if(FAILED(hres) && hres != E_PENDING) {
1508             stop_binding(binding, hres, NULL);
1509             IBinding_Release(&binding->IBinding_iface);
1510
1511             return hres;
1512         }
1513     }
1514
1515     while(!(binding->bindf & BINDF_ASYNCHRONOUS) &&
1516           !(binding->state & BINDING_STOPPED)) {
1517         MsgWaitForMultipleObjects(0, NULL, FALSE, 5000, QS_POSTMESSAGE);
1518         while (PeekMessageW(&msg, binding->notif_hwnd, WM_USER, WM_USER+117, PM_REMOVE|PM_NOYIELD)) {
1519             TranslateMessage(&msg);
1520             DispatchMessageW(&msg);
1521         }
1522     }
1523
1524     *ret = binding;
1525     return S_OK;
1526 }
1527
1528 HRESULT bind_to_storage(IUri *uri, IBindCtx *pbc, REFIID riid, void **ppv)
1529 {
1530     Binding *binding = NULL, *binding_ctx;
1531     HRESULT hres;
1532
1533     binding_ctx = get_bctx_binding(pbc);
1534
1535     hres = start_binding(NULL, binding_ctx, uri, pbc, FALSE, riid, &binding);
1536     if(binding_ctx)
1537         IBinding_Release(&binding_ctx->IBinding_iface);
1538     if(FAILED(hres))
1539         return hres;
1540
1541     if(binding->hres == S_OK && binding->download_state != BEFORE_DOWNLOAD /* FIXME */) {
1542         if((binding->state & BINDING_STOPPED) && (binding->state & BINDING_LOCKED))
1543             IInternetProtocolEx_UnlockRequest(&binding->protocol->IInternetProtocolEx_iface);
1544
1545         hres = binding->stgmed_obj->vtbl->get_result(binding->stgmed_obj, binding->bindf, ppv);
1546     }else if(binding->bindf & BINDF_ASYNCHRONOUS) {
1547         hres = MK_S_ASYNCHRONOUS;
1548     }else {
1549         hres = FAILED(binding->hres) ? binding->hres : S_OK;
1550     }
1551
1552     IBinding_Release(&binding->IBinding_iface);
1553
1554     return hres;
1555 }
1556
1557 HRESULT bind_to_object(IMoniker *mon, IUri *uri, IBindCtx *pbc, REFIID riid, void **ppv)
1558 {
1559     Binding *binding;
1560     HRESULT hres;
1561
1562     *ppv = NULL;
1563
1564     hres = start_binding(mon, NULL, uri, pbc, TRUE, riid, &binding);
1565     if(FAILED(hres))
1566         return hres;
1567
1568     if(binding->hres != S_OK) {
1569         hres = SUCCEEDED(binding->hres) ? S_OK : binding->hres;
1570     }else if(binding->bindf & BINDF_ASYNCHRONOUS) {
1571         hres = MK_S_ASYNCHRONOUS;
1572     }else {
1573         *ppv = binding->obj;
1574         IUnknown_AddRef(binding->obj);
1575         hres = S_OK;
1576     }
1577
1578     IBinding_Release(&binding->IBinding_iface);
1579
1580     return hres;
1581 }