MeshCentral/MeshAgent
1
0
Fork 0
mirror of https://github.com/Ylianst/MeshAgent synced 2025-12-26 21:23:21 +00:00
Code Issues Releases Wiki Activity

Added tracker

Browse Source
This commit is contained in:
Bryan Roe
2021-01-23 00:44:52 -08:00
parent 9bb81b946a
commit 4be310f809
4 changed files with 59 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
microstack/ILibParsers.c
View File

@@ -997,6 +997,18 @@ typedef struct ILibBaseChain
int lastDescriptorCount;
}ILibBaseChain;
#if defined(ILIBMEMTRACK) && !defined(ILIBCHAIN_GLOBAL_LOCK)
size_t ILib_NativeAllocSize = 0;
ILibSpinLock ILib_MemoryTrackLock = 0;
void* ILibMemory_InitEx(void *ptr, size_t primarySize, size_t extraSize, ILibMemory_Types memType)
{
ILibSpinLock_Lock(&ILib_MemoryTrackLock);
ILib_NativeAllocSize += (primarySize + extraSize);
ILibSpinLock_UnLock(&ILib_MemoryTrackLock);
return(ILibMemory_Init(ptr, primarySize, extraSize, memType));
}
#endif
void* ILibMemory_AllocateA_Init(void *buffer)
{
@@ -1031,6 +1043,19 @@ void* ILibMemory_SmartReAllocate(void *ptr, size_t len)
memmove_s(ILibMemory_Extra(ret) - sizeof(ILibMemory_Header) + offset, originalExtraSize + sizeof(ILibMemory_Header), ILibMemory_Extra(ret) - sizeof(ILibMemory_Header), originalExtraSize + sizeof(ILibMemory_Header));
}
#if defined(ILIBMEMTRACK) && !defined(ILIBCHAIN_GLOBAL_LOCK)
ILibSpinLock_Lock(&ILib_MemoryTrackLock);
if (newRawSize > originalRawSize)
{
ILib_NativeAllocSize += (newRawSize - originalRawSize);
}
else
{
ILib_NativeAllocSize -= (originalRawSize - newRawSize);
}
ILibSpinLock_UnLock(&ILib_MemoryTrackLock);
#endif
ILibMemory_Size(ret) = len;
return(ret);
}
@@ -1076,6 +1101,7 @@ void* ILibMemory_Init(void *ptr, size_t primarySize, size_t extraSize, ILibMemor
return(primary);
}
void ILibMemory_SecureZero(void *ptr, size_t len)
{
#if !defined(MICROSTACK_NOTLS)
@@ -1102,6 +1128,12 @@ void ILibMemory_Free(void *ptr)
{
if (ILibMemory_CanaryOK(ptr) && ILibMemory_MemType(ptr) == ILibMemory_Types_HEAP)
{
#if defined(ILIBMEMTRACK) && !defined(ILIBCHAIN_GLOBAL_LOCK)
ILibSpinLock_Lock(&ILib_MemoryTrackLock);
ILib_NativeAllocSize -= (ILibMemory_Size(ptr) + ILibMemory_ExtraSize(ptr));
ILibSpinLock_UnLock(&ILib_MemoryTrackLock);
#endif
if (ILibMemory_ExtraSize(ptr) > 0)
{
ILibMemory_SecureZero(ILibMemory_RawPtr(ILibMemory_Extra(ptr)), sizeof(ILibMemory_Header));