d3d9/tests: Fix test failure for IDirect3D9::CreateDevice in test_set_stream_source.
[wine] / dlls / urlmon / bindprot.c
1 /*
2  * Copyright 2007-2009 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 "wine/debug.h"
21
22 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
23
24 typedef struct BindProtocol BindProtocol;
25
26 struct _task_header_t;
27
28 typedef void (*task_proc_t)(BindProtocol*,struct _task_header_t*);
29
30 typedef struct _task_header_t {
31     task_proc_t proc;
32     struct _task_header_t *next;
33 } task_header_t;
34
35 struct BindProtocol {
36     const IInternetProtocolVtbl      *lpIInternetProtocolVtbl;
37     const IInternetBindInfoVtbl      *lpInternetBindInfoVtbl;
38     const IInternetPriorityVtbl      *lpInternetPriorityVtbl;
39     const IServiceProviderVtbl       *lpServiceProviderVtbl;
40     const IInternetProtocolSinkVtbl  *lpIInternetProtocolSinkVtbl;
41     const IWinInetHttpInfoVtbl       *lpIWinInetHttpInfoVtbl;
42
43     const IInternetProtocolVtbl *lpIInternetProtocolHandlerVtbl;
44
45     LONG ref;
46
47     IInternetProtocol *protocol;
48     IInternetProtocol *protocol_handler;
49     IInternetBindInfo *bind_info;
50     IInternetProtocolSink *protocol_sink;
51     IServiceProvider *service_provider;
52     IWinInetInfo *wininet_info;
53
54     LONG priority;
55
56     BOOL reported_result;
57     BOOL reported_mime;
58     BOOL from_urlmon;
59     DWORD pi;
60
61     DWORD apartment_thread;
62     HWND notif_hwnd;
63     DWORD continue_call;
64
65     CRITICAL_SECTION section;
66     task_header_t *task_queue_head, *task_queue_tail;
67
68     BYTE *buf;
69     DWORD buf_size;
70     LPWSTR mime;
71     LPWSTR url;
72     ProtocolProxy *filter_proxy;
73 };
74
75 #define BINDINFO(x)  ((IInternetBindInfo*) &(x)->lpInternetBindInfoVtbl)
76 #define PRIORITY(x)  ((IInternetPriority*) &(x)->lpInternetPriorityVtbl)
77 #define HTTPINFO(x)  ((IWinInetHttpInfo*)  &(x)->lpIWinInetHttpInfoVtbl)
78 #define SERVPROV(x)  ((IServiceProvider*)  &(x)->lpServiceProviderVtbl)
79
80 #define PROTOCOLHANDLER(x)  ((IInternetProtocol*)  &(x)->lpIInternetProtocolHandlerVtbl)
81
82 #define BUFFER_SIZE     2048
83 #define MIME_TEST_SIZE  255
84
85 #define WM_MK_CONTINUE   (WM_USER+101)
86 #define WM_MK_RELEASE    (WM_USER+102)
87
88 static LRESULT WINAPI notif_wnd_proc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
89 {
90     switch(msg) {
91     case WM_MK_CONTINUE: {
92         BindProtocol *This = (BindProtocol*)lParam;
93         task_header_t *task;
94
95         while(1) {
96             EnterCriticalSection(&This->section);
97
98             task = This->task_queue_head;
99             if(task) {
100                 This->task_queue_head = task->next;
101                 if(!This->task_queue_head)
102                     This->task_queue_tail = NULL;
103             }
104
105             LeaveCriticalSection(&This->section);
106
107             if(!task)
108                 break;
109
110             This->continue_call++;
111             task->proc(This, task);
112             This->continue_call--;
113         }
114
115         IInternetProtocol_Release(PROTOCOL(This));
116         return 0;
117     }
118     case WM_MK_RELEASE: {
119         tls_data_t *data = get_tls_data();
120
121         if(!--data->notif_hwnd_cnt) {
122             DestroyWindow(hwnd);
123             data->notif_hwnd = NULL;
124         }
125     }
126     }
127
128     return DefWindowProcW(hwnd, msg, wParam, lParam);
129 }
130
131 HWND get_notif_hwnd(void)
132 {
133     static ATOM wnd_class = 0;
134     tls_data_t *tls_data;
135
136     static const WCHAR wszURLMonikerNotificationWindow[] =
137         {'U','R','L',' ','M','o','n','i','k','e','r',' ',
138          'N','o','t','i','f','i','c','a','t','i','o','n',' ','W','i','n','d','o','w',0};
139
140     tls_data = get_tls_data();
141     if(!tls_data)
142         return NULL;
143
144     if(tls_data->notif_hwnd_cnt) {
145         tls_data->notif_hwnd_cnt++;
146         return tls_data->notif_hwnd;
147     }
148
149     if(!wnd_class) {
150         static WNDCLASSEXW wndclass = {
151             sizeof(wndclass), 0,
152             notif_wnd_proc, 0, 0,
153             NULL, NULL, NULL, NULL, NULL,
154             wszURLMonikerNotificationWindow,
155             NULL
156         };
157
158         wndclass.hInstance = hProxyDll;
159
160         wnd_class = RegisterClassExW(&wndclass);
161         if (!wnd_class && GetLastError() == ERROR_CLASS_ALREADY_EXISTS)
162             wnd_class = 1;
163     }
164
165     tls_data->notif_hwnd = CreateWindowExW(0, wszURLMonikerNotificationWindow,
166             wszURLMonikerNotificationWindow, 0, 0, 0, 0, 0, HWND_MESSAGE,
167             NULL, hProxyDll, NULL);
168     if(tls_data->notif_hwnd)
169         tls_data->notif_hwnd_cnt++;
170
171     TRACE("hwnd = %p\n", tls_data->notif_hwnd);
172
173     return tls_data->notif_hwnd;
174 }
175
176 void release_notif_hwnd(HWND hwnd)
177 {
178     tls_data_t *data = get_tls_data();
179
180     if(!data)
181         return;
182
183     if(data->notif_hwnd != hwnd) {
184         PostMessageW(data->notif_hwnd, WM_MK_RELEASE, 0, 0);
185         return;
186     }
187
188     if(!--data->notif_hwnd_cnt) {
189         DestroyWindow(data->notif_hwnd);
190         data->notif_hwnd = NULL;
191     }
192 }
193
194 static void push_task(BindProtocol *This, task_header_t *task, task_proc_t proc)
195 {
196     BOOL do_post = FALSE;
197
198     task->proc = proc;
199     task->next = NULL;
200
201     EnterCriticalSection(&This->section);
202
203     if(This->task_queue_tail) {
204         This->task_queue_tail->next = task;
205         This->task_queue_tail = task;
206     }else {
207         This->task_queue_tail = This->task_queue_head = task;
208         do_post = !This->continue_call;
209     }
210
211     LeaveCriticalSection(&This->section);
212
213     if(do_post) {
214         IInternetProtocol_AddRef(PROTOCOL(This));
215         PostMessageW(This->notif_hwnd, WM_MK_CONTINUE, 0, (LPARAM)This);
216     }
217 }
218
219 static inline BOOL do_direct_notif(BindProtocol *This)
220 {
221     return !(This->pi & PI_APARTMENTTHREADED) || (This->apartment_thread == GetCurrentThreadId() && !This->continue_call);
222 }
223
224 static HRESULT handle_mime_filter(BindProtocol *This, IInternetProtocol *mime_filter, LPCWSTR mime)
225 {
226     PROTOCOLFILTERDATA filter_data = { sizeof(PROTOCOLFILTERDATA), NULL, NULL, NULL, 0 };
227     IInternetProtocolSink *protocol_sink, *old_sink;
228     ProtocolProxy *filter_proxy;
229     HRESULT hres;
230
231     hres = IInternetProtocol_QueryInterface(mime_filter, &IID_IInternetProtocolSink, (void**)&protocol_sink);
232     if(FAILED(hres))
233         return hres;
234
235     hres = create_protocol_proxy(PROTOCOLHANDLER(This), This->protocol_sink, &filter_proxy);
236     if(FAILED(hres)) {
237         IInternetProtocolSink_Release(protocol_sink);
238         return hres;
239     }
240
241     old_sink = This->protocol_sink;
242     This->protocol_sink = protocol_sink;
243     This->filter_proxy = filter_proxy;
244
245     IInternetProtocol_AddRef(mime_filter);
246     This->protocol_handler = mime_filter;
247
248     filter_data.pProtocol = PROTOCOL(filter_proxy);
249     hres = IInternetProtocol_Start(mime_filter, mime, PROTSINK(filter_proxy), BINDINFO(This),
250             PI_FILTER_MODE|PI_FORCE_ASYNC, (HANDLE_PTR)&filter_data);
251     if(FAILED(hres)) {
252         IInternetProtocolSink_Release(old_sink);
253         return hres;
254     }
255
256     IInternetProtocolSink_ReportProgress(old_sink, BINDSTATUS_LOADINGMIMEHANDLER, NULL);
257     IInternetProtocolSink_Release(old_sink);
258
259     This->pi &= ~PI_MIMEVERIFICATION; /* FIXME: more tests */
260     return S_OK;
261 }
262
263 static void mime_available(BindProtocol *This, LPCWSTR mime, BOOL verified)
264 {
265     IInternetProtocol *mime_filter;
266     HRESULT hres;
267
268     heap_free(This->mime);
269     This->mime = NULL;
270
271     mime_filter = get_mime_filter(mime);
272     if(mime_filter) {
273         TRACE("Got mime filter for %s\n", debugstr_w(mime));
274
275         hres = handle_mime_filter(This, mime_filter, mime);
276         IInternetProtocol_Release(mime_filter);
277         if(FAILED(hres))
278             FIXME("MIME filter failed: %08x\n", hres);
279     }else {
280         This->mime = heap_strdupW(mime);
281
282         if(verified || !(This->pi & PI_MIMEVERIFICATION)) {
283             This->reported_mime = TRUE;
284
285             if(This->protocol_sink)
286                 IInternetProtocolSink_ReportProgress(This->protocol_sink, BINDSTATUS_MIMETYPEAVAILABLE, mime);
287         }
288     }
289 }
290
291 #define PROTOCOL_THIS(iface) DEFINE_THIS(BindProtocol, IInternetProtocol, iface)
292
293 static HRESULT WINAPI BindProtocol_QueryInterface(IInternetProtocol *iface, REFIID riid, void **ppv)
294 {
295     BindProtocol *This = PROTOCOL_THIS(iface);
296
297     *ppv = NULL;
298     if(IsEqualGUID(&IID_IUnknown, riid)) {
299         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
300         *ppv = PROTOCOL(This);
301     }else if(IsEqualGUID(&IID_IInternetProtocolRoot, riid)) {
302         TRACE("(%p)->(IID_IInternetProtocolRoot %p)\n", This, ppv);
303         *ppv = PROTOCOL(This);
304     }else if(IsEqualGUID(&IID_IInternetProtocol, riid)) {
305         TRACE("(%p)->(IID_IInternetProtocol %p)\n", This, ppv);
306         *ppv = PROTOCOL(This);
307     }else if(IsEqualGUID(&IID_IInternetBindInfo, riid)) {
308         TRACE("(%p)->(IID_IInternetBindInfo %p)\n", This, ppv);
309         *ppv = BINDINFO(This);
310     }else if(IsEqualGUID(&IID_IInternetPriority, riid)) {
311         TRACE("(%p)->(IID_IInternetPriority %p)\n", This, ppv);
312         *ppv = PRIORITY(This);
313     }else if(IsEqualGUID(&IID_IAuthenticate, riid)) {
314         FIXME("(%p)->(IID_IAuthenticate %p)\n", This, ppv);
315     }else if(IsEqualGUID(&IID_IServiceProvider, riid)) {
316         TRACE("(%p)->(IID_IServiceProvider %p)\n", This, ppv);
317         *ppv = SERVPROV(This);
318     }else if(IsEqualGUID(&IID_IInternetProtocolSink, riid)) {
319         TRACE("(%p)->(IID_IInternetProtocolSink %p)\n", This, ppv);
320         *ppv = PROTSINK(This);
321     }else if(IsEqualGUID(&IID_IWinInetInfo, riid)) {
322         TRACE("(%p)->(IID_IWinInetInfo %p)\n", This, ppv);
323
324         if(This->protocol) {
325             IWinInetInfo *inet_info;
326             HRESULT hres;
327
328             hres = IInternetProtocol_QueryInterface(This->protocol, &IID_IWinInetInfo, (void**)&inet_info);
329             if(SUCCEEDED(hres)) {
330                 *ppv = HTTPINFO(This);
331                 IWinInetInfo_Release(inet_info);
332             }
333         }
334     }else if(IsEqualGUID(&IID_IWinInetHttpInfo, riid)) {
335         TRACE("(%p)->(IID_IWinInetHttpInfo %p)\n", This, ppv);
336
337         if(This->protocol) {
338             IWinInetHttpInfo *http_info;
339             HRESULT hres;
340
341             hres = IInternetProtocol_QueryInterface(This->protocol, &IID_IWinInetHttpInfo, (void**)&http_info);
342             if(SUCCEEDED(hres)) {
343                 *ppv = HTTPINFO(This);
344                 IWinInetHttpInfo_Release(http_info);
345             }
346         }
347     }else {
348         WARN("not supported interface %s\n", debugstr_guid(riid));
349     }
350
351     if(!*ppv)
352         return E_NOINTERFACE;
353
354     IUnknown_AddRef((IUnknown*)*ppv);
355     return S_OK;
356 }
357
358 static ULONG WINAPI BindProtocol_AddRef(IInternetProtocol *iface)
359 {
360     BindProtocol *This = PROTOCOL_THIS(iface);
361     LONG ref = InterlockedIncrement(&This->ref);
362     TRACE("(%p) ref=%d\n", This, ref);
363     return ref;
364 }
365
366 static ULONG WINAPI BindProtocol_Release(IInternetProtocol *iface)
367 {
368     BindProtocol *This = PROTOCOL_THIS(iface);
369     LONG ref = InterlockedDecrement(&This->ref);
370
371     TRACE("(%p) ref=%d\n", This, ref);
372
373     if(!ref) {
374         if(This->wininet_info)
375             IWinInetInfo_Release(This->wininet_info);
376         if(This->protocol)
377             IInternetProtocol_Release(This->protocol);
378         if(This->bind_info)
379             IInternetBindInfo_Release(This->bind_info);
380         if(This->protocol_handler && This->protocol_handler != PROTOCOLHANDLER(This))
381             IInternetProtocol_Release(This->protocol_handler);
382         if(This->filter_proxy)
383             IInternetProtocol_Release(PROTOCOL(This->filter_proxy));
384
385         set_binding_sink(PROTOCOL(This), NULL, NULL);
386
387         if(This->notif_hwnd)
388             release_notif_hwnd(This->notif_hwnd);
389         DeleteCriticalSection(&This->section);
390
391         heap_free(This->mime);
392         heap_free(This->url);
393         heap_free(This);
394
395         URLMON_UnlockModule();
396     }
397
398     return ref;
399 }
400
401 static HRESULT WINAPI BindProtocol_Start(IInternetProtocol *iface, LPCWSTR szUrl,
402         IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo,
403         DWORD grfPI, HANDLE_PTR dwReserved)
404 {
405     BindProtocol *This = PROTOCOL_THIS(iface);
406
407     TRACE("(%p)->(%s %p %p %08x %lx)\n", This, debugstr_w(szUrl), pOIProtSink,
408             pOIBindInfo, grfPI, dwReserved);
409
410     return IInternetProtocol_Start(This->protocol_handler, szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved);
411 }
412
413 static HRESULT WINAPI BindProtocol_Continue(IInternetProtocol *iface, PROTOCOLDATA *pProtocolData)
414 {
415     BindProtocol *This = PROTOCOL_THIS(iface);
416
417     TRACE("(%p)->(%p)\n", This, pProtocolData);
418
419     return IInternetProtocol_Continue(This->protocol_handler, pProtocolData);
420 }
421
422 static HRESULT WINAPI BindProtocol_Abort(IInternetProtocol *iface, HRESULT hrReason,
423         DWORD dwOptions)
424 {
425     BindProtocol *This = PROTOCOL_THIS(iface);
426
427     TRACE("(%p)->(%08x %08x)\n", This, hrReason, dwOptions);
428
429     return IInternetProtocol_Abort(This->protocol_handler, hrReason, dwOptions);
430 }
431
432 static HRESULT WINAPI BindProtocol_Terminate(IInternetProtocol *iface, DWORD dwOptions)
433 {
434     BindProtocol *This = PROTOCOL_THIS(iface);
435
436     TRACE("(%p)->(%08x)\n", This, dwOptions);
437
438     return IInternetProtocol_Terminate(This->protocol_handler, dwOptions);
439 }
440
441 static HRESULT WINAPI BindProtocol_Suspend(IInternetProtocol *iface)
442 {
443     BindProtocol *This = PROTOCOL_THIS(iface);
444     FIXME("(%p)\n", This);
445     return E_NOTIMPL;
446 }
447
448 static HRESULT WINAPI BindProtocol_Resume(IInternetProtocol *iface)
449 {
450     BindProtocol *This = PROTOCOL_THIS(iface);
451     FIXME("(%p)\n", This);
452     return E_NOTIMPL;
453 }
454
455 static HRESULT WINAPI BindProtocol_Read(IInternetProtocol *iface, void *pv,
456         ULONG cb, ULONG *pcbRead)
457 {
458     BindProtocol *This = PROTOCOL_THIS(iface);
459
460     TRACE("(%p)->(%p %u %p)\n", This, pv, cb, pcbRead);
461
462     if(pcbRead)
463         *pcbRead = 0;
464     return IInternetProtocol_Read(This->protocol_handler, pv, cb, pcbRead);
465 }
466
467 static HRESULT WINAPI BindProtocol_Seek(IInternetProtocol *iface, LARGE_INTEGER dlibMove,
468         DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
469 {
470     BindProtocol *This = PROTOCOL_THIS(iface);
471     FIXME("(%p)->(%d %d %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
472     return E_NOTIMPL;
473 }
474
475 static HRESULT WINAPI BindProtocol_LockRequest(IInternetProtocol *iface, DWORD dwOptions)
476 {
477     BindProtocol *This = PROTOCOL_THIS(iface);
478
479     TRACE("(%p)->(%08x)\n", This, dwOptions);
480
481     return IInternetProtocol_LockRequest(This->protocol_handler, dwOptions);
482 }
483
484 static HRESULT WINAPI BindProtocol_UnlockRequest(IInternetProtocol *iface)
485 {
486     BindProtocol *This = PROTOCOL_THIS(iface);
487
488     TRACE("(%p)\n", This);
489
490     return IInternetProtocol_UnlockRequest(This->protocol_handler);
491 }
492
493 void set_binding_sink(IInternetProtocol *bind_protocol, IInternetProtocolSink *sink, IInternetBindInfo *bind_info)
494 {
495     BindProtocol *This = PROTOCOL_THIS(bind_protocol);
496     IInternetProtocolSink *prev_sink;
497     IServiceProvider *service_provider = NULL;
498
499     if(sink)
500         IInternetProtocolSink_AddRef(sink);
501     prev_sink = InterlockedExchangePointer((void**)&This->protocol_sink, sink);
502     if(prev_sink)
503         IInternetProtocolSink_Release(prev_sink);
504
505     if(sink)
506         IInternetProtocolSink_QueryInterface(sink, &IID_IServiceProvider, (void**)&service_provider);
507     service_provider = InterlockedExchangePointer((void**)&This->service_provider, service_provider);
508     if(service_provider)
509         IServiceProvider_Release(service_provider);
510
511     if(bind_info)
512         IInternetBindInfo_AddRef(bind_info);
513     bind_info = InterlockedExchangePointer((void**)&This->bind_info, bind_info);
514     if(bind_info)
515         IInternetBindInfo_Release(bind_info);
516 }
517
518 IWinInetInfo *get_wininet_info(IInternetProtocol *bind_protocol)
519 {
520     BindProtocol *This = PROTOCOL_THIS(bind_protocol);
521
522     return This->wininet_info;
523 }
524
525 #undef PROTOCOL_THIS
526
527 static const IInternetProtocolVtbl BindProtocolVtbl = {
528     BindProtocol_QueryInterface,
529     BindProtocol_AddRef,
530     BindProtocol_Release,
531     BindProtocol_Start,
532     BindProtocol_Continue,
533     BindProtocol_Abort,
534     BindProtocol_Terminate,
535     BindProtocol_Suspend,
536     BindProtocol_Resume,
537     BindProtocol_Read,
538     BindProtocol_Seek,
539     BindProtocol_LockRequest,
540     BindProtocol_UnlockRequest
541 };
542
543 #define PROTOCOLHANDLER_THIS(iface) DEFINE_THIS(BindProtocol, IInternetProtocolHandler, iface)
544
545 static HRESULT WINAPI ProtocolHandler_QueryInterface(IInternetProtocol *iface, REFIID riid, void **ppv)
546 {
547     ERR("should not be called\n");
548     return E_NOINTERFACE;
549 }
550
551 static ULONG WINAPI ProtocolHandler_AddRef(IInternetProtocol *iface)
552 {
553     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
554     return IInternetProtocol_AddRef(PROTOCOL(This));
555 }
556
557 static ULONG WINAPI ProtocolHandler_Release(IInternetProtocol *iface)
558 {
559     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
560     return IInternetProtocol_Release(PROTOCOL(This));
561 }
562
563 static HRESULT WINAPI ProtocolHandler_Start(IInternetProtocol *iface, LPCWSTR szUrl,
564         IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo,
565         DWORD grfPI, HANDLE_PTR dwReserved)
566 {
567     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
568     IInternetProtocol *protocol = NULL;
569     IInternetPriority *priority;
570     IServiceProvider *service_provider;
571     BOOL urlmon_protocol = FALSE;
572     CLSID clsid = IID_NULL;
573     LPOLESTR clsid_str;
574     HRESULT hres;
575
576     TRACE("(%p)->(%s %p %p %08x %lx)\n", This, debugstr_w(szUrl), pOIProtSink,
577             pOIBindInfo, grfPI, dwReserved);
578
579     if(!szUrl || !pOIProtSink || !pOIBindInfo)
580         return E_INVALIDARG;
581
582     This->pi = grfPI;
583     This->url = heap_strdupW(szUrl);
584
585     hres = IInternetProtocolSink_QueryInterface(pOIProtSink, &IID_IServiceProvider,
586                                                 (void**)&service_provider);
587     if(SUCCEEDED(hres)) {
588         /* FIXME: What's protocol CLSID here? */
589         IServiceProvider_QueryService(service_provider, &IID_IInternetProtocol,
590                                       &IID_IInternetProtocol, (void**)&protocol);
591         IServiceProvider_Release(service_provider);
592     }
593
594     if(!protocol) {
595         IClassFactory *cf;
596         IUnknown *unk;
597
598         hres = get_protocol_handler(szUrl, &clsid, &urlmon_protocol, &cf);
599         if(FAILED(hres))
600             return hres;
601
602         if(This->from_urlmon) {
603             hres = IClassFactory_CreateInstance(cf, NULL, &IID_IInternetProtocol, (void**)&protocol);
604             IClassFactory_Release(cf);
605             if(FAILED(hres))
606                 return hres;
607         }else {
608             hres = IClassFactory_CreateInstance(cf, (IUnknown*)BINDINFO(This),
609                     &IID_IUnknown, (void**)&unk);
610             IClassFactory_Release(cf);
611             if(FAILED(hres))
612                 return hres;
613
614             hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocol, (void**)&protocol);
615             IUnknown_Release(unk);
616             if(FAILED(hres))
617                 return hres;
618         }
619     }
620
621     StringFromCLSID(&clsid, &clsid_str);
622     IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_PROTOCOLCLASSID, clsid_str);
623     CoTaskMemFree(clsid_str);
624
625     This->protocol = protocol;
626
627     if(urlmon_protocol)
628         IInternetProtocol_QueryInterface(protocol, &IID_IWinInetInfo, (void**)&This->wininet_info);
629
630     set_binding_sink(PROTOCOL(This), pOIProtSink, pOIBindInfo);
631
632     hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetPriority, (void**)&priority);
633     if(SUCCEEDED(hres)) {
634         IInternetPriority_SetPriority(priority, This->priority);
635         IInternetPriority_Release(priority);
636     }
637
638     return IInternetProtocol_Start(protocol, szUrl, PROTSINK(This), BINDINFO(This), 0, 0);
639 }
640
641 static HRESULT WINAPI ProtocolHandler_Continue(IInternetProtocol *iface, PROTOCOLDATA *pProtocolData)
642 {
643     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
644     HRESULT hres;
645
646     TRACE("(%p)->(%p)\n", This, pProtocolData);
647
648     hres = IInternetProtocol_Continue(This->protocol, pProtocolData);
649
650     heap_free(pProtocolData);
651     return hres;
652 }
653
654 static HRESULT WINAPI ProtocolHandler_Abort(IInternetProtocol *iface, HRESULT hrReason,
655         DWORD dwOptions)
656 {
657     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
658
659     TRACE("(%p)->(%08x %08x)\n", This, hrReason, dwOptions);
660
661     if(This->protocol && !This->reported_result)
662         return IInternetProtocol_Abort(This->protocol, hrReason, dwOptions);
663
664     return S_OK;
665 }
666
667 static HRESULT WINAPI ProtocolHandler_Terminate(IInternetProtocol *iface, DWORD dwOptions)
668 {
669     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
670
671     TRACE("(%p)->(%08x)\n", This, dwOptions);
672
673     if(!This->reported_result)
674         return E_FAIL;
675
676     IInternetProtocol_Terminate(This->protocol, 0);
677
678     if(This->filter_proxy) {
679         IInternetProtocol_Release(PROTOCOL(This->filter_proxy));
680         This->filter_proxy = NULL;
681     }
682
683     set_binding_sink(PROTOCOL(This), NULL, NULL);
684
685     if(This->bind_info) {
686         IInternetBindInfo_Release(This->bind_info);
687         This->bind_info = NULL;
688     }
689
690     return S_OK;
691 }
692
693 static HRESULT WINAPI ProtocolHandler_Suspend(IInternetProtocol *iface)
694 {
695     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
696     FIXME("(%p)\n", This);
697     return E_NOTIMPL;
698 }
699
700 static HRESULT WINAPI ProtocolHandler_Resume(IInternetProtocol *iface)
701 {
702     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
703     FIXME("(%p)\n", This);
704     return E_NOTIMPL;
705 }
706
707 static HRESULT WINAPI ProtocolHandler_Read(IInternetProtocol *iface, void *pv,
708         ULONG cb, ULONG *pcbRead)
709 {
710     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
711     ULONG read = 0;
712     HRESULT hres = S_OK;
713
714     TRACE("(%p)->(%p %u %p)\n", This, pv, cb, pcbRead);
715
716     if(This->buf_size) {
717         read = min(cb, This->buf_size);
718         memcpy(pv, This->buf, read);
719
720         if(read == This->buf_size) {
721             heap_free(This->buf);
722             This->buf = NULL;
723         }else {
724             memmove(This->buf, This->buf+cb, This->buf_size-cb);
725         }
726
727         This->buf_size -= read;
728     }
729
730     if(read < cb) {
731         ULONG cread = 0;
732
733         hres = IInternetProtocol_Read(This->protocol, (BYTE*)pv+read, cb-read, &cread);
734         read += cread;
735     }
736
737     *pcbRead = read;
738     return hres;
739 }
740
741 static HRESULT WINAPI ProtocolHandler_Seek(IInternetProtocol *iface, LARGE_INTEGER dlibMove,
742         DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
743 {
744     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
745     FIXME("(%p)->(%d %d %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
746     return E_NOTIMPL;
747 }
748
749 static HRESULT WINAPI ProtocolHandler_LockRequest(IInternetProtocol *iface, DWORD dwOptions)
750 {
751     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
752
753     TRACE("(%p)->(%08x)\n", This, dwOptions);
754
755     return IInternetProtocol_LockRequest(This->protocol, dwOptions);
756 }
757
758 static HRESULT WINAPI ProtocolHandler_UnlockRequest(IInternetProtocol *iface)
759 {
760     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
761
762     TRACE("(%p)\n", This);
763
764     return IInternetProtocol_UnlockRequest(This->protocol);
765 }
766
767 #undef PROTOCOL_THIS
768
769 static const IInternetProtocolVtbl InternetProtocolHandlerVtbl = {
770     ProtocolHandler_QueryInterface,
771     ProtocolHandler_AddRef,
772     ProtocolHandler_Release,
773     ProtocolHandler_Start,
774     ProtocolHandler_Continue,
775     ProtocolHandler_Abort,
776     ProtocolHandler_Terminate,
777     ProtocolHandler_Suspend,
778     ProtocolHandler_Resume,
779     ProtocolHandler_Read,
780     ProtocolHandler_Seek,
781     ProtocolHandler_LockRequest,
782     ProtocolHandler_UnlockRequest
783 };
784
785 #define BINDINFO_THIS(iface) DEFINE_THIS(BindProtocol, InternetBindInfo, iface)
786
787 static HRESULT WINAPI BindInfo_QueryInterface(IInternetBindInfo *iface,
788         REFIID riid, void **ppv)
789 {
790     BindProtocol *This = BINDINFO_THIS(iface);
791     return IInternetProtocol_QueryInterface(PROTOCOL(This), riid, ppv);
792 }
793
794 static ULONG WINAPI BindInfo_AddRef(IInternetBindInfo *iface)
795 {
796     BindProtocol *This = BINDINFO_THIS(iface);
797     return IBinding_AddRef(PROTOCOL(This));
798 }
799
800 static ULONG WINAPI BindInfo_Release(IInternetBindInfo *iface)
801 {
802     BindProtocol *This = BINDINFO_THIS(iface);
803     return IBinding_Release(PROTOCOL(This));
804 }
805
806 static HRESULT WINAPI BindInfo_GetBindInfo(IInternetBindInfo *iface,
807         DWORD *grfBINDF, BINDINFO *pbindinfo)
808 {
809     BindProtocol *This = BINDINFO_THIS(iface);
810     HRESULT hres;
811
812     TRACE("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo);
813
814     hres = IInternetBindInfo_GetBindInfo(This->bind_info, grfBINDF, pbindinfo);
815     if(FAILED(hres)) {
816         WARN("GetBindInfo failed: %08x\n", hres);
817         return hres;
818     }
819
820     *grfBINDF |= BINDF_FROMURLMON;
821     return hres;
822 }
823
824 static HRESULT WINAPI BindInfo_GetBindString(IInternetBindInfo *iface,
825         ULONG ulStringType, LPOLESTR *ppwzStr, ULONG cEl, ULONG *pcElFetched)
826 {
827     BindProtocol *This = BINDINFO_THIS(iface);
828
829     TRACE("(%p)->(%d %p %d %p)\n", This, ulStringType, ppwzStr, cEl, pcElFetched);
830
831     return IInternetBindInfo_GetBindString(This->bind_info, ulStringType, ppwzStr, cEl, pcElFetched);
832 }
833
834 #undef BINDFO_THIS
835
836 static const IInternetBindInfoVtbl InternetBindInfoVtbl = {
837     BindInfo_QueryInterface,
838     BindInfo_AddRef,
839     BindInfo_Release,
840     BindInfo_GetBindInfo,
841     BindInfo_GetBindString
842 };
843
844 #define PRIORITY_THIS(iface) DEFINE_THIS(BindProtocol, InternetPriority, iface)
845
846 static HRESULT WINAPI InternetPriority_QueryInterface(IInternetPriority *iface,
847         REFIID riid, void **ppv)
848 {
849     BindProtocol *This = PRIORITY_THIS(iface);
850     return IInternetProtocol_QueryInterface(PROTOCOL(This), riid, ppv);
851 }
852
853 static ULONG WINAPI InternetPriority_AddRef(IInternetPriority *iface)
854 {
855     BindProtocol *This = PRIORITY_THIS(iface);
856     return IInternetProtocol_AddRef(PROTOCOL(This));
857 }
858
859 static ULONG WINAPI InternetPriority_Release(IInternetPriority *iface)
860 {
861     BindProtocol *This = PRIORITY_THIS(iface);
862     return IInternetProtocol_Release(PROTOCOL(This));
863 }
864
865 static HRESULT WINAPI InternetPriority_SetPriority(IInternetPriority *iface, LONG nPriority)
866 {
867     BindProtocol *This = PRIORITY_THIS(iface);
868
869     TRACE("(%p)->(%d)\n", This, nPriority);
870
871     This->priority = nPriority;
872     return S_OK;
873 }
874
875 static HRESULT WINAPI InternetPriority_GetPriority(IInternetPriority *iface, LONG *pnPriority)
876 {
877     BindProtocol *This = PRIORITY_THIS(iface);
878
879     TRACE("(%p)->(%p)\n", This, pnPriority);
880
881     *pnPriority = This->priority;
882     return S_OK;
883 }
884
885 #undef PRIORITY_THIS
886
887 static const IInternetPriorityVtbl InternetPriorityVtbl = {
888     InternetPriority_QueryInterface,
889     InternetPriority_AddRef,
890     InternetPriority_Release,
891     InternetPriority_SetPriority,
892     InternetPriority_GetPriority
893
894 };
895
896 #define PROTSINK_THIS(iface) DEFINE_THIS(BindProtocol, IInternetProtocolSink, iface)
897
898 static HRESULT WINAPI BPInternetProtocolSink_QueryInterface(IInternetProtocolSink *iface,
899         REFIID riid, void **ppv)
900 {
901     BindProtocol *This = PROTSINK_THIS(iface);
902     return IInternetProtocol_QueryInterface(PROTOCOL(This), riid, ppv);
903 }
904
905 static ULONG WINAPI BPInternetProtocolSink_AddRef(IInternetProtocolSink *iface)
906 {
907     BindProtocol *This = PROTSINK_THIS(iface);
908     return IInternetProtocol_AddRef(PROTOCOL(This));
909 }
910
911 static ULONG WINAPI BPInternetProtocolSink_Release(IInternetProtocolSink *iface)
912 {
913     BindProtocol *This = PROTSINK_THIS(iface);
914     return IInternetProtocol_Release(PROTOCOL(This));
915 }
916
917 typedef struct {
918     task_header_t header;
919     PROTOCOLDATA *data;
920 } switch_task_t;
921
922 static void switch_proc(BindProtocol *bind, task_header_t *t)
923 {
924     switch_task_t *task = (switch_task_t*)t;
925
926     IInternetProtocol_Continue(bind->protocol_handler, task->data);
927
928     heap_free(task);
929 }
930
931 static HRESULT WINAPI BPInternetProtocolSink_Switch(IInternetProtocolSink *iface,
932         PROTOCOLDATA *pProtocolData)
933 {
934     BindProtocol *This = PROTSINK_THIS(iface);
935     PROTOCOLDATA *data;
936
937     TRACE("(%p)->(%p)\n", This, pProtocolData);
938
939     TRACE("flags %x state %x data %p cb %u\n", pProtocolData->grfFlags, pProtocolData->dwState,
940           pProtocolData->pData, pProtocolData->cbData);
941
942     data = heap_alloc(sizeof(PROTOCOLDATA));
943     if(!data)
944         return E_OUTOFMEMORY;
945     memcpy(data, pProtocolData, sizeof(PROTOCOLDATA));
946
947     if(!do_direct_notif(This)) {
948         switch_task_t *task;
949
950         task = heap_alloc(sizeof(switch_task_t));
951         if(!task)
952             return E_OUTOFMEMORY;
953
954         task->data = data;
955
956         push_task(This, &task->header, switch_proc);
957         return S_OK;
958     }
959
960     if(!This->protocol_sink) {
961         IInternetProtocol_Continue(This->protocol_handler, data);
962         return S_OK;
963     }
964
965     return IInternetProtocolSink_Switch(This->protocol_sink, data);
966 }
967
968 static void report_progress(BindProtocol *This, ULONG status_code, LPCWSTR status_text)
969 {
970     switch(status_code) {
971     case BINDSTATUS_FINDINGRESOURCE:
972     case BINDSTATUS_CONNECTING:
973     case BINDSTATUS_REDIRECTING:
974     case BINDSTATUS_BEGINDOWNLOADDATA:
975     case BINDSTATUS_SENDINGREQUEST:
976     case BINDSTATUS_CACHEFILENAMEAVAILABLE:
977     case BINDSTATUS_DIRECTBIND:
978     case BINDSTATUS_ACCEPTRANGES:
979         if(This->protocol_sink)
980             IInternetProtocolSink_ReportProgress(This->protocol_sink, status_code, status_text);
981         break;
982
983     case BINDSTATUS_MIMETYPEAVAILABLE:
984         mime_available(This, status_text, FALSE);
985         break;
986
987     case BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE:
988         mime_available(This, status_text, TRUE);
989         break;
990
991     default:
992         FIXME("unsupported ulStatusCode %u\n", status_code);
993     }
994 }
995
996 typedef struct {
997     task_header_t header;
998
999     ULONG status_code;
1000     LPWSTR status_text;
1001 } on_progress_task_t;
1002
1003 static void on_progress_proc(BindProtocol *This, task_header_t *t)
1004 {
1005     on_progress_task_t *task = (on_progress_task_t*)t;
1006
1007     report_progress(This, task->status_code, task->status_text);
1008
1009     heap_free(task->status_text);
1010     heap_free(task);
1011 }
1012
1013 static HRESULT WINAPI BPInternetProtocolSink_ReportProgress(IInternetProtocolSink *iface,
1014         ULONG ulStatusCode, LPCWSTR szStatusText)
1015 {
1016     BindProtocol *This = PROTSINK_THIS(iface);
1017
1018     TRACE("(%p)->(%u %s)\n", This, ulStatusCode, debugstr_w(szStatusText));
1019
1020     if(do_direct_notif(This)) {
1021         report_progress(This, ulStatusCode, szStatusText);
1022     }else {
1023         on_progress_task_t *task;
1024
1025         task = heap_alloc(sizeof(on_progress_task_t));
1026
1027         task->status_code = ulStatusCode;
1028         task->status_text = heap_strdupW(szStatusText);
1029
1030         push_task(This, &task->header, on_progress_proc);
1031     }
1032
1033     return S_OK;
1034 }
1035
1036 static HRESULT report_data(BindProtocol *This, DWORD bscf, ULONG progress, ULONG progress_max)
1037 {
1038     if(!This->protocol_sink)
1039         return S_OK;
1040
1041     if((This->pi & PI_MIMEVERIFICATION) && !This->reported_mime) {
1042         BYTE buf[BUFFER_SIZE];
1043         DWORD read = 0;
1044         LPWSTR mime;
1045         HRESULT hres;
1046
1047         do {
1048             read = 0;
1049             hres = IInternetProtocol_Read(This->protocol, buf,
1050                     sizeof(buf)-This->buf_size, &read);
1051             if(FAILED(hres) && hres != E_PENDING)
1052                 return hres;
1053
1054             if(!This->buf) {
1055                 This->buf = heap_alloc(BUFFER_SIZE);
1056                 if(!This->buf)
1057                     return E_OUTOFMEMORY;
1058             }else if(read + This->buf_size > BUFFER_SIZE) {
1059                 BYTE *tmp;
1060
1061                 tmp = heap_realloc(This->buf, read+This->buf_size);
1062                 if(!tmp)
1063                     return E_OUTOFMEMORY;
1064                 This->buf = tmp;
1065             }
1066
1067             memcpy(This->buf+This->buf_size, buf, read);
1068             This->buf_size += read;
1069         }while(This->buf_size < MIME_TEST_SIZE && hres == S_OK);
1070
1071         if(This->buf_size < MIME_TEST_SIZE && hres != S_FALSE)
1072             return S_OK;
1073
1074         bscf = BSCF_FIRSTDATANOTIFICATION;
1075         if(hres == S_FALSE)
1076             bscf |= BSCF_LASTDATANOTIFICATION|BSCF_DATAFULLYAVAILABLE;
1077
1078         if(!This->reported_mime) {
1079             hres = FindMimeFromData(NULL, This->url, This->buf, min(This->buf_size, MIME_TEST_SIZE),
1080                     This->mime, 0, &mime, 0);
1081             if(FAILED(hres))
1082                 return hres;
1083
1084             mime_available(This, mime, TRUE);
1085             CoTaskMemFree(mime);
1086         }
1087     }
1088
1089     if(!This->protocol_sink)
1090         return S_OK;
1091
1092     return IInternetProtocolSink_ReportData(This->protocol_sink, bscf, progress, progress_max);
1093 }
1094
1095 typedef struct {
1096     task_header_t header;
1097     DWORD bscf;
1098     ULONG progress;
1099     ULONG progress_max;
1100 } report_data_task_t;
1101
1102 static void report_data_proc(BindProtocol *This, task_header_t *t)
1103 {
1104     report_data_task_t *task = (report_data_task_t*)t;
1105
1106     report_data(This, task->bscf, task->progress, task->progress_max);
1107     heap_free(task);
1108 }
1109
1110 static HRESULT WINAPI BPInternetProtocolSink_ReportData(IInternetProtocolSink *iface,
1111         DWORD grfBSCF, ULONG ulProgress, ULONG ulProgressMax)
1112 {
1113     BindProtocol *This = PROTSINK_THIS(iface);
1114
1115     TRACE("(%p)->(%d %u %u)\n", This, grfBSCF, ulProgress, ulProgressMax);
1116
1117     if(!This->protocol_sink)
1118         return S_OK;
1119
1120     if(!do_direct_notif(This)) {
1121         report_data_task_t *task;
1122
1123         task = heap_alloc(sizeof(report_data_task_t));
1124         if(!task)
1125             return E_OUTOFMEMORY;
1126
1127         task->bscf = grfBSCF;
1128         task->progress = ulProgress;
1129         task->progress_max = ulProgressMax;
1130
1131         push_task(This, &task->header, report_data_proc);
1132         return S_OK;
1133     }
1134
1135     return report_data(This, grfBSCF, ulProgress, ulProgressMax);
1136 }
1137
1138 typedef struct {
1139     task_header_t header;
1140
1141     HRESULT hres;
1142     DWORD err;
1143     LPWSTR str;
1144 } report_result_task_t;
1145
1146 static void report_result_proc(BindProtocol *This, task_header_t *t)
1147 {
1148     report_result_task_t *task = (report_result_task_t*)t;
1149
1150     if(This->protocol_sink)
1151         IInternetProtocolSink_ReportResult(This->protocol_sink, task->hres, task->err, task->str);
1152
1153     heap_free(task->str);
1154     heap_free(task);
1155 }
1156
1157 static HRESULT WINAPI BPInternetProtocolSink_ReportResult(IInternetProtocolSink *iface,
1158         HRESULT hrResult, DWORD dwError, LPCWSTR szResult)
1159 {
1160     BindProtocol *This = PROTSINK_THIS(iface);
1161
1162     TRACE("(%p)->(%08x %d %s)\n", This, hrResult, dwError, debugstr_w(szResult));
1163
1164     if(!This->protocol_sink)
1165         return E_FAIL;
1166
1167     This->reported_result = TRUE;
1168
1169     if(!do_direct_notif(This)) {
1170         report_result_task_t *task;
1171
1172         task = heap_alloc(sizeof(report_result_task_t));
1173         if(!task)
1174             return E_OUTOFMEMORY;
1175
1176         task->hres = hrResult;
1177         task->err = dwError;
1178         task->str = heap_strdupW(szResult);
1179
1180         push_task(This, &task->header, report_result_proc);
1181         return S_OK;
1182     }
1183
1184     return IInternetProtocolSink_ReportResult(This->protocol_sink, hrResult, dwError, szResult);
1185 }
1186
1187 #undef PROTSINK_THIS
1188
1189 static const IInternetProtocolSinkVtbl InternetProtocolSinkVtbl = {
1190     BPInternetProtocolSink_QueryInterface,
1191     BPInternetProtocolSink_AddRef,
1192     BPInternetProtocolSink_Release,
1193     BPInternetProtocolSink_Switch,
1194     BPInternetProtocolSink_ReportProgress,
1195     BPInternetProtocolSink_ReportData,
1196     BPInternetProtocolSink_ReportResult
1197 };
1198
1199 #define INETINFO_THIS(iface) DEFINE_THIS(BindProtocol, IWinInetHttpInfo, iface)
1200
1201 static HRESULT WINAPI WinInetHttpInfo_QueryInterface(IWinInetHttpInfo *iface, REFIID riid, void **ppv)
1202 {
1203     BindProtocol *This = INETINFO_THIS(iface);
1204     return IInternetProtocol_QueryInterface(PROTOCOL(This), riid, ppv);
1205 }
1206
1207 static ULONG WINAPI WinInetHttpInfo_AddRef(IWinInetHttpInfo *iface)
1208 {
1209     BindProtocol *This = INETINFO_THIS(iface);
1210     return IInternetProtocol_AddRef(PROTOCOL(This));
1211 }
1212
1213 static ULONG WINAPI WinInetHttpInfo_Release(IWinInetHttpInfo *iface)
1214 {
1215     BindProtocol *This = INETINFO_THIS(iface);
1216     return IInternetProtocol_Release(PROTOCOL(This));
1217 }
1218
1219 static HRESULT WINAPI WinInetHttpInfo_QueryOption(IWinInetHttpInfo *iface, DWORD dwOption,
1220         void *pBuffer, DWORD *pcbBuffer)
1221 {
1222     BindProtocol *This = INETINFO_THIS(iface);
1223     FIXME("(%p)->(%x %p %p)\n", This, dwOption, pBuffer, pcbBuffer);
1224     return E_NOTIMPL;
1225 }
1226
1227 static HRESULT WINAPI WinInetHttpInfo_QueryInfo(IWinInetHttpInfo *iface, DWORD dwOption,
1228         void *pBuffer, DWORD *pcbBuffer, DWORD *pdwFlags, DWORD *pdwReserved)
1229 {
1230     BindProtocol *This = INETINFO_THIS(iface);
1231     FIXME("(%p)->(%x %p %p %p %p)\n", This, dwOption, pBuffer, pcbBuffer, pdwFlags, pdwReserved);
1232     return E_NOTIMPL;
1233 }
1234
1235 #undef INETINFO_THIS
1236
1237 static const IWinInetHttpInfoVtbl WinInetHttpInfoVtbl = {
1238     WinInetHttpInfo_QueryInterface,
1239     WinInetHttpInfo_AddRef,
1240     WinInetHttpInfo_Release,
1241     WinInetHttpInfo_QueryOption,
1242     WinInetHttpInfo_QueryInfo
1243 };
1244
1245 #define SERVPROV_THIS(iface) DEFINE_THIS(BindProtocol, ServiceProvider, iface)
1246
1247 static HRESULT WINAPI BPServiceProvider_QueryInterface(IServiceProvider *iface,
1248         REFIID riid, void **ppv)
1249 {
1250     BindProtocol *This = SERVPROV_THIS(iface);
1251     return IInternetProtocol_QueryInterface(PROTOCOL(This), riid, ppv);
1252 }
1253
1254 static ULONG WINAPI BPServiceProvider_AddRef(IServiceProvider *iface)
1255 {
1256     BindProtocol *This = SERVPROV_THIS(iface);
1257     return IInternetProtocol_AddRef(PROTOCOL(This));
1258 }
1259
1260 static ULONG WINAPI BPServiceProvider_Release(IServiceProvider *iface)
1261 {
1262     BindProtocol *This = SERVPROV_THIS(iface);
1263     return IInternetProtocol_Release(PROTOCOL(This));
1264 }
1265
1266 static HRESULT WINAPI BPServiceProvider_QueryService(IServiceProvider *iface,
1267         REFGUID guidService, REFIID riid, void **ppv)
1268 {
1269     BindProtocol *This = SERVPROV_THIS(iface);
1270
1271     TRACE("(%p)->(%s %s %p)\n", This, debugstr_guid(guidService), debugstr_guid(riid), ppv);
1272
1273     if(!This->service_provider)
1274         return E_NOINTERFACE;
1275
1276     return IServiceProvider_QueryService(This->service_provider, guidService, riid, ppv);
1277 }
1278
1279 #undef SERVPROV_THIS
1280
1281 static const IServiceProviderVtbl ServiceProviderVtbl = {
1282     BPServiceProvider_QueryInterface,
1283     BPServiceProvider_AddRef,
1284     BPServiceProvider_Release,
1285     BPServiceProvider_QueryService
1286 };
1287
1288 HRESULT create_binding_protocol(LPCWSTR url, BOOL from_urlmon, IInternetProtocol **protocol)
1289 {
1290     BindProtocol *ret = heap_alloc_zero(sizeof(BindProtocol));
1291
1292     ret->lpIInternetProtocolVtbl        = &BindProtocolVtbl;
1293     ret->lpInternetBindInfoVtbl         = &InternetBindInfoVtbl;
1294     ret->lpInternetPriorityVtbl         = &InternetPriorityVtbl;
1295     ret->lpServiceProviderVtbl          = &ServiceProviderVtbl;
1296     ret->lpIInternetProtocolSinkVtbl    = &InternetProtocolSinkVtbl;
1297     ret->lpIInternetProtocolHandlerVtbl = &InternetProtocolHandlerVtbl;
1298     ret->lpIWinInetHttpInfoVtbl         = &WinInetHttpInfoVtbl;
1299
1300     ret->ref = 1;
1301     ret->from_urlmon = from_urlmon;
1302     ret->apartment_thread = GetCurrentThreadId();
1303     ret->notif_hwnd = get_notif_hwnd();
1304     ret->protocol_handler = PROTOCOLHANDLER(ret);
1305     InitializeCriticalSection(&ret->section);
1306
1307     URLMON_LockModule();
1308
1309     *protocol = PROTOCOL(ret);
1310     return S_OK;
1311 }