[SDK:CMLIB] Use Hive->Allocate / Hive->Free where possible.

This commit is contained in:
Hermès Bélusca-Maïto 2022-03-27 18:44:53 +02:00
parent b79fbe2333
commit 6a03fb7099
No known key found for this signature in database
GPG key ID: 3B2539C65E7B93D0
2 changed files with 6 additions and 10 deletions

View file

@ -793,9 +793,7 @@ CmpMarkIndexDirty(IN PHHIVE Hive,
SearchName.Length = CmpCompressedNameSize(Node->Name,
Node->NameLength);
SearchName.MaximumLength = SearchName.Length;
SearchName.Buffer = CmpAllocate(SearchName.Length,
TRUE,
TAG_CM);
SearchName.Buffer = Hive->Allocate(SearchName.Length, TRUE, TAG_CM);
if (!SearchName.Buffer)
{
/* Fail */
@ -889,7 +887,7 @@ CmpMarkIndexDirty(IN PHHIVE Hive,
if (Child != HCELL_NIL)
{
/* We found it, free the name now */
if (IsCompressed) CmpFree(SearchName.Buffer, 0);
if (IsCompressed) Hive->Free(SearchName.Buffer, 0);
/* Release the parent key */
HvReleaseCell(Hive, ParentKey);
@ -914,7 +912,7 @@ Quickie:
if (CellToRelease != HCELL_NIL) HvReleaseCell(Hive, CellToRelease);
/* Free the search name and return failure */
if (IsCompressed) CmpFree(SearchName.Buffer, 0);
if (IsCompressed) Hive->Free(SearchName.Buffer, 0);
return FALSE;
}
@ -1731,9 +1729,7 @@ CmpRemoveSubKey(IN PHHIVE Hive,
if (SearchName.MaximumLength > sizeof(Buffer))
{
/* Allocate one */
SearchName.Buffer = CmpAllocate(SearchName.Length,
TRUE,
TAG_CM);
SearchName.Buffer = Hive->Allocate(SearchName.Length, TRUE, TAG_CM);
if (!SearchName.Buffer) return FALSE;
}
else
@ -1872,7 +1868,7 @@ Exit:
if ((IsCompressed) && (SearchName.MaximumLength > sizeof(Buffer)))
{
/* Free the buffer we allocated */
CmpFree(SearchName.Buffer, 0);
Hive->Free(SearchName.Buffer, 0);
}
/* Return the result */

View file

@ -194,7 +194,7 @@ CmpValueToData(IN PHHIVE Hive,
if (BufferAllocated)
{
/* Free the buffer and bugcheck */
CmpFree(Buffer, 0);
Hive->Free(Buffer, 0);
KeBugCheckEx(REGISTRY_ERROR, 8, 0, (ULONG_PTR)Hive, (ULONG_PTR)Value);
}