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