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

[NTOSKRNL] Taggify some allocs

Browse source
This commit is contained in:
Pierre Schweitzer 2019-06-10 14:54:55 +02:00
parent a5daa8894d
commit 651ba03c6d
No known key found for this signature in database
GPG key ID: 7545556C3D585B0B
2 changed files with 12 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
ntoskrnl/include/internal/tag.h
View file

@ -178,6 +178,8 @@
#define TAG_LUID 'uLeS' #define TAG_LUID 'uLeS'
#define TAG_PRIVILEGE_SET 'rPeS' #define TAG_PRIVILEGE_SET 'rPeS'
#define TAG_TOKEN_DYNAMIC 'dTeS' #define TAG_TOKEN_DYNAMIC 'dTeS'
#define TAG_SE_HANDLES_TAB 'aHeS'
#define TAG_SE_DIR_BUFFER 'bDeS'
/* LPC Tags */ /* LPC Tags */
#define TAG_LPC_MESSAGE 'McpL' #define TAG_LPC_MESSAGE 'McpL'

16
ntoskrnl/se/srm.c
View file

@ -489,7 +489,9 @@ SepCleanupLUIDDeviceMapDirectory(
* All handles have been already closed * All handles have been already closed
*/ */
AllocateLinksAgain: AllocateLinksAgain:
LinksBuffer = ExAllocatePoolWithTag(PagedPool, LinksSize, 'aHeS'); LinksBuffer = ExAllocatePoolWithTag(PagedPool,
LinksSize,
TAG_SE_HANDLES_TAB);
if (LinksBuffer == NULL) if (LinksBuffer == NULL)
{ {
/* /*
@ -504,7 +506,7 @@ AllocateLinksAgain:
*/ */
if (DirectoryInfo != NULL) if (DirectoryInfo != NULL)
{ {
ExFreePoolWithTag(DirectoryInfo, 0); ExFreePoolWithTag(DirectoryInfo, TAG_SE_DIR_BUFFER);
} }
if (!UseCurrentProc) if (!UseCurrentProc)
@ -551,7 +553,9 @@ AllocateLinksAgain:
} }
/* And reallocate a bigger one */ /* And reallocate a bigger one */
DirectoryInfo = ExAllocatePoolWithTag(PagedPool, DirInfoLength, 'bDeS'); DirectoryInfo = ExAllocatePoolWithTag(PagedPool,
DirInfoLength,
TAG_SE_DIR_BUFFER);
/* Fail if we cannot allocate */ /* Fail if we cannot allocate */
if (DirectoryInfo == NULL) if (DirectoryInfo == NULL)
{ {
@ -584,7 +588,7 @@ AllocateLinksAgain:
/* Allow 20 more HANDLEs */ /* Allow 20 more HANDLEs */
LinksCount += 20; LinksCount += 20;
CurrentLinks = 0; CurrentLinks = 0;
ExFreePoolWithTag(LinksBuffer, 'aHeS'); ExFreePoolWithTag(LinksBuffer, TAG_SE_HANDLES_TAB);
LinksSize = LinksCount * sizeof(HANDLE); LinksSize = LinksCount * sizeof(HANDLE);
/* And reloop again */ /* And reloop again */
@ -627,12 +631,12 @@ AllocateLinksAgain:
ZwClose(LinksBuffer[i]); ZwClose(LinksBuffer[i]);
} }
/* And free our links buffer */ /* And free our links buffer */
ExFreePoolWithTag(LinksBuffer, 'aHeS'); ExFreePoolWithTag(LinksBuffer, TAG_SE_HANDLES_TAB);
/* Free our directory info buffer - it might be NULL if we failed realloc */ /* Free our directory info buffer - it might be NULL if we failed realloc */
if (DirectoryInfo != NULL) if (DirectoryInfo != NULL)
{ {
ExFreePoolWithTag(DirectoryInfo, 'bDeS'); ExFreePoolWithTag(DirectoryInfo, TAG_SE_DIR_BUFFER);
} }
/* Close our session directory */ /* Close our session directory */