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 struct BindProtocol BindProtocol;
26 struct _task_header_t;
28 typedef void (*task_proc_t)(BindProtocol*,struct _task_header_t*);
30 typedef struct _task_header_t {
32 struct _task_header_t *next;
36 const IInternetProtocolVtbl *lpIInternetProtocolVtbl;
37 const IInternetBindInfoVtbl *lpInternetBindInfoVtbl;
38 const IInternetPriorityVtbl *lpInternetPriorityVtbl;
39 const IServiceProviderVtbl *lpServiceProviderVtbl;
40 const IInternetProtocolSinkVtbl *lpIInternetProtocolSinkVtbl;
41 const IWinInetHttpInfoVtbl *lpIWinInetHttpInfoVtbl;
43 const IInternetProtocolVtbl *lpIInternetProtocolHandlerVtbl;
47 IInternetProtocol *protocol;
48 IInternetProtocol *protocol_handler;
49 IInternetBindInfo *bind_info;
50 IInternetProtocolSink *protocol_sink;
51 IServiceProvider *service_provider;
52 IWinInetInfo *wininet_info;
61 DWORD apartment_thread;
65 CRITICAL_SECTION section;
66 task_header_t *task_queue_head, *task_queue_tail;
72 ProtocolProxy *filter_proxy;
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)
80 #define PROTOCOLHANDLER(x) ((IInternetProtocol*) &(x)->lpIInternetProtocolHandlerVtbl)
82 #define BUFFER_SIZE 2048
83 #define MIME_TEST_SIZE 255
85 #define WM_MK_CONTINUE (WM_USER+101)
86 #define WM_MK_RELEASE (WM_USER+102)
88 static LRESULT WINAPI notif_wnd_proc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
91 case WM_MK_CONTINUE: {
92 BindProtocol *This = (BindProtocol*)lParam;
96 EnterCriticalSection(&This->section);
98 task = This->task_queue_head;
100 This->task_queue_head = task->next;
101 if(!This->task_queue_head)
102 This->task_queue_tail = NULL;
105 LeaveCriticalSection(&This->section);
110 This->continue_call++;
111 task->proc(This, task);
112 This->continue_call--;
115 IInternetProtocol_Release(PROTOCOL(This));
118 case WM_MK_RELEASE: {
119 tls_data_t *data = get_tls_data();
121 if(!--data->notif_hwnd_cnt) {
123 data->notif_hwnd = NULL;
128 return DefWindowProcW(hwnd, msg, wParam, lParam);
131 HWND get_notif_hwnd(void)
133 static ATOM wnd_class = 0;
134 tls_data_t *tls_data;
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};
140 tls_data = get_tls_data();
144 if(tls_data->notif_hwnd_cnt) {
145 tls_data->notif_hwnd_cnt++;
146 return tls_data->notif_hwnd;
150 static WNDCLASSEXW wndclass = {
152 notif_wnd_proc, 0, 0,
153 NULL, NULL, NULL, NULL, NULL,
154 wszURLMonikerNotificationWindow,
158 wndclass.hInstance = URLMON_hInstance;
160 wnd_class = RegisterClassExW(&wndclass);
161 if (!wnd_class && GetLastError() == ERROR_CLASS_ALREADY_EXISTS)
165 tls_data->notif_hwnd = CreateWindowExW(0, wszURLMonikerNotificationWindow,
166 wszURLMonikerNotificationWindow, 0, 0, 0, 0, 0, HWND_MESSAGE,
167 NULL, URLMON_hInstance, NULL);
168 if(tls_data->notif_hwnd)
169 tls_data->notif_hwnd_cnt++;
171 TRACE("hwnd = %p\n", tls_data->notif_hwnd);
173 return tls_data->notif_hwnd;
176 void release_notif_hwnd(HWND hwnd)
178 tls_data_t *data = get_tls_data();
183 if(data->notif_hwnd != hwnd) {
184 PostMessageW(data->notif_hwnd, WM_MK_RELEASE, 0, 0);
188 if(!--data->notif_hwnd_cnt) {
189 DestroyWindow(data->notif_hwnd);
190 data->notif_hwnd = NULL;
194 static void push_task(BindProtocol *This, task_header_t *task, task_proc_t proc)
196 BOOL do_post = FALSE;
201 EnterCriticalSection(&This->section);
203 if(This->task_queue_tail) {
204 This->task_queue_tail->next = task;
205 This->task_queue_tail = task;
207 This->task_queue_tail = This->task_queue_head = task;
211 LeaveCriticalSection(&This->section);
214 IInternetProtocol_AddRef(PROTOCOL(This));
215 PostMessageW(This->notif_hwnd, WM_MK_CONTINUE, 0, (LPARAM)This);
219 static inline BOOL do_direct_notif(BindProtocol *This)
221 return !(This->pi & PI_APARTMENTTHREADED) || (This->apartment_thread == GetCurrentThreadId() && !This->continue_call);
224 static HRESULT handle_mime_filter(BindProtocol *This, IInternetProtocol *mime_filter, LPCWSTR mime)
226 PROTOCOLFILTERDATA filter_data = { sizeof(PROTOCOLFILTERDATA), NULL, NULL, NULL, 0 };
227 IInternetProtocolSink *protocol_sink, *old_sink;
228 ProtocolProxy *filter_proxy;
231 hres = IInternetProtocol_QueryInterface(mime_filter, &IID_IInternetProtocolSink, (void**)&protocol_sink);
235 hres = create_protocol_proxy(PROTOCOLHANDLER(This), This->protocol_sink, &filter_proxy);
237 IInternetProtocolSink_Release(protocol_sink);
241 old_sink = This->protocol_sink;
242 This->protocol_sink = protocol_sink;
243 This->filter_proxy = filter_proxy;
245 IInternetProtocol_AddRef(mime_filter);
246 This->protocol_handler = mime_filter;
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);
252 IInternetProtocolSink_Release(old_sink);
256 IInternetProtocolSink_ReportProgress(old_sink, BINDSTATUS_LOADINGMIMEHANDLER, NULL);
257 IInternetProtocolSink_Release(old_sink);
259 This->pi &= ~PI_MIMEVERIFICATION; /* FIXME: more tests */
263 static void mime_available(BindProtocol *This, LPCWSTR mime, BOOL verified)
265 IInternetProtocol *mime_filter;
268 heap_free(This->mime);
271 mime_filter = get_mime_filter(mime);
273 TRACE("Got mime filter for %s\n", debugstr_w(mime));
275 hres = handle_mime_filter(This, mime_filter, mime);
276 IInternetProtocol_Release(mime_filter);
278 FIXME("MIME filter failed: %08x\n", hres);
280 This->mime = heap_strdupW(mime);
282 if(verified || !(This->pi & PI_MIMEVERIFICATION)) {
283 This->reported_mime = TRUE;
285 if(This->protocol_sink)
286 IInternetProtocolSink_ReportProgress(This->protocol_sink, BINDSTATUS_MIMETYPEAVAILABLE, mime);
291 #define PROTOCOL_THIS(iface) DEFINE_THIS(BindProtocol, IInternetProtocol, iface)
293 static HRESULT WINAPI BindProtocol_QueryInterface(IInternetProtocol *iface, REFIID riid, void **ppv)
295 BindProtocol *This = PROTOCOL_THIS(iface);
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);
325 IWinInetInfo *inet_info;
328 hres = IInternetProtocol_QueryInterface(This->protocol, &IID_IWinInetInfo, (void**)&inet_info);
329 if(SUCCEEDED(hres)) {
330 *ppv = HTTPINFO(This);
331 IWinInetInfo_Release(inet_info);
334 }else if(IsEqualGUID(&IID_IWinInetHttpInfo, riid)) {
335 TRACE("(%p)->(IID_IWinInetHttpInfo %p)\n", This, ppv);
338 IWinInetHttpInfo *http_info;
341 hres = IInternetProtocol_QueryInterface(This->protocol, &IID_IWinInetHttpInfo, (void**)&http_info);
342 if(SUCCEEDED(hres)) {
343 *ppv = HTTPINFO(This);
344 IWinInetHttpInfo_Release(http_info);
348 WARN("not supported interface %s\n", debugstr_guid(riid));
352 return E_NOINTERFACE;
354 IUnknown_AddRef((IUnknown*)*ppv);
358 static ULONG WINAPI BindProtocol_AddRef(IInternetProtocol *iface)
360 BindProtocol *This = PROTOCOL_THIS(iface);
361 LONG ref = InterlockedIncrement(&This->ref);
362 TRACE("(%p) ref=%d\n", This, ref);
366 static ULONG WINAPI BindProtocol_Release(IInternetProtocol *iface)
368 BindProtocol *This = PROTOCOL_THIS(iface);
369 LONG ref = InterlockedDecrement(&This->ref);
371 TRACE("(%p) ref=%d\n", This, ref);
374 if(This->wininet_info)
375 IWinInetInfo_Release(This->wininet_info);
377 IInternetProtocol_Release(This->protocol);
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));
385 set_binding_sink(PROTOCOL(This), NULL);
388 release_notif_hwnd(This->notif_hwnd);
389 DeleteCriticalSection(&This->section);
391 heap_free(This->mime);
392 heap_free(This->url);
395 URLMON_UnlockModule();
401 static HRESULT WINAPI BindProtocol_Start(IInternetProtocol *iface, LPCWSTR szUrl,
402 IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo,
403 DWORD grfPI, HANDLE_PTR dwReserved)
405 BindProtocol *This = PROTOCOL_THIS(iface);
407 TRACE("(%p)->(%s %p %p %08x %lx)\n", This, debugstr_w(szUrl), pOIProtSink,
408 pOIBindInfo, grfPI, dwReserved);
410 return IInternetProtocol_Start(This->protocol_handler, szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved);
413 static HRESULT WINAPI BindProtocol_Continue(IInternetProtocol *iface, PROTOCOLDATA *pProtocolData)
415 BindProtocol *This = PROTOCOL_THIS(iface);
417 TRACE("(%p)->(%p)\n", This, pProtocolData);
419 return IInternetProtocol_Continue(This->protocol_handler, pProtocolData);
422 static HRESULT WINAPI BindProtocol_Abort(IInternetProtocol *iface, HRESULT hrReason,
425 BindProtocol *This = PROTOCOL_THIS(iface);
426 FIXME("(%p)->(%08x %08x)\n", This, hrReason, dwOptions);
430 static HRESULT WINAPI BindProtocol_Terminate(IInternetProtocol *iface, DWORD dwOptions)
432 BindProtocol *This = PROTOCOL_THIS(iface);
434 TRACE("(%p)->(%08x)\n", This, dwOptions);
436 return IInternetProtocol_Terminate(This->protocol_handler, dwOptions);
439 static HRESULT WINAPI BindProtocol_Suspend(IInternetProtocol *iface)
441 BindProtocol *This = PROTOCOL_THIS(iface);
442 FIXME("(%p)\n", This);
446 static HRESULT WINAPI BindProtocol_Resume(IInternetProtocol *iface)
448 BindProtocol *This = PROTOCOL_THIS(iface);
449 FIXME("(%p)\n", This);
453 static HRESULT WINAPI BindProtocol_Read(IInternetProtocol *iface, void *pv,
454 ULONG cb, ULONG *pcbRead)
456 BindProtocol *This = PROTOCOL_THIS(iface);
458 TRACE("(%p)->(%p %u %p)\n", This, pv, cb, pcbRead);
462 return IInternetProtocol_Read(This->protocol_handler, pv, cb, pcbRead);
465 static HRESULT WINAPI BindProtocol_Seek(IInternetProtocol *iface, LARGE_INTEGER dlibMove,
466 DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
468 BindProtocol *This = PROTOCOL_THIS(iface);
469 FIXME("(%p)->(%d %d %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
473 static HRESULT WINAPI BindProtocol_LockRequest(IInternetProtocol *iface, DWORD dwOptions)
475 BindProtocol *This = PROTOCOL_THIS(iface);
477 TRACE("(%p)->(%08x)\n", This, dwOptions);
479 return IInternetProtocol_LockRequest(This->protocol_handler, dwOptions);
482 static HRESULT WINAPI BindProtocol_UnlockRequest(IInternetProtocol *iface)
484 BindProtocol *This = PROTOCOL_THIS(iface);
486 TRACE("(%p)\n", This);
488 return IInternetProtocol_UnlockRequest(This->protocol_handler);
491 void set_binding_sink(IInternetProtocol *bind_protocol, IInternetProtocolSink *sink)
493 BindProtocol *This = PROTOCOL_THIS(bind_protocol);
494 IInternetProtocolSink *prev_sink;
495 IServiceProvider *service_provider = NULL;
498 IInternetProtocolSink_AddRef(sink);
499 prev_sink = InterlockedExchangePointer((void**)&This->protocol_sink, sink);
501 IInternetProtocolSink_Release(prev_sink);
504 IInternetProtocolSink_QueryInterface(sink, &IID_IServiceProvider, (void**)&service_provider);
505 service_provider = InterlockedExchangePointer((void**)&This->service_provider, service_provider);
507 IServiceProvider_Release(service_provider);
510 IWinInetInfo *get_wininet_info(IInternetProtocol *bind_protocol)
512 BindProtocol *This = PROTOCOL_THIS(bind_protocol);
514 return This->wininet_info;
519 static const IInternetProtocolVtbl BindProtocolVtbl = {
520 BindProtocol_QueryInterface,
522 BindProtocol_Release,
524 BindProtocol_Continue,
526 BindProtocol_Terminate,
527 BindProtocol_Suspend,
531 BindProtocol_LockRequest,
532 BindProtocol_UnlockRequest
535 #define PROTOCOLHANDLER_THIS(iface) DEFINE_THIS(BindProtocol, IInternetProtocolHandler, iface)
537 static HRESULT WINAPI ProtocolHandler_QueryInterface(IInternetProtocol *iface, REFIID riid, void **ppv)
539 ERR("should not be called\n");
540 return E_NOINTERFACE;
543 static ULONG WINAPI ProtocolHandler_AddRef(IInternetProtocol *iface)
545 BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
546 return IInternetProtocol_AddRef(PROTOCOL(This));
549 static ULONG WINAPI ProtocolHandler_Release(IInternetProtocol *iface)
551 BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
552 return IInternetProtocol_Release(PROTOCOL(This));
555 static HRESULT WINAPI ProtocolHandler_Start(IInternetProtocol *iface, LPCWSTR szUrl,
556 IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo,
557 DWORD grfPI, HANDLE_PTR dwReserved)
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;
568 TRACE("(%p)->(%s %p %p %08x %lx)\n", This, debugstr_w(szUrl), pOIProtSink,
569 pOIBindInfo, grfPI, dwReserved);
571 if(!szUrl || !pOIProtSink || !pOIBindInfo)
575 This->url = heap_strdupW(szUrl);
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);
590 hres = get_protocol_handler(szUrl, &clsid, &urlmon_protocol, &cf);
594 if(This->from_urlmon) {
595 hres = IClassFactory_CreateInstance(cf, NULL, &IID_IInternetProtocol, (void**)&protocol);
596 IClassFactory_Release(cf);
600 hres = IClassFactory_CreateInstance(cf, (IUnknown*)BINDINFO(This),
601 &IID_IUnknown, (void**)&unk);
602 IClassFactory_Release(cf);
606 hres = IUnknown_QueryInterface(unk, &IID_IInternetProtocol, (void**)&protocol);
607 IUnknown_Release(unk);
613 StringFromCLSID(&clsid, &clsid_str);
614 IInternetProtocolSink_ReportProgress(pOIProtSink, BINDSTATUS_PROTOCOLCLASSID, clsid_str);
615 CoTaskMemFree(clsid_str);
617 This->protocol = protocol;
620 IInternetProtocol_QueryInterface(protocol, &IID_IWinInetInfo, (void**)&This->wininet_info);
622 IInternetBindInfo_AddRef(pOIBindInfo);
623 This->bind_info = pOIBindInfo;
625 set_binding_sink(PROTOCOL(This), pOIProtSink);
627 hres = IInternetProtocol_QueryInterface(protocol, &IID_IInternetPriority, (void**)&priority);
628 if(SUCCEEDED(hres)) {
629 IInternetPriority_SetPriority(priority, This->priority);
630 IInternetPriority_Release(priority);
633 return IInternetProtocol_Start(protocol, szUrl, PROTSINK(This), BINDINFO(This), 0, 0);
636 static HRESULT WINAPI ProtocolHandler_Continue(IInternetProtocol *iface, PROTOCOLDATA *pProtocolData)
638 BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
641 TRACE("(%p)->(%p)\n", This, pProtocolData);
643 hres = IInternetProtocol_Continue(This->protocol, pProtocolData);
645 heap_free(pProtocolData);
649 static HRESULT WINAPI ProtocolHandler_Abort(IInternetProtocol *iface, HRESULT hrReason,
652 BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
653 FIXME("(%p)->(%08x %08x)\n", This, hrReason, dwOptions);
657 static HRESULT WINAPI ProtocolHandler_Terminate(IInternetProtocol *iface, DWORD dwOptions)
659 BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
661 TRACE("(%p)->(%08x)\n", This, dwOptions);
663 if(!This->reported_result)
666 IInternetProtocol_Terminate(This->protocol, 0);
668 if(This->filter_proxy) {
669 IInternetProtocol_Release(PROTOCOL(This->filter_proxy));
670 This->filter_proxy = NULL;
673 set_binding_sink(PROTOCOL(This), NULL);
675 if(This->bind_info) {
676 IInternetBindInfo_Release(This->bind_info);
677 This->bind_info = NULL;
683 static HRESULT WINAPI ProtocolHandler_Suspend(IInternetProtocol *iface)
685 BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
686 FIXME("(%p)\n", This);
690 static HRESULT WINAPI ProtocolHandler_Resume(IInternetProtocol *iface)
692 BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
693 FIXME("(%p)\n", This);
697 static HRESULT WINAPI ProtocolHandler_Read(IInternetProtocol *iface, void *pv,
698 ULONG cb, ULONG *pcbRead)
700 BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
704 TRACE("(%p)->(%p %u %p)\n", This, pv, cb, pcbRead);
707 read = min(cb, This->buf_size);
708 memcpy(pv, This->buf, read);
710 if(read == This->buf_size) {
711 heap_free(This->buf);
714 memmove(This->buf, This->buf+cb, This->buf_size-cb);
717 This->buf_size -= read;
723 hres = IInternetProtocol_Read(This->protocol, (BYTE*)pv+read, cb-read, &cread);
731 static HRESULT WINAPI ProtocolHandler_Seek(IInternetProtocol *iface, LARGE_INTEGER dlibMove,
732 DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
734 BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
735 FIXME("(%p)->(%d %d %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
739 static HRESULT WINAPI ProtocolHandler_LockRequest(IInternetProtocol *iface, DWORD dwOptions)
741 BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
743 TRACE("(%p)->(%08x)\n", This, dwOptions);
745 return IInternetProtocol_LockRequest(This->protocol, dwOptions);
748 static HRESULT WINAPI ProtocolHandler_UnlockRequest(IInternetProtocol *iface)
750 BindProtocol *This = PROTOCOLHANDLER_THIS(iface);
752 TRACE("(%p)\n", This);
754 return IInternetProtocol_UnlockRequest(This->protocol);
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
775 #define BINDINFO_THIS(iface) DEFINE_THIS(BindProtocol, InternetBindInfo, iface)
777 static HRESULT WINAPI BindInfo_QueryInterface(IInternetBindInfo *iface,
778 REFIID riid, void **ppv)
780 BindProtocol *This = BINDINFO_THIS(iface);
781 return IInternetProtocol_QueryInterface(PROTOCOL(This), riid, ppv);
784 static ULONG WINAPI BindInfo_AddRef(IInternetBindInfo *iface)
786 BindProtocol *This = BINDINFO_THIS(iface);
787 return IBinding_AddRef(PROTOCOL(This));
790 static ULONG WINAPI BindInfo_Release(IInternetBindInfo *iface)
792 BindProtocol *This = BINDINFO_THIS(iface);
793 return IBinding_Release(PROTOCOL(This));
796 static HRESULT WINAPI BindInfo_GetBindInfo(IInternetBindInfo *iface,
797 DWORD *grfBINDF, BINDINFO *pbindinfo)
799 BindProtocol *This = BINDINFO_THIS(iface);
802 TRACE("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo);
804 hres = IInternetBindInfo_GetBindInfo(This->bind_info, grfBINDF, pbindinfo);
806 WARN("GetBindInfo failed: %08x\n", hres);
810 *grfBINDF |= BINDF_FROMURLMON;
814 static HRESULT WINAPI BindInfo_GetBindString(IInternetBindInfo *iface,
815 ULONG ulStringType, LPOLESTR *ppwzStr, ULONG cEl, ULONG *pcElFetched)
817 BindProtocol *This = BINDINFO_THIS(iface);
819 TRACE("(%p)->(%d %p %d %p)\n", This, ulStringType, ppwzStr, cEl, pcElFetched);
821 return IInternetBindInfo_GetBindString(This->bind_info, ulStringType, ppwzStr, cEl, pcElFetched);
826 static const IInternetBindInfoVtbl InternetBindInfoVtbl = {
827 BindInfo_QueryInterface,
830 BindInfo_GetBindInfo,
831 BindInfo_GetBindString
834 #define PRIORITY_THIS(iface) DEFINE_THIS(BindProtocol, InternetPriority, iface)
836 static HRESULT WINAPI InternetPriority_QueryInterface(IInternetPriority *iface,
837 REFIID riid, void **ppv)
839 BindProtocol *This = PRIORITY_THIS(iface);
840 return IInternetProtocol_QueryInterface(PROTOCOL(This), riid, ppv);
843 static ULONG WINAPI InternetPriority_AddRef(IInternetPriority *iface)
845 BindProtocol *This = PRIORITY_THIS(iface);
846 return IInternetProtocol_AddRef(PROTOCOL(This));
849 static ULONG WINAPI InternetPriority_Release(IInternetPriority *iface)
851 BindProtocol *This = PRIORITY_THIS(iface);
852 return IInternetProtocol_Release(PROTOCOL(This));
855 static HRESULT WINAPI InternetPriority_SetPriority(IInternetPriority *iface, LONG nPriority)
857 BindProtocol *This = PRIORITY_THIS(iface);
859 TRACE("(%p)->(%d)\n", This, nPriority);
861 This->priority = nPriority;
865 static HRESULT WINAPI InternetPriority_GetPriority(IInternetPriority *iface, LONG *pnPriority)
867 BindProtocol *This = PRIORITY_THIS(iface);
869 TRACE("(%p)->(%p)\n", This, pnPriority);
871 *pnPriority = This->priority;
877 static const IInternetPriorityVtbl InternetPriorityVtbl = {
878 InternetPriority_QueryInterface,
879 InternetPriority_AddRef,
880 InternetPriority_Release,
881 InternetPriority_SetPriority,
882 InternetPriority_GetPriority
886 #define PROTSINK_THIS(iface) DEFINE_THIS(BindProtocol, IInternetProtocolSink, iface)
888 static HRESULT WINAPI BPInternetProtocolSink_QueryInterface(IInternetProtocolSink *iface,
889 REFIID riid, void **ppv)
891 BindProtocol *This = PROTSINK_THIS(iface);
892 return IInternetProtocol_QueryInterface(PROTOCOL(This), riid, ppv);
895 static ULONG WINAPI BPInternetProtocolSink_AddRef(IInternetProtocolSink *iface)
897 BindProtocol *This = PROTSINK_THIS(iface);
898 return IInternetProtocol_AddRef(PROTOCOL(This));
901 static ULONG WINAPI BPInternetProtocolSink_Release(IInternetProtocolSink *iface)
903 BindProtocol *This = PROTSINK_THIS(iface);
904 return IInternetProtocol_Release(PROTOCOL(This));
908 task_header_t header;
912 static void switch_proc(BindProtocol *bind, task_header_t *t)
914 switch_task_t *task = (switch_task_t*)t;
916 IInternetProtocol_Continue(bind->protocol_handler, task->data);
921 static HRESULT WINAPI BPInternetProtocolSink_Switch(IInternetProtocolSink *iface,
922 PROTOCOLDATA *pProtocolData)
924 BindProtocol *This = PROTSINK_THIS(iface);
927 TRACE("(%p)->(%p)\n", This, pProtocolData);
929 TRACE("flags %x state %x data %p cb %u\n", pProtocolData->grfFlags, pProtocolData->dwState,
930 pProtocolData->pData, pProtocolData->cbData);
932 data = heap_alloc(sizeof(PROTOCOLDATA));
934 return E_OUTOFMEMORY;
935 memcpy(data, pProtocolData, sizeof(PROTOCOLDATA));
937 if(!do_direct_notif(This)) {
940 task = heap_alloc(sizeof(switch_task_t));
942 return E_OUTOFMEMORY;
946 push_task(This, &task->header, switch_proc);
950 if(!This->protocol_sink) {
951 IInternetProtocol_Continue(This->protocol_handler, data);
955 return IInternetProtocolSink_Switch(This->protocol_sink, data);
958 static void report_progress(BindProtocol *This, ULONG status_code, LPCWSTR status_text)
960 switch(status_code) {
961 case BINDSTATUS_FINDINGRESOURCE:
962 case BINDSTATUS_CONNECTING:
963 case BINDSTATUS_BEGINDOWNLOADDATA:
964 case BINDSTATUS_SENDINGREQUEST:
965 case BINDSTATUS_CACHEFILENAMEAVAILABLE:
966 case BINDSTATUS_DIRECTBIND:
967 case BINDSTATUS_ACCEPTRANGES:
968 if(This->protocol_sink)
969 IInternetProtocolSink_ReportProgress(This->protocol_sink, status_code, status_text);
972 case BINDSTATUS_MIMETYPEAVAILABLE:
973 mime_available(This, status_text, FALSE);
976 case BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE:
977 mime_available(This, status_text, TRUE);
981 FIXME("unsupported ulStatusCode %u\n", status_code);
986 task_header_t header;
990 } on_progress_task_t;
992 static void on_progress_proc(BindProtocol *This, task_header_t *t)
994 on_progress_task_t *task = (on_progress_task_t*)t;
996 report_progress(This, task->status_code, task->status_text);
998 heap_free(task->status_text);
1002 static HRESULT WINAPI BPInternetProtocolSink_ReportProgress(IInternetProtocolSink *iface,
1003 ULONG ulStatusCode, LPCWSTR szStatusText)
1005 BindProtocol *This = PROTSINK_THIS(iface);
1007 TRACE("(%p)->(%u %s)\n", This, ulStatusCode, debugstr_w(szStatusText));
1009 if(do_direct_notif(This)) {
1010 report_progress(This, ulStatusCode, szStatusText);
1012 on_progress_task_t *task;
1014 task = heap_alloc(sizeof(on_progress_task_t));
1016 task->status_code = ulStatusCode;
1017 task->status_text = heap_strdupW(szStatusText);
1019 push_task(This, &task->header, on_progress_proc);
1025 static HRESULT report_data(BindProtocol *This, DWORD bscf, ULONG progress, ULONG progress_max)
1027 if(!This->protocol_sink)
1030 if((This->pi & PI_MIMEVERIFICATION) && !This->reported_mime) {
1031 BYTE buf[BUFFER_SIZE];
1038 hres = IInternetProtocol_Read(This->protocol, buf,
1039 sizeof(buf)-This->buf_size, &read);
1040 if(FAILED(hres) && hres != E_PENDING)
1044 This->buf = heap_alloc(BUFFER_SIZE);
1046 return E_OUTOFMEMORY;
1047 }else if(read + This->buf_size > BUFFER_SIZE) {
1050 tmp = heap_realloc(This->buf, read+This->buf_size);
1052 return E_OUTOFMEMORY;
1056 memcpy(This->buf+This->buf_size, buf, read);
1057 This->buf_size += read;
1058 }while(This->buf_size < MIME_TEST_SIZE && hres == S_OK);
1060 if(This->buf_size < MIME_TEST_SIZE && hres != S_FALSE)
1063 bscf = BSCF_FIRSTDATANOTIFICATION;
1065 bscf |= BSCF_LASTDATANOTIFICATION|BSCF_DATAFULLYAVAILABLE;
1067 if(!This->reported_mime) {
1068 hres = FindMimeFromData(NULL, This->url, This->buf, min(This->buf_size, MIME_TEST_SIZE),
1069 This->mime, 0, &mime, 0);
1073 mime_available(This, mime, TRUE);
1074 CoTaskMemFree(mime);
1078 if(!This->protocol_sink)
1081 return IInternetProtocolSink_ReportData(This->protocol_sink, bscf, progress, progress_max);
1085 task_header_t header;
1089 } report_data_task_t;
1091 static void report_data_proc(BindProtocol *This, task_header_t *t)
1093 report_data_task_t *task = (report_data_task_t*)t;
1095 report_data(This, task->bscf, task->progress, task->progress_max);
1099 static HRESULT WINAPI BPInternetProtocolSink_ReportData(IInternetProtocolSink *iface,
1100 DWORD grfBSCF, ULONG ulProgress, ULONG ulProgressMax)
1102 BindProtocol *This = PROTSINK_THIS(iface);
1104 TRACE("(%p)->(%d %u %u)\n", This, grfBSCF, ulProgress, ulProgressMax);
1106 if(!This->protocol_sink)
1109 if(!do_direct_notif(This)) {
1110 report_data_task_t *task;
1112 task = heap_alloc(sizeof(report_data_task_t));
1114 return E_OUTOFMEMORY;
1116 task->bscf = grfBSCF;
1117 task->progress = ulProgress;
1118 task->progress_max = ulProgressMax;
1120 push_task(This, &task->header, report_data_proc);
1124 return report_data(This, grfBSCF, ulProgress, ulProgressMax);
1128 task_header_t header;
1133 } report_result_task_t;
1135 static void report_result_proc(BindProtocol *This, task_header_t *t)
1137 report_result_task_t *task = (report_result_task_t*)t;
1139 if(This->protocol_sink)
1140 IInternetProtocolSink_ReportResult(This->protocol_sink, task->hres, task->err, task->str);
1142 heap_free(task->str);
1146 static HRESULT WINAPI BPInternetProtocolSink_ReportResult(IInternetProtocolSink *iface,
1147 HRESULT hrResult, DWORD dwError, LPCWSTR szResult)
1149 BindProtocol *This = PROTSINK_THIS(iface);
1151 TRACE("(%p)->(%08x %d %s)\n", This, hrResult, dwError, debugstr_w(szResult));
1153 if(!This->protocol_sink)
1156 This->reported_result = TRUE;
1158 if(!do_direct_notif(This)) {
1159 report_result_task_t *task;
1161 task = heap_alloc(sizeof(report_result_task_t));
1163 return E_OUTOFMEMORY;
1165 task->hres = hrResult;
1166 task->err = dwError;
1167 task->str = heap_strdupW(szResult);
1169 push_task(This, &task->header, report_result_proc);
1173 return IInternetProtocolSink_ReportResult(This->protocol_sink, hrResult, dwError, szResult);
1176 #undef PROTSINK_THIS
1178 static const IInternetProtocolSinkVtbl InternetProtocolSinkVtbl = {
1179 BPInternetProtocolSink_QueryInterface,
1180 BPInternetProtocolSink_AddRef,
1181 BPInternetProtocolSink_Release,
1182 BPInternetProtocolSink_Switch,
1183 BPInternetProtocolSink_ReportProgress,
1184 BPInternetProtocolSink_ReportData,
1185 BPInternetProtocolSink_ReportResult
1188 #define INETINFO_THIS(iface) DEFINE_THIS(BindProtocol, IWinInetHttpInfo, iface)
1190 static HRESULT WINAPI WinInetHttpInfo_QueryInterface(IWinInetHttpInfo *iface, REFIID riid, void **ppv)
1192 BindProtocol *This = INETINFO_THIS(iface);
1193 return IInternetProtocol_QueryInterface(PROTOCOL(This), riid, ppv);
1196 static ULONG WINAPI WinInetHttpInfo_AddRef(IWinInetHttpInfo *iface)
1198 BindProtocol *This = INETINFO_THIS(iface);
1199 return IInternetProtocol_AddRef(PROTOCOL(This));
1202 static ULONG WINAPI WinInetHttpInfo_Release(IWinInetHttpInfo *iface)
1204 BindProtocol *This = INETINFO_THIS(iface);
1205 return IInternetProtocol_Release(PROTOCOL(This));
1208 static HRESULT WINAPI WinInetHttpInfo_QueryOption(IWinInetHttpInfo *iface, DWORD dwOption,
1209 void *pBuffer, DWORD *pcbBuffer)
1211 BindProtocol *This = INETINFO_THIS(iface);
1212 FIXME("(%p)->(%x %p %p)\n", This, dwOption, pBuffer, pcbBuffer);
1216 static HRESULT WINAPI WinInetHttpInfo_QueryInfo(IWinInetHttpInfo *iface, DWORD dwOption,
1217 void *pBuffer, DWORD *pcbBuffer, DWORD *pdwFlags, DWORD *pdwReserved)
1219 BindProtocol *This = INETINFO_THIS(iface);
1220 FIXME("(%p)->(%x %p %p %p %p)\n", This, dwOption, pBuffer, pcbBuffer, pdwFlags, pdwReserved);
1224 #undef INETINFO_THIS
1226 static const IWinInetHttpInfoVtbl WinInetHttpInfoVtbl = {
1227 WinInetHttpInfo_QueryInterface,
1228 WinInetHttpInfo_AddRef,
1229 WinInetHttpInfo_Release,
1230 WinInetHttpInfo_QueryOption,
1231 WinInetHttpInfo_QueryInfo
1234 #define SERVPROV_THIS(iface) DEFINE_THIS(BindProtocol, ServiceProvider, iface)
1236 static HRESULT WINAPI BPServiceProvider_QueryInterface(IServiceProvider *iface,
1237 REFIID riid, void **ppv)
1239 BindProtocol *This = SERVPROV_THIS(iface);
1240 return IInternetProtocol_QueryInterface(PROTOCOL(This), riid, ppv);
1243 static ULONG WINAPI BPServiceProvider_AddRef(IServiceProvider *iface)
1245 BindProtocol *This = SERVPROV_THIS(iface);
1246 return IInternetProtocol_AddRef(PROTOCOL(This));
1249 static ULONG WINAPI BPServiceProvider_Release(IServiceProvider *iface)
1251 BindProtocol *This = SERVPROV_THIS(iface);
1252 return IInternetProtocol_Release(PROTOCOL(This));
1255 static HRESULT WINAPI BPServiceProvider_QueryService(IServiceProvider *iface,
1256 REFGUID guidService, REFIID riid, void **ppv)
1258 BindProtocol *This = SERVPROV_THIS(iface);
1260 TRACE("(%p)->(%s %s %p)\n", This, debugstr_guid(guidService), debugstr_guid(riid), ppv);
1262 if(!This->service_provider)
1263 return E_NOINTERFACE;
1265 return IServiceProvider_QueryService(This->service_provider, guidService, riid, ppv);
1268 #undef SERVPROV_THIS
1270 static const IServiceProviderVtbl ServiceProviderVtbl = {
1271 BPServiceProvider_QueryInterface,
1272 BPServiceProvider_AddRef,
1273 BPServiceProvider_Release,
1274 BPServiceProvider_QueryService
1277 HRESULT create_binding_protocol(LPCWSTR url, BOOL from_urlmon, IInternetProtocol **protocol)
1279 BindProtocol *ret = heap_alloc_zero(sizeof(BindProtocol));
1281 ret->lpIInternetProtocolVtbl = &BindProtocolVtbl;
1282 ret->lpInternetBindInfoVtbl = &InternetBindInfoVtbl;
1283 ret->lpInternetPriorityVtbl = &InternetPriorityVtbl;
1284 ret->lpServiceProviderVtbl = &ServiceProviderVtbl;
1285 ret->lpIInternetProtocolSinkVtbl = &InternetProtocolSinkVtbl;
1286 ret->lpIInternetProtocolHandlerVtbl = &InternetProtocolHandlerVtbl;
1287 ret->lpIWinInetHttpInfoVtbl = &WinInetHttpInfoVtbl;
1290 ret->from_urlmon = from_urlmon;
1291 ret->apartment_thread = GetCurrentThreadId();
1292 ret->notif_hwnd = get_notif_hwnd();
1293 ret->protocol_handler = PROTOCOLHANDLER(ret);
1294 InitializeCriticalSection(&ret->section);
1296 URLMON_LockModule();
1298 *protocol = PROTOCOL(ret);