ded010144de973f80325acb1824bcd7108dddaa9
[reactos.git] / rosapps / applications / downloader / download.c
1 /* PROJECT: ReactOS Downloader (was GetFirefox)
2 * LICENSE: GPL - See COPYING in the top level directory
3 * FILE: base/applications/downloader/download.c
4 * PURPOSE: Displaying a download dialog
5 * COPYRIGHT: Copyright 2001 John R. Sheets (for CodeWeavers)
6 * Copyright 2004 Mike McCormack (for CodeWeavers)
7 * Copyright 2005 Ge van Geldorp (gvg@reactos.org)
8 * Copyright 2007 Dmitry Chapyshev (lentind@yandex.ru)
9 */
10 /*
11 * Based on Wine dlls/shdocvw/shdocvw_main.c
12 *
13 * This library is free software; you can redistribute it and/or
14 * modify it under the terms of the GNU Lesser General Public
15 * License as published by the Free Software Foundation; either
16 * version 2.1 of the License, or (at your option) any later version.
17 *
18 * This library is distributed in the hope that it will be useful,
19 * but WITHOUT ANY WARRANTY; without even the implied warranty of
20 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
21 * Lesser General Public License for more details.
22 *
23 * You should have received a copy of the GNU Lesser General Public
24 * License along with this library; if not, write to the Free Software
25 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
26 */
27
28 #define COBJMACROS
29 #define WIN32_NO_STATUS
30 #include <windows.h>
31 #include <commctrl.h>
32 #include <shlobj.h>
33 #include <shlwapi.h>
34 #include <urlmon.h>
35
36 #include "resources.h"
37 #include "structures.h"
38
39 #define NDEBUG
40 #include <debug.h>
41
42 extern struct Application* SelectedApplication;
43 extern WCHAR Strings [STRING_COUNT][MAX_STRING_LENGHT];
44
45 typedef struct _IBindStatusCallbackImpl
46 {
47 const IBindStatusCallbackVtbl *vtbl;
48 LONG ref;
49 HWND hDialog;
50 BOOL *pbCancelled;
51 } IBindStatusCallbackImpl;
52
53 static HRESULT WINAPI
54 dlQueryInterface(IBindStatusCallback* This, REFIID riid, void** ppvObject)
55 {
56 if (NULL == ppvObject)
57 {
58 return E_POINTER;
59 }
60
61 if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IBindStatusCallback))
62 {
63 IBindStatusCallback_AddRef( This );
64 *ppvObject = This;
65 return S_OK;
66 }
67
68 return E_NOINTERFACE;
69 }
70
71 static ULONG WINAPI
72 dlAddRef(IBindStatusCallback* iface)
73 {
74 IBindStatusCallbackImpl *This = (IBindStatusCallbackImpl *) iface;
75
76 return InterlockedIncrement(&This->ref);
77 }
78
79 static ULONG WINAPI
80 dlRelease(IBindStatusCallback* iface)
81 {
82 IBindStatusCallbackImpl *This = (IBindStatusCallbackImpl *) iface;
83 DWORD ref = InterlockedDecrement(&This->ref);
84
85 if( !ref )
86 {
87 DestroyWindow( This->hDialog );
88 HeapFree(GetProcessHeap(), 0, This);
89 }
90
91 return ref;
92 }
93
94 static HRESULT WINAPI
95 dlOnStartBinding(IBindStatusCallback* iface, DWORD dwReserved, IBinding* pib)
96 {
97 DPRINT1("OnStartBinding not implemented\n");
98
99 return S_OK;
100 }
101
102 static HRESULT WINAPI
103 dlGetPriority(IBindStatusCallback* iface, LONG* pnPriority)
104 {
105 DPRINT1("GetPriority not implemented\n");
106
107 return S_OK;
108 }
109
110 static HRESULT WINAPI
111 dlOnLowResource( IBindStatusCallback* iface, DWORD reserved)
112 {
113 DPRINT1("OnLowResource not implemented\n");
114
115 return S_OK;
116 }
117
118 static HRESULT WINAPI
119 dlOnProgress(IBindStatusCallback* iface, ULONG ulProgress,
120 ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText)
121 {
122 IBindStatusCallbackImpl *This = (IBindStatusCallbackImpl *) iface;
123 HWND Item;
124 LONG r;
125 WCHAR OldText[100];
126
127 Item = GetDlgItem(This->hDialog, IDC_PROGRESS);
128 if (NULL != Item && 0 != ulProgressMax)
129 {
130 SendMessageW(Item, PBM_SETPOS, (ulProgress * 100) / ulProgressMax, 0);
131 }
132
133 Item = GetDlgItem(This->hDialog, IDC_STATUS);
134 if (NULL != Item && NULL != szStatusText)
135 {
136 SendMessageW(Item, WM_GETTEXT, sizeof(OldText) / sizeof(OldText[0]),
137 (LPARAM) OldText);
138 if (sizeof(OldText) / sizeof(OldText[0]) - 1 <= wcslen(OldText) || 0 != wcscmp(OldText, szStatusText))
139 {
140 SendMessageW(Item, WM_SETTEXT, 0, (LPARAM) szStatusText);
141 }
142 }
143
144 SetLastError(0);
145 r = GetWindowLongPtrW(This->hDialog, GWLP_USERDATA);
146 if (0 != r || 0 != GetLastError())
147 {
148 *This->pbCancelled = TRUE;
149 DPRINT("Cancelled\n");
150 return E_ABORT;
151 }
152
153 return S_OK;
154 }
155
156 static HRESULT WINAPI
157 dlOnStopBinding(IBindStatusCallback* iface, HRESULT hresult, LPCWSTR szError)
158 {
159 DPRINT1("OnStopBinding not implemented\n");
160
161 return S_OK;
162 }
163
164 static HRESULT WINAPI
165 dlGetBindInfo(IBindStatusCallback* iface, DWORD* grfBINDF, BINDINFO* pbindinfo)
166 {
167 DPRINT1("GetBindInfo not implemented\n");
168
169 return S_OK;
170 }
171
172 static HRESULT WINAPI
173 dlOnDataAvailable(IBindStatusCallback* iface, DWORD grfBSCF,
174 DWORD dwSize, FORMATETC* pformatetc, STGMEDIUM* pstgmed)
175 {
176 DPRINT1("OnDataAvailable implemented\n");
177
178 return S_OK;
179 }
180
181 static HRESULT WINAPI
182 dlOnObjectAvailable(IBindStatusCallback* iface, REFIID riid, IUnknown* punk)
183 {
184 DPRINT1("OnObjectAvailable implemented\n");
185
186 return S_OK;
187 }
188
189 static const IBindStatusCallbackVtbl dlVtbl =
190 {
191 dlQueryInterface,
192 dlAddRef,
193 dlRelease,
194 dlOnStartBinding,
195 dlGetPriority,
196 dlOnLowResource,
197 dlOnProgress,
198 dlOnStopBinding,
199 dlGetBindInfo,
200 dlOnDataAvailable,
201 dlOnObjectAvailable
202 };
203
204 static IBindStatusCallback*
205 CreateDl(HWND Dlg, BOOL *pbCancelled)
206 {
207 IBindStatusCallbackImpl *This;
208
209 This = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IBindStatusCallbackImpl));
210 if (!This)
211 return NULL;
212
213 This->vtbl = &dlVtbl;
214 This->ref = 1;
215 This->hDialog = Dlg;
216 This->pbCancelled = pbCancelled;
217
218 return (IBindStatusCallback*) This;
219 }
220
221 static DWORD WINAPI
222 ThreadFunc(LPVOID Context)
223 {
224 //static const WCHAR szUrl[] = DownloadUrl;
225 IBindStatusCallback *dl;
226 WCHAR path[MAX_PATH];
227 LPWSTR p;
228 STARTUPINFOW si;
229 PROCESS_INFORMATION pi;
230 HWND Dlg = (HWND) Context;
231 DWORD r;
232 BOOL bCancelled = FALSE;
233 BOOL bTempfile = FALSE;
234 HKEY hKey;
235 DWORD dwSize = MAX_PATH;
236
237 /* built the path for the download */
238 p = wcsrchr(SelectedApplication->Location, L'/');
239 if (NULL == p)
240 {
241 goto end;
242 }
243
244 /* Create default download path */
245 if (GetWindowsDirectory(path, sizeof(path) / sizeof(WCHAR)))
246 {
247 WCHAR DPath[256];
248 int i;
249 for (i = 0; i < 4; i++)
250 {
251 if (i == 3)
252 {
253 DPath[i] = '\0';
254 break;
255 }
256 DPath[i] = path[i];
257 }
258 LoadString(GetModuleHandle(NULL), IDS_DOWNLOAD_FOLDER, path, sizeof(path) / sizeof(WCHAR));
259 wcscat((LPWSTR)DPath, path);
260 wcscpy(path, DPath);
261 }
262
263 if (RegOpenKey(HKEY_LOCAL_MACHINE,
264 TEXT("Software\\ReactOS\\Downloader"),
265 &hKey) == ERROR_SUCCESS)
266 {
267 if ((RegQueryValueEx(hKey,
268 L"DownloadFolder",
269 NULL,
270 NULL,
271 (LPBYTE)&path,
272 &dwSize) != ERROR_SUCCESS) && (path[0] == 0))
273 {
274 goto end;
275 }
276 }
277
278 if (GetFileAttributes(path) == 0xFFFFFFFF)
279 if (!CreateDirectory((LPCTSTR)path,NULL))
280 {
281 goto end;
282 }
283 wcscat(path, L"\\");
284 wcscat(path, p + 1);
285
286 /* download it */
287 bTempfile = TRUE;
288 dl = CreateDl(Context, &bCancelled);
289 r = URLDownloadToFileW(NULL, SelectedApplication->Location, path, 0, dl);
290 if (NULL != dl)
291 {
292 IBindStatusCallback_Release(dl);
293 }
294 if (S_OK != r)
295 {
296 MessageBoxW(0,Strings[IDS_DOWNLOAD_ERROR],0,0);
297 goto end;
298 }
299 else if (bCancelled)
300 {
301 goto end;
302 }
303 ShowWindow(Dlg, SW_HIDE);
304
305 /* run it */
306 memset(&si, 0, sizeof(si));
307 si.cb = sizeof(si);
308 r = CreateProcessW(path, NULL, NULL, NULL, 0, 0, NULL, NULL, &si, &pi);
309 if (0 == r)
310 {
311 goto end;
312 }
313 CloseHandle(pi.hThread);
314 WaitForSingleObject(pi.hProcess, INFINITE);
315 CloseHandle(pi.hProcess);
316
317 end:
318 if (bTempfile)
319 {
320 if (bCancelled)
321 DeleteFileW(path);
322 else
323 {
324 DWORD dwSize = sizeof(DWORD);
325 DWORD dwValue, dwType = REG_DWORD;
326 if (RegQueryValueEx(hKey,
327 L"DeleteInstaller",
328 NULL,
329 &dwType,
330 (LPBYTE)&dwValue,
331 &dwSize) == ERROR_SUCCESS)
332 if (dwValue == 0x1)
333 DeleteFileW(path);
334 RegCloseKey(hKey);
335 }
336 }
337 EndDialog(Dlg, 0);
338 return 0;
339 }
340
341 INT_PTR CALLBACK
342 DownloadProc(HWND Dlg, UINT Msg, WPARAM wParam, LPARAM lParam)
343 {
344 HANDLE Thread;
345 DWORD ThreadId;
346 HWND Item;
347
348 switch (Msg)
349 {
350 case WM_INITDIALOG:/*
351 Icon = LoadIconW((HINSTANCE) GetWindowLongPtr(Dlg, GWLP_HINSTANCE),
352 MAKEINTRESOURCEW(IDI_ICON_MAIN));
353 if (NULL != Icon)
354 {
355 SendMessageW(Dlg, WM_SETICON, ICON_BIG, (LPARAM) Icon);
356 SendMessageW(Dlg, WM_SETICON, ICON_SMALL, (LPARAM) Icon);
357 }*/
358 SetWindowLongPtrW(Dlg, GWLP_USERDATA, 0);
359 Item = GetDlgItem(Dlg, IDC_PROGRESS);
360 if (NULL != Item)
361 {
362 SendMessageW(Item, PBM_SETRANGE, 0, MAKELPARAM(0,100));
363 SendMessageW(Item, PBM_SETPOS, 0, 0);
364 }/*
365 Item = GetDlgItem(Dlg, IDC_REMOVE);
366 if (NULL != Item)
367 {
368 if (GetShortcutName(ShortcutName) &&
369 INVALID_FILE_ATTRIBUTES != GetFileAttributesW(ShortcutName))
370 {
371 SendMessageW(Item, BM_SETCHECK, BST_CHECKED, 0);
372 }
373 else
374 {
375 SendMessageW(Item, BM_SETCHECK, BST_UNCHECKED, 0);
376 ShowWindow(Item, SW_HIDE);
377 }
378 }*/
379 Thread = CreateThread(NULL, 0, ThreadFunc, Dlg, 0, &ThreadId);
380 if (NULL == Thread)
381 {
382 return FALSE;
383 }
384 CloseHandle(Thread);
385 return TRUE;
386
387 case WM_COMMAND:
388 if (wParam == IDCANCEL)
389 {
390 SetWindowLongPtrW(Dlg, GWLP_USERDATA, 1);
391 PostMessage(Dlg, WM_CLOSE, 0, 0);
392 }
393 return FALSE;
394
395 case WM_CLOSE:
396 EndDialog(Dlg, 0);
397 return TRUE;
398
399 default:
400 return FALSE;
401 }
402 }