comctl32: Validate the day of month when scrolling through years.
[wine] / dlls / appwiz.cpl / addons.c
1 /*
2  * Copyright 2006-2010 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 "config.h"
20
21 #include <stdarg.h>
22 #include <fcntl.h>
23 #include <stdio.h>
24 #ifdef HAVE_UNISTD_H
25 # include <unistd.h>
26 #endif
27
28 #define COBJMACROS
29 #define NONAMELESSUNION
30 #define NONAMELESSSTRUCT
31
32 #include "windef.h"
33 #include "winbase.h"
34 #include "winuser.h"
35 #include "cpl.h"
36 #include "winreg.h"
37 #include "ole2.h"
38 #include "commctrl.h"
39 #include "advpub.h"
40 #include "wininet.h"
41 #include "shellapi.h"
42 #include "urlmon.h"
43 #include "msi.h"
44
45 #include "appwiz.h"
46 #include "res.h"
47
48 #include "wine/debug.h"
49 #include "wine/unicode.h"
50 #include "wine/library.h"
51
52 WINE_DEFAULT_DEBUG_CHANNEL(appwizcpl);
53
54 #define GECKO_VERSION "1.2.0"
55
56 #ifdef __i386__
57 #define ARCH_STRING "x86"
58 #define GECKO_SHA "6964d1877668ab7da07a60f6dcf23fb0e261a808"
59 #elif defined(__x86_64__)
60 #define ARCH_STRING "x86_64"
61 #define GECKO_SHA "3ac3c3e880e40f7763824866372ffc56128f0abd"
62 #else
63 #define ARCH_STRING ""
64 #define GECKO_SHA "???"
65 #endif
66
67 #define GECKO_FILE_NAME "wine_gecko-" GECKO_VERSION "-" ARCH_STRING ".msi"
68
69 static const WCHAR mshtml_keyW[] =
70     {'S','o','f','t','w','a','r','e',
71      '\\','W','i','n','e',
72      '\\','M','S','H','T','M','L',0};
73
74 static HWND install_dialog = NULL;
75 static LPWSTR url = NULL;
76
77 static inline char *heap_strdupWtoA(LPCWSTR str)
78 {
79     char *ret = NULL;
80
81     if(str) {
82         DWORD size = WideCharToMultiByte(CP_ACP, 0, str, -1, NULL, 0, NULL, NULL);
83         ret = heap_alloc(size);
84         WideCharToMultiByte(CP_ACP, 0, str, -1, ret, size, NULL, NULL);
85     }
86
87     return ret;
88 }
89
90 /* SHA definitions are copied from advapi32. They aren't available in headers. */
91
92 typedef struct {
93    ULONG Unknown[6];
94    ULONG State[5];
95    ULONG Count[2];
96    UCHAR Buffer[64];
97 } SHA_CTX, *PSHA_CTX;
98
99 void WINAPI A_SHAInit(PSHA_CTX);
100 void WINAPI A_SHAUpdate(PSHA_CTX,const unsigned char*,UINT);
101 void WINAPI A_SHAFinal(PSHA_CTX,PULONG);
102
103 static BOOL sha_check(const WCHAR *file_name)
104 {
105     const unsigned char *file_map;
106     HANDLE file, map;
107     ULONG sha[5];
108     char buf[2*sizeof(sha)+1];
109     SHA_CTX ctx;
110     DWORD size, i;
111
112     file = CreateFileW(file_name, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_READONLY, NULL);
113     if(file == INVALID_HANDLE_VALUE)
114         return FALSE;
115
116     size = GetFileSize(file, NULL);
117
118     map = CreateFileMappingW(file, NULL, PAGE_READONLY, 0, 0, NULL);
119     CloseHandle(file);
120     if(!map)
121         return FALSE;
122
123     file_map = MapViewOfFile(map, FILE_MAP_READ, 0, 0, 0);
124     CloseHandle(map);
125     if(!file_map)
126         return FALSE;
127
128     A_SHAInit(&ctx);
129     A_SHAUpdate(&ctx, file_map, size);
130     A_SHAFinal(&ctx, sha);
131
132     UnmapViewOfFile(file_map);
133
134     for(i=0; i < sizeof(sha); i++)
135         sprintf(buf + i*2, "%02x", *((unsigned char*)sha+i));
136
137     if(strcmp(buf, GECKO_SHA)) {
138         WCHAR message[256];
139
140         WARN("Got %s, expected %s\n", buf, GECKO_SHA);
141
142         if(LoadStringW(hInst, IDS_INVALID_SHA, message, sizeof(message)/sizeof(WCHAR)))
143             MessageBoxW(NULL, message, NULL, MB_ICONERROR);
144
145         return FALSE;
146     }
147
148     return TRUE;
149 }
150
151 static void set_status(DWORD id)
152 {
153     HWND status = GetDlgItem(install_dialog, ID_DWL_STATUS);
154     WCHAR buf[64];
155
156     LoadStringW(hInst, id, buf, sizeof(buf)/sizeof(WCHAR));
157     SendMessageW(status, WM_SETTEXT, 0, (LPARAM)buf);
158 }
159
160 static BOOL install_file(const WCHAR *file_name)
161 {
162     ULONG res;
163
164     res = MsiInstallProductW(file_name, NULL);
165     if(res != ERROR_SUCCESS) {
166         ERR("MsiInstallProduct failed: %u\n", res);
167         return FALSE;
168     }
169
170     return TRUE;
171 }
172
173 static BOOL install_from_unix_file(const char *file_name)
174 {
175     LPWSTR dos_file_name;
176     int fd;
177     BOOL ret;
178
179     static WCHAR * (CDECL *wine_get_dos_file_name)(const char*);
180     static const WCHAR kernel32W[] = {'k','e','r','n','e','l','3','2','.','d','l','l',0};
181
182     fd = open(file_name, O_RDONLY);
183     if(fd == -1) {
184         TRACE("%s not found\n", debugstr_a(file_name));
185         return FALSE;
186     }
187
188     close(fd);
189
190     if(!wine_get_dos_file_name)
191         wine_get_dos_file_name = (void*)GetProcAddress(GetModuleHandleW(kernel32W), "wine_get_dos_file_name");
192
193     if(wine_get_dos_file_name) { /* Wine UNIX mode */
194         dos_file_name = wine_get_dos_file_name(file_name);
195         if(!dos_file_name) {
196             ERR("Could not get dos file name of %s\n", debugstr_a(file_name));
197             return FALSE;
198         }
199     } else { /* Windows mode */
200         UINT res;
201         WARN("Could not get wine_get_dos_file_name function, calling install_cab directly.\n");
202         res = MultiByteToWideChar( CP_ACP, 0, file_name, -1, 0, 0);
203         dos_file_name = heap_alloc (res*sizeof(WCHAR));
204         MultiByteToWideChar( CP_ACP, 0, file_name, -1, dos_file_name, res);
205     }
206
207     ret = install_file(dos_file_name);
208
209     heap_free(dos_file_name);
210     return ret;
211 }
212
213 static BOOL install_from_registered_dir(void)
214 {
215     char *file_name;
216     HKEY hkey;
217     DWORD res, type, size = MAX_PATH;
218     BOOL ret;
219
220     /* @@ Wine registry key: HKCU\Software\Wine\MSHTML */
221     res = RegOpenKeyW(HKEY_CURRENT_USER, mshtml_keyW, &hkey);
222     if(res != ERROR_SUCCESS)
223         return FALSE;
224
225     file_name = heap_alloc(size+sizeof(GECKO_FILE_NAME));
226     res = RegGetValueA(hkey, NULL, "GeckoCabDir", RRF_RT_ANY, &type, (PBYTE)file_name, &size);
227     if(res == ERROR_MORE_DATA) {
228         file_name = heap_realloc(file_name, size+sizeof(GECKO_FILE_NAME));
229         res = RegGetValueA(hkey, NULL, "GeckoCabDir", RRF_RT_ANY, &type, (PBYTE)file_name, &size);
230     }
231     RegCloseKey(hkey);
232     if(res != ERROR_SUCCESS || (type != REG_SZ && type != REG_EXPAND_SZ)) {
233         heap_free(file_name);
234         return FALSE;
235     }
236
237     strcat(file_name, GECKO_FILE_NAME);
238
239     TRACE("Trying %s\n", debugstr_a(file_name));
240
241     ret = install_from_unix_file(file_name);
242
243     heap_free(file_name);
244     return ret;
245 }
246
247 static BOOL install_from_default_dir(void)
248 {
249     const char *data_dir, *subdir;
250     char *file_name;
251     int len, len2;
252     BOOL ret;
253
254     if((data_dir = wine_get_data_dir()))
255         subdir = "/gecko/";
256     else if((data_dir = wine_get_build_dir()))
257         subdir = "/../gecko/";
258     else
259         return FALSE;
260
261     len = strlen(data_dir);
262     len2 = strlen(subdir);
263
264     file_name = heap_alloc(len+len2+sizeof(GECKO_FILE_NAME));
265     memcpy(file_name, data_dir, len);
266     memcpy(file_name+len, subdir, len2);
267     memcpy(file_name+len+len2, GECKO_FILE_NAME, sizeof(GECKO_FILE_NAME));
268
269     ret = install_from_unix_file(file_name);
270
271     heap_free(file_name);
272
273     if (!ret)
274         ret = install_from_unix_file(INSTALL_DATADIR "/wine/gecko/" GECKO_FILE_NAME);
275     if (!ret && strcmp(INSTALL_DATADIR, "/usr/share"))
276         ret = install_from_unix_file("/usr/share/wine/gecko/" GECKO_FILE_NAME);
277     return ret;
278 }
279
280 static HRESULT WINAPI InstallCallback_QueryInterface(IBindStatusCallback *iface,
281         REFIID riid, void **ppv)
282 {
283     if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IBindStatusCallback, riid)) {
284         *ppv = iface;
285         return S_OK;
286     }
287
288     return E_INVALIDARG;
289 }
290
291 static ULONG WINAPI InstallCallback_AddRef(IBindStatusCallback *iface)
292 {
293     return 2;
294 }
295
296 static ULONG WINAPI InstallCallback_Release(IBindStatusCallback *iface)
297 {
298     return 1;
299 }
300
301 static HRESULT WINAPI InstallCallback_OnStartBinding(IBindStatusCallback *iface,
302         DWORD dwReserved, IBinding *pib)
303 {
304     set_status(IDS_DOWNLOADING);
305     return S_OK;
306 }
307
308 static HRESULT WINAPI InstallCallback_GetPriority(IBindStatusCallback *iface,
309         LONG *pnPriority)
310 {
311     return E_NOTIMPL;
312 }
313
314 static HRESULT WINAPI InstallCallback_OnLowResource(IBindStatusCallback *iface,
315        DWORD dwReserved)
316 {
317     return E_NOTIMPL;
318 }
319
320 static HRESULT WINAPI InstallCallback_OnProgress(IBindStatusCallback *iface, ULONG ulProgress,
321         ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText)
322 {
323     HWND progress = GetDlgItem(install_dialog, ID_DWL_PROGRESS);
324
325     if(ulProgressMax)
326         SendMessageW(progress, PBM_SETRANGE32, 0, ulProgressMax);
327     if(ulProgress)
328         SendMessageW(progress, PBM_SETPOS, ulProgress, 0);
329
330     return S_OK;
331 }
332
333 static HRESULT WINAPI InstallCallback_OnStopBinding(IBindStatusCallback *iface,
334         HRESULT hresult, LPCWSTR szError)
335 {
336     if(FAILED(hresult)) {
337         ERR("Binding failed %08x\n", hresult);
338         return S_OK;
339     }
340
341     set_status(IDS_INSTALLING);
342     return S_OK;
343 }
344
345 static HRESULT WINAPI InstallCallback_GetBindInfo(IBindStatusCallback *iface,
346         DWORD* grfBINDF, BINDINFO* pbindinfo)
347 {
348     /* FIXME */
349     *grfBINDF = 0;
350     return S_OK;
351 }
352
353 static HRESULT WINAPI InstallCallback_OnDataAvailable(IBindStatusCallback *iface, DWORD grfBSCF,
354         DWORD dwSize, FORMATETC* pformatetc, STGMEDIUM* pstgmed)
355 {
356     ERR("\n");
357     return E_NOTIMPL;
358 }
359
360 static HRESULT WINAPI InstallCallback_OnObjectAvailable(IBindStatusCallback *iface,
361         REFIID riid, IUnknown* punk)
362 {
363     ERR("\n");
364     return E_NOTIMPL;
365 }
366
367 static const IBindStatusCallbackVtbl InstallCallbackVtbl = {
368     InstallCallback_QueryInterface,
369     InstallCallback_AddRef,
370     InstallCallback_Release,
371     InstallCallback_OnStartBinding,
372     InstallCallback_GetPriority,
373     InstallCallback_OnLowResource,
374     InstallCallback_OnProgress,
375     InstallCallback_OnStopBinding,
376     InstallCallback_GetBindInfo,
377     InstallCallback_OnDataAvailable,
378     InstallCallback_OnObjectAvailable
379 };
380
381 static IBindStatusCallback InstallCallback = { &InstallCallbackVtbl };
382
383 static LPWSTR get_url(void)
384 {
385     HKEY hkey;
386     DWORD res, type;
387     DWORD size = INTERNET_MAX_URL_LENGTH*sizeof(WCHAR);
388     DWORD returned_size;
389     LPWSTR url;
390
391     static const WCHAR wszGeckoUrl[] = {'G','e','c','k','o','U','r','l',0};
392     static const WCHAR httpW[] = {'h','t','t','p'};
393     static const WCHAR arch_formatW[] = {'?','a','r','c','h','='};
394     static const WCHAR v_formatW[] = {'&','v','='};
395
396     /* @@ Wine registry key: HKCU\Software\Wine\MSHTML */
397     res = RegOpenKeyW(HKEY_CURRENT_USER, mshtml_keyW, &hkey);
398     if(res != ERROR_SUCCESS)
399         return NULL;
400
401     url = heap_alloc(size);
402     returned_size = size;
403
404     res = RegQueryValueExW(hkey, wszGeckoUrl, NULL, &type, (LPBYTE)url, &returned_size);
405     RegCloseKey(hkey);
406     if(res != ERROR_SUCCESS || type != REG_SZ) {
407         heap_free(url);
408         return NULL;
409     }
410
411     if(returned_size > sizeof(httpW) && !memcmp(url, httpW, sizeof(httpW))) {
412         DWORD len;
413
414         len = strlenW(url);
415         memcpy(url+len, arch_formatW, sizeof(arch_formatW));
416         len += sizeof(arch_formatW)/sizeof(WCHAR);
417         len += MultiByteToWideChar(CP_ACP, 0, ARCH_STRING, sizeof(ARCH_STRING), url+len, size/sizeof(WCHAR)-len)-1;
418         memcpy(url+len, v_formatW, sizeof(v_formatW));
419         len += sizeof(v_formatW)/sizeof(WCHAR);
420         MultiByteToWideChar(CP_ACP, 0, GECKO_VERSION, -1, url+len, size/sizeof(WCHAR)-len);
421     }
422
423     TRACE("Got URL %s\n", debugstr_w(url));
424     return url;
425 }
426
427 static DWORD WINAPI download_proc(PVOID arg)
428 {
429     WCHAR tmp_dir[MAX_PATH], tmp_file[MAX_PATH];
430     HRESULT hres;
431
432     GetTempPathW(sizeof(tmp_dir)/sizeof(WCHAR), tmp_dir);
433     GetTempFileNameW(tmp_dir, NULL, 0, tmp_file);
434
435     TRACE("using temp file %s\n", debugstr_w(tmp_file));
436
437     hres = URLDownloadToFileW(NULL, url, tmp_file, 0, &InstallCallback);
438     if(FAILED(hres)) {
439         ERR("URLDownloadToFile failed: %08x\n", hres);
440         return 0;
441     }
442
443     if(sha_check(tmp_file))
444         install_file(tmp_file);
445     DeleteFileW(tmp_file);
446     EndDialog(install_dialog, 0);
447     return 0;
448 }
449
450 static INT_PTR CALLBACK installer_proc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
451 {
452     switch(msg) {
453     case WM_INITDIALOG:
454         ShowWindow(GetDlgItem(hwnd, ID_DWL_PROGRESS), SW_HIDE);
455         install_dialog = hwnd;
456         return TRUE;
457
458     case WM_COMMAND:
459         switch(wParam) {
460         case IDCANCEL:
461             EndDialog(hwnd, 0);
462             return FALSE;
463
464         case ID_DWL_INSTALL:
465             ShowWindow(GetDlgItem(hwnd, ID_DWL_PROGRESS), SW_SHOW);
466             EnableWindow(GetDlgItem(hwnd, ID_DWL_INSTALL), 0);
467             EnableWindow(GetDlgItem(hwnd, IDCANCEL), 0); /* FIXME */
468             CreateThread(NULL, 0, download_proc, NULL, 0, NULL);
469             return FALSE;
470         }
471     }
472
473     return FALSE;
474 }
475
476 BOOL install_wine_gecko(void)
477 {
478     if(!*ARCH_STRING)
479         return FALSE;
480
481     /*
482      * Try to find Gecko .cab file in following order:
483      * - directory stored in GeckoCabDir value of HKCU/Wine/Software/MSHTML key
484      * - $datadir/gecko/
485      * - $INSTALL_DATADIR/wine/gecko/
486      * - /usr/share/wine/gecko/
487      * - download from URL stored in GeckoUrl value of HKCU/Wine/Software/MSHTML key
488      */
489     if(!install_from_registered_dir()
490        && !install_from_default_dir()
491        && (url = get_url()))
492         DialogBoxW(hInst, MAKEINTRESOURCEW(ID_DWL_DIALOG), 0, installer_proc);
493
494     heap_free(url);
495     url = NULL;
496     return TRUE;
497 }