/*
* PROJECT: ReactOS Setup Library
* LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
- * PURPOSE: NT 5.x family (MS Windows <= 2003, and ReactOS)
- * boot loaders management.
+ * PURPOSE: Boot Stores Management functionality, with support for
+ * NT 5.x family (MS Windows <= 2003, and ReactOS) bootloaders.
* COPYRIGHT: Copyright 2017-2018 Hermes Belusca-Maito
*/
(*POPEN_BOOT_STORE)(
OUT PVOID* Handle,
IN HANDLE PartitionDirectoryHandle, // OPTIONAL
- IN NTOS_BOOT_LOADER_TYPE Type,
+ IN BOOT_STORE_TYPE Type,
IN BOOLEAN CreateNew);
typedef NTSTATUS
typedef struct _NTOS_BOOT_LOADER_FILES
{
- NTOS_BOOT_LOADER_TYPE Type;
+ BOOT_STORE_TYPE Type;
PCZZWSTR LoaderExecutables;
PCWSTR LoaderConfigurationFile;
POPEN_BOOT_STORE OpenBootStore;
*/
typedef struct _BOOT_STORE_CONTEXT
{
- NTOS_BOOT_LOADER_TYPE Type;
+ BOOT_STORE_TYPE Type;
// PNTOS_BOOT_LOADER_FILES ??
/*
PVOID PrivateData;
OpenIniBootLoaderStore(
OUT PVOID* Handle,
IN HANDLE PartitionDirectoryHandle, // OPTIONAL
- IN NTOS_BOOT_LOADER_TYPE Type,
+ IN BOOT_STORE_TYPE Type,
IN BOOLEAN CreateNew);
static NTSTATUS
/* FUNCTIONS ****************************************************************/
NTSTATUS
-FindNTOSBootLoader( // By handle
+FindBootStore( // By handle
IN HANDLE PartitionDirectoryHandle, // OPTIONAL
- IN NTOS_BOOT_LOADER_TYPE Type,
- OUT PULONG Version OPTIONAL)
+ IN BOOT_STORE_TYPE Type,
+ OUT PULONG VersionNumber OPTIONAL)
// OUT PHANDLE ConfigFileHande OPTIONAL ????
{
PCWSTR LoaderExecutable;
if (Type >= BldrTypeMax)
return STATUS_INVALID_PARAMETER;
- if (Version)
- *Version = 0;
+ if (VersionNumber)
+ *VersionNumber = 0;
/* Check whether any of the loader executables exist */
LoaderExecutable = NtosBootLoaders[Type].LoaderExecutables;
}
/* Check for loader version if needed */
- if (Version)
+ if (VersionNumber)
{
- *Version = 0;
- // TODO: Check for BLDR version ONLY if Version != NULL
+ *VersionNumber = 0;
+ // TODO: Check for BLDR version!
}
/* Check whether the loader configuration file exists */
OpenIniBootLoaderStore(
OUT PVOID* Handle,
IN HANDLE PartitionDirectoryHandle, // OPTIONAL
- IN NTOS_BOOT_LOADER_TYPE Type,
+ IN BOOT_STORE_TYPE Type,
IN BOOLEAN CreateNew)
{
NTSTATUS Status;
NTSTATUS
-OpenNTOSBootLoaderStoreByHandle(
+OpenBootStoreByHandle(
OUT PVOID* Handle,
IN HANDLE PartitionDirectoryHandle, // OPTIONAL
- IN NTOS_BOOT_LOADER_TYPE Type,
+ IN BOOT_STORE_TYPE Type,
IN BOOLEAN CreateNew)
{
/*
}
NTSTATUS
-OpenNTOSBootLoaderStore_UStr(
+OpenBootStore_UStr(
OUT PVOID* Handle,
IN PUNICODE_STRING SystemPartitionPath,
- IN NTOS_BOOT_LOADER_TYPE Type,
+ IN BOOT_STORE_TYPE Type,
IN BOOLEAN CreateNew)
{
NTSTATUS Status;
return Status;
}
- Status = OpenNTOSBootLoaderStoreByHandle(Handle, PartitionDirectoryHandle, Type, CreateNew);
+ Status = OpenBootStoreByHandle(Handle, PartitionDirectoryHandle, Type, CreateNew);
/* Done! */
NtClose(PartitionDirectoryHandle);
}
NTSTATUS
-OpenNTOSBootLoaderStore(
+OpenBootStore(
OUT PVOID* Handle,
IN PCWSTR SystemPartition,
- IN NTOS_BOOT_LOADER_TYPE Type,
+ IN BOOT_STORE_TYPE Type,
IN BOOLEAN CreateNew)
{
UNICODE_STRING SystemPartitionPath;
RtlInitUnicodeString(&SystemPartitionPath, SystemPartition);
- return OpenNTOSBootLoaderStore_UStr(Handle, &SystemPartitionPath, Type, CreateNew);
+ return OpenBootStore_UStr(Handle, &SystemPartitionPath, Type, CreateNew);
}
NTSTATUS
-CloseNTOSBootLoaderStore(
+CloseBootStore(
IN PVOID Handle)
{
PBOOT_STORE_CONTEXT BootStore = (PBOOT_STORE_CONTEXT)Handle;
CreateNTOSEntry(
IN PBOOT_STORE_INI_CONTEXT BootStore,
IN ULONG_PTR BootEntryKey,
- IN PNTOS_BOOT_ENTRY BootEntry)
+ IN PBOOT_STORE_ENTRY BootEntry)
{
PINICACHESECTION IniSection;
PWCHAR Section = (PWCHAR)BootEntryKey;
/* Create a new section */
IniSection = IniCacheAppendSection(BootStore->IniCache, Section);
- // if (_wcsicmp(BootEntry->Version, L"Windows2003") == 0)
if (BootEntry->OsOptionsLength >= sizeof(NTOS_OPTIONS) &&
RtlCompareMemory(&BootEntry->OsOptions /* Signature */,
NTOS_OPTIONS_SIGNATURE,
L"Options", (PWSTR)Options->OsLoadOptions);
}
else
- // if (_wcsicmp(BootEntry->Version, L"BootSector") == 0)
if (BootEntry->OsOptionsLength >= sizeof(BOOT_SECTOR_OPTIONS) &&
RtlCompareMemory(&BootEntry->OsOptions /* Signature */,
BOOT_SECTOR_OPTIONS_SIGNATURE,
}
else
{
- DPRINT1("Unsupported BootType '%S'\n", BootEntry->Version);
+ // DPRINT1("Unsupported BootType %lu/'%*.s'\n",
+ // BootEntry->OsOptionsLength, 8, &BootEntry->OsOptions);
+ DPRINT1("Unsupported BootType %lu\n", BootEntry->OsOptionsLength);
}
return STATUS_SUCCESS;
}
NTSTATUS
-AddNTOSBootEntry(
+AddBootStoreEntry(
IN PVOID Handle,
- IN PNTOS_BOOT_ENTRY BootEntry,
+ IN PBOOT_STORE_ENTRY BootEntry,
IN ULONG_PTR BootEntryKey)
{
PBOOT_STORE_CONTEXT BootStore = (PBOOT_STORE_CONTEXT)Handle;
if (BootStore->Type == FreeLdr)
{
+ if (BootEntry->Version != FreeLdr)
+ return STATUS_INVALID_PARAMETER;
+
return CreateNTOSEntry((PBOOT_STORE_INI_CONTEXT)BootStore,
BootEntryKey, BootEntry);
}
ULONG BufferLength;
PCWSTR InstallName, OsOptions;
// ULONG InstallNameLength, OsOptionsLength;
+ BOOLEAN IsNameNotQuoted;
+
+ if (BootEntry->Version != NtLdr)
+ return STATUS_INVALID_PARAMETER;
- // if (_wcsicmp(BootEntry->Version, L"Windows2003") != 0)
if (BootEntry->OsOptionsLength < sizeof(NTOS_OPTIONS) ||
RtlCompareMemory(&BootEntry->OsOptions /* Signature */,
NTOS_OPTIONS_SIGNATURE,
RTL_FIELD_SIZE(NTOS_OPTIONS, Signature)) !=
RTL_FIELD_SIZE(NTOS_OPTIONS, Signature))
{
- DPRINT1("Unsupported BootType '%S'\n", BootEntry->Version);
+ // DPRINT1("Unsupported BootType '%S'\n", BootEntry->Version);
+ DPRINT1("Unsupported BootType %lu\n", BootEntry->OsOptionsLength);
return STATUS_SUCCESS; // STATUS_NOT_SUPPORTED;
}
// if (InstallNameLength == 0) InstallName = NULL;
// if (OsOptionsLength == 0) OsOptions = NULL;
- BufferLength = 2 /* Quotes for FriendlyName*/ + wcslen(InstallName);
+ IsNameNotQuoted = (InstallName[0] != L'\"' || InstallName[wcslen(InstallName)-1] != L'\"');
+
+ BufferLength = (IsNameNotQuoted ? 2 /* Quotes for FriendlyName*/ : 0) + wcslen(InstallName);
if (OsOptions)
BufferLength += 1 /* Space between FriendlyName and options */ + wcslen(OsOptions);
BufferLength++; /* NULL-termination */
if (!Buffer)
return STATUS_INSUFFICIENT_RESOURCES;
- wcscpy(Buffer, L"\"");
- wcscat(Buffer, InstallName);
- wcscat(Buffer, L"\"");
+ *Buffer = UNICODE_NULL;
+ if (IsNameNotQuoted) RtlStringCchCatW(Buffer, BufferLength, L"\"");
+ RtlStringCchCatW(Buffer, BufferLength, InstallName);
+ if (IsNameNotQuoted) RtlStringCchCatW(Buffer, BufferLength, L"\"");
if (OsOptions)
{
- wcscat(Buffer, L" ");
- wcscat(Buffer, OsOptions);
+ RtlStringCchCatW(Buffer, BufferLength, L" ");
+ RtlStringCchCatW(Buffer, BufferLength, OsOptions);
}
/* Insert the entry into the "Operating Systems" section */
}
NTSTATUS
-DeleteNTOSBootEntry(
+DeleteBootStoreEntry(
IN PVOID Handle,
IN ULONG_PTR BootEntryKey)
{
}
NTSTATUS
-ModifyNTOSBootEntry(
+ModifyBootStoreEntry(
IN PVOID Handle,
- IN PNTOS_BOOT_ENTRY BootEntry)
+ IN PBOOT_STORE_ENTRY BootEntry)
{
PBOOT_STORE_CONTEXT BootStore = (PBOOT_STORE_CONTEXT)Handle;
}
NTSTATUS
-QueryNTOSBootEntry(
+QueryBootStoreEntry(
IN PVOID Handle,
IN ULONG_PTR BootEntryKey,
- OUT PNTOS_BOOT_ENTRY BootEntry) // Technically this should be PNTOS_BOOT_ENTRY*
+ OUT PBOOT_STORE_ENTRY BootEntry) // Technically this should be PBOOT_STORE_ENTRY*
{
PBOOT_STORE_CONTEXT BootStore = (PBOOT_STORE_CONTEXT)Handle;
}
NTSTATUS
-QueryNTOSBootOptions(
+QueryBootStoreOptions(
IN PVOID Handle,
- IN OUT PNTOS_BOOT_OPTIONS BootOptions
+ IN OUT PBOOT_STORE_OPTIONS BootOptions
/* , IN PULONG BootOptionsLength */ )
{
NTSTATUS Status = STATUS_SUCCESS;
if (BootStore->Type == FreeLdr)
{
+ BootOptions->Version = FreeLdr;
+
Status = IniCacheGetKey(((PBOOT_STORE_INI_CONTEXT)BootStore)->OptionsIniSection,
L"DefaultOS", (PWCHAR*)&BootOptions->CurrentBootEntryKey);
if (!NT_SUCCESS(Status))
}
else if (BootStore->Type == NtLdr)
{
+ BootOptions->Version = NtLdr;
+
Status = IniCacheGetKey(((PBOOT_STORE_INI_CONTEXT)BootStore)->OptionsIniSection,
L"default", (PWCHAR*)&BootOptions->CurrentBootEntryKey);
if (!NT_SUCCESS(Status))
}
NTSTATUS
-SetNTOSBootOptions(
+SetBootStoreOptions(
IN PVOID Handle,
- IN PNTOS_BOOT_OPTIONS BootOptions,
+ IN PBOOT_STORE_OPTIONS BootOptions,
IN ULONG FieldsToChange)
{
PBOOT_STORE_CONTEXT BootStore = (PBOOT_STORE_CONTEXT)Handle;
return STATUS_NOT_SUPPORTED;
}
+ if (BootOptions->Version != FreeLdr)
+ return STATUS_INVALID_PARAMETER;
+
//
// TODO: Depending on the flags set in 'FieldsToChange',
// change either one or both these bootloader options.
NULL, INSERT_LAST,
L"DefaultOS", (PWCHAR)BootOptions->CurrentBootEntryKey);
- StringCchPrintfW(TimeoutStr, ARRAYSIZE(TimeoutStr), L"%d", BootOptions->Timeout);
+ RtlStringCchPrintfW(TimeoutStr, ARRAYSIZE(TimeoutStr), L"%d", BootOptions->Timeout);
IniCacheInsertKey(((PBOOT_STORE_INI_CONTEXT)BootStore)->OptionsIniSection,
NULL, INSERT_LAST,
L"TimeOut", TimeoutStr);
PINICACHEITERATOR Iterator;
PINICACHESECTION OsIniSection;
PWCHAR SectionName, KeyData;
- UCHAR xxBootEntry[FIELD_OFFSET(NTOS_BOOT_ENTRY, OsOptions) +
+ UCHAR xxBootEntry[FIELD_OFFSET(BOOT_STORE_ENTRY, OsOptions) +
max(sizeof(NTOS_OPTIONS), sizeof(BOOT_SECTOR_OPTIONS))];
- PNTOS_BOOT_ENTRY BootEntry = (PNTOS_BOOT_ENTRY)&xxBootEntry;
+ PBOOT_STORE_ENTRY BootEntry = (PBOOT_STORE_ENTRY)&xxBootEntry;
PWCHAR Buffer;
/* Enumerate all the valid installations listed in the "Operating Systems" section */
DPRINT1("Boot entry '%S' in OS section '%S'\n", InstallName, SectionName);
- BootEntry->Version = NULL;
+ BootEntry->Version = FreeLdr;
BootEntry->BootEntryKey = MAKESTRKEY(SectionName);
BootEntry->FriendlyName = InstallName;
BootEntry->BootFilePath = NULL;
/* BootType is Windows2003 */
PNTOS_OPTIONS Options = (PNTOS_OPTIONS)&BootEntry->OsOptions;
- BootEntry->Version = L"Windows2003";
- DPRINT1("This is a '%S' boot entry\n", BootEntry->Version);
+ DPRINT1("This is a '%S' boot entry\n", KeyData);
BootEntry->OsOptionsLength = sizeof(NTOS_OPTIONS);
RtlCopyMemory(Options->Signature,
/* BootType is BootSector */
PBOOT_SECTOR_OPTIONS Options = (PBOOT_SECTOR_OPTIONS)&BootEntry->OsOptions;
- BootEntry->Version = L"BootSector";
- DPRINT1("This is a '%S' boot entry\n", BootEntry->Version);
+ DPRINT1("This is a '%S' boot entry\n", KeyData);
BootEntry->OsOptionsLength = sizeof(BOOT_SECTOR_OPTIONS);
RtlCopyMemory(Options->Signature,
else
{
DPRINT1("Unrecognized BootType value '%S'\n", KeyData);
- // BootEntry->Version = KeyData;
// goto DoEnum;
}
NTSTATUS Status = STATUS_SUCCESS;
PINICACHEITERATOR Iterator;
PWCHAR SectionName, KeyData;
- UCHAR xxBootEntry[FIELD_OFFSET(NTOS_BOOT_ENTRY, OsOptions) + sizeof(NTOS_OPTIONS)];
- PNTOS_BOOT_ENTRY BootEntry = (PNTOS_BOOT_ENTRY)&xxBootEntry;
+ UCHAR xxBootEntry[FIELD_OFFSET(BOOT_STORE_ENTRY, OsOptions) + sizeof(NTOS_OPTIONS)];
+ PBOOT_STORE_ENTRY BootEntry = (PBOOT_STORE_ENTRY)&xxBootEntry;
PNTOS_OPTIONS Options = (PNTOS_OPTIONS)&BootEntry->OsOptions;
PWCHAR Buffer;
ULONG BufferLength;
DPRINT1("Boot entry '%S' in OS section (path) '%S'\n", InstallName, SectionName);
// SectionName == SystemRoot;
- BootEntry->Version = L"Windows2003";
+ BootEntry->Version = NtLdr;
BootEntry->BootEntryKey = 0; // FIXME??
BootEntry->FriendlyName = InstallName;
BootEntry->BootFilePath = NULL;
}
NTSTATUS
-EnumerateNTOSBootEntries(
+EnumerateBootStoreEntries(
IN PVOID Handle,
// IN ULONG Flags, // Determine which data to retrieve
IN PENUM_BOOT_ENTRIES_ROUTINE EnumBootEntriesRoutine,
projects / reactos.git / blobdiff