[URLMON] Sync with Wine Staging 4.18. CORE-16441
[reactos.git] / dll / win32 / urlmon / binding.c
1 /*
2 * Copyright 2005-2007 Jacek Caban for CodeWeavers
3 *
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17 */
18
19 #define NONAMELESSUNION
20
21 #include "urlmon_main.h"
22 #include "winreg.h"
23 #include "shlwapi.h"
24
25 #include "wine/debug.h"
26
27 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
28
29 static WCHAR cbinding_contextW[] = {'C','B','i','n','d','i','n','g',' ','C','o','n','t','e','x','t',0};
30 static WCHAR bscb_holderW[] = { '_','B','S','C','B','_','H','o','l','d','e','r','_',0 };
31
32 typedef struct {
33 IUnknown IUnknown_iface;
34
35 LONG ref;
36
37 IInternetProtocolEx *protocol;
38
39 HANDLE file;
40 HRESULT hres;
41
42 LPWSTR cache_file;
43 } stgmed_buf_t;
44
45 typedef struct _stgmed_obj_t stgmed_obj_t;
46
47 typedef struct {
48 void (*release)(stgmed_obj_t*);
49 HRESULT (*fill_stgmed)(stgmed_obj_t*,STGMEDIUM*);
50 HRESULT (*get_result)(stgmed_obj_t*,DWORD,void**);
51 } stgmed_obj_vtbl;
52
53 struct _stgmed_obj_t {
54 const stgmed_obj_vtbl *vtbl;
55 };
56
57 typedef enum {
58 BEFORE_DOWNLOAD,
59 DOWNLOADING,
60 END_DOWNLOAD
61 } download_state_t;
62
63 #define BINDING_LOCKED 0x0001
64 #define BINDING_STOPPED 0x0002
65 #define BINDING_OBJAVAIL 0x0004
66 #define BINDING_ABORTED 0x0008
67
68 typedef struct {
69 IBinding IBinding_iface;
70 IInternetProtocolSink IInternetProtocolSink_iface;
71 IInternetBindInfo IInternetBindInfo_iface;
72 IWinInetHttpInfo IWinInetHttpInfo_iface;
73 IServiceProvider IServiceProvider_iface;
74
75 LONG ref;
76
77 IBindStatusCallback *callback;
78 IServiceProvider *service_provider;
79
80 BindProtocol *protocol;
81
82 stgmed_buf_t *stgmed_buf;
83 stgmed_obj_t *stgmed_obj;
84
85 BINDINFO bindinfo;
86 DWORD bindf;
87 BOOL to_object;
88 LPWSTR mime;
89 UINT clipboard_format;
90 LPWSTR url;
91 LPWSTR redirect_url;
92 IID iid;
93 BOOL report_mime;
94 BOOL use_cache_file;
95 DWORD state;
96 HRESULT hres;
97 CLSID clsid;
98 download_state_t download_state;
99 IUnknown *obj;
100 IMoniker *mon;
101 IBindCtx *bctx;
102 HWND notif_hwnd;
103
104 CRITICAL_SECTION section;
105 } Binding;
106
107 static void read_protocol_data(stgmed_buf_t *stgmed_buf)
108 {
109 BYTE buf[8192];
110 DWORD read;
111 HRESULT hres;
112
113 do hres = IInternetProtocolEx_Read(stgmed_buf->protocol, buf, sizeof(buf), &read);
114 while(hres == S_OK);
115 }
116
117 static void dump_BINDINFO(BINDINFO *bi)
118 {
119 static const char * const BINDINFOF_str[] = {
120 "#0",
121 "BINDINFOF_URLENCODESTGMEDDATA",
122 "BINDINFOF_URLENCODEDEXTRAINFO"
123 };
124
125 static const char * const BINDVERB_str[] = {
126 "BINDVERB_GET",
127 "BINDVERB_POST",
128 "BINDVERB_PUT",
129 "BINDVERB_CUSTOM"
130 };
131
132 TRACE("\n"
133 "BINDINFO = {\n"
134 " %d, %s,\n"
135 " {%d, %p, %p},\n"
136 " %s,\n"
137 " %s,\n"
138 " %s,\n"
139 " %d, %08x, %d, %d\n"
140 " {%d %p %x},\n"
141 " %s\n"
142 " %p, %d\n"
143 "}\n",
144
145 bi->cbSize, debugstr_w(bi->szExtraInfo),
146 bi->stgmedData.tymed, bi->stgmedData.u.hGlobal, bi->stgmedData.pUnkForRelease,
147 bi->grfBindInfoF > BINDINFOF_URLENCODEDEXTRAINFO
148 ? "unknown" : BINDINFOF_str[bi->grfBindInfoF],
149 bi->dwBindVerb > BINDVERB_CUSTOM
150 ? "unknown" : BINDVERB_str[bi->dwBindVerb],
151 debugstr_w(bi->szCustomVerb),
152 bi->cbstgmedData, bi->dwOptions, bi->dwOptionsFlags, bi->dwCodePage,
153 bi->securityAttributes.nLength,
154 bi->securityAttributes.lpSecurityDescriptor,
155 bi->securityAttributes.bInheritHandle,
156 debugstr_guid(&bi->iid),
157 bi->pUnk, bi->dwReserved
158 );
159 }
160
161 static void mime_available(Binding *This, LPCWSTR mime)
162 {
163 heap_free(This->mime);
164 This->mime = heap_strdupW(mime);
165
166 if(!This->mime || !This->report_mime)
167 return;
168
169 IBindStatusCallback_OnProgress(This->callback, 0, 0, BINDSTATUS_MIMETYPEAVAILABLE, This->mime);
170
171 This->clipboard_format = RegisterClipboardFormatW(This->mime);
172 }
173
174 static void stop_binding(Binding *binding, HRESULT hres, LPCWSTR str)
175 {
176 if(binding->state & BINDING_LOCKED) {
177 IInternetProtocolEx_UnlockRequest(&binding->protocol->IInternetProtocolEx_iface);
178 binding->state &= ~BINDING_LOCKED;
179 }
180
181 if(!(binding->state & BINDING_STOPPED)) {
182 binding->state |= BINDING_STOPPED;
183
184 binding->hres = hres;
185 IBindStatusCallback_OnStopBinding(binding->callback, hres, str);
186 }
187 }
188
189 static LPWSTR get_mime_clsid(LPCWSTR mime, CLSID *clsid)
190 {
191 LPWSTR key_name, ret;
192 DWORD res, type, size;
193 HKEY hkey;
194 int len;
195 HRESULT hres;
196
197 static const WCHAR mime_keyW[] =
198 {'M','I','M','E','\\','D','a','t','a','b','a','s','e','\\',
199 'C','o','n','t','e','n','t',' ','T','y','p','e','\\'};
200 static const WCHAR clsidW[] = {'C','L','S','I','D',0};
201
202 len = lstrlenW(mime)+1;
203 key_name = heap_alloc(sizeof(mime_keyW) + len*sizeof(WCHAR));
204 memcpy(key_name, mime_keyW, sizeof(mime_keyW));
205 lstrcpyW(key_name + ARRAY_SIZE(mime_keyW), mime);
206
207 res = RegOpenKeyW(HKEY_CLASSES_ROOT, key_name, &hkey);
208 heap_free(key_name);
209 if(res != ERROR_SUCCESS) {
210 WARN("Could not open MIME key: %x\n", res);
211 return NULL;
212 }
213
214 size = 50*sizeof(WCHAR);
215 ret = heap_alloc(size);
216 res = RegQueryValueExW(hkey, clsidW, NULL, &type, (LPBYTE)ret, &size);
217 RegCloseKey(hkey);
218 if(res != ERROR_SUCCESS) {
219 WARN("Could not get CLSID: %08x\n", res);
220 heap_free(ret);
221 return NULL;
222 }
223
224 hres = CLSIDFromString(ret, clsid);
225 if(FAILED(hres)) {
226 WARN("Could not parse CLSID: %08x\n", hres);
227 heap_free(ret);
228 return NULL;
229 }
230
231 return ret;
232 }
233
234 static void load_doc_mon(Binding *binding, IPersistMoniker *persist)
235 {
236 IBindCtx *bctx;
237 HRESULT hres;
238
239 hres = CreateAsyncBindCtxEx(binding->bctx, 0, NULL, NULL, &bctx, 0);
240 if(FAILED(hres)) {
241 WARN("CreateAsyncBindCtxEx failed: %08x\n", hres);
242 return;
243 }
244
245 IBindCtx_RevokeObjectParam(bctx, bscb_holderW);
246 IBindCtx_RegisterObjectParam(bctx, cbinding_contextW, (IUnknown*)&binding->IBinding_iface);
247
248 hres = IPersistMoniker_Load(persist, binding->download_state == END_DOWNLOAD, binding->mon, bctx, 0x12);
249 IBindCtx_RevokeObjectParam(bctx, cbinding_contextW);
250 IBindCtx_Release(bctx);
251 if(FAILED(hres))
252 FIXME("Load failed: %08x\n", hres);
253 }
254
255 static HRESULT create_mime_object(Binding *binding, const CLSID *clsid, LPCWSTR clsid_str)
256 {
257 IPersistMoniker *persist;
258 HRESULT hres;
259
260 hres = CoCreateInstance(clsid, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
261 &binding->iid, (void**)&binding->obj);
262 if(FAILED(hres)) {
263 WARN("CoCreateInstance failed: %08x\n", hres);
264 return INET_E_CANNOT_INSTANTIATE_OBJECT;
265 }
266
267 binding->state |= BINDING_OBJAVAIL;
268
269 hres = IUnknown_QueryInterface(binding->obj, &IID_IPersistMoniker, (void**)&persist);
270 if(SUCCEEDED(hres)) {
271 IMonikerProp *prop;
272
273 hres = IPersistMoniker_QueryInterface(persist, &IID_IMonikerProp, (void**)&prop);
274 if(SUCCEEDED(hres)) {
275 IMonikerProp_PutProperty(prop, MIMETYPEPROP, binding->mime);
276 IMonikerProp_PutProperty(prop, CLASSIDPROP, clsid_str);
277 IMonikerProp_Release(prop);
278 }
279
280 load_doc_mon(binding, persist);
281
282 IPersistMoniker_Release(persist);
283 }else {
284 FIXME("Could not get IPersistMoniker: %08x\n", hres);
285 /* FIXME: Try query IPersistFile */
286 }
287
288 IBindStatusCallback_OnObjectAvailable(binding->callback, &binding->iid, binding->obj);
289
290 return S_OK;
291 }
292
293 static void create_object(Binding *binding)
294 {
295 LPWSTR clsid_str;
296 CLSID clsid;
297 HRESULT hres;
298
299 if(!binding->mime) {
300 FIXME("MIME not available\n");
301 return;
302 }
303
304 if((clsid_str = get_mime_clsid(binding->mime, &clsid)))
305 IBindStatusCallback_OnProgress(binding->callback, 0, 0, BINDSTATUS_CLASSIDAVAILABLE, clsid_str);
306
307 IBindStatusCallback_OnProgress(binding->callback, 0, 0, BINDSTATUS_BEGINSYNCOPERATION, NULL);
308
309 if(clsid_str) {
310 hres = create_mime_object(binding, &clsid, clsid_str);
311 heap_free(clsid_str);
312 }else {
313 FIXME("Could not find object for MIME %s\n", debugstr_w(binding->mime));
314 hres = REGDB_E_CLASSNOTREG;
315 }
316
317 IBindStatusCallback_OnProgress(binding->callback, 0, 0, BINDSTATUS_ENDSYNCOPERATION, NULL);
318 binding->clsid = CLSID_NULL;
319
320 stop_binding(binding, hres, NULL);
321 if(FAILED(hres))
322 IInternetProtocolEx_Terminate(&binding->protocol->IInternetProtocolEx_iface, 0);
323 }
324
325 static void cache_file_available(Binding *This, const WCHAR *file_name)
326 {
327 heap_free(This->stgmed_buf->cache_file);
328 This->stgmed_buf->cache_file = heap_strdupW(file_name);
329
330 if(This->use_cache_file) {
331 This->stgmed_buf->file = CreateFileW(file_name, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL,
332 OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
333 if(This->stgmed_buf->file == INVALID_HANDLE_VALUE)
334 WARN("CreateFile failed: %u\n", GetLastError());
335 }
336 }
337
338 static inline stgmed_buf_t *impl_from_IUnknown(IUnknown *iface)
339 {
340 return CONTAINING_RECORD(iface, stgmed_buf_t, IUnknown_iface);
341 }
342
343 static HRESULT WINAPI StgMedUnk_QueryInterface(IUnknown *iface, REFIID riid, void **ppv)
344 {
345 stgmed_buf_t *This = impl_from_IUnknown(iface);
346
347 *ppv = NULL;
348
349 if(IsEqualGUID(riid, &IID_IUnknown)) {
350 TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
351
352 *ppv = &This->IUnknown_iface;
353 IUnknown_AddRef(&This->IUnknown_iface);
354 return S_OK;
355 }
356
357 TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
358 return E_NOINTERFACE;
359 }
360
361 static ULONG WINAPI StgMedUnk_AddRef(IUnknown *iface)
362 {
363 stgmed_buf_t *This = impl_from_IUnknown(iface);
364 LONG ref = InterlockedIncrement(&This->ref);
365
366 TRACE("(%p) ref=%d\n", This, ref);
367
368 return ref;
369 }
370
371 static ULONG WINAPI StgMedUnk_Release(IUnknown *iface)
372 {
373 stgmed_buf_t *This = impl_from_IUnknown(iface);
374 LONG ref = InterlockedDecrement(&This->ref);
375
376 TRACE("(%p) ref=%d\n", This, ref);
377
378 if(!ref) {
379 if(This->file != INVALID_HANDLE_VALUE)
380 CloseHandle(This->file);
381 IInternetProtocolEx_Release(This->protocol);
382 heap_free(This->cache_file);
383 heap_free(This);
384
385 URLMON_UnlockModule();
386 }
387
388 return ref;
389 }
390
391 static const IUnknownVtbl StgMedUnkVtbl = {
392 StgMedUnk_QueryInterface,
393 StgMedUnk_AddRef,
394 StgMedUnk_Release
395 };
396
397 static stgmed_buf_t *create_stgmed_buf(IInternetProtocolEx *protocol)
398 {
399 stgmed_buf_t *ret = heap_alloc(sizeof(*ret));
400
401 ret->IUnknown_iface.lpVtbl = &StgMedUnkVtbl;
402 ret->ref = 1;
403 ret->file = INVALID_HANDLE_VALUE;
404 ret->hres = S_OK;
405 ret->cache_file = NULL;
406
407 IInternetProtocolEx_AddRef(protocol);
408 ret->protocol = protocol;
409
410 URLMON_LockModule();
411
412 return ret;
413 }
414
415 typedef struct {
416 stgmed_obj_t stgmed_obj;
417 IStream IStream_iface;
418
419 LONG ref;
420
421 stgmed_buf_t *buf;
422 } ProtocolStream;
423
424 static inline ProtocolStream *impl_from_IStream(IStream *iface)
425 {
426 return CONTAINING_RECORD(iface, ProtocolStream, IStream_iface);
427 }
428
429 static HRESULT WINAPI ProtocolStream_QueryInterface(IStream *iface,
430 REFIID riid, void **ppv)
431 {
432 ProtocolStream *This = impl_from_IStream(iface);
433
434 *ppv = NULL;
435
436 if(IsEqualGUID(&IID_IUnknown, riid)) {
437 TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
438 *ppv = &This->IStream_iface;
439 }else if(IsEqualGUID(&IID_ISequentialStream, riid)) {
440 TRACE("(%p)->(IID_ISequentialStream %p)\n", This, ppv);
441 *ppv = &This->IStream_iface;
442 }else if(IsEqualGUID(&IID_IStream, riid)) {
443 TRACE("(%p)->(IID_IStream %p)\n", This, ppv);
444 *ppv = &This->IStream_iface;
445 }
446
447 if(*ppv) {
448 IStream_AddRef(&This->IStream_iface);
449 return S_OK;
450 }
451
452 WARN("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
453 return E_NOINTERFACE;
454 }
455
456 static ULONG WINAPI ProtocolStream_AddRef(IStream *iface)
457 {
458 ProtocolStream *This = impl_from_IStream(iface);
459 LONG ref = InterlockedIncrement(&This->ref);
460
461 TRACE("(%p) ref=%d\n", This, ref);
462
463 return ref;
464 }
465
466 static ULONG WINAPI ProtocolStream_Release(IStream *iface)
467 {
468 ProtocolStream *This = impl_from_IStream(iface);
469 LONG ref = InterlockedDecrement(&This->ref);
470
471 TRACE("(%p) ref=%d\n", This, ref);
472
473 if(!ref) {
474 IUnknown_Release(&This->buf->IUnknown_iface);
475 heap_free(This);
476
477 URLMON_UnlockModule();
478 }
479
480 return ref;
481 }
482
483 static HRESULT WINAPI ProtocolStream_Read(IStream *iface, void *pv,
484 ULONG cb, ULONG *pcbRead)
485 {
486 ProtocolStream *This = impl_from_IStream(iface);
487 DWORD read = 0;
488 HRESULT hres;
489
490 TRACE("(%p)->(%p %d %p)\n", This, pv, cb, pcbRead);
491
492 if(This->buf->file == INVALID_HANDLE_VALUE) {
493 hres = This->buf->hres = IInternetProtocolEx_Read(This->buf->protocol, (PBYTE)pv, cb, &read);
494 }else {
495 hres = ReadFile(This->buf->file, pv, cb, &read, NULL) ? S_OK : INET_E_DOWNLOAD_FAILURE;
496 }
497
498 if (pcbRead)
499 *pcbRead = read;
500
501 if(hres == E_PENDING)
502 return E_PENDING;
503 else if(FAILED(hres))
504 FIXME("Read failed: %08x\n", hres);
505
506 return read ? S_OK : S_FALSE;
507 }
508
509 static HRESULT WINAPI ProtocolStream_Write(IStream *iface, const void *pv,
510 ULONG cb, ULONG *pcbWritten)
511 {
512 ProtocolStream *This = impl_from_IStream(iface);
513
514 TRACE("(%p)->(%p %d %p)\n", This, pv, cb, pcbWritten);
515
516 return STG_E_ACCESSDENIED;
517 }
518
519 static HRESULT WINAPI ProtocolStream_Seek(IStream *iface, LARGE_INTEGER dlibMove,
520 DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
521 {
522 ProtocolStream *This = impl_from_IStream(iface);
523 LARGE_INTEGER new_pos;
524 DWORD method;
525
526 TRACE("(%p)->(%d %08x %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
527
528 if(This->buf->file == INVALID_HANDLE_VALUE) {
529 /* We should probably call protocol handler's Seek. */
530 FIXME("no cache file, not supported\n");
531 return E_FAIL;
532 }
533
534 switch(dwOrigin) {
535 case STREAM_SEEK_SET:
536 method = FILE_BEGIN;
537 break;
538 case STREAM_SEEK_CUR:
539 method = FILE_CURRENT;
540 break;
541 case STREAM_SEEK_END:
542 method = FILE_END;
543 break;
544 default:
545 WARN("Invalid origin %x\n", dwOrigin);
546 return E_FAIL;
547 }
548
549 if(!SetFilePointerEx(This->buf->file, dlibMove, &new_pos, method)) {
550 FIXME("SetFilePointerEx failed: %u\n", GetLastError());
551 return E_FAIL;
552 }
553
554 if(plibNewPosition)
555 plibNewPosition->QuadPart = new_pos.QuadPart;
556 return S_OK;
557 }
558
559 static HRESULT WINAPI ProtocolStream_SetSize(IStream *iface, ULARGE_INTEGER libNewSize)
560 {
561 ProtocolStream *This = impl_from_IStream(iface);
562 FIXME("(%p)->(%d)\n", This, libNewSize.u.LowPart);
563 return E_NOTIMPL;
564 }
565
566 static HRESULT WINAPI ProtocolStream_CopyTo(IStream *iface, IStream *pstm,
567 ULARGE_INTEGER cb, ULARGE_INTEGER *pcbRead, ULARGE_INTEGER *pcbWritten)
568 {
569 ProtocolStream *This = impl_from_IStream(iface);
570 FIXME("(%p)->(%p %d %p %p)\n", This, pstm, cb.u.LowPart, pcbRead, pcbWritten);
571 return E_NOTIMPL;
572 }
573
574 static HRESULT WINAPI ProtocolStream_Commit(IStream *iface, DWORD grfCommitFlags)
575 {
576 ProtocolStream *This = impl_from_IStream(iface);
577
578 TRACE("(%p)->(%08x)\n", This, grfCommitFlags);
579
580 return E_NOTIMPL;
581 }
582
583 static HRESULT WINAPI ProtocolStream_Revert(IStream *iface)
584 {
585 ProtocolStream *This = impl_from_IStream(iface);
586
587 TRACE("(%p)\n", This);
588
589 return E_NOTIMPL;
590 }
591
592 static HRESULT WINAPI ProtocolStream_LockRegion(IStream *iface, ULARGE_INTEGER libOffset,
593 ULARGE_INTEGER cb, DWORD dwLockType)
594 {
595 ProtocolStream *This = impl_from_IStream(iface);
596 FIXME("(%p)->(%d %d %d)\n", This, libOffset.u.LowPart, cb.u.LowPart, dwLockType);
597 return E_NOTIMPL;
598 }
599
600 static HRESULT WINAPI ProtocolStream_UnlockRegion(IStream *iface,
601 ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
602 {
603 ProtocolStream *This = impl_from_IStream(iface);
604 FIXME("(%p)->(%d %d %d)\n", This, libOffset.u.LowPart, cb.u.LowPart, dwLockType);
605 return E_NOTIMPL;
606 }
607
608 static HRESULT WINAPI ProtocolStream_Stat(IStream *iface, STATSTG *pstatstg,
609 DWORD dwStatFlag)
610 {
611 ProtocolStream *This = impl_from_IStream(iface);
612 TRACE("(%p)->(%p %08x)\n", This, pstatstg, dwStatFlag);
613
614 if(!pstatstg)
615 return E_FAIL;
616
617 memset(pstatstg, 0, sizeof(STATSTG));
618
619 if(!(dwStatFlag&STATFLAG_NONAME) && This->buf->cache_file) {
620 pstatstg->pwcsName = CoTaskMemAlloc((lstrlenW(This->buf->cache_file)+1)*sizeof(WCHAR));
621 if(!pstatstg->pwcsName)
622 return STG_E_INSUFFICIENTMEMORY;
623
624 lstrcpyW(pstatstg->pwcsName, This->buf->cache_file);
625 }
626
627 pstatstg->type = STGTY_STREAM;
628 if(This->buf->file != INVALID_HANDLE_VALUE) {
629 GetFileSizeEx(This->buf->file, (PLARGE_INTEGER)&pstatstg->cbSize);
630 GetFileTime(This->buf->file, &pstatstg->ctime, &pstatstg->atime, &pstatstg->mtime);
631 if(pstatstg->cbSize.QuadPart)
632 pstatstg->grfMode = GENERIC_READ;
633 }
634
635 return S_OK;
636 }
637
638 static HRESULT WINAPI ProtocolStream_Clone(IStream *iface, IStream **ppstm)
639 {
640 ProtocolStream *This = impl_from_IStream(iface);
641 FIXME("(%p)->(%p)\n", This, ppstm);
642 return E_NOTIMPL;
643 }
644
645 static const IStreamVtbl ProtocolStreamVtbl = {
646 ProtocolStream_QueryInterface,
647 ProtocolStream_AddRef,
648 ProtocolStream_Release,
649 ProtocolStream_Read,
650 ProtocolStream_Write,
651 ProtocolStream_Seek,
652 ProtocolStream_SetSize,
653 ProtocolStream_CopyTo,
654 ProtocolStream_Commit,
655 ProtocolStream_Revert,
656 ProtocolStream_LockRegion,
657 ProtocolStream_UnlockRegion,
658 ProtocolStream_Stat,
659 ProtocolStream_Clone
660 };
661
662 static void stgmed_stream_release(stgmed_obj_t *obj)
663 {
664 ProtocolStream *stream = (ProtocolStream*)obj;
665 IStream_Release(&stream->IStream_iface);
666 }
667
668 static HRESULT stgmed_stream_fill_stgmed(stgmed_obj_t *obj, STGMEDIUM *stgmed)
669 {
670 ProtocolStream *stream = (ProtocolStream*)obj;
671
672 stgmed->tymed = TYMED_ISTREAM;
673 stgmed->u.pstm = &stream->IStream_iface;
674 stgmed->pUnkForRelease = &stream->buf->IUnknown_iface;
675
676 return S_OK;
677 }
678
679 static HRESULT stgmed_stream_get_result(stgmed_obj_t *obj, DWORD bindf, void **result)
680 {
681 ProtocolStream *stream = (ProtocolStream*)obj;
682
683 if(!(bindf & BINDF_ASYNCHRONOUS) && stream->buf->file == INVALID_HANDLE_VALUE
684 && stream->buf->hres != S_FALSE)
685 return INET_E_DATA_NOT_AVAILABLE;
686
687 IStream_AddRef(&stream->IStream_iface);
688 *result = &stream->IStream_iface;
689 return S_OK;
690 }
691
692 static const stgmed_obj_vtbl stgmed_stream_vtbl = {
693 stgmed_stream_release,
694 stgmed_stream_fill_stgmed,
695 stgmed_stream_get_result
696 };
697
698 typedef struct {
699 stgmed_obj_t stgmed_obj;
700 stgmed_buf_t *buf;
701 } stgmed_file_obj_t;
702
703 static stgmed_obj_t *create_stgmed_stream(stgmed_buf_t *buf)
704 {
705 ProtocolStream *ret = heap_alloc(sizeof(ProtocolStream));
706
707 ret->stgmed_obj.vtbl = &stgmed_stream_vtbl;
708 ret->IStream_iface.lpVtbl = &ProtocolStreamVtbl;
709 ret->ref = 1;
710
711 IUnknown_AddRef(&buf->IUnknown_iface);
712 ret->buf = buf;
713
714 URLMON_LockModule();
715
716 return &ret->stgmed_obj;
717 }
718
719 static void stgmed_file_release(stgmed_obj_t *obj)
720 {
721 stgmed_file_obj_t *file_obj = (stgmed_file_obj_t*)obj;
722
723 IUnknown_Release(&file_obj->buf->IUnknown_iface);
724 heap_free(file_obj);
725 }
726
727 static HRESULT stgmed_file_fill_stgmed(stgmed_obj_t *obj, STGMEDIUM *stgmed)
728 {
729 stgmed_file_obj_t *file_obj = (stgmed_file_obj_t*)obj;
730
731 if(!file_obj->buf->cache_file) {
732 WARN("cache_file not set\n");
733 return INET_E_DATA_NOT_AVAILABLE;
734 }
735
736 read_protocol_data(file_obj->buf);
737
738 stgmed->tymed = TYMED_FILE;
739 stgmed->u.lpszFileName = file_obj->buf->cache_file;
740 stgmed->pUnkForRelease = &file_obj->buf->IUnknown_iface;
741
742 return S_OK;
743 }
744
745 static HRESULT stgmed_file_get_result(stgmed_obj_t *obj, DWORD bindf, void **result)
746 {
747 return bindf & BINDF_ASYNCHRONOUS ? MK_S_ASYNCHRONOUS : S_OK;
748 }
749
750 static const stgmed_obj_vtbl stgmed_file_vtbl = {
751 stgmed_file_release,
752 stgmed_file_fill_stgmed,
753 stgmed_file_get_result
754 };
755
756 static stgmed_obj_t *create_stgmed_file(stgmed_buf_t *buf)
757 {
758 stgmed_file_obj_t *ret = heap_alloc(sizeof(*ret));
759
760 ret->stgmed_obj.vtbl = &stgmed_file_vtbl;
761
762 IUnknown_AddRef(&buf->IUnknown_iface);
763 ret->buf = buf;
764
765 return &ret->stgmed_obj;
766 }
767
768 static inline Binding *impl_from_IBinding(IBinding *iface)
769 {
770 return CONTAINING_RECORD(iface, Binding, IBinding_iface);
771 }
772
773 static HRESULT WINAPI Binding_QueryInterface(IBinding *iface, REFIID riid, void **ppv)
774 {
775 Binding *This = impl_from_IBinding(iface);
776
777 *ppv = NULL;
778
779 if(IsEqualGUID(&IID_IUnknown, riid)) {
780 TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
781 *ppv = &This->IBinding_iface;
782 }else if(IsEqualGUID(&IID_IBinding, riid)) {
783 TRACE("(%p)->(IID_IBinding %p)\n", This, ppv);
784 *ppv = &This->IBinding_iface;
785 }else if(IsEqualGUID(&IID_IInternetProtocolSink, riid)) {
786 TRACE("(%p)->(IID_IInternetProtocolSink %p)\n", This, ppv);
787 *ppv = &This->IInternetProtocolSink_iface;
788 }else if(IsEqualGUID(&IID_IInternetBindInfo, riid)) {
789 TRACE("(%p)->(IID_IInternetBindInfo %p)\n", This, ppv);
790 *ppv = &This->IInternetBindInfo_iface;
791 }else if(IsEqualGUID(&IID_IServiceProvider, riid)) {
792 TRACE("(%p)->(IID_IServiceProvider %p)\n", This, ppv);
793 *ppv = &This->IServiceProvider_iface;
794 }else if(IsEqualGUID(&IID_IWinInetInfo, riid)) {
795 IWinInetInfo *wininet_info;
796 HRESULT hres;
797
798 TRACE("(%p)->(IID_IWinInetInfo %p)\n", This, ppv);
799
800 /* NOTE: This violidates COM rules, but tests prove that we should do it */
801 hres = IInternetProtocolEx_QueryInterface(&This->protocol->IInternetProtocolEx_iface,
802 &IID_IWinInetInfo, (void**)&wininet_info);
803 if(SUCCEEDED(hres)) {
804 IWinInetInfo_Release(wininet_info);
805 *ppv = &This->IWinInetHttpInfo_iface;
806 }
807 }else if(IsEqualGUID(&IID_IWinInetHttpInfo, riid)) {
808 IWinInetHttpInfo *http_info;
809 HRESULT hres;
810
811 TRACE("(%p)->(IID_IWinInetHttpInfo %p)\n", This, ppv);
812
813 /* NOTE: This violidates COM rules, but tests prove that we should do it */
814 hres = IInternetProtocolEx_QueryInterface(&This->protocol->IInternetProtocolEx_iface,
815 &IID_IWinInetHttpInfo, (void**)&http_info);
816 if(SUCCEEDED(hres)) {
817 IWinInetHttpInfo_Release(http_info);
818 *ppv = &This->IWinInetHttpInfo_iface;
819 }
820 }
821
822 if(*ppv) {
823 IBinding_AddRef(&This->IBinding_iface);
824 return S_OK;
825 }
826
827 WARN("Unsupported interface %s\n", debugstr_guid(riid));
828 return E_NOINTERFACE;
829 }
830
831 static ULONG WINAPI Binding_AddRef(IBinding *iface)
832 {
833 Binding *This = impl_from_IBinding(iface);
834 LONG ref = InterlockedIncrement(&This->ref);
835
836 TRACE("(%p) ref=%d\n", This, ref);
837
838 return ref;
839 }
840
841 static ULONG WINAPI Binding_Release(IBinding *iface)
842 {
843 Binding *This = impl_from_IBinding(iface);
844 LONG ref = InterlockedDecrement(&This->ref);
845
846 TRACE("(%p) ref=%d\n", This, ref);
847
848 if(!ref) {
849 if(This->notif_hwnd)
850 release_notif_hwnd(This->notif_hwnd);
851 if(This->mon)
852 IMoniker_Release(This->mon);
853 if(This->callback)
854 IBindStatusCallback_Release(This->callback);
855 if(This->protocol)
856 IInternetProtocolEx_Release(&This->protocol->IInternetProtocolEx_iface);
857 if(This->service_provider)
858 IServiceProvider_Release(This->service_provider);
859 if(This->stgmed_buf)
860 IUnknown_Release(&This->stgmed_buf->IUnknown_iface);
861 if(This->stgmed_obj)
862 This->stgmed_obj->vtbl->release(This->stgmed_obj);
863 if(This->obj)
864 IUnknown_Release(This->obj);
865 if(This->bctx)
866 IBindCtx_Release(This->bctx);
867
868 ReleaseBindInfo(&This->bindinfo);
869 This->section.DebugInfo->Spare[0] = 0;
870 DeleteCriticalSection(&This->section);
871 SysFreeString(This->url);
872 heap_free(This->mime);
873 heap_free(This->redirect_url);
874 heap_free(This);
875
876 URLMON_UnlockModule();
877 }
878
879 return ref;
880 }
881
882 static HRESULT WINAPI Binding_Abort(IBinding *iface)
883 {
884 Binding *This = impl_from_IBinding(iface);
885 HRESULT hres;
886
887 TRACE("(%p)\n", This);
888
889 if(This->state & BINDING_ABORTED)
890 return E_FAIL;
891
892 hres = IInternetProtocolEx_Abort(&This->protocol->IInternetProtocolEx_iface, E_ABORT,
893 ERROR_SUCCESS);
894 if(FAILED(hres))
895 return hres;
896
897 This->state |= BINDING_ABORTED;
898 return S_OK;
899 }
900
901 static HRESULT WINAPI Binding_Suspend(IBinding *iface)
902 {
903 Binding *This = impl_from_IBinding(iface);
904 FIXME("(%p)\n", This);
905 return E_NOTIMPL;
906 }
907
908 static HRESULT WINAPI Binding_Resume(IBinding *iface)
909 {
910 Binding *This = impl_from_IBinding(iface);
911 FIXME("(%p)\n", This);
912 return E_NOTIMPL;
913 }
914
915 static HRESULT WINAPI Binding_SetPriority(IBinding *iface, LONG nPriority)
916 {
917 Binding *This = impl_from_IBinding(iface);
918 FIXME("(%p)->(%d)\n", This, nPriority);
919 return E_NOTIMPL;
920 }
921
922 static HRESULT WINAPI Binding_GetPriority(IBinding *iface, LONG *pnPriority)
923 {
924 Binding *This = impl_from_IBinding(iface);
925 FIXME("(%p)->(%p)\n", This, pnPriority);
926 return E_NOTIMPL;
927 }
928
929 static HRESULT WINAPI Binding_GetBindResult(IBinding *iface, CLSID *pclsidProtocol,
930 DWORD *pdwResult, LPOLESTR *pszResult, DWORD *pdwReserved)
931 {
932 Binding *This = impl_from_IBinding(iface);
933
934 TRACE("(%p)->(%p %p %p %p)\n", This, pclsidProtocol, pdwResult, pszResult, pdwReserved);
935
936 if(!pdwResult || !pszResult || pdwReserved)
937 return E_INVALIDARG;
938
939 if(!(This->state & BINDING_STOPPED)) {
940 *pclsidProtocol = CLSID_NULL;
941 *pdwResult = 0;
942 *pszResult = NULL;
943 return S_OK;
944 }
945
946 *pclsidProtocol = This->hres==S_OK ? CLSID_NULL : This->clsid;
947 *pdwResult = This->hres;
948 *pszResult = NULL;
949 return S_OK;
950 }
951
952 static const IBindingVtbl BindingVtbl = {
953 Binding_QueryInterface,
954 Binding_AddRef,
955 Binding_Release,
956 Binding_Abort,
957 Binding_Suspend,
958 Binding_Resume,
959 Binding_SetPriority,
960 Binding_GetPriority,
961 Binding_GetBindResult
962 };
963
964 static Binding *get_bctx_binding(IBindCtx *bctx)
965 {
966 IBinding *binding;
967 IUnknown *unk;
968 HRESULT hres;
969
970 hres = IBindCtx_GetObjectParam(bctx, cbinding_contextW, &unk);
971 if(FAILED(hres))
972 return NULL;
973
974 hres = IUnknown_QueryInterface(unk, &IID_IBinding, (void**)&binding);
975 IUnknown_Release(unk);
976 if(FAILED(hres))
977 return NULL;
978
979 if (binding->lpVtbl != &BindingVtbl)
980 return NULL;
981 return impl_from_IBinding(binding);
982 }
983
984 static inline Binding *impl_from_IInternetProtocolSink(IInternetProtocolSink *iface)
985 {
986 return CONTAINING_RECORD(iface, Binding, IInternetProtocolSink_iface);
987 }
988
989 static HRESULT WINAPI InternetProtocolSink_QueryInterface(IInternetProtocolSink *iface,
990 REFIID riid, void **ppv)
991 {
992 Binding *This = impl_from_IInternetProtocolSink(iface);
993 return IBinding_QueryInterface(&This->IBinding_iface, riid, ppv);
994 }
995
996 static ULONG WINAPI InternetProtocolSink_AddRef(IInternetProtocolSink *iface)
997 {
998 Binding *This = impl_from_IInternetProtocolSink(iface);
999 return IBinding_AddRef(&This->IBinding_iface);
1000 }
1001
1002 static ULONG WINAPI InternetProtocolSink_Release(IInternetProtocolSink *iface)
1003 {
1004 Binding *This = impl_from_IInternetProtocolSink(iface);
1005 return IBinding_Release(&This->IBinding_iface);
1006 }
1007
1008 static HRESULT WINAPI InternetProtocolSink_Switch(IInternetProtocolSink *iface,
1009 PROTOCOLDATA *pProtocolData)
1010 {
1011 Binding *This = impl_from_IInternetProtocolSink(iface);
1012
1013 WARN("(%p)->(%p)\n", This, pProtocolData);
1014
1015 return E_FAIL;
1016 }
1017
1018 static void on_progress(Binding *This, ULONG progress, ULONG progress_max,
1019 ULONG status_code, LPCWSTR status_text)
1020 {
1021 IBindStatusCallback_OnProgress(This->callback, progress, progress_max,
1022 status_code, status_text);
1023 }
1024
1025 static HRESULT WINAPI InternetProtocolSink_ReportProgress(IInternetProtocolSink *iface,
1026 ULONG ulStatusCode, LPCWSTR szStatusText)
1027 {
1028 Binding *This = impl_from_IInternetProtocolSink(iface);
1029
1030 TRACE("(%p)->(%s %s)\n", This, debugstr_bindstatus(ulStatusCode), debugstr_w(szStatusText));
1031
1032 switch(ulStatusCode) {
1033 case BINDSTATUS_FINDINGRESOURCE:
1034 on_progress(This, 0, 0, BINDSTATUS_FINDINGRESOURCE, szStatusText);
1035 break;
1036 case BINDSTATUS_CONNECTING:
1037 on_progress(This, 0, 0, BINDSTATUS_CONNECTING, szStatusText);
1038 break;
1039 case BINDSTATUS_REDIRECTING:
1040 heap_free(This->redirect_url);
1041 This->redirect_url = heap_strdupW(szStatusText);
1042 on_progress(This, 0, 0, BINDSTATUS_REDIRECTING, szStatusText);
1043 break;
1044 case BINDSTATUS_BEGINDOWNLOADDATA:
1045 break;
1046 case BINDSTATUS_SENDINGREQUEST:
1047 on_progress(This, 0, 0, BINDSTATUS_SENDINGREQUEST, szStatusText);
1048 break;
1049 case BINDSTATUS_PROTOCOLCLASSID:
1050 CLSIDFromString(szStatusText, &This->clsid);
1051 break;
1052 case BINDSTATUS_MIMETYPEAVAILABLE:
1053 case BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE:
1054 mime_available(This, szStatusText);
1055 break;
1056 case BINDSTATUS_CACHEFILENAMEAVAILABLE:
1057 cache_file_available(This, szStatusText);
1058 break;
1059 case BINDSTATUS_DECODING:
1060 IBindStatusCallback_OnProgress(This->callback, 0, 0, BINDSTATUS_DECODING, szStatusText);
1061 break;
1062 case BINDSTATUS_LOADINGMIMEHANDLER:
1063 on_progress(This, 0, 0, BINDSTATUS_LOADINGMIMEHANDLER, szStatusText);
1064 break;
1065 case BINDSTATUS_DIRECTBIND: /* FIXME: Handle BINDSTATUS_DIRECTBIND in BindProtocol */
1066 This->report_mime = FALSE;
1067 break;
1068 case BINDSTATUS_ACCEPTRANGES:
1069 break;
1070 default:
1071 FIXME("Unhandled status code %d\n", ulStatusCode);
1072 return E_NOTIMPL;
1073 };
1074
1075 return S_OK;
1076 }
1077
1078 static void report_data(Binding *This, DWORD bscf, ULONG progress, ULONG progress_max)
1079 {
1080 FORMATETC formatetc = {0, NULL, 1, -1, TYMED_ISTREAM};
1081 BOOL sent_begindownloaddata = FALSE;
1082
1083 TRACE("(%p)->(%d %u %u)\n", This, bscf, progress, progress_max);
1084
1085 if(This->download_state == END_DOWNLOAD || (This->state & BINDING_ABORTED)) {
1086 read_protocol_data(This->stgmed_buf);
1087 return;
1088 }
1089
1090 if(This->state & BINDING_STOPPED)
1091 return;
1092
1093 if(This->stgmed_buf->file != INVALID_HANDLE_VALUE)
1094 read_protocol_data(This->stgmed_buf);
1095
1096 if(This->download_state == BEFORE_DOWNLOAD) {
1097 This->download_state = DOWNLOADING;
1098 sent_begindownloaddata = TRUE;
1099 IBindStatusCallback_OnProgress(This->callback, progress, progress_max,
1100 BINDSTATUS_BEGINDOWNLOADDATA, This->url);
1101
1102 if(This->stgmed_buf->cache_file)
1103 IBindStatusCallback_OnProgress(This->callback, progress, progress_max,
1104 BINDSTATUS_CACHEFILENAMEAVAILABLE, This->stgmed_buf->cache_file);
1105 }
1106
1107 if(This->stgmed_buf->hres == S_FALSE || (bscf & BSCF_LASTDATANOTIFICATION)) {
1108 This->download_state = END_DOWNLOAD;
1109 IBindStatusCallback_OnProgress(This->callback, progress, progress_max,
1110 BINDSTATUS_ENDDOWNLOADDATA, This->url);
1111 }else if(!sent_begindownloaddata) {
1112 IBindStatusCallback_OnProgress(This->callback, progress, progress_max,
1113 BINDSTATUS_DOWNLOADINGDATA, This->url);
1114 }
1115
1116 if(This->state & (BINDING_STOPPED|BINDING_ABORTED))
1117 return;
1118
1119 if(This->to_object) {
1120 if(!(This->state & BINDING_OBJAVAIL)) {
1121 IBinding_AddRef(&This->IBinding_iface);
1122 create_object(This);
1123 IBinding_Release(&This->IBinding_iface);
1124 }
1125 }else {
1126 STGMEDIUM stgmed;
1127 HRESULT hres;
1128
1129 if(!(This->state & BINDING_LOCKED)) {
1130 HRESULT hres = IInternetProtocolEx_LockRequest(
1131 &This->protocol->IInternetProtocolEx_iface, 0);
1132 if(SUCCEEDED(hres))
1133 This->state |= BINDING_LOCKED;
1134 }
1135
1136 hres = This->stgmed_obj->vtbl->fill_stgmed(This->stgmed_obj, &stgmed);
1137 if(FAILED(hres)) {
1138 stop_binding(This, hres, NULL);
1139 return;
1140 }
1141
1142 formatetc.tymed = stgmed.tymed;
1143 formatetc.cfFormat = This->clipboard_format;
1144
1145 hres = IBindStatusCallback_OnDataAvailable(This->callback, bscf, progress,
1146 &formatetc, &stgmed);
1147 if(hres != S_OK) {
1148 if(This->download_state != END_DOWNLOAD) {
1149 This->download_state = END_DOWNLOAD;
1150 IBindStatusCallback_OnProgress(This->callback, progress, progress_max,
1151 BINDSTATUS_ENDDOWNLOADDATA, This->url);
1152 }
1153
1154 WARN("OnDataAvailable returned %x\n", hres);
1155 stop_binding(This, hres, NULL);
1156 return;
1157 }
1158
1159 if(This->download_state == END_DOWNLOAD)
1160 stop_binding(This, S_OK, NULL);
1161 }
1162 }
1163
1164 static HRESULT WINAPI InternetProtocolSink_ReportData(IInternetProtocolSink *iface,
1165 DWORD grfBSCF, ULONG ulProgress, ULONG ulProgressMax)
1166 {
1167 Binding *This = impl_from_IInternetProtocolSink(iface);
1168
1169 TRACE("(%p)->(%d %u %u)\n", This, grfBSCF, ulProgress, ulProgressMax);
1170
1171 report_data(This, grfBSCF, ulProgress, ulProgressMax);
1172 return S_OK;
1173 }
1174
1175 static HRESULT WINAPI InternetProtocolSink_ReportResult(IInternetProtocolSink *iface,
1176 HRESULT hrResult, DWORD dwError, LPCWSTR szResult)
1177 {
1178 Binding *This = impl_from_IInternetProtocolSink(iface);
1179
1180 TRACE("(%p)->(%08x %d %s)\n", This, hrResult, dwError, debugstr_w(szResult));
1181
1182 stop_binding(This, hrResult, szResult);
1183
1184 IInternetProtocolEx_Terminate(&This->protocol->IInternetProtocolEx_iface, 0);
1185 return S_OK;
1186 }
1187
1188 static const IInternetProtocolSinkVtbl InternetProtocolSinkVtbl = {
1189 InternetProtocolSink_QueryInterface,
1190 InternetProtocolSink_AddRef,
1191 InternetProtocolSink_Release,
1192 InternetProtocolSink_Switch,
1193 InternetProtocolSink_ReportProgress,
1194 InternetProtocolSink_ReportData,
1195 InternetProtocolSink_ReportResult
1196 };
1197
1198 static inline Binding *impl_from_IInternetBindInfo(IInternetBindInfo *iface)
1199 {
1200 return CONTAINING_RECORD(iface, Binding, IInternetBindInfo_iface);
1201 }
1202
1203 static HRESULT WINAPI InternetBindInfo_QueryInterface(IInternetBindInfo *iface,
1204 REFIID riid, void **ppv)
1205 {
1206 Binding *This = impl_from_IInternetBindInfo(iface);
1207 return IBinding_QueryInterface(&This->IBinding_iface, riid, ppv);
1208 }
1209
1210 static ULONG WINAPI InternetBindInfo_AddRef(IInternetBindInfo *iface)
1211 {
1212 Binding *This = impl_from_IInternetBindInfo(iface);
1213 return IBinding_AddRef(&This->IBinding_iface);
1214 }
1215
1216 static ULONG WINAPI InternetBindInfo_Release(IInternetBindInfo *iface)
1217 {
1218 Binding *This = impl_from_IInternetBindInfo(iface);
1219 return IBinding_Release(&This->IBinding_iface);
1220 }
1221
1222 static HRESULT WINAPI InternetBindInfo_GetBindInfo(IInternetBindInfo *iface,
1223 DWORD *grfBINDF, BINDINFO *pbindinfo)
1224 {
1225 Binding *This = impl_from_IInternetBindInfo(iface);
1226
1227 TRACE("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo);
1228
1229 *grfBINDF = This->bindf;
1230 return CopyBindInfo(&This->bindinfo, pbindinfo);
1231 }
1232
1233 static HRESULT WINAPI InternetBindInfo_GetBindString(IInternetBindInfo *iface,
1234 ULONG ulStringType, LPOLESTR *ppwzStr, ULONG cEl, ULONG *pcElFetched)
1235 {
1236 Binding *This = impl_from_IInternetBindInfo(iface);
1237
1238 TRACE("(%p)->(%d %p %d %p)\n", This, ulStringType, ppwzStr, cEl, pcElFetched);
1239
1240 switch(ulStringType) {
1241 case BINDSTRING_ACCEPT_MIMES: {
1242 static const WCHAR wszMimes[] = {'*','/','*',0};
1243
1244 if(!ppwzStr || !pcElFetched)
1245 return E_INVALIDARG;
1246
1247 ppwzStr[0] = CoTaskMemAlloc(sizeof(wszMimes));
1248 memcpy(ppwzStr[0], wszMimes, sizeof(wszMimes));
1249 *pcElFetched = 1;
1250 return S_OK;
1251 }
1252 case BINDSTRING_USER_AGENT: {
1253 IInternetBindInfo *bindinfo = NULL;
1254 HRESULT hres;
1255
1256 hres = IBindStatusCallback_QueryInterface(This->callback, &IID_IInternetBindInfo,
1257 (void**)&bindinfo);
1258 if(FAILED(hres))
1259 return hres;
1260
1261 hres = IInternetBindInfo_GetBindString(bindinfo, ulStringType, ppwzStr,
1262 cEl, pcElFetched);
1263 IInternetBindInfo_Release(bindinfo);
1264
1265 return hres;
1266 }
1267 case BINDSTRING_URL: {
1268 DWORD size = (SysStringLen(This->url)+1) * sizeof(WCHAR);
1269
1270 if(!ppwzStr || !pcElFetched)
1271 return E_INVALIDARG;
1272
1273 *ppwzStr = CoTaskMemAlloc(size);
1274 memcpy(*ppwzStr, This->url, size);
1275 *pcElFetched = 1;
1276 return S_OK;
1277 }
1278 }
1279
1280 FIXME("not supported string type %d\n", ulStringType);
1281 return E_NOTIMPL;
1282 }
1283
1284 static const IInternetBindInfoVtbl InternetBindInfoVtbl = {
1285 InternetBindInfo_QueryInterface,
1286 InternetBindInfo_AddRef,
1287 InternetBindInfo_Release,
1288 InternetBindInfo_GetBindInfo,
1289 InternetBindInfo_GetBindString
1290 };
1291
1292 static inline Binding *impl_from_IWinInetHttpInfo(IWinInetHttpInfo *iface)
1293 {
1294 return CONTAINING_RECORD(iface, Binding, IWinInetHttpInfo_iface);
1295 }
1296
1297 static HRESULT WINAPI WinInetHttpInfo_QueryInterface(IWinInetHttpInfo *iface, REFIID riid, void **ppv)
1298 {
1299 Binding *This = impl_from_IWinInetHttpInfo(iface);
1300 return IBinding_QueryInterface(&This->IBinding_iface, riid, ppv);
1301 }
1302
1303 static ULONG WINAPI WinInetHttpInfo_AddRef(IWinInetHttpInfo *iface)
1304 {
1305 Binding *This = impl_from_IWinInetHttpInfo(iface);
1306 return IBinding_AddRef(&This->IBinding_iface);
1307 }
1308
1309 static ULONG WINAPI WinInetHttpInfo_Release(IWinInetHttpInfo *iface)
1310 {
1311 Binding *This = impl_from_IWinInetHttpInfo(iface);
1312 return IBinding_Release(&This->IBinding_iface);
1313 }
1314
1315 static HRESULT WINAPI WinInetHttpInfo_QueryOption(IWinInetHttpInfo *iface, DWORD dwOption,
1316 void *pBuffer, DWORD *pcbBuffer)
1317 {
1318 Binding *This = impl_from_IWinInetHttpInfo(iface);
1319 IWinInetInfo *wininet_info;
1320 HRESULT hres;
1321
1322 TRACE("(%p)->(%x %p %p)\n", This, dwOption, pBuffer, pcbBuffer);
1323
1324 hres = IInternetProtocolEx_QueryInterface(&This->protocol->IInternetProtocolEx_iface,
1325 &IID_IWinInetInfo, (void**)&wininet_info);
1326 if(FAILED(hres))
1327 return E_FAIL;
1328
1329 hres = IWinInetInfo_QueryOption(wininet_info, dwOption, pBuffer, pcbBuffer);
1330 IWinInetInfo_Release(wininet_info);
1331 return hres;
1332 }
1333
1334 static HRESULT WINAPI WinInetHttpInfo_QueryInfo(IWinInetHttpInfo *iface, DWORD dwOption,
1335 void *pBuffer, DWORD *pcbBuffer, DWORD *pdwFlags, DWORD *pdwReserved)
1336 {
1337 Binding *This = impl_from_IWinInetHttpInfo(iface);
1338 IWinInetHttpInfo *http_info;
1339 HRESULT hres;
1340
1341 TRACE("(%p)->(%x %p %p %p %p)\n", This, dwOption, pBuffer, pcbBuffer, pdwFlags, pdwReserved);
1342
1343 hres = IInternetProtocolEx_QueryInterface(&This->protocol->IInternetProtocolEx_iface,
1344 &IID_IWinInetHttpInfo, (void**)&http_info);
1345 if(FAILED(hres))
1346 return E_FAIL;
1347
1348 hres = IWinInetHttpInfo_QueryInfo(http_info, dwOption, pBuffer, pcbBuffer, pdwFlags, pdwReserved);
1349 IWinInetHttpInfo_Release(http_info);
1350 return hres;
1351 }
1352
1353 static const IWinInetHttpInfoVtbl WinInetHttpInfoVtbl = {
1354 WinInetHttpInfo_QueryInterface,
1355 WinInetHttpInfo_AddRef,
1356 WinInetHttpInfo_Release,
1357 WinInetHttpInfo_QueryOption,
1358 WinInetHttpInfo_QueryInfo
1359 };
1360
1361 static inline Binding *impl_from_IServiceProvider(IServiceProvider *iface)
1362 {
1363 return CONTAINING_RECORD(iface, Binding, IServiceProvider_iface);
1364 }
1365
1366 static HRESULT WINAPI ServiceProvider_QueryInterface(IServiceProvider *iface,
1367 REFIID riid, void **ppv)
1368 {
1369 Binding *This = impl_from_IServiceProvider(iface);
1370 return IBinding_QueryInterface(&This->IBinding_iface, riid, ppv);
1371 }
1372
1373 static ULONG WINAPI ServiceProvider_AddRef(IServiceProvider *iface)
1374 {
1375 Binding *This = impl_from_IServiceProvider(iface);
1376 return IBinding_AddRef(&This->IBinding_iface);
1377 }
1378
1379 static ULONG WINAPI ServiceProvider_Release(IServiceProvider *iface)
1380 {
1381 Binding *This = impl_from_IServiceProvider(iface);
1382 return IBinding_Release(&This->IBinding_iface);
1383 }
1384
1385 static HRESULT WINAPI ServiceProvider_QueryService(IServiceProvider *iface,
1386 REFGUID guidService, REFIID riid, void **ppv)
1387 {
1388 Binding *This = impl_from_IServiceProvider(iface);
1389 HRESULT hres;
1390
1391 TRACE("(%p)->(%s %s %p)\n", This, debugstr_guid(guidService), debugstr_guid(riid), ppv);
1392
1393 if(This->service_provider) {
1394 hres = IServiceProvider_QueryService(This->service_provider, guidService,
1395 riid, ppv);
1396 if(SUCCEEDED(hres))
1397 return hres;
1398 }
1399
1400 WARN("unknown service %s\n", debugstr_guid(guidService));
1401 return E_NOINTERFACE;
1402 }
1403
1404 static const IServiceProviderVtbl ServiceProviderVtbl = {
1405 ServiceProvider_QueryInterface,
1406 ServiceProvider_AddRef,
1407 ServiceProvider_Release,
1408 ServiceProvider_QueryService
1409 };
1410
1411 static HRESULT get_callback(IBindCtx *pbc, IBindStatusCallback **callback)
1412 {
1413 IUnknown *unk;
1414 HRESULT hres;
1415
1416 hres = IBindCtx_GetObjectParam(pbc, bscb_holderW, &unk);
1417 if(FAILED(hres))
1418 return create_default_callback(callback);
1419
1420 hres = IUnknown_QueryInterface(unk, &IID_IBindStatusCallback, (void**)callback);
1421 IUnknown_Release(unk);
1422 return hres;
1423 }
1424
1425 static BOOL is_urlmon_protocol(IUri *uri)
1426 {
1427 DWORD scheme;
1428 HRESULT hres;
1429
1430 hres = IUri_GetScheme(uri, &scheme);
1431 if(FAILED(hres))
1432 return FALSE;
1433
1434 switch(scheme) {
1435 case URL_SCHEME_FILE:
1436 case URL_SCHEME_FTP:
1437 case URL_SCHEME_GOPHER:
1438 case URL_SCHEME_HTTP:
1439 case URL_SCHEME_HTTPS:
1440 case URL_SCHEME_MK:
1441 return TRUE;
1442 }
1443
1444 return FALSE;
1445 }
1446
1447 static HRESULT Binding_Create(IMoniker *mon, Binding *binding_ctx, IUri *uri, IBindCtx *pbc,
1448 BOOL to_obj, REFIID riid, Binding **binding)
1449 {
1450 Binding *ret;
1451 HRESULT hres;
1452
1453 URLMON_LockModule();
1454
1455 ret = heap_alloc_zero(sizeof(Binding));
1456
1457 ret->IBinding_iface.lpVtbl = &BindingVtbl;
1458 ret->IInternetProtocolSink_iface.lpVtbl = &InternetProtocolSinkVtbl;
1459 ret->IInternetBindInfo_iface.lpVtbl = &InternetBindInfoVtbl;
1460 ret->IWinInetHttpInfo_iface.lpVtbl = &WinInetHttpInfoVtbl;
1461 ret->IServiceProvider_iface.lpVtbl = &ServiceProviderVtbl;
1462
1463 ret->ref = 1;
1464
1465 ret->to_object = to_obj;
1466 ret->iid = *riid;
1467 ret->notif_hwnd = get_notif_hwnd();
1468 ret->report_mime = !binding_ctx;
1469 ret->download_state = BEFORE_DOWNLOAD;
1470
1471 if(to_obj) {
1472 IBindCtx_AddRef(pbc);
1473 ret->bctx = pbc;
1474 }
1475
1476 if(mon) {
1477 IMoniker_AddRef(mon);
1478 ret->mon = mon;
1479 }
1480
1481 ret->bindinfo.cbSize = sizeof(BINDINFO);
1482
1483 InitializeCriticalSection(&ret->section);
1484 ret->section.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": Binding.section");
1485
1486 hres = get_callback(pbc, &ret->callback);
1487 if(FAILED(hres)) {
1488 WARN("Could not get IBindStatusCallback\n");
1489 IBinding_Release(&ret->IBinding_iface);
1490 return hres;
1491 }
1492
1493 IBindStatusCallback_QueryInterface(ret->callback, &IID_IServiceProvider,
1494 (void**)&ret->service_provider);
1495
1496 if(binding_ctx) {
1497 ret->protocol = binding_ctx->protocol;
1498 IInternetProtocolEx_AddRef(&ret->protocol->IInternetProtocolEx_iface);
1499 }else {
1500 hres = create_binding_protocol(&ret->protocol);
1501 if(FAILED(hres)) {
1502 WARN("Could not get protocol handler\n");
1503 IBinding_Release(&ret->IBinding_iface);
1504 return hres;
1505 }
1506 }
1507
1508 hres = IBindStatusCallback_GetBindInfo(ret->callback, &ret->bindf, &ret->bindinfo);
1509 if(FAILED(hres)) {
1510 WARN("GetBindInfo failed: %08x\n", hres);
1511 IBinding_Release(&ret->IBinding_iface);
1512 return hres;
1513 }
1514
1515 TRACE("bindf %08x\n", ret->bindf);
1516 dump_BINDINFO(&ret->bindinfo);
1517
1518 ret->bindf |= BINDF_FROMURLMON;
1519 if(to_obj)
1520 ret->bindinfo.dwOptions |= 0x100000;
1521
1522 if(!(ret->bindf & BINDF_ASYNCHRONOUS) || !(ret->bindf & BINDF_PULLDATA)) {
1523 ret->bindf |= BINDF_NEEDFILE;
1524 ret->use_cache_file = TRUE;
1525 }else if(!is_urlmon_protocol(uri)) {
1526 ret->bindf |= BINDF_NEEDFILE;
1527 }
1528
1529 hres = IUri_GetDisplayUri(uri, &ret->url);
1530 if(FAILED(hres)) {
1531 IBinding_Release(&ret->IBinding_iface);
1532 return hres;
1533 }
1534
1535 if(binding_ctx) {
1536 ret->stgmed_buf = binding_ctx->stgmed_buf;
1537 IUnknown_AddRef(&ret->stgmed_buf->IUnknown_iface);
1538 ret->clipboard_format = binding_ctx->clipboard_format;
1539 }else {
1540 ret->stgmed_buf = create_stgmed_buf(&ret->protocol->IInternetProtocolEx_iface);
1541 }
1542
1543 if(to_obj) {
1544 ret->stgmed_obj = NULL;
1545 }else if(IsEqualGUID(&IID_IStream, riid)) {
1546 ret->stgmed_obj = create_stgmed_stream(ret->stgmed_buf);
1547 }else if(IsEqualGUID(&IID_IUnknown, riid)) {
1548 ret->bindf |= BINDF_NEEDFILE;
1549 ret->stgmed_obj = create_stgmed_file(ret->stgmed_buf);
1550 }else {
1551 FIXME("Unsupported riid %s\n", debugstr_guid(riid));
1552 IBinding_Release(&ret->IBinding_iface);
1553 return E_NOTIMPL;
1554 }
1555
1556 *binding = ret;
1557 return S_OK;
1558 }
1559
1560 static HRESULT start_binding(IMoniker *mon, Binding *binding_ctx, IUri *uri, IBindCtx *pbc,
1561 BOOL to_obj, REFIID riid, Binding **ret)
1562 {
1563 Binding *binding = NULL;
1564 HRESULT hres;
1565 MSG msg;
1566
1567 hres = Binding_Create(mon, binding_ctx, uri, pbc, to_obj, riid, &binding);
1568 if(FAILED(hres))
1569 return hres;
1570
1571 hres = IBindStatusCallback_OnStartBinding(binding->callback, 0, &binding->IBinding_iface);
1572 if(FAILED(hres)) {
1573 WARN("OnStartBinding failed: %08x\n", hres);
1574 if(hres != E_ABORT && hres != E_NOTIMPL)
1575 hres = INET_E_DOWNLOAD_FAILURE;
1576
1577 stop_binding(binding, hres, NULL);
1578 IBinding_Release(&binding->IBinding_iface);
1579 return hres;
1580 }
1581
1582 if(binding_ctx) {
1583 set_binding_sink(binding->protocol, &binding->IInternetProtocolSink_iface,
1584 &binding->IInternetBindInfo_iface);
1585 if(binding_ctx->redirect_url)
1586 IBindStatusCallback_OnProgress(binding->callback, 0, 0, BINDSTATUS_REDIRECTING, binding_ctx->redirect_url);
1587 report_data(binding, BSCF_FIRSTDATANOTIFICATION | (binding_ctx->download_state == END_DOWNLOAD ? BSCF_LASTDATANOTIFICATION : 0),
1588 0, 0);
1589 }else {
1590 hres = IInternetProtocolEx_StartEx(&binding->protocol->IInternetProtocolEx_iface, uri,
1591 &binding->IInternetProtocolSink_iface, &binding->IInternetBindInfo_iface,
1592 PI_APARTMENTTHREADED|PI_MIMEVERIFICATION, 0);
1593
1594 TRACE("start ret %08x\n", hres);
1595
1596 if(FAILED(hres) && hres != E_PENDING) {
1597 stop_binding(binding, hres, NULL);
1598 IBinding_Release(&binding->IBinding_iface);
1599
1600 return hres;
1601 }
1602 }
1603
1604 while(!(binding->bindf & BINDF_ASYNCHRONOUS) &&
1605 !(binding->state & BINDING_STOPPED)) {
1606 MsgWaitForMultipleObjects(0, NULL, FALSE, 5000, QS_POSTMESSAGE);
1607 while (PeekMessageW(&msg, binding->notif_hwnd, WM_USER, WM_USER+117, PM_REMOVE|PM_NOYIELD)) {
1608 TranslateMessage(&msg);
1609 DispatchMessageW(&msg);
1610 }
1611 }
1612
1613 *ret = binding;
1614 return S_OK;
1615 }
1616
1617 HRESULT bind_to_storage(IUri *uri, IBindCtx *pbc, REFIID riid, void **ppv)
1618 {
1619 Binding *binding = NULL, *binding_ctx;
1620 HRESULT hres;
1621
1622 binding_ctx = get_bctx_binding(pbc);
1623
1624 hres = start_binding(NULL, binding_ctx, uri, pbc, FALSE, riid, &binding);
1625 if(binding_ctx)
1626 IBinding_Release(&binding_ctx->IBinding_iface);
1627 if(FAILED(hres))
1628 return hres;
1629
1630 if(binding->hres == S_OK && binding->download_state != BEFORE_DOWNLOAD /* FIXME */) {
1631 if((binding->state & BINDING_STOPPED) && (binding->state & BINDING_LOCKED))
1632 IInternetProtocolEx_UnlockRequest(&binding->protocol->IInternetProtocolEx_iface);
1633
1634 hres = binding->stgmed_obj->vtbl->get_result(binding->stgmed_obj, binding->bindf, ppv);
1635 }else if(binding->bindf & BINDF_ASYNCHRONOUS) {
1636 hres = MK_S_ASYNCHRONOUS;
1637 }else {
1638 hres = FAILED(binding->hres) ? binding->hres : S_OK;
1639 }
1640
1641 IBinding_Release(&binding->IBinding_iface);
1642
1643 return hres;
1644 }
1645
1646 HRESULT bind_to_object(IMoniker *mon, IUri *uri, IBindCtx *pbc, REFIID riid, void **ppv)
1647 {
1648 Binding *binding;
1649 HRESULT hres;
1650
1651 *ppv = NULL;
1652
1653 hres = start_binding(mon, NULL, uri, pbc, TRUE, riid, &binding);
1654 if(FAILED(hres))
1655 return hres;
1656
1657 if(binding->hres != S_OK) {
1658 hres = SUCCEEDED(binding->hres) ? S_OK : binding->hres;
1659 }else if(binding->bindf & BINDF_ASYNCHRONOUS) {
1660 hres = MK_S_ASYNCHRONOUS;
1661 }else {
1662 *ppv = binding->obj;
1663 IUnknown_AddRef(binding->obj);
1664 hres = S_OK;
1665 }
1666
1667 IBinding_Release(&binding->IBinding_iface);
1668
1669 return hres;
1670 }