dlls/mshtml/install.c

   1 /*
   2  * Copyright 2006-2007 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 #ifdef HAVE_UNISTD_H
  24 # include <unistd.h>
  25 #endif
  26
  27 #define COBJMACROS
  28 #define NONAMELESSUNION
  29 #define NONAMELESSSTRUCT
  30
  31 #include "windef.h"
  32 #include "winbase.h"
  33 #include "winuser.h"
  34 #include "winreg.h"
  35 #include "ole2.h"
  36 #include "commctrl.h"
  37 #include "advpub.h"
  38 #include "wininet.h"
  39 #include "shellapi.h"
  40
  41 #include "wine/debug.h"
  42 #include "wine/unicode.h"
  43 #include "wine/library.h"
  44
  45 #include "mshtml_private.h"
  46 #include "resource.h"
  47
  48 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  49
  50 #define GECKO_FILE_NAME "wine_gecko-" GECKO_VERSION ".cab"
  51
  52 static const WCHAR mshtml_keyW[] =
  53     {'S','o','f','t','w','a','r','e',
  54      '\\','W','i','n','e',
  55      '\\','M','S','H','T','M','L',0};
  56
  57 static HWND install_dialog = NULL;
  58 static LPWSTR tmp_file_name = NULL;
  59 static HANDLE tmp_file = INVALID_HANDLE_VALUE;
  60 static LPWSTR url = NULL;
  61
  62 static void clean_up(void)
  63 {
  64     if(tmp_file != INVALID_HANDLE_VALUE)
  65         CloseHandle(tmp_file);
  66
  67     if(tmp_file_name) {
  68         DeleteFileW(tmp_file_name);
  69         heap_free(tmp_file_name);
  70         tmp_file_name = NULL;
  71     }
  72
  73     if(tmp_file != INVALID_HANDLE_VALUE) {
  74         CloseHandle(tmp_file);
  75         tmp_file = INVALID_HANDLE_VALUE;
  76     }
  77
  78     if(install_dialog)
  79         EndDialog(install_dialog, 0);
  80 }
  81
  82 static void set_status(DWORD id)
  83 {
  84     HWND status = GetDlgItem(install_dialog, ID_DWL_STATUS);
  85     WCHAR buf[64];
  86
  87     LoadStringW(hInst, id, buf, sizeof(buf)/sizeof(WCHAR));
  88     SendMessageW(status, WM_SETTEXT, 0, (LPARAM)buf);
  89 }
  90
  91 static void set_registry(LPCSTR install_dir)
  92 {
  93     WCHAR mshtml_key[100];
  94     LPWSTR gecko_path;
  95     HKEY hkey;
  96     DWORD res, len, size;
  97
  98     static const WCHAR wszGeckoPath[] = {'G','e','c','k','o','P','a','t','h',0};
  99     static const WCHAR wszWineGecko[] = {'w','i','n','e','_','g','e','c','k','o',0};
 100
 101     memcpy(mshtml_key, mshtml_keyW, sizeof(mshtml_keyW));
 102     mshtml_key[sizeof(mshtml_keyW)/sizeof(WCHAR)-1] = '\\';
 103     MultiByteToWideChar(CP_ACP, 0, GECKO_VERSION, sizeof(GECKO_VERSION),
 104             mshtml_key+sizeof(mshtml_keyW)/sizeof(WCHAR),
 105             (sizeof(mshtml_key)-sizeof(mshtml_keyW))/sizeof(WCHAR));
 106
 107     /* @@ Wine registry key: HKCU\Software\Wine\MSHTML\<version> */
 108     res = RegCreateKeyW(HKEY_CURRENT_USER, mshtml_key, &hkey);
 109     if(res != ERROR_SUCCESS) {
 110         ERR("Faild to create MSHTML key: %d\n", res);
 111         return;
 112     }
 113
 114     len = MultiByteToWideChar(CP_ACP, 0, install_dir, -1, NULL, 0)-1;
 115     gecko_path = heap_alloc((len+1)*sizeof(WCHAR)+sizeof(wszWineGecko));
 116     MultiByteToWideChar(CP_ACP, 0, install_dir, -1, gecko_path, (len+1)*sizeof(WCHAR));
 117
 118     if (len && gecko_path[len-1] != '\\')
 119         gecko_path[len++] = '\\';
 120
 121     memcpy(gecko_path+len, wszWineGecko, sizeof(wszWineGecko));
 122
 123     size = len*sizeof(WCHAR)+sizeof(wszWineGecko);
 124     res = RegSetValueExW(hkey, wszGeckoPath, 0, REG_SZ, (LPVOID)gecko_path,
 125                        len*sizeof(WCHAR)+sizeof(wszWineGecko));
 126     heap_free(gecko_path);
 127     RegCloseKey(hkey);
 128     if(res != ERROR_SUCCESS)
 129         ERR("Failed to set GeckoPath value: %08x\n", res);
 130 }
 131
 132 static BOOL install_cab(LPCWSTR file_name)
 133 {
 134     HMODULE advpack;
 135     char install_dir[MAX_PATH];
 136     typeof(ExtractFilesA) *pExtractFilesA;
 137     LPSTR file_name_a;
 138     DWORD res, len;
 139     HRESULT hres;
 140
 141     static const WCHAR wszAdvpack[] = {'a','d','v','p','a','c','k','.','d','l','l',0};
 142
 143     TRACE("(%s)\n", debugstr_w(file_name));
 144
 145     GetWindowsDirectoryA(install_dir, sizeof(install_dir));
 146     strcat(install_dir, "\\gecko\\");
 147     res = CreateDirectoryA(install_dir, NULL);
 148     if(!res && GetLastError() != ERROR_ALREADY_EXISTS) {
 149         ERR("Could not create directory: %08u\n", GetLastError());
 150         return FALSE;
 151     }
 152
 153     strcat(install_dir, GECKO_VERSION);
 154     res = CreateDirectoryA(install_dir, NULL);
 155     if(!res && GetLastError() != ERROR_ALREADY_EXISTS) {
 156         ERR("Could not create directory: %08u\n", GetLastError());
 157         return FALSE;
 158     }
 159
 160     advpack = LoadLibraryW(wszAdvpack);
 161     pExtractFilesA = (typeof(ExtractFilesA)*)GetProcAddress(advpack, "ExtractFiles");
 162
 163     len = WideCharToMultiByte(CP_ACP, 0, file_name, -1, NULL, 0, NULL, NULL);
 164     file_name_a = heap_alloc(len);
 165     WideCharToMultiByte(CP_ACP, 0, file_name, -1, file_name_a, -1, NULL, NULL);
 166
 167     /* FIXME: Use unicode version (not yet implemented) */
 168     hres = pExtractFilesA(file_name_a, install_dir, 0, NULL, NULL, 0);
 169     FreeLibrary(advpack);
 170     heap_free(file_name_a);
 171     if(FAILED(hres)) {
 172         ERR("Could not extract package: %08x\n", hres);
 173         clean_up();
 174         return FALSE;
 175     }
 176
 177     set_registry(install_dir);
 178     clean_up();
 179
 180     return TRUE;
 181 }
 182
 183 static BOOL install_from_unix_file(const char *file_name)
 184 {
 185     LPWSTR dos_file_name;
 186     int fd;
 187     BOOL ret;
 188
 189     static WCHAR *(*wine_get_dos_file_name)(const char*);
 190     static const WCHAR kernel32W[] = {'k','e','r','n','e','l','3','2','.','d','l','l',0};
 191
 192     fd = open(file_name, O_RDONLY);
 193     if(fd == -1) {
 194         TRACE("%s not found\n", debugstr_a(file_name));
 195         return FALSE;
 196     }
 197
 198     close(fd);
 199
 200     if(!wine_get_dos_file_name) {
 201         wine_get_dos_file_name = (void*)GetProcAddress(GetModuleHandleW(kernel32W), "wine_get_dos_file_name");
 202         if(!wine_get_dos_file_name) {
 203             ERR("Could not get wine_get_dos_file_name function.\n");
 204             return FALSE;
 205         }
 206     }
 207
 208     dos_file_name = wine_get_dos_file_name(file_name);
 209     if(!file_name) {
 210         ERR("Could not get dos file name of %s\n", debugstr_a(file_name));
 211         return FALSE;
 212     }
 213
 214     ret = install_cab(dos_file_name);
 215
 216     heap_free(dos_file_name);
 217     return ret;
 218 }
 219
 220 static BOOL install_from_registered_dir(void)
 221 {
 222     char *file_name;
 223     HKEY hkey;
 224     DWORD res, type, size = MAX_PATH;
 225     BOOL ret;
 226
 227     /* @@ Wine registry key: HKCU\Software\Wine\MSHTML */
 228     res = RegOpenKeyW(HKEY_CURRENT_USER, mshtml_keyW, &hkey);
 229     if(res != ERROR_SUCCESS)
 230         return FALSE;
 231
 232     file_name = heap_alloc(size+sizeof(GECKO_FILE_NAME));
 233     res = RegQueryValueExA(hkey, "GeckoCabDir", NULL, &type, (PBYTE)file_name, &size);
 234     if(res == ERROR_MORE_DATA) {
 235         file_name = heap_realloc(file_name, size+sizeof(GECKO_FILE_NAME));
 236         res = RegQueryValueExA(hkey, "GeckoCabDir", NULL, &type, (PBYTE)file_name, &size);
 237     }
 238     RegCloseKey(hkey);
 239     if(res != ERROR_SUCCESS || type != REG_SZ) {
 240         heap_free(file_name);
 241         return FALSE;
 242     }
 243
 244     strcat(file_name, GECKO_FILE_NAME);
 245
 246     TRACE("Trying %s\n", debugstr_a(file_name));
 247
 248     ret = install_from_unix_file(file_name);
 249
 250     heap_free(file_name);
 251     return ret;
 252 }
 253
 254 static BOOL install_from_default_dir(void)
 255 {
 256     const char *data_dir, *subdir;
 257     char *file_name;
 258     int len, len2;
 259     BOOL ret;
 260
 261     if((data_dir = wine_get_data_dir()))
 262         subdir = "/gecko/";
 263     else if((data_dir = wine_get_build_dir()))
 264         subdir = "/../gecko/";
 265     else
 266         return FALSE;
 267
 268     len = strlen(data_dir);
 269     len2 = strlen(subdir);
 270
 271     file_name = heap_alloc(len+len2+sizeof(GECKO_FILE_NAME));
 272     memcpy(file_name, data_dir, len);
 273     memcpy(file_name+len, subdir, len2);
 274     memcpy(file_name+len+len2, GECKO_FILE_NAME, sizeof(GECKO_FILE_NAME));
 275
 276     ret = install_from_unix_file(file_name);
 277
 278     heap_free(file_name);
 279     return ret;
 280 }
 281
 282 static HRESULT WINAPI InstallCallback_QueryInterface(IBindStatusCallback *iface,
 283                                                      REFIID riid, void **ppv)
 284 {
 285     if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IBindStatusCallback, riid)) {
 286         *ppv = iface;
 287         return S_OK;
 288     }
 289
 290     return E_INVALIDARG;
 291 }
 292
 293 static ULONG WINAPI InstallCallback_AddRef(IBindStatusCallback *iface)
 294 {
 295     return 2;
 296 }
 297
 298 static ULONG WINAPI InstallCallback_Release(IBindStatusCallback *iface)
 299 {
 300     return 1;
 301 }
 302
 303 static HRESULT WINAPI InstallCallback_OnStartBinding(IBindStatusCallback *iface,
 304         DWORD dwReserved, IBinding *pib)
 305 {
 306     WCHAR tmp_dir[MAX_PATH];
 307
 308     set_status(IDS_DOWNLOADING);
 309
 310     GetTempPathW(sizeof(tmp_dir)/sizeof(WCHAR), tmp_dir);
 311
 312     tmp_file_name = heap_alloc(MAX_PATH*sizeof(WCHAR));
 313     GetTempFileNameW(tmp_dir, NULL, 0, tmp_file_name);
 314
 315     TRACE("creating temp file %s\n", debugstr_w(tmp_file_name));
 316
 317     tmp_file = CreateFileW(tmp_file_name, GENERIC_WRITE, 0, NULL,
 318                            CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
 319
 320     if(tmp_file == INVALID_HANDLE_VALUE) {
 321         ERR("Could not create file: %d\n", GetLastError());
 322         clean_up();
 323         return E_FAIL;
 324     }
 325
 326     return S_OK;
 327 }
 328
 329 static HRESULT WINAPI InstallCallback_GetPriority(IBindStatusCallback *iface,
 330         LONG *pnPriority)
 331 {
 332     return E_NOTIMPL;
 333 }
 334
 335 static HRESULT WINAPI InstallCallback_OnLowResource(IBindStatusCallback *iface,
 336        DWORD dwReserved)
 337 {
 338     return E_NOTIMPL;
 339 }
 340
 341 static HRESULT WINAPI InstallCallback_OnProgress(IBindStatusCallback *iface, ULONG ulProgress,
 342         ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText)
 343 {
 344     HWND progress = GetDlgItem(install_dialog, ID_DWL_PROGRESS);
 345
 346     if(ulProgressMax)
 347         SendMessageW(progress, PBM_SETRANGE32, 0, ulProgressMax);
 348     if(ulProgress)
 349         SendMessageW(progress, PBM_SETPOS, ulProgress, 0);
 350
 351     return S_OK;
 352 }
 353
 354 static HRESULT WINAPI InstallCallback_OnStopBinding(IBindStatusCallback *iface,
 355         HRESULT hresult, LPCWSTR szError)
 356 {
 357     if(FAILED(hresult)) {
 358         ERR("Binding failed %08x\n", hresult);
 359         clean_up();
 360         return S_OK;
 361     }
 362
 363     CloseHandle(tmp_file);
 364     tmp_file = INVALID_HANDLE_VALUE;
 365
 366     set_status(IDS_INSTALLING);
 367
 368     install_cab(tmp_file_name);
 369
 370     return S_OK;
 371 }
 372
 373 static HRESULT WINAPI InstallCallback_GetBindInfo(IBindStatusCallback *iface,
 374         DWORD* grfBINDF, BINDINFO* pbindinfo)
 375 {
 376     /* FIXME */
 377     *grfBINDF = 0;
 378     return S_OK;
 379 }
 380
 381 static HRESULT WINAPI InstallCallback_OnDataAvailable(IBindStatusCallback *iface, DWORD grfBSCF,
 382         DWORD dwSize, FORMATETC* pformatetc, STGMEDIUM* pstgmed)
 383 {
 384     IStream *str = pstgmed->u.pstm;
 385     BYTE buf[1024];
 386     DWORD size;
 387     HRESULT hres;
 388
 389     do {
 390         size = 0;
 391         hres = IStream_Read(str, buf, sizeof(buf), &size);
 392         if(size)
 393             WriteFile(tmp_file, buf, size, NULL, NULL);
 394     }while(hres == S_OK);
 395
 396     return S_OK;
 397 }
 398
 399 static HRESULT WINAPI InstallCallback_OnObjectAvailable(IBindStatusCallback *iface,
 400         REFIID riid, IUnknown* punk)
 401 {
 402     ERR("\n");
 403     return E_NOTIMPL;
 404 }
 405
 406 static const IBindStatusCallbackVtbl InstallCallbackVtbl = {
 407     InstallCallback_QueryInterface,
 408     InstallCallback_AddRef,
 409     InstallCallback_Release,
 410     InstallCallback_OnStartBinding,
 411     InstallCallback_GetPriority,
 412     InstallCallback_OnLowResource,
 413     InstallCallback_OnProgress,
 414     InstallCallback_OnStopBinding,
 415     InstallCallback_GetBindInfo,
 416     InstallCallback_OnDataAvailable,
 417     InstallCallback_OnObjectAvailable
 418 };
 419
 420 static IBindStatusCallback InstallCallback = { &InstallCallbackVtbl };
 421
 422 static LPWSTR get_url(void)
 423 {
 424     HKEY hkey;
 425     DWORD res, type;
 426     DWORD size = INTERNET_MAX_URL_LENGTH*sizeof(WCHAR);
 427     LPWSTR url;
 428
 429     static const WCHAR wszGeckoUrl[] = {'G','e','c','k','o','U','r','l',0};
 430     static const WCHAR httpW[] = {'h','t','t','p'};
 431     static const WCHAR v_formatW[] = {'?','v','=',0};
 432
 433     /* @@ Wine registry key: HKCU\Software\Wine\MSHTML */
 434     res = RegOpenKeyW(HKEY_CURRENT_USER, mshtml_keyW, &hkey);
 435     if(res != ERROR_SUCCESS)
 436         return NULL;
 437
 438     url = heap_alloc(size);
 439
 440     res = RegQueryValueExW(hkey, wszGeckoUrl, NULL, &type, (LPBYTE)url, &size);
 441     RegCloseKey(hkey);
 442     if(res != ERROR_SUCCESS || type != REG_SZ) {
 443         heap_free(url);
 444         return NULL;
 445     }
 446
 447     if(size > sizeof(httpW) && !memcmp(url, httpW, sizeof(httpW))) {
 448         strcatW(url, v_formatW);
 449         MultiByteToWideChar(CP_ACP, 0, GECKO_VERSION, -1, url+strlenW(url), -1);
 450     }
 451
 452     TRACE("Got URL %s\n", debugstr_w(url));
 453     return url;
 454 }
 455
 456 static DWORD WINAPI download_proc(PVOID arg)
 457 {
 458     IMoniker *mon;
 459     IBindCtx *bctx;
 460     IStream *str = NULL;
 461     HRESULT hres;
 462
 463     CreateURLMoniker(NULL, url, &mon);
 464     heap_free(url);
 465     url = NULL;
 466
 467     CreateAsyncBindCtx(0, &InstallCallback, 0, &bctx);
 468
 469     hres = IMoniker_BindToStorage(mon, bctx, NULL, &IID_IStream, (void**)&str);
 470     IBindCtx_Release(bctx);
 471     if(FAILED(hres)) {
 472         ERR("BindToStorage failed: %08x\n", hres);
 473         return 0;
 474     }
 475
 476     if(str)
 477         IStream_Release(str);
 478
 479     return 0;
 480 }
 481
 482 static INT_PTR CALLBACK installer_proc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
 483 {
 484     switch(msg) {
 485     case WM_INITDIALOG:
 486         ShowWindow(GetDlgItem(hwnd, ID_DWL_PROGRESS), SW_HIDE);
 487         install_dialog = hwnd;
 488         return TRUE;
 489
 490     case WM_COMMAND:
 491         switch(wParam) {
 492         case IDCANCEL:
 493             EndDialog(hwnd, 0);
 494             return FALSE;
 495
 496         case ID_DWL_INSTALL:
 497             ShowWindow(GetDlgItem(hwnd, ID_DWL_PROGRESS), SW_SHOW);
 498             EnableWindow(GetDlgItem(hwnd, ID_DWL_INSTALL), 0);
 499             EnableWindow(GetDlgItem(hwnd, IDCANCEL), 0); /* FIXME */
 500             CreateThread(NULL, 0, download_proc, NULL, 0, NULL);
 501             return FALSE;
 502         }
 503     }
 504
 505     return FALSE;
 506 }
 507
 508 BOOL install_wine_gecko(BOOL silent)
 509 {
 510     HANDLE hsem;
 511
 512     SetLastError(ERROR_SUCCESS);
 513     hsem = CreateSemaphoreA( NULL, 0, 1, "mshtml_install_semaphore");
 514
 515     if(GetLastError() == ERROR_ALREADY_EXISTS) {
 516         WaitForSingleObject(hsem, INFINITE);
 517     }else {
 518         /*
 519          * Try to find Gecko .cab file in following order:
 520          * - directory stored in GeckoCabDir value of HKCU/Software/MSHTML key
 521          * - $datadir/gecko
 522          * - download from URL stored in GeckoUrl value of HKCU/Software/MSHTML key
 523          */
 524         if(!install_from_registered_dir()
 525            && !install_from_default_dir()
 526            && !silent && (url = get_url()))
 527             DialogBoxW(hInst, MAKEINTRESOURCEW(ID_DWL_DIALOG), 0, installer_proc);
 528     }
 529
 530     ReleaseSemaphore(hsem, 1, NULL);
 531     CloseHandle(hsem);
 532
 533     return TRUE;
 534 }