2 * Copyright 2006-2007 Jacek Caban for CodeWeavers
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.
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.
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
28 #define NONAMELESSUNION
29 #define NONAMELESSSTRUCT
41 #include "wine/debug.h"
42 #include "wine/unicode.h"
43 #include "wine/library.h"
45 #include "mshtml_private.h"
48 WINE_DEFAULT_DEBUG_CHANNEL(mshtml
);
51 #define GECKO_ARCH "x86"
56 #define GECKO_FILE_NAME "wine_gecko-" GECKO_VERSION "-" GECKO_ARCH ".cab"
58 static const WCHAR mshtml_keyW
[] =
59 {'S','o','f','t','w','a','r','e',
61 '\\','M','S','H','T','M','L',0};
63 static const CHAR mshtml_keyA
[] =
64 {'S','o','f','t','w','a','r','e',
66 '\\','M','S','H','T','M','L',0};
68 static HWND install_dialog
= NULL
;
69 static LPWSTR tmp_file_name
= NULL
;
70 static HANDLE tmp_file
= INVALID_HANDLE_VALUE
;
71 static LPWSTR url
= NULL
;
73 static void clean_up(void)
75 if(tmp_file
!= INVALID_HANDLE_VALUE
)
76 CloseHandle(tmp_file
);
79 DeleteFileW(tmp_file_name
);
80 heap_free(tmp_file_name
);
84 if(tmp_file
!= INVALID_HANDLE_VALUE
) {
85 CloseHandle(tmp_file
);
86 tmp_file
= INVALID_HANDLE_VALUE
;
90 EndDialog(install_dialog
, 0);
93 static void set_status(DWORD id
)
95 HWND status
= GetDlgItem(install_dialog
, ID_DWL_STATUS
);
98 LoadStringW(hInst
, id
, buf
, sizeof(buf
)/sizeof(WCHAR
));
99 SendMessageW(status
, WM_SETTEXT
, 0, (LPARAM
)buf
);
102 static void set_registry(LPCSTR install_dir
)
104 WCHAR mshtml_key
[100];
109 static const WCHAR wszGeckoPath
[] = {'G','e','c','k','o','P','a','t','h',0};
110 static const WCHAR wszWineGecko
[] = {'w','i','n','e','_','g','e','c','k','o',0};
112 memcpy(mshtml_key
, mshtml_keyW
, sizeof(mshtml_keyW
));
113 mshtml_key
[sizeof(mshtml_keyW
)/sizeof(WCHAR
)-1] = '\\';
114 MultiByteToWideChar(CP_ACP
, 0, GECKO_VERSION
, sizeof(GECKO_VERSION
),
115 mshtml_key
+sizeof(mshtml_keyW
)/sizeof(WCHAR
),
116 (sizeof(mshtml_key
)-sizeof(mshtml_keyW
))/sizeof(WCHAR
));
118 /* @@ Wine registry key: HKCU\Software\Wine\MSHTML\<version> */
119 res
= RegCreateKeyW(HKEY_CURRENT_USER
, mshtml_key
, &hkey
);
120 if(res
!= ERROR_SUCCESS
) {
121 ERR("Faild to create MSHTML key: %d\n", res
);
125 len
= MultiByteToWideChar(CP_ACP
, 0, install_dir
, -1, NULL
, 0)-1;
126 gecko_path
= heap_alloc((len
+1)*sizeof(WCHAR
)+sizeof(wszWineGecko
));
127 MultiByteToWideChar(CP_ACP
, 0, install_dir
, -1, gecko_path
, len
+1);
129 if (len
&& gecko_path
[len
-1] != '\\')
130 gecko_path
[len
++] = '\\';
132 memcpy(gecko_path
+len
, wszWineGecko
, sizeof(wszWineGecko
));
134 res
= RegSetValueExW(hkey
, wszGeckoPath
, 0, REG_SZ
, (LPVOID
)gecko_path
,
135 len
*sizeof(WCHAR
)+sizeof(wszWineGecko
));
136 heap_free(gecko_path
);
138 if(res
!= ERROR_SUCCESS
)
139 ERR("Failed to set GeckoPath value: %08x\n", res
);
142 static BOOL
install_cab(LPCWSTR file_name
)
145 char install_dir
[MAX_PATH
];
146 HRESULT (WINAPI
*pExtractFilesA
)(LPCSTR
,LPCSTR
,DWORD
,LPCSTR
,LPVOID
,DWORD
);
151 static const WCHAR wszAdvpack
[] = {'a','d','v','p','a','c','k','.','d','l','l',0};
153 TRACE("(%s)\n", debugstr_w(file_name
));
155 GetSystemDirectoryA(install_dir
, sizeof(install_dir
));
156 strcat(install_dir
, "\\gecko\\");
157 res
= CreateDirectoryA(install_dir
, NULL
);
158 if(!res
&& GetLastError() != ERROR_ALREADY_EXISTS
) {
159 ERR("Could not create directory: %08u\n", GetLastError());
163 strcat(install_dir
, GECKO_VERSION
);
164 res
= CreateDirectoryA(install_dir
, NULL
);
165 if(!res
&& GetLastError() != ERROR_ALREADY_EXISTS
) {
166 ERR("Could not create directory: %08u\n", GetLastError());
170 advpack
= LoadLibraryW(wszAdvpack
);
171 pExtractFilesA
= (void *)GetProcAddress(advpack
, "ExtractFiles");
173 /* FIXME: Use unicode version (not yet implemented) */
174 file_name_a
= heap_strdupWtoA(file_name
);
175 hres
= pExtractFilesA(file_name_a
, install_dir
, 0, NULL
, NULL
, 0);
176 FreeLibrary(advpack
);
177 heap_free(file_name_a
);
179 ERR("Could not extract package: %08x\n", hres
);
184 set_registry(install_dir
);
190 static BOOL
install_from_unix_file(const char *file_name
)
192 LPWSTR dos_file_name
;
196 static WCHAR
* (CDECL
*wine_get_dos_file_name
)(const char*);
197 static const WCHAR kernel32W
[] = {'k','e','r','n','e','l','3','2','.','d','l','l',0};
199 fd
= open(file_name
, O_RDONLY
);
201 TRACE("%s not found\n", debugstr_a(file_name
));
207 if(!wine_get_dos_file_name
)
208 wine_get_dos_file_name
= (void*)GetProcAddress(GetModuleHandleW(kernel32W
), "wine_get_dos_file_name");
210 if(wine_get_dos_file_name
) { /* Wine UNIX mode */
211 dos_file_name
= wine_get_dos_file_name(file_name
);
213 ERR("Could not get dos file name of %s\n", debugstr_a(file_name
));
216 } else { /* Windows mode */
218 WARN("Could not get wine_get_dos_file_name function, calling install_cab directly.\n");
219 res
= MultiByteToWideChar( CP_ACP
, 0, file_name
, -1, 0, 0);
220 dos_file_name
= heap_alloc (res
*sizeof(WCHAR
));
221 MultiByteToWideChar( CP_ACP
, 0, file_name
, -1, dos_file_name
, res
);
224 ret
= install_cab(dos_file_name
);
226 heap_free(dos_file_name
);
230 static BOOL
install_from_registered_dir(void)
233 DWORD res
, type
, size
= MAX_PATH
;
236 file_name
= heap_alloc(size
+sizeof(GECKO_FILE_NAME
));
237 /* @@ Wine registry key: HKCU\Software\Wine\MSHTML */
238 res
= RegGetValueA(HKEY_CURRENT_USER
, mshtml_keyA
, "GeckoCabDir", RRF_RT_ANY
, &type
, (PBYTE
)file_name
, &size
);
239 if(res
== ERROR_MORE_DATA
) {
240 file_name
= heap_realloc(file_name
, size
+sizeof(GECKO_FILE_NAME
));
241 res
= RegGetValueA(HKEY_CURRENT_USER
, mshtml_keyA
, "GeckoCabDir", RRF_RT_ANY
, &type
, (PBYTE
)file_name
, &size
);
244 if(res
!= ERROR_SUCCESS
|| (type
!= REG_SZ
&& type
!= REG_EXPAND_SZ
)) {
245 heap_free(file_name
);
249 strcat(file_name
, GECKO_FILE_NAME
);
251 TRACE("Trying %s\n", debugstr_a(file_name
));
253 ret
= install_from_unix_file(file_name
);
255 heap_free(file_name
);
259 static BOOL
install_from_default_dir(void)
261 const char *data_dir
, *subdir
;
266 if((data_dir
= wine_get_data_dir()))
268 else if((data_dir
= wine_get_build_dir()))
269 subdir
= "/../gecko/";
273 len
= strlen(data_dir
);
274 len2
= strlen(subdir
);
276 file_name
= heap_alloc(len
+len2
+sizeof(GECKO_FILE_NAME
));
277 memcpy(file_name
, data_dir
, len
);
278 memcpy(file_name
+len
, subdir
, len2
);
279 memcpy(file_name
+len
+len2
, GECKO_FILE_NAME
, sizeof(GECKO_FILE_NAME
));
281 ret
= install_from_unix_file(file_name
);
283 heap_free(file_name
);
287 static HRESULT WINAPI
InstallCallback_QueryInterface(IBindStatusCallback
*iface
,
288 REFIID riid
, void **ppv
)
290 if(IsEqualGUID(&IID_IUnknown
, riid
) || IsEqualGUID(&IID_IBindStatusCallback
, riid
)) {
298 static ULONG WINAPI
InstallCallback_AddRef(IBindStatusCallback
*iface
)
303 static ULONG WINAPI
InstallCallback_Release(IBindStatusCallback
*iface
)
308 static HRESULT WINAPI
InstallCallback_OnStartBinding(IBindStatusCallback
*iface
,
309 DWORD dwReserved
, IBinding
*pib
)
311 WCHAR tmp_dir
[MAX_PATH
];
313 set_status(IDS_DOWNLOADING
);
315 GetTempPathW(sizeof(tmp_dir
)/sizeof(WCHAR
), tmp_dir
);
317 tmp_file_name
= heap_alloc(MAX_PATH
*sizeof(WCHAR
));
318 GetTempFileNameW(tmp_dir
, NULL
, 0, tmp_file_name
);
320 TRACE("creating temp file %s\n", debugstr_w(tmp_file_name
));
322 tmp_file
= CreateFileW(tmp_file_name
, GENERIC_WRITE
, 0, NULL
,
323 CREATE_ALWAYS
, FILE_ATTRIBUTE_NORMAL
, NULL
);
325 if(tmp_file
== INVALID_HANDLE_VALUE
) {
326 ERR("Could not create file: %d\n", GetLastError());
334 static HRESULT WINAPI
InstallCallback_GetPriority(IBindStatusCallback
*iface
,
340 static HRESULT WINAPI
InstallCallback_OnLowResource(IBindStatusCallback
*iface
,
346 static HRESULT WINAPI
InstallCallback_OnProgress(IBindStatusCallback
*iface
, ULONG ulProgress
,
347 ULONG ulProgressMax
, ULONG ulStatusCode
, LPCWSTR szStatusText
)
349 HWND progress
= GetDlgItem(install_dialog
, ID_DWL_PROGRESS
);
352 SendMessageW(progress
, PBM_SETRANGE32
, 0, ulProgressMax
);
354 SendMessageW(progress
, PBM_SETPOS
, ulProgress
, 0);
359 static HRESULT WINAPI
InstallCallback_OnStopBinding(IBindStatusCallback
*iface
,
360 HRESULT hresult
, LPCWSTR szError
)
362 if(FAILED(hresult
)) {
363 ERR("Binding failed %08x\n", hresult
);
368 CloseHandle(tmp_file
);
369 tmp_file
= INVALID_HANDLE_VALUE
;
371 set_status(IDS_INSTALLING
);
373 install_cab(tmp_file_name
);
378 static HRESULT WINAPI
InstallCallback_GetBindInfo(IBindStatusCallback
*iface
,
379 DWORD
* grfBINDF
, BINDINFO
* pbindinfo
)
386 static HRESULT WINAPI
InstallCallback_OnDataAvailable(IBindStatusCallback
*iface
, DWORD grfBSCF
,
387 DWORD dwSize
, FORMATETC
* pformatetc
, STGMEDIUM
* pstgmed
)
389 IStream
*str
= pstgmed
->u
.pstm
;
398 hres
= IStream_Read(str
, buf
, sizeof(buf
), &size
);
400 WriteFile(tmp_file
, buf
, size
, &written
, NULL
);
401 }while(hres
== S_OK
);
406 static HRESULT WINAPI
InstallCallback_OnObjectAvailable(IBindStatusCallback
*iface
,
407 REFIID riid
, IUnknown
* punk
)
413 static const IBindStatusCallbackVtbl InstallCallbackVtbl
= {
414 InstallCallback_QueryInterface
,
415 InstallCallback_AddRef
,
416 InstallCallback_Release
,
417 InstallCallback_OnStartBinding
,
418 InstallCallback_GetPriority
,
419 InstallCallback_OnLowResource
,
420 InstallCallback_OnProgress
,
421 InstallCallback_OnStopBinding
,
422 InstallCallback_GetBindInfo
,
423 InstallCallback_OnDataAvailable
,
424 InstallCallback_OnObjectAvailable
427 static IBindStatusCallback InstallCallback
= { &InstallCallbackVtbl
};
429 static LPWSTR
get_url(void)
433 DWORD size
= INTERNET_MAX_URL_LENGTH
*sizeof(WCHAR
);
437 static const WCHAR wszGeckoUrl
[] = {'G','e','c','k','o','U','r','l',0};
438 static const WCHAR httpW
[] = {'h','t','t','p'};
439 static const WCHAR arch_formatW
[] = {'?','a','r','c','h','='};
440 static const WCHAR v_formatW
[] = {'&','v','='};
442 /* @@ Wine registry key: HKCU\Software\Wine\MSHTML */
443 res
= RegOpenKeyW(HKEY_CURRENT_USER
, mshtml_keyW
, &hkey
);
444 if(res
!= ERROR_SUCCESS
)
447 url
= heap_alloc(size
);
448 returned_size
= size
;
450 res
= RegQueryValueExW(hkey
, wszGeckoUrl
, NULL
, &type
, (LPBYTE
)url
, &returned_size
);
452 if(res
!= ERROR_SUCCESS
|| type
!= REG_SZ
) {
457 if(returned_size
> sizeof(httpW
) && !memcmp(url
, httpW
, sizeof(httpW
))) {
461 memcpy(url
+len
, arch_formatW
, sizeof(arch_formatW
));
462 len
+= sizeof(arch_formatW
)/sizeof(WCHAR
);
463 len
+= MultiByteToWideChar(CP_ACP
, 0, GECKO_ARCH
, sizeof(GECKO_ARCH
), url
+len
, size
/sizeof(WCHAR
)-len
)-1;
464 memcpy(url
+len
, v_formatW
, sizeof(v_formatW
));
465 len
+= sizeof(v_formatW
)/sizeof(WCHAR
);
466 MultiByteToWideChar(CP_ACP
, 0, GECKO_VERSION
, -1, url
+len
, size
/sizeof(WCHAR
)-len
);
469 TRACE("Got URL %s\n", debugstr_w(url
));
473 static DWORD WINAPI
download_proc(PVOID arg
)
480 CreateURLMoniker(NULL
, url
, &mon
);
484 CreateAsyncBindCtx(0, &InstallCallback
, 0, &bctx
);
486 hres
= IMoniker_BindToStorage(mon
, bctx
, NULL
, &IID_IStream
, (void**)&str
);
487 IBindCtx_Release(bctx
);
489 ERR("BindToStorage failed: %08x\n", hres
);
494 IStream_Release(str
);
499 static INT_PTR CALLBACK
installer_proc(HWND hwnd
, UINT msg
, WPARAM wParam
, LPARAM lParam
)
503 ShowWindow(GetDlgItem(hwnd
, ID_DWL_PROGRESS
), SW_HIDE
);
504 install_dialog
= hwnd
;
514 ShowWindow(GetDlgItem(hwnd
, ID_DWL_PROGRESS
), SW_SHOW
);
515 EnableWindow(GetDlgItem(hwnd
, ID_DWL_INSTALL
), 0);
516 EnableWindow(GetDlgItem(hwnd
, IDCANCEL
), 0); /* FIXME */
517 CreateThread(NULL
, 0, download_proc
, NULL
, 0, NULL
);
525 BOOL
install_wine_gecko(BOOL silent
)
532 SetLastError(ERROR_SUCCESS
);
533 hsem
= CreateSemaphoreA( NULL
, 0, 1, "mshtml_install_semaphore");
535 if(GetLastError() == ERROR_ALREADY_EXISTS
) {
536 WaitForSingleObject(hsem
, INFINITE
);
539 * Try to find Gecko .cab file in following order:
540 * - directory stored in GeckoCabDir value of HKCU/Software/MSHTML key
542 * - download from URL stored in GeckoUrl value of HKCU/Software/MSHTML key
544 if(!install_from_registered_dir()
545 && !install_from_default_dir()
546 && !silent
&& (url
= get_url()))
547 DialogBoxW(hInst
, MAKEINTRESOURCEW(ID_DWL_DIALOG
), 0, installer_proc
);
550 ReleaseSemaphore(hsem
, 1, NULL
);