server: Move socket async activation to sock_poll_event.
[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);
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     FIXME("(%p)->(%08x %08x)\n", This, hrReason, dwOptions);
427     return E_NOTIMPL;
428 }
429
430 static HRESULT WINAPI BindProtocol_Terminate(IInternetProtocol *iface, DWORD dwOptions)
431 {
432     BindProtocol *This = PROTOCOL_THIS(iface);
433
434     TRACE("(%p)->(%08x)\n", This, dwOptions);
435
436     return IInternetProtocol_Terminate(This->protocol_handler, dwOptions);
437 }
438
439 static HRESULT WINAPI BindProtocol_Suspend(IInternetProtocol *iface)
440 {
441     BindProtocol *This = PROTOCOL_THIS(iface);
442     FIXME("(%p)\n", This);
443     return E_NOTIMPL;
444 }
445
446 static HRESULT WINAPI BindProtocol_Resume(IInternetProtocol *iface)
447 {
448     BindProtocol *This = PROTOCOL_THIS(iface);
449     FIXME("(%p)\n", This);
450     return E_NOTIMPL;
451 }
452
453 static HRESULT WINAPI BindProtocol_Read(IInternetProtocol *iface, void *pv,
454         ULONG cb, ULONG *pcbRead)
455 {
456     BindProtocol *This = PROTOCOL_THIS(iface);
457
458     TRACE("(%p)->(%p %u %p)\n", This, pv, cb, pcbRead);
459
460     if(pcbRead)
461         *pcbRead = 0;
462     return IInternetProtocol_Read(This->protocol_handler, pv, cb, pcbRead);
463 }
464
465 static HRESULT WINAPI BindProtocol_Seek(IInternetProtocol *iface, LARGE_INTEGER dlibMove,
466         DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
467 {
468     BindProtocol *This = PROTOCOL_THIS(iface);
469     FIXME("(%p)->(%d %d %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
470     return E_NOTIMPL;
471 }
472
473 static HRESULT WINAPI BindProtocol_LockRequest(IInternetProtocol *iface, DWORD dwOptions)
474 {
475     BindProtocol *This = PROTOCOL_THIS(iface);
476
477     TRACE("(%p)->(%08x)\n", This, dwOptions);
478
479     return IInternetProtocol_LockRequest(This->protocol_handler, dwOptions);
480 }
481
482 static HRESULT WINAPI BindProtocol_UnlockRequest(IInternetProtocol *iface)
483 {
484     BindProtocol *This = PROTOCOL_THIS(iface);
485
486     TRACE("(%p)\n", This);
487
488     return IInternetProtocol_UnlockRequest(This->protocol_handler);
489 }
490
491 void set_binding_sink(IInternetProtocol *bind_protocol, IInternetProtocolSink *sink)
492 {
493     BindProtocol *This = PROTOCOL_THIS(bind_protocol);
494     IInternetProtocolSink *prev_sink;
495     IServiceProvider *service_provider = NULL;
496
497     if(sink)
498         IInternetProtocolSink_AddRef(sink);
499     prev_sink = InterlockedExchangePointer((void**)&This->protocol_sink, sink);
500     if(prev_sink)
501         IInternetProtocolSink_Release(prev_sink);
502
503     if(sink)
504         IInternetProtocolSink_QueryInterface(sink, &IID_IServiceProvider, (void**)&service_provider);
505     service_provider = InterlockedExchangePointer((void**)&This->service_provider, service_provider);
506     if(service_provider)
507         IServiceProvider_Release(service_provider);
508 }
509
510 IWinInetInfo *get_wininet_info(IInternetProtocol *bind_protocol)
511 {
512     BindProtocol *This = PROTOCOL_THIS(bind_protocol);
513
514     return This->wininet_info;
515 }
516
517 #undef PROTOCOL_THIS
518
519 static const IInternetProtocolVtbl BindProtocolVtbl = {
520     BindProtocol_QueryInterface,
521     BindProtocol_AddRef,
522     BindProtocol_Release,
523     BindProtocol_Start,
524     BindProtocol_Continue,
525     BindProtocol_Abort,
526     BindProtocol_Terminate,
527     BindProtocol_Suspend,
528     BindProtocol_Resume,
529     BindProtocol_Read,
530     BindProtocol_Seek,
531     BindProtocol_LockRequest,
532     BindProtocol_UnlockRequest
533 };
534
535 #define PROTOCOLHANDLER_THIS(iface) DEFINE_THIS(BindProtocol, IInternetProtocolHandler, iface)
536
537 static HRESULT WINAPI ProtocolHandler_QueryInterface(IInternetProtocol *iface, REFIID riid, void **ppv)
538 {
539     ERR("should not be called\n");
540     return E_NOINTERFACE;
541 }
542
543 static ULONG WINAPI ProtocolHandler_AddRef(IInternetProtocol *iface)
544 {
545     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
546     return IInternetProtocol_AddRef(PROTOCOL(This));
547 }
548
549 static ULONG WINAPI ProtocolHandler_Release(IInternetProtocol *iface)
550 {
551     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
552     return IInternetProtocol_Release(PROTOCOL(This));
553 }
554
555 static HRESULT WINAPI ProtocolHandler_Start(IInternetProtocol *iface, LPCWSTR szUrl,
556         IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo,
557         DWORD grfPI, HANDLE_PTR dwReserved)
558 {
559     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
560     IInternetProtocol *protocol = NULL;
561     IInternetPriority *priority;
562     IServiceProvider *service_provider;
563     BOOL urlmon_protocol = FALSE;
564     CLSID clsid = IID_NULL;
565     LPOLESTR clsid_str;
566     HRESULT hres;
567
568     TRACE("(%p)->(%s %p %p %08x %lx)\n", This, debugstr_w(szUrl), pOIProtSink,
569             pOIBindInfo, grfPI, dwReserved);
570
571     if(!szUrl || !pOIProtSink || !pOIBindInfo)
572         return E_INVALIDARG;
573
574     This->pi = grfPI;
575     This->url = heap_strdupW(szUrl);
576
577     hres = IInternetProtocolSink_QueryInterface(pOIProtSink, &IID_IServiceProvider,
578                                                 (void**)&service_provider);
579     if(SUCCEEDED(hres)) {
580         /* FIXME: What's protocol CLSID here? */
581         IServiceProvider_QueryService(service_provider, &IID_IInternetProtocol,
582                                       &IID_IInternetProtocol, (void**)&protocol);
583         IServiceProvider_Release(service_provider);
584     }
585
586     if(!protocol) {
587         IClassFactory *cf;
588         IUnknown *unk;
589
590         hres = get_protocol_handler(szUrl, &clsid, &urlmon_protocol, &cf);
591         if(FAILED(hres))
592             return hres;
593
594         if(This->from_urlmon) {
595             hres = IClassFactory_CreateInstance(cf, NULL, &IID_IInternetProtocol, (void**)&protocol);
596             IClassFactory_Release(cf);
597             if(FAILED(hres))
598                 return hres;
599         }else {
600             hres = IClassFactory_CreateInstance(cf, (IUnknown*)BINDINFO(This),
601                     &IID_IUnknown, (void**)&unk);
602             IClassFactory_Release(cf);
603             if(FAILED(hres))
604                 return hres;
605
606             hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocol, (void**)&protocol);
607             IUnknown_Release(unk);
608             if(FAILED(hres))
609                 return hres;
610         }
611     }
612
613     StringFromCLSID(&clsid, &clsid_str);
614     IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_PROTOCOLCLASSID, clsid_str);
615     CoTaskMemFree(clsid_str);
616
617     This->protocol = protocol;
618
619     if(urlmon_protocol)
620         IInternetProtocol_QueryInterface(protocol, &IID_IWinInetInfo, (void**)&This->wininet_info);
621
622     IInternetBindInfo_AddRef(pOIBindInfo);
623     This->bind_info = pOIBindInfo;
624
625     set_binding_sink(PROTOCOL(This), pOIProtSink);
626
627     hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetPriority, (void**)&priority);
628     if(SUCCEEDED(hres)) {
629         IInternetPriority_SetPriority(priority, This->priority);
630         IInternetPriority_Release(priority);
631     }
632
633     return IInternetProtocol_Start(protocol, szUrl, PROTSINK(This), BINDINFO(This), 0, 0);
634 }
635
636 static HRESULT WINAPI ProtocolHandler_Continue(IInternetProtocol *iface, PROTOCOLDATA *pProtocolData)
637 {
638     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
639     HRESULT hres;
640
641     TRACE("(%p)->(%p)\n", This, pProtocolData);
642
643     hres = IInternetProtocol_Continue(This->protocol, pProtocolData);
644
645     heap_free(pProtocolData);
646     return hres;
647 }
648
649 static HRESULT WINAPI ProtocolHandler_Abort(IInternetProtocol *iface, HRESULT hrReason,
650         DWORD dwOptions)
651 {
652     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
653     FIXME("(%p)->(%08x %08x)\n", This, hrReason, dwOptions);
654     return E_NOTIMPL;
655 }
656
657 static HRESULT WINAPI ProtocolHandler_Terminate(IInternetProtocol *iface, DWORD dwOptions)
658 {
659     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
660
661     TRACE("(%p)->(%08x)\n", This, dwOptions);
662
663     if(!This->reported_result)
664         return E_FAIL;
665
666     IInternetProtocol_Terminate(This->protocol, 0);
667
668     if(This->filter_proxy) {
669         IInternetProtocol_Release(PROTOCOL(This->filter_proxy));
670         This->filter_proxy = NULL;
671     }
672
673     set_binding_sink(PROTOCOL(This), NULL);
674
675     if(This->bind_info) {
676         IInternetBindInfo_Release(This->bind_info);
677         This->bind_info = NULL;
678     }
679
680     return S_OK;
681 }
682
683 static HRESULT WINAPI ProtocolHandler_Suspend(IInternetProtocol *iface)
684 {
685     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
686     FIXME("(%p)\n", This);
687     return E_NOTIMPL;
688 }
689
690 static HRESULT WINAPI ProtocolHandler_Resume(IInternetProtocol *iface)
691 {
692     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
693     FIXME("(%p)\n", This);
694     return E_NOTIMPL;
695 }
696
697 static HRESULT WINAPI ProtocolHandler_Read(IInternetProtocol *iface, void *pv,
698         ULONG cb, ULONG *pcbRead)
699 {
700     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
701     ULONG read = 0;
702     HRESULT hres = S_OK;
703
704     TRACE("(%p)->(%p %u %p)\n", This, pv, cb, pcbRead);
705
706     if(This->buf_size) {
707         read = min(cb, This->buf_size);
708         memcpy(pv, This->buf, read);
709
710         if(read == This->buf_size) {
711             heap_free(This->buf);
712             This->buf = NULL;
713         }else {
714             memmove(This->buf, This->buf+cb, This->buf_size-cb);
715         }
716
717         This->buf_size -= read;
718     }
719
720     if(read < cb) {
721         ULONG cread = 0;
722
723         hres = IInternetProtocol_Read(This->protocol, (BYTE*)pv+read, cb-read, &cread);
724         read += cread;
725     }
726
727     *pcbRead = read;
728     return hres;
729 }
730
731 static HRESULT WINAPI ProtocolHandler_Seek(IInternetProtocol *iface, LARGE_INTEGER dlibMove,
732         DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
733 {
734     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
735     FIXME("(%p)->(%d %d %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
736     return E_NOTIMPL;
737 }
738
739 static HRESULT WINAPI ProtocolHandler_LockRequest(IInternetProtocol *iface, DWORD dwOptions)
740 {
741     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
742
743     TRACE("(%p)->(%08x)\n", This, dwOptions);
744
745     return IInternetProtocol_LockRequest(This->protocol, dwOptions);
746 }
747
748 static HRESULT WINAPI ProtocolHandler_UnlockRequest(IInternetProtocol *iface)
749 {
750     BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
751
752     TRACE("(%p)\n", This);
753
754     return IInternetProtocol_UnlockRequest(This->protocol);
755 }
756
757 #undef PROTOCOL_THIS
758
759 static const IInternetProtocolVtbl InternetProtocolHandlerVtbl = {
760     ProtocolHandler_QueryInterface,
761     ProtocolHandler_AddRef,
762     ProtocolHandler_Release,
763     ProtocolHandler_Start,
764     ProtocolHandler_Continue,
765     ProtocolHandler_Abort,
766     ProtocolHandler_Terminate,
767     ProtocolHandler_Suspend,
768     ProtocolHandler_Resume,
769     ProtocolHandler_Read,
770     ProtocolHandler_Seek,
771     ProtocolHandler_LockRequest,
772     ProtocolHandler_UnlockRequest
773 };
774
775 #define BINDINFO_THIS(iface) DEFINE_THIS(BindProtocol, InternetBindInfo, iface)
776
777 static HRESULT WINAPI BindInfo_QueryInterface(IInternetBindInfo *iface,
778         REFIID riid, void **ppv)
779 {
780     BindProtocol *This = BINDINFO_THIS(iface);
781     return IInternetProtocol_QueryInterface(PROTOCOL(This), riid, ppv);
782 }
783
784 static ULONG WINAPI BindInfo_AddRef(IInternetBindInfo *iface)
785 {
786     BindProtocol *This = BINDINFO_THIS(iface);
787     return IBinding_AddRef(PROTOCOL(This));
788 }
789
790 static ULONG WINAPI BindInfo_Release(IInternetBindInfo *iface)
791 {
792     BindProtocol *This = BINDINFO_THIS(iface);
793     return IBinding_Release(PROTOCOL(This));
794 }
795
796 static HRESULT WINAPI BindInfo_GetBindInfo(IInternetBindInfo *iface,
797         DWORD *grfBINDF, BINDINFO *pbindinfo)
798 {
799     BindProtocol *This = BINDINFO_THIS(iface);
800     HRESULT hres;
801
802     TRACE("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo);
803
804     hres = IInternetBindInfo_GetBindInfo(This->bind_info, grfBINDF, pbindinfo);
805     if(FAILED(hres)) {
806         WARN("GetBindInfo failed: %08x\n", hres);
807         return hres;
808     }
809
810     *grfBINDF |= BINDF_FROMURLMON;
811     return hres;
812 }
813
814 static HRESULT WINAPI BindInfo_GetBindString(IInternetBindInfo *iface,
815         ULONG ulStringType, LPOLESTR *ppwzStr, ULONG cEl, ULONG *pcElFetched)
816 {
817     BindProtocol *This = BINDINFO_THIS(iface);
818
819     TRACE("(%p)->(%d %p %d %p)\n", This, ulStringType, ppwzStr, cEl, pcElFetched);
820
821     return IInternetBindInfo_GetBindString(This->bind_info, ulStringType, ppwzStr, cEl, pcElFetched);
822 }
823
824 #undef BINDFO_THIS
825
826 static const IInternetBindInfoVtbl InternetBindInfoVtbl = {
827     BindInfo_QueryInterface,
828     BindInfo_AddRef,
829     BindInfo_Release,
830     BindInfo_GetBindInfo,
831     BindInfo_GetBindString
832 };
833
834 #define PRIORITY_THIS(iface) DEFINE_THIS(BindProtocol, InternetPriority, iface)
835
836 static HRESULT WINAPI InternetPriority_QueryInterface(IInternetPriority *iface,
837         REFIID riid, void **ppv)
838 {
839     BindProtocol *This = PRIORITY_THIS(iface);
840     return IInternetProtocol_QueryInterface(PROTOCOL(This), riid, ppv);
841 }
842
843 static ULONG WINAPI InternetPriority_AddRef(IInternetPriority *iface)
844 {
845     BindProtocol *This = PRIORITY_THIS(iface);
846     return IInternetProtocol_AddRef(PROTOCOL(This));
847 }
848
849 static ULONG WINAPI InternetPriority_Release(IInternetPriority *iface)
850 {
851     BindProtocol *This = PRIORITY_THIS(iface);
852     return IInternetProtocol_Release(PROTOCOL(This));
853 }
854
855 static HRESULT WINAPI InternetPriority_SetPriority(IInternetPriority *iface, LONG nPriority)
856 {
857     BindProtocol *This = PRIORITY_THIS(iface);
858
859     TRACE("(%p)->(%d)\n", This, nPriority);
860
861     This->priority = nPriority;
862     return S_OK;
863 }
864
865 static HRESULT WINAPI InternetPriority_GetPriority(IInternetPriority *iface, LONG *pnPriority)
866 {
867     BindProtocol *This = PRIORITY_THIS(iface);
868
869     TRACE("(%p)->(%p)\n", This, pnPriority);
870
871     *pnPriority = This->priority;
872     return S_OK;
873 }
874
875 #undef PRIORITY_THIS
876
877 static const IInternetPriorityVtbl InternetPriorityVtbl = {
878     InternetPriority_QueryInterface,
879     InternetPriority_AddRef,
880     InternetPriority_Release,
881     InternetPriority_SetPriority,
882     InternetPriority_GetPriority
883
884 };
885
886 #define PROTSINK_THIS(iface) DEFINE_THIS(BindProtocol, IInternetProtocolSink, iface)
887
888 static HRESULT WINAPI BPInternetProtocolSink_QueryInterface(IInternetProtocolSink *iface,
889         REFIID riid, void **ppv)
890 {
891     BindProtocol *This = PROTSINK_THIS(iface);
892     return IInternetProtocol_QueryInterface(PROTOCOL(This), riid, ppv);
893 }
894
895 static ULONG WINAPI BPInternetProtocolSink_AddRef(IInternetProtocolSink *iface)
896 {
897     BindProtocol *This = PROTSINK_THIS(iface);
898     return IInternetProtocol_AddRef(PROTOCOL(This));
899 }
900
901 static ULONG WINAPI BPInternetProtocolSink_Release(IInternetProtocolSink *iface)
902 {
903     BindProtocol *This = PROTSINK_THIS(iface);
904     return IInternetProtocol_Release(PROTOCOL(This));
905 }
906
907 typedef struct {
908     task_header_t header;
909     PROTOCOLDATA *data;
910 } switch_task_t;
911
912 static void switch_proc(BindProtocol *bind, task_header_t *t)
913 {
914     switch_task_t *task = (switch_task_t*)t;
915
916     IInternetProtocol_Continue(bind->protocol_handler, task->data);
917
918     heap_free(task);
919 }
920
921 static HRESULT WINAPI BPInternetProtocolSink_Switch(IInternetProtocolSink *iface,
922         PROTOCOLDATA *pProtocolData)
923 {
924     BindProtocol *This = PROTSINK_THIS(iface);
925     PROTOCOLDATA *data;
926
927     TRACE("(%p)->(%p)\n", This, pProtocolData);
928
929     TRACE("flags %x state %x data %p cb %u\n", pProtocolData->grfFlags, pProtocolData->dwState,
930           pProtocolData->pData, pProtocolData->cbData);
931
932     data = heap_alloc(sizeof(PROTOCOLDATA));
933     if(!data)
934         return E_OUTOFMEMORY;
935     memcpy(data, pProtocolData, sizeof(PROTOCOLDATA));
936
937     if(!do_direct_notif(This)) {
938         switch_task_t *task;
939
940         task = heap_alloc(sizeof(switch_task_t));
941         if(!task)
942             return E_OUTOFMEMORY;
943
944         task->data = data;
945
946         push_task(This, &task->header, switch_proc);
947         return S_OK;
948     }
949
950     if(!This->protocol_sink) {
951         IInternetProtocol_Continue(This->protocol_handler, data);
952         return S_OK;
953     }
954
955     return IInternetProtocolSink_Switch(This->protocol_sink, data);
956 }
957
958 static void report_progress(BindProtocol *This, ULONG status_code, LPCWSTR status_text)
959 {
960     switch(status_code) {
961     case BINDSTATUS_FINDINGRESOURCE:
962     case BINDSTATUS_CONNECTING:
963     case BINDSTATUS_REDIRECTING:
964     case BINDSTATUS_BEGINDOWNLOADDATA:
965     case BINDSTATUS_SENDINGREQUEST:
966     case BINDSTATUS_CACHEFILENAMEAVAILABLE:
967     case BINDSTATUS_DIRECTBIND:
968     case BINDSTATUS_ACCEPTRANGES:
969         if(This->protocol_sink)
970             IInternetProtocolSink_ReportProgress(This->protocol_sink, status_code, status_text);
971         break;
972
973     case BINDSTATUS_MIMETYPEAVAILABLE:
974         mime_available(This, status_text, FALSE);
975         break;
976
977     case BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE:
978         mime_available(This, status_text, TRUE);
979         break;
980
981     default:
982         FIXME("unsupported ulStatusCode %u\n", status_code);
983     }
984 }
985
986 typedef struct {
987     task_header_t header;
988
989     ULONG status_code;
990     LPWSTR status_text;
991 } on_progress_task_t;
992
993 static void on_progress_proc(BindProtocol *This, task_header_t *t)
994 {
995     on_progress_task_t *task = (on_progress_task_t*)t;
996
997     report_progress(This, task->status_code, task->status_text);
998
999     heap_free(task->status_text);
1000     heap_free(task);
1001 }
1002
1003 static HRESULT WINAPI BPInternetProtocolSink_ReportProgress(IInternetProtocolSink *iface,
1004         ULONG ulStatusCode, LPCWSTR szStatusText)
1005 {
1006     BindProtocol *This = PROTSINK_THIS(iface);
1007
1008     TRACE("(%p)->(%u %s)\n", This, ulStatusCode, debugstr_w(szStatusText));
1009
1010     if(do_direct_notif(This)) {
1011         report_progress(This, ulStatusCode, szStatusText);
1012     }else {
1013         on_progress_task_t *task;
1014
1015         task = heap_alloc(sizeof(on_progress_task_t));
1016
1017         task->status_code = ulStatusCode;
1018         task->status_text = heap_strdupW(szStatusText);
1019
1020         push_task(This, &task->header, on_progress_proc);
1021     }
1022
1023     return S_OK;
1024 }
1025
1026 static HRESULT report_data(BindProtocol *This, DWORD bscf, ULONG progress, ULONG progress_max)
1027 {
1028     if(!This->protocol_sink)
1029         return S_OK;
1030
1031     if((This->pi & PI_MIMEVERIFICATION) && !This->reported_mime) {
1032         BYTE buf[BUFFER_SIZE];
1033         DWORD read = 0;
1034         LPWSTR mime;
1035         HRESULT hres;
1036
1037         do {
1038             read = 0;
1039             hres = IInternetProtocol_Read(This->protocol, buf,
1040                     sizeof(buf)-This->buf_size, &read);
1041             if(FAILED(hres) && hres != E_PENDING)
1042                 return hres;
1043
1044             if(!This->buf) {
1045                 This->buf = heap_alloc(BUFFER_SIZE);
1046                 if(!This->buf)
1047                     return E_OUTOFMEMORY;
1048             }else if(read + This->buf_size > BUFFER_SIZE) {
1049                 BYTE *tmp;
1050
1051                 tmp = heap_realloc(This->buf, read+This->buf_size);
1052                 if(!tmp)
1053                     return E_OUTOFMEMORY;
1054                 This->buf = tmp;
1055             }
1056
1057             memcpy(This->buf+This->buf_size, buf, read);
1058             This->buf_size += read;
1059         }while(This->buf_size < MIME_TEST_SIZE && hres == S_OK);
1060
1061         if(This->buf_size < MIME_TEST_SIZE && hres != S_FALSE)
1062             return S_OK;
1063
1064         bscf = BSCF_FIRSTDATANOTIFICATION;
1065         if(hres == S_FALSE)
1066             bscf |= BSCF_LASTDATANOTIFICATION|BSCF_DATAFULLYAVAILABLE;
1067
1068         if(!This->reported_mime) {
1069             hres = FindMimeFromData(NULL, This->url, This->buf, min(This->buf_size, MIME_TEST_SIZE),
1070                     This->mime, 0, &mime, 0);
1071             if(FAILED(hres))
1072                 return hres;
1073
1074             mime_available(This, mime, TRUE);
1075             CoTaskMemFree(mime);
1076         }
1077     }
1078
1079     if(!This->protocol_sink)
1080         return S_OK;
1081
1082     return IInternetProtocolSink_ReportData(This->protocol_sink, bscf, progress, progress_max);
1083 }
1084
1085 typedef struct {
1086     task_header_t header;
1087     DWORD bscf;
1088     ULONG progress;
1089     ULONG progress_max;
1090 } report_data_task_t;
1091
1092 static void report_data_proc(BindProtocol *This, task_header_t *t)
1093 {
1094     report_data_task_t *task = (report_data_task_t*)t;
1095
1096     report_data(This, task->bscf, task->progress, task->progress_max);
1097     heap_free(task);
1098 }
1099
1100 static HRESULT WINAPI BPInternetProtocolSink_ReportData(IInternetProtocolSink *iface,
1101         DWORD grfBSCF, ULONG ulProgress, ULONG ulProgressMax)
1102 {
1103     BindProtocol *This = PROTSINK_THIS(iface);
1104
1105     TRACE("(%p)->(%d %u %u)\n", This, grfBSCF, ulProgress, ulProgressMax);
1106
1107     if(!This->protocol_sink)
1108         return S_OK;
1109
1110     if(!do_direct_notif(This)) {
1111         report_data_task_t *task;
1112
1113         task = heap_alloc(sizeof(report_data_task_t));
1114         if(!task)
1115             return E_OUTOFMEMORY;
1116
1117         task->bscf = grfBSCF;
1118         task->progress = ulProgress;
1119         task->progress_max = ulProgressMax;
1120
1121         push_task(This, &task->header, report_data_proc);
1122         return S_OK;
1123     }
1124
1125     return report_data(This, grfBSCF, ulProgress, ulProgressMax);
1126 }
1127
1128 typedef struct {
1129     task_header_t header;
1130
1131     HRESULT hres;
1132     DWORD err;
1133     LPWSTR str;
1134 } report_result_task_t;
1135
1136 static void report_result_proc(BindProtocol *This, task_header_t *t)
1137 {
1138     report_result_task_t *task = (report_result_task_t*)t;
1139
1140     if(This->protocol_sink)
1141         IInternetProtocolSink_ReportResult(This->protocol_sink, task->hres, task->err, task->str);
1142
1143     heap_free(task->str);
1144     heap_free(task);
1145 }
1146
1147 static HRESULT WINAPI BPInternetProtocolSink_ReportResult(IInternetProtocolSink *iface,
1148         HRESULT hrResult, DWORD dwError, LPCWSTR szResult)
1149 {
1150     BindProtocol *This = PROTSINK_THIS(iface);
1151
1152     TRACE("(%p)->(%08x %d %s)\n", This, hrResult, dwError, debugstr_w(szResult));
1153
1154     if(!This->protocol_sink)
1155         return E_FAIL;
1156
1157     This->reported_result = TRUE;
1158
1159     if(!do_direct_notif(This)) {
1160         report_result_task_t *task;
1161
1162         task = heap_alloc(sizeof(report_result_task_t));
1163         if(!task)
1164             return E_OUTOFMEMORY;
1165
1166         task->hres = hrResult;
1167         task->err = dwError;
1168         task->str = heap_strdupW(szResult);
1169
1170         push_task(This, &task->header, report_result_proc);
1171         return S_OK;
1172     }
1173
1174     return IInternetProtocolSink_ReportResult(This->protocol_sink, hrResult, dwError, szResult);
1175 }
1176
1177 #undef PROTSINK_THIS
1178
1179 static const IInternetProtocolSinkVtbl InternetProtocolSinkVtbl = {
1180     BPInternetProtocolSink_QueryInterface,
1181     BPInternetProtocolSink_AddRef,
1182     BPInternetProtocolSink_Release,
1183     BPInternetProtocolSink_Switch,
1184     BPInternetProtocolSink_ReportProgress,
1185     BPInternetProtocolSink_ReportData,
1186     BPInternetProtocolSink_ReportResult
1187 };
1188
1189 #define INETINFO_THIS(iface) DEFINE_THIS(BindProtocol, IWinInetHttpInfo, iface)
1190
1191 static HRESULT WINAPI WinInetHttpInfo_QueryInterface(IWinInetHttpInfo *iface, REFIID riid, void **ppv)
1192 {
1193     BindProtocol *This = INETINFO_THIS(iface);
1194     return IInternetProtocol_QueryInterface(PROTOCOL(This), riid, ppv);
1195 }
1196
1197 static ULONG WINAPI WinInetHttpInfo_AddRef(IWinInetHttpInfo *iface)
1198 {
1199     BindProtocol *This = INETINFO_THIS(iface);
1200     return IInternetProtocol_AddRef(PROTOCOL(This));
1201 }
1202
1203 static ULONG WINAPI WinInetHttpInfo_Release(IWinInetHttpInfo *iface)
1204 {
1205     BindProtocol *This = INETINFO_THIS(iface);
1206     return IInternetProtocol_Release(PROTOCOL(This));
1207 }
1208
1209 static HRESULT WINAPI WinInetHttpInfo_QueryOption(IWinInetHttpInfo *iface, DWORD dwOption,
1210         void *pBuffer, DWORD *pcbBuffer)
1211 {
1212     BindProtocol *This = INETINFO_THIS(iface);
1213     FIXME("(%p)->(%x %p %p)\n", This, dwOption, pBuffer, pcbBuffer);
1214     return E_NOTIMPL;
1215 }
1216
1217 static HRESULT WINAPI WinInetHttpInfo_QueryInfo(IWinInetHttpInfo *iface, DWORD dwOption,
1218         void *pBuffer, DWORD *pcbBuffer, DWORD *pdwFlags, DWORD *pdwReserved)
1219 {
1220     BindProtocol *This = INETINFO_THIS(iface);
1221     FIXME("(%p)->(%x %p %p %p %p)\n", This, dwOption, pBuffer, pcbBuffer, pdwFlags, pdwReserved);
1222     return E_NOTIMPL;
1223 }
1224
1225 #undef INETINFO_THIS
1226
1227 static const IWinInetHttpInfoVtbl WinInetHttpInfoVtbl = {
1228     WinInetHttpInfo_QueryInterface,
1229     WinInetHttpInfo_AddRef,
1230     WinInetHttpInfo_Release,
1231     WinInetHttpInfo_QueryOption,
1232     WinInetHttpInfo_QueryInfo
1233 };
1234
1235 #define SERVPROV_THIS(iface) DEFINE_THIS(BindProtocol, ServiceProvider, iface)
1236
1237 static HRESULT WINAPI BPServiceProvider_QueryInterface(IServiceProvider *iface,
1238         REFIID riid, void **ppv)
1239 {
1240     BindProtocol *This = SERVPROV_THIS(iface);
1241     return IInternetProtocol_QueryInterface(PROTOCOL(This), riid, ppv);
1242 }
1243
1244 static ULONG WINAPI BPServiceProvider_AddRef(IServiceProvider *iface)
1245 {
1246     BindProtocol *This = SERVPROV_THIS(iface);
1247     return IInternetProtocol_AddRef(PROTOCOL(This));
1248 }
1249
1250 static ULONG WINAPI BPServiceProvider_Release(IServiceProvider *iface)
1251 {
1252     BindProtocol *This = SERVPROV_THIS(iface);
1253     return IInternetProtocol_Release(PROTOCOL(This));
1254 }
1255
1256 static HRESULT WINAPI BPServiceProvider_QueryService(IServiceProvider *iface,
1257         REFGUID guidService, REFIID riid, void **ppv)
1258 {
1259     BindProtocol *This = SERVPROV_THIS(iface);
1260
1261     TRACE("(%p)->(%s %s %p)\n", This, debugstr_guid(guidService), debugstr_guid(riid), ppv);
1262
1263     if(!This->service_provider)
1264         return E_NOINTERFACE;
1265
1266     return IServiceProvider_QueryService(This->service_provider, guidService, riid, ppv);
1267 }
1268
1269 #undef SERVPROV_THIS
1270
1271 static const IServiceProviderVtbl ServiceProviderVtbl = {
1272     BPServiceProvider_QueryInterface,
1273     BPServiceProvider_AddRef,
1274     BPServiceProvider_Release,
1275     BPServiceProvider_QueryService
1276 };
1277
1278 HRESULT create_binding_protocol(LPCWSTR url, BOOL from_urlmon, IInternetProtocol **protocol)
1279 {
1280     BindProtocol *ret = heap_alloc_zero(sizeof(BindProtocol));
1281
1282     ret->lpIInternetProtocolVtbl        = &BindProtocolVtbl;
1283     ret->lpInternetBindInfoVtbl         = &InternetBindInfoVtbl;
1284     ret->lpInternetPriorityVtbl         = &InternetPriorityVtbl;
1285     ret->lpServiceProviderVtbl          = &ServiceProviderVtbl;
1286     ret->lpIInternetProtocolSinkVtbl    = &InternetProtocolSinkVtbl;
1287     ret->lpIInternetProtocolHandlerVtbl = &InternetProtocolHandlerVtbl;
1288     ret->lpIWinInetHttpInfoVtbl         = &WinInetHttpInfoVtbl;
1289
1290     ret->ref = 1;
1291     ret->from_urlmon = from_urlmon;
1292     ret->apartment_thread = GetCurrentThreadId();
1293     ret->notif_hwnd = get_notif_hwnd();
1294     ret->protocol_handler = PROTOCOLHANDLER(ret);
1295     InitializeCriticalSection(&ret->section);
1296
1297     URLMON_LockModule();
1298
1299     *protocol = PROTOCOL(ret);
1300     return S_OK;
1301 }