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