Check return value of NtQuerySystemInformation in EngQuerySystemAttribute
CID 513027

svn path=/trunk/; revision=63088
This commit is contained in:
Timo Kreuzer 2014-05-01 08:55:04 +00:00
parent fc4372b2b8
commit 305c6b4fcc

View file

@ -245,22 +245,34 @@ EngQuerySystemAttribute(
{
SYSTEM_BASIC_INFORMATION sbi;
SYSTEM_PROCESSOR_INFORMATION spi;
NTSTATUS status;
switch (CapNum)
{
case EngNumberOfProcessors:
NtQuerySystemInformation(SystemBasicInformation,
&sbi,
sizeof(SYSTEM_BASIC_INFORMATION),
NULL);
status = NtQuerySystemInformation(SystemBasicInformation,
&sbi,
sizeof(SYSTEM_BASIC_INFORMATION),
NULL);
if (!NT_SUCCESS(status))
{
DPRINT1("Failed to query basic information: 0x%ls\n", status);
return FALSE;
}
*pCapability = sbi.NumberOfProcessors;
return TRUE;
case EngProcessorFeature:
NtQuerySystemInformation(SystemProcessorInformation,
&spi,
sizeof(SYSTEM_PROCESSOR_INFORMATION),
NULL);
status = NtQuerySystemInformation(SystemProcessorInformation,
&spi,
sizeof(SYSTEM_PROCESSOR_INFORMATION),
NULL);
if (!NT_SUCCESS(status))
{
DPRINT1("Failed to query processor information: 0x%ls\n", status);
return FALSE;
}
*pCapability = spi.ProcessorFeatureBits;
return TRUE;