programs/wusa/main.c

   1 /*
   2  * Copyright 2012 Austin English
   3  * Copyright 2015 Michael Müller
   4  * Copyright 2015 Sebastian Lackner
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <windows.h>
  22 #include <fcntl.h>
  23 #include <fdi.h>
  24 #include <shlwapi.h>
  25
  26 #include "wine/debug.h"
  27 #include "wine/list.h"
  28 #include "wusa.h"
  29
  30 WINE_DEFAULT_DEBUG_CHANNEL(wusa);
  31
  32 struct strbuf
  33 {
  34     WCHAR *buf;
  35     DWORD pos;
  36     DWORD len;
  37 };
  38
  39 struct installer_tempdir
  40 {
  41     struct list entry;
  42     WCHAR *path;
  43 };
  44
  45 struct installer_state
  46 {
  47     BOOL norestart;
  48     BOOL quiet;
  49     struct list tempdirs;
  50     struct list assemblies;
  51     struct list updates;
  52 };
  53
  54 static void * CDECL cabinet_alloc(ULONG cb)
  55 {
  56     return heap_alloc(cb);
  57 }
  58
  59 static void CDECL cabinet_free(void *pv)
  60 {
  61     heap_free(pv);
  62 }
  63
  64 static INT_PTR CDECL cabinet_open(char *pszFile, int oflag, int pmode)
  65 {
  66     DWORD dwAccess = 0;
  67     DWORD dwShareMode = 0;
  68     DWORD dwCreateDisposition = OPEN_EXISTING;
  69
  70     switch (oflag & _O_ACCMODE)
  71     {
  72     case _O_RDONLY:
  73         dwAccess = GENERIC_READ;
  74         dwShareMode = FILE_SHARE_READ | FILE_SHARE_DELETE;
  75         break;
  76     case _O_WRONLY:
  77         dwAccess = GENERIC_WRITE;
  78         dwShareMode = FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE;
  79         break;
  80     case _O_RDWR:
  81         dwAccess = GENERIC_READ | GENERIC_WRITE;
  82         dwShareMode = FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE;
  83         break;
  84     }
  85
  86     if ((oflag & (_O_CREAT | _O_EXCL)) == (_O_CREAT | _O_EXCL))
  87         dwCreateDisposition = CREATE_NEW;
  88     else if (oflag & _O_CREAT)
  89         dwCreateDisposition = CREATE_ALWAYS;
  90
  91     return (INT_PTR)CreateFileA(pszFile, dwAccess, dwShareMode, NULL, dwCreateDisposition, 0, NULL);
  92 }
  93
  94 static UINT CDECL cabinet_read(INT_PTR hf, void *pv, UINT cb)
  95 {
  96     HANDLE handle = (HANDLE)hf;
  97     DWORD read;
  98
  99     if (ReadFile(handle, pv, cb, &read, NULL))
 100         return read;
 101
 102     return 0;
 103 }
 104
 105 static UINT CDECL cabinet_write(INT_PTR hf, void *pv, UINT cb)
 106 {
 107     HANDLE handle = (HANDLE)hf;
 108     DWORD written;
 109
 110     if (WriteFile(handle, pv, cb, &written, NULL))
 111         return written;
 112
 113     return 0;
 114 }
 115
 116 static int CDECL cabinet_close(INT_PTR hf)
 117 {
 118     HANDLE handle = (HANDLE)hf;
 119     return CloseHandle(handle) ? 0 : -1;
 120 }
 121
 122 static LONG CDECL cabinet_seek(INT_PTR hf, LONG dist, int seektype)
 123 {
 124     HANDLE handle = (HANDLE)hf;
 125     /* flags are compatible and so are passed straight through */
 126     return SetFilePointer(handle, dist, NULL, seektype);
 127 }
 128
 129 static WCHAR *path_combine(const WCHAR *path, const WCHAR *filename)
 130 {
 131     WCHAR *result;
 132     DWORD length;
 133
 134     if (!path || !filename) return NULL;
 135     length = lstrlenW(path) + lstrlenW(filename) + 2;
 136     if (!(result = heap_alloc(length * sizeof(WCHAR)))) return NULL;
 137
 138     lstrcpyW(result, path);
 139     if (result[0] && result[lstrlenW(result) - 1] != '\\') lstrcatW(result, L"\\");
 140     lstrcatW(result, filename);
 141     return result;
 142 }
 143
 144 static WCHAR *get_uncompressed_path(PFDINOTIFICATION pfdin)
 145 {
 146     WCHAR *file = strdupAtoW(pfdin->psz1);
 147     WCHAR *path = path_combine(pfdin->pv, file);
 148     heap_free(file);
 149     return path;
 150 }
 151
 152 static BOOL is_directory(const WCHAR *path)
 153 {
 154     DWORD attrs = GetFileAttributesW(path);
 155     if (attrs == INVALID_FILE_ATTRIBUTES) return FALSE;
 156     return (attrs & FILE_ATTRIBUTE_DIRECTORY) != 0;
 157 }
 158
 159 static BOOL create_directory(const WCHAR *path)
 160 {
 161     if (is_directory(path)) return TRUE;
 162     if (CreateDirectoryW(path, NULL)) return TRUE;
 163     return (GetLastError() == ERROR_ALREADY_EXISTS);
 164 }
 165
 166 static BOOL create_parent_directory(const WCHAR *filename)
 167 {
 168     WCHAR *p, *path = strdupW(filename);
 169     BOOL ret = FALSE;
 170
 171     if (!path) return FALSE;
 172     if (!PathRemoveFileSpecW(path)) goto done;
 173     if (is_directory(path))
 174     {
 175         ret = TRUE;
 176         goto done;
 177     }
 178
 179     for (p = path; *p; p++)
 180     {
 181         if (*p != '\\') continue;
 182         *p = 0;
 183         if (!create_directory(path)) goto done;
 184         *p = '\\';
 185     }
 186     ret = create_directory(path);
 187
 188 done:
 189     heap_free(path);
 190     return ret;
 191 }
 192
 193 static INT_PTR cabinet_copy_file(FDINOTIFICATIONTYPE fdint, PFDINOTIFICATION pfdin)
 194 {
 195     HANDLE handle = INVALID_HANDLE_VALUE;
 196     WCHAR *file;
 197     DWORD attrs;
 198
 199     if (!(file = get_uncompressed_path(pfdin)))
 200         return -1;
 201
 202     TRACE("Extracting %s -> %s\n", debugstr_a(pfdin->psz1), debugstr_w(file));
 203
 204     if (create_parent_directory(file))
 205     {
 206         attrs = pfdin->attribs;
 207         if (!attrs) attrs = FILE_ATTRIBUTE_NORMAL;
 208         handle = CreateFileW(file, GENERIC_READ | GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, attrs, NULL);
 209     }
 210
 211     heap_free(file);
 212     return (handle != INVALID_HANDLE_VALUE) ? (INT_PTR)handle : -1;
 213 }
 214
 215 static INT_PTR cabinet_close_file_info(FDINOTIFICATIONTYPE fdint, PFDINOTIFICATION pfdin)
 216 {
 217     HANDLE handle = (HANDLE)pfdin->hf;
 218     CloseHandle(handle);
 219     return 1;
 220 }
 221
 222 static INT_PTR CDECL cabinet_notify(FDINOTIFICATIONTYPE fdint, PFDINOTIFICATION pfdin)
 223 {
 224     switch (fdint)
 225     {
 226     case fdintPARTIAL_FILE:
 227         FIXME("fdintPARTIAL_FILE not implemented\n");
 228         return 0;
 229
 230     case fdintNEXT_CABINET:
 231         FIXME("fdintNEXT_CABINET not implemented\n");
 232         return 0;
 233
 234     case fdintCOPY_FILE:
 235         return cabinet_copy_file(fdint, pfdin);
 236
 237     case fdintCLOSE_FILE_INFO:
 238         return cabinet_close_file_info(fdint, pfdin);
 239
 240     default:
 241         return 0;
 242     }
 243 }
 244
 245 static BOOL extract_cabinet(const WCHAR *filename, const WCHAR *destination)
 246 {
 247     char *filenameA = NULL;
 248     BOOL ret = FALSE;
 249     HFDI hfdi;
 250     ERF erf;
 251
 252     hfdi = FDICreate(cabinet_alloc, cabinet_free, cabinet_open, cabinet_read,
 253                      cabinet_write, cabinet_close, cabinet_seek, 0, &erf);
 254     if (!hfdi) return FALSE;
 255
 256     if ((filenameA = strdupWtoA(filename)))
 257     {
 258         ret = FDICopy(hfdi, filenameA, NULL, 0, cabinet_notify, NULL, (void *)destination);
 259         heap_free(filenameA);
 260     }
 261
 262     FDIDestroy(hfdi);
 263     return ret;
 264 }
 265
 266 static const WCHAR *create_temp_directory(struct installer_state *state)
 267 {
 268     static UINT id;
 269     struct installer_tempdir *entry;
 270     WCHAR tmp[MAX_PATH];
 271
 272     if (!GetTempPathW(ARRAY_SIZE(tmp), tmp)) return NULL;
 273     if (!(entry = heap_alloc(sizeof(*entry)))) return NULL;
 274     if (!(entry->path = heap_alloc((MAX_PATH + 20) * sizeof(WCHAR))))
 275     {
 276         heap_free(entry);
 277         return NULL;
 278     }
 279     for (;;)
 280     {
 281         if (!GetTempFileNameW(tmp, L"msu", ++id, entry->path))
 282         {
 283             heap_free(entry->path);
 284             heap_free(entry);
 285             return NULL;
 286         }
 287         if (CreateDirectoryW(entry->path, NULL)) break;
 288     }
 289
 290     list_add_tail(&state->tempdirs, &entry->entry);
 291     return entry->path;
 292 }
 293
 294 static BOOL delete_directory(const WCHAR *path)
 295 {
 296     WIN32_FIND_DATAW data;
 297     WCHAR *full_path;
 298     HANDLE search;
 299
 300     if (!(full_path = path_combine(path, L"*"))) return FALSE;
 301     search = FindFirstFileW(full_path, &data);
 302     heap_free(full_path);
 303
 304     if (search != INVALID_HANDLE_VALUE)
 305     {
 306         do
 307         {
 308             if (!wcscmp(data.cFileName, L".")) continue;
 309             if (!wcscmp(data.cFileName, L"..")) continue;
 310             if (!(full_path = path_combine(path, data.cFileName))) continue;
 311             if (data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
 312                 delete_directory(full_path);
 313             else
 314                 DeleteFileW(full_path);
 315             heap_free(full_path);
 316         }
 317         while (FindNextFileW(search, &data));
 318         FindClose(search);
 319     }
 320
 321     return RemoveDirectoryW(path);
 322 }
 323
 324 static void installer_cleanup(struct installer_state *state)
 325 {
 326     struct installer_tempdir *tempdir, *tempdir2;
 327     struct assembly_entry *assembly, *assembly2;
 328     struct dependency_entry *dependency, *dependency2;
 329
 330     LIST_FOR_EACH_ENTRY_SAFE(tempdir, tempdir2, &state->tempdirs, struct installer_tempdir, entry)
 331     {
 332         list_remove(&tempdir->entry);
 333         delete_directory(tempdir->path);
 334         heap_free(tempdir->path);
 335         heap_free(tempdir);
 336     }
 337     LIST_FOR_EACH_ENTRY_SAFE(assembly, assembly2, &state->assemblies, struct assembly_entry, entry)
 338     {
 339         list_remove(&assembly->entry);
 340         free_assembly(assembly);
 341     }
 342     LIST_FOR_EACH_ENTRY_SAFE(dependency, dependency2, &state->updates, struct dependency_entry, entry)
 343     {
 344         list_remove(&dependency->entry);
 345         free_dependency(dependency);
 346     }
 347 }
 348
 349 static BOOL str_ends_with(const WCHAR *str, const WCHAR *suffix)
 350 {
 351     DWORD str_len = lstrlenW(str), suffix_len = lstrlenW(suffix);
 352     if (suffix_len > str_len) return FALSE;
 353     return !wcsicmp(str + str_len - suffix_len, suffix);
 354 }
 355
 356 static BOOL load_assemblies_from_cab(const WCHAR *filename, struct installer_state *state)
 357 {
 358     struct assembly_entry *assembly;
 359     const WCHAR *temp_path;
 360     WIN32_FIND_DATAW data;
 361     HANDLE search;
 362     WCHAR *path;
 363
 364     TRACE("Processing cab file %s\n", debugstr_w(filename));
 365
 366     if (!(temp_path = create_temp_directory(state))) return FALSE;
 367     if (!extract_cabinet(filename, temp_path))
 368     {
 369         ERR("Failed to extract %s\n", debugstr_w(filename));
 370         return FALSE;
 371     }
 372
 373     if (!(path = path_combine(temp_path, L"_manifest_.cix.xml"))) return FALSE;
 374     if (GetFileAttributesW(path) != INVALID_FILE_ATTRIBUTES)
 375     {
 376         FIXME("Cabinet uses proprietary msdelta file compression which is not (yet) supported\n");
 377         FIXME("Installation of msu file will most likely fail\n");
 378     }
 379     heap_free(path);
 380
 381     if (!(path = path_combine(temp_path, L"*"))) return FALSE;
 382     search = FindFirstFileW(path, &data);
 383     heap_free(path);
 384
 385     if (search != INVALID_HANDLE_VALUE)
 386     {
 387         do
 388         {
 389             if (data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) continue;
 390             if (!str_ends_with(data.cFileName, L".manifest") &&
 391                 !str_ends_with(data.cFileName, L".mum")) continue;
 392             if (!(path = path_combine(temp_path, data.cFileName))) continue;
 393             if ((assembly = load_manifest(path)))
 394                 list_add_tail(&state->assemblies, &assembly->entry);
 395             heap_free(path);
 396         }
 397         while (FindNextFileW(search, &data));
 398         FindClose(search);
 399     }
 400
 401     return TRUE;
 402 }
 403
 404 static BOOL compare_assembly_string(const WCHAR *str1, const WCHAR *str2)
 405 {
 406     return !wcscmp(str1, str2) || !wcscmp(str1, L"*") || !wcscmp(str2, L"*");
 407 }
 408
 409 static struct assembly_entry *lookup_assembly(struct list *manifest_list, struct assembly_identity *identity)
 410 {
 411     struct assembly_entry *assembly;
 412
 413     LIST_FOR_EACH_ENTRY(assembly, manifest_list, struct assembly_entry, entry)
 414     {
 415         if (wcsicmp(assembly->identity.name, identity->name)) continue;
 416         if (!compare_assembly_string(assembly->identity.architecture, identity->architecture)) continue;
 417         if (!compare_assembly_string(assembly->identity.language, identity->language)) continue;
 418         if (!compare_assembly_string(assembly->identity.pubkey_token, identity->pubkey_token)) continue;
 419         if (!compare_assembly_string(assembly->identity.version, identity->version))
 420         {
 421             WARN("Ignoring version difference for %s (expected %s, found %s)\n",
 422                  debugstr_w(identity->name), debugstr_w(identity->version), debugstr_w(assembly->identity.version));
 423         }
 424         return assembly;
 425     }
 426
 427     return NULL;
 428 }
 429
 430 static WCHAR *get_assembly_source(struct assembly_entry *assembly)
 431 {
 432     WCHAR *p, *path = strdupW(assembly->filename);
 433     if (path && (p = wcsrchr(path, '.'))) *p = 0;
 434     return path;
 435 }
 436
 437 static BOOL strbuf_init(struct strbuf *buf)
 438 {
 439     buf->pos = 0;
 440     buf->len = 64;
 441     buf->buf = heap_alloc(buf->len * sizeof(WCHAR));
 442     return buf->buf != NULL;
 443 }
 444
 445 static void strbuf_free(struct strbuf *buf)
 446 {
 447     heap_free(buf->buf);
 448     buf->buf = NULL;
 449 }
 450
 451 static BOOL strbuf_append(struct strbuf *buf, const WCHAR *str, DWORD len)
 452 {
 453     DWORD new_len;
 454     WCHAR *new_buf;
 455
 456     if (!buf->buf) return FALSE;
 457     if (!str) return TRUE;
 458
 459     if (len == ~0U) len = lstrlenW(str);
 460     if (buf->pos + len + 1 > buf->len)
 461     {
 462         new_len = max(buf->pos + len + 1, buf->len * 2);
 463         new_buf = heap_realloc(buf->buf, new_len * sizeof(WCHAR));
 464         if (!new_buf)
 465         {
 466             strbuf_free(buf);
 467             return FALSE;
 468         }
 469         buf->buf = new_buf;
 470         buf->len = new_len;
 471     }
 472
 473     memcpy(&buf->buf[buf->pos], str, len * sizeof(WCHAR));
 474     buf->buf[buf->pos + len] = 0;
 475     buf->pos += len;
 476     return TRUE;
 477 }
 478
 479 static WCHAR *lookup_expression(struct assembly_entry *assembly, const WCHAR *key)
 480 {
 481     WCHAR path[MAX_PATH];
 482
 483     if (!wcscmp(key, L"runtime.system32"))
 484     {
 485 #ifdef __x86_64__
 486         if (!wcscmp(assembly->identity.architecture, L"x86"))
 487         {
 488             GetSystemWow64DirectoryW(path, ARRAY_SIZE(path));
 489             return strdupW(path);
 490         }
 491 #endif
 492         GetSystemDirectoryW(path, ARRAY_SIZE(path));
 493         return strdupW(path);
 494     }
 495     if (!wcscmp(key, L"runtime.windows"))
 496     {
 497         GetWindowsDirectoryW(path, ARRAY_SIZE(path));
 498         return strdupW(path);
 499     }
 500
 501     FIXME("Unknown expression %s\n", debugstr_w(key));
 502     return NULL;
 503 }
 504
 505 static WCHAR *expand_expression(struct assembly_entry *assembly, const WCHAR *expression)
 506 {
 507     const WCHAR *pos, *next;
 508     WCHAR *key, *value;
 509     struct strbuf buf;
 510
 511     if (!expression || !strbuf_init(&buf)) return NULL;
 512
 513     for (pos = expression; (next = wcsstr(pos, L"$(")); pos = next + 1)
 514     {
 515         strbuf_append(&buf, pos, next - pos);
 516         pos = next + 2;
 517         if (!(next = wcsstr(pos, L")")))
 518         {
 519             strbuf_append(&buf, L"$(", 2);
 520             break;
 521         }
 522
 523         if (!(key = strdupWn(pos, next - pos))) goto error;
 524         value = lookup_expression(assembly, key);
 525         heap_free(key);
 526         if (!value) goto error;
 527         strbuf_append(&buf, value, ~0U);
 528         heap_free(value);
 529     }
 530
 531     strbuf_append(&buf, pos, ~0U);
 532     return buf.buf;
 533
 534 error:
 535     FIXME("Couldn't resolve expression %s\n", debugstr_w(expression));
 536     strbuf_free(&buf);
 537     return NULL;
 538 }
 539
 540 static BOOL install_files_copy(struct assembly_entry *assembly, const WCHAR *source_path, struct fileop_entry *fileop, BOOL dryrun)
 541 {
 542     WCHAR *target_path, *target, *source = NULL;
 543     BOOL ret = FALSE;
 544
 545     if (!(target_path = expand_expression(assembly, fileop->target))) return FALSE;
 546     if (!(target = path_combine(target_path, fileop->source))) goto error;
 547     if (!(source = path_combine(source_path, fileop->source))) goto error;
 548
 549     if (dryrun)
 550     {
 551         if (!(ret = PathFileExistsW(source)))
 552         {
 553             ERR("Required file %s not found\n", debugstr_w(source));
 554             goto error;
 555         }
 556     }
 557     else
 558     {
 559         TRACE("Copying %s -> %s\n", debugstr_w(source), debugstr_w(target));
 560
 561         if (!create_parent_directory(target))
 562         {
 563             ERR("Failed to create parent directory for %s\n", debugstr_w(target));
 564             goto error;
 565         }
 566         if (!(ret = CopyFileExW(source, target, NULL, NULL, NULL, 0)))
 567         {
 568             ERR("Failed to copy %s to %s\n", debugstr_w(source), debugstr_w(target));
 569             goto error;
 570         }
 571     }
 572
 573 error:
 574     heap_free(target_path);
 575     heap_free(target);
 576     heap_free(source);
 577     return ret;
 578 }
 579
 580 static BOOL install_files(struct assembly_entry *assembly, BOOL dryrun)
 581 {
 582     struct fileop_entry *fileop;
 583     WCHAR *source_path;
 584     BOOL ret = TRUE;
 585
 586     if (!(source_path = get_assembly_source(assembly)))
 587     {
 588         ERR("Failed to get assembly source directory\n");
 589         return FALSE;
 590     }
 591
 592     LIST_FOR_EACH_ENTRY(fileop, &assembly->fileops, struct fileop_entry, entry)
 593     {
 594         if (!(ret = install_files_copy(assembly, source_path, fileop, dryrun))) break;
 595     }
 596
 597     heap_free(source_path);
 598     return ret;
 599 }
 600
 601 static WCHAR *split_registry_key(WCHAR *key, HKEY *root)
 602 {
 603     DWORD size;
 604     WCHAR *p;
 605
 606     if (!(p = wcschr(key, '\\'))) return NULL;
 607     size = p - key;
 608
 609     if (lstrlenW(L"HKEY_CLASSES_ROOT") == size && !wcsncmp(key, L"HKEY_CLASSES_ROOT", size))
 610         *root = HKEY_CLASSES_ROOT;
 611     else if (lstrlenW(L"HKEY_CURRENT_CONFIG") == size && !wcsncmp(key, L"HKEY_CURRENT_CONFIG", size))
 612         *root = HKEY_CURRENT_CONFIG;
 613     else if (lstrlenW(L"HKEY_CURRENT_USER") == size && !wcsncmp(key, L"HKEY_CURRENT_USER", size))
 614         *root = HKEY_CURRENT_USER;
 615     else if (lstrlenW(L"HKEY_LOCAL_MACHINE") == size && !wcsncmp(key, L"HKEY_LOCAL_MACHINE", size))
 616         *root = HKEY_LOCAL_MACHINE;
 617     else if (lstrlenW(L"HKEY_USERS") == size && !wcsncmp(key, L"HKEY_USERS", size))
 618         *root = HKEY_USERS;
 619     else
 620     {
 621         FIXME("Unknown root key %s\n", debugstr_wn(key, size));
 622         return NULL;
 623     }
 624
 625     return p + 1;
 626 }
 627
 628 static BOOL install_registry_string(struct assembly_entry *assembly, HKEY key, struct registrykv_entry *registrykv, DWORD type, BOOL dryrun)
 629 {
 630     DWORD value_size;
 631     WCHAR *value = expand_expression(assembly, registrykv->value);
 632     BOOL ret = TRUE;
 633
 634     if (registrykv->value && !value)
 635         return FALSE;
 636
 637     value_size = value ? (lstrlenW(value) + 1) * sizeof(WCHAR) : 0;
 638     if (!dryrun && RegSetValueExW(key, registrykv->name, 0, type, (void *)value, value_size))
 639     {
 640         ERR("Failed to set registry key %s\n", debugstr_w(registrykv->name));
 641         ret = FALSE;
 642     }
 643
 644     heap_free(value);
 645     return ret;
 646 }
 647
 648 static WCHAR *parse_multisz(const WCHAR *input, DWORD *size)
 649 {
 650     const WCHAR *pos, *next;
 651     struct strbuf buf;
 652
 653     *size = 0;
 654     if (!input || !input[0] || !strbuf_init(&buf)) return NULL;
 655
 656     for (pos = input; pos[0] == '"'; pos++)
 657     {
 658         pos++;
 659         if (!(next = wcsstr(pos, L"\""))) goto error;
 660         strbuf_append(&buf, pos, next - pos);
 661         strbuf_append(&buf, L"", ARRAY_SIZE(L""));
 662
 663         pos = next + 1;
 664         if (!pos[0]) break;
 665         if (pos[0] != ',')
 666         {
 667             FIXME("Error while parsing REG_MULTI_SZ string: Expected comma but got '%c'\n", pos[0]);
 668             goto error;
 669         }
 670     }
 671
 672     if (pos[0])
 673     {
 674         FIXME("Error while parsing REG_MULTI_SZ string: Garbage at end of string\n");
 675         goto error;
 676     }
 677
 678     strbuf_append(&buf, L"", ARRAY_SIZE(L""));
 679     *size = buf.pos * sizeof(WCHAR);
 680     return buf.buf;
 681
 682 error:
 683     strbuf_free(&buf);
 684     return NULL;
 685 }
 686
 687 static BOOL install_registry_multisz(struct assembly_entry *assembly, HKEY key, struct registrykv_entry *registrykv, BOOL dryrun)
 688 {
 689     DWORD value_size;
 690     WCHAR *value = parse_multisz(registrykv->value, &value_size);
 691     BOOL ret = TRUE;
 692
 693     if (registrykv->value && registrykv->value[0] && !value)
 694         return FALSE;
 695
 696     if (!dryrun && RegSetValueExW(key, registrykv->name, 0, REG_MULTI_SZ, (void *)value, value_size))
 697     {
 698         ERR("Failed to set registry key %s\n", debugstr_w(registrykv->name));
 699         ret = FALSE;
 700     }
 701
 702     heap_free(value);
 703     return ret;
 704 }
 705
 706 static BOOL install_registry_dword(struct assembly_entry *assembly, HKEY key, struct registrykv_entry *registrykv, BOOL dryrun)
 707 {
 708     DWORD value = registrykv->value_type ? wcstoul(registrykv->value_type, NULL, 16) : 0;
 709     BOOL ret = TRUE;
 710
 711     if (!dryrun && RegSetValueExW(key, registrykv->name, 0, REG_DWORD, (void *)&value, sizeof(value)))
 712     {
 713         ERR("Failed to set registry key %s\n", debugstr_w(registrykv->name));
 714         ret = FALSE;
 715     }
 716
 717     return ret;
 718 }
 719
 720 static BYTE *parse_hex(const WCHAR *input, DWORD *size)
 721 {
 722     WCHAR number[3] = {0, 0, 0};
 723     BYTE *output, *p;
 724     int length;
 725
 726     *size = 0;
 727     if (!input) return NULL;
 728     length = lstrlenW(input);
 729     if (length & 1) return NULL;
 730     length >>= 1;
 731
 732     if (!(output = heap_alloc(length))) return NULL;
 733     for (p = output; *input; input += 2)
 734     {
 735         number[0] = input[0];
 736         number[1] = input[1];
 737         *p++ = wcstoul(number, 0, 16);
 738     }
 739     *size = length;
 740     return output;
 741 }
 742
 743 static BOOL install_registry_binary(struct assembly_entry *assembly, HKEY key, struct registrykv_entry *registrykv, BOOL dryrun)
 744 {
 745     DWORD value_size;
 746     BYTE *value = parse_hex(registrykv->value, &value_size);
 747     BOOL ret = TRUE;
 748
 749     if (registrykv->value && !value)
 750         return FALSE;
 751
 752     if (!dryrun && RegSetValueExW(key, registrykv->name, 0, REG_BINARY, value, value_size))
 753     {
 754         ERR("Failed to set registry key %s\n", debugstr_w(registrykv->name));
 755         ret = FALSE;
 756     }
 757
 758     heap_free(value);
 759     return ret;
 760 }
 761
 762 static BOOL install_registry_value(struct assembly_entry *assembly, HKEY key, struct registrykv_entry *registrykv, BOOL dryrun)
 763 {
 764     TRACE("Setting registry key %s = %s\n", debugstr_w(registrykv->name), debugstr_w(registrykv->value));
 765
 766     if (!wcscmp(registrykv->value_type, L"REG_SZ"))
 767         return install_registry_string(assembly, key, registrykv, REG_SZ, dryrun);
 768     if (!wcscmp(registrykv->value_type, L"REG_EXPAND_SZ"))
 769         return install_registry_string(assembly, key, registrykv, REG_EXPAND_SZ, dryrun);
 770     if (!wcscmp(registrykv->value_type, L"REG_MULTI_SZ"))
 771         return install_registry_multisz(assembly, key, registrykv, dryrun);
 772     if (!wcscmp(registrykv->value_type, L"REG_DWORD"))
 773         return install_registry_dword(assembly, key, registrykv, dryrun);
 774     if (!wcscmp(registrykv->value_type, L"REG_BINARY"))
 775         return install_registry_binary(assembly, key, registrykv, dryrun);
 776
 777     FIXME("Unsupported registry value type %s\n", debugstr_w(registrykv->value_type));
 778     return FALSE;
 779 }
 780
 781 static BOOL install_registry(struct assembly_entry *assembly, BOOL dryrun)
 782 {
 783     struct registryop_entry *registryop;
 784     struct registrykv_entry *registrykv;
 785     HKEY root, subkey;
 786     WCHAR *path;
 787     REGSAM sam = KEY_ALL_ACCESS;
 788     BOOL ret = TRUE;
 789
 790 #ifdef __x86_64__
 791     if (!wcscmp(assembly->identity.architecture, L"x86")) sam |= KEY_WOW64_32KEY;
 792 #endif
 793
 794     LIST_FOR_EACH_ENTRY(registryop, &assembly->registryops, struct registryop_entry, entry)
 795     {
 796         if (!(path = split_registry_key(registryop->key, &root)))
 797         {
 798             ret = FALSE;
 799             break;
 800         }
 801
 802         TRACE("Processing registry key %s\n", debugstr_w(registryop->key));
 803
 804         if (!dryrun && RegCreateKeyExW(root, path, 0, NULL, 0, sam, NULL, &subkey, NULL))
 805         {
 806             ERR("Failed to open registry key %s\n", debugstr_w(registryop->key));
 807             ret = FALSE;
 808             break;
 809         }
 810
 811         LIST_FOR_EACH_ENTRY(registrykv, &registryop->keyvalues, struct registrykv_entry, entry)
 812         {
 813             if (!(ret = install_registry_value(assembly, subkey, registrykv, dryrun))) break;
 814         }
 815
 816         if (!dryrun) RegCloseKey(subkey);
 817         if (!ret) break;
 818     }
 819
 820     return ret;
 821 }
 822
 823 static BOOL install_assembly(struct list *manifest_list, struct assembly_identity *identity, BOOL dryrun)
 824 {
 825     struct dependency_entry *dependency;
 826     struct assembly_entry *assembly;
 827     const WCHAR *name;
 828
 829     if (!(assembly = lookup_assembly(manifest_list, identity)))
 830     {
 831         FIXME("Assembly %s not found\n", debugstr_w(identity->name));
 832         return FALSE;
 833     }
 834
 835     name = assembly->identity.name;
 836
 837     if (assembly->status == ASSEMBLY_STATUS_INSTALLED)
 838     {
 839         TRACE("Assembly %s already installed\n", debugstr_w(name));
 840         return TRUE;
 841     }
 842     if (assembly->status == ASSEMBLY_STATUS_IN_PROGRESS)
 843     {
 844         ERR("Assembly %s caused circular dependency\n", debugstr_w(name));
 845         return FALSE;
 846     }
 847
 848 #ifdef __i386__
 849     if (!wcscmp(assembly->identity.architecture, L"amd64"))
 850     {
 851         ERR("Cannot install amd64 assembly in 32-bit prefix\n");
 852         return FALSE;
 853     }
 854 #endif
 855
 856     assembly->status = ASSEMBLY_STATUS_IN_PROGRESS;
 857
 858     LIST_FOR_EACH_ENTRY(dependency, &assembly->dependencies, struct dependency_entry, entry)
 859     {
 860         if (!install_assembly(manifest_list, &dependency->identity, dryrun)) return FALSE;
 861     }
 862
 863     TRACE("Installing assembly %s%s\n", debugstr_w(name), dryrun ? " (dryrun)" : "");
 864
 865     if (!install_files(assembly, dryrun))
 866     {
 867         ERR("Failed to install all files for %s\n", debugstr_w(name));
 868         return FALSE;
 869     }
 870
 871     if (!install_registry(assembly, dryrun))
 872     {
 873         ERR("Failed to install registry keys for %s\n", debugstr_w(name));
 874         return FALSE;
 875     }
 876
 877     TRACE("Installation of %s finished\n", debugstr_w(name));
 878
 879     assembly->status = ASSEMBLY_STATUS_INSTALLED;
 880     return TRUE;
 881 }
 882
 883 static BOOL install_updates(struct installer_state *state, BOOL dryrun)
 884 {
 885     struct dependency_entry *dependency;
 886     LIST_FOR_EACH_ENTRY(dependency, &state->updates, struct dependency_entry, entry)
 887     {
 888         if (!install_assembly(&state->assemblies, &dependency->identity, dryrun))
 889         {
 890             ERR("Failed to install update %s\n", debugstr_w(dependency->identity.name));
 891             return FALSE;
 892         }
 893     }
 894     return TRUE;
 895 }
 896
 897 static void set_assembly_status(struct list *manifest_list, DWORD status)
 898 {
 899     struct assembly_entry *assembly;
 900     LIST_FOR_EACH_ENTRY(assembly, manifest_list, struct assembly_entry, entry)
 901     {
 902         assembly->status = status;
 903     }
 904 }
 905
 906 static BOOL install_msu(const WCHAR *filename, struct installer_state *state)
 907 {
 908     const WCHAR *temp_path;
 909     WIN32_FIND_DATAW data;
 910     HANDLE search;
 911     WCHAR *path;
 912     BOOL ret = FALSE;
 913
 914     list_init(&state->tempdirs);
 915     list_init(&state->assemblies);
 916     list_init(&state->updates);
 917     CoInitialize(NULL);
 918
 919     TRACE("Processing msu file %s\n", debugstr_w(filename));
 920
 921     if (!(temp_path = create_temp_directory(state))) return FALSE;
 922     if (!extract_cabinet(filename, temp_path))
 923     {
 924         ERR("Failed to extract %s\n", debugstr_w(filename));
 925         goto done;
 926     }
 927
 928     /* load all manifests from contained cabinet archives */
 929     if (!(path = path_combine(temp_path, L"*.cab"))) goto done;
 930     search = FindFirstFileW(path, &data);
 931     heap_free(path);
 932
 933     if (search != INVALID_HANDLE_VALUE)
 934     {
 935         do
 936         {
 937             if (data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) continue;
 938             if (!wcsicmp(data.cFileName, L"WSUSSCAN.cab")) continue;
 939             if (!(path = path_combine(temp_path, data.cFileName))) continue;
 940             if (!load_assemblies_from_cab(path, state))
 941                 ERR("Failed to load all manifests from %s, ignoring\n", debugstr_w(path));
 942             heap_free(path);
 943         }
 944         while (FindNextFileW(search, &data));
 945         FindClose(search);
 946     }
 947
 948     /* load all update descriptions */
 949     if (!(path = path_combine(temp_path, L"*.xml"))) goto done;
 950     search = FindFirstFileW(path, &data);
 951     heap_free(path);
 952
 953     if (search != INVALID_HANDLE_VALUE)
 954     {
 955         do
 956         {
 957             if (data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) continue;
 958             if (!(path = path_combine(temp_path, data.cFileName))) continue;
 959             if (!load_update(path, &state->updates))
 960                 ERR("Failed to load all updates from %s, ignoring\n", debugstr_w(path));
 961             heap_free(path);
 962         }
 963         while (FindNextFileW(search, &data));
 964         FindClose(search);
 965     }
 966
 967     /* dump package information (for debugging) */
 968     if (TRACE_ON(wusa))
 969     {
 970         struct dependency_entry *dependency;
 971         struct assembly_entry *assembly;
 972
 973         TRACE("List of updates:\n");
 974         LIST_FOR_EACH_ENTRY(dependency, &state->updates, struct dependency_entry, entry)
 975             TRACE(" * %s\n", debugstr_w(dependency->identity.name));
 976
 977         TRACE("List of manifests (with dependencies):\n");
 978         LIST_FOR_EACH_ENTRY(assembly, &state->assemblies, struct assembly_entry, entry)
 979         {
 980             TRACE(" * %s\n", debugstr_w(assembly->identity.name));
 981             LIST_FOR_EACH_ENTRY(dependency, &assembly->dependencies, struct dependency_entry, entry)
 982                 TRACE("   -> %s\n", debugstr_w(dependency->identity.name));
 983         }
 984     }
 985
 986     if (list_empty(&state->updates))
 987     {
 988         ERR("No updates found, probably incompatible MSU file format?\n");
 989         goto done;
 990     }
 991
 992     /* perform dry run */
 993     set_assembly_status(&state->assemblies, ASSEMBLY_STATUS_NONE);
 994     if (!install_updates(state, TRUE))
 995     {
 996         ERR("Dry run failed, aborting installation\n");
 997         goto done;
 998     }
 999
1000     /* installation */
1001     set_assembly_status(&state->assemblies, ASSEMBLY_STATUS_NONE);
1002     if (!install_updates(state, FALSE))
1003     {
1004         ERR("Installation failed\n");
1005         goto done;
1006     }
1007
1008     TRACE("Installation finished\n");
1009     ret = TRUE;
1010
1011 done:
1012     installer_cleanup(state);
1013     return ret;
1014 }
1015
1016 static void restart_as_x86_64(void)
1017 {
1018     WCHAR filename[MAX_PATH];
1019     PROCESS_INFORMATION pi;
1020     STARTUPINFOW si;
1021     DWORD exit_code = 1;
1022     void *redir;
1023
1024     memset(&si, 0, sizeof(si));
1025     si.cb = sizeof(si);
1026     GetSystemDirectoryW( filename, MAX_PATH );
1027     wcscat( filename, L"\\wusa.exe" );
1028
1029     Wow64DisableWow64FsRedirection(&redir);
1030     if (CreateProcessW(filename, GetCommandLineW(), NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi))
1031     {
1032         TRACE("Restarting %s\n", wine_dbgstr_w(filename));
1033         WaitForSingleObject(pi.hProcess, INFINITE);
1034         GetExitCodeProcess(pi.hProcess, &exit_code);
1035         CloseHandle(pi.hProcess);
1036         CloseHandle(pi.hThread);
1037     }
1038     else ERR("Failed to restart 64-bit %s, err %u\n", wine_dbgstr_w(filename), GetLastError());
1039     Wow64RevertWow64FsRedirection(redir);
1040
1041     ExitProcess(exit_code);
1042 }
1043
1044 int __cdecl wmain(int argc, WCHAR *argv[])
1045 {
1046     struct installer_state state;
1047     const WCHAR *filename = NULL;
1048     BOOL is_wow64;
1049     int i;
1050
1051     if (IsWow64Process( GetCurrentProcess(), &is_wow64 ) && is_wow64) restart_as_x86_64();
1052
1053     state.norestart = FALSE;
1054     state.quiet = FALSE;
1055
1056     if (TRACE_ON(wusa))
1057     {
1058         TRACE("Command line:");
1059         for (i = 0; i < argc; i++)
1060             TRACE(" %s", wine_dbgstr_w(argv[i]));
1061         TRACE("\n");
1062     }
1063
1064     for (i = 1; i < argc; i++)
1065     {
1066         if (argv[i][0] == '/')
1067         {
1068             if (!wcscmp(argv[i], L"/norestart"))
1069                 state.norestart = TRUE;
1070             else if (!wcscmp(argv[i], L"/quiet"))
1071                 state.quiet = TRUE;
1072             else
1073                 FIXME("Unknown option: %s\n", wine_dbgstr_w(argv[i]));
1074         }
1075         else if (!filename)
1076             filename = argv[i];
1077         else
1078             FIXME("Unknown option: %s\n", wine_dbgstr_w(argv[i]));
1079     }
1080
1081     if (!filename)
1082     {
1083         FIXME("Missing filename argument\n");
1084         return 1;
1085     }
1086
1087     return !install_msu(filename, &state);
1088 }