* Sync to trunk HEAD (r53298).
[reactos.git] / dll / win32 / urlmon / download.c
1 /*
2 * Copyright 2008 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 #include "urlmon_main.h"
20 #include "wine/debug.h"
21
22 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
23
24 typedef struct {
25 IBindStatusCallback IBindStatusCallback_iface;
26 IServiceProvider IServiceProvider_iface;
27
28 LONG ref;
29
30 IBindStatusCallback *callback;
31 IBinding *binding;
32 LPWSTR file_name;
33 LPWSTR cache_file;
34 } DownloadBSC;
35
36 static inline DownloadBSC *impl_from_IBindStatusCallback(IBindStatusCallback *iface)
37 {
38 return CONTAINING_RECORD(iface, DownloadBSC, IBindStatusCallback_iface);
39 }
40
41 static inline DownloadBSC *impl_from_IServiceProvider(IServiceProvider *iface)
42 {
43 return CONTAINING_RECORD(iface, DownloadBSC, IServiceProvider_iface);
44 }
45
46 static HRESULT WINAPI DownloadBSC_QueryInterface(IBindStatusCallback *iface,
47 REFIID riid, void **ppv)
48 {
49 DownloadBSC *This = impl_from_IBindStatusCallback(iface);
50
51 *ppv = NULL;
52
53 if(IsEqualGUID(&IID_IUnknown, riid)) {
54 TRACE("(%p)->(IID_IUnknown, %p)\n", This, ppv);
55 *ppv = &This->IBindStatusCallback_iface;
56 }else if(IsEqualGUID(&IID_IBindStatusCallback, riid)) {
57 TRACE("(%p)->(IID_IBindStatusCallback, %p)\n", This, ppv);
58 *ppv = &This->IBindStatusCallback_iface;
59 }else if(IsEqualGUID(&IID_IServiceProvider, riid)) {
60 TRACE("(%p)->(IID_IServiceProvider, %p)\n", This, ppv);
61 *ppv = &This->IServiceProvider_iface;
62 }
63
64 if(*ppv) {
65 IBindStatusCallback_AddRef((IUnknown*)*ppv);
66 return S_OK;
67 }
68
69 TRACE("Unsupported riid = %s\n", debugstr_guid(riid));
70 return E_NOINTERFACE;
71 }
72
73 static ULONG WINAPI DownloadBSC_AddRef(IBindStatusCallback *iface)
74 {
75 DownloadBSC *This = impl_from_IBindStatusCallback(iface);
76 LONG ref = InterlockedIncrement(&This->ref);
77
78 TRACE("(%p) ref = %d\n", This, ref);
79
80 return ref;
81 }
82
83 static ULONG WINAPI DownloadBSC_Release(IBindStatusCallback *iface)
84 {
85 DownloadBSC *This = impl_from_IBindStatusCallback(iface);
86 LONG ref = InterlockedDecrement(&This->ref);
87
88 TRACE("(%p) ref = %d\n", This, ref);
89
90 if(!ref) {
91 if(This->callback)
92 IBindStatusCallback_Release(This->callback);
93 if(This->binding)
94 IBinding_Release(This->binding);
95 heap_free(This->file_name);
96 heap_free(This->cache_file);
97 heap_free(This);
98 }
99
100 return ref;
101 }
102
103 static HRESULT WINAPI DownloadBSC_OnStartBinding(IBindStatusCallback *iface,
104 DWORD dwReserved, IBinding *pbind)
105 {
106 DownloadBSC *This = impl_from_IBindStatusCallback(iface);
107 HRESULT hres = S_OK;
108
109 TRACE("(%p)->(%d %p)\n", This, dwReserved, pbind);
110
111 if(This->callback) {
112 hres = IBindStatusCallback_OnStartBinding(This->callback, dwReserved, pbind);
113
114 IBinding_AddRef(pbind);
115 This->binding = pbind;
116 }
117
118 /* Windows seems to ignore E_NOTIMPL if it's returned from the client. */
119 return hres == E_NOTIMPL ? S_OK : hres;
120 }
121
122 static HRESULT WINAPI DownloadBSC_GetPriority(IBindStatusCallback *iface, LONG *pnPriority)
123 {
124 DownloadBSC *This = impl_from_IBindStatusCallback(iface);
125 FIXME("(%p)->(%p)\n", This, pnPriority);
126 return E_NOTIMPL;
127 }
128
129 static HRESULT WINAPI DownloadBSC_OnLowResource(IBindStatusCallback *iface, DWORD reserved)
130 {
131 DownloadBSC *This = impl_from_IBindStatusCallback(iface);
132 FIXME("(%p)->(%d)\n", This, reserved);
133 return E_NOTIMPL;
134 }
135
136 static HRESULT on_progress(DownloadBSC *This, ULONG progress, ULONG progress_max, ULONG status_code, LPCWSTR status_text)
137 {
138 HRESULT hres;
139
140 if(!This->callback)
141 return S_OK;
142
143 hres = IBindStatusCallback_OnProgress(This->callback, progress, progress_max, status_code, status_text);
144 if(hres == E_ABORT) {
145 if(This->binding)
146 IBinding_Abort(This->binding);
147 else
148 FIXME("No binding, not sure what to do!\n");
149 }
150
151 return hres;
152 }
153
154 static HRESULT WINAPI DownloadBSC_OnProgress(IBindStatusCallback *iface, ULONG ulProgress,
155 ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText)
156 {
157 DownloadBSC *This = impl_from_IBindStatusCallback(iface);
158 HRESULT hres = S_OK;
159
160 TRACE("%p)->(%u %u %u %s)\n", This, ulProgress, ulProgressMax, ulStatusCode,
161 debugstr_w(szStatusText));
162
163 switch(ulStatusCode) {
164 case BINDSTATUS_CONNECTING:
165 case BINDSTATUS_BEGINDOWNLOADDATA:
166 case BINDSTATUS_DOWNLOADINGDATA:
167 case BINDSTATUS_ENDDOWNLOADDATA:
168 case BINDSTATUS_SENDINGREQUEST:
169 case BINDSTATUS_MIMETYPEAVAILABLE:
170 hres = on_progress(This, ulProgress, ulProgressMax, ulStatusCode, szStatusText);
171 break;
172
173 case BINDSTATUS_CACHEFILENAMEAVAILABLE:
174 hres = on_progress(This, ulProgress, ulProgressMax, ulStatusCode, szStatusText);
175 This->cache_file = heap_strdupW(szStatusText);
176 break;
177
178 case BINDSTATUS_FINDINGRESOURCE: /* FIXME */
179 break;
180
181 default:
182 FIXME("Unsupported status %u\n", ulStatusCode);
183 }
184
185 return hres;
186 }
187
188 static HRESULT WINAPI DownloadBSC_OnStopBinding(IBindStatusCallback *iface,
189 HRESULT hresult, LPCWSTR szError)
190 {
191 DownloadBSC *This = impl_from_IBindStatusCallback(iface);
192
193 TRACE("(%p)->(%08x %s)\n", This, hresult, debugstr_w(szError));
194
195 if(This->file_name) {
196 if(This->cache_file) {
197 BOOL b;
198
199 b = CopyFileW(This->cache_file, This->file_name, FALSE);
200 if(!b)
201 FIXME("CopyFile failed: %u\n", GetLastError());
202 }else {
203 FIXME("No cache file\n");
204 }
205 }
206
207 if(This->callback)
208 IBindStatusCallback_OnStopBinding(This->callback, hresult, szError);
209
210 if(This->binding) {
211 IBinding_Release(This->binding);
212 This->binding = NULL;
213 }
214
215 return S_OK;
216 }
217
218 static HRESULT WINAPI DownloadBSC_GetBindInfo(IBindStatusCallback *iface,
219 DWORD *grfBINDF, BINDINFO *pbindinfo)
220 {
221 DownloadBSC *This = impl_from_IBindStatusCallback(iface);
222 DWORD bindf = 0;
223
224 TRACE("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo);
225
226 if(This->callback) {
227 BINDINFO bindinfo;
228 HRESULT hres;
229
230 memset(&bindinfo, 0, sizeof(bindinfo));
231 bindinfo.cbSize = sizeof(bindinfo);
232
233 hres = IBindStatusCallback_GetBindInfo(This->callback, &bindf, &bindinfo);
234 if(SUCCEEDED(hres))
235 ReleaseBindInfo(&bindinfo);
236 }
237
238 *grfBINDF = BINDF_PULLDATA | BINDF_NEEDFILE | (bindf & BINDF_ENFORCERESTRICTED);
239 return S_OK;
240 }
241
242 static HRESULT WINAPI DownloadBSC_OnDataAvailable(IBindStatusCallback *iface,
243 DWORD grfBSCF, DWORD dwSize, FORMATETC *pformatetc, STGMEDIUM *pstgmed)
244 {
245 DownloadBSC *This = impl_from_IBindStatusCallback(iface);
246
247 TRACE("(%p)->(%08x %d %p %p)\n", This, grfBSCF, dwSize, pformatetc, pstgmed);
248
249 return S_OK;
250 }
251
252 static HRESULT WINAPI DownloadBSC_OnObjectAvailable(IBindStatusCallback *iface,
253 REFIID riid, IUnknown *punk)
254 {
255 DownloadBSC *This = impl_from_IBindStatusCallback(iface);
256 FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), punk);
257 return E_NOTIMPL;
258 }
259
260 static const IBindStatusCallbackVtbl BindStatusCallbackVtbl = {
261 DownloadBSC_QueryInterface,
262 DownloadBSC_AddRef,
263 DownloadBSC_Release,
264 DownloadBSC_OnStartBinding,
265 DownloadBSC_GetPriority,
266 DownloadBSC_OnLowResource,
267 DownloadBSC_OnProgress,
268 DownloadBSC_OnStopBinding,
269 DownloadBSC_GetBindInfo,
270 DownloadBSC_OnDataAvailable,
271 DownloadBSC_OnObjectAvailable
272 };
273
274 static HRESULT WINAPI DwlServiceProvider_QueryInterface(IServiceProvider *iface,
275 REFIID riid, void **ppv)
276 {
277 DownloadBSC *This = impl_from_IServiceProvider(iface);
278 return IBindStatusCallback_QueryInterface(&This->IBindStatusCallback_iface, riid, ppv);
279 }
280
281 static ULONG WINAPI DwlServiceProvider_AddRef(IServiceProvider *iface)
282 {
283 DownloadBSC *This = impl_from_IServiceProvider(iface);
284 return IBindStatusCallback_AddRef(&This->IBindStatusCallback_iface);
285 }
286
287 static ULONG WINAPI DwlServiceProvider_Release(IServiceProvider *iface)
288 {
289 DownloadBSC *This = impl_from_IServiceProvider(iface);
290 return IBindStatusCallback_Release(&This->IBindStatusCallback_iface);
291 }
292
293 static HRESULT WINAPI DwlServiceProvider_QueryService(IServiceProvider *iface,
294 REFGUID guidService, REFIID riid, void **ppv)
295 {
296 DownloadBSC *This = impl_from_IServiceProvider(iface);
297 IServiceProvider *serv_prov;
298 HRESULT hres;
299
300 TRACE("(%p)->(%s %s %p)\n", This, debugstr_guid(guidService), debugstr_guid(riid), ppv);
301
302 if(!This->callback)
303 return E_NOINTERFACE;
304
305 hres = IBindStatusCallback_QueryInterface(This->callback, riid, ppv);
306 if(SUCCEEDED(hres))
307 return S_OK;
308
309 hres = IBindStatusCallback_QueryInterface(This->callback, &IID_IServiceProvider, (void**)&serv_prov);
310 if(SUCCEEDED(hres)) {
311 hres = IServiceProvider_QueryService(serv_prov, guidService, riid, ppv);
312 IServiceProvider_Release(serv_prov);
313 return hres;
314 }
315
316 return E_NOINTERFACE;
317 }
318
319 static const IServiceProviderVtbl ServiceProviderVtbl = {
320 DwlServiceProvider_QueryInterface,
321 DwlServiceProvider_AddRef,
322 DwlServiceProvider_Release,
323 DwlServiceProvider_QueryService
324 };
325
326 static HRESULT DownloadBSC_Create(IBindStatusCallback *callback, LPCWSTR file_name, IBindStatusCallback **ret_callback)
327 {
328 DownloadBSC *ret = heap_alloc(sizeof(*ret));
329
330 ret->IBindStatusCallback_iface.lpVtbl = &BindStatusCallbackVtbl;
331 ret->IServiceProvider_iface.lpVtbl = &ServiceProviderVtbl;
332 ret->ref = 1;
333 ret->file_name = heap_strdupW(file_name);
334 ret->cache_file = NULL;
335 ret->binding = NULL;
336
337 if(callback)
338 IBindStatusCallback_AddRef(callback);
339 ret->callback = callback;
340
341 *ret_callback = &ret->IBindStatusCallback_iface;
342 return S_OK;
343 }
344
345 HRESULT create_default_callback(IBindStatusCallback **ret)
346 {
347 IBindStatusCallback *callback;
348 HRESULT hres;
349
350 hres = DownloadBSC_Create(NULL, NULL, &callback);
351 if(FAILED(hres))
352 return hres;
353
354 hres = wrap_callback(callback, ret);
355 IBindStatusCallback_Release(callback);
356 return hres;
357 }
358
359 /***********************************************************************
360 * URLDownloadToFileW (URLMON.@)
361 *
362 * Downloads URL szURL to file szFileName and call lpfnCB callback to
363 * report progress.
364 *
365 * PARAMS
366 * pCaller [I] controlling IUnknown interface.
367 * szURL [I] URL of the file to download
368 * szFileName [I] file name to store the content of the URL
369 * dwReserved [I] reserved - set to 0
370 * lpfnCB [I] callback for progress report
371 *
372 * RETURNS
373 * S_OK on success
374 */
375 HRESULT WINAPI URLDownloadToFileW(LPUNKNOWN pCaller, LPCWSTR szURL, LPCWSTR szFileName,
376 DWORD dwReserved, LPBINDSTATUSCALLBACK lpfnCB)
377 {
378 IBindStatusCallback *callback;
379 IUnknown *unk;
380 IMoniker *mon;
381 IBindCtx *bindctx;
382 HRESULT hres;
383
384 TRACE("(%p %s %s %d %p)\n", pCaller, debugstr_w(szURL), debugstr_w(szFileName), dwReserved, lpfnCB);
385
386 if(pCaller)
387 FIXME("pCaller not supported\n");
388
389 hres = DownloadBSC_Create(lpfnCB, szFileName, &callback);
390 if(FAILED(hres))
391 return hres;
392
393 hres = CreateAsyncBindCtx(0, callback, NULL, &bindctx);
394 IBindStatusCallback_Release(callback);
395 if(FAILED(hres))
396 return hres;
397
398 hres = CreateURLMoniker(NULL, szURL, &mon);
399 if(FAILED(hres)) {
400 IBindCtx_Release(bindctx);
401 return hres;
402 }
403
404 hres = IMoniker_BindToStorage(mon, bindctx, NULL, &IID_IUnknown, (void**)&unk);
405 IMoniker_Release(mon);
406 IBindCtx_Release(bindctx);
407
408 if(unk)
409 IUnknown_Release(unk);
410
411 return hres == MK_S_ASYNCHRONOUS ? S_OK : hres;
412 }
413
414 /***********************************************************************
415 * URLDownloadToFileA (URLMON.@)
416 *
417 * Downloads URL szURL to rile szFileName and call lpfnCB callback to
418 * report progress.
419 *
420 * PARAMS
421 * pCaller [I] controlling IUnknown interface.
422 * szURL [I] URL of the file to download
423 * szFileName [I] file name to store the content of the URL
424 * dwReserved [I] reserved - set to 0
425 * lpfnCB [I] callback for progress report
426 *
427 * RETURNS
428 * S_OK on success
429 */
430 HRESULT WINAPI URLDownloadToFileA(LPUNKNOWN pCaller, LPCSTR szURL, LPCSTR szFileName, DWORD dwReserved,
431 LPBINDSTATUSCALLBACK lpfnCB)
432 {
433 LPWSTR urlW, file_nameW;
434 HRESULT hres;
435
436 TRACE("(%p %s %s %d %p)\n", pCaller, debugstr_a(szURL), debugstr_a(szFileName), dwReserved, lpfnCB);
437
438 urlW = heap_strdupAtoW(szURL);
439 file_nameW = heap_strdupAtoW(szFileName);
440
441 hres = URLDownloadToFileW(pCaller, urlW, file_nameW, dwReserved, lpfnCB);
442
443 heap_free(urlW);
444 heap_free(file_nameW);
445
446 return hres;
447 }