xref: /reactos/dll/win32/syssetup/install.c (revision bbabe248) 
1 /*
2  * COPYRIGHT:         See COPYING in the top level directory
3  * PROJECT:           ReactOS system libraries
4  * PURPOSE:           System setup
5  * FILE:              dll/win32/syssetup/install.c
6  * PROGRAMER:         Eric Kohl
7  */
8 
9 /* INCLUDES *****************************************************************/
10 
11 #include "precomp.h"
12 
13 #define COBJMACROS
14 
15 #include <io.h>
16 #include <wincon.h>
17 #include <winnls.h>
18 #include <winsvc.h>
19 #include <userenv.h>
20 #include <shlobj.h>
21 #include <shlwapi.h>
22 #include <shobjidl.h>
23 #include <rpcproxy.h>
24 #include <ndk/cmfuncs.h>
25 
26 #define NDEBUG
27 #include <debug.h>
28 
29 //DWORD WINAPI
30 //CMP_WaitNoPendingInstallEvents(DWORD dwTimeout);
31 
32 DWORD WINAPI
33 SetupStartService(LPCWSTR lpServiceName, BOOL bWait);
34 
35 /* GLOBALS ******************************************************************/
36 
37 HINF hSysSetupInf = INVALID_HANDLE_VALUE;
38 ADMIN_INFO AdminInfo;
39 
40 typedef struct _DLG_DATA
41 {
42     HBITMAP hLogoBitmap;
43     HBITMAP hBarBitmap;
44     HWND hWndBarCtrl;
45     DWORD BarCounter;
46     DWORD BarWidth;
47     DWORD BarHeight;
48 } DLG_DATA, *PDLG_DATA;
49 
50 /* FUNCTIONS ****************************************************************/
51 
52 static VOID
53 FatalError(char *pszFmt,...)
54 {
55     char szBuffer[512];
56     va_list ap;
57 
58     va_start(ap, pszFmt);
59     vsprintf(szBuffer, pszFmt, ap);
60     va_end(ap);
61 
62     LogItem(NULL, L"Failed");
63 
64     strcat(szBuffer, "\nRebooting now!");
65     MessageBoxA(NULL,
66                 szBuffer,
67                 "ReactOS Setup",
68                 MB_OK);
69 }
70 
71 static HRESULT
72 CreateShellLink(
73     LPCWSTR pszLinkPath,
74     LPCWSTR pszCmd,
75     LPCWSTR pszArg,
76     LPCWSTR pszDir,
77     LPCWSTR pszIconPath,
78     INT iIconNr,
79     LPCWSTR pszComment)
80 {
81     IShellLinkW *psl;
82     IPersistFile *ppf;
83 
84     HRESULT hr = CoCreateInstance(&CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, &IID_IShellLink, (LPVOID*)&psl);
85 
86     if (SUCCEEDED(hr))
87     {
88         hr = IShellLinkW_SetPath(psl, pszCmd);
89 
90         if (pszArg)
91             hr = IShellLinkW_SetArguments(psl, pszArg);
92 
93         if (pszDir)
94             hr = IShellLinkW_SetWorkingDirectory(psl, pszDir);
95 
96         if (pszIconPath)
97             hr = IShellLinkW_SetIconLocation(psl, pszIconPath, iIconNr);
98 
99         if (pszComment)
100             hr = IShellLinkW_SetDescription(psl, pszComment);
101 
102         hr = IShellLinkW_QueryInterface(psl, &IID_IPersistFile, (LPVOID*)&ppf);
103 
104         if (SUCCEEDED(hr))
105         {
106             hr = IPersistFile_Save(ppf, pszLinkPath, TRUE);
107             IPersistFile_Release(ppf);
108         }
109 
110         IShellLinkW_Release(psl);
111     }
112 
113     return hr;
114 }
115 
116 
117 static BOOL
118 CreateShortcut(
119     LPCWSTR pszFolder,
120     LPCWSTR pszName,
121     LPCWSTR pszCommand,
122     LPCWSTR pszDescription,
123     INT iIconNr,
124     LPCWSTR pszWorkingDir)
125 {
126     DWORD dwLen;
127     LPWSTR Ptr;
128     LPWSTR lpFilePart;
129     WCHAR szPath[MAX_PATH];
130     WCHAR szWorkingDirBuf[MAX_PATH];
131 
132     /* If no working directory is provided, try to compute a default one */
133     if (pszWorkingDir == NULL || pszWorkingDir[0] == L'\0')
134     {
135         if (ExpandEnvironmentStringsW(pszCommand, szPath, ARRAYSIZE(szPath)) == 0)
136             wcscpy(szPath, pszCommand);
137 
138         dwLen = GetFullPathNameW(szPath,
139                                  ARRAYSIZE(szWorkingDirBuf),
140                                  szWorkingDirBuf,
141                                  &lpFilePart);
142         if (dwLen != 0 && dwLen <= ARRAYSIZE(szWorkingDirBuf))
143         {
144             /* Since those should only be called with (.exe) files,
145                lpFilePart has not to be NULL */
146             ASSERT(lpFilePart != NULL);
147 
148             /* We're only interested in the path. Cut the file name off.
149                Also remove the trailing backslash unless the working directory
150                is only going to be a drive, i.e. C:\ */
151             *(lpFilePart--) = L'\0';
152             if (!(lpFilePart - szWorkingDirBuf == 2 &&
153                   szWorkingDirBuf[1] == L':' && szWorkingDirBuf[2] == L'\\'))
154             {
155                 *lpFilePart = L'\0';
156             }
157             pszWorkingDir = szWorkingDirBuf;
158         }
159     }
160 
161     /* If we failed to compute a working directory, just do not use one */
162     if (pszWorkingDir && pszWorkingDir[0] == L'\0')
163         pszWorkingDir = NULL;
164 
165     /* Build the shortcut file name */
166     wcscpy(szPath, pszFolder);
167     Ptr = PathAddBackslash(szPath);
168     wcscpy(Ptr, pszName);
169 
170     /* Create the shortcut */
171     return SUCCEEDED(CreateShellLink(szPath,
172                                      pszCommand,
173                                      L"",
174                                      pszWorkingDir,
175                                      /* Special value to indicate no icon */
176                                      (iIconNr != -1 ? pszCommand : NULL),
177                                      iIconNr,
178                                      pszDescription));
179 }
180 
181 
182 static BOOL CreateShortcutsFromSection(HINF hinf, LPWSTR pszSection, LPCWSTR pszFolder)
183 {
184     INFCONTEXT Context;
185     DWORD dwFieldCount;
186     INT iIconNr;
187     WCHAR szCommand[MAX_PATH];
188     WCHAR szName[MAX_PATH];
189     WCHAR szDescription[MAX_PATH];
190     WCHAR szDirectory[MAX_PATH];
191 
192     if (!SetupFindFirstLine(hinf, pszSection, NULL, &Context))
193         return FALSE;
194 
195     do
196     {
197         dwFieldCount = SetupGetFieldCount(&Context);
198         if (dwFieldCount < 3)
199             continue;
200 
201         if (!SetupGetStringFieldW(&Context, 1, szCommand, ARRAYSIZE(szCommand), NULL))
202             continue;
203 
204         if (!SetupGetStringFieldW(&Context, 2, szName, ARRAYSIZE(szName), NULL))
205             continue;
206 
207         if (!SetupGetStringFieldW(&Context, 3, szDescription, ARRAYSIZE(szDescription), NULL))
208             continue;
209 
210         if (dwFieldCount < 4 || !SetupGetIntField(&Context, 4, &iIconNr))
211             iIconNr = -1; /* Special value to indicate no icon */
212 
213         if (dwFieldCount < 5 || !SetupGetStringFieldW(&Context, 5, szDirectory, ARRAYSIZE(szDirectory), NULL))
214             szDirectory[0] = L'\0';
215 
216         wcscat(szName, L".lnk");
217 
218         CreateShortcut(pszFolder, szName, szCommand, szDescription, iIconNr, szDirectory);
219 
220     } while (SetupFindNextLine(&Context, &Context));
221 
222     return TRUE;
223 }
224 
225 static BOOL CreateShortcuts(HINF hinf, LPCWSTR szSection)
226 {
227     INFCONTEXT Context;
228     WCHAR szPath[MAX_PATH];
229     WCHAR szFolder[MAX_PATH];
230     WCHAR szFolderSection[MAX_PATH];
231     INT csidl;
232 
233     CoInitialize(NULL);
234 
235     if (!SetupFindFirstLine(hinf, szSection, NULL, &Context))
236         return FALSE;
237 
238     do
239     {
240         if (SetupGetFieldCount(&Context) < 2)
241             continue;
242 
243         if (!SetupGetStringFieldW(&Context, 0, szFolderSection, ARRAYSIZE(szFolderSection), NULL))
244             continue;
245 
246         if (!SetupGetIntField(&Context, 1, &csidl))
247             continue;
248 
249         if (!SetupGetStringFieldW(&Context, 2, szFolder, ARRAYSIZE(szFolder), NULL))
250             continue;
251 
252         if (FAILED(SHGetFolderPathAndSubDirW(NULL, csidl|CSIDL_FLAG_CREATE, (HANDLE)-1, SHGFP_TYPE_DEFAULT, szFolder, szPath)))
253             continue;
254 
255         CreateShortcutsFromSection(hinf, szFolderSection, szPath);
256 
257     } while (SetupFindNextLine(&Context, &Context));
258 
259     CoUninitialize();
260 
261     return TRUE;
262 }
263 
264 static VOID
265 CreateTempDir(
266     IN LPCWSTR VarName)
267 {
268     WCHAR szTempDir[MAX_PATH];
269     WCHAR szBuffer[MAX_PATH];
270     DWORD dwLength;
271     HKEY hKey;
272 
273     if (RegOpenKeyExW(HKEY_LOCAL_MACHINE,
274                       L"SYSTEM\\CurrentControlSet\\Control\\Session Manager\\Environment",
275                       0,
276                       KEY_QUERY_VALUE,
277                       &hKey) != ERROR_SUCCESS)
278     {
279         FatalError("Error: %lu\n", GetLastError());
280         return;
281     }
282 
283     /* Get temp dir */
284     dwLength = sizeof(szBuffer);
285     if (RegQueryValueExW(hKey,
286                          VarName,
287                          NULL,
288                          NULL,
289                          (LPBYTE)szBuffer,
290                          &dwLength) != ERROR_SUCCESS)
291     {
292         FatalError("Error: %lu\n", GetLastError());
293         goto cleanup;
294     }
295 
296     /* Expand it */
297     if (!ExpandEnvironmentStringsW(szBuffer, szTempDir, ARRAYSIZE(szTempDir)))
298     {
299         FatalError("Error: %lu\n", GetLastError());
300         goto cleanup;
301     }
302 
303     /* Create profiles directory */
304     if (!CreateDirectoryW(szTempDir, NULL))
305     {
306         if (GetLastError() != ERROR_ALREADY_EXISTS)
307         {
308             FatalError("Error: %lu\n", GetLastError());
309             goto cleanup;
310         }
311     }
312 
313 cleanup:
314     RegCloseKey(hKey);
315 }
316 
317 static BOOL
318 InstallSysSetupInfDevices(VOID)
319 {
320     INFCONTEXT InfContext;
321     WCHAR szLineBuffer[256];
322     DWORD dwLineLength;
323 
324     if (!SetupFindFirstLineW(hSysSetupInf,
325                             L"DeviceInfsToInstall",
326                             NULL,
327                             &InfContext))
328     {
329         return FALSE;
330     }
331 
332     do
333     {
334         if (!SetupGetStringFieldW(&InfContext,
335                                   0,
336                                   szLineBuffer,
337                                   ARRAYSIZE(szLineBuffer),
338                                   &dwLineLength))
339         {
340             return FALSE;
341         }
342 
343         if (!SetupDiInstallClassW(NULL, szLineBuffer, DI_QUIETINSTALL, NULL))
344         {
345             return FALSE;
346         }
347     }
348     while (SetupFindNextLine(&InfContext, &InfContext));
349 
350     return TRUE;
351 }
352 
353 static BOOL
354 InstallSysSetupInfComponents(VOID)
355 {
356     INFCONTEXT InfContext;
357     WCHAR szNameBuffer[256];
358     WCHAR szSectionBuffer[256];
359     HINF hComponentInf = INVALID_HANDLE_VALUE;
360 
361     if (!SetupFindFirstLineW(hSysSetupInf,
362                              L"Infs.Always",
363                              NULL,
364                              &InfContext))
365     {
366         DPRINT("No Inf.Always section found\n");
367     }
368     else
369     {
370         do
371         {
372             if (!SetupGetStringFieldW(&InfContext,
373                                       1, // Get the component name
374                                       szNameBuffer,
375                                       ARRAYSIZE(szNameBuffer),
376                                       NULL))
377             {
378                 FatalError("Error while trying to get component name\n");
379                 return FALSE;
380             }
381 
382             if (!SetupGetStringFieldW(&InfContext,
383                                       2, // Get the component install section
384                                       szSectionBuffer,
385                                       ARRAYSIZE(szSectionBuffer),
386                                       NULL))
387             {
388                 FatalError("Error while trying to get component install section\n");
389                 return FALSE;
390             }
391 
392             DPRINT("Trying to execute install section '%S' from '%S'\n", szSectionBuffer, szNameBuffer);
393 
394             hComponentInf = SetupOpenInfFileW(szNameBuffer,
395                                               NULL,
396                                               INF_STYLE_WIN4,
397                                               NULL);
398 
399             if (hComponentInf == INVALID_HANDLE_VALUE)
400             {
401                 FatalError("SetupOpenInfFileW() failed to open '%S' (Error: %lu)\n", szNameBuffer, GetLastError());
402                 return FALSE;
403             }
404 
405             if (!SetupInstallFromInfSectionW(NULL,
406                                              hComponentInf,
407                                              szSectionBuffer,
408                                              SPINST_ALL,
409                                              NULL,
410                                              NULL,
411                                              SP_COPY_NEWER,
412                                              SetupDefaultQueueCallbackW,
413                                              NULL,
414                                              NULL,
415                                              NULL))
416            {
417                 FatalError("Error while trying to install : %S (Error: %lu)\n", szNameBuffer, GetLastError());
418                 SetupCloseInfFile(hComponentInf);
419                 return FALSE;
420            }
421 
422            SetupCloseInfFile(hComponentInf);
423         }
424         while (SetupFindNextLine(&InfContext, &InfContext));
425     }
426 
427     return TRUE;
428 }
429 
430 
431 
432 BOOL
433 RegisterTypeLibraries(HINF hinf, LPCWSTR szSection)
434 {
435     INFCONTEXT InfContext;
436     BOOL res;
437     WCHAR szName[MAX_PATH];
438     WCHAR szPath[MAX_PATH];
439     INT csidl;
440     LPWSTR p;
441     HMODULE hmod;
442     HRESULT hret;
443 
444     /* Begin iterating the entries in the inf section */
445     res = SetupFindFirstLine(hinf, szSection, NULL, &InfContext);
446     if (!res) return FALSE;
447 
448     do
449     {
450         /* Get the name of the current type library */
451         if (!SetupGetStringFieldW(&InfContext, 1, szName, ARRAYSIZE(szName), NULL))
452         {
453             FatalError("SetupGetStringFieldW failed\n");
454             continue;
455         }
456 
457         if (!SetupGetIntField(&InfContext, 2, &csidl))
458             csidl = CSIDL_SYSTEM;
459 
460         hret = SHGetFolderPathW(NULL, csidl, NULL, 0, szPath);
461         if (FAILED(hret))
462         {
463             FatalError("SHGetFolderPathW failed hret=0x%lx\n", hret);
464             continue;
465         }
466 
467         p = PathAddBackslash(szPath);
468         wcscpy(p, szName);
469 
470         hmod = LoadLibraryW(szPath);
471         if (hmod == NULL)
472         {
473             FatalError("LoadLibraryW failed\n");
474             continue;
475         }
476 
477         __wine_register_resources(hmod);
478 
479     } while (SetupFindNextLine(&InfContext, &InfContext));
480 
481     return TRUE;
482 }
483 
484 static BOOL
485 EnableUserModePnpManager(VOID)
486 {
487     SC_HANDLE hSCManager = NULL;
488     SC_HANDLE hService = NULL;
489     SERVICE_STATUS_PROCESS ServiceStatus;
490     BOOL bRet = FALSE;
491     DWORD BytesNeeded, WaitTime;
492 
493     hSCManager = OpenSCManagerW(NULL, NULL, SC_MANAGER_ENUMERATE_SERVICE);
494     if (hSCManager == NULL)
495     {
496         DPRINT1("Unable to open the service control manager.\n");
497         DPRINT1("Last Error %d\n", GetLastError());
498         goto cleanup;
499     }
500 
501     hService = OpenServiceW(hSCManager,
502                             L"PlugPlay",
503                             SERVICE_CHANGE_CONFIG | SERVICE_START | SERVICE_QUERY_STATUS);
504     if (hService == NULL)
505     {
506         DPRINT1("Unable to open PlugPlay service\n");
507         goto cleanup;
508     }
509 
510     bRet = ChangeServiceConfigW(hService,
511                                 SERVICE_NO_CHANGE,
512                                 SERVICE_AUTO_START,
513                                 SERVICE_NO_CHANGE,
514                                 NULL, NULL, NULL,
515                                 NULL, NULL, NULL, NULL);
516     if (!bRet)
517     {
518         DPRINT1("Unable to change the service configuration\n");
519         goto cleanup;
520     }
521 
522     bRet = StartServiceW(hService, 0, NULL);
523     if (!bRet && (GetLastError() != ERROR_SERVICE_ALREADY_RUNNING))
524     {
525         DPRINT1("Unable to start service\n");
526         goto cleanup;
527     }
528 
529     while (TRUE)
530     {
531         bRet = QueryServiceStatusEx(hService,
532                                     SC_STATUS_PROCESS_INFO,
533                                     (LPBYTE)&ServiceStatus,
534                                     sizeof(ServiceStatus),
535                                     &BytesNeeded);
536         if (!bRet)
537         {
538             DPRINT1("QueryServiceStatusEx() failed for PlugPlay service (error 0x%x)\n", GetLastError());
539             goto cleanup;
540         }
541 
542         if (ServiceStatus.dwCurrentState != SERVICE_START_PENDING)
543             break;
544 
545         WaitTime = ServiceStatus.dwWaitHint / 10;
546         if (WaitTime < 1000) WaitTime = 1000;
547         else if (WaitTime > 10000) WaitTime = 10000;
548         Sleep(WaitTime);
549     };
550 
551     if (ServiceStatus.dwCurrentState != SERVICE_RUNNING)
552     {
553         bRet = FALSE;
554         DPRINT1("Failed to start PlugPlay service\n");
555         goto cleanup;
556     }
557 
558     bRet = TRUE;
559 
560 cleanup:
561     if (hService != NULL)
562         CloseServiceHandle(hService);
563     if (hSCManager != NULL)
564         CloseServiceHandle(hSCManager);
565     return bRet;
566 }
567 
568 static INT_PTR CALLBACK
569 StatusMessageWindowProc(
570     IN HWND hwndDlg,
571     IN UINT uMsg,
572     IN WPARAM wParam,
573     IN LPARAM lParam)
574 {
575     PDLG_DATA pDlgData;
576     UNREFERENCED_PARAMETER(wParam);
577 
578     pDlgData = (PDLG_DATA)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
579 
580     /* pDlgData is required for each case except WM_INITDIALOG */
581     if (uMsg != WM_INITDIALOG && pDlgData == NULL) return FALSE;
582 
583     switch (uMsg)
584     {
585         case WM_INITDIALOG:
586         {
587             BITMAP bm;
588             WCHAR szMsg[256];
589 
590             /* Allocate pDlgData */
591             pDlgData = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*pDlgData));
592             if (pDlgData)
593             {
594                 /* Set pDlgData to GWLP_USERDATA, so we can get it for new messages */
595                 SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pDlgData);
596 
597                 /* Load bitmaps */
598                 pDlgData->hLogoBitmap = LoadImageW(hDllInstance,
599                                                     MAKEINTRESOURCEW(IDB_REACTOS), IMAGE_BITMAP,
600                                                     0, 0, LR_DEFAULTCOLOR);
601 
602                 pDlgData->hBarBitmap = LoadImageW(hDllInstance, MAKEINTRESOURCEW(IDB_LINE),
603                                                 IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
604                 GetObject(pDlgData->hBarBitmap, sizeof(bm), &bm);
605                 pDlgData->BarWidth = bm.bmWidth;
606                 pDlgData->BarHeight = bm.bmHeight;
607 
608                 if (pDlgData->hLogoBitmap && pDlgData->hBarBitmap)
609                 {
610                     if (SetTimer(hwndDlg, IDT_BAR, 20, NULL) == 0)
611                     {
612                         DPRINT1("SetTimer(IDT_BAR) failed: %lu\n", GetLastError());
613                     }
614 
615                     /* Get the animation bar control */
616                     pDlgData->hWndBarCtrl = GetDlgItem(hwndDlg, IDC_BAR);
617                 }
618             }
619 
620             /* Get and set status text */
621             if (!LoadStringW(hDllInstance, IDS_STATUS_INSTALL_DEV, szMsg, ARRAYSIZE(szMsg)))
622                 return FALSE;
623             SetDlgItemTextW(hwndDlg, IDC_STATUSLABEL, szMsg);
624 
625             return TRUE;
626         }
627 
628         case WM_TIMER:
629         {
630             if (pDlgData->hBarBitmap)
631             {
632                 /*
633                  * Default rotation bar image width is 413 (same as logo)
634                  * We can divide 413 by 7 without remainder
635                  */
636                 pDlgData->BarCounter = (pDlgData->BarCounter + 7) % pDlgData->BarWidth;
637                 InvalidateRect(pDlgData->hWndBarCtrl, NULL, FALSE);
638                 UpdateWindow(pDlgData->hWndBarCtrl);
639             }
640             return TRUE;
641         }
642 
643         case WM_DRAWITEM:
644         {
645             LPDRAWITEMSTRUCT lpDis = (LPDRAWITEMSTRUCT)lParam;
646 
647             if (lpDis->CtlID != IDC_BAR)
648             {
649                 return FALSE;
650             }
651 
652             if (pDlgData->hBarBitmap)
653             {
654                 HDC hdcMem;
655                 HGDIOBJ hOld;
656                 DWORD off = pDlgData->BarCounter;
657                 DWORD iw = pDlgData->BarWidth;
658                 DWORD ih = pDlgData->BarHeight;
659 
660                 hdcMem = CreateCompatibleDC(lpDis->hDC);
661                 hOld = SelectObject(hdcMem, pDlgData->hBarBitmap);
662                 BitBlt(lpDis->hDC, off, 0, iw - off, ih, hdcMem, 0, 0, SRCCOPY);
663                 BitBlt(lpDis->hDC, 0, 0, off, ih, hdcMem, iw - off, 0, SRCCOPY);
664                 SelectObject(hdcMem, hOld);
665                 DeleteDC(hdcMem);
666                 return TRUE;
667             }
668             return FALSE;
669         }
670 
671         case WM_DESTROY:
672         {
673             if (pDlgData->hBarBitmap)
674             {
675                 KillTimer(hwndDlg, IDT_BAR);
676             }
677 
678             DeleteObject(pDlgData->hLogoBitmap);
679             DeleteObject(pDlgData->hBarBitmap);
680             HeapFree(GetProcessHeap(), 0, pDlgData);
681             return TRUE;
682         }
683     }
684     return FALSE;
685 }
686 
687 static DWORD WINAPI
688 ShowStatusMessageThread(
689     IN LPVOID lpParameter)
690 {
691     HWND hWnd;
692     MSG Msg;
693     UNREFERENCED_PARAMETER(lpParameter);
694 
695     hWnd = CreateDialogParam(hDllInstance,
696                              MAKEINTRESOURCE(IDD_STATUSWINDOW_DLG),
697                              GetDesktopWindow(),
698                              StatusMessageWindowProc,
699                              (LPARAM)NULL);
700     if (!hWnd)
701         return 0;
702 
703     ShowWindow(hWnd, SW_SHOW);
704 
705     /* Message loop for the Status window */
706     while (GetMessage(&Msg, NULL, 0, 0))
707     {
708         TranslateMessage(&Msg);
709         DispatchMessage(&Msg);
710     }
711 
712     EndDialog(hWnd, 0);
713 
714     return 0;
715 }
716 
717 static LONG
718 ReadRegSzKey(
719     IN HKEY hKey,
720     IN LPCWSTR pszKey,
721     OUT LPWSTR* pValue)
722 {
723     LONG rc;
724     DWORD dwType;
725     DWORD cbData = 0;
726     LPWSTR pwszValue;
727 
728     if (!pValue)
729         return ERROR_INVALID_PARAMETER;
730 
731     *pValue = NULL;
732     rc = RegQueryValueExW(hKey, pszKey, NULL, &dwType, NULL, &cbData);
733     if (rc != ERROR_SUCCESS)
734         return rc;
735     if (dwType != REG_SZ)
736         return ERROR_FILE_NOT_FOUND;
737     pwszValue = HeapAlloc(GetProcessHeap(), 0, cbData + sizeof(WCHAR));
738     if (!pwszValue)
739         return ERROR_NOT_ENOUGH_MEMORY;
740     rc = RegQueryValueExW(hKey, pszKey, NULL, NULL, (LPBYTE)pwszValue, &cbData);
741     if (rc != ERROR_SUCCESS)
742     {
743         HeapFree(GetProcessHeap(), 0, pwszValue);
744         return rc;
745     }
746     /* NULL-terminate the string */
747     pwszValue[cbData / sizeof(WCHAR)] = '\0';
748 
749     *pValue = pwszValue;
750     return ERROR_SUCCESS;
751 }
752 
753 static BOOL
754 IsConsoleBoot(VOID)
755 {
756     HKEY hControlKey = NULL;
757     LPWSTR pwszSystemStartOptions = NULL;
758     LPWSTR pwszCurrentOption, pwszNextOption; /* Pointers into SystemStartOptions */
759     BOOL bConsoleBoot = FALSE;
760     LONG rc;
761 
762     rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
763                        L"SYSTEM\\CurrentControlSet\\Control",
764                        0,
765                        KEY_QUERY_VALUE,
766                        &hControlKey);
767     if (rc != ERROR_SUCCESS)
768         goto cleanup;
769 
770     rc = ReadRegSzKey(hControlKey, L"SystemStartOptions", &pwszSystemStartOptions);
771     if (rc != ERROR_SUCCESS)
772         goto cleanup;
773 
774     /* Check for CONSOLE switch in SystemStartOptions */
775     pwszCurrentOption = pwszSystemStartOptions;
776     while (pwszCurrentOption)
777     {
778         pwszNextOption = wcschr(pwszCurrentOption, L' ');
779         if (pwszNextOption)
780             *pwszNextOption = L'\0';
781         if (wcsicmp(pwszCurrentOption, L"CONSOLE") == 0)
782         {
783             DPRINT("Found %S. Switching to console boot\n", pwszCurrentOption);
784             bConsoleBoot = TRUE;
785             goto cleanup;
786         }
787         pwszCurrentOption = pwszNextOption ? pwszNextOption + 1 : NULL;
788     }
789 
790 cleanup:
791     if (hControlKey != NULL)
792         RegCloseKey(hControlKey);
793     if (pwszSystemStartOptions)
794         HeapFree(GetProcessHeap(), 0, pwszSystemStartOptions);
795     return bConsoleBoot;
796 }
797 
798 static BOOL
799 CommonInstall(VOID)
800 {
801     HANDLE hThread = NULL;
802     BOOL bResult = FALSE;
803 
804     hSysSetupInf = SetupOpenInfFileW(L"syssetup.inf",
805                                      NULL,
806                                      INF_STYLE_WIN4,
807                                      NULL);
808     if (hSysSetupInf == INVALID_HANDLE_VALUE)
809     {
810         FatalError("SetupOpenInfFileW() failed to open 'syssetup.inf' (Error: %lu)\n", GetLastError());
811         return FALSE;
812     }
813 
814     if (!InstallSysSetupInfDevices())
815     {
816         FatalError("InstallSysSetupInfDevices() failed!\n");
817         goto Exit;
818     }
819 
820     if(!InstallSysSetupInfComponents())
821     {
822         FatalError("InstallSysSetupInfComponents() failed!\n");
823         goto Exit;
824     }
825 
826     if (!IsConsoleBoot())
827     {
828         hThread = CreateThread(NULL,
829                                0,
830                                ShowStatusMessageThread,
831                                NULL,
832                                0,
833                                NULL);
834     }
835 
836     if (!EnableUserModePnpManager())
837     {
838         FatalError("EnableUserModePnpManager() failed!\n");
839         goto Exit;
840     }
841 
842     if (CMP_WaitNoPendingInstallEvents(INFINITE) != WAIT_OBJECT_0)
843     {
844         FatalError("CMP_WaitNoPendingInstallEvents() failed!\n");
845         goto Exit;
846     }
847 
848     bResult = TRUE;
849 
850 Exit:
851 
852     if (bResult == FALSE)
853     {
854         SetupCloseInfFile(hSysSetupInf);
855     }
856 
857     if (hThread != NULL)
858     {
859         PostThreadMessage(GetThreadId(hThread), WM_QUIT, 0, 0);
860         WaitForSingleObject(hThread, INFINITE);
861         CloseHandle(hThread);
862     }
863 
864     return bResult;
865 }
866 
867 static
868 DWORD
869 InstallLiveCD(VOID)
870 {
871     STARTUPINFOW StartupInfo;
872     PROCESS_INFORMATION ProcessInformation;
873     BOOL bRes;
874 
875     if (!CommonInstall())
876         goto error;
877 
878     /* Install the TCP/IP protocol driver */
879     bRes = InstallNetworkComponent(L"MS_TCPIP");
880     if (!bRes && GetLastError() != ERROR_FILE_NOT_FOUND)
881     {
882         DPRINT("InstallNetworkComponent() failed with error 0x%lx\n", GetLastError());
883     }
884     else
885     {
886         /* Start the TCP/IP protocol driver */
887         SetupStartService(L"Tcpip", FALSE);
888         SetupStartService(L"Dhcp", FALSE);
889         SetupStartService(L"Dnscache", FALSE);
890     }
891 
892     /* Register components */
893     _SEH2_TRY
894     {
895         if (!SetupInstallFromInfSectionW(NULL,
896                                          hSysSetupInf, L"RegistrationPhase2",
897                                          SPINST_ALL,
898                                          0, NULL, 0, NULL, NULL, NULL, NULL))
899         {
900             DPRINT1("SetupInstallFromInfSectionW failed!\n");
901         }
902 
903         RegisterTypeLibraries(hSysSetupInf, L"TypeLibraries");
904     }
905     _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
906     {
907         DPRINT1("Catching exception\n");
908     }
909     _SEH2_END;
910 
911     SetupCloseInfFile(hSysSetupInf);
912 
913     /* Run the shell */
914     ZeroMemory(&StartupInfo, sizeof(StartupInfo));
915     StartupInfo.cb = sizeof(StartupInfo);
916     bRes = CreateProcessW(L"userinit.exe",
917                           NULL,
918                           NULL,
919                           NULL,
920                           FALSE,
921                           0,
922                           NULL,
923                           NULL,
924                           &StartupInfo,
925                           &ProcessInformation);
926     if (!bRes)
927         goto error;
928 
929     CloseHandle(ProcessInformation.hThread);
930     CloseHandle(ProcessInformation.hProcess);
931 
932     return 0;
933 
934 error:
935     MessageBoxW(
936         NULL,
937         L"Failed to load LiveCD! You can shutdown your computer, or press ENTER to reboot.",
938         L"ReactOS LiveCD",
939         MB_OK);
940     return 0;
941 }
942 
943 
944 static BOOL
945 SetSetupType(DWORD dwSetupType)
946 {
947     DWORD dwError;
948     HKEY hKey;
949 
950     dwError = RegOpenKeyExW(
951         HKEY_LOCAL_MACHINE,
952         L"SYSTEM\\Setup",
953         0,
954         KEY_SET_VALUE,
955         &hKey);
956     if (dwError != ERROR_SUCCESS)
957         return FALSE;
958 
959     dwError = RegSetValueExW(
960         hKey,
961         L"SetupType",
962         0,
963         REG_DWORD,
964         (LPBYTE)&dwSetupType,
965         sizeof(DWORD));
966     RegCloseKey(hKey);
967     if (dwError != ERROR_SUCCESS)
968         return FALSE;
969 
970     return TRUE;
971 }
972 
973 static DWORD CALLBACK
974 HotkeyThread(LPVOID Parameter)
975 {
976     ATOM hotkey;
977     MSG msg;
978 
979     DPRINT("HotkeyThread start\n");
980 
981     hotkey = GlobalAddAtomW(L"Setup Shift+F10 Hotkey");
982 
983     if (!RegisterHotKey(NULL, hotkey, MOD_SHIFT, VK_F10))
984         DPRINT1("RegisterHotKey failed with %lu\n", GetLastError());
985 
986     while (GetMessage(&msg, NULL, 0, 0))
987     {
988         if (msg.hwnd == NULL && msg.message == WM_HOTKEY && msg.wParam == hotkey)
989         {
990             STARTUPINFOW si = { sizeof(si) };
991             PROCESS_INFORMATION pi;
992 
993             if (CreateProcessW(L"cmd.exe",
994                                NULL,
995                                NULL,
996                                NULL,
997                                FALSE,
998                                CREATE_NEW_CONSOLE,
999                                NULL,
1000                                NULL,
1001                                &si,
1002                                &pi))
1003             {
1004                 CloseHandle(pi.hProcess);
1005                 CloseHandle(pi.hThread);
1006             }
1007             else
1008             {
1009                 DPRINT1("Failed to launch command prompt: %lu\n", GetLastError());
1010             }
1011         }
1012     }
1013 
1014     UnregisterHotKey(NULL, hotkey);
1015     GlobalDeleteAtom(hotkey);
1016 
1017     DPRINT("HotkeyThread terminate\n");
1018     return 0;
1019 }
1020 
1021 
1022 static
1023 BOOL
1024 InitializeProgramFilesDir(VOID)
1025 {
1026     LONG Error;
1027     HKEY hKey;
1028     DWORD dwLength;
1029     WCHAR szProgramFilesDirPath[MAX_PATH];
1030     WCHAR szCommonFilesDirPath[MAX_PATH];
1031     WCHAR szBuffer[MAX_PATH];
1032 
1033     /* Load 'Program Files' location */
1034     if (!LoadStringW(hDllInstance,
1035                      IDS_PROGRAMFILES,
1036                      szBuffer,
1037                      ARRAYSIZE(szBuffer)))
1038     {
1039         DPRINT1("Error: %lu\n", GetLastError());
1040         return FALSE;
1041     }
1042 
1043     if (!LoadStringW(hDllInstance,
1044                      IDS_COMMONFILES,
1045                      szCommonFilesDirPath,
1046                      ARRAYSIZE(szCommonFilesDirPath)))
1047     {
1048         DPRINT1("Warning: %lu\n", GetLastError());
1049     }
1050 
1051     /* Expand it */
1052     if (!ExpandEnvironmentStringsW(szBuffer,
1053                                    szProgramFilesDirPath,
1054                                    ARRAYSIZE(szProgramFilesDirPath)))
1055     {
1056         DPRINT1("Error: %lu\n", GetLastError());
1057         return FALSE;
1058     }
1059 
1060     wcscpy(szBuffer, szProgramFilesDirPath);
1061     wcscat(szBuffer, L"\\");
1062     wcscat(szBuffer, szCommonFilesDirPath);
1063 
1064     if (!ExpandEnvironmentStringsW(szBuffer,
1065                                    szCommonFilesDirPath,
1066                                    ARRAYSIZE(szCommonFilesDirPath)))
1067     {
1068         DPRINT1("Warning: %lu\n", GetLastError());
1069     }
1070 
1071     /* Store it */
1072     Error = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
1073                           L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion",
1074                           0,
1075                           KEY_SET_VALUE,
1076                           &hKey);
1077     if (Error != ERROR_SUCCESS)
1078     {
1079         DPRINT1("Error: %lu\n", Error);
1080         return FALSE;
1081     }
1082 
1083     dwLength = (wcslen(szProgramFilesDirPath) + 1) * sizeof(WCHAR);
1084     Error = RegSetValueExW(hKey,
1085                            L"ProgramFilesDir",
1086                            0,
1087                            REG_SZ,
1088                            (LPBYTE)szProgramFilesDirPath,
1089                            dwLength);
1090     if (Error != ERROR_SUCCESS)
1091     {
1092         DPRINT1("Error: %lu\n", Error);
1093         RegCloseKey(hKey);
1094         return FALSE;
1095     }
1096 
1097     dwLength = (wcslen(szCommonFilesDirPath) + 1) * sizeof(WCHAR);
1098     Error = RegSetValueExW(hKey,
1099                            L"CommonFilesDir",
1100                            0,
1101                            REG_SZ,
1102                            (LPBYTE)szCommonFilesDirPath,
1103                            dwLength);
1104     if (Error != ERROR_SUCCESS)
1105     {
1106         DPRINT1("Warning: %lu\n", Error);
1107     }
1108 
1109     RegCloseKey(hKey);
1110 
1111     /* Create directory */
1112     // FIXME: Security!
1113     if (!CreateDirectoryW(szProgramFilesDirPath, NULL))
1114     {
1115         if (GetLastError() != ERROR_ALREADY_EXISTS)
1116         {
1117             DPRINT1("Error: %lu\n", GetLastError());
1118             return FALSE;
1119         }
1120     }
1121 
1122     /* Create directory */
1123     // FIXME: Security!
1124     if (!CreateDirectoryW(szCommonFilesDirPath, NULL))
1125     {
1126         if (GetLastError() != ERROR_ALREADY_EXISTS)
1127         {
1128             DPRINT1("Warning: %lu\n", GetLastError());
1129             // return FALSE;
1130         }
1131     }
1132 
1133     return TRUE;
1134 }
1135 
1136 
1137 static
1138 VOID
1139 InitializeDefaultUserLocale(VOID)
1140 {
1141     WCHAR szBuffer[80];
1142     PWSTR ptr;
1143     HKEY hLocaleKey;
1144     DWORD ret;
1145     DWORD dwSize;
1146     LCID lcid;
1147     INT i;
1148 
1149     struct {LCTYPE LCType; PWSTR pValue;} LocaleData[] = {
1150         /* Number */
1151         {LOCALE_SDECIMAL, L"sDecimal"},
1152         {LOCALE_STHOUSAND, L"sThousand"},
1153         {LOCALE_SNEGATIVESIGN, L"sNegativeSign"},
1154         {LOCALE_SPOSITIVESIGN, L"sPositiveSign"},
1155         {LOCALE_SGROUPING, L"sGrouping"},
1156         {LOCALE_SLIST, L"sList"},
1157         {LOCALE_SNATIVEDIGITS, L"sNativeDigits"},
1158         {LOCALE_INEGNUMBER, L"iNegNumber"},
1159         {LOCALE_IDIGITS, L"iDigits"},
1160         {LOCALE_ILZERO, L"iLZero"},
1161         {LOCALE_IMEASURE, L"iMeasure"},
1162         {LOCALE_IDIGITSUBSTITUTION, L"NumShape"},
1163 
1164         /* Currency */
1165         {LOCALE_SCURRENCY, L"sCurrency"},
1166         {LOCALE_SMONDECIMALSEP, L"sMonDecimalSep"},
1167         {LOCALE_SMONTHOUSANDSEP, L"sMonThousandSep"},
1168         {LOCALE_SMONGROUPING, L"sMonGrouping"},
1169         {LOCALE_ICURRENCY, L"iCurrency"},
1170         {LOCALE_INEGCURR, L"iNegCurr"},
1171         {LOCALE_ICURRDIGITS, L"iCurrDigits"},
1172 
1173         /* Time */
1174         {LOCALE_STIMEFORMAT, L"sTimeFormat"},
1175         {LOCALE_STIME, L"sTime"},
1176         {LOCALE_S1159, L"s1159"},
1177         {LOCALE_S2359, L"s2359"},
1178         {LOCALE_ITIME, L"iTime"},
1179         {LOCALE_ITIMEMARKPOSN, L"iTimePrefix"},
1180         {LOCALE_ITLZERO, L"iTLZero"},
1181 
1182         /* Date */
1183         {LOCALE_SLONGDATE, L"sLongDate"},
1184         {LOCALE_SSHORTDATE, L"sShortDate"},
1185         {LOCALE_SDATE, L"sDate"},
1186         {LOCALE_IFIRSTDAYOFWEEK, L"iFirstDayOfWeek"},
1187         {LOCALE_IFIRSTWEEKOFYEAR, L"iFirstWeekOfYear"},
1188         {LOCALE_IDATE, L"iDate"},
1189         {LOCALE_ICALENDARTYPE, L"iCalendarType"},
1190 
1191         /* Misc */
1192         {LOCALE_SCOUNTRY, L"sCountry"},
1193         {LOCALE_SABBREVLANGNAME, L"sLanguage"},
1194         {LOCALE_ICOUNTRY, L"iCountry"},
1195         {0, NULL}};
1196 
1197     ret = RegOpenKeyExW(HKEY_USERS,
1198                         L".DEFAULT\\Control Panel\\International",
1199                         0,
1200                         KEY_READ | KEY_WRITE,
1201                         &hLocaleKey);
1202     if (ret != ERROR_SUCCESS)
1203     {
1204         return;
1205     }
1206 
1207     dwSize = 9 * sizeof(WCHAR);
1208     ret = RegQueryValueExW(hLocaleKey,
1209                            L"Locale",
1210                            NULL,
1211                            NULL,
1212                            (PBYTE)szBuffer,
1213                            &dwSize);
1214     if (ret != ERROR_SUCCESS)
1215         goto done;
1216 
1217     lcid = (LCID)wcstoul(szBuffer, &ptr, 16);
1218     if (lcid == 0)
1219         goto done;
1220 
1221     i = 0;
1222     while (LocaleData[i].pValue != NULL)
1223     {
1224         if (GetLocaleInfoW(lcid,
1225                            LocaleData[i].LCType | LOCALE_NOUSEROVERRIDE,
1226                            szBuffer,
1227                            ARRAYSIZE(szBuffer)))
1228         {
1229             RegSetValueExW(hLocaleKey,
1230                            LocaleData[i].pValue,
1231                            0,
1232                            REG_SZ,
1233                            (PBYTE)szBuffer,
1234                            (wcslen(szBuffer) + 1) * sizeof(WCHAR));
1235         }
1236 
1237         i++;
1238     }
1239 
1240 done:
1241     RegCloseKey(hLocaleKey);
1242 }
1243 
1244 
1245 static
1246 DWORD
1247 SaveDefaultUserHive(VOID)
1248 {
1249     WCHAR szDefaultUserHive[MAX_PATH];
1250     HKEY hUserKey = NULL;
1251     DWORD cchSize;
1252     DWORD dwError;
1253 
1254     DPRINT("SaveDefaultUserHive()\n");
1255 
1256     cchSize = ARRAYSIZE(szDefaultUserHive);
1257     GetDefaultUserProfileDirectoryW(szDefaultUserHive, &cchSize);
1258 
1259     wcscat(szDefaultUserHive, L"\\ntuser.dat");
1260 
1261     dwError = RegOpenKeyExW(HKEY_USERS,
1262                             L".DEFAULT",
1263                             0,
1264                             KEY_READ,
1265                             &hUserKey);
1266     if (dwError != ERROR_SUCCESS)
1267     {
1268         DPRINT1("RegOpenKeyExW() failed (Error %lu)\n", dwError);
1269         return dwError;
1270     }
1271 
1272     pSetupEnablePrivilege(L"SeBackupPrivilege", TRUE);
1273 
1274     /* Save the Default hive */
1275     dwError = RegSaveKeyExW(hUserKey,
1276                             szDefaultUserHive,
1277                             NULL,
1278                             REG_STANDARD_FORMAT);
1279     if (dwError == ERROR_ALREADY_EXISTS)
1280     {
1281         WCHAR szBackupHive[MAX_PATH];
1282 
1283         /* Build the backup hive file name by replacing the extension */
1284         wcscpy(szBackupHive, szDefaultUserHive);
1285         wcscpy(&szBackupHive[wcslen(szBackupHive) - 4], L".bak");
1286 
1287         /* Back up the existing default user hive by renaming it, replacing any possible existing old backup */
1288         if (!MoveFileExW(szDefaultUserHive,
1289                          szBackupHive,
1290                          MOVEFILE_REPLACE_EXISTING))
1291         {
1292             dwError = GetLastError();
1293             DPRINT1("Failed to create a default-user hive backup '%S', MoveFileExW failed (Error %lu)\n",
1294                     szBackupHive, dwError);
1295         }
1296         else
1297         {
1298             /* The backup has been done, retry saving the Default hive */
1299             dwError = RegSaveKeyExW(hUserKey,
1300                                     szDefaultUserHive,
1301                                     NULL,
1302                                     REG_STANDARD_FORMAT);
1303         }
1304     }
1305     if (dwError != ERROR_SUCCESS)
1306     {
1307         DPRINT1("RegSaveKeyExW() failed (Error %lu)\n", dwError);
1308     }
1309 
1310     pSetupEnablePrivilege(L"SeBackupPrivilege", FALSE);
1311 
1312     RegCloseKey(hUserKey);
1313 
1314     return dwError;
1315 }
1316 
1317 
1318 static
1319 DWORD
1320 InstallReactOS(VOID)
1321 {
1322     WCHAR szBuffer[MAX_PATH];
1323     HANDLE token;
1324     TOKEN_PRIVILEGES privs;
1325     HKEY hKey;
1326     HINF hShortcutsInf;
1327     HANDLE hHotkeyThread;
1328     BOOL ret;
1329 
1330     InitializeSetupActionLog(FALSE);
1331     LogItem(NULL, L"Installing ReactOS");
1332 
1333     CreateTempDir(L"TEMP");
1334     CreateTempDir(L"TMP");
1335 
1336     if (!InitializeProgramFilesDir())
1337     {
1338         FatalError("InitializeProgramFilesDir() failed");
1339         return 0;
1340     }
1341 
1342     if (!InitializeProfiles())
1343     {
1344         FatalError("InitializeProfiles() failed");
1345         return 0;
1346     }
1347 
1348     InitializeDefaultUserLocale();
1349 
1350     if (GetWindowsDirectoryW(szBuffer, ARRAYSIZE(szBuffer)))
1351     {
1352         if (RegOpenKeyExW(HKEY_LOCAL_MACHINE,
1353                           L"SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion",
1354                           0,
1355                           KEY_WRITE,
1356                           &hKey) == ERROR_SUCCESS)
1357         {
1358             RegSetValueExW(hKey,
1359                            L"PathName",
1360                            0,
1361                            REG_SZ,
1362                            (LPBYTE)szBuffer,
1363                            (wcslen(szBuffer) + 1) * sizeof(WCHAR));
1364 
1365             RegSetValueExW(hKey,
1366                            L"SystemRoot",
1367                            0,
1368                            REG_SZ,
1369                            (LPBYTE)szBuffer,
1370                            (wcslen(szBuffer) + 1) * sizeof(WCHAR));
1371 
1372             RegCloseKey(hKey);
1373         }
1374 
1375         PathAddBackslash(szBuffer);
1376         wcscat(szBuffer, L"system");
1377         CreateDirectory(szBuffer, NULL);
1378     }
1379 
1380     if (SaveDefaultUserHive() != ERROR_SUCCESS)
1381     {
1382         FatalError("SaveDefaultUserHive() failed");
1383         return 0;
1384     }
1385 
1386     if (!CopySystemProfile(0))
1387     {
1388         FatalError("CopySystemProfile() failed");
1389         return 0;
1390     }
1391 
1392     hHotkeyThread = CreateThread(NULL, 0, HotkeyThread, NULL, 0, NULL);
1393 
1394     if (!CommonInstall())
1395         return 0;
1396 
1397     /* Install the TCP/IP protocol driver */
1398     ret = InstallNetworkComponent(L"MS_TCPIP");
1399     if (!ret && GetLastError() != ERROR_FILE_NOT_FOUND)
1400     {
1401         DPRINT("InstallNetworkComponent() failed with error 0x%lx\n", GetLastError());
1402     }
1403     else
1404     {
1405         /* Start the TCP/IP protocol driver */
1406         SetupStartService(L"Tcpip", FALSE);
1407         SetupStartService(L"Dhcp", FALSE);
1408         SetupStartService(L"Dnscache", FALSE);
1409     }
1410 
1411     InstallWizard();
1412 
1413     InstallSecurity();
1414 
1415     SetAutoAdminLogon();
1416 
1417     hShortcutsInf = SetupOpenInfFileW(L"shortcuts.inf",
1418                                       NULL,
1419                                       INF_STYLE_WIN4,
1420                                       NULL);
1421     if (hShortcutsInf == INVALID_HANDLE_VALUE)
1422     {
1423         FatalError("Failed to open shortcuts.inf");
1424         return 0;
1425     }
1426 
1427     if (!CreateShortcuts(hShortcutsInf, L"ShortcutFolders"))
1428     {
1429         FatalError("CreateShortcuts() failed");
1430         return 0;
1431     }
1432 
1433     SetupCloseInfFile(hShortcutsInf);
1434 
1435     hShortcutsInf = SetupOpenInfFileW(L"rosapps_shortcuts.inf",
1436                                        NULL,
1437                                        INF_STYLE_WIN4,
1438                                        NULL);
1439     if (hShortcutsInf != INVALID_HANDLE_VALUE)
1440     {
1441         if (!CreateShortcuts(hShortcutsInf, L"ShortcutFolders"))
1442         {
1443             FatalError("CreateShortcuts(rosapps) failed");
1444             return 0;
1445         }
1446         SetupCloseInfFile(hShortcutsInf);
1447     }
1448 
1449     SetupCloseInfFile(hSysSetupInf);
1450     SetSetupType(0);
1451 
1452     if (hHotkeyThread)
1453     {
1454         PostThreadMessage(GetThreadId(hHotkeyThread), WM_QUIT, 0, 0);
1455         CloseHandle(hHotkeyThread);
1456     }
1457 
1458     LogItem(NULL, L"Installing ReactOS done");
1459     TerminateSetupActionLog();
1460 
1461     if (AdminInfo.Name != NULL)
1462         RtlFreeHeap(RtlGetProcessHeap(), 0, AdminInfo.Name);
1463 
1464     if (AdminInfo.Domain != NULL)
1465         RtlFreeHeap(RtlGetProcessHeap(), 0, AdminInfo.Domain);
1466 
1467     if (AdminInfo.Password != NULL)
1468         RtlFreeHeap(RtlGetProcessHeap(), 0, AdminInfo.Password);
1469 
1470     /* Get shutdown privilege */
1471     if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &token))
1472     {
1473         FatalError("OpenProcessToken() failed!");
1474         return 0;
1475     }
1476     if (!LookupPrivilegeValue(NULL,
1477                               SE_SHUTDOWN_NAME,
1478                               &privs.Privileges[0].Luid))
1479     {
1480         FatalError("LookupPrivilegeValue() failed!");
1481         return 0;
1482     }
1483     privs.PrivilegeCount = 1;
1484     privs.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
1485     if (AdjustTokenPrivileges(token,
1486                               FALSE,
1487                               &privs,
1488                               0,
1489                               (PTOKEN_PRIVILEGES)NULL,
1490                               NULL) == 0)
1491     {
1492         FatalError("AdjustTokenPrivileges() failed!");
1493         return 0;
1494     }
1495 
1496     ExitWindowsEx(EWX_REBOOT, 0);
1497     return 0;
1498 }
1499 
1500 
1501 /*
1502  * Standard Windows-compatible export, which dispatches
1503  * to either 'InstallReactOS' or 'InstallLiveCD'.
1504  */
1505 INT
1506 WINAPI
1507 InstallWindowsNt(INT argc, WCHAR** argv)
1508 {
1509     INT i;
1510     PWSTR p;
1511 
1512     for (i = 0; i < argc; ++i)
1513     {
1514         p = argv[i];
1515         if (*p == L'-')
1516         {
1517             p++;
1518 
1519             // NOTE: On Windows, "mini" means "minimal UI", and can be used
1520             // in addition to "newsetup"; these options are not exclusive.
1521             if (_wcsicmp(p, L"newsetup") == 0)
1522                 return (INT)InstallReactOS();
1523             else if (_wcsicmp(p, L"mini") == 0)
1524                 return (INT)InstallLiveCD();
1525 
1526             /* Add support for other switches */
1527         }
1528     }
1529 
1530     return 0;
1531 }
1532 
1533 
1534 /*
1535  * @unimplemented
1536  */
1537 DWORD WINAPI
1538 SetupChangeFontSize(
1539     IN HANDLE hWnd,
1540     IN LPCWSTR lpszFontSize)
1541 {
1542     SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
1543     return FALSE;
1544 }
1545 
1546 /*
1547  * @unimplemented
1548  */
1549 DWORD WINAPI
1550 SetupChangeLocaleEx(HWND hWnd,
1551                     LCID Lcid,
1552                     LPCWSTR lpSrcRootPath,
1553                     char Unknown,
1554                     DWORD dwUnused1,
1555                     DWORD dwUnused2)
1556 {
1557     SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
1558     return FALSE;
1559 }
1560 
1561 /*
1562  * @implemented
1563  */
1564 DWORD WINAPI
1565 SetupChangeLocale(HWND hWnd, LCID Lcid)
1566 {
1567     return SetupChangeLocaleEx(hWnd, Lcid, NULL, 0, 0, 0);
1568 }
1569 
1570 
1571 DWORD
1572 WINAPI
1573 SetupStartService(
1574     LPCWSTR lpServiceName,
1575     BOOL bWait)
1576 {
1577     SC_HANDLE hManager = NULL;
1578     SC_HANDLE hService = NULL;
1579     DWORD dwError = ERROR_SUCCESS;
1580 
1581     hManager = OpenSCManagerW(NULL,
1582                               NULL,
1583                               SC_MANAGER_ALL_ACCESS);
1584     if (hManager == NULL)
1585     {
1586         dwError = GetLastError();
1587         goto done;
1588     }
1589 
1590     hService = OpenServiceW(hManager,
1591                             lpServiceName,
1592                             SERVICE_START);
1593     if (hService == NULL)
1594     {
1595         dwError = GetLastError();
1596         goto done;
1597     }
1598 
1599     if (!StartService(hService, 0, NULL))
1600     {
1601         dwError = GetLastError();
1602         goto done;
1603     }
1604 
1605 done:
1606     if (hService != NULL)
1607         CloseServiceHandle(hService);
1608 
1609     if (hManager != NULL)
1610         CloseServiceHandle(hManager);
1611 
1612     return dwError;
1613 }
1614