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

[NTOSKRNL]

Browse source 
Implement querying class data in CmpQueryKeyData

svn path=/trunk/; revision=61658
This commit is contained in:
Timo Kreuzer 2014-01-17 21:45:53 +00:00
parent 1f09821059
commit f9a9a6e8de
1 changed files with 59 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

65
reactos/ntoskrnl/config/cmapi.c
View file

@ -380,9 +380,10 @@ CmpQueryKeyData(IN PHHIVE Hive,
IN OUT PULONG ResultLength) IN OUT PULONG ResultLength)
{ {
NTSTATUS Status; NTSTATUS Status;
ULONG Size, SizeLeft, MinimumSize; ULONG Size, SizeLeft, MinimumSize, Offset;
PKEY_INFORMATION Info = (PKEY_INFORMATION)KeyInformation; PKEY_INFORMATION Info = (PKEY_INFORMATION)KeyInformation;
USHORT NameLength; USHORT NameLength;
PVOID ClassData;
/* Check if the value is compressed */ /* Check if the value is compressed */
if (Node->Flags & KEY_COMP_NAME) if (Node->Flags & KEY_COMP_NAME)
@ -516,8 +517,36 @@ CmpQueryKeyData(IN PHHIVE Hive,
/* Check if the node has a class */ /* Check if the node has a class */
if (Node->ClassLength > 0) if (Node->ClassLength > 0)
{ {
/* It does. We don't support these yet */ /* Set the class offset */
ASSERTMSG("Classes not supported\n", FALSE); Offset = FIELD_OFFSET(KEY_NODE_INFORMATION, Name) + NameLength;
Offset = ALIGN_UP_BY(Offset, sizeof(ULONG));
Info->KeyNodeInformation.ClassOffset = Offset;
/* Get the class data */
ClassData = HvGetCell(Hive, Node->Class);
if (ClassData == NULL)
{
Status = STATUS_INSUFFICIENT_RESOURCES;
break;
}
/* Check if we can copy anything */
if (Length > Offset)
{
/* Copy the class data */
RtlCopyMemory((PUCHAR)Info + Offset,
ClassData,
min(Node->ClassLength, Length - Offset));
}
/* Check if the buffer was large enough */
if (Length < Offset + Node->ClassLength)
{
Status = STATUS_BUFFER_OVERFLOW;
}
/* Release the class cell */
HvReleaseCell(Hive, Node->Class);
} }
else else
{ {
@ -563,13 +592,37 @@ CmpQueryKeyData(IN PHHIVE Hive,
/* Check if we have a class */ /* Check if we have a class */
if (Node->ClassLength > 0) if (Node->ClassLength > 0)
{ {
/* We do, but we currently don't support this */ /* Set the class offset */
ASSERTMSG("Classes not supported\n", FALSE); Offset = FIELD_OFFSET(KEY_FULL_INFORMATION, Class);
Info->KeyFullInformation.ClassOffset = Offset;
/* Get the class data */
ClassData = HvGetCell(Hive, Node->Class);
if (ClassData == NULL)
{
Status = STATUS_INSUFFICIENT_RESOURCES;
break;
}
/* Copy the class data */
NT_ASSERT(Length > Offset);
RtlCopyMemory(Info->KeyFullInformation.Class,
ClassData,
min(Node->ClassLength, Length - Offset));
/* Check if the buffer was large enough */
if (Length < Offset + Node->ClassLength)
{
Status = STATUS_BUFFER_OVERFLOW;
}
/* Release the class cell */
HvReleaseCell(Hive, Node->Class);
} }
else else
{ {
/* We don't have a class, so set offset to -1, not 0! */ /* We don't have a class, so set offset to -1, not 0! */
Info->KeyNodeInformation.ClassOffset = 0xFFFFFFFF; Info->KeyFullInformation.ClassOffset = 0xFFFFFFFF;
} }
break; break;