blob: 453d6a7183a1e8274fa613f1f9944ecfdf3d6a15 [file] [log] [blame]
Jacek Caban0dcf2a42005-11-14 15:12:43 +00001/*
2 * Copyright 2005 Jacek Caban
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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 */
18
19#include <stdarg.h>
20
21#define COBJMACROS
22#define NONAMELESSUNION
23#define NONAMELESSSTRUCT
24
25#include "windef.h"
26#include "winbase.h"
27#include "winuser.h"
28#include "ole2.h"
29#include "urlmon.h"
30#include "urlmon_main.h"
31
32#include "wine/debug.h"
33#include "wine/unicode.h"
34
35WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
36
Jacek Caban762e2692006-01-09 17:18:06 +010037typedef struct ProtocolStream ProtocolStream;
38
Jacek Caban0dcf2a42005-11-14 15:12:43 +000039typedef struct {
40 const IBindingVtbl *lpBindingVtbl;
41 const IInternetProtocolSinkVtbl *lpInternetProtocolSinkVtbl;
42 const IInternetBindInfoVtbl *lpInternetBindInfoVtbl;
43 const IServiceProviderVtbl *lpServiceProviderVtbl;
44
45 LONG ref;
46
47 IBindStatusCallback *callback;
48 IInternetProtocol *protocol;
Jacek Caban6744b9b2006-01-05 13:42:32 +010049 IServiceProvider *service_provider;
Jacek Caban762e2692006-01-09 17:18:06 +010050 ProtocolStream *stream;
Jacek Caban0dcf2a42005-11-14 15:12:43 +000051
52 BINDINFO bindinfo;
53 DWORD bindf;
54 LPWSTR mime;
55 LPWSTR url;
56} Binding;
57
Jacek Caban762e2692006-01-09 17:18:06 +010058struct ProtocolStream {
59 const IStreamVtbl *lpStreamVtbl;
60
61 LONG ref;
62
63 IInternetProtocol *protocol;
64
65 BYTE buf[1024*8];
66 DWORD buf_size;
67};
68
Jacek Caban0dcf2a42005-11-14 15:12:43 +000069#define BINDING(x) ((IBinding*) &(x)->lpBindingVtbl)
70#define PROTSINK(x) ((IInternetProtocolSink*) &(x)->lpInternetProtocolSinkVtbl)
71#define BINDINF(x) ((IInternetBindInfo*) &(x)->lpInternetBindInfoVtbl)
72#define SERVPROV(x) ((IServiceProvider*) &(x)->lpServiceProviderVtbl)
73
Jacek Caban762e2692006-01-09 17:18:06 +010074#define STREAM(x) ((IStream*) &(x)->lpStreamVtbl)
75
Jacek Cabanb235d792006-01-05 13:42:43 +010076static HRESULT WINAPI HttpNegotiate_QueryInterface(IHttpNegotiate2 *iface,
77 REFIID riid, void **ppv)
78{
79 *ppv = NULL;
80
81 if(IsEqualGUID(&IID_IUnknown, riid)) {
82 TRACE("(IID_IUnknown %p)\n", ppv);
83 *ppv = iface;
84 }else if(IsEqualGUID(&IID_IHttpNegotiate, riid)) {
85 TRACE("(IID_IHttpNegotiate %p)\n", ppv);
86 *ppv = iface;
87 }else if(IsEqualGUID(&IID_IHttpNegotiate2, riid)) {
88 TRACE("(IID_IHttpNegotiate2 %p)\n", ppv);
89 *ppv = iface;
90 }
91
92 if(*ppv) {
93 IHttpNegotiate2_AddRef(iface);
94 return S_OK;
95 }
96
97 WARN("Unsupported interface %s\n", debugstr_guid(riid));
98 return E_NOINTERFACE;
99}
100
101static ULONG WINAPI HttpNegotiate_AddRef(IHttpNegotiate2 *iface)
102{
103 URLMON_LockModule();
104 return 2;
105}
106
107static ULONG WINAPI HttpNegotiate_Release(IHttpNegotiate2 *iface)
108{
109 URLMON_UnlockModule();
110 return 1;
111}
112
113static HRESULT WINAPI HttpNegotiate_BeginningTransaction(IHttpNegotiate2 *iface,
114 LPCWSTR szURL, LPCWSTR szHeaders, DWORD dwReserved, LPWSTR *pszAdditionalHeaders)
115{
116 FIXME("(%s %s %ld %p)\n", debugstr_w(szURL), debugstr_w(szHeaders), dwReserved,
117 pszAdditionalHeaders);
118 return E_NOTIMPL;
119}
120
121static HRESULT WINAPI HttpNegotiate_OnResponse(IHttpNegotiate2 *iface, DWORD dwResponseCode,
122 LPCWSTR szResponseHeaders, LPCWSTR szRequestHeaders,
123 LPWSTR *pszAdditionalRequestHeaders)
124{
125 FIXME("(%ld %s %s %p)\n", dwResponseCode, debugstr_w(szResponseHeaders),
126 debugstr_w(szRequestHeaders), pszAdditionalRequestHeaders);
127 return E_NOTIMPL;
128}
129
130static HRESULT WINAPI HttpNegotiate_GetRootSecurityId(IHttpNegotiate2 *iface,
131 BYTE *pbSecurityId, DWORD *pcbSecurityId, DWORD_PTR dwReserved)
132{
133 FIXME("(%p %p %ld)\n", pbSecurityId, pcbSecurityId, dwReserved);
134 return E_NOTIMPL;
135}
136
137static const IHttpNegotiate2Vtbl HttpNegotiate2Vtbl = {
138 HttpNegotiate_QueryInterface,
139 HttpNegotiate_AddRef,
140 HttpNegotiate_Release,
141 HttpNegotiate_BeginningTransaction,
142 HttpNegotiate_OnResponse,
143 HttpNegotiate_GetRootSecurityId
144};
145
146static IHttpNegotiate2 HttpNegotiate = { &HttpNegotiate2Vtbl };
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000147
Jacek Caban762e2692006-01-09 17:18:06 +0100148#define STREAM_THIS(iface) DEFINE_THIS(ProtocolStream, Stream, iface)
149
150static HRESULT WINAPI ProtocolStream_QueryInterface(IStream *iface,
151 REFIID riid, void **ppv)
152{
153 ProtocolStream *This = STREAM_THIS(iface);
154
155 *ppv = NULL;
156
157 if(IsEqualGUID(&IID_IUnknown, riid)) {
158 TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
159 *ppv = STREAM(This);
160 }else if(IsEqualGUID(&IID_ISequentialStream, riid)) {
161 TRACE("(%p)->(IID_ISequentialStream %p)\n", This, ppv);
162 *ppv = STREAM(This);
163 }else if(IsEqualGUID(&IID_IStream, riid)) {
164 TRACE("(%p)->(IID_IStream %p)\n", This, ppv);
165 *ppv = STREAM(This);
166 }
167
168 if(*ppv) {
169 IStream_AddRef(STREAM(This));
170 return S_OK;
171 }
172
173 WARN("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
174 return E_NOINTERFACE;
175}
176
177static ULONG WINAPI ProtocolStream_AddRef(IStream *iface)
178{
179 ProtocolStream *This = STREAM_THIS(iface);
180 LONG ref = InterlockedIncrement(&This->ref);
181
182 TRACE("(%p) ref=%ld\n", This, ref);
183
184 return ref;
185}
186
187static ULONG WINAPI ProtocolStream_Release(IStream *iface)
188{
189 ProtocolStream *This = STREAM_THIS(iface);
190 LONG ref = InterlockedDecrement(&This->ref);
191
192 TRACE("(%p) ref=%ld\n", This, ref);
193
194 if(!ref) {
195 IInternetProtocol_Release(This->protocol);
196 HeapFree(GetProcessHeap(), 0, This);
197 }
198
199 return ref;
200}
201
202static HRESULT WINAPI ProtocolStream_Read(IStream *iface, void *pv,
203 ULONG cb, ULONG *pcbRead)
204{
205 ProtocolStream *This = STREAM_THIS(iface);
206 DWORD read = 0, pread = 0;
207
208 TRACE("(%p)->(%p %ld %p)\n", This, pv, cb, pcbRead);
209
210 if(This->buf_size) {
211 read = cb;
212
213 if(read > This->buf_size)
214 read = This->buf_size;
215
216 memcpy(pv, This->buf, read);
217
218 if(read < This->buf_size)
219 memmove(This->buf, This->buf+read, This->buf_size-read);
220 This->buf_size -= read;
221 }
222
223 if(read == cb) {
224 *pcbRead = read;
225 return S_OK;
226 }
227
228 IInternetProtocol_Read(This->protocol, (PBYTE)pv+read, cb-read, &pread);
229 *pcbRead = read + pread;
230
231 return read || pread ? S_OK : S_FALSE;
232}
233
234static HRESULT WINAPI ProtocolStream_Write(IStream *iface, const void *pv,
235 ULONG cb, ULONG *pcbWritten)
236{
237 ProtocolStream *This = STREAM_THIS(iface);
238
239 TRACE("(%p)->(%p %ld %p)\n", This, pv, cb, pcbWritten);
240
241 return STG_E_ACCESSDENIED;
242}
243
244static HRESULT WINAPI ProtocolStream_Seek(IStream *iface, LARGE_INTEGER dlibMove,
245 DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
246{
247 ProtocolStream *This = STREAM_THIS(iface);
248 FIXME("(%p)->(%ld %08lx %p)\n", This, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
249 return E_NOTIMPL;
250}
251
252static HRESULT WINAPI ProtocolStream_SetSize(IStream *iface, ULARGE_INTEGER libNewSize)
253{
254 ProtocolStream *This = STREAM_THIS(iface);
255 FIXME("(%p)->(%ld)\n", This, libNewSize.u.LowPart);
256 return E_NOTIMPL;
257}
258
259static HRESULT WINAPI ProtocolStream_CopyTo(IStream *iface, IStream *pstm,
260 ULARGE_INTEGER cb, ULARGE_INTEGER *pcbRead, ULARGE_INTEGER *pcbWritten)
261{
262 ProtocolStream *This = STREAM_THIS(iface);
263 FIXME("(%p)->(%p %ld %p %p)\n", This, pstm, cb.u.LowPart, pcbRead, pcbWritten);
264 return E_NOTIMPL;
265}
266
267static HRESULT WINAPI ProtocolStream_Commit(IStream *iface, DWORD grfCommitFlags)
268{
269 ProtocolStream *This = STREAM_THIS(iface);
270
271 TRACE("(%p)->(%08lx)\n", This, grfCommitFlags);
272
273 return E_NOTIMPL;
274}
275
276static HRESULT WINAPI ProtocolStream_Revert(IStream *iface)
277{
278 ProtocolStream *This = STREAM_THIS(iface);
279
280 TRACE("(%p)\n", This);
281
282 return E_NOTIMPL;
283}
284
285static HRESULT WINAPI ProtocolStream_LockRegion(IStream *iface, ULARGE_INTEGER libOffset,
286 ULARGE_INTEGER cb, DWORD dwLockType)
287{
288 ProtocolStream *This = STREAM_THIS(iface);
289 FIXME("(%p)->(%ld %ld %ld)\n", This, libOffset.u.LowPart, cb.u.LowPart, dwLockType);
290 return E_NOTIMPL;
291}
292
293static HRESULT WINAPI ProtocolStream_UnlockRegion(IStream *iface,
294 ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
295{
296 ProtocolStream *This = STREAM_THIS(iface);
297 FIXME("(%p)->(%ld %ld %ld)\n", This, libOffset.u.LowPart, cb.u.LowPart, dwLockType);
298 return E_NOTIMPL;
299}
300
301static HRESULT WINAPI ProtocolStream_Stat(IStream *iface, STATSTG *pstatstg,
302 DWORD dwStatFlag)
303{
304 ProtocolStream *This = STREAM_THIS(iface);
305 FIXME("(%p)->(%p %08lx)\n", This, pstatstg, dwStatFlag);
306 return E_NOTIMPL;
307}
308
309static HRESULT WINAPI ProtocolStream_Clone(IStream *iface, IStream **ppstm)
310{
311 ProtocolStream *This = STREAM_THIS(iface);
312 FIXME("(%p)->(%p)\n", This, ppstm);
313 return E_NOTIMPL;
314}
315
316#undef STREAM_THIS
317
318static const IStreamVtbl ProtocolStreamVtbl = {
319 ProtocolStream_QueryInterface,
320 ProtocolStream_AddRef,
321 ProtocolStream_Release,
322 ProtocolStream_Read,
323 ProtocolStream_Write,
324 ProtocolStream_Seek,
325 ProtocolStream_SetSize,
326 ProtocolStream_CopyTo,
327 ProtocolStream_Commit,
328 ProtocolStream_Revert,
329 ProtocolStream_LockRegion,
330 ProtocolStream_UnlockRegion,
331 ProtocolStream_Stat,
332 ProtocolStream_Clone
333};
334
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000335#define BINDING_THIS(iface) DEFINE_THIS(Binding, Binding, iface)
336
Jacek Caban762e2692006-01-09 17:18:06 +0100337static ProtocolStream *create_stream(IInternetProtocol *protocol)
338{
339 ProtocolStream *ret = HeapAlloc(GetProcessHeap(), 0, sizeof(ProtocolStream));
340
341 ret->lpStreamVtbl = &ProtocolStreamVtbl;
342 ret->ref = 1;
343 ret->buf_size = 0;
344
345 IInternetProtocol_AddRef(protocol);
346 ret->protocol = protocol;
347
348 return ret;
349}
350
351static void fill_stream_buffer(ProtocolStream *This)
352{
353 DWORD read = 0;
354
355 IInternetProtocol_Read(This->protocol, This->buf+This->buf_size,
356 sizeof(This->buf)-This->buf_size, &read);
357 This->buf_size += read;
358}
359
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000360static HRESULT WINAPI Binding_QueryInterface(IBinding *iface, REFIID riid, void **ppv)
361{
362 Binding *This = BINDING_THIS(iface);
363
364 *ppv = NULL;
365
366 if(IsEqualGUID(&IID_IUnknown, riid)) {
367 TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
368 *ppv = BINDING(This);
369 }else if(IsEqualGUID(&IID_IBinding, riid)) {
370 TRACE("(%p)->(IID_IBinding %p)\n", This, ppv);
371 *ppv = BINDING(This);
372 }else if(IsEqualGUID(&IID_IInternetProtocolSink, riid)) {
373 TRACE("(%p)->(IID_IInternetProtocolSink %p)\n", This, ppv);
374 *ppv = PROTSINK(This);
375 }else if(IsEqualGUID(&IID_IInternetBindInfo, riid)) {
376 TRACE("(%p)->(IID_IInternetBindInfo %p)\n", This, ppv);
377 *ppv = BINDINF(This);
378 }else if(IsEqualGUID(&IID_IServiceProvider, riid)) {
379 TRACE("(%p)->(IID_IServiceProvider %p)\n", This, ppv);
380 *ppv = SERVPROV(This);
381 }
382
383 if(*ppv)
384 return S_OK;
385
386 WARN("Unsupported interface %s\n", debugstr_guid(riid));
387 return E_NOINTERFACE;
388}
389
390static ULONG WINAPI Binding_AddRef(IBinding *iface)
391{
392 Binding *This = BINDING_THIS(iface);
393 LONG ref = InterlockedIncrement(&This->ref);
394
395 TRACE("(%p) ref=%ld\n", This, ref);
396
397 return ref;
398}
399
400static ULONG WINAPI Binding_Release(IBinding *iface)
401{
402 Binding *This = BINDING_THIS(iface);
403 LONG ref = InterlockedDecrement(&This->ref);
404
405 TRACE("(%p) ref=%ld\n", This, ref);
406
407 if(!ref) {
408 if(This->callback)
409 IBindStatusCallback_Release(This->callback);
410 if(This->protocol)
411 IInternetProtocol_Release(This->protocol);
Jacek Caban6744b9b2006-01-05 13:42:32 +0100412 if(This->service_provider)
413 IServiceProvider_Release(This->service_provider);
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000414 if(This->stream)
Jacek Caban762e2692006-01-09 17:18:06 +0100415 IStream_Release(STREAM(This->stream));
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000416
417 ReleaseBindInfo(&This->bindinfo);
418 HeapFree(GetProcessHeap(), 0, This->mime);
419 HeapFree(GetProcessHeap(), 0, This->url);
420
421 HeapFree(GetProcessHeap(), 0, This);
422 }
423
424 return ref;
425}
426
427static HRESULT WINAPI Binding_Abort(IBinding *iface)
428{
429 Binding *This = BINDING_THIS(iface);
430 FIXME("(%p)\n", This);
431 return E_NOTIMPL;
432}
433
434static HRESULT WINAPI Binding_Suspend(IBinding *iface)
435{
436 Binding *This = BINDING_THIS(iface);
437 FIXME("(%p)\n", This);
438 return E_NOTIMPL;
439}
440
441static HRESULT WINAPI Binding_Resume(IBinding *iface)
442{
443 Binding *This = BINDING_THIS(iface);
444 FIXME("(%p)\n", This);
445 return E_NOTIMPL;
446}
447
448static HRESULT WINAPI Binding_SetPriority(IBinding *iface, LONG nPriority)
449{
450 Binding *This = BINDING_THIS(iface);
451 FIXME("(%p)->(%ld)\n", This, nPriority);
452 return E_NOTIMPL;
453}
454
455static HRESULT WINAPI Binding_GetPriority(IBinding *iface, LONG *pnPriority)
456{
457 Binding *This = BINDING_THIS(iface);
458 FIXME("(%p)->(%p)\n", This, pnPriority);
459 return E_NOTIMPL;
460}
461
462static HRESULT WINAPI Binding_GetBindResult(IBinding *iface, CLSID *pclsidProtocol,
463 DWORD *pdwResult, LPOLESTR *pszResult, DWORD *pdwReserved)
464{
465 Binding *This = BINDING_THIS(iface);
466 FIXME("(%p)->(%p %p %p %p)\n", This, pclsidProtocol, pdwResult, pszResult, pdwReserved);
467 return E_NOTIMPL;
468}
469
470#undef BINDING_THIS
471
472static const IBindingVtbl BindingVtbl = {
473 Binding_QueryInterface,
474 Binding_AddRef,
475 Binding_Release,
476 Binding_Abort,
477 Binding_Suspend,
478 Binding_Resume,
479 Binding_SetPriority,
480 Binding_GetPriority,
481 Binding_GetBindResult
482};
483
484#define PROTSINK_THIS(iface) DEFINE_THIS(Binding, InternetProtocolSink, iface)
485
486static HRESULT WINAPI InternetProtocolSink_QueryInterface(IInternetProtocolSink *iface,
487 REFIID riid, void **ppv)
488{
489 Binding *This = PROTSINK_THIS(iface);
490 return IBinding_QueryInterface(BINDING(This), riid, ppv);
491}
492
493static ULONG WINAPI InternetProtocolSink_AddRef(IInternetProtocolSink *iface)
494{
495 Binding *This = PROTSINK_THIS(iface);
496 return IBinding_AddRef(BINDING(This));
497}
498
499static ULONG WINAPI InternetProtocolSink_Release(IInternetProtocolSink *iface)
500{
501 Binding *This = PROTSINK_THIS(iface);
502 return IBinding_Release(BINDING(This));
503}
504
505static HRESULT WINAPI InternetProtocolSink_Switch(IInternetProtocolSink *iface,
506 PROTOCOLDATA *pProtocolData)
507{
508 Binding *This = PROTSINK_THIS(iface);
509 FIXME("(%p)->(%p)\n", This, pProtocolData);
510 return E_NOTIMPL;
511}
512
513static HRESULT WINAPI InternetProtocolSink_ReportProgress(IInternetProtocolSink *iface,
514 ULONG ulStatusCode, LPCWSTR szStatusText)
515{
516 Binding *This = PROTSINK_THIS(iface);
517
518 TRACE("(%p)->(%lu %s)\n", This, ulStatusCode, debugstr_w(szStatusText));
519
520 switch(ulStatusCode) {
521 case BINDSTATUS_MIMETYPEAVAILABLE: {
522 int len = strlenW(szStatusText)+1;
523 This->mime = HeapAlloc(GetProcessHeap(), 0, len*sizeof(WCHAR));
524 memcpy(This->mime, szStatusText, len*sizeof(WCHAR));
525 break;
526 }
Jacek Caban0cceb332005-12-26 13:27:10 +0100527 case BINDSTATUS_SENDINGREQUEST:
528 IBindStatusCallback_OnProgress(This->callback, 0, 0, BINDSTATUS_SENDINGREQUEST,
529 szStatusText);
530 break;
531 case BINDSTATUS_VERIFIEDMIMETYPEAVAILABLE:
532 IBindStatusCallback_OnProgress(This->callback, 0, 0,
533 BINDSTATUS_MIMETYPEAVAILABLE, szStatusText);
534 break;
535 case BINDSTATUS_CACHEFILENAMEAVAILABLE:
536 break;
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000537 default:
538 FIXME("Unhandled status code %ld\n", ulStatusCode);
539 return E_NOTIMPL;
540 };
541
542 return S_OK;
543}
544
545static HRESULT WINAPI InternetProtocolSink_ReportData(IInternetProtocolSink *iface,
546 DWORD grfBSCF, ULONG ulProgress, ULONG ulProgressMax)
547{
548 Binding *This = PROTSINK_THIS(iface);
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000549 STGMEDIUM stgmed;
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000550
551 TRACE("(%p)->(%ld %lu %lu)\n", This, grfBSCF, ulProgress, ulProgressMax);
552
553 if(grfBSCF & BSCF_FIRSTDATANOTIFICATION) {
554 if(This->mime)
555 IBindStatusCallback_OnProgress(This->callback, ulProgress, ulProgressMax,
556 BINDSTATUS_MIMETYPEAVAILABLE, This->mime);
557 IBindStatusCallback_OnProgress(This->callback, ulProgress, ulProgressMax,
558 BINDSTATUS_BEGINDOWNLOADDATA, This->url);
559 }
560
561 if(grfBSCF & BSCF_LASTDATANOTIFICATION)
562 IBindStatusCallback_OnProgress(This->callback, ulProgress, ulProgressMax,
563 BINDSTATUS_ENDDOWNLOADDATA, This->url);
564
565 if(grfBSCF & BSCF_FIRSTDATANOTIFICATION)
566 IInternetProtocol_LockRequest(This->protocol, 0);
Jacek Caban762e2692006-01-09 17:18:06 +0100567
568 fill_stream_buffer(This->stream);
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000569
570 stgmed.tymed = TYMED_ISTREAM;
Jacek Caban762e2692006-01-09 17:18:06 +0100571 stgmed.u.pstm = STREAM(This->stream);
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000572
Jacek Caban762e2692006-01-09 17:18:06 +0100573 IBindStatusCallback_OnDataAvailable(This->callback, grfBSCF, This->stream->buf_size,
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000574 NULL /* FIXME */, &stgmed);
575
576 if(grfBSCF & BSCF_LASTDATANOTIFICATION)
577 IBindStatusCallback_OnStopBinding(This->callback, S_OK, NULL);
578
579 return S_OK;
580}
581
582static HRESULT WINAPI InternetProtocolSink_ReportResult(IInternetProtocolSink *iface,
583 HRESULT hrResult, DWORD dwError, LPCWSTR szResult)
584{
585 Binding *This = PROTSINK_THIS(iface);
586 FIXME("(%p)->(%08lx %ld %s)\n", This, hrResult, dwError, debugstr_w(szResult));
587 return E_NOTIMPL;
588}
589
590#undef PROTSINK_THIS
591
592static const IInternetProtocolSinkVtbl InternetProtocolSinkVtbl = {
593 InternetProtocolSink_QueryInterface,
594 InternetProtocolSink_AddRef,
595 InternetProtocolSink_Release,
596 InternetProtocolSink_Switch,
597 InternetProtocolSink_ReportProgress,
598 InternetProtocolSink_ReportData,
599 InternetProtocolSink_ReportResult
600};
601
602#define BINDINF_THIS(iface) DEFINE_THIS(Binding, InternetBindInfo, iface)
603
604static HRESULT WINAPI InternetBindInfo_QueryInterface(IInternetBindInfo *iface,
605 REFIID riid, void **ppv)
606{
607 Binding *This = BINDINF_THIS(iface);
608 return IBinding_QueryInterface(BINDING(This), riid, ppv);
609}
610
611static ULONG WINAPI InternetBindInfo_AddRef(IInternetBindInfo *iface)
612{
613 Binding *This = BINDINF_THIS(iface);
614 return IBinding_AddRef(BINDING(This));
615}
616
617static ULONG WINAPI InternetBindInfo_Release(IInternetBindInfo *iface)
618{
619 Binding *This = BINDINF_THIS(iface);
620 return IBinding_Release(BINDING(This));
621}
622
623static HRESULT WINAPI InternetBindInfo_GetBindInfo(IInternetBindInfo *iface,
624 DWORD *grfBINDF, BINDINFO *pbindinfo)
625{
626 Binding *This = BINDINF_THIS(iface);
627
628 TRACE("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo);
629
630 *grfBINDF = This->bindf;
631
632 memcpy(pbindinfo, &This->bindinfo, sizeof(BINDINFO));
633
634 if(pbindinfo->szExtraInfo || pbindinfo->szCustomVerb)
635 FIXME("copy strings\n");
636
637 if(pbindinfo->pUnk)
638 IUnknown_AddRef(pbindinfo->pUnk);
639
640 return S_OK;
641}
642
643static HRESULT WINAPI InternetBindInfo_GetBindString(IInternetBindInfo *iface,
644 ULONG ulStringType, LPOLESTR *ppwzStr, ULONG cEl, ULONG *pcElFetched)
645{
646 Binding *This = BINDINF_THIS(iface);
Jacek Cabanc4489682006-01-05 12:17:59 +0100647
648 TRACE("(%p)->(%ld %p %ld %p)\n", This, ulStringType, ppwzStr, cEl, pcElFetched);
649
650 switch(ulStringType) {
651 case BINDSTRING_ACCEPT_MIMES: {
652 static const WCHAR wszMimes[] = {'*','/','*',0};
653
654 if(!ppwzStr || !pcElFetched)
655 return E_INVALIDARG;
656
657 ppwzStr[0] = CoTaskMemAlloc(sizeof(wszMimes));
658 memcpy(ppwzStr[0], wszMimes, sizeof(wszMimes));
659 *pcElFetched = 1;
660 return S_OK;
661 }
662 case BINDSTRING_USER_AGENT: {
663 IInternetBindInfo *bindinfo = NULL;
664 HRESULT hres;
665
666 hres = IBindStatusCallback_QueryInterface(This->callback, &IID_IInternetBindInfo,
667 (void**)&bindinfo);
668 if(FAILED(hres))
669 return hres;
670
671 hres = IInternetBindInfo_GetBindString(bindinfo, ulStringType, ppwzStr,
672 cEl, pcElFetched);
673 IInternetBindInfo_Release(bindinfo);
674
675 return hres;
676 }
677 }
678
679 FIXME("not supported string type %ld\n", ulStringType);
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000680 return E_NOTIMPL;
681}
682
683#undef BINDF_THIS
684
685static const IInternetBindInfoVtbl InternetBindInfoVtbl = {
686 InternetBindInfo_QueryInterface,
687 InternetBindInfo_AddRef,
688 InternetBindInfo_Release,
689 InternetBindInfo_GetBindInfo,
690 InternetBindInfo_GetBindString
691};
692
693#define SERVPROV_THIS(iface) DEFINE_THIS(Binding, ServiceProvider, iface)
694
695static HRESULT WINAPI ServiceProvider_QueryInterface(IServiceProvider *iface,
696 REFIID riid, void **ppv)
697{
698 Binding *This = SERVPROV_THIS(iface);
699 return IBinding_QueryInterface(BINDING(This), riid, ppv);
700}
701
702static ULONG WINAPI ServiceProvider_AddRef(IServiceProvider *iface)
703{
704 Binding *This = SERVPROV_THIS(iface);
705 return IBinding_AddRef(BINDING(This));
706}
707
708static ULONG WINAPI ServiceProvider_Release(IServiceProvider *iface)
709{
710 Binding *This = SERVPROV_THIS(iface);
711 return IBinding_Release(BINDING(This));
712}
713
714static HRESULT WINAPI ServiceProvider_QueryService(IServiceProvider *iface,
715 REFGUID guidService, REFIID riid, void **ppv)
716{
717 Binding *This = SERVPROV_THIS(iface);
Jacek Cabanb235d792006-01-05 13:42:43 +0100718 HRESULT hres;
719
720 TRACE("(%p)->(%s %s %p)\n", This, debugstr_guid(guidService), debugstr_guid(riid), ppv);
721
722 if(This->service_provider) {
723 hres = IServiceProvider_QueryService(This->service_provider, guidService,
724 riid, ppv);
725 if(SUCCEEDED(hres))
726 return hres;
727 }
728
729 if(IsEqualGUID(&IID_IHttpNegotiate, guidService)
730 || IsEqualGUID(&IID_IHttpNegotiate2, guidService))
731 return IHttpNegotiate2_QueryInterface(&HttpNegotiate, riid, ppv);
732
733 WARN("unknown service %s\n", debugstr_guid(guidService));
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000734 return E_NOTIMPL;
735}
736
737#undef SERVPROV_THIS
738
739static const IServiceProviderVtbl ServiceProviderVtbl = {
740 ServiceProvider_QueryInterface,
741 ServiceProvider_AddRef,
742 ServiceProvider_Release,
743 ServiceProvider_QueryService
744};
745
746static HRESULT get_callback(IBindCtx *pbc, IBindStatusCallback **callback)
747{
748 HRESULT hres;
749
750 static WCHAR wszBSCBHolder[] = { '_','B','S','C','B','_','H','o','l','d','e','r','_',0 };
751
752 hres = IBindCtx_GetObjectParam(pbc, wszBSCBHolder, (IUnknown**)callback);
753 if(FAILED(hres))
Jacek Caban90c70f72005-12-16 12:40:39 +0100754 return MK_E_SYNTAX;
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000755
756 return S_OK;
757}
758
759static HRESULT get_protocol(Binding *This, LPCWSTR url)
760{
761 IUnknown *unk = NULL;
762 IClassFactory *cf = NULL;
763 HRESULT hres;
764
765 hres = IBindStatusCallback_QueryInterface(This->callback, &IID_IInternetProtocol,
766 (void**)&This->protocol);
767 if(SUCCEEDED(hres))
768 return S_OK;
769
Jacek Caban6744b9b2006-01-05 13:42:32 +0100770 if(This->service_provider) {
771 hres = IServiceProvider_QueryService(This->service_provider, &IID_IInternetProtocol,
772 &IID_IInternetProtocol, (void**)&This->protocol);
773 if(SUCCEEDED(hres))
774 return S_OK;
775 }
776
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000777 hres = get_protocol_iface(url, &unk);
778 if(FAILED(hres))
779 return hres;
780
781 hres = IUnknown_QueryInterface(unk, &IID_IClassFactory, (void**)&cf);
782 IUnknown_Release(unk);
783 if(FAILED(hres))
784 return hres;
785
786 hres = IClassFactory_CreateInstance(cf, NULL, &IID_IInternetProtocol, (void**)&This->protocol);
787 IClassFactory_Release(cf);
788
789 return hres;
790}
791
792static HRESULT Binding_Create(LPCWSTR url, IBindCtx *pbc, REFIID riid, Binding **binding)
793{
794 Binding *ret;
795 int len;
796 HRESULT hres;
797
Jacek Caban0cceb332005-12-26 13:27:10 +0100798 static const WCHAR wszFile[] = {'f','i','l','e',':'};
799
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000800 if(!IsEqualGUID(&IID_IStream, riid)) {
801 FIXME("Unsupported riid %s\n", debugstr_guid(riid));
802 return E_NOTIMPL;
803 }
804
805 ret = HeapAlloc(GetProcessHeap(), 0, sizeof(Binding));
806
807 ret->lpBindingVtbl = &BindingVtbl;
808 ret->lpInternetProtocolSinkVtbl = &InternetProtocolSinkVtbl;
809 ret->lpInternetBindInfoVtbl = &InternetBindInfoVtbl;
810 ret->lpServiceProviderVtbl = &ServiceProviderVtbl;
811
812 ret->ref = 1;
813
814 ret->callback = NULL;
815 ret->protocol = NULL;
Jacek Caban6744b9b2006-01-05 13:42:32 +0100816 ret->service_provider = NULL;
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000817 ret->stream = NULL;
818 ret->mime = NULL;
819 ret->url = NULL;
820
821 memset(&ret->bindinfo, 0, sizeof(BINDINFO));
822 ret->bindinfo.cbSize = sizeof(BINDINFO);
823 ret->bindf = 0;
824
825 hres = get_callback(pbc, &ret->callback);
826 if(FAILED(hres)) {
827 WARN("Could not get IBindStatusCallback\n");
828 IBinding_Release(BINDING(ret));
829 return hres;
830 }
831
Jacek Caban6744b9b2006-01-05 13:42:32 +0100832 IBindStatusCallback_QueryInterface(ret->callback, &IID_IServiceProvider,
833 (void**)&ret->service_provider);
834
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000835 hres = get_protocol(ret, url);
836 if(FAILED(hres)) {
837 WARN("Could not get protocol handler\n");
838 IBinding_Release(BINDING(ret));
839 return hres;
840 }
841
842 hres = IBindStatusCallback_GetBindInfo(ret->callback, &ret->bindf, &ret->bindinfo);
843 if(FAILED(hres)) {
844 WARN("GetBindInfo failed: %08lx\n", hres);
845 IBinding_Release(BINDING(ret));
846 return hres;
847 }
848
Jacek Caban0cceb332005-12-26 13:27:10 +0100849 ret->bindf |= BINDF_FROMURLMON;
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000850
851 len = strlenW(url)+1;
Jacek Caban0cceb332005-12-26 13:27:10 +0100852
853 if(len < sizeof(wszFile)/sizeof(WCHAR) || memcmp(wszFile, url, sizeof(wszFile)))
854 ret->bindf |= BINDF_NEEDFILE;
855
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000856 ret->url = HeapAlloc(GetProcessHeap(), 0, len*sizeof(WCHAR));
857 memcpy(ret->url, url, len*sizeof(WCHAR));
858
Jacek Caban762e2692006-01-09 17:18:06 +0100859 ret->stream = create_stream(ret->protocol);
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000860
861 *binding = ret;
862 return S_OK;
863}
864
865HRESULT start_binding(LPCWSTR url, IBindCtx *pbc, REFIID riid, void **ppv)
866{
867 Binding *binding = NULL;
868 HRESULT hres;
869
870 *ppv = NULL;
871
872 hres = Binding_Create(url, pbc, riid, &binding);
873 if(FAILED(hres))
874 return hres;
875
876 hres = IBindStatusCallback_OnStartBinding(binding->callback, 0, BINDING(binding));
877 if(FAILED(hres)) {
878 WARN("OnStartBinding failed: %08lx\n", hres);
879 IBindStatusCallback_OnStopBinding(binding->callback, 0x800c0008, NULL);
880 IBinding_Release(BINDING(binding));
881 return hres;
882 }
883
884 hres = IInternetProtocol_Start(binding->protocol, url, PROTSINK(binding),
885 BINDINF(binding), 0, 0);
886 IInternetProtocol_Terminate(binding->protocol, 0);
887
888 if(SUCCEEDED(hres)) {
889 IInternetProtocol_UnlockRequest(binding->protocol);
890 }else {
891 WARN("Start failed: %08lx\n", hres);
892 IBindStatusCallback_OnStopBinding(binding->callback, S_OK, NULL);
893 }
894
Jacek Caban762e2692006-01-09 17:18:06 +0100895 IStream_AddRef(STREAM(binding->stream));
Jacek Caban0dcf2a42005-11-14 15:12:43 +0000896 *ppv = binding->stream;
897
898 IBinding_Release(BINDING(binding));
899
900 return hres;
901}