Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-06-19 19:28:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NTOS:CM] Implement support for KeyValuePartialInformationAlign64. CORE-17358

Browse source
This commit is contained in:
Thomas Faber 2020-10-30 17:29:29 +01:00
parent b2cf64094c
commit 507c442473
No known key found for this signature in database
GPG key ID: 076E7C3D44720826
1 changed files with 35 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
ntoskrnl/config/cmvalche.c
View file

@ -574,16 +574,23 @@ CmpQueryKeyValueData(IN PCM_KEY_CONTROL_BLOCK Kcb,
/* Partial information requested (no name or alignment!) */
case KeyValuePartialInformation:
case KeyValuePartialInformationAlign64:
/* Check if this is a small key and compute key size */
IsSmall = CmpIsKeyValueSmall(&KeySize,
CellData->u.KeyValue.DataLength);
/* Calculate the total size required */
Size = FIELD_OFFSET(KEY_VALUE_PARTIAL_INFORMATION, Data) + KeySize;
/* And this is the least we can work with */
MinimumSize = FIELD_OFFSET(KEY_VALUE_PARTIAL_INFORMATION, Data);
/* Calculate the total size required and the least we can work with */
if (KeyValueInformationClass == KeyValuePartialInformationAlign64)
{
Size = FIELD_OFFSET(KEY_VALUE_PARTIAL_INFORMATION_ALIGN64, Data) + KeySize;
MinimumSize = FIELD_OFFSET(KEY_VALUE_PARTIAL_INFORMATION_ALIGN64, Data);
}
else
{
Size = FIELD_OFFSET(KEY_VALUE_PARTIAL_INFORMATION, Data) + KeySize;
MinimumSize = FIELD_OFFSET(KEY_VALUE_PARTIAL_INFORMATION, Data);
}
/* Tell the caller the size we'll finally need, and set success */
*ResultLength = Size;
@ -598,9 +605,17 @@ CmpQueryKeyValueData(IN PCM_KEY_CONTROL_BLOCK Kcb,
}
/* Fill out the basic information */
Info->KeyValuePartialInformation.TitleIndex = 0;
Info->KeyValuePartialInformation.Type = CellData->u.KeyValue.Type;
Info->KeyValuePartialInformation.DataLength = KeySize;
if (KeyValueInformationClass == KeyValuePartialInformationAlign64)
{
Info->KeyValuePartialInformationAlign64.Type = CellData->u.KeyValue.Type;
Info->KeyValuePartialInformationAlign64.DataLength = KeySize;
}
else
{
Info->KeyValuePartialInformation.TitleIndex = 0;
Info->KeyValuePartialInformation.Type = CellData->u.KeyValue.Type;
Info->KeyValuePartialInformation.DataLength = KeySize;
}
/* Now check if the key had any data */
if (KeySize > 0)
@ -649,9 +664,18 @@ CmpQueryKeyValueData(IN PCM_KEY_CONTROL_BLOCK Kcb,
if (Buffer)
{
/* Copy the data into the aligned offset */
RtlCopyMemory(Info->KeyValuePartialInformation.Data,
Buffer,
Size);
if (KeyValueInformationClass == KeyValuePartialInformationAlign64)
{
RtlCopyMemory(Info->KeyValuePartialInformationAlign64.Data,
Buffer,
Size);
}
else
{
RtlCopyMemory(Info->KeyValuePartialInformation.Data,
Buffer,
Size);
}
}
}