reactos/dll/win32/syssetup/install.c

   1 /*
   2  *  ReactOS kernel
   3  *  Copyright (C) 2003 ReactOS Team
   4  *
   5  *  This program is free software; you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation; either version 2 of the License, or
   8  *  (at your option) any later version.
   9  *
  10  *  This program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License along
  16  *  with this program; if not, write to the Free Software Foundation, Inc.,
  17  *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18  */
  19 /*
  20  * COPYRIGHT:         See COPYING in the top level directory
  21  * PROJECT:           ReactOS system libraries
  22  * PURPOSE:           System setup
  23  * FILE:              dll/win32/syssetup/install.c
  24  * PROGRAMER:         Eric Kohl
  25  */
  26
  27 /* INCLUDES *****************************************************************/
  28
  29 #define WIN32_NO_STATUS
  30 #include <windows.h>
  31 #define NTOS_MODE_USER
  32 #include <ndk/ntndk.h>
  33
  34 #include <commctrl.h>
  35 #include <stdio.h>
  36 #include <io.h>
  37 #include <tchar.h>
  38 #include <stdlib.h>
  39
  40 #include <samlib/samlib.h>
  41 #include <syssetup/syssetup.h>
  42 #include <userenv.h>
  43 #include <setupapi.h>
  44
  45 #include <shlobj.h>
  46 #include <objidl.h>
  47 #include <shlwapi.h>
  48
  49 #include "globals.h"
  50 #include "resource.h"
  51
  52 #define NDEBUG
  53 #include <debug.h>
  54
  55 DWORD WINAPI
  56 CMP_WaitNoPendingInstallEvents(DWORD dwTimeout);
  57
  58 /* GLOBALS ******************************************************************/
  59
  60 PSID DomainSid = NULL;
  61 PSID AdminSid = NULL;
  62
  63 HINF hSysSetupInf = INVALID_HANDLE_VALUE;
  64
  65 /* FUNCTIONS ****************************************************************/
  66
  67 static VOID
  68 DebugPrint(char* fmt,...)
  69 {
  70     char buffer[512];
  71     va_list ap;
  72
  73     va_start(ap, fmt);
  74     vsprintf(buffer, fmt, ap);
  75     va_end(ap);
  76
  77     LogItem(SYSSETUP_SEVERITY_FATAL_ERROR, L"Failed");
  78
  79     strcat(buffer, "\nRebooting now!");
  80     MessageBoxA(NULL,
  81                 buffer,
  82                 "ReactOS Setup",
  83                 MB_OK);
  84 }
  85
  86
  87 HRESULT CreateShellLink(LPCTSTR linkPath, LPCTSTR cmd, LPCTSTR arg, LPCTSTR dir, LPCTSTR iconPath, int icon_nr, LPCTSTR comment)
  88 {
  89     IShellLink* psl;
  90     IPersistFile* ppf;
  91 #ifndef _UNICODE
  92     WCHAR buffer[MAX_PATH];
  93 #endif /* _UNICODE */
  94
  95     HRESULT hr = CoCreateInstance(&CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, &IID_IShellLink, (LPVOID*)&psl);
  96
  97     if (SUCCEEDED(hr))
  98     {
  99         hr = psl->lpVtbl->SetPath(psl, cmd);
 100
 101         if (arg)
 102         {
 103             hr = psl->lpVtbl->SetArguments(psl, arg);
 104         }
 105
 106         if (dir)
 107         {
 108             hr = psl->lpVtbl->SetWorkingDirectory(psl, dir);
 109         }
 110
 111         if (iconPath)
 112         {
 113             hr = psl->lpVtbl->SetIconLocation(psl, iconPath, icon_nr);
 114         }
 115
 116         if (comment)
 117         {
 118             hr = psl->lpVtbl->SetDescription(psl, comment);
 119         }
 120
 121         hr = psl->lpVtbl->QueryInterface(psl, &IID_IPersistFile, (LPVOID*)&ppf);
 122
 123         if (SUCCEEDED(hr))
 124         {
 125 #ifdef _UNICODE
 126             hr = ppf->lpVtbl->Save(ppf, linkPath, TRUE);
 127 #else /* _UNICODE */
 128             MultiByteToWideChar(CP_ACP, 0, linkPath, -1, buffer, MAX_PATH);
 129
 130             hr = ppf->lpVtbl->Save(ppf, buffer, TRUE);
 131 #endif /* _UNICODE */
 132
 133             ppf->lpVtbl->Release(ppf);
 134         }
 135
 136         psl->lpVtbl->Release(psl);
 137     }
 138
 139     return hr;
 140 }
 141
 142
 143 static BOOL
 144 CreateShortcut(int csidl, LPCTSTR folder, UINT nIdName, LPCTSTR command, UINT nIdTitle, BOOL bCheckExistence)
 145 {
 146     TCHAR path[MAX_PATH];
 147     TCHAR exeName[MAX_PATH];
 148     TCHAR title[256];
 149     TCHAR name[256];
 150     LPTSTR p = path;
 151     TCHAR szWorkingDir[MAX_PATH];
 152     LPTSTR lpWorkingDir = NULL;
 153     LPTSTR lpFilePart;
 154     DWORD dwLen;
 155
 156     if (ExpandEnvironmentStrings(command,
 157                                  path,
 158                                  sizeof(path) / sizeof(path[0])) == 0)
 159     {
 160         _tcscpy(path,
 161                 command);
 162     }
 163
 164     if (bCheckExistence)
 165     {
 166         if ((_taccess(path, 0 )) == -1)
 167             /* Expected error, don't return FALSE */
 168             return TRUE;
 169     }
 170
 171     dwLen = GetFullPathName(path,
 172                             sizeof(szWorkingDir) / sizeof(szWorkingDir[0]),
 173                             szWorkingDir,
 174                             &lpFilePart);
 175     if (dwLen != 0 && dwLen <= sizeof(szWorkingDir) / sizeof(szWorkingDir[0]))
 176     {
 177         /* Save the file name */
 178         _tcscpy(exeName, lpFilePart);
 179
 180         if (lpFilePart != NULL)
 181         {
 182             /* We're only interested in the path. Cut the file name off.
 183                Also remove the trailing backslash unless the working directory
 184                is only going to be a drive, ie. C:\ */
 185             *(lpFilePart--) = _T('\0');
 186             if (!(lpFilePart - szWorkingDir == 2 && szWorkingDir[1] == _T(':') &&
 187                   szWorkingDir[2] == _T('\\')))
 188             {
 189                 *lpFilePart = _T('\0');
 190             }
 191         }
 192
 193         lpWorkingDir = szWorkingDir;
 194     }
 195
 196
 197     if (!SHGetSpecialFolderPath(0, path, csidl, TRUE))
 198         return FALSE;
 199
 200     if (folder)
 201     {
 202         p = PathAddBackslash(p);
 203         _tcscpy(p, folder);
 204     }
 205
 206     p = PathAddBackslash(p);
 207
 208     if (!LoadString(hDllInstance, nIdName, name, sizeof(name)/sizeof(name[0])))
 209         return FALSE;
 210     _tcscpy(p, name);
 211
 212     if (!LoadString(hDllInstance, nIdTitle, title, sizeof(title)/sizeof(title[0])))
 213         return FALSE;
 214
 215     // FIXME: we should pass 'command' straight in here, but shell32 doesn't expand it
 216     return SUCCEEDED(CreateShellLink(path, exeName, _T(""), lpWorkingDir, NULL, 0, title));
 217 }
 218
 219
 220 static BOOL
 221 CreateShortcutFolder(int csidl, UINT nID, LPTSTR name, int nameLen)
 222 {
 223     TCHAR path[MAX_PATH];
 224     LPTSTR p;
 225
 226     if (!SHGetSpecialFolderPath(0, path, csidl, TRUE))
 227         return FALSE;
 228
 229     if (!LoadString(hDllInstance, nID, name, nameLen))
 230         return FALSE;
 231
 232     p = PathAddBackslash(path);
 233     _tcscpy(p, name);
 234
 235     return CreateDirectory(path, NULL) || GetLastError()==ERROR_ALREADY_EXISTS;
 236 }
 237
 238
 239 static BOOL
 240 CreateRandomSid(
 241     OUT PSID *Sid)
 242 {
 243     SID_IDENTIFIER_AUTHORITY SystemAuthority = {SECURITY_NT_AUTHORITY};
 244     LARGE_INTEGER SystemTime;
 245     PULONG Seed;
 246     NTSTATUS Status;
 247
 248     NtQuerySystemTime(&SystemTime);
 249     Seed = &SystemTime.u.LowPart;
 250
 251     Status = RtlAllocateAndInitializeSid(
 252         &SystemAuthority,
 253         4,
 254         SECURITY_NT_NON_UNIQUE,
 255         RtlUniform(Seed),
 256         RtlUniform(Seed),
 257         RtlUniform(Seed),
 258         SECURITY_NULL_RID,
 259         SECURITY_NULL_RID,
 260         SECURITY_NULL_RID,
 261         SECURITY_NULL_RID,
 262         Sid);
 263     return NT_SUCCESS(Status);
 264 }
 265
 266
 267 static VOID
 268 AppendRidToSid(
 269     OUT PSID *Dst,
 270     IN PSID Src,
 271     IN ULONG NewRid)
 272 {
 273     ULONG Rid[8] = {0, 0, 0, 0, 0, 0, 0, 0};
 274     UCHAR RidCount;
 275     ULONG i;
 276
 277     RidCount = *RtlSubAuthorityCountSid (Src);
 278
 279     for (i = 0; i < RidCount; i++)
 280         Rid[i] = *RtlSubAuthoritySid (Src, i);
 281
 282     if (RidCount < 8)
 283     {
 284         Rid[RidCount] = NewRid;
 285         RidCount++;
 286     }
 287
 288     RtlAllocateAndInitializeSid(
 289         RtlIdentifierAuthoritySid(Src),
 290         RidCount,
 291         Rid[0],
 292         Rid[1],
 293         Rid[2],
 294         Rid[3],
 295         Rid[4],
 296         Rid[5],
 297         Rid[6],
 298         Rid[7],
 299         Dst);
 300 }
 301
 302
 303 static VOID
 304 CreateTempDir(
 305     IN LPCWSTR VarName)
 306 {
 307     WCHAR szTempDir[MAX_PATH];
 308     WCHAR szBuffer[MAX_PATH];
 309     DWORD dwLength;
 310     HKEY hKey;
 311
 312     if (RegOpenKeyExW(HKEY_LOCAL_MACHINE,
 313                      L"SYSTEM\\CurrentControlSet\\Control\\Session Manager\\Environment",
 314                      0,
 315                      KEY_QUERY_VALUE,
 316                      &hKey))
 317     {
 318         DebugPrint("Error: %lu\n", GetLastError());
 319         return;
 320     }
 321
 322     /* Get temp dir */
 323     dwLength = MAX_PATH * sizeof(WCHAR);
 324     if (RegQueryValueExW(hKey,
 325                         VarName,
 326                         NULL,
 327                         NULL,
 328                         (LPBYTE)szBuffer,
 329                         &dwLength))
 330     {
 331         DebugPrint("Error: %lu\n", GetLastError());
 332         RegCloseKey(hKey);
 333         return;
 334     }
 335
 336     /* Expand it */
 337     if (!ExpandEnvironmentStringsW(szBuffer,
 338                                   szTempDir,
 339                                   MAX_PATH))
 340     {
 341         DebugPrint("Error: %lu\n", GetLastError());
 342         RegCloseKey(hKey);
 343         return;
 344     }
 345
 346     /* Create profiles directory */
 347     if (!CreateDirectoryW(szTempDir, NULL))
 348     {
 349         if (GetLastError() != ERROR_ALREADY_EXISTS)
 350         {
 351             DebugPrint("Error: %lu\n", GetLastError());
 352             RegCloseKey(hKey);
 353             return;
 354         }
 355     }
 356
 357     RegCloseKey(hKey);
 358 }
 359
 360
 361 BOOL
 362 InstallSysSetupInfDevices(VOID)
 363 {
 364     INFCONTEXT InfContext;
 365     WCHAR LineBuffer[256];
 366     DWORD LineLength;
 367
 368     if (!SetupFindFirstLineW(hSysSetupInf,
 369                             L"DeviceInfsToInstall",
 370                             NULL,
 371                             &InfContext))
 372     {
 373         return FALSE;
 374     }
 375
 376     do
 377     {
 378         if (!SetupGetStringFieldW(&InfContext,
 379                                  0,
 380                                  LineBuffer,
 381                                  sizeof(LineBuffer)/sizeof(LineBuffer[0]),
 382                                  &LineLength))
 383         {
 384             return FALSE;
 385         }
 386
 387         if (!SetupDiInstallClassW(NULL, LineBuffer, DI_QUIETINSTALL, NULL))
 388         {
 389             return FALSE;
 390         }
 391     }
 392     while (SetupFindNextLine(&InfContext, &InfContext));
 393
 394     return TRUE;
 395 }
 396 BOOL
 397 InstallSysSetupInfComponents(VOID)
 398 {
 399     INFCONTEXT InfContext;
 400     WCHAR NameBuffer[256];
 401     WCHAR SectionBuffer[256];
 402     HINF hComponentInf = INVALID_HANDLE_VALUE;
 403
 404     if (!SetupFindFirstLineW(hSysSetupInf,
 405                             L"Infs.Always",
 406                             NULL,
 407                             &InfContext))
 408     {
 409         DPRINT("No Inf.Always section found\n");
 410     }
 411     else
 412     {
 413         do
 414         {
 415             if (!SetupGetStringFieldW(&InfContext,
 416                                      1, // Get the component name
 417                                      NameBuffer,
 418                                      sizeof(NameBuffer)/sizeof(NameBuffer[0]),
 419                                      NULL))
 420             {
 421                 DebugPrint("Error while trying to get component name \n");
 422                 return FALSE;
 423             }
 424
 425             if (!SetupGetStringFieldW(&InfContext,
 426                                      2, // Get the component install section
 427                                      SectionBuffer,
 428                                      sizeof(SectionBuffer)/sizeof(SectionBuffer[0]),
 429                                      NULL))
 430             {
 431                 DebugPrint("Error while trying to get component install section \n");
 432                 return FALSE;
 433             }
 434
 435             DPRINT("Trying to execute install section '%S' from '%S' \n", SectionBuffer , NameBuffer);
 436
 437             hComponentInf = SetupOpenInfFileW(NameBuffer,
 438                                               NULL,
 439                                               INF_STYLE_WIN4,
 440                                               NULL);
 441
 442             if (hComponentInf == INVALID_HANDLE_VALUE)
 443             {
 444                 DebugPrint("SetupOpenInfFileW() failed to open '%S' (Error: %lu)\n", NameBuffer ,GetLastError());
 445                 return FALSE;
 446             }
 447
 448             if (!SetupInstallFromInfSectionW(NULL,
 449                                             hComponentInf,
 450                                             SectionBuffer,
 451                                             SPINST_ALL,
 452                                             NULL,
 453                                             NULL,
 454                                             SP_COPY_NEWER,
 455                                             SetupDefaultQueueCallbackW,
 456                                             NULL,
 457                                             NULL,
 458                                             NULL))
 459            {
 460                 DebugPrint("Error while trying to install : %S (Error: %lu)\n", NameBuffer, GetLastError());
 461                 SetupCloseInfFile(hComponentInf);
 462                 return FALSE;
 463            }
 464
 465            SetupCloseInfFile(hComponentInf);
 466         }
 467         while (SetupFindNextLine(&InfContext, &InfContext));
 468     }
 469
 470     return TRUE;
 471 }
 472
 473 static BOOL
 474 EnableUserModePnpManager(VOID)
 475 {
 476     SC_HANDLE hSCManager = NULL;
 477     SC_HANDLE hService = NULL;
 478     BOOL ret = FALSE;
 479
 480     hSCManager = OpenSCManagerW(NULL, NULL, SC_MANAGER_ENUMERATE_SERVICE);
 481     if (hSCManager == NULL)
 482     {
 483         DPRINT1("Unable to open the service control manager.\n");
 484         goto cleanup;
 485     }
 486
 487     hService = OpenServiceW(hSCManager,
 488                             L"PlugPlay",
 489                             SERVICE_CHANGE_CONFIG | SERVICE_START);
 490     if (hService == NULL)
 491     {
 492         DPRINT1("Unable to open PlugPlay service\n");
 493         goto cleanup;
 494     }
 495
 496     ret = ChangeServiceConfigW(hService,
 497                                SERVICE_NO_CHANGE,
 498                                SERVICE_AUTO_START,
 499                                SERVICE_NO_CHANGE,
 500                                NULL, NULL, NULL,
 501                                NULL, NULL, NULL, NULL);
 502     if (!ret)
 503     {
 504         DPRINT1("Unable to change the service configuration\n");
 505         goto cleanup;
 506     }
 507
 508     ret = StartServiceW(hService, 0, NULL);
 509     if ((!ret) && (GetLastError() != ERROR_SERVICE_ALREADY_RUNNING))
 510     {
 511         DPRINT1("Unable to start service\n");
 512         goto cleanup;
 513     }
 514
 515     ret = TRUE;
 516
 517 cleanup:
 518     if (hSCManager != NULL)
 519         CloseServiceHandle(hSCManager);
 520     if (hService != NULL)
 521         CloseServiceHandle(hService);
 522     return ret;
 523 }
 524
 525
 526 static INT_PTR CALLBACK
 527 StatusMessageWindowProc(
 528     IN HWND hwndDlg,
 529     IN UINT uMsg,
 530     IN WPARAM wParam,
 531     IN LPARAM lParam)
 532 {
 533     UNREFERENCED_PARAMETER(wParam);
 534
 535     switch (uMsg)
 536     {
 537         case WM_INITDIALOG:
 538         {
 539             WCHAR szMsg[256];
 540
 541             if (!LoadStringW(hDllInstance, IDS_STATUS_INSTALL_DEV, szMsg, sizeof(szMsg)/sizeof(szMsg[0])))
 542                 return FALSE;
 543             SetDlgItemTextW(hwndDlg, IDC_STATUSLABEL, szMsg);
 544             return TRUE;
 545         }
 546     }
 547     return FALSE;
 548 }
 549
 550
 551 static DWORD WINAPI
 552 ShowStatusMessageThread(
 553     IN LPVOID lpParameter)
 554 {
 555     HWND *phWnd = (HWND *)lpParameter;
 556     HWND hWnd;
 557     MSG Msg;
 558
 559     hWnd = CreateDialogParam(
 560         hDllInstance,
 561         MAKEINTRESOURCE(IDD_STATUSWINDOW_DLG),
 562         GetDesktopWindow(),
 563         StatusMessageWindowProc,
 564         (LPARAM)NULL);
 565     if (!hWnd)
 566         return 0;
 567     *phWnd = hWnd;
 568
 569     ShowWindow(hWnd, SW_SHOW);
 570
 571     /* Message loop for the Status window */
 572     while (GetMessage(&Msg, NULL, 0, 0))
 573     {
 574         TranslateMessage(&Msg);
 575         DispatchMessage(&Msg);
 576     }
 577
 578     return 0;
 579 }
 580
 581 static LONG
 582 ReadRegSzKey(
 583     IN HKEY hKey,
 584     IN LPCWSTR pszKey,
 585     OUT LPWSTR* pValue)
 586 {
 587     LONG rc;
 588     DWORD dwType;
 589     DWORD cbData = 0;
 590     LPWSTR Value;
 591
 592     if (!pValue)
 593         return ERROR_INVALID_PARAMETER;
 594
 595     *pValue = NULL;
 596     rc = RegQueryValueExW(hKey, pszKey, NULL, &dwType, NULL, &cbData);
 597     if (rc != ERROR_SUCCESS)
 598         return rc;
 599     if (dwType != REG_SZ)
 600         return ERROR_FILE_NOT_FOUND;
 601     Value = HeapAlloc(GetProcessHeap(), 0, cbData + sizeof(WCHAR));
 602     if (!Value)
 603         return ERROR_NOT_ENOUGH_MEMORY;
 604     rc = RegQueryValueExW(hKey, pszKey, NULL, NULL, (LPBYTE)Value, &cbData);
 605     if (rc != ERROR_SUCCESS)
 606     {
 607         HeapFree(GetProcessHeap(), 0, Value);
 608         return rc;
 609     }
 610     /* NULL-terminate the string */
 611     Value[cbData / sizeof(WCHAR)] = '\0';
 612
 613     *pValue = Value;
 614     return ERROR_SUCCESS;
 615 }
 616
 617 static BOOL
 618 IsConsoleBoot(VOID)
 619 {
 620     HKEY ControlKey = NULL;
 621     LPWSTR SystemStartOptions = NULL;
 622     LPWSTR CurrentOption, NextOption; /* Pointers into SystemStartOptions */
 623     BOOL ConsoleBoot = FALSE;
 624     LONG rc;
 625
 626     rc = RegOpenKeyExW(
 627         HKEY_LOCAL_MACHINE,
 628         L"SYSTEM\\CurrentControlSet\\Control",
 629         0,
 630         KEY_QUERY_VALUE,
 631         &ControlKey);
 632
 633     rc = ReadRegSzKey(ControlKey, L"SystemStartOptions", &SystemStartOptions);
 634     if (rc != ERROR_SUCCESS)
 635         goto cleanup;
 636
 637     /* Check for CMDCONS in SystemStartOptions */
 638     CurrentOption = SystemStartOptions;
 639     while (CurrentOption)
 640     {
 641         NextOption = wcschr(CurrentOption, L' ');
 642         if (NextOption)
 643             *NextOption = L'\0';
 644         if (wcsicmp(CurrentOption, L"CONSOLE") == 0)
 645         {
 646             DPRINT("Found %S. Switching to console boot\n", CurrentOption);
 647             ConsoleBoot = TRUE;
 648             goto cleanup;
 649         }
 650         CurrentOption = NextOption ? NextOption + 1 : NULL;
 651     }
 652
 653 cleanup:
 654     if (ControlKey != NULL)
 655         RegCloseKey(ControlKey);
 656     HeapFree(GetProcessHeap(), 0, SystemStartOptions);
 657     return ConsoleBoot;
 658 }
 659
 660 static BOOL
 661 CommonInstall(VOID)
 662 {
 663     HWND hWnd = NULL;
 664
 665     hSysSetupInf = SetupOpenInfFileW(
 666         L"syssetup.inf",
 667         NULL,
 668         INF_STYLE_WIN4,
 669         NULL);
 670     if (hSysSetupInf == INVALID_HANDLE_VALUE)
 671     {
 672         DebugPrint("SetupOpenInfFileW() failed to open 'syssetup.inf' (Error: %lu)\n", GetLastError());
 673         return FALSE;
 674     }
 675
 676     if (!InstallSysSetupInfDevices())
 677     {
 678         DebugPrint("InstallSysSetupInfDevices() failed!\n");
 679         SetupCloseInfFile(hSysSetupInf);
 680         return FALSE;
 681     }
 682
 683     if(!InstallSysSetupInfComponents())
 684     {
 685         DebugPrint("InstallSysSetupInfComponents() failed!\n");
 686         SetupCloseInfFile(hSysSetupInf);
 687         return FALSE;
 688     }
 689
 690     if (!IsConsoleBoot())
 691     {
 692         CreateThread(
 693             NULL,
 694             0,
 695             ShowStatusMessageThread,
 696             (LPVOID)&hWnd,
 697             0,
 698             NULL);
 699     }
 700
 701     if (!EnableUserModePnpManager())
 702     {
 703        DebugPrint("EnableUserModePnpManager() failed!\n");
 704        SetupCloseInfFile(hSysSetupInf);
 705        EndDialog(hWnd, 0);
 706        return FALSE;
 707     }
 708
 709     if (CMP_WaitNoPendingInstallEvents(INFINITE) != WAIT_OBJECT_0)
 710     {
 711       DebugPrint("CMP_WaitNoPendingInstallEvents() failed!\n");
 712       SetupCloseInfFile(hSysSetupInf);
 713       EndDialog(hWnd, 0);
 714       return FALSE;
 715     }
 716
 717     EndDialog(hWnd, 0);
 718     return TRUE;
 719 }
 720
 721 DWORD WINAPI
 722 InstallLiveCD(IN HINSTANCE hInstance)
 723 {
 724     STARTUPINFOW StartupInfo;
 725     PROCESS_INFORMATION ProcessInformation;
 726     BOOL res;
 727
 728     if (!CommonInstall())
 729         goto cleanup;
 730     SetupCloseInfFile(hSysSetupInf);
 731
 732     /* Run the shell */
 733     StartupInfo.cb = sizeof(STARTUPINFOW);
 734     StartupInfo.lpReserved = NULL;
 735     StartupInfo.lpDesktop = NULL;
 736     StartupInfo.lpTitle = NULL;
 737     StartupInfo.dwFlags = 0;
 738     StartupInfo.cbReserved2 = 0;
 739     StartupInfo.lpReserved2 = 0;
 740     res = CreateProcessW(
 741         L"userinit.exe",
 742         NULL,
 743         NULL,
 744         NULL,
 745         FALSE,
 746         0,
 747         NULL,
 748         NULL,
 749         &StartupInfo,
 750         &ProcessInformation);
 751     if (!res)
 752         goto cleanup;
 753
 754     return 0;
 755
 756 cleanup:
 757     MessageBoxW(
 758         NULL,
 759         L"You can shutdown your computer, or press ENTER to reboot",
 760         L"ReactOS LiveCD",
 761         MB_OK);
 762     return 0;
 763 }
 764
 765
 766 static BOOL
 767 CreateShortcuts(VOID)
 768 {
 769     TCHAR szFolder[256];
 770
 771     CoInitialize(NULL);
 772
 773     /* Create desktop shortcuts */
 774     CreateShortcut(CSIDL_DESKTOP, NULL, IDS_SHORT_CMD, _T("%SystemRoot%\\system32\\cmd.exe"), IDS_CMT_CMD, TRUE);
 775     CreateShortcut(CSIDL_DESKTOP, NULL, IDS_SHORT_DOWNLOADER, _T("%SystemRoot%\\system32\\rapps.exe"), IDS_CMT_DOWNLOADER, TRUE);
 776
 777     /* Create program startmenu shortcuts */
 778     CreateShortcut(CSIDL_PROGRAMS, NULL, IDS_SHORT_EXPLORER, _T("%SystemRoot%\\explorer.exe"), IDS_CMT_EXPLORER, TRUE);
 779     CreateShortcut(CSIDL_PROGRAMS, NULL, IDS_SHORT_DOWNLOADER, _T("%SystemRoot%\\system32\\rapps.exe"), IDS_CMT_DOWNLOADER, TRUE);
 780
 781     /* Create administrative tools startmenu shortcuts */
 782     CreateShortcut(CSIDL_COMMON_ADMINTOOLS, NULL, IDS_SHORT_SERVICE, _T("%SystemRoot%\\system32\\servman.exe"), IDS_CMT_SERVMAN, TRUE);
 783     CreateShortcut(CSIDL_COMMON_ADMINTOOLS, NULL, IDS_SHORT_DEVICE, _T("%SystemRoot%\\system32\\devmgmt.exe"), IDS_CMT_DEVMGMT, TRUE);
 784     CreateShortcut(CSIDL_COMMON_ADMINTOOLS, NULL, IDS_SHORT_EVENTVIEW, _T("%SystemRoot%\\system32\\eventvwr.exe"), IDS_CMT_EVENTVIEW, TRUE);
 785     CreateShortcut(CSIDL_COMMON_ADMINTOOLS, NULL, IDS_SHORT_MSCONFIG, _T("%SystemRoot%\\system32\\msconfig.exe"), IDS_CMT_MSCONFIG, TRUE);
 786
 787     /* Create and fill Accessories subfolder */
 788     if (CreateShortcutFolder(CSIDL_PROGRAMS, IDS_ACCESSORIES, szFolder, sizeof(szFolder)/sizeof(szFolder[0])))
 789     {
 790         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_CALC, _T("%SystemRoot%\\system32\\calc.exe"), IDS_CMT_CALC, TRUE);
 791         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_CMD, _T("%SystemRoot%\\system32\\cmd.exe"), IDS_CMT_CMD, TRUE);
 792         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_NOTEPAD, _T("%SystemRoot%\\system32\\notepad.exe"), IDS_CMT_NOTEPAD, TRUE);
 793         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_RDESKTOP, _T("%SystemRoot%\\system32\\mstsc.exe"), IDS_CMT_RDESKTOP, TRUE);
 794         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_SNAP, _T("%SystemRoot%\\system32\\screenshot.exe"), IDS_CMT_SCREENSHOT, TRUE);
 795         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_WORDPAD, _T("%SystemRoot%\\system32\\wordpad.exe"), IDS_CMT_WORDPAD, TRUE);
 796         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_PAINT, _T("%SystemRoot%\\system32\\paint.exe"), IDS_CMT_PAINT, TRUE);
 797     }
 798
 799     /* Create System Tools subfolder and fill if the exe is available */
 800     if (CreateShortcutFolder(CSIDL_PROGRAMS, IDS_SYS_TOOLS, szFolder, sizeof(szFolder)/sizeof(szFolder[0])))
 801     {
 802         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_CHARMAP, _T("%SystemRoot%\\system32\\charmap.exe"), IDS_CMT_CHARMAP, TRUE);
 803         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_KBSWITCH, _T("%SystemRoot%\\system32\\kbswitch.exe"), IDS_CMT_KBSWITCH, TRUE);
 804         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_REGEDIT, _T("%SystemRoot%\\regedit.exe"), IDS_CMT_REGEDIT, TRUE);
 805         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_DXDIAG, _T("%SystemRoot%\\system32\\dxdiag.exe"), IDS_CMT_DXDIAG, TRUE);
 806     }
 807
 808     /* Create Accessibility subfolder and fill if the exe is available */
 809     if (CreateShortcutFolder(CSIDL_PROGRAMS, IDS_SYS_ACCESSIBILITY, szFolder, sizeof(szFolder)/sizeof(szFolder[0])))
 810     {
 811         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_MAGNIFY, _T("%SystemRoot%\\system32\\magnify.exe"), IDS_CMT_MAGNIFY, TRUE);
 812     }
 813
 814     /* Create Entertainment subfolder and fill if the exe is available */
 815     if (CreateShortcutFolder(CSIDL_PROGRAMS, IDS_SYS_ENTERTAINMENT, szFolder, sizeof(szFolder)/sizeof(szFolder[0])))
 816     {
 817         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_MPLAY32, _T("%SystemRoot%\\system32\\mplay32.exe"), IDS_CMT_MPLAY32, TRUE);
 818         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_SNDVOL32, _T("%SystemRoot%\\system32\\sndvol32.exe"), IDS_CMT_SNDVOL32, TRUE);
 819         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_SNDREC32, _T("%SystemRoot%\\system32\\sndrec32.exe"), IDS_CMT_SNDREC32, TRUE);
 820     }
 821
 822     /* Create Games subfolder and fill if the exe is available */
 823     if (CreateShortcutFolder(CSIDL_PROGRAMS, IDS_GAMES, szFolder, sizeof(szFolder)/sizeof(szFolder[0])))
 824     {
 825         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_SOLITAIRE, _T("%SystemRoot%\\system32\\sol.exe"), IDS_CMT_SOLITAIRE, TRUE);
 826         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_WINEMINE, _T("%SystemRoot%\\system32\\winmine.exe"), IDS_CMT_WINEMINE, TRUE);
 827         CreateShortcut(CSIDL_PROGRAMS, szFolder, IDS_SHORT_SPIDER, _T("%SystemRoot%\\system32\\spider.exe"), IDS_CMT_SPIDER, TRUE);
 828     }
 829
 830     CoUninitialize();
 831
 832     return TRUE;
 833 }
 834
 835 static BOOL
 836 SetSetupType(DWORD dwSetupType)
 837 {
 838     DWORD dwError;
 839     HKEY hKey;
 840
 841     dwError = RegOpenKeyExW(
 842         HKEY_LOCAL_MACHINE,
 843         L"SYSTEM\\Setup",
 844         0,
 845         KEY_SET_VALUE,
 846         &hKey);
 847     if (dwError != ERROR_SUCCESS)
 848         return FALSE;
 849
 850     dwError = RegSetValueExW(
 851         hKey,
 852         L"SetupType",
 853         0,
 854         REG_DWORD,
 855         (LPBYTE)&dwSetupType,
 856         sizeof(DWORD));
 857     RegCloseKey(hKey);
 858     if (dwError != ERROR_SUCCESS)
 859         return FALSE;
 860
 861     return TRUE;
 862 }
 863
 864 DWORD WINAPI
 865 InstallReactOS(HINSTANCE hInstance)
 866 {
 867     TCHAR szBuffer[MAX_PATH];
 868     DWORD LastError;
 869     HANDLE token;
 870     TOKEN_PRIVILEGES privs;
 871     HKEY hKey;
 872
 873     InitializeSetupActionLog(FALSE);
 874     LogItem(SYSSETUP_SEVERITY_INFORMATION, L"Installing ReactOS");
 875
 876     if (!InitializeProfiles())
 877     {
 878         DebugPrint("InitializeProfiles() failed");
 879         return 0;
 880     }
 881
 882     if (!CreateShortcuts())
 883     {
 884         DebugPrint("InitializeProfiles() failed");
 885         return 0;
 886     }
 887
 888     /* Initialize the Security Account Manager (SAM) */
 889     if (!SamInitializeSAM())
 890     {
 891         DebugPrint("SamInitializeSAM() failed!");
 892         return 0;
 893     }
 894
 895     /* Create the semi-random Domain-SID */
 896     if (!CreateRandomSid(&DomainSid))
 897     {
 898         DebugPrint("Domain-SID creation failed!");
 899         return 0;
 900     }
 901
 902     /* Set the Domain SID (aka Computer SID) */
 903     if (!SamSetDomainSid(DomainSid))
 904     {
 905         DebugPrint("SamSetDomainSid() failed!");
 906         RtlFreeSid(DomainSid);
 907         return 0;
 908     }
 909
 910     /* Append the Admin-RID */
 911     AppendRidToSid(&AdminSid, DomainSid, DOMAIN_USER_RID_ADMIN);
 912
 913     CreateTempDir(L"TEMP");
 914     CreateTempDir(L"TMP");
 915
 916     if (GetWindowsDirectory(szBuffer, sizeof(szBuffer) / sizeof(TCHAR)))
 917     {
 918         if (RegOpenKeyExW(HKEY_LOCAL_MACHINE,
 919                           L"SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion",
 920                           0,
 921                           KEY_WRITE,
 922                           &hKey) == ERROR_SUCCESS)
 923         {
 924             RegSetValueExW(hKey,
 925                            L"PathName",
 926                            0,
 927                            REG_SZ,
 928                            (LPBYTE)szBuffer,
 929                            (wcslen(szBuffer) + 1) * sizeof(WCHAR));
 930
 931             RegSetValueExW(hKey,
 932                            L"SystemRoot",
 933                            0,
 934                            REG_SZ,
 935                            (LPBYTE)szBuffer,
 936                            (wcslen(szBuffer) + 1) * sizeof(WCHAR));
 937
 938             RegCloseKey(hKey);
 939         }
 940
 941         PathAddBackslash(szBuffer);
 942         _tcscat(szBuffer, _T("system"));
 943         CreateDirectory(szBuffer, NULL);
 944     }
 945
 946     if (!CommonInstall())
 947         return 0;
 948
 949     InstallWizard();
 950
 951     /* Create the Administrator account */
 952     if (!SamCreateUser(L"Administrator", L"", AdminSid))
 953     {
 954         /* Check what the error was.
 955          * If the Admin Account already exists, then it means Setup
 956          * wasn't allowed to finish properly. Instead of rebooting
 957          * and not completing it, let it restart instead
 958          */
 959         LastError = GetLastError();
 960         if (LastError != ERROR_USER_EXISTS)
 961         {
 962             DebugPrint("SamCreateUser() failed!");
 963             RtlFreeSid(AdminSid);
 964             RtlFreeSid(DomainSid);
 965             return 0;
 966         }
 967     }
 968
 969     RtlFreeSid(AdminSid);
 970     RtlFreeSid(DomainSid);
 971
 972     /* ROS HACK, as long as NtUnloadKey is not implemented */
 973     {
 974         NTSTATUS Status = NtUnloadKey(NULL);
 975         if (Status == STATUS_NOT_IMPLEMENTED)
 976         {
 977             /* Create the Administrator profile */
 978             PROFILEINFOW ProfileInfo;
 979             HANDLE hToken;
 980             BOOL ret;
 981 #define LOGON32_LOGON_NETWORK 3
 982             ret = LogonUserW(L"Administrator", L"", L"", LOGON32_LOGON_NETWORK, LOGON32_PROVIDER_DEFAULT, &hToken);
 983             if (!ret)
 984             {
 985                 DebugPrint("LogonUserW() failed!");
 986                 return 0;
 987             }
 988             ZeroMemory(&ProfileInfo, sizeof(PROFILEINFOW));
 989             ProfileInfo.dwSize = sizeof(PROFILEINFOW);
 990             ProfileInfo.lpUserName = L"Administrator";
 991             ProfileInfo.dwFlags = PI_NOUI;
 992             LoadUserProfileW(hToken, &ProfileInfo);
 993             CloseHandle(hToken);
 994         }
 995         else
 996         {
 997             DPRINT1("ROS HACK not needed anymore. Please remove it\n");
 998         }
 999     }
1000     /* END OF ROS HACK */
1001
1002     SetupCloseInfFile(hSysSetupInf);
1003     SetSetupType(0);
1004
1005     LogItem(SYSSETUP_SEVERITY_INFORMATION, L"Installing ReactOS done");
1006     TerminateSetupActionLog();
1007
1008     /* Get shutdown privilege */
1009     if (! OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &token))
1010     {
1011         DebugPrint("OpenProcessToken() failed!");
1012         return 0;
1013     }
1014     if (!LookupPrivilegeValue(
1015         NULL,
1016         SE_SHUTDOWN_NAME,
1017         &privs.Privileges[0].Luid))
1018     {
1019         DebugPrint("LookupPrivilegeValue() failed!");
1020         return 0;
1021     }
1022     privs.PrivilegeCount = 1;
1023     privs.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
1024     if (AdjustTokenPrivileges(
1025         token,
1026         FALSE,
1027         &privs,
1028         0,
1029         (PTOKEN_PRIVILEGES)NULL,
1030         NULL) == 0)
1031     {
1032         DebugPrint("AdjustTokenPrivileges() failed!");
1033         return 0;
1034     }
1035
1036     ExitWindowsEx(EWX_REBOOT, 0);
1037     return 0;
1038 }
1039
1040
1041 /*
1042  * @unimplemented
1043  */
1044 DWORD WINAPI
1045 SetupChangeFontSize(
1046     IN HANDLE hWnd,
1047     IN LPCWSTR lpszFontSize)
1048 {
1049     SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
1050     return FALSE;
1051 }
1052
1053 /*
1054  * @unimplemented
1055  */
1056 DWORD WINAPI
1057 SetupChangeLocaleEx(HWND hWnd,
1058                     LCID Lcid,
1059                     LPCWSTR lpSrcRootPath,
1060                     char Unknown,
1061                     DWORD dwUnused1,
1062                     DWORD dwUnused2)
1063 {
1064     SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
1065     return FALSE;
1066 }
1067
1068 /*
1069  * @implemented
1070  */
1071 DWORD WINAPI
1072 SetupChangeLocale(HWND hWnd, LCID Lcid)
1073 {
1074     return SetupChangeLocaleEx(hWnd, Lcid, NULL, 0, 0, 0);
1075 }