2 * Copyright 2005-2007 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"
21 static WCHAR cbinding_contextW
[] = {'C','B','i','n','d','i','n','g',' ','C','o','n','t','e','x','t',0};
22 static WCHAR bscb_holderW
[] = { '_','B','S','C','B','_','H','o','l','d','e','r','_',0 };
25 IUnknown IUnknown_iface
;
29 IInternetProtocolEx
*protocol
;
37 typedef struct _stgmed_obj_t stgmed_obj_t
;
40 void (*release
)(stgmed_obj_t
*);
41 HRESULT (*fill_stgmed
)(stgmed_obj_t
*,STGMEDIUM
*);
42 HRESULT (*get_result
)(stgmed_obj_t
*,DWORD
,void**);
45 struct _stgmed_obj_t
{
46 const stgmed_obj_vtbl
*vtbl
;
55 #define BINDING_LOCKED 0x0001
56 #define BINDING_STOPPED 0x0002
57 #define BINDING_OBJAVAIL 0x0004
58 #define BINDING_ABORTED 0x0008
61 IBinding IBinding_iface
;
62 IInternetProtocolSink IInternetProtocolSink_iface
;
63 IInternetBindInfo IInternetBindInfo_iface
;
64 IWinInetHttpInfo IWinInetHttpInfo_iface
;
65 IServiceProvider IServiceProvider_iface
;
69 IBindStatusCallback
*callback
;
70 IServiceProvider
*service_provider
;
72 BindProtocol
*protocol
;
74 stgmed_buf_t
*stgmed_buf
;
75 stgmed_obj_t
*stgmed_obj
;
81 UINT clipboard_format
;
90 download_state_t download_state
;
96 CRITICAL_SECTION section
;
99 static void read_protocol_data(stgmed_buf_t
*stgmed_buf
)
105 do hres
= IInternetProtocolEx_Read(stgmed_buf
->protocol
, buf
, sizeof(buf
), &read
);
109 static void dump_BINDINFO(BINDINFO
*bi
)
111 static const char * const BINDINFOF_str
[] = {
113 "BINDINFOF_URLENCODESTGMEDDATA",
114 "BINDINFOF_URLENCODEDEXTRAINFO"
117 static const char * const BINDVERB_str
[] = {
131 " %d, %08x, %d, %d\n"
137 bi
->cbSize
, debugstr_w(bi
->szExtraInfo
),
138 bi
->stgmedData
.tymed
, bi
->stgmedData
.u
.hGlobal
, bi
->stgmedData
.pUnkForRelease
,
139 bi
->grfBindInfoF
> BINDINFOF_URLENCODEDEXTRAINFO
140 ? "unknown" : BINDINFOF_str
[bi
->grfBindInfoF
],
141 bi
->dwBindVerb
> BINDVERB_CUSTOM
142 ? "unknown" : BINDVERB_str
[bi
->dwBindVerb
],
143 debugstr_w(bi
->szCustomVerb
),
144 bi
->cbstgmedData
, bi
->dwOptions
, bi
->dwOptionsFlags
, bi
->dwCodePage
,
145 bi
->securityAttributes
.nLength
,
146 bi
->securityAttributes
.lpSecurityDescriptor
,
147 bi
->securityAttributes
.bInheritHandle
,
148 debugstr_guid(&bi
->iid
),
149 bi
->pUnk
, bi
->dwReserved
153 static void mime_available(Binding
*This
, LPCWSTR mime
)
155 heap_free(This
->mime
);
156 This
->mime
= heap_strdupW(mime
);
158 if(!This
->mime
|| !This
->report_mime
)
161 IBindStatusCallback_OnProgress(This
->callback
, 0, 0, BINDSTATUS_MIMETYPEAVAILABLE
, This
->mime
);
163 This
->clipboard_format
= RegisterClipboardFormatW(This
->mime
);
166 static void stop_binding(Binding
*binding
, HRESULT hres
, LPCWSTR str
)
168 if(binding
->state
& BINDING_LOCKED
) {
169 IInternetProtocolEx_UnlockRequest(&binding
->protocol
->IInternetProtocolEx_iface
);
170 binding
->state
&= ~BINDING_LOCKED
;
173 if(!(binding
->state
& BINDING_STOPPED
)) {
174 binding
->state
|= BINDING_STOPPED
;
176 binding
->hres
= hres
;
177 IBindStatusCallback_OnStopBinding(binding
->callback
, hres
, str
);
181 static LPWSTR
get_mime_clsid(LPCWSTR mime
, CLSID
*clsid
)
183 LPWSTR key_name
, ret
;
184 DWORD res
, type
, size
;
189 static const WCHAR mime_keyW
[] =
190 {'M','I','M','E','\\','D','a','t','a','b','a','s','e','\\',
191 'C','o','n','t','e','n','t',' ','T','y','p','e','\\'};
192 static const WCHAR clsidW
[] = {'C','L','S','I','D',0};
194 len
= strlenW(mime
)+1;
195 key_name
= heap_alloc(sizeof(mime_keyW
) + len
*sizeof(WCHAR
));
196 memcpy(key_name
, mime_keyW
, sizeof(mime_keyW
));
197 strcpyW(key_name
+ sizeof(mime_keyW
)/sizeof(WCHAR
), mime
);
199 res
= RegOpenKeyW(HKEY_CLASSES_ROOT
, key_name
, &hkey
);
201 if(res
!= ERROR_SUCCESS
) {
202 WARN("Could not open MIME key: %x\n", res
);
206 size
= 50*sizeof(WCHAR
);
207 ret
= heap_alloc(size
);
208 res
= RegQueryValueExW(hkey
, clsidW
, NULL
, &type
, (LPBYTE
)ret
, &size
);
210 if(res
!= ERROR_SUCCESS
) {
211 WARN("Could not get CLSID: %08x\n", res
);
216 hres
= CLSIDFromString(ret
, clsid
);
218 WARN("Could not parse CLSID: %08x\n", hres
);
226 static void load_doc_mon(Binding
*binding
, IPersistMoniker
*persist
)
231 hres
= CreateAsyncBindCtxEx(binding
->bctx
, 0, NULL
, NULL
, &bctx
, 0);
233 WARN("CreateAsyncBindCtxEx failed: %08x\n", hres
);
237 IBindCtx_RevokeObjectParam(bctx
, bscb_holderW
);
238 IBindCtx_RegisterObjectParam(bctx
, cbinding_contextW
, (IUnknown
*)&binding
->IBinding_iface
);
240 hres
= IPersistMoniker_Load(persist
, binding
->download_state
== END_DOWNLOAD
, binding
->mon
, bctx
, 0x12);
241 IBindCtx_RevokeObjectParam(bctx
, cbinding_contextW
);
242 IBindCtx_Release(bctx
);
244 FIXME("Load failed: %08x\n", hres
);
247 static HRESULT
create_mime_object(Binding
*binding
, const CLSID
*clsid
, LPCWSTR clsid_str
)
249 IPersistMoniker
*persist
;
252 hres
= CoCreateInstance(clsid
, NULL
, CLSCTX_INPROC_SERVER
|CLSCTX_INPROC_HANDLER
,
253 &binding
->iid
, (void**)&binding
->obj
);
255 WARN("CoCreateInstance failed: %08x\n", hres
);
256 return INET_E_CANNOT_INSTANTIATE_OBJECT
;
259 binding
->state
|= BINDING_OBJAVAIL
;
261 hres
= IUnknown_QueryInterface(binding
->obj
, &IID_IPersistMoniker
, (void**)&persist
);
262 if(SUCCEEDED(hres
)) {
265 hres
= IPersistMoniker_QueryInterface(persist
, &IID_IMonikerProp
, (void**)&prop
);
266 if(SUCCEEDED(hres
)) {
267 IMonikerProp_PutProperty(prop
, MIMETYPEPROP
, binding
->mime
);
268 IMonikerProp_PutProperty(prop
, CLASSIDPROP
, clsid_str
);
269 IMonikerProp_Release(prop
);
272 load_doc_mon(binding
, persist
);
274 IPersistMoniker_Release(persist
);
276 FIXME("Could not get IPersistMoniker: %08x\n", hres
);
277 /* FIXME: Try query IPersistFile */
280 IBindStatusCallback_OnObjectAvailable(binding
->callback
, &binding
->iid
, binding
->obj
);
285 static void create_object(Binding
*binding
)
292 FIXME("MIME not available\n");
296 if((clsid_str
= get_mime_clsid(binding
->mime
, &clsid
)))
297 IBindStatusCallback_OnProgress(binding
->callback
, 0, 0, BINDSTATUS_CLASSIDAVAILABLE
, clsid_str
);
299 IBindStatusCallback_OnProgress(binding
->callback
, 0, 0, BINDSTATUS_BEGINSYNCOPERATION
, NULL
);
302 hres
= create_mime_object(binding
, &clsid
, clsid_str
);
303 heap_free(clsid_str
);
305 FIXME("Could not find object for MIME %s\n", debugstr_w(binding
->mime
));
306 hres
= REGDB_E_CLASSNOTREG
;
309 IBindStatusCallback_OnProgress(binding
->callback
, 0, 0, BINDSTATUS_ENDSYNCOPERATION
, NULL
);
310 binding
->clsid
= CLSID_NULL
;
312 stop_binding(binding
, hres
, NULL
);
314 IInternetProtocolEx_Terminate(&binding
->protocol
->IInternetProtocolEx_iface
, 0);
317 static void cache_file_available(Binding
*This
, const WCHAR
*file_name
)
319 heap_free(This
->stgmed_buf
->cache_file
);
320 This
->stgmed_buf
->cache_file
= heap_strdupW(file_name
);
322 if(This
->use_cache_file
) {
323 This
->stgmed_buf
->file
= CreateFileW(file_name
, GENERIC_READ
, FILE_SHARE_READ
|FILE_SHARE_WRITE
, NULL
,
324 OPEN_EXISTING
, FILE_ATTRIBUTE_NORMAL
, NULL
);
325 if(This
->stgmed_buf
->file
== INVALID_HANDLE_VALUE
)
326 WARN("CreateFile failed: %u\n", GetLastError());
330 static inline stgmed_buf_t
*impl_from_IUnknown(IUnknown
*iface
)
332 return CONTAINING_RECORD(iface
, stgmed_buf_t
, IUnknown_iface
);
335 static HRESULT WINAPI
StgMedUnk_QueryInterface(IUnknown
*iface
, REFIID riid
, void **ppv
)
337 stgmed_buf_t
*This
= impl_from_IUnknown(iface
);
341 if(IsEqualGUID(riid
, &IID_IUnknown
)) {
342 TRACE("(%p)->(IID_IUnknown %p)\n", This
, ppv
);
344 *ppv
= &This
->IUnknown_iface
;
345 IUnknown_AddRef(&This
->IUnknown_iface
);
349 TRACE("(%p)->(%s %p)\n", This
, debugstr_guid(riid
), ppv
);
350 return E_NOINTERFACE
;
353 static ULONG WINAPI
StgMedUnk_AddRef(IUnknown
*iface
)
355 stgmed_buf_t
*This
= impl_from_IUnknown(iface
);
356 LONG ref
= InterlockedIncrement(&This
->ref
);
358 TRACE("(%p) ref=%d\n", This
, ref
);
363 static ULONG WINAPI
StgMedUnk_Release(IUnknown
*iface
)
365 stgmed_buf_t
*This
= impl_from_IUnknown(iface
);
366 LONG ref
= InterlockedDecrement(&This
->ref
);
368 TRACE("(%p) ref=%d\n", This
, ref
);
371 if(This
->file
!= INVALID_HANDLE_VALUE
)
372 CloseHandle(This
->file
);
373 IInternetProtocolEx_Release(This
->protocol
);
374 heap_free(This
->cache_file
);
377 URLMON_UnlockModule();
383 static const IUnknownVtbl StgMedUnkVtbl
= {
384 StgMedUnk_QueryInterface
,
389 static stgmed_buf_t
*create_stgmed_buf(IInternetProtocolEx
*protocol
)
391 stgmed_buf_t
*ret
= heap_alloc(sizeof(*ret
));
393 ret
->IUnknown_iface
.lpVtbl
= &StgMedUnkVtbl
;
395 ret
->file
= INVALID_HANDLE_VALUE
;
397 ret
->cache_file
= NULL
;
399 IInternetProtocolEx_AddRef(protocol
);
400 ret
->protocol
= protocol
;
408 stgmed_obj_t stgmed_obj
;
409 IStream IStream_iface
;
416 static inline ProtocolStream
*impl_from_IStream(IStream
*iface
)
418 return CONTAINING_RECORD(iface
, ProtocolStream
, IStream_iface
);
421 static HRESULT WINAPI
ProtocolStream_QueryInterface(IStream
*iface
,
422 REFIID riid
, void **ppv
)
424 ProtocolStream
*This
= impl_from_IStream(iface
);
428 if(IsEqualGUID(&IID_IUnknown
, riid
)) {
429 TRACE("(%p)->(IID_IUnknown %p)\n", This
, ppv
);
430 *ppv
= &This
->IStream_iface
;
431 }else if(IsEqualGUID(&IID_ISequentialStream
, riid
)) {
432 TRACE("(%p)->(IID_ISequentialStream %p)\n", This
, ppv
);
433 *ppv
= &This
->IStream_iface
;
434 }else if(IsEqualGUID(&IID_IStream
, riid
)) {
435 TRACE("(%p)->(IID_IStream %p)\n", This
, ppv
);
436 *ppv
= &This
->IStream_iface
;
440 IStream_AddRef(&This
->IStream_iface
);
444 WARN("(%p)->(%s %p)\n", This
, debugstr_guid(riid
), ppv
);
445 return E_NOINTERFACE
;
448 static ULONG WINAPI
ProtocolStream_AddRef(IStream
*iface
)
450 ProtocolStream
*This
= impl_from_IStream(iface
);
451 LONG ref
= InterlockedIncrement(&This
->ref
);
453 TRACE("(%p) ref=%d\n", This
, ref
);
458 static ULONG WINAPI
ProtocolStream_Release(IStream
*iface
)
460 ProtocolStream
*This
= impl_from_IStream(iface
);
461 LONG ref
= InterlockedDecrement(&This
->ref
);
463 TRACE("(%p) ref=%d\n", This
, ref
);
466 IUnknown_Release(&This
->buf
->IUnknown_iface
);
469 URLMON_UnlockModule();
475 static HRESULT WINAPI
ProtocolStream_Read(IStream
*iface
, void *pv
,
476 ULONG cb
, ULONG
*pcbRead
)
478 ProtocolStream
*This
= impl_from_IStream(iface
);
482 TRACE("(%p)->(%p %d %p)\n", This
, pv
, cb
, pcbRead
);
484 if(This
->buf
->file
== INVALID_HANDLE_VALUE
) {
485 hres
= This
->buf
->hres
= IInternetProtocolEx_Read(This
->buf
->protocol
, (PBYTE
)pv
, cb
, &read
);
487 hres
= ReadFile(This
->buf
->file
, pv
, cb
, &read
, NULL
) ? S_OK
: INET_E_DOWNLOAD_FAILURE
;
493 if(hres
== E_PENDING
)
495 else if(FAILED(hres
))
496 FIXME("Read failed: %08x\n", hres
);
498 return read
? S_OK
: S_FALSE
;
501 static HRESULT WINAPI
ProtocolStream_Write(IStream
*iface
, const void *pv
,
502 ULONG cb
, ULONG
*pcbWritten
)
504 ProtocolStream
*This
= impl_from_IStream(iface
);
506 TRACE("(%p)->(%p %d %p)\n", This
, pv
, cb
, pcbWritten
);
508 return STG_E_ACCESSDENIED
;
511 static HRESULT WINAPI
ProtocolStream_Seek(IStream
*iface
, LARGE_INTEGER dlibMove
,
512 DWORD dwOrigin
, ULARGE_INTEGER
*plibNewPosition
)
514 ProtocolStream
*This
= impl_from_IStream(iface
);
515 FIXME("(%p)->(%d %08x %p)\n", This
, dlibMove
.u
.LowPart
, dwOrigin
, plibNewPosition
);
519 static HRESULT WINAPI
ProtocolStream_SetSize(IStream
*iface
, ULARGE_INTEGER libNewSize
)
521 ProtocolStream
*This
= impl_from_IStream(iface
);
522 FIXME("(%p)->(%d)\n", This
, libNewSize
.u
.LowPart
);
526 static HRESULT WINAPI
ProtocolStream_CopyTo(IStream
*iface
, IStream
*pstm
,
527 ULARGE_INTEGER cb
, ULARGE_INTEGER
*pcbRead
, ULARGE_INTEGER
*pcbWritten
)
529 ProtocolStream
*This
= impl_from_IStream(iface
);
530 FIXME("(%p)->(%p %d %p %p)\n", This
, pstm
, cb
.u
.LowPart
, pcbRead
, pcbWritten
);
534 static HRESULT WINAPI
ProtocolStream_Commit(IStream
*iface
, DWORD grfCommitFlags
)
536 ProtocolStream
*This
= impl_from_IStream(iface
);
538 TRACE("(%p)->(%08x)\n", This
, grfCommitFlags
);
543 static HRESULT WINAPI
ProtocolStream_Revert(IStream
*iface
)
545 ProtocolStream
*This
= impl_from_IStream(iface
);
547 TRACE("(%p)\n", This
);
552 static HRESULT WINAPI
ProtocolStream_LockRegion(IStream
*iface
, ULARGE_INTEGER libOffset
,
553 ULARGE_INTEGER cb
, DWORD dwLockType
)
555 ProtocolStream
*This
= impl_from_IStream(iface
);
556 FIXME("(%p)->(%d %d %d)\n", This
, libOffset
.u
.LowPart
, cb
.u
.LowPart
, dwLockType
);
560 static HRESULT WINAPI
ProtocolStream_UnlockRegion(IStream
*iface
,
561 ULARGE_INTEGER libOffset
, ULARGE_INTEGER cb
, DWORD dwLockType
)
563 ProtocolStream
*This
= impl_from_IStream(iface
);
564 FIXME("(%p)->(%d %d %d)\n", This
, libOffset
.u
.LowPart
, cb
.u
.LowPart
, dwLockType
);
568 static HRESULT WINAPI
ProtocolStream_Stat(IStream
*iface
, STATSTG
*pstatstg
,
571 ProtocolStream
*This
= impl_from_IStream(iface
);
572 TRACE("(%p)->(%p %08x)\n", This
, pstatstg
, dwStatFlag
);
577 memset(pstatstg
, 0, sizeof(STATSTG
));
579 if(!(dwStatFlag
&STATFLAG_NONAME
) && This
->buf
->cache_file
) {
580 pstatstg
->pwcsName
= CoTaskMemAlloc((lstrlenW(This
->buf
->cache_file
)+1)*sizeof(WCHAR
));
581 if(!pstatstg
->pwcsName
)
582 return STG_E_INSUFFICIENTMEMORY
;
584 lstrcpyW(pstatstg
->pwcsName
, This
->buf
->cache_file
);
587 pstatstg
->type
= STGTY_STREAM
;
588 if(This
->buf
->file
!= INVALID_HANDLE_VALUE
) {
589 GetFileSizeEx(This
->buf
->file
, (PLARGE_INTEGER
)&pstatstg
->cbSize
);
590 GetFileTime(This
->buf
->file
, &pstatstg
->ctime
, &pstatstg
->atime
, &pstatstg
->mtime
);
591 if(pstatstg
->cbSize
.QuadPart
)
592 pstatstg
->grfMode
= GENERIC_READ
;
598 static HRESULT WINAPI
ProtocolStream_Clone(IStream
*iface
, IStream
**ppstm
)
600 ProtocolStream
*This
= impl_from_IStream(iface
);
601 FIXME("(%p)->(%p)\n", This
, ppstm
);
605 static const IStreamVtbl ProtocolStreamVtbl
= {
606 ProtocolStream_QueryInterface
,
607 ProtocolStream_AddRef
,
608 ProtocolStream_Release
,
610 ProtocolStream_Write
,
612 ProtocolStream_SetSize
,
613 ProtocolStream_CopyTo
,
614 ProtocolStream_Commit
,
615 ProtocolStream_Revert
,
616 ProtocolStream_LockRegion
,
617 ProtocolStream_UnlockRegion
,
622 static void stgmed_stream_release(stgmed_obj_t
*obj
)
624 ProtocolStream
*stream
= (ProtocolStream
*)obj
;
625 IStream_Release(&stream
->IStream_iface
);
628 static HRESULT
stgmed_stream_fill_stgmed(stgmed_obj_t
*obj
, STGMEDIUM
*stgmed
)
630 ProtocolStream
*stream
= (ProtocolStream
*)obj
;
632 stgmed
->tymed
= TYMED_ISTREAM
;
633 stgmed
->u
.pstm
= &stream
->IStream_iface
;
634 stgmed
->pUnkForRelease
= &stream
->buf
->IUnknown_iface
;
639 static HRESULT
stgmed_stream_get_result(stgmed_obj_t
*obj
, DWORD bindf
, void **result
)
641 ProtocolStream
*stream
= (ProtocolStream
*)obj
;
643 if(!(bindf
& BINDF_ASYNCHRONOUS
) && stream
->buf
->file
== INVALID_HANDLE_VALUE
644 && stream
->buf
->hres
!= S_FALSE
)
645 return INET_E_DATA_NOT_AVAILABLE
;
647 IStream_AddRef(&stream
->IStream_iface
);
648 *result
= &stream
->IStream_iface
;
652 static const stgmed_obj_vtbl stgmed_stream_vtbl
= {
653 stgmed_stream_release
,
654 stgmed_stream_fill_stgmed
,
655 stgmed_stream_get_result
659 stgmed_obj_t stgmed_obj
;
663 static stgmed_obj_t
*create_stgmed_stream(stgmed_buf_t
*buf
)
665 ProtocolStream
*ret
= heap_alloc(sizeof(ProtocolStream
));
667 ret
->stgmed_obj
.vtbl
= &stgmed_stream_vtbl
;
668 ret
->IStream_iface
.lpVtbl
= &ProtocolStreamVtbl
;
671 IUnknown_AddRef(&buf
->IUnknown_iface
);
676 return &ret
->stgmed_obj
;
679 static void stgmed_file_release(stgmed_obj_t
*obj
)
681 stgmed_file_obj_t
*file_obj
= (stgmed_file_obj_t
*)obj
;
683 IUnknown_Release(&file_obj
->buf
->IUnknown_iface
);
687 static HRESULT
stgmed_file_fill_stgmed(stgmed_obj_t
*obj
, STGMEDIUM
*stgmed
)
689 stgmed_file_obj_t
*file_obj
= (stgmed_file_obj_t
*)obj
;
691 if(!file_obj
->buf
->cache_file
) {
692 WARN("cache_file not set\n");
693 return INET_E_DATA_NOT_AVAILABLE
;
696 read_protocol_data(file_obj
->buf
);
698 stgmed
->tymed
= TYMED_FILE
;
699 stgmed
->u
.lpszFileName
= file_obj
->buf
->cache_file
;
700 stgmed
->pUnkForRelease
= &file_obj
->buf
->IUnknown_iface
;
705 static HRESULT
stgmed_file_get_result(stgmed_obj_t
*obj
, DWORD bindf
, void **result
)
707 return bindf
& BINDF_ASYNCHRONOUS
? MK_S_ASYNCHRONOUS
: S_OK
;
710 static const stgmed_obj_vtbl stgmed_file_vtbl
= {
712 stgmed_file_fill_stgmed
,
713 stgmed_file_get_result
716 static stgmed_obj_t
*create_stgmed_file(stgmed_buf_t
*buf
)
718 stgmed_file_obj_t
*ret
= heap_alloc(sizeof(*ret
));
720 ret
->stgmed_obj
.vtbl
= &stgmed_file_vtbl
;
722 IUnknown_AddRef(&buf
->IUnknown_iface
);
725 return &ret
->stgmed_obj
;
728 static inline Binding
*impl_from_IBinding(IBinding
*iface
)
730 return CONTAINING_RECORD(iface
, Binding
, IBinding_iface
);
733 static HRESULT WINAPI
Binding_QueryInterface(IBinding
*iface
, REFIID riid
, void **ppv
)
735 Binding
*This
= impl_from_IBinding(iface
);
739 if(IsEqualGUID(&IID_IUnknown
, riid
)) {
740 TRACE("(%p)->(IID_IUnknown %p)\n", This
, ppv
);
741 *ppv
= &This
->IBinding_iface
;
742 }else if(IsEqualGUID(&IID_IBinding
, riid
)) {
743 TRACE("(%p)->(IID_IBinding %p)\n", This
, ppv
);
744 *ppv
= &This
->IBinding_iface
;
745 }else if(IsEqualGUID(&IID_IInternetProtocolSink
, riid
)) {
746 TRACE("(%p)->(IID_IInternetProtocolSink %p)\n", This
, ppv
);
747 *ppv
= &This
->IInternetProtocolSink_iface
;
748 }else if(IsEqualGUID(&IID_IInternetBindInfo
, riid
)) {
749 TRACE("(%p)->(IID_IInternetBindInfo %p)\n", This
, ppv
);
750 *ppv
= &This
->IInternetBindInfo_iface
;
751 }else if(IsEqualGUID(&IID_IServiceProvider
, riid
)) {
752 TRACE("(%p)->(IID_IServiceProvider %p)\n", This
, ppv
);
753 *ppv
= &This
->IServiceProvider_iface
;
754 }else if(IsEqualGUID(&IID_IWinInetInfo
, riid
)) {
755 TRACE("(%p)->(IID_IWinInetInfo %p)\n", This
, ppv
);
757 /* NOTE: This violidates COM rules, but tests prove that we should do it */
758 if(!This
->protocol
->wininet_info
)
759 return E_NOINTERFACE
;
761 *ppv
= &This
->IWinInetHttpInfo_iface
;
762 }else if(IsEqualGUID(&IID_IWinInetHttpInfo
, riid
)) {
763 TRACE("(%p)->(IID_IWinInetHttpInfo %p)\n", This
, ppv
);
765 if(!This
->protocol
->wininet_http_info
)
766 return E_NOINTERFACE
;
768 *ppv
= &This
->IWinInetHttpInfo_iface
;
772 IBinding_AddRef(&This
->IBinding_iface
);
776 WARN("Unsupported interface %s\n", debugstr_guid(riid
));
777 return E_NOINTERFACE
;
780 static ULONG WINAPI
Binding_AddRef(IBinding
*iface
)
782 Binding
*This
= impl_from_IBinding(iface
);
783 LONG ref
= InterlockedIncrement(&This
->ref
);
785 TRACE("(%p) ref=%d\n", This
, ref
);
790 static ULONG WINAPI
Binding_Release(IBinding
*iface
)
792 Binding
*This
= impl_from_IBinding(iface
);
793 LONG ref
= InterlockedDecrement(&This
->ref
);
795 TRACE("(%p) ref=%d\n", This
, ref
);
799 release_notif_hwnd(This
->notif_hwnd
);
801 IMoniker_Release(This
->mon
);
803 IBindStatusCallback_Release(This
->callback
);
805 IInternetProtocolEx_Release(&This
->protocol
->IInternetProtocolEx_iface
);
806 if(This
->service_provider
)
807 IServiceProvider_Release(This
->service_provider
);
809 IUnknown_Release(&This
->stgmed_buf
->IUnknown_iface
);
811 This
->stgmed_obj
->vtbl
->release(This
->stgmed_obj
);
813 IUnknown_Release(This
->obj
);
815 IBindCtx_Release(This
->bctx
);
817 ReleaseBindInfo(&This
->bindinfo
);
818 This
->section
.DebugInfo
->Spare
[0] = 0;
819 DeleteCriticalSection(&This
->section
);
820 SysFreeString(This
->url
);
821 heap_free(This
->mime
);
822 heap_free(This
->redirect_url
);
825 URLMON_UnlockModule();
831 static HRESULT WINAPI
Binding_Abort(IBinding
*iface
)
833 Binding
*This
= impl_from_IBinding(iface
);
836 TRACE("(%p)\n", This
);
838 if(This
->state
& BINDING_ABORTED
)
841 hres
= IInternetProtocolEx_Abort(&This
->protocol
->IInternetProtocolEx_iface
, E_ABORT
,
846 This
->state
|= BINDING_ABORTED
;
850 static HRESULT WINAPI
Binding_Suspend(IBinding
*iface
)
852 Binding
*This
= impl_from_IBinding(iface
);
853 FIXME("(%p)\n", This
);
857 static HRESULT WINAPI
Binding_Resume(IBinding
*iface
)
859 Binding
*This
= impl_from_IBinding(iface
);
860 FIXME("(%p)\n", This
);
864 static HRESULT WINAPI
Binding_SetPriority(IBinding
*iface
, LONG nPriority
)
866 Binding
*This
= impl_from_IBinding(iface
);
867 FIXME("(%p)->(%d)\n", This
, nPriority
);
871 static HRESULT WINAPI
Binding_GetPriority(IBinding
*iface
, LONG
*pnPriority
)
873 Binding
*This
= impl_from_IBinding(iface
);
874 FIXME("(%p)->(%p)\n", This
, pnPriority
);
878 static HRESULT WINAPI
Binding_GetBindResult(IBinding
*iface
, CLSID
*pclsidProtocol
,
879 DWORD
*pdwResult
, LPOLESTR
*pszResult
, DWORD
*pdwReserved
)
881 Binding
*This
= impl_from_IBinding(iface
);
883 TRACE("(%p)->(%p %p %p %p)\n", This
, pclsidProtocol
, pdwResult
, pszResult
, pdwReserved
);
885 if(!pdwResult
|| !pszResult
|| pdwReserved
)
888 if(!(This
->state
& BINDING_STOPPED
)) {
889 *pclsidProtocol
= CLSID_NULL
;
895 *pclsidProtocol
= This
->hres
==S_OK
? CLSID_NULL
: This
->clsid
;
896 *pdwResult
= This
->hres
;
901 static const IBindingVtbl BindingVtbl
= {
902 Binding_QueryInterface
,
910 Binding_GetBindResult
913 static Binding
*get_bctx_binding(IBindCtx
*bctx
)
919 hres
= IBindCtx_GetObjectParam(bctx
, cbinding_contextW
, &unk
);
923 hres
= IUnknown_QueryInterface(unk
, &IID_IBinding
, (void**)&binding
);
924 IUnknown_Release(unk
);
928 if (binding
->lpVtbl
!= &BindingVtbl
)
930 return impl_from_IBinding(binding
);
933 static inline Binding
*impl_from_IInternetProtocolSink(IInternetProtocolSink
*iface
)
935 return CONTAINING_RECORD(iface
, Binding
, IInternetProtocolSink_iface
);
938 static HRESULT WINAPI
InternetProtocolSink_QueryInterface(IInternetProtocolSink
*iface
,
939 REFIID riid
, void **ppv
)
941 Binding
*This
= impl_from_IInternetProtocolSink(iface
);
942 return IBinding_QueryInterface(&This
->IBinding_iface
, riid
, ppv
);
945 static ULONG WINAPI
InternetProtocolSink_AddRef(IInternetProtocolSink
*iface
)
947 Binding
*This
= impl_from_IInternetProtocolSink(iface
);
948 return IBinding_AddRef(&This
->IBinding_iface
);
951 static ULONG WINAPI
InternetProtocolSink_Release(IInternetProtocolSink
*iface
)
953 Binding
*This
= impl_from_IInternetProtocolSink(iface
);
954 return IBinding_Release(&This
->IBinding_iface
);
957 static HRESULT WINAPI
InternetProtocolSink_Switch(IInternetProtocolSink
*iface
,
958 PROTOCOLDATA
*pProtocolData
)
960 Binding
*This
= impl_from_IInternetProtocolSink(iface
);
962 WARN("(%p)->(%p)\n", This
, pProtocolData
);
967 static void on_progress(Binding
*This
, ULONG progress
, ULONG progress_max
,
968 ULONG status_code
, LPCWSTR status_text
)
970 IBindStatusCallback_OnProgress(This
->callback
, progress
, progress_max
,
971 status_code
, status_text
);
974 static HRESULT WINAPI
InternetProtocolSink_ReportProgress(IInternetProtocolSink
*iface
,
975 ULONG ulStatusCode
, LPCWSTR szStatusText
)
977 Binding
*This
= impl_from_IInternetProtocolSink(iface
);
979 TRACE("(%p)->(%s %s)\n", This
, debugstr_bindstatus(ulStatusCode
), debugstr_w(szStatusText
));
981 switch(ulStatusCode
) {
982 case BINDSTATUS_FINDINGRESOURCE
:
983 on_progress(This
, 0, 0, BINDSTATUS_FINDINGRESOURCE
, szStatusText
);
985 case BINDSTATUS_CONNECTING
:
986 on_progress(This
, 0, 0, BINDSTATUS_CONNECTING
, szStatusText
);
988 case BINDSTATUS_REDIRECTING
:
989 heap_free(This
->redirect_url
);
990 This
->redirect_url
= heap_strdupW(szStatusText
);
991 on_progress(This
, 0, 0, BINDSTATUS_REDIRECTING
, szStatusText
);
993 case BINDSTATUS_BEGINDOWNLOADDATA
:
995 case BINDSTATUS_SENDINGREQUEST
:
996 on_progress(This
, 0, 0, BINDSTATUS_SENDINGREQUEST
, szStatusText
);
998 case BINDSTATUS_PROTOCOLCLASSID
:
999 CLSIDFromString(szStatusText
, &This
->clsid
);
1001 case BINDSTATUS_MIMETYPEAVAILABLE
:
1002 case BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE
:
1003 mime_available(This
, szStatusText
);
1005 case BINDSTATUS_CACHEFILENAMEAVAILABLE
:
1006 cache_file_available(This
, szStatusText
);
1008 case BINDSTATUS_DECODING
:
1009 IBindStatusCallback_OnProgress(This
->callback
, 0, 0, BINDSTATUS_DECODING
, szStatusText
);
1011 case BINDSTATUS_LOADINGMIMEHANDLER
:
1012 on_progress(This
, 0, 0, BINDSTATUS_LOADINGMIMEHANDLER
, szStatusText
);
1014 case BINDSTATUS_DIRECTBIND
: /* FIXME: Handle BINDSTATUS_DIRECTBIND in BindProtocol */
1015 This
->report_mime
= FALSE
;
1017 case BINDSTATUS_ACCEPTRANGES
:
1020 FIXME("Unhandled status code %d\n", ulStatusCode
);
1027 static void report_data(Binding
*This
, DWORD bscf
, ULONG progress
, ULONG progress_max
)
1029 FORMATETC formatetc
= {0, NULL
, 1, -1, TYMED_ISTREAM
};
1030 BOOL sent_begindownloaddata
= FALSE
;
1032 TRACE("(%p)->(%d %u %u)\n", This
, bscf
, progress
, progress_max
);
1034 if(This
->download_state
== END_DOWNLOAD
|| (This
->state
& BINDING_ABORTED
)) {
1035 read_protocol_data(This
->stgmed_buf
);
1039 if(This
->state
& BINDING_STOPPED
)
1042 if(This
->stgmed_buf
->file
!= INVALID_HANDLE_VALUE
)
1043 read_protocol_data(This
->stgmed_buf
);
1045 if(This
->download_state
== BEFORE_DOWNLOAD
) {
1046 This
->download_state
= DOWNLOADING
;
1047 sent_begindownloaddata
= TRUE
;
1048 IBindStatusCallback_OnProgress(This
->callback
, progress
, progress_max
,
1049 BINDSTATUS_BEGINDOWNLOADDATA
, This
->url
);
1051 if(This
->stgmed_buf
->cache_file
)
1052 IBindStatusCallback_OnProgress(This
->callback
, progress
, progress_max
,
1053 BINDSTATUS_CACHEFILENAMEAVAILABLE
, This
->stgmed_buf
->cache_file
);
1056 if(This
->stgmed_buf
->hres
== S_FALSE
|| (bscf
& BSCF_LASTDATANOTIFICATION
)) {
1057 This
->download_state
= END_DOWNLOAD
;
1058 IBindStatusCallback_OnProgress(This
->callback
, progress
, progress_max
,
1059 BINDSTATUS_ENDDOWNLOADDATA
, This
->url
);
1060 }else if(!sent_begindownloaddata
) {
1061 IBindStatusCallback_OnProgress(This
->callback
, progress
, progress_max
,
1062 BINDSTATUS_DOWNLOADINGDATA
, This
->url
);
1065 if(This
->state
& (BINDING_STOPPED
|BINDING_ABORTED
))
1068 if(This
->to_object
) {
1069 if(!(This
->state
& BINDING_OBJAVAIL
)) {
1070 IBinding_AddRef(&This
->IBinding_iface
);
1071 create_object(This
);
1072 IBinding_Release(&This
->IBinding_iface
);
1078 if(!(This
->state
& BINDING_LOCKED
)) {
1079 HRESULT hres
= IInternetProtocolEx_LockRequest(
1080 &This
->protocol
->IInternetProtocolEx_iface
, 0);
1082 This
->state
|= BINDING_LOCKED
;
1085 hres
= This
->stgmed_obj
->vtbl
->fill_stgmed(This
->stgmed_obj
, &stgmed
);
1087 stop_binding(This
, hres
, NULL
);
1091 formatetc
.tymed
= stgmed
.tymed
;
1092 formatetc
.cfFormat
= This
->clipboard_format
;
1094 hres
= IBindStatusCallback_OnDataAvailable(This
->callback
, bscf
, progress
,
1095 &formatetc
, &stgmed
);
1097 if(This
->download_state
!= END_DOWNLOAD
) {
1098 This
->download_state
= END_DOWNLOAD
;
1099 IBindStatusCallback_OnProgress(This
->callback
, progress
, progress_max
,
1100 BINDSTATUS_ENDDOWNLOADDATA
, This
->url
);
1103 WARN("OnDataAvailable returned %x\n", hres
);
1104 stop_binding(This
, hres
, NULL
);
1108 if(This
->download_state
== END_DOWNLOAD
)
1109 stop_binding(This
, S_OK
, NULL
);
1113 static HRESULT WINAPI
InternetProtocolSink_ReportData(IInternetProtocolSink
*iface
,
1114 DWORD grfBSCF
, ULONG ulProgress
, ULONG ulProgressMax
)
1116 Binding
*This
= impl_from_IInternetProtocolSink(iface
);
1118 TRACE("(%p)->(%d %u %u)\n", This
, grfBSCF
, ulProgress
, ulProgressMax
);
1120 report_data(This
, grfBSCF
, ulProgress
, ulProgressMax
);
1124 static HRESULT WINAPI
InternetProtocolSink_ReportResult(IInternetProtocolSink
*iface
,
1125 HRESULT hrResult
, DWORD dwError
, LPCWSTR szResult
)
1127 Binding
*This
= impl_from_IInternetProtocolSink(iface
);
1129 TRACE("(%p)->(%08x %d %s)\n", This
, hrResult
, dwError
, debugstr_w(szResult
));
1131 stop_binding(This
, hrResult
, szResult
);
1133 IInternetProtocolEx_Terminate(&This
->protocol
->IInternetProtocolEx_iface
, 0);
1137 static const IInternetProtocolSinkVtbl InternetProtocolSinkVtbl
= {
1138 InternetProtocolSink_QueryInterface
,
1139 InternetProtocolSink_AddRef
,
1140 InternetProtocolSink_Release
,
1141 InternetProtocolSink_Switch
,
1142 InternetProtocolSink_ReportProgress
,
1143 InternetProtocolSink_ReportData
,
1144 InternetProtocolSink_ReportResult
1147 static inline Binding
*impl_from_IInternetBindInfo(IInternetBindInfo
*iface
)
1149 return CONTAINING_RECORD(iface
, Binding
, IInternetBindInfo_iface
);
1152 static HRESULT WINAPI
InternetBindInfo_QueryInterface(IInternetBindInfo
*iface
,
1153 REFIID riid
, void **ppv
)
1155 Binding
*This
= impl_from_IInternetBindInfo(iface
);
1156 return IBinding_QueryInterface(&This
->IBinding_iface
, riid
, ppv
);
1159 static ULONG WINAPI
InternetBindInfo_AddRef(IInternetBindInfo
*iface
)
1161 Binding
*This
= impl_from_IInternetBindInfo(iface
);
1162 return IBinding_AddRef(&This
->IBinding_iface
);
1165 static ULONG WINAPI
InternetBindInfo_Release(IInternetBindInfo
*iface
)
1167 Binding
*This
= impl_from_IInternetBindInfo(iface
);
1168 return IBinding_Release(&This
->IBinding_iface
);
1171 static HRESULT WINAPI
InternetBindInfo_GetBindInfo(IInternetBindInfo
*iface
,
1172 DWORD
*grfBINDF
, BINDINFO
*pbindinfo
)
1174 Binding
*This
= impl_from_IInternetBindInfo(iface
);
1176 TRACE("(%p)->(%p %p)\n", This
, grfBINDF
, pbindinfo
);
1178 *grfBINDF
= This
->bindf
;
1179 return CopyBindInfo(&This
->bindinfo
, pbindinfo
);
1182 static HRESULT WINAPI
InternetBindInfo_GetBindString(IInternetBindInfo
*iface
,
1183 ULONG ulStringType
, LPOLESTR
*ppwzStr
, ULONG cEl
, ULONG
*pcElFetched
)
1185 Binding
*This
= impl_from_IInternetBindInfo(iface
);
1187 TRACE("(%p)->(%d %p %d %p)\n", This
, ulStringType
, ppwzStr
, cEl
, pcElFetched
);
1189 switch(ulStringType
) {
1190 case BINDSTRING_ACCEPT_MIMES
: {
1191 static const WCHAR wszMimes
[] = {'*','/','*',0};
1193 if(!ppwzStr
|| !pcElFetched
)
1194 return E_INVALIDARG
;
1196 ppwzStr
[0] = CoTaskMemAlloc(sizeof(wszMimes
));
1197 memcpy(ppwzStr
[0], wszMimes
, sizeof(wszMimes
));
1201 case BINDSTRING_USER_AGENT
: {
1202 IInternetBindInfo
*bindinfo
= NULL
;
1205 hres
= IBindStatusCallback_QueryInterface(This
->callback
, &IID_IInternetBindInfo
,
1210 hres
= IInternetBindInfo_GetBindString(bindinfo
, ulStringType
, ppwzStr
,
1212 IInternetBindInfo_Release(bindinfo
);
1216 case BINDSTRING_URL
: {
1217 DWORD size
= (SysStringLen(This
->url
)+1) * sizeof(WCHAR
);
1219 if(!ppwzStr
|| !pcElFetched
)
1220 return E_INVALIDARG
;
1222 *ppwzStr
= CoTaskMemAlloc(size
);
1223 memcpy(*ppwzStr
, This
->url
, size
);
1229 FIXME("not supported string type %d\n", ulStringType
);
1233 static const IInternetBindInfoVtbl InternetBindInfoVtbl
= {
1234 InternetBindInfo_QueryInterface
,
1235 InternetBindInfo_AddRef
,
1236 InternetBindInfo_Release
,
1237 InternetBindInfo_GetBindInfo
,
1238 InternetBindInfo_GetBindString
1241 static inline Binding
*impl_from_IWinInetHttpInfo(IWinInetHttpInfo
*iface
)
1243 return CONTAINING_RECORD(iface
, Binding
, IWinInetHttpInfo_iface
);
1246 static HRESULT WINAPI
WinInetHttpInfo_QueryInterface(IWinInetHttpInfo
*iface
, REFIID riid
, void **ppv
)
1248 Binding
*This
= impl_from_IWinInetHttpInfo(iface
);
1249 return IBinding_QueryInterface(&This
->IBinding_iface
, riid
, ppv
);
1252 static ULONG WINAPI
WinInetHttpInfo_AddRef(IWinInetHttpInfo
*iface
)
1254 Binding
*This
= impl_from_IWinInetHttpInfo(iface
);
1255 return IBinding_AddRef(&This
->IBinding_iface
);
1258 static ULONG WINAPI
WinInetHttpInfo_Release(IWinInetHttpInfo
*iface
)
1260 Binding
*This
= impl_from_IWinInetHttpInfo(iface
);
1261 return IBinding_Release(&This
->IBinding_iface
);
1264 static HRESULT WINAPI
WinInetHttpInfo_QueryOption(IWinInetHttpInfo
*iface
, DWORD dwOption
,
1265 void *pBuffer
, DWORD
*pcbBuffer
)
1267 Binding
*This
= impl_from_IWinInetHttpInfo(iface
);
1268 TRACE("(%p)->(%x %p %p)\n", This
, dwOption
, pBuffer
, pcbBuffer
);
1270 if(!This
->protocol
->wininet_info
)
1273 return IWinInetInfo_QueryOption(This
->protocol
->wininet_info
,
1274 dwOption
, pBuffer
, pcbBuffer
);
1277 static HRESULT WINAPI
WinInetHttpInfo_QueryInfo(IWinInetHttpInfo
*iface
, DWORD dwOption
,
1278 void *pBuffer
, DWORD
*pcbBuffer
, DWORD
*pdwFlags
, DWORD
*pdwReserved
)
1280 Binding
*This
= impl_from_IWinInetHttpInfo(iface
);
1281 TRACE("(%p)->(%x %p %p %p %p)\n", This
, dwOption
, pBuffer
, pcbBuffer
, pdwFlags
, pdwReserved
);
1283 if(!This
->protocol
->wininet_http_info
)
1286 return IWinInetHttpInfo_QueryInfo(This
->protocol
->wininet_http_info
,
1287 dwOption
, pBuffer
, pcbBuffer
, pdwFlags
, pdwReserved
);
1290 static const IWinInetHttpInfoVtbl WinInetHttpInfoVtbl
= {
1291 WinInetHttpInfo_QueryInterface
,
1292 WinInetHttpInfo_AddRef
,
1293 WinInetHttpInfo_Release
,
1294 WinInetHttpInfo_QueryOption
,
1295 WinInetHttpInfo_QueryInfo
1298 static inline Binding
*impl_from_IServiceProvider(IServiceProvider
*iface
)
1300 return CONTAINING_RECORD(iface
, Binding
, IServiceProvider_iface
);
1303 static HRESULT WINAPI
ServiceProvider_QueryInterface(IServiceProvider
*iface
,
1304 REFIID riid
, void **ppv
)
1306 Binding
*This
= impl_from_IServiceProvider(iface
);
1307 return IBinding_QueryInterface(&This
->IBinding_iface
, riid
, ppv
);
1310 static ULONG WINAPI
ServiceProvider_AddRef(IServiceProvider
*iface
)
1312 Binding
*This
= impl_from_IServiceProvider(iface
);
1313 return IBinding_AddRef(&This
->IBinding_iface
);
1316 static ULONG WINAPI
ServiceProvider_Release(IServiceProvider
*iface
)
1318 Binding
*This
= impl_from_IServiceProvider(iface
);
1319 return IBinding_Release(&This
->IBinding_iface
);
1322 static HRESULT WINAPI
ServiceProvider_QueryService(IServiceProvider
*iface
,
1323 REFGUID guidService
, REFIID riid
, void **ppv
)
1325 Binding
*This
= impl_from_IServiceProvider(iface
);
1328 TRACE("(%p)->(%s %s %p)\n", This
, debugstr_guid(guidService
), debugstr_guid(riid
), ppv
);
1330 if(This
->service_provider
) {
1331 hres
= IServiceProvider_QueryService(This
->service_provider
, guidService
,
1337 WARN("unknown service %s\n", debugstr_guid(guidService
));
1338 return E_NOINTERFACE
;
1341 static const IServiceProviderVtbl ServiceProviderVtbl
= {
1342 ServiceProvider_QueryInterface
,
1343 ServiceProvider_AddRef
,
1344 ServiceProvider_Release
,
1345 ServiceProvider_QueryService
1348 static HRESULT
get_callback(IBindCtx
*pbc
, IBindStatusCallback
**callback
)
1353 hres
= IBindCtx_GetObjectParam(pbc
, bscb_holderW
, &unk
);
1355 return create_default_callback(callback
);
1357 hres
= IUnknown_QueryInterface(unk
, &IID_IBindStatusCallback
, (void**)callback
);
1358 IUnknown_Release(unk
);
1362 static BOOL
is_urlmon_protocol(IUri
*uri
)
1367 hres
= IUri_GetScheme(uri
, &scheme
);
1372 case URL_SCHEME_FILE
:
1373 case URL_SCHEME_FTP
:
1374 case URL_SCHEME_GOPHER
:
1375 case URL_SCHEME_HTTP
:
1376 case URL_SCHEME_HTTPS
:
1384 static HRESULT
Binding_Create(IMoniker
*mon
, Binding
*binding_ctx
, IUri
*uri
, IBindCtx
*pbc
,
1385 BOOL to_obj
, REFIID riid
, Binding
**binding
)
1390 URLMON_LockModule();
1392 ret
= heap_alloc_zero(sizeof(Binding
));
1394 ret
->IBinding_iface
.lpVtbl
= &BindingVtbl
;
1395 ret
->IInternetProtocolSink_iface
.lpVtbl
= &InternetProtocolSinkVtbl
;
1396 ret
->IInternetBindInfo_iface
.lpVtbl
= &InternetBindInfoVtbl
;
1397 ret
->IWinInetHttpInfo_iface
.lpVtbl
= &WinInetHttpInfoVtbl
;
1398 ret
->IServiceProvider_iface
.lpVtbl
= &ServiceProviderVtbl
;
1402 ret
->to_object
= to_obj
;
1404 ret
->notif_hwnd
= get_notif_hwnd();
1405 ret
->report_mime
= !binding_ctx
;
1406 ret
->download_state
= BEFORE_DOWNLOAD
;
1409 IBindCtx_AddRef(pbc
);
1414 IMoniker_AddRef(mon
);
1418 ret
->bindinfo
.cbSize
= sizeof(BINDINFO
);
1420 InitializeCriticalSection(&ret
->section
);
1421 ret
->section
.DebugInfo
->Spare
[0] = (DWORD_PTR
)(__FILE__
": Binding.section");
1423 hres
= get_callback(pbc
, &ret
->callback
);
1425 WARN("Could not get IBindStatusCallback\n");
1426 IBinding_Release(&ret
->IBinding_iface
);
1430 IBindStatusCallback_QueryInterface(ret
->callback
, &IID_IServiceProvider
,
1431 (void**)&ret
->service_provider
);
1434 ret
->protocol
= binding_ctx
->protocol
;
1435 IInternetProtocolEx_AddRef(&ret
->protocol
->IInternetProtocolEx_iface
);
1437 hres
= create_binding_protocol(TRUE
, &ret
->protocol
);
1439 WARN("Could not get protocol handler\n");
1440 IBinding_Release(&ret
->IBinding_iface
);
1445 hres
= IBindStatusCallback_GetBindInfo(ret
->callback
, &ret
->bindf
, &ret
->bindinfo
);
1447 WARN("GetBindInfo failed: %08x\n", hres
);
1448 IBinding_Release(&ret
->IBinding_iface
);
1452 TRACE("bindf %08x\n", ret
->bindf
);
1453 dump_BINDINFO(&ret
->bindinfo
);
1455 ret
->bindf
|= BINDF_FROMURLMON
;
1457 ret
->bindinfo
.dwOptions
|= 0x100000;
1459 if(!(ret
->bindf
& BINDF_ASYNCHRONOUS
) || !(ret
->bindf
& BINDF_PULLDATA
)) {
1460 ret
->bindf
|= BINDF_NEEDFILE
;
1461 ret
->use_cache_file
= TRUE
;
1462 }else if(!is_urlmon_protocol(uri
)) {
1463 ret
->bindf
|= BINDF_NEEDFILE
;
1466 hres
= IUri_GetDisplayUri(uri
, &ret
->url
);
1468 IBinding_Release(&ret
->IBinding_iface
);
1473 ret
->stgmed_buf
= binding_ctx
->stgmed_buf
;
1474 IUnknown_AddRef(&ret
->stgmed_buf
->IUnknown_iface
);
1475 ret
->clipboard_format
= binding_ctx
->clipboard_format
;
1477 ret
->stgmed_buf
= create_stgmed_buf(&ret
->protocol
->IInternetProtocolEx_iface
);
1481 ret
->stgmed_obj
= NULL
;
1482 }else if(IsEqualGUID(&IID_IStream
, riid
)) {
1483 ret
->stgmed_obj
= create_stgmed_stream(ret
->stgmed_buf
);
1484 }else if(IsEqualGUID(&IID_IUnknown
, riid
)) {
1485 ret
->bindf
|= BINDF_NEEDFILE
;
1486 ret
->stgmed_obj
= create_stgmed_file(ret
->stgmed_buf
);
1488 FIXME("Unsupported riid %s\n", debugstr_guid(riid
));
1489 IBinding_Release(&ret
->IBinding_iface
);
1497 static HRESULT
start_binding(IMoniker
*mon
, Binding
*binding_ctx
, IUri
*uri
, IBindCtx
*pbc
,
1498 BOOL to_obj
, REFIID riid
, Binding
**ret
)
1500 Binding
*binding
= NULL
;
1504 hres
= Binding_Create(mon
, binding_ctx
, uri
, pbc
, to_obj
, riid
, &binding
);
1508 hres
= IBindStatusCallback_OnStartBinding(binding
->callback
, 0, &binding
->IBinding_iface
);
1510 WARN("OnStartBinding failed: %08x\n", hres
);
1511 if(hres
!= E_ABORT
&& hres
!= E_NOTIMPL
)
1512 hres
= INET_E_DOWNLOAD_FAILURE
;
1514 stop_binding(binding
, hres
, NULL
);
1515 IBinding_Release(&binding
->IBinding_iface
);
1520 set_binding_sink(binding
->protocol
, &binding
->IInternetProtocolSink_iface
,
1521 &binding
->IInternetBindInfo_iface
);
1522 if(binding_ctx
->redirect_url
)
1523 IBindStatusCallback_OnProgress(binding
->callback
, 0, 0, BINDSTATUS_REDIRECTING
, binding_ctx
->redirect_url
);
1524 report_data(binding
, BSCF_FIRSTDATANOTIFICATION
| (binding_ctx
->download_state
== END_DOWNLOAD
? BSCF_LASTDATANOTIFICATION
: 0),
1527 hres
= IInternetProtocolEx_StartEx(&binding
->protocol
->IInternetProtocolEx_iface
, uri
,
1528 &binding
->IInternetProtocolSink_iface
, &binding
->IInternetBindInfo_iface
,
1529 PI_APARTMENTTHREADED
|PI_MIMEVERIFICATION
, 0);
1531 TRACE("start ret %08x\n", hres
);
1533 if(FAILED(hres
) && hres
!= E_PENDING
) {
1534 stop_binding(binding
, hres
, NULL
);
1535 IBinding_Release(&binding
->IBinding_iface
);
1541 while(!(binding
->bindf
& BINDF_ASYNCHRONOUS
) &&
1542 !(binding
->state
& BINDING_STOPPED
)) {
1543 MsgWaitForMultipleObjects(0, NULL
, FALSE
, 5000, QS_POSTMESSAGE
);
1544 while (PeekMessageW(&msg
, binding
->notif_hwnd
, WM_USER
, WM_USER
+117, PM_REMOVE
|PM_NOYIELD
)) {
1545 TranslateMessage(&msg
);
1546 DispatchMessageW(&msg
);
1554 HRESULT
bind_to_storage(IUri
*uri
, IBindCtx
*pbc
, REFIID riid
, void **ppv
)
1556 Binding
*binding
= NULL
, *binding_ctx
;
1559 binding_ctx
= get_bctx_binding(pbc
);
1561 hres
= start_binding(NULL
, binding_ctx
, uri
, pbc
, FALSE
, riid
, &binding
);
1563 IBinding_Release(&binding_ctx
->IBinding_iface
);
1567 if(binding
->hres
== S_OK
&& binding
->download_state
!= BEFORE_DOWNLOAD
/* FIXME */) {
1568 if((binding
->state
& BINDING_STOPPED
) && (binding
->state
& BINDING_LOCKED
))
1569 IInternetProtocolEx_UnlockRequest(&binding
->protocol
->IInternetProtocolEx_iface
);
1571 hres
= binding
->stgmed_obj
->vtbl
->get_result(binding
->stgmed_obj
, binding
->bindf
, ppv
);
1572 }else if(binding
->bindf
& BINDF_ASYNCHRONOUS
) {
1573 hres
= MK_S_ASYNCHRONOUS
;
1575 hres
= FAILED(binding
->hres
) ? binding
->hres
: S_OK
;
1578 IBinding_Release(&binding
->IBinding_iface
);
1583 HRESULT
bind_to_object(IMoniker
*mon
, IUri
*uri
, IBindCtx
*pbc
, REFIID riid
, void **ppv
)
1590 hres
= start_binding(mon
, NULL
, uri
, pbc
, TRUE
, riid
, &binding
);
1594 if(binding
->hres
!= S_OK
) {
1595 hres
= SUCCEEDED(binding
->hres
) ? S_OK
: binding
->hres
;
1596 }else if(binding
->bindf
& BINDF_ASYNCHRONOUS
) {
1597 hres
= MK_S_ASYNCHRONOUS
;
1599 *ppv
= binding
->obj
;
1600 IUnknown_AddRef(binding
->obj
);
1604 IBinding_Release(&binding
->IBinding_iface
);