Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-02-24 09:25:10 +00:00
Code Issues Projects Releases Packages Wiki Activity

[NTOS:IO]

Browse source 
- Use pool tagging

svn path=/trunk/; revision=74496
This commit is contained in:
Thomas Faber 2017-05-08 06:54:35 +00:00
parent 55bf6ea8ff
commit ba1fcb6554
1 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
reactos/ntoskrnl/io/iomgr/driver.c
View file

@ -898,7 +898,9 @@ IopInitializeBuiltinDriver(IN PLDR_DATA_TABLE_ENTRY BootLdrEntry)
IopDisplayLoadingMessage(ModuleName); IopDisplayLoadingMessage(ModuleName);
InbvIndicateProgress(); InbvIndicateProgress();
Buffer = ExAllocatePool(PagedPool, ModuleName->Length + sizeof(UNICODE_NULL)); Buffer = ExAllocatePoolWithTag(PagedPool,
ModuleName->Length + sizeof(UNICODE_NULL),
TAG_IO);
if (Buffer == NULL) if (Buffer == NULL)
{ {
return STATUS_INSUFFICIENT_RESOURCES; return STATUS_INSUFFICIENT_RESOURCES;
@ -924,7 +926,7 @@ IopInitializeBuiltinDriver(IN PLDR_DATA_TABLE_ENTRY BootLdrEntry)
* Strip the file extension from ServiceName * Strip the file extension from ServiceName
*/ */
Success = RtlCreateUnicodeString(&ServiceName, FileNameWithoutPath); Success = RtlCreateUnicodeString(&ServiceName, FileNameWithoutPath);
ExFreePool(Buffer); ExFreePoolWithTag(Buffer, TAG_IO);
if (!Success) if (!Success)
{ {
return STATUS_INSUFFICIENT_RESOURCES; return STATUS_INSUFFICIENT_RESOURCES;
@ -1284,7 +1286,9 @@ IopUnloadDriver(PUNICODE_STRING DriverServiceName, BOOLEAN UnloadPnpDrivers)
*/ */
ObjectName.Length = ((USHORT)wcslen(Start) + 8) * sizeof(WCHAR); ObjectName.Length = ((USHORT)wcslen(Start) + 8) * sizeof(WCHAR);
ObjectName.MaximumLength = ObjectName.Length + sizeof(WCHAR); ObjectName.MaximumLength = ObjectName.Length + sizeof(WCHAR);
ObjectName.Buffer = ExAllocatePool(PagedPool, ObjectName.MaximumLength); ObjectName.Buffer = ExAllocatePoolWithTag(PagedPool,
ObjectName.MaximumLength,
TAG_IO);
if (!ObjectName.Buffer) return STATUS_INSUFFICIENT_RESOURCES; if (!ObjectName.Buffer) return STATUS_INSUFFICIENT_RESOURCES;
wcscpy(ObjectName.Buffer, DRIVER_ROOT_NAME); wcscpy(ObjectName.Buffer, DRIVER_ROOT_NAME);
memcpy(ObjectName.Buffer + 8, Start, ObjectName.Length - 8 * sizeof(WCHAR)); memcpy(ObjectName.Buffer + 8, Start, ObjectName.Length - 8 * sizeof(WCHAR));
@ -1305,12 +1309,12 @@ IopUnloadDriver(PUNICODE_STRING DriverServiceName, BOOLEAN UnloadPnpDrivers)
if (!NT_SUCCESS(Status)) if (!NT_SUCCESS(Status))
{ {
DPRINT1("Can't locate driver object for %wZ\n", &ObjectName); DPRINT1("Can't locate driver object for %wZ\n", &ObjectName);
ExFreePool(ObjectName.Buffer); ExFreePoolWithTag(ObjectName.Buffer, TAG_IO);
return Status; return Status;
} }
/* Free the buffer for driver object name */ /* Free the buffer for driver object name */
ExFreePool(ObjectName.Buffer); ExFreePoolWithTag(ObjectName.Buffer, TAG_IO);
/* Check that driver is not already unloading */ /* Check that driver is not already unloading */
if (DriverObject->Flags & DRVO_UNLOAD_INVOKED) if (DriverObject->Flags & DRVO_UNLOAD_INVOKED)