2 * Copyright 2007-2009 Jacek Caban for CodeWeavers
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.
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.
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
19 #include "urlmon_main.h"
20 #include "wine/debug.h"
22 WINE_DEFAULT_DEBUG_CHANNEL(urlmon
);
24 typedef void (*task_proc_t
)(BindProtocol
*,task_header_t
*);
26 struct _task_header_t
{
31 #define BUFFER_SIZE 2048
32 #define MIME_TEST_SIZE 255
34 #define WM_MK_CONTINUE (WM_USER+101)
35 #define WM_MK_RELEASE (WM_USER+102)
37 static void process_tasks(BindProtocol
*This
)
42 EnterCriticalSection(&This
->section
);
44 task
= This
->task_queue_head
;
46 This
->task_queue_head
= task
->next
;
47 if(!This
->task_queue_head
)
48 This
->task_queue_tail
= NULL
;
51 LeaveCriticalSection(&This
->section
);
56 This
->continue_call
++;
57 task
->proc(This
, task
);
58 This
->continue_call
--;
62 static LRESULT WINAPI
notif_wnd_proc(HWND hwnd
, UINT msg
, WPARAM wParam
, LPARAM lParam
)
65 case WM_MK_CONTINUE
: {
66 BindProtocol
*This
= (BindProtocol
*)lParam
;
70 IInternetProtocolEx_Release(&This
->IInternetProtocolEx_iface
);
74 tls_data_t
*data
= get_tls_data();
76 if(!--data
->notif_hwnd_cnt
) {
78 data
->notif_hwnd
= NULL
;
83 return DefWindowProcW(hwnd
, msg
, wParam
, lParam
);
86 static const WCHAR wszURLMonikerNotificationWindow
[] =
87 {'U','R','L',' ','M','o','n','i','k','e','r',' ',
88 'N','o','t','i','f','i','c','a','t','i','o','n',' ','W','i','n','d','o','w',0};
90 static ATOM notif_wnd_class
;
92 static BOOL WINAPI
register_notif_wnd_class(INIT_ONCE
*once
, void *param
, void **context
)
94 static WNDCLASSEXW wndclass
= {
95 sizeof(wndclass
), 0, notif_wnd_proc
, 0, 0,
96 NULL
, NULL
, NULL
, NULL
, NULL
,
97 wszURLMonikerNotificationWindow
, NULL
100 wndclass
.hInstance
= hProxyDll
;
101 notif_wnd_class
= RegisterClassExW(&wndclass
);
105 void unregister_notif_wnd_class(void)
108 UnregisterClassW(MAKEINTRESOURCEW(notif_wnd_class
), hProxyDll
);
111 HWND
get_notif_hwnd(void)
113 tls_data_t
*tls_data
;
115 static INIT_ONCE init_once
= INIT_ONCE_STATIC_INIT
;
117 tls_data
= get_tls_data();
121 if(tls_data
->notif_hwnd_cnt
) {
122 tls_data
->notif_hwnd_cnt
++;
123 return tls_data
->notif_hwnd
;
126 InitOnceExecuteOnce(&init_once
, register_notif_wnd_class
, NULL
, NULL
);
130 tls_data
->notif_hwnd
= CreateWindowExW(0, MAKEINTRESOURCEW(notif_wnd_class
),
131 wszURLMonikerNotificationWindow
, 0, 0, 0, 0, 0, HWND_MESSAGE
,
132 NULL
, hProxyDll
, NULL
);
133 if(tls_data
->notif_hwnd
)
134 tls_data
->notif_hwnd_cnt
++;
136 TRACE("hwnd = %p\n", tls_data
->notif_hwnd
);
138 return tls_data
->notif_hwnd
;
141 void release_notif_hwnd(HWND hwnd
)
143 tls_data_t
*data
= get_tls_data();
148 if(data
->notif_hwnd
!= hwnd
) {
149 PostMessageW(data
->notif_hwnd
, WM_MK_RELEASE
, 0, 0);
153 if(!--data
->notif_hwnd_cnt
) {
154 DestroyWindow(data
->notif_hwnd
);
155 data
->notif_hwnd
= NULL
;
159 static void push_task(BindProtocol
*This
, task_header_t
*task
, task_proc_t proc
)
161 BOOL do_post
= FALSE
;
166 EnterCriticalSection(&This
->section
);
168 if(This
->task_queue_tail
) {
169 This
->task_queue_tail
->next
= task
;
170 This
->task_queue_tail
= task
;
172 This
->task_queue_tail
= This
->task_queue_head
= task
;
173 do_post
= !This
->continue_call
;
176 LeaveCriticalSection(&This
->section
);
179 IInternetProtocolEx_AddRef(&This
->IInternetProtocolEx_iface
);
180 PostMessageW(This
->notif_hwnd
, WM_MK_CONTINUE
, 0, (LPARAM
)This
);
184 static inline BOOL
is_apartment_thread(BindProtocol
*This
)
186 return This
->apartment_thread
== GetCurrentThreadId();
189 static inline BOOL
do_direct_notif(BindProtocol
*This
)
191 return !(This
->pi
& PI_APARTMENTTHREADED
) || (is_apartment_thread(This
) && !This
->continue_call
);
194 static HRESULT
handle_mime_filter(BindProtocol
*This
, IInternetProtocol
*mime_filter
)
196 PROTOCOLFILTERDATA filter_data
= { sizeof(PROTOCOLFILTERDATA
), NULL
, NULL
, NULL
, 0 };
199 hres
= IInternetProtocol_QueryInterface(mime_filter
, &IID_IInternetProtocolSink
, (void**)&This
->protocol_sink_handler
);
201 This
->protocol_sink_handler
= &This
->default_protocol_handler
.IInternetProtocolSink_iface
;
205 IInternetProtocol_AddRef(mime_filter
);
206 This
->protocol_handler
= mime_filter
;
208 filter_data
.pProtocol
= &This
->default_protocol_handler
.IInternetProtocol_iface
;
209 hres
= IInternetProtocol_Start(mime_filter
, This
->mime
, &This
->default_protocol_handler
.IInternetProtocolSink_iface
,
210 &This
->IInternetBindInfo_iface
, PI_FILTER_MODE
|PI_FORCE_ASYNC
,
211 (HANDLE_PTR
)&filter_data
);
213 IInternetProtocolSink_Release(This
->protocol_sink_handler
);
214 IInternetProtocol_Release(This
->protocol_handler
);
215 This
->protocol_sink_handler
= &This
->default_protocol_handler
.IInternetProtocolSink_iface
;
216 This
->protocol_handler
= &This
->default_protocol_handler
.IInternetProtocol_iface
;
220 /* NOTE: IE9 calls it on the new protocol_sink. It doesn't make sense so it seems to be a bug there. */
221 IInternetProtocolSink_ReportProgress(This
->protocol_sink
, BINDSTATUS_LOADINGMIMEHANDLER
, NULL
);
226 static void mime_available(BindProtocol
*This
, LPCWSTR mime
, BOOL verified
)
228 IInternetProtocol
*mime_filter
;
231 heap_free(This
->mime
);
232 This
->mime
= heap_strdupW(mime
);
234 if(This
->protocol_handler
==&This
->default_protocol_handler
.IInternetProtocol_iface
235 && (mime_filter
= get_mime_filter(mime
))) {
236 TRACE("Got mime filter for %s\n", debugstr_w(mime
));
238 hres
= handle_mime_filter(This
, mime_filter
);
239 IInternetProtocol_Release(mime_filter
);
241 FIXME("MIME filter failed: %08x\n", hres
);
244 if(This
->reported_mime
|| verified
|| !(This
->pi
& PI_MIMEVERIFICATION
)) {
245 This
->reported_mime
= TRUE
;
246 IInternetProtocolSink_ReportProgress(This
->protocol_sink
, BINDSTATUS_MIMETYPEAVAILABLE
, mime
);
250 static inline BindProtocol
*impl_from_IInternetProtocolEx(IInternetProtocolEx
*iface
)
252 return CONTAINING_RECORD(iface
, BindProtocol
, IInternetProtocolEx_iface
);
255 static HRESULT WINAPI
BindProtocol_QueryInterface(IInternetProtocolEx
*iface
, REFIID riid
, void **ppv
)
257 BindProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
260 if(IsEqualGUID(&IID_IUnknown
, riid
)) {
261 TRACE("(%p)->(IID_IUnknown %p)\n", This
, ppv
);
262 *ppv
= &This
->IInternetProtocolEx_iface
;
263 }else if(IsEqualGUID(&IID_IInternetProtocolRoot
, riid
)) {
264 TRACE("(%p)->(IID_IInternetProtocolRoot %p)\n", This
, ppv
);
265 *ppv
= &This
->IInternetProtocolEx_iface
;
266 }else if(IsEqualGUID(&IID_IInternetProtocol
, riid
)) {
267 TRACE("(%p)->(IID_IInternetProtocol %p)\n", This
, ppv
);
268 *ppv
= &This
->IInternetProtocolEx_iface
;
269 }else if(IsEqualGUID(&IID_IInternetProtocolEx
, riid
)) {
270 TRACE("(%p)->(IID_IInternetProtocolEx %p)\n", This
, ppv
);
271 *ppv
= &This
->IInternetProtocolEx_iface
;
272 }else if(IsEqualGUID(&IID_IInternetBindInfo
, riid
)) {
273 TRACE("(%p)->(IID_IInternetBindInfo %p)\n", This
, ppv
);
274 *ppv
= &This
->IInternetBindInfo_iface
;
275 }else if(IsEqualGUID(&IID_IInternetPriority
, riid
)) {
276 TRACE("(%p)->(IID_IInternetPriority %p)\n", This
, ppv
);
277 *ppv
= &This
->IInternetPriority_iface
;
278 }else if(IsEqualGUID(&IID_IAuthenticate
, riid
)) {
279 FIXME("(%p)->(IID_IAuthenticate %p)\n", This
, ppv
);
280 }else if(IsEqualGUID(&IID_IServiceProvider
, riid
)) {
281 TRACE("(%p)->(IID_IServiceProvider %p)\n", This
, ppv
);
282 *ppv
= &This
->IServiceProvider_iface
;
283 }else if(IsEqualGUID(&IID_IInternetProtocolSink
, riid
)) {
284 TRACE("(%p)->(IID_IInternetProtocolSink %p)\n", This
, ppv
);
285 *ppv
= &This
->IInternetProtocolSink_iface
;
286 }else if(This
->protocol_unk
) {
288 hres
= IUnknown_QueryInterface(This
->protocol_unk
, riid
, ppv
);
289 TRACE("(%p) aggregated handler returned %08x for %s\n", This
, hres
, debugstr_guid(riid
));
292 WARN("not supported interface %s\n", debugstr_guid(riid
));
296 return E_NOINTERFACE
;
298 IUnknown_AddRef((IUnknown
*)*ppv
);
302 static ULONG WINAPI
BindProtocol_AddRef(IInternetProtocolEx
*iface
)
304 BindProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
305 LONG ref
= InterlockedIncrement(&This
->ref
);
306 TRACE("(%p) ref=%d\n", This
, ref
);
310 static void release_protocol_handler(BindProtocol
*This
)
312 if(This
->protocol_unk
) {
313 IUnknown_Release(This
->protocol_unk
);
314 This
->protocol_unk
= NULL
;
317 IInternetProtocol_Release(This
->protocol
);
318 This
->protocol
= NULL
;
320 if(This
->protocol_handler
&& This
->protocol_handler
!= &This
->default_protocol_handler
.IInternetProtocol_iface
) {
321 IInternetProtocol_Release(This
->protocol_handler
);
322 This
->protocol_handler
= &This
->default_protocol_handler
.IInternetProtocol_iface
;
324 if(This
->protocol_sink_handler
&&
325 This
->protocol_sink_handler
!= &This
->default_protocol_handler
.IInternetProtocolSink_iface
) {
326 IInternetProtocolSink_Release(This
->protocol_sink_handler
);
327 This
->protocol_sink_handler
= &This
->default_protocol_handler
.IInternetProtocolSink_iface
;
331 static ULONG WINAPI
BindProtocol_Release(IInternetProtocolEx
*iface
)
333 BindProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
334 LONG ref
= InterlockedDecrement(&This
->ref
);
336 TRACE("(%p) ref=%d\n", This
, ref
);
339 release_protocol_handler(This
);
340 if(This
->redirect_callback
)
341 IBindCallbackRedirect_Release(This
->redirect_callback
);
343 IInternetBindInfo_Release(This
->bind_info
);
345 IUri_Release(This
->uri
);
346 SysFreeString(This
->display_uri
);
348 set_binding_sink(This
, NULL
, NULL
);
351 release_notif_hwnd(This
->notif_hwnd
);
352 This
->section
.DebugInfo
->Spare
[0] = 0;
353 DeleteCriticalSection(&This
->section
);
355 heap_free(This
->mime
);
358 URLMON_UnlockModule();
364 static HRESULT WINAPI
BindProtocol_Start(IInternetProtocolEx
*iface
, LPCWSTR szUrl
,
365 IInternetProtocolSink
*pOIProtSink
, IInternetBindInfo
*pOIBindInfo
,
366 DWORD grfPI
, HANDLE_PTR dwReserved
)
368 BindProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
372 TRACE("(%p)->(%s %p %p %08x %lx)\n", This
, debugstr_w(szUrl
), pOIProtSink
,
373 pOIBindInfo
, grfPI
, dwReserved
);
375 hres
= CreateUri(szUrl
, Uri_CREATE_FILE_USE_DOS_PATH
, 0, &uri
);
379 hres
= IInternetProtocolEx_StartEx(&This
->IInternetProtocolEx_iface
, uri
, pOIProtSink
,
380 pOIBindInfo
, grfPI
, (HANDLE
*)dwReserved
);
386 static HRESULT WINAPI
BindProtocol_Continue(IInternetProtocolEx
*iface
, PROTOCOLDATA
*pProtocolData
)
388 BindProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
390 TRACE("(%p)->(%p)\n", This
, pProtocolData
);
392 return IInternetProtocol_Continue(This
->protocol_handler
, pProtocolData
);
395 static HRESULT WINAPI
BindProtocol_Abort(IInternetProtocolEx
*iface
, HRESULT hrReason
,
398 BindProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
400 TRACE("(%p)->(%08x %08x)\n", This
, hrReason
, dwOptions
);
402 return IInternetProtocol_Abort(This
->protocol_handler
, hrReason
, dwOptions
);
405 static HRESULT WINAPI
BindProtocol_Terminate(IInternetProtocolEx
*iface
, DWORD dwOptions
)
407 BindProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
409 TRACE("(%p)->(%08x)\n", This
, dwOptions
);
411 return IInternetProtocol_Terminate(This
->protocol_handler
, dwOptions
);
414 static HRESULT WINAPI
BindProtocol_Suspend(IInternetProtocolEx
*iface
)
416 BindProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
417 FIXME("(%p)\n", This
);
421 static HRESULT WINAPI
BindProtocol_Resume(IInternetProtocolEx
*iface
)
423 BindProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
424 FIXME("(%p)\n", This
);
428 static HRESULT WINAPI
BindProtocol_Read(IInternetProtocolEx
*iface
, void *pv
,
429 ULONG cb
, ULONG
*pcbRead
)
431 BindProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
433 TRACE("(%p)->(%p %u %p)\n", This
, pv
, cb
, pcbRead
);
437 return IInternetProtocol_Read(This
->protocol_handler
, pv
, cb
, pcbRead
);
440 static HRESULT WINAPI
BindProtocol_Seek(IInternetProtocolEx
*iface
, LARGE_INTEGER dlibMove
,
441 DWORD dwOrigin
, ULARGE_INTEGER
*plibNewPosition
)
443 BindProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
444 FIXME("(%p)->(%d %d %p)\n", This
, dlibMove
.u
.LowPart
, dwOrigin
, plibNewPosition
);
448 static HRESULT WINAPI
BindProtocol_LockRequest(IInternetProtocolEx
*iface
, DWORD dwOptions
)
450 BindProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
452 TRACE("(%p)->(%08x)\n", This
, dwOptions
);
454 return IInternetProtocol_LockRequest(This
->protocol_handler
, dwOptions
);
457 static HRESULT WINAPI
BindProtocol_UnlockRequest(IInternetProtocolEx
*iface
)
459 BindProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
461 TRACE("(%p)\n", This
);
463 return IInternetProtocol_UnlockRequest(This
->protocol_handler
);
466 static HRESULT WINAPI
BindProtocol_StartEx(IInternetProtocolEx
*iface
, IUri
*pUri
,
467 IInternetProtocolSink
*pOIProtSink
, IInternetBindInfo
*pOIBindInfo
,
468 DWORD grfPI
, HANDLE
*dwReserved
)
470 BindProtocol
*This
= impl_from_IInternetProtocolEx(iface
);
471 IInternetProtocol
*protocol
= NULL
;
472 IInternetProtocolEx
*protocolex
;
473 IInternetPriority
*priority
;
474 IServiceProvider
*service_provider
;
475 CLSID clsid
= IID_NULL
;
476 IUnknown
*protocol_unk
= NULL
;
480 TRACE("(%p)->(%p %p %p %08x %p)\n", This
, pUri
, pOIProtSink
, pOIBindInfo
, grfPI
, dwReserved
);
482 if(!pUri
|| !pOIProtSink
|| !pOIBindInfo
)
488 SysFreeString(This
->display_uri
);
489 IUri_Release(This
->uri
);
494 hres
= IInternetProtocolSink_QueryInterface(pOIProtSink
, &IID_IServiceProvider
,
495 (void**)&service_provider
);
496 if(SUCCEEDED(hres
)) {
497 /* FIXME: What's protocol CLSID here? */
498 IServiceProvider_QueryService(service_provider
, &IID_IInternetProtocol
,
499 &IID_IInternetProtocol
, (void**)&protocol
);
500 IServiceProvider_Release(service_provider
);
506 hres
= get_protocol_handler(pUri
, &clsid
, &cf
);
510 hres
= IClassFactory_CreateInstance(cf
, (IUnknown
*)&This
->IInternetBindInfo_iface
,
511 &IID_IUnknown
, (void**)&protocol_unk
);
512 IClassFactory_Release(cf
);
516 hres
= IUnknown_QueryInterface(protocol_unk
, &IID_IInternetProtocol
, (void**)&protocol
);
518 IUnknown_Release(protocol_unk
);
523 StringFromCLSID(&clsid
, &clsid_str
);
524 IInternetProtocolSink_ReportProgress(pOIProtSink
, BINDSTATUS_PROTOCOLCLASSID
, clsid_str
);
525 CoTaskMemFree(clsid_str
);
527 This
->protocol_unk
= protocol_unk
;
528 This
->protocol
= protocol
;
531 protocol_unk
= (IUnknown
*)protocol
;
533 set_binding_sink(This
, pOIProtSink
, pOIBindInfo
);
535 hres
= IUnknown_QueryInterface(protocol_unk
, &IID_IInternetPriority
, (void**)&priority
);
536 if(SUCCEEDED(hres
)) {
537 IInternetPriority_SetPriority(priority
, This
->priority
);
538 IInternetPriority_Release(priority
);
541 hres
= IUnknown_QueryInterface(protocol_unk
, &IID_IInternetProtocolEx
, (void**)&protocolex
);
542 if(SUCCEEDED(hres
)) {
543 hres
= IInternetProtocolEx_StartEx(protocolex
, pUri
, &This
->IInternetProtocolSink_iface
,
544 &This
->IInternetBindInfo_iface
, 0, NULL
);
545 IInternetProtocolEx_Release(protocolex
);
547 hres
= IUri_GetDisplayUri(pUri
, &This
->display_uri
);
551 hres
= IInternetProtocol_Start(protocol
, This
->display_uri
, &This
->IInternetProtocolSink_iface
,
552 &This
->IInternetBindInfo_iface
, 0, 0);
560 void set_binding_sink(BindProtocol
*This
, IInternetProtocolSink
*sink
, IInternetBindInfo
*bind_info
)
562 IInternetProtocolSink
*prev_sink
;
563 IServiceProvider
*service_provider
= NULL
;
566 IInternetProtocolSink_AddRef(sink
);
567 prev_sink
= InterlockedExchangePointer((void**)&This
->protocol_sink
, sink
);
569 IInternetProtocolSink_Release(prev_sink
);
572 IInternetProtocolSink_QueryInterface(sink
, &IID_IServiceProvider
, (void**)&service_provider
);
573 service_provider
= InterlockedExchangePointer((void**)&This
->service_provider
, service_provider
);
575 IServiceProvider_Release(service_provider
);
578 IInternetBindInfo_AddRef(bind_info
);
579 bind_info
= InterlockedExchangePointer((void**)&This
->bind_info
, bind_info
);
581 IInternetBindInfo_Release(bind_info
);
584 static const IInternetProtocolExVtbl BindProtocolVtbl
= {
585 BindProtocol_QueryInterface
,
587 BindProtocol_Release
,
589 BindProtocol_Continue
,
591 BindProtocol_Terminate
,
592 BindProtocol_Suspend
,
596 BindProtocol_LockRequest
,
597 BindProtocol_UnlockRequest
,
601 static inline BindProtocol
*impl_from_IInternetProtocol(IInternetProtocol
*iface
)
603 return CONTAINING_RECORD(iface
, BindProtocol
, default_protocol_handler
.IInternetProtocol_iface
);
606 static HRESULT WINAPI
ProtocolHandler_QueryInterface(IInternetProtocol
*iface
, REFIID riid
, void **ppv
)
608 BindProtocol
*This
= impl_from_IInternetProtocol(iface
);
611 if(IsEqualGUID(&IID_IUnknown
, riid
)) {
612 TRACE("(%p)->(IID_IUnknown %p)\n", This
, ppv
);
613 *ppv
= &This
->default_protocol_handler
.IInternetProtocol_iface
;
614 }else if(IsEqualGUID(&IID_IInternetProtocolRoot
, riid
)) {
615 TRACE("(%p)->(IID_IInternetProtocolRoot %p)\n", This
, ppv
);
616 *ppv
= &This
->default_protocol_handler
.IInternetProtocol_iface
;
617 }else if(IsEqualGUID(&IID_IInternetProtocol
, riid
)) {
618 TRACE("(%p)->(IID_IInternetProtocol %p)\n", This
, ppv
);
619 *ppv
= &This
->default_protocol_handler
.IInternetProtocol_iface
;
620 }else if(IsEqualGUID(&IID_IInternetProtocolSink
, riid
)) {
621 TRACE("(%p)->(IID_IInternetProtocolSink %p)\n", This
, ppv
);
622 *ppv
= &This
->default_protocol_handler
.IInternetProtocolSink_iface
;
626 IInternetProtocol_AddRef(iface
);
630 WARN("not supported interface %s\n", debugstr_guid(riid
));
631 return E_NOINTERFACE
;
634 static ULONG WINAPI
ProtocolHandler_AddRef(IInternetProtocol
*iface
)
636 BindProtocol
*This
= impl_from_IInternetProtocol(iface
);
637 return IInternetProtocolEx_AddRef(&This
->IInternetProtocolEx_iface
);
640 static ULONG WINAPI
ProtocolHandler_Release(IInternetProtocol
*iface
)
642 BindProtocol
*This
= impl_from_IInternetProtocol(iface
);
643 return IInternetProtocolEx_Release(&This
->IInternetProtocolEx_iface
);
646 static HRESULT WINAPI
ProtocolHandler_Start(IInternetProtocol
*iface
, LPCWSTR szUrl
,
647 IInternetProtocolSink
*pOIProtSink
, IInternetBindInfo
*pOIBindInfo
,
648 DWORD grfPI
, HANDLE_PTR dwReserved
)
650 ERR("Should not be called\n");
654 static HRESULT WINAPI
ProtocolHandler_Continue(IInternetProtocol
*iface
, PROTOCOLDATA
*pProtocolData
)
656 BindProtocol
*This
= impl_from_IInternetProtocol(iface
);
657 IInternetProtocol
*protocol
;
660 TRACE("(%p)->(%p)\n", This
, pProtocolData
);
662 /* FIXME: This should not be needed. */
663 if(!This
->protocol
&& This
->protocol_unk
) {
664 hres
= IUnknown_QueryInterface(This
->protocol_unk
, &IID_IInternetProtocol
, (void**)&protocol
);
668 IInternetProtocol_AddRef(protocol
= This
->protocol
);
671 hres
= IInternetProtocol_Continue(protocol
, pProtocolData
);
673 heap_free(pProtocolData
);
674 IInternetProtocol_Release(protocol
);
678 static HRESULT WINAPI
ProtocolHandler_Abort(IInternetProtocol
*iface
, HRESULT hrReason
,
681 BindProtocol
*This
= impl_from_IInternetProtocol(iface
);
683 TRACE("(%p)->(%08x %08x)\n", This
, hrReason
, dwOptions
);
685 if(This
->protocol
&& !This
->reported_result
)
686 return IInternetProtocol_Abort(This
->protocol
, hrReason
, dwOptions
);
691 static HRESULT WINAPI
ProtocolHandler_Terminate(IInternetProtocol
*iface
, DWORD dwOptions
)
693 BindProtocol
*This
= impl_from_IInternetProtocol(iface
);
695 TRACE("(%p)->(%08x)\n", This
, dwOptions
);
697 if(!This
->reported_result
)
700 /* This may get released in Terminate call. */
701 IInternetProtocolEx_AddRef(&This
->IInternetProtocolEx_iface
);
704 IInternetProtocol_Terminate(This
->protocol
, 0);
705 IInternetProtocol_Release(This
->protocol
);
706 This
->protocol
= NULL
;
709 set_binding_sink(This
, NULL
, NULL
);
711 if(This
->bind_info
) {
712 IInternetBindInfo_Release(This
->bind_info
);
713 This
->bind_info
= NULL
;
716 if(This
->redirect_callback
) {
717 IBindCallbackRedirect_Release(This
->redirect_callback
);
718 This
->redirect_callback
= NULL
;
721 IInternetProtocolEx_Release(&This
->IInternetProtocolEx_iface
);
725 static HRESULT WINAPI
ProtocolHandler_Suspend(IInternetProtocol
*iface
)
727 BindProtocol
*This
= impl_from_IInternetProtocol(iface
);
728 FIXME("(%p)\n", This
);
732 static HRESULT WINAPI
ProtocolHandler_Resume(IInternetProtocol
*iface
)
734 BindProtocol
*This
= impl_from_IInternetProtocol(iface
);
735 FIXME("(%p)\n", This
);
739 static HRESULT WINAPI
ProtocolHandler_Read(IInternetProtocol
*iface
, void *pv
,
740 ULONG cb
, ULONG
*pcbRead
)
742 BindProtocol
*This
= impl_from_IInternetProtocol(iface
);
746 TRACE("(%p)->(%p %u %p)\n", This
, pv
, cb
, pcbRead
);
749 read
= min(cb
, This
->buf_size
);
750 memcpy(pv
, This
->buf
, read
);
752 if(read
== This
->buf_size
) {
753 heap_free(This
->buf
);
756 memmove(This
->buf
, This
->buf
+cb
, This
->buf_size
-cb
);
759 This
->buf_size
-= read
;
763 IInternetProtocol
*protocol
;
766 /* FIXME: We shouldn't need it, but out binding code currently depends on it. */
767 if(!This
->protocol
&& This
->protocol_unk
) {
768 hres
= IUnknown_QueryInterface(This
->protocol_unk
, &IID_IInternetProtocol
,
773 protocol
= This
->protocol
;
776 if(is_apartment_thread(This
))
777 This
->continue_call
++;
778 hres
= IInternetProtocol_Read(protocol
, (BYTE
*)pv
+read
, cb
-read
, &cread
);
779 if(is_apartment_thread(This
))
780 This
->continue_call
--;
784 IInternetProtocol_Release(protocol
);
791 static HRESULT WINAPI
ProtocolHandler_Seek(IInternetProtocol
*iface
, LARGE_INTEGER dlibMove
,
792 DWORD dwOrigin
, ULARGE_INTEGER
*plibNewPosition
)
794 BindProtocol
*This
= impl_from_IInternetProtocol(iface
);
795 FIXME("(%p)->(%d %d %p)\n", This
, dlibMove
.u
.LowPart
, dwOrigin
, plibNewPosition
);
799 static HRESULT WINAPI
ProtocolHandler_LockRequest(IInternetProtocol
*iface
, DWORD dwOptions
)
801 BindProtocol
*This
= impl_from_IInternetProtocol(iface
);
803 TRACE("(%p)->(%08x)\n", This
, dwOptions
);
805 return IInternetProtocol_LockRequest(This
->protocol
, dwOptions
);
808 static HRESULT WINAPI
ProtocolHandler_UnlockRequest(IInternetProtocol
*iface
)
810 BindProtocol
*This
= impl_from_IInternetProtocol(iface
);
812 TRACE("(%p)\n", This
);
814 return IInternetProtocol_UnlockRequest(This
->protocol
);
817 static const IInternetProtocolVtbl InternetProtocolHandlerVtbl
= {
818 ProtocolHandler_QueryInterface
,
819 ProtocolHandler_AddRef
,
820 ProtocolHandler_Release
,
821 ProtocolHandler_Start
,
822 ProtocolHandler_Continue
,
823 ProtocolHandler_Abort
,
824 ProtocolHandler_Terminate
,
825 ProtocolHandler_Suspend
,
826 ProtocolHandler_Resume
,
827 ProtocolHandler_Read
,
828 ProtocolHandler_Seek
,
829 ProtocolHandler_LockRequest
,
830 ProtocolHandler_UnlockRequest
833 static inline BindProtocol
*impl_from_IInternetProtocolSinkHandler(IInternetProtocolSink
*iface
)
835 return CONTAINING_RECORD(iface
, BindProtocol
, default_protocol_handler
.IInternetProtocolSink_iface
);
838 static HRESULT WINAPI
ProtocolSinkHandler_QueryInterface(IInternetProtocolSink
*iface
,
839 REFIID riid
, void **ppvObject
)
841 BindProtocol
*This
= impl_from_IInternetProtocolSinkHandler(iface
);
842 return IInternetProtocol_QueryInterface(&This
->default_protocol_handler
.IInternetProtocol_iface
,
846 static ULONG WINAPI
ProtocolSinkHandler_AddRef(IInternetProtocolSink
*iface
)
848 BindProtocol
*This
= impl_from_IInternetProtocolSinkHandler(iface
);
849 return IInternetProtocolEx_AddRef(&This
->IInternetProtocolEx_iface
);
852 static ULONG WINAPI
ProtocolSinkHandler_Release(IInternetProtocolSink
*iface
)
854 BindProtocol
*This
= impl_from_IInternetProtocolSinkHandler(iface
);
855 return IInternetProtocolEx_Release(&This
->IInternetProtocolEx_iface
);
858 static HRESULT WINAPI
ProtocolSinkHandler_Switch(IInternetProtocolSink
*iface
,
859 PROTOCOLDATA
*pProtocolData
)
861 BindProtocol
*This
= impl_from_IInternetProtocolSinkHandler(iface
);
863 TRACE("(%p)->(%p)\n", This
, pProtocolData
);
865 if(!This
->protocol_sink
) {
866 IInternetProtocol_Continue(This
->protocol_handler
, pProtocolData
);
870 return IInternetProtocolSink_Switch(This
->protocol_sink
, pProtocolData
);
873 static HRESULT WINAPI
ProtocolSinkHandler_ReportProgress(IInternetProtocolSink
*iface
,
874 ULONG status_code
, LPCWSTR status_text
)
876 BindProtocol
*This
= impl_from_IInternetProtocolSinkHandler(iface
);
878 TRACE("(%p)->(%s %s)\n", This
, debugstr_bindstatus(status_code
), debugstr_w(status_text
));
880 if(!This
->protocol_sink
)
883 switch(status_code
) {
884 case BINDSTATUS_FINDINGRESOURCE
:
885 case BINDSTATUS_CONNECTING
:
886 case BINDSTATUS_REDIRECTING
:
887 case BINDSTATUS_SENDINGREQUEST
:
888 case BINDSTATUS_CACHEFILENAMEAVAILABLE
:
889 case BINDSTATUS_DIRECTBIND
:
890 case BINDSTATUS_ACCEPTRANGES
:
891 case BINDSTATUS_DECODING
:
892 IInternetProtocolSink_ReportProgress(This
->protocol_sink
, status_code
, status_text
);
895 case BINDSTATUS_BEGINDOWNLOADDATA
:
896 IInternetProtocolSink_ReportData(This
->protocol_sink
, This
->bscf
, This
->progress
, This
->progress_max
);
899 case BINDSTATUS_MIMETYPEAVAILABLE
:
900 mime_available(This
, status_text
, FALSE
);
903 case BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE
:
904 mime_available(This
, status_text
, TRUE
);
908 FIXME("unsupported ulStatusCode %u\n", status_code
);
914 static HRESULT WINAPI
ProtocolSinkHandler_ReportData(IInternetProtocolSink
*iface
,
915 DWORD bscf
, ULONG progress
, ULONG progress_max
)
917 BindProtocol
*This
= impl_from_IInternetProtocolSinkHandler(iface
);
919 TRACE("(%p)->(%x %u %u)\n", This
, bscf
, progress
, progress_max
);
922 This
->progress
= progress
;
923 This
->progress_max
= progress_max
;
925 if(!This
->protocol_sink
)
928 if((This
->pi
& PI_MIMEVERIFICATION
) && !This
->reported_mime
) {
929 BYTE buf
[BUFFER_SIZE
];
936 if(is_apartment_thread(This
))
937 This
->continue_call
++;
938 hres
= IInternetProtocol_Read(This
->protocol
, buf
,
939 sizeof(buf
)-This
->buf_size
, &read
);
940 if(is_apartment_thread(This
))
941 This
->continue_call
--;
942 if(FAILED(hres
) && hres
!= E_PENDING
)
946 This
->buf
= heap_alloc(BUFFER_SIZE
);
948 return E_OUTOFMEMORY
;
949 }else if(read
+ This
->buf_size
> BUFFER_SIZE
) {
952 tmp
= heap_realloc(This
->buf
, read
+This
->buf_size
);
954 return E_OUTOFMEMORY
;
958 memcpy(This
->buf
+This
->buf_size
, buf
, read
);
959 This
->buf_size
+= read
;
960 }while(This
->buf_size
< MIME_TEST_SIZE
&& hres
== S_OK
);
962 if(This
->buf_size
< MIME_TEST_SIZE
&& hres
!= S_FALSE
)
965 bscf
= BSCF_FIRSTDATANOTIFICATION
;
967 bscf
|= BSCF_LASTDATANOTIFICATION
|BSCF_DATAFULLYAVAILABLE
;
969 if(!This
->reported_mime
) {
972 hres
= IUri_GetRawUri(This
->uri
, &raw_uri
);
976 hres
= FindMimeFromData(NULL
, raw_uri
, This
->buf
, min(This
->buf_size
, MIME_TEST_SIZE
),
977 This
->mime
, 0, &mime
, 0);
978 SysFreeString(raw_uri
);
982 heap_free(This
->mime
);
983 This
->mime
= heap_strdupW(mime
);
985 This
->reported_mime
= TRUE
;
986 if(This
->protocol_sink
)
987 IInternetProtocolSink_ReportProgress(This
->protocol_sink
, BINDSTATUS_MIMETYPEAVAILABLE
, This
->mime
);
991 if(!This
->protocol_sink
)
994 return IInternetProtocolSink_ReportData(This
->protocol_sink
, bscf
, progress
, progress_max
);
997 static HRESULT
handle_redirect(BindProtocol
*This
, const WCHAR
*url
)
1001 if(This
->redirect_callback
) {
1002 VARIANT_BOOL cancel
= VARIANT_FALSE
;
1003 IBindCallbackRedirect_Redirect(This
->redirect_callback
, url
, &cancel
);
1005 return INET_E_REDIRECT_FAILED
;
1008 if(This
->protocol_sink
) {
1009 hres
= IInternetProtocolSink_ReportProgress(This
->protocol_sink
, BINDSTATUS_REDIRECTING
, url
);
1014 IInternetProtocol_Terminate(This
->protocol
, 0); /* should this be done in StartEx? */
1015 release_protocol_handler(This
);
1017 return IInternetProtocolEx_Start(&This
->IInternetProtocolEx_iface
, url
, This
->protocol_sink
, This
->bind_info
, This
->pi
, 0);
1020 static HRESULT WINAPI
ProtocolSinkHandler_ReportResult(IInternetProtocolSink
*iface
,
1021 HRESULT hrResult
, DWORD dwError
, LPCWSTR szResult
)
1023 BindProtocol
*This
= impl_from_IInternetProtocolSinkHandler(iface
);
1025 TRACE("(%p)->(%08x %d %s)\n", This
, hrResult
, dwError
, debugstr_w(szResult
));
1027 if(hrResult
== INET_E_REDIRECT_FAILED
) {
1028 hrResult
= handle_redirect(This
, szResult
);
1029 if(hrResult
== S_OK
)
1034 if(This
->protocol_sink
)
1035 return IInternetProtocolSink_ReportResult(This
->protocol_sink
, hrResult
, dwError
, szResult
);
1039 static const IInternetProtocolSinkVtbl InternetProtocolSinkHandlerVtbl
= {
1040 ProtocolSinkHandler_QueryInterface
,
1041 ProtocolSinkHandler_AddRef
,
1042 ProtocolSinkHandler_Release
,
1043 ProtocolSinkHandler_Switch
,
1044 ProtocolSinkHandler_ReportProgress
,
1045 ProtocolSinkHandler_ReportData
,
1046 ProtocolSinkHandler_ReportResult
1049 static inline BindProtocol
*impl_from_IInternetBindInfo(IInternetBindInfo
*iface
)
1051 return CONTAINING_RECORD(iface
, BindProtocol
, IInternetBindInfo_iface
);
1054 static HRESULT WINAPI
BindInfo_QueryInterface(IInternetBindInfo
*iface
,
1055 REFIID riid
, void **ppv
)
1057 BindProtocol
*This
= impl_from_IInternetBindInfo(iface
);
1058 return IInternetProtocolEx_QueryInterface(&This
->IInternetProtocolEx_iface
, riid
, ppv
);
1061 static ULONG WINAPI
BindInfo_AddRef(IInternetBindInfo
*iface
)
1063 BindProtocol
*This
= impl_from_IInternetBindInfo(iface
);
1064 return IInternetProtocolEx_AddRef(&This
->IInternetProtocolEx_iface
);
1067 static ULONG WINAPI
BindInfo_Release(IInternetBindInfo
*iface
)
1069 BindProtocol
*This
= impl_from_IInternetBindInfo(iface
);
1070 return IInternetProtocolEx_Release(&This
->IInternetProtocolEx_iface
);
1073 static HRESULT WINAPI
BindInfo_GetBindInfo(IInternetBindInfo
*iface
,
1074 DWORD
*grfBINDF
, BINDINFO
*pbindinfo
)
1076 BindProtocol
*This
= impl_from_IInternetBindInfo(iface
);
1079 TRACE("(%p)->(%p %p)\n", This
, grfBINDF
, pbindinfo
);
1081 hres
= IInternetBindInfo_GetBindInfo(This
->bind_info
, grfBINDF
, pbindinfo
);
1083 WARN("GetBindInfo failed: %08x\n", hres
);
1087 if((pbindinfo
->dwOptions
& BINDINFO_OPTIONS_DISABLEAUTOREDIRECTS
) && !This
->redirect_callback
) {
1088 IServiceProvider
*service_provider
;
1090 hres
= IInternetProtocolSink_QueryInterface(This
->protocol_sink
, &IID_IServiceProvider
, (void**)&service_provider
);
1091 if(SUCCEEDED(hres
)) {
1092 hres
= IServiceProvider_QueryService(service_provider
, &IID_IBindCallbackRedirect
, &IID_IBindCallbackRedirect
,
1093 (void**)&This
->redirect_callback
);
1094 IServiceProvider_Release(service_provider
);
1098 *grfBINDF
|= BINDF_FROMURLMON
;
1102 static HRESULT WINAPI
BindInfo_GetBindString(IInternetBindInfo
*iface
,
1103 ULONG ulStringType
, LPOLESTR
*ppwzStr
, ULONG cEl
, ULONG
*pcElFetched
)
1105 BindProtocol
*This
= impl_from_IInternetBindInfo(iface
);
1107 TRACE("(%p)->(%d %p %d %p)\n", This
, ulStringType
, ppwzStr
, cEl
, pcElFetched
);
1109 return IInternetBindInfo_GetBindString(This
->bind_info
, ulStringType
, ppwzStr
, cEl
, pcElFetched
);
1112 static const IInternetBindInfoVtbl InternetBindInfoVtbl
= {
1113 BindInfo_QueryInterface
,
1116 BindInfo_GetBindInfo
,
1117 BindInfo_GetBindString
1120 static inline BindProtocol
*impl_from_IInternetPriority(IInternetPriority
*iface
)
1122 return CONTAINING_RECORD(iface
, BindProtocol
, IInternetPriority_iface
);
1125 static HRESULT WINAPI
InternetPriority_QueryInterface(IInternetPriority
*iface
,
1126 REFIID riid
, void **ppv
)
1128 BindProtocol
*This
= impl_from_IInternetPriority(iface
);
1129 return IInternetProtocolEx_QueryInterface(&This
->IInternetProtocolEx_iface
, riid
, ppv
);
1132 static ULONG WINAPI
InternetPriority_AddRef(IInternetPriority
*iface
)
1134 BindProtocol
*This
= impl_from_IInternetPriority(iface
);
1135 return IInternetProtocolEx_AddRef(&This
->IInternetProtocolEx_iface
);
1138 static ULONG WINAPI
InternetPriority_Release(IInternetPriority
*iface
)
1140 BindProtocol
*This
= impl_from_IInternetPriority(iface
);
1141 return IInternetProtocolEx_Release(&This
->IInternetProtocolEx_iface
);
1144 static HRESULT WINAPI
InternetPriority_SetPriority(IInternetPriority
*iface
, LONG nPriority
)
1146 BindProtocol
*This
= impl_from_IInternetPriority(iface
);
1148 TRACE("(%p)->(%d)\n", This
, nPriority
);
1150 This
->priority
= nPriority
;
1154 static HRESULT WINAPI
InternetPriority_GetPriority(IInternetPriority
*iface
, LONG
*pnPriority
)
1156 BindProtocol
*This
= impl_from_IInternetPriority(iface
);
1158 TRACE("(%p)->(%p)\n", This
, pnPriority
);
1160 *pnPriority
= This
->priority
;
1164 static const IInternetPriorityVtbl InternetPriorityVtbl
= {
1165 InternetPriority_QueryInterface
,
1166 InternetPriority_AddRef
,
1167 InternetPriority_Release
,
1168 InternetPriority_SetPriority
,
1169 InternetPriority_GetPriority
1173 static inline BindProtocol
*impl_from_IInternetProtocolSink(IInternetProtocolSink
*iface
)
1175 return CONTAINING_RECORD(iface
, BindProtocol
, IInternetProtocolSink_iface
);
1178 static HRESULT WINAPI
BPInternetProtocolSink_QueryInterface(IInternetProtocolSink
*iface
,
1179 REFIID riid
, void **ppv
)
1181 BindProtocol
*This
= impl_from_IInternetProtocolSink(iface
);
1182 return IInternetProtocolEx_QueryInterface(&This
->IInternetProtocolEx_iface
, riid
, ppv
);
1185 static ULONG WINAPI
BPInternetProtocolSink_AddRef(IInternetProtocolSink
*iface
)
1187 BindProtocol
*This
= impl_from_IInternetProtocolSink(iface
);
1188 return IInternetProtocolEx_AddRef(&This
->IInternetProtocolEx_iface
);
1191 static ULONG WINAPI
BPInternetProtocolSink_Release(IInternetProtocolSink
*iface
)
1193 BindProtocol
*This
= impl_from_IInternetProtocolSink(iface
);
1194 return IInternetProtocolEx_Release(&This
->IInternetProtocolEx_iface
);
1198 task_header_t header
;
1202 static void switch_proc(BindProtocol
*bind
, task_header_t
*t
)
1204 switch_task_t
*task
= (switch_task_t
*)t
;
1206 IInternetProtocol_Continue(bind
->protocol_handler
, task
->data
);
1211 static HRESULT WINAPI
BPInternetProtocolSink_Switch(IInternetProtocolSink
*iface
,
1212 PROTOCOLDATA
*pProtocolData
)
1214 BindProtocol
*This
= impl_from_IInternetProtocolSink(iface
);
1217 TRACE("(%p)->(%p)\n", This
, pProtocolData
);
1219 TRACE("flags %x state %x data %p cb %u\n", pProtocolData
->grfFlags
, pProtocolData
->dwState
,
1220 pProtocolData
->pData
, pProtocolData
->cbData
);
1222 data
= heap_alloc(sizeof(PROTOCOLDATA
));
1224 return E_OUTOFMEMORY
;
1225 memcpy(data
, pProtocolData
, sizeof(PROTOCOLDATA
));
1227 if((This
->pi
&PI_APARTMENTTHREADED
&& pProtocolData
->grfFlags
&PI_FORCE_ASYNC
)
1228 || !do_direct_notif(This
)) {
1229 switch_task_t
*task
;
1231 task
= heap_alloc(sizeof(switch_task_t
));
1235 return E_OUTOFMEMORY
;
1240 push_task(This
, &task
->header
, switch_proc
);
1244 return IInternetProtocolSink_Switch(This
->protocol_sink_handler
, data
);
1248 task_header_t header
;
1252 } on_progress_task_t
;
1254 static void on_progress_proc(BindProtocol
*This
, task_header_t
*t
)
1256 on_progress_task_t
*task
= (on_progress_task_t
*)t
;
1258 IInternetProtocolSink_ReportProgress(This
->protocol_sink_handler
, task
->status_code
, task
->status_text
);
1260 heap_free(task
->status_text
);
1264 static HRESULT WINAPI
BPInternetProtocolSink_ReportProgress(IInternetProtocolSink
*iface
,
1265 ULONG ulStatusCode
, LPCWSTR szStatusText
)
1267 BindProtocol
*This
= impl_from_IInternetProtocolSink(iface
);
1269 TRACE("(%p)->(%u %s)\n", This
, ulStatusCode
, debugstr_w(szStatusText
));
1271 if(do_direct_notif(This
)) {
1272 IInternetProtocolSink_ReportProgress(This
->protocol_sink_handler
, ulStatusCode
, szStatusText
);
1274 on_progress_task_t
*task
;
1276 task
= heap_alloc(sizeof(on_progress_task_t
));
1278 task
->status_code
= ulStatusCode
;
1279 task
->status_text
= heap_strdupW(szStatusText
);
1281 push_task(This
, &task
->header
, on_progress_proc
);
1288 task_header_t header
;
1292 } report_data_task_t
;
1294 static void report_data_proc(BindProtocol
*This
, task_header_t
*t
)
1296 report_data_task_t
*task
= (report_data_task_t
*)t
;
1298 IInternetProtocolSink_ReportData(This
->protocol_sink_handler
,
1299 task
->bscf
, task
->progress
, task
->progress_max
);
1304 static HRESULT WINAPI
BPInternetProtocolSink_ReportData(IInternetProtocolSink
*iface
,
1305 DWORD grfBSCF
, ULONG ulProgress
, ULONG ulProgressMax
)
1307 BindProtocol
*This
= impl_from_IInternetProtocolSink(iface
);
1309 TRACE("(%p)->(%x %u %u)\n", This
, grfBSCF
, ulProgress
, ulProgressMax
);
1311 if(!This
->protocol_sink
)
1314 if(!do_direct_notif(This
)) {
1315 report_data_task_t
*task
;
1317 task
= heap_alloc(sizeof(report_data_task_t
));
1319 return E_OUTOFMEMORY
;
1321 task
->bscf
= grfBSCF
;
1322 task
->progress
= ulProgress
;
1323 task
->progress_max
= ulProgressMax
;
1325 push_task(This
, &task
->header
, report_data_proc
);
1329 return IInternetProtocolSink_ReportData(This
->protocol_sink_handler
,
1330 grfBSCF
, ulProgress
, ulProgressMax
);
1334 task_header_t header
;
1339 } report_result_task_t
;
1341 static void report_result_proc(BindProtocol
*This
, task_header_t
*t
)
1343 report_result_task_t
*task
= (report_result_task_t
*)t
;
1345 IInternetProtocolSink_ReportResult(This
->protocol_sink_handler
, task
->hres
, task
->err
, task
->str
);
1347 heap_free(task
->str
);
1351 static HRESULT WINAPI
BPInternetProtocolSink_ReportResult(IInternetProtocolSink
*iface
,
1352 HRESULT hrResult
, DWORD dwError
, LPCWSTR szResult
)
1354 BindProtocol
*This
= impl_from_IInternetProtocolSink(iface
);
1356 TRACE("(%p)->(%08x %d %s)\n", This
, hrResult
, dwError
, debugstr_w(szResult
));
1358 if(!This
->protocol_sink
)
1360 This
->reported_result
= TRUE
;
1362 if(!do_direct_notif(This
)) {
1363 report_result_task_t
*task
;
1365 task
= heap_alloc(sizeof(report_result_task_t
));
1367 return E_OUTOFMEMORY
;
1369 task
->hres
= hrResult
;
1370 task
->err
= dwError
;
1371 task
->str
= heap_strdupW(szResult
);
1373 push_task(This
, &task
->header
, report_result_proc
);
1377 return IInternetProtocolSink_ReportResult(This
->protocol_sink_handler
, hrResult
, dwError
, szResult
);
1380 static const IInternetProtocolSinkVtbl InternetProtocolSinkVtbl
= {
1381 BPInternetProtocolSink_QueryInterface
,
1382 BPInternetProtocolSink_AddRef
,
1383 BPInternetProtocolSink_Release
,
1384 BPInternetProtocolSink_Switch
,
1385 BPInternetProtocolSink_ReportProgress
,
1386 BPInternetProtocolSink_ReportData
,
1387 BPInternetProtocolSink_ReportResult
1390 static inline BindProtocol
*impl_from_IServiceProvider(IServiceProvider
*iface
)
1392 return CONTAINING_RECORD(iface
, BindProtocol
, IServiceProvider_iface
);
1395 static HRESULT WINAPI
BPServiceProvider_QueryInterface(IServiceProvider
*iface
,
1396 REFIID riid
, void **ppv
)
1398 BindProtocol
*This
= impl_from_IServiceProvider(iface
);
1399 return IInternetProtocolEx_QueryInterface(&This
->IInternetProtocolEx_iface
, riid
, ppv
);
1402 static ULONG WINAPI
BPServiceProvider_AddRef(IServiceProvider
*iface
)
1404 BindProtocol
*This
= impl_from_IServiceProvider(iface
);
1405 return IInternetProtocolEx_AddRef(&This
->IInternetProtocolEx_iface
);
1408 static ULONG WINAPI
BPServiceProvider_Release(IServiceProvider
*iface
)
1410 BindProtocol
*This
= impl_from_IServiceProvider(iface
);
1411 return IInternetProtocolEx_Release(&This
->IInternetProtocolEx_iface
);
1414 static HRESULT WINAPI
BPServiceProvider_QueryService(IServiceProvider
*iface
,
1415 REFGUID guidService
, REFIID riid
, void **ppv
)
1417 BindProtocol
*This
= impl_from_IServiceProvider(iface
);
1419 TRACE("(%p)->(%s %s %p)\n", This
, debugstr_guid(guidService
), debugstr_guid(riid
), ppv
);
1421 if(!This
->service_provider
)
1422 return E_NOINTERFACE
;
1424 return IServiceProvider_QueryService(This
->service_provider
, guidService
, riid
, ppv
);
1427 static const IServiceProviderVtbl ServiceProviderVtbl
= {
1428 BPServiceProvider_QueryInterface
,
1429 BPServiceProvider_AddRef
,
1430 BPServiceProvider_Release
,
1431 BPServiceProvider_QueryService
1434 HRESULT
create_binding_protocol(BindProtocol
**protocol
)
1436 BindProtocol
*ret
= heap_alloc_zero(sizeof(BindProtocol
));
1438 ret
->IInternetProtocolEx_iface
.lpVtbl
= &BindProtocolVtbl
;
1439 ret
->IInternetBindInfo_iface
.lpVtbl
= &InternetBindInfoVtbl
;
1440 ret
->IInternetPriority_iface
.lpVtbl
= &InternetPriorityVtbl
;
1441 ret
->IServiceProvider_iface
.lpVtbl
= &ServiceProviderVtbl
;
1442 ret
->IInternetProtocolSink_iface
.lpVtbl
= &InternetProtocolSinkVtbl
;
1444 ret
->default_protocol_handler
.IInternetProtocol_iface
.lpVtbl
= &InternetProtocolHandlerVtbl
;
1445 ret
->default_protocol_handler
.IInternetProtocolSink_iface
.lpVtbl
= &InternetProtocolSinkHandlerVtbl
;
1448 ret
->apartment_thread
= GetCurrentThreadId();
1449 ret
->notif_hwnd
= get_notif_hwnd();
1450 ret
->protocol_handler
= &ret
->default_protocol_handler
.IInternetProtocol_iface
;
1451 ret
->protocol_sink_handler
= &ret
->default_protocol_handler
.IInternetProtocolSink_iface
;
1452 InitializeCriticalSection(&ret
->section
);
1453 ret
->section
.DebugInfo
->Spare
[0] = (DWORD_PTR
)(__FILE__
": BindProtocol.section");
1455 URLMON_LockModule();