base/setup/lib/setuplib.c

   1 /*
   2  * COPYRIGHT:       See COPYING in the top level directory
   3  * PROJECT:         ReactOS Setup Library
   4  * FILE:            base/setup/lib/setuplib.c
   5  * PURPOSE:         Setup Library - Main initialization helpers
   6  * PROGRAMMERS:     Casper S. Hornstrup (chorns@users.sourceforge.net)
   7  *                  Hermes Belusca-Maito (hermes.belusca@sfr.fr)
   8  */
   9
  10 /* INCLUDES *****************************************************************/
  11
  12 #include "precomp.h"
  13 #include "filesup.h"
  14 #include "infsupp.h"
  15 #include "inicache.h"
  16
  17 #include "setuplib.h"
  18
  19 #define NDEBUG
  20 #include <debug.h>
  21
  22
  23 /* GLOBALS ******************************************************************/
  24
  25 /* FUNCTIONS ****************************************************************/
  26
  27 VOID
  28 CheckUnattendedSetup(
  29     IN OUT PUSETUP_DATA pSetupData)
  30 {
  31     INFCONTEXT Context;
  32     HINF UnattendInf;
  33     UINT ErrorLine;
  34     INT IntValue;
  35     PWCHAR Value;
  36     WCHAR UnattendInfPath[MAX_PATH];
  37
  38     CombinePaths(UnattendInfPath, ARRAYSIZE(UnattendInfPath), 2,
  39                  pSetupData->SourcePath.Buffer, L"unattend.inf");
  40
  41     DPRINT("UnattendInf path: '%S'\n", UnattendInfPath);
  42
  43     if (DoesFileExist(NULL, UnattendInfPath) == FALSE)
  44     {
  45         DPRINT("Does not exist: %S\n", UnattendInfPath);
  46         return;
  47     }
  48
  49     /* Load 'unattend.inf' from installation media */
  50     UnattendInf = SetupOpenInfFileExW(UnattendInfPath,
  51                                       NULL,
  52                                       INF_STYLE_OLDNT,
  53                                       pSetupData->LanguageId,
  54                                       &ErrorLine);
  55
  56     if (UnattendInf == INVALID_HANDLE_VALUE)
  57     {
  58         DPRINT("SetupOpenInfFileExW() failed\n");
  59         return;
  60     }
  61
  62     /* Open 'Unattend' section */
  63     if (!SetupFindFirstLineW(UnattendInf, L"Unattend", L"Signature", &Context))
  64     {
  65         DPRINT("SetupFindFirstLineW() failed for section 'Unattend'\n");
  66         goto Quit;
  67     }
  68
  69     /* Get pointer 'Signature' key */
  70     if (!INF_GetData(&Context, NULL, &Value))
  71     {
  72         DPRINT("INF_GetData() failed for key 'Signature'\n");
  73         goto Quit;
  74     }
  75
  76     /* Check 'Signature' string */
  77     if (_wcsicmp(Value, L"$ReactOS$") != 0)
  78     {
  79         DPRINT("Signature not $ReactOS$\n");
  80         INF_FreeData(Value);
  81         goto Quit;
  82     }
  83
  84     INF_FreeData(Value);
  85
  86     /* Check if Unattend setup is enabled */
  87     if (!SetupFindFirstLineW(UnattendInf, L"Unattend", L"UnattendSetupEnabled", &Context))
  88     {
  89         DPRINT("Can't find key 'UnattendSetupEnabled'\n");
  90         goto Quit;
  91     }
  92
  93     if (!INF_GetData(&Context, NULL, &Value))
  94     {
  95         DPRINT("Can't read key 'UnattendSetupEnabled'\n");
  96         goto Quit;
  97     }
  98
  99     if (_wcsicmp(Value, L"yes") != 0)
 100     {
 101         DPRINT("Unattend setup is disabled by 'UnattendSetupEnabled' key!\n");
 102         INF_FreeData(Value);
 103         goto Quit;
 104     }
 105
 106     INF_FreeData(Value);
 107
 108     /* Search for 'DestinationDiskNumber' in the 'Unattend' section */
 109     if (!SetupFindFirstLineW(UnattendInf, L"Unattend", L"DestinationDiskNumber", &Context))
 110     {
 111         DPRINT("SetupFindFirstLine() failed for key 'DestinationDiskNumber'\n");
 112         goto Quit;
 113     }
 114
 115     if (!SetupGetIntField(&Context, 1, &IntValue))
 116     {
 117         DPRINT("SetupGetIntField() failed for key 'DestinationDiskNumber'\n");
 118         goto Quit;
 119     }
 120
 121     pSetupData->DestinationDiskNumber = (LONG)IntValue;
 122
 123     /* Search for 'DestinationPartitionNumber' in the 'Unattend' section */
 124     if (!SetupFindFirstLineW(UnattendInf, L"Unattend", L"DestinationPartitionNumber", &Context))
 125     {
 126         DPRINT("SetupFindFirstLine() failed for key 'DestinationPartitionNumber'\n");
 127         goto Quit;
 128     }
 129
 130     if (!SetupGetIntField(&Context, 1, &IntValue))
 131     {
 132         DPRINT("SetupGetIntField() failed for key 'DestinationPartitionNumber'\n");
 133         goto Quit;
 134     }
 135
 136     pSetupData->DestinationPartitionNumber = (LONG)IntValue;
 137
 138     /* Search for 'InstallationDirectory' in the 'Unattend' section (optional) */
 139     if (SetupFindFirstLineW(UnattendInf, L"Unattend", L"InstallationDirectory", &Context))
 140     {
 141         /* Get pointer 'InstallationDirectory' key */
 142         if (!INF_GetData(&Context, NULL, &Value))
 143         {
 144             DPRINT("INF_GetData() failed for key 'InstallationDirectory'\n");
 145             goto Quit;
 146         }
 147
 148         RtlStringCchCopyW(pSetupData->InstallationDirectory,
 149                           ARRAYSIZE(pSetupData->InstallationDirectory),
 150                           Value);
 151
 152         INF_FreeData(Value);
 153     }
 154
 155     IsUnattendedSetup = TRUE;
 156     DPRINT("Running unattended setup\n");
 157
 158     /* Search for 'MBRInstallType' in the 'Unattend' section */
 159     pSetupData->MBRInstallType = -1;
 160     if (SetupFindFirstLineW(UnattendInf, L"Unattend", L"MBRInstallType", &Context))
 161     {
 162         if (SetupGetIntField(&Context, 1, &IntValue))
 163         {
 164             pSetupData->MBRInstallType = IntValue;
 165         }
 166     }
 167
 168     /* Search for 'FormatPartition' in the 'Unattend' section */
 169     pSetupData->FormatPartition = 0;
 170     if (SetupFindFirstLineW(UnattendInf, L"Unattend", L"FormatPartition", &Context))
 171     {
 172         if (SetupGetIntField(&Context, 1, &IntValue))
 173         {
 174             pSetupData->FormatPartition = IntValue;
 175         }
 176     }
 177
 178     pSetupData->AutoPartition = 0;
 179     if (SetupFindFirstLineW(UnattendInf, L"Unattend", L"AutoPartition", &Context))
 180     {
 181         if (SetupGetIntField(&Context, 1, &IntValue))
 182         {
 183             pSetupData->AutoPartition = IntValue;
 184         }
 185     }
 186
 187     /* Search for LocaleID in the 'Unattend' section */
 188     if (SetupFindFirstLineW(UnattendInf, L"Unattend", L"LocaleID", &Context))
 189     {
 190         if (INF_GetData(&Context, NULL, &Value))
 191         {
 192             LONG Id = wcstol(Value, NULL, 16);
 193             RtlStringCchPrintfW(pSetupData->LocaleID,
 194                                 ARRAYSIZE(pSetupData->LocaleID),
 195                                 L"%08lx", Id);
 196             INF_FreeData(Value);
 197        }
 198     }
 199
 200 Quit:
 201     SetupCloseInfFile(UnattendInf);
 202 }
 203
 204 VOID
 205 InstallSetupInfFile(
 206     IN OUT PUSETUP_DATA pSetupData)
 207 {
 208     NTSTATUS Status;
 209     PINICACHE IniCache;
 210
 211 #if 0 // HACK FIXME!
 212     PINICACHE UnattendCache;
 213     PINICACHEITERATOR Iterator;
 214 #else
 215     // PCWSTR CrLf = L"\r\n";
 216     PCSTR CrLf = "\r\n";
 217     HANDLE FileHandle, UnattendFileHandle, SectionHandle;
 218     FILE_STANDARD_INFORMATION FileInfo;
 219     ULONG FileSize;
 220     PVOID ViewBase;
 221     UNICODE_STRING FileName;
 222     OBJECT_ATTRIBUTES ObjectAttributes;
 223     IO_STATUS_BLOCK IoStatusBlock;
 224 #endif
 225
 226     PINICACHESECTION IniSection;
 227     WCHAR PathBuffer[MAX_PATH];
 228     WCHAR UnattendInfPath[MAX_PATH];
 229
 230     /* Create a $winnt$.inf file with default entries */
 231     IniCache = IniCacheCreate();
 232     if (!IniCache)
 233         return;
 234
 235     IniSection = IniCacheAppendSection(IniCache, L"SetupParams");
 236     if (IniSection)
 237     {
 238         /* Key "skipmissingfiles" */
 239         // RtlStringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
 240                             // L"\"%s\"", L"WinNt5.2");
 241         // IniCacheInsertKey(IniSection, NULL, INSERT_LAST,
 242                           // L"Version", PathBuffer);
 243     }
 244
 245     IniSection = IniCacheAppendSection(IniCache, L"Data");
 246     if (IniSection)
 247     {
 248         RtlStringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
 249                             L"\"%s\"", IsUnattendedSetup ? L"yes" : L"no");
 250         IniCacheInsertKey(IniSection, NULL, INSERT_LAST,
 251                           L"UnattendedInstall", PathBuffer);
 252
 253         // "floppylessbootpath" (yes/no)
 254
 255         RtlStringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
 256                             L"\"%s\"", L"winnt");
 257         IniCacheInsertKey(IniSection, NULL, INSERT_LAST,
 258                           L"ProductType", PathBuffer);
 259
 260         RtlStringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
 261                             L"\"%s\\\"", pSetupData->SourceRootPath.Buffer);
 262         IniCacheInsertKey(IniSection, NULL, INSERT_LAST,
 263                           L"SourcePath", PathBuffer);
 264
 265         // "floppyless" ("0")
 266     }
 267
 268 #if 0
 269
 270     /* TODO: Append the standard unattend.inf file */
 271     CombinePaths(UnattendInfPath, ARRAYSIZE(UnattendInfPath), 2,
 272                  pSetupData->SourcePath.Buffer, L"unattend.inf");
 273     if (DoesFileExist(NULL, UnattendInfPath) == FALSE)
 274     {
 275         DPRINT("Does not exist: %S\n", UnattendInfPath);
 276         goto Quit;
 277     }
 278
 279     Status = IniCacheLoad(&UnattendCache, UnattendInfPath, FALSE);
 280     if (!NT_SUCCESS(Status))
 281     {
 282         DPRINT1("Cannot load %S as an INI file!\n", UnattendInfPath);
 283         goto Quit;
 284     }
 285
 286     IniCacheDestroy(UnattendCache);
 287
 288 Quit:
 289     CombinePaths(PathBuffer, ARRAYSIZE(PathBuffer), 2,
 290                  pSetupData->DestinationPath.Buffer, L"System32\\$winnt$.inf");
 291     IniCacheSave(IniCache, PathBuffer);
 292     IniCacheDestroy(IniCache);
 293
 294 #else
 295
 296     CombinePaths(PathBuffer, ARRAYSIZE(PathBuffer), 2,
 297                  pSetupData->DestinationPath.Buffer, L"System32\\$winnt$.inf");
 298     IniCacheSave(IniCache, PathBuffer);
 299     IniCacheDestroy(IniCache);
 300
 301     /* TODO: Append the standard unattend.inf file */
 302     CombinePaths(UnattendInfPath, ARRAYSIZE(UnattendInfPath), 2,
 303                  pSetupData->SourcePath.Buffer, L"unattend.inf");
 304     if (DoesFileExist(NULL, UnattendInfPath) == FALSE)
 305     {
 306         DPRINT("Does not exist: %S\n", UnattendInfPath);
 307         return;
 308     }
 309
 310     RtlInitUnicodeString(&FileName, PathBuffer);
 311     InitializeObjectAttributes(&ObjectAttributes,
 312                                &FileName,
 313                                OBJ_CASE_INSENSITIVE | OBJ_OPENIF,
 314                                NULL,
 315                                NULL);
 316     Status = NtOpenFile(&FileHandle,
 317                         FILE_APPEND_DATA | SYNCHRONIZE,
 318                         &ObjectAttributes,
 319                         &IoStatusBlock,
 320                         FILE_SHARE_READ,
 321                         FILE_SYNCHRONOUS_IO_NONALERT | FILE_NON_DIRECTORY_FILE);
 322     if (!NT_SUCCESS(Status))
 323     {
 324         DPRINT1("Cannot load %S as an INI file!\n", PathBuffer);
 325         return;
 326     }
 327
 328     /* Query the file size */
 329     Status = NtQueryInformationFile(FileHandle,
 330                                     &IoStatusBlock,
 331                                     &FileInfo,
 332                                     sizeof(FileInfo),
 333                                     FileStandardInformation);
 334     if (!NT_SUCCESS(Status))
 335     {
 336         DPRINT("NtQueryInformationFile() failed (Status %lx)\n", Status);
 337         FileInfo.EndOfFile.QuadPart = 0ULL;
 338     }
 339
 340     Status = OpenAndMapFile(NULL,
 341                             UnattendInfPath,
 342                             &UnattendFileHandle,
 343                             &SectionHandle,
 344                             &ViewBase,
 345                             &FileSize,
 346                             FALSE);
 347     if (!NT_SUCCESS(Status))
 348     {
 349         DPRINT1("Cannot load %S !\n", UnattendInfPath);
 350         NtClose(FileHandle);
 351         return;
 352     }
 353
 354     /* Write to the INI file */
 355
 356     /* "\r\n" */
 357     Status = NtWriteFile(FileHandle,
 358                          NULL,
 359                          NULL,
 360                          NULL,
 361                          &IoStatusBlock,
 362                          (PVOID)CrLf,
 363                          2 * sizeof(CHAR), // 2 * sizeof(WCHAR),
 364                          &FileInfo.EndOfFile,
 365                          NULL);
 366
 367     Status = NtWriteFile(FileHandle,
 368                          NULL,
 369                          NULL,
 370                          NULL,
 371                          &IoStatusBlock,
 372                          ViewBase,
 373                          FileSize,
 374                          NULL,
 375                          NULL);
 376     if (!NT_SUCCESS(Status))
 377     {
 378         DPRINT("NtWriteFile() failed (Status %lx)\n", Status);
 379     }
 380
 381     /* Finally, unmap and close the file */
 382     UnMapAndCloseFile(UnattendFileHandle, SectionHandle, ViewBase);
 383
 384     NtClose(FileHandle);
 385 #endif
 386 }
 387
 388 NTSTATUS
 389 GetSourcePaths(
 390     OUT PUNICODE_STRING SourcePath,
 391     OUT PUNICODE_STRING SourceRootPath,
 392     OUT PUNICODE_STRING SourceRootDir)
 393 {
 394     NTSTATUS Status;
 395     HANDLE Handle;
 396     OBJECT_ATTRIBUTES ObjectAttributes;
 397     UCHAR ImageFileBuffer[sizeof(UNICODE_STRING) + MAX_PATH * sizeof(WCHAR)];
 398     PUNICODE_STRING InstallSourcePath = (PUNICODE_STRING)&ImageFileBuffer;
 399     WCHAR SystemRootBuffer[MAX_PATH] = L"";
 400     UNICODE_STRING SystemRootPath = RTL_CONSTANT_STRING(L"\\SystemRoot");
 401     ULONG BufferSize;
 402     PWCHAR Ptr;
 403
 404     /* Determine the installation source path via the full path of the installer */
 405     RtlInitEmptyUnicodeString(InstallSourcePath,
 406                               (PWSTR)((ULONG_PTR)ImageFileBuffer + sizeof(UNICODE_STRING)),
 407                               sizeof(ImageFileBuffer) - sizeof(UNICODE_STRING)
 408             /* Reserve space for a NULL terminator */ - sizeof(UNICODE_NULL));
 409     BufferSize = sizeof(ImageFileBuffer);
 410     Status = NtQueryInformationProcess(NtCurrentProcess(),
 411                                        ProcessImageFileName,
 412                                        InstallSourcePath,
 413                                        BufferSize,
 414                                        NULL);
 415     // STATUS_INFO_LENGTH_MISMATCH or STATUS_BUFFER_TOO_SMALL ?
 416     if (!NT_SUCCESS(Status))
 417         return Status;
 418
 419     /* Manually NULL-terminate */
 420     InstallSourcePath->Buffer[InstallSourcePath->Length / sizeof(WCHAR)] = UNICODE_NULL;
 421
 422     /* Strip the trailing file name */
 423     Ptr = wcsrchr(InstallSourcePath->Buffer, OBJ_NAME_PATH_SEPARATOR);
 424     if (Ptr)
 425         *Ptr = UNICODE_NULL;
 426     InstallSourcePath->Length = wcslen(InstallSourcePath->Buffer) * sizeof(WCHAR);
 427
 428
 429     /*
 430      * Now resolve the full path to \SystemRoot. In case it prefixes
 431      * the installation source path determined from the full path of
 432      * the installer, we use instead the resolved \SystemRoot as the
 433      * installation source path.
 434      * Otherwise, we use instead the path from the full installer path.
 435      */
 436
 437     InitializeObjectAttributes(&ObjectAttributes,
 438                                &SystemRootPath,
 439                                OBJ_CASE_INSENSITIVE,
 440                                NULL,
 441                                NULL);
 442
 443     Status = NtOpenSymbolicLinkObject(&Handle,
 444                                       SYMBOLIC_LINK_QUERY,
 445                                       &ObjectAttributes);
 446     if (!NT_SUCCESS(Status))
 447     {
 448         /*
 449          * We failed at opening the \SystemRoot link (usually due to wrong
 450          * access rights). Do not consider this as a fatal error, but use
 451          * instead the image file path as the installation source path.
 452          */
 453         DPRINT1("NtOpenSymbolicLinkObject(%wZ) failed with Status 0x%08lx\n",
 454                 &SystemRootPath, Status);
 455         goto InitPaths;
 456     }
 457
 458     RtlInitEmptyUnicodeString(&SystemRootPath,
 459                               SystemRootBuffer,
 460                               sizeof(SystemRootBuffer));
 461
 462     Status = NtQuerySymbolicLinkObject(Handle,
 463                                        &SystemRootPath,
 464                                        &BufferSize);
 465     NtClose(Handle);
 466
 467     if (!NT_SUCCESS(Status))
 468         return Status; // Unexpected error
 469
 470     /* Check whether the resolved \SystemRoot is a prefix of the image file path */
 471     if (RtlPrefixUnicodeString(&SystemRootPath, InstallSourcePath, TRUE))
 472     {
 473         /* Yes it is, so we use instead SystemRoot as the installation source path */
 474         InstallSourcePath = &SystemRootPath;
 475     }
 476
 477
 478 InitPaths:
 479     /*
 480      * Retrieve the different source path components
 481      */
 482     RtlCreateUnicodeString(SourcePath, InstallSourcePath->Buffer);
 483
 484     /* Strip trailing directory */
 485     Ptr = wcsrchr(InstallSourcePath->Buffer, OBJ_NAME_PATH_SEPARATOR);
 486     if (Ptr)
 487     {
 488         RtlCreateUnicodeString(SourceRootDir, Ptr);
 489         *Ptr = UNICODE_NULL;
 490     }
 491     else
 492     {
 493         RtlCreateUnicodeString(SourceRootDir, L"");
 494     }
 495
 496     RtlCreateUnicodeString(SourceRootPath, InstallSourcePath->Buffer);
 497
 498     return STATUS_SUCCESS;
 499 }
 500
 501 ERROR_NUMBER
 502 LoadSetupInf(
 503     OUT HINF* SetupInf,
 504     IN OUT PUSETUP_DATA pSetupData)
 505 {
 506     INFCONTEXT Context;
 507     UINT ErrorLine;
 508     INT IntValue;
 509     PWCHAR Value;
 510     WCHAR FileNameBuffer[MAX_PATH];
 511
 512     CombinePaths(FileNameBuffer, ARRAYSIZE(FileNameBuffer), 2,
 513                  pSetupData->SourcePath.Buffer, L"txtsetup.sif");
 514
 515     DPRINT("SetupInf path: '%S'\n", FileNameBuffer);
 516
 517     *SetupInf = SetupOpenInfFileExW(FileNameBuffer,
 518                                    NULL,
 519                                    INF_STYLE_WIN4 | INF_STYLE_OLDNT,
 520                                    pSetupData->LanguageId,
 521                                    &ErrorLine);
 522
 523     if (*SetupInf == INVALID_HANDLE_VALUE)
 524         return ERROR_LOAD_TXTSETUPSIF;
 525
 526     /* Open 'Version' section */
 527     if (!SetupFindFirstLineW(*SetupInf, L"Version", L"Signature", &Context))
 528         return ERROR_CORRUPT_TXTSETUPSIF;
 529
 530     /* Get pointer 'Signature' key */
 531     if (!INF_GetData(&Context, NULL, &Value))
 532         return ERROR_CORRUPT_TXTSETUPSIF;
 533
 534     /* Check 'Signature' string */
 535     if (_wcsicmp(Value, L"$ReactOS$") != 0)
 536     {
 537         INF_FreeData(Value);
 538         return ERROR_SIGNATURE_TXTSETUPSIF;
 539     }
 540
 541     INF_FreeData(Value);
 542
 543     /* Open 'DiskSpaceRequirements' section */
 544     if (!SetupFindFirstLineW(*SetupInf, L"DiskSpaceRequirements", L"FreeSysPartDiskSpace", &Context))
 545         return ERROR_CORRUPT_TXTSETUPSIF;
 546
 547     pSetupData->RequiredPartitionDiskSpace = ~0;
 548
 549     /* Get the 'FreeSysPartDiskSpace' value */
 550     if (!SetupGetIntField(&Context, 1, &IntValue))
 551         return ERROR_CORRUPT_TXTSETUPSIF;
 552
 553     pSetupData->RequiredPartitionDiskSpace = (ULONG)IntValue;
 554
 555     //
 556     // TODO: Support "SetupSourceDevice" and "SetupSourcePath" in txtsetup.sif
 557     // See CORE-9023
 558     //
 559
 560     /* Search for 'DefaultPath' in the 'SetupData' section */
 561     if (SetupFindFirstLineW(*SetupInf, L"SetupData", L"DefaultPath", &Context))
 562     {
 563         /* Get pointer 'DefaultPath' key */
 564         if (!INF_GetData(&Context, NULL, &Value))
 565             return ERROR_CORRUPT_TXTSETUPSIF;
 566
 567         RtlStringCchCopyW(pSetupData->InstallationDirectory,
 568                           ARRAYSIZE(pSetupData->InstallationDirectory),
 569                           Value);
 570
 571         INF_FreeData(Value);
 572     }
 573
 574     return ERROR_SUCCESS;
 575 }
 576
 577 /* EOF */