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