Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-05-07 10:46:58 +00:00
Code Issues Projects Releases Packages Wiki Activity

[NTOS:MM] Split MmVerifyImageIsOkForMpUse() into an auxiliary inlined helper.

Browse source 
This helper is used when an existing NtHeader is already available.
This commit is contained in:
Hermès Bélusca-Maïto 2025-03-03 20:29:36 +01:00
parent ccf1e97aa1
commit 7c23a2e38e
No known key found for this signature in database
GPG key ID: 3B2539C65E7B93D0
1 changed files with 26 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

43
ntoskrnl/mm/ARM3/sysldr.c
View file

@ -2716,28 +2716,37 @@ MiEnablePagingOfDriver(IN PLDR_DATA_TABLE_ENTRY LdrEntry)
if (PointerPte) MiSetPagingOfDriver(PointerPte, LastPte);
}
FORCEINLINE
BOOLEAN
MiVerifyImageIsOkForMpUse(
_In_ PIMAGE_NT_HEADERS NtHeaders)
{
/* Fail if we have 2+ CPUs, but the image is only safe for UP */
if ((KeNumberProcessors > 1) &&
(NtHeaders->FileHeader.Characteristics & IMAGE_FILE_UP_SYSTEM_ONLY))
{
return FALSE;
}
/* Otherwise, it's safe to use */
return TRUE;
}
// TODO: Use this function to verify that the loaded boot drivers
// (in ExpLoadBootSymbols) are compatible with MP.
BOOLEAN
NTAPI
MmVerifyImageIsOkForMpUse(IN PVOID BaseAddress)
MmVerifyImageIsOkForMpUse(
_In_ PVOID BaseAddress)
{
PIMAGE_NT_HEADERS NtHeader;
PIMAGE_NT_HEADERS NtHeaders;
PAGED_CODE();
/* Get NT Headers */
NtHeader = RtlImageNtHeader(BaseAddress);
if (NtHeader)
{
/* Check if this image is only safe for UP while we have 2+ CPUs */
if ((KeNumberProcessors > 1) &&
(NtHeader->FileHeader.Characteristics & IMAGE_FILE_UP_SYSTEM_ONLY))
{
/* Fail */
return FALSE;
}
}
/* Otherwise, it's safe */
return TRUE;
/* Get the NT headers. If none, suppose the image
* is safe to use, otherwise invoke the helper. */
NtHeaders = RtlImageNtHeader(BaseAddress);
if (!NtHeaders)
return TRUE;
return MiVerifyImageIsOkForMpUse(NtHeaders);
}
NTSTATUS