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