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