AMajor/SoftEtherVPN
1
0
Fork 0
mirror of https://github.com/SoftEtherVPN/SoftEtherVPN.git synced 2024-11-22 17:39:53 +03:00
Code Releases Activity

Mayaqua: ability to toggle memory tracking at runtime, various logic improvements

Browse Source
This commit is contained in:
Davide Beatrici 2018-11-18 02:57:14 +01:00
parent 4e7d946214
commit c48de5924a
6 changed files with 53 additions and 152 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/Mayaqua/Mayaqua.c
View File

@ -552,8 +552,11 @@ void InitMayaqua(bool memcheck, bool debug, int argc, char **argv)
// Initialize the Kernel status // Initialize the Kernel status
InitKernelStatus(); InitKernelStatus();
if (IsTrackingEnabled())
{
// Initialize the tracking // Initialize the tracking
InitTracking(); InitTracking();
}
// Initialization of thread pool // Initialization of thread pool
InitThreading(); InitThreading();
@ -696,7 +699,8 @@ void FreeMayaqua()
// Release of thread pool // Release of thread pool
FreeThreading(); FreeThreading();
#ifndef VPN_SPEED if (IsTrackingEnabled())
{
// Show the kernel status // Show the kernel status
if (g_debug) if (g_debug)
{ {
@ -708,10 +712,10 @@ void FreeMayaqua()
{ {
PrintDebugInformation(); PrintDebugInformation();
} }
#endif // VPN_SPEED
// Release the tracking // Release the tracking
FreeTracking(); FreeTracking();
}
// Release of the kernel status // Release of the kernel status
FreeKernelStatus(); FreeKernelStatus();

20
src/Mayaqua/Mayaqua.h
View File

@ -125,10 +125,8 @@
// Macro for the release flag // Macro for the release flag
#ifdef VPN_SPEED #ifdef VPN_SPEED
#define DONT_USE_KERNEL_STATUS // Do not update the kernel status
#define WIN32_USE_HEAP_API_FOR_MEMORY // Use the heap API to allocate memory #define WIN32_USE_HEAP_API_FOR_MEMORY // Use the heap API to allocate memory
#define WIN32_NO_DEBUG_HELP_DLL // Do not call the DLL for debugging #define WIN32_NO_DEBUG_HELP_DLL // Do not call the DLL for debugging
#define DONT_CHECK_HEAP // Do not check the status of the heap
#define DONT_ALLOW_RUN_ON_DEBUGGER // Do not allow running on the debugger #define DONT_ALLOW_RUN_ON_DEBUGGER // Do not allow running on the debugger
#endif // VPN_SPEED #endif // VPN_SPEED
@ -412,43 +410,35 @@ extern BOOL kernel_status_inited;
#define KS_GETMAX64(id) (kernel_status_max[id]) #define KS_GETMAX64(id) (kernel_status_max[id])
#define KS_GETMAX(id) ((UINT)KS_GETMAX64(id)) #define KS_GETMAX(id) ((UINT)KS_GETMAX64(id))
#ifdef DONT_USE_KERNEL_STATUS // Operations of the kernel status
// Disable operations of the kernel status
#define KS_INC(id)
#define KS_DEC(id)
#define KS_ADD(id, n)
#define KS_SUB(id, n)
#else // DONT_USE_KERNEL_STATUS
// Enable operations of the kernel status
#define KS_INC(id) \ #define KS_INC(id) \
if (kernel_status_inited) { \ if (IsTrackingEnabled()) { \
KS_LOCK(id); \ KS_LOCK(id); \
kernel_status[id]++; \ kernel_status[id]++; \
kernel_status_max[id] = MAX(kernel_status_max[id], kernel_status[id]); \ kernel_status_max[id] = MAX(kernel_status_max[id], kernel_status[id]); \
KS_UNLOCK(id); \ KS_UNLOCK(id); \
} }
#define KS_DEC(id) \ #define KS_DEC(id) \
if (kernel_status_inited) { \ if (IsTrackingEnabled()) { \
KS_LOCK(id); \ KS_LOCK(id); \
kernel_status[id]--; \ kernel_status[id]--; \
kernel_status_max[id] = MAX(kernel_status_max[id], kernel_status[id]); \ kernel_status_max[id] = MAX(kernel_status_max[id], kernel_status[id]); \
KS_UNLOCK(id); \ KS_UNLOCK(id); \
} }
#define KS_ADD(id, n) \ #define KS_ADD(id, n) \
if (kernel_status_inited) { \ if (IsTrackingEnabled()) { \
KS_LOCK(id); \ KS_LOCK(id); \
kernel_status[id] += n; \ kernel_status[id] += n; \
kernel_status_max[id] = MAX(kernel_status_max[id], kernel_status[id]); \ kernel_status_max[id] = MAX(kernel_status_max[id], kernel_status[id]); \
KS_UNLOCK(id); \ KS_UNLOCK(id); \
} }
#define KS_SUB(id, n) \ #define KS_SUB(id, n) \
if (kernel_status_inited) { \ if (IsTrackingEnabled()) { \
KS_LOCK(id); \ KS_LOCK(id); \
kernel_status[id] -= n; \ kernel_status[id] -= n; \
kernel_status_max[id] = MAX(kernel_status_max[id], kernel_status[id]); \ kernel_status_max[id] = MAX(kernel_status_max[id], kernel_status[id]); \
KS_UNLOCK(id); \ KS_UNLOCK(id); \
} }
#endif // DONT_USE_KERNEL_STATUS
// Kernel status // Kernel status
// String related // String related

57
src/Mayaqua/Memory.c
View File

@ -972,10 +972,6 @@ SK *NewSkEx(bool no_compact)
s->p = Malloc(sizeof(void *) * s->num_reserved); s->p = Malloc(sizeof(void *) * s->num_reserved);
s->no_compact = no_compact; s->no_compact = no_compact;
#ifndef DONT_USE_KERNEL_STATUS
// TrackNewObj(POINTER_TO_UINT64(s), "SK", 0);
#endif // DONT_USE_KERNEL_STATUS
// KS // KS
KS_INC(KS_NEWSK_COUNT); KS_INC(KS_NEWSK_COUNT);
@ -1011,10 +1007,6 @@ void CleanupSk(SK *s)
DeleteLock(s->lock); DeleteLock(s->lock);
Free(s); Free(s);
#ifndef DONT_USE_KERNEL_STATUS
// TrackDeleteObj(POINTER_TO_UINT64(s));
#endif // DONT_USE_KERNEL_STATUS
// KS // KS
KS_INC(KS_FREESK_COUNT); KS_INC(KS_FREESK_COUNT);
} }
@ -1278,10 +1270,6 @@ void CleanupQueue(QUEUE *q)
DeleteLock(q->lock); DeleteLock(q->lock);
Free(q); Free(q);
#ifndef DONT_USE_KERNEL_STATUS
// TrackDeleteObj(POINTER_TO_UINT64(q));
#endif // DONT_USE_KERNEL_STATUS
// KS // KS
KS_INC(KS_FREEQUEUE_COUNT); KS_INC(KS_FREEQUEUE_COUNT);
} }
@ -1297,10 +1285,6 @@ QUEUE *NewQueue()
q->num_item = 0; q->num_item = 0;
q->fifo = NewFifo(); q->fifo = NewFifo();
#ifndef DONT_USE_KERNEL_STATUS
// TrackNewObj(POINTER_TO_UINT64(q), "QUEUE", 0);
#endif // DONT_USE_KERNEL_STATUS
// KS // KS
KS_INC(KS_NEWQUEUE_COUNT); KS_INC(KS_NEWQUEUE_COUNT);
@ -1316,10 +1300,6 @@ QUEUE *NewQueueFast()
q->num_item = 0; q->num_item = 0;
q->fifo = NewFifoFast(); q->fifo = NewFifoFast();
#ifndef DONT_USE_KERNEL_STATUS
// TrackNewObj(POINTER_TO_UINT64(q), "QUEUE", 0);
#endif // DONT_USE_KERNEL_STATUS
// KS // KS
KS_INC(KS_NEWQUEUE_COUNT); KS_INC(KS_NEWQUEUE_COUNT);
@ -1783,10 +1763,6 @@ void CleanupList(LIST *o)
// KS // KS
KS_INC(KS_FREELIST_COUNT); KS_INC(KS_FREELIST_COUNT);
#ifndef DONT_USE_KERNEL_STATUS
// TrackDeleteObj(POINTER_TO_UINT64(o));
#endif // DONT_USE_KERNEL_STATUS
} }
// Check whether the specified number is already in the list // Check whether the specified number is already in the list
@ -2145,10 +2121,6 @@ LIST *NewListEx2(COMPARE *cmp, bool fast, bool fast_malloc)
o->cmp = cmp; o->cmp = cmp;
o->sorted = true; o->sorted = true;
#ifndef DONT_USE_KERNEL_STATUS
// TrackNewObj(POINTER_TO_UINT64(o), "LIST", 0);
#endif //DONT_USE_KERNEL_STATUS
// KS // KS
KS_INC(KS_NEWLIST_COUNT); KS_INC(KS_NEWLIST_COUNT);
@ -2422,10 +2394,6 @@ void CleanupFifo(FIFO *f)
Free(f->p); Free(f->p);
Free(f); Free(f);
#ifndef DONT_USE_KERNEL_STATUS
// TrackDeleteObj(POINTER_TO_UINT64(f));
#endif //DONT_USE_KERNEL_STATUS
// KS // KS
KS_INC(KS_FREEFIFO_COUNT); KS_INC(KS_FREEFIFO_COUNT);
} }
@ -2472,10 +2440,6 @@ FIFO *NewFifoEx2(bool fast, bool fixed)
f->p = Malloc(FIFO_INIT_MEM_SIZE); f->p = Malloc(FIFO_INIT_MEM_SIZE);
f->fixed = false; f->fixed = false;
#ifndef DONT_USE_KERNEL_STATUS
// TrackNewObj(POINTER_TO_UINT64(f), "FIFO", 0);
#endif // DONT_USE_KERNEL_STATUS
// KS // KS
KS_INC(KS_NEWFIFO_COUNT); KS_INC(KS_NEWFIFO_COUNT);
@ -2793,10 +2757,6 @@ BUF *NewBuf()
b->Current = 0; b->Current = 0;
b->SizeReserved = INIT_BUF_SIZE; b->SizeReserved = INIT_BUF_SIZE;
#ifndef DONT_USE_KERNEL_STATUS
// TrackNewObj(POINTER_TO_UINT64(b), "BUF", 0);
#endif // DONT_USE_KERNEL_STATUS
// KS // KS
KS_INC(KS_NEWBUF_COUNT); KS_INC(KS_NEWBUF_COUNT);
KS_INC(KS_CURRENT_BUF_COUNT); KS_INC(KS_CURRENT_BUF_COUNT);
@ -3241,10 +3201,6 @@ void FreeBuf(BUF *b)
// KS // KS
KS_INC(KS_FREEBUF_COUNT); KS_INC(KS_FREEBUF_COUNT);
KS_DEC(KS_CURRENT_BUF_COUNT); KS_DEC(KS_CURRENT_BUF_COUNT);
#ifndef DONT_USE_KERNEL_STATUS
// TrackDeleteObj(POINTER_TO_UINT64(b));
#endif // DONT_USE_KERNEL_STATUS
} }
// Compare BUFs whether two are identical // Compare BUFs whether two are identical
@ -3802,7 +3758,11 @@ void Free(void *addr)
// Check the memtag // Check the memtag
void CheckMemTag(MEMTAG *tag) void CheckMemTag(MEMTAG *tag)
{ {
#ifndef DONT_CHECK_HEAP if (IsTrackingEnabled() == false)
{
return;
}
// Validate arguments // Validate arguments
if (tag == NULL) if (tag == NULL)
{ {
@ -3815,7 +3775,6 @@ void CheckMemTag(MEMTAG *tag)
AbortExitEx("CheckMemTag: tag->Magic != MEMTAG_MAGIC"); AbortExitEx("CheckMemTag: tag->Magic != MEMTAG_MAGIC");
return; return;
} }
#endif // DONT_CHECK_HEAP
} }
// ZeroMalloc // ZeroMalloc
@ -3859,9 +3818,7 @@ void *InternalMalloc(UINT size)
OSSleep(MEMORY_SLEEP_TIME); OSSleep(MEMORY_SLEEP_TIME);
} }
#ifndef DONT_USE_KERNEL_STATUS
TrackNewObj(POINTER_TO_UINT64(addr), "MEM", size); TrackNewObj(POINTER_TO_UINT64(addr), "MEM", size);
#endif //DONT_USE_KERNEL_STATUS
return addr; return addr;
} }
@ -3879,9 +3836,7 @@ void InternalFree(void *addr)
KS_DEC(KS_CURRENT_MEM_COUNT); KS_DEC(KS_CURRENT_MEM_COUNT);
KS_INC(KS_FREE_COUNT); KS_INC(KS_FREE_COUNT);
#ifndef DONT_USE_KERNEL_STATUS
TrackDeleteObj(POINTER_TO_UINT64(addr)); TrackDeleteObj(POINTER_TO_UINT64(addr));
#endif // DONT_USE_KERNEL_STATUS
// Memory release // Memory release
OSMemoryFree(addr); OSMemoryFree(addr);
@ -3914,9 +3869,7 @@ void *InternalReAlloc(void *addr, UINT size)
OSSleep(MEMORY_SLEEP_TIME); OSSleep(MEMORY_SLEEP_TIME);
} }
#ifndef DONT_USE_KERNEL_STATUS
TrackChangeObjSize(POINTER_TO_UINT64(addr), size, POINTER_TO_UINT64(new_addr)); TrackChangeObjSize(POINTER_TO_UINT64(addr), size, POINTER_TO_UINT64(new_addr));
#endif // DONT_USE_KERNEL_STATUS
return new_addr; return new_addr;
} }

10
src/Mayaqua/Table.c
View File

@ -1071,8 +1071,6 @@ void FreeTable()
return; return;
} }
TrackingDisable();
num = LIST_NUM(TableList); num = LIST_NUM(TableList);
tables = ToArray(TableList); tables = ToArray(TableList);
for (i = 0;i < num;i++) for (i = 0;i < num;i++)
@ -1088,8 +1086,6 @@ void FreeTable()
Free(tables); Free(tables);
Zero(old_table_name, sizeof(old_table_name)); Zero(old_table_name, sizeof(old_table_name));
TrackingEnable();
} }
// Read a string table from the buffer // Read a string table from the buffer
@ -1479,8 +1475,6 @@ bool LoadTableW(wchar_t *filename)
Zero(replace_name, sizeof(replace_name)); Zero(replace_name, sizeof(replace_name));
TrackingDisable();
b = ReadDump("@table_name.txt"); b = ReadDump("@table_name.txt");
if (b != NULL) if (b != NULL)
{ {
@ -1500,9 +1494,5 @@ bool LoadTableW(wchar_t *filename)
ret = LoadTableMain(filename); ret = LoadTableMain(filename);
TrackingEnable();
return ret; return ret;
} }

48
src/Mayaqua/Tracking.c
View File

@ -127,27 +127,14 @@ static LOCK *obj_lock;
static LOCK *obj_id_lock; static LOCK *obj_id_lock;
static UINT obj_id; static UINT obj_id;
static LOCK *cs_lock; static LOCK *cs_lock;
static bool disable_tracking = false;
static TRACKING_LIST **hashlist; static TRACKING_LIST **hashlist;
static bool do_not_get_callstack; static bool do_not_get_callstack;
// Enable the tracking
void TrackingEnable()
{
disable_tracking = false;
}
// Disable the tracking
void TrackingDisable()
{
disable_tracking = true;
}
// Get whether the tracking is enabled // Get whether the tracking is enabled
bool IsTrackingEnabled() bool IsTrackingEnabled()
{ {
return !disable_tracking; return (IsDebug() || IsMemCheck()) && kernel_status_inited;
} }
// Memory debug menu // Memory debug menu
@ -482,14 +469,9 @@ void TrackNewObj(UINT64 addr, char *name, UINT size)
return; return;
} }
if (IsMemCheck() == false) if ((IsTrackingEnabled() && IsMemCheck()) == false)
{
// Don't track in the release mode
return;
}
if (disable_tracking)
{ {
// Don't track in detail if the memory check option is not specified
return; return;
} }
@ -528,14 +510,9 @@ void TrackDeleteObj(UINT64 addr)
return; return;
} }
if (IsMemCheck() == false) if ((IsTrackingEnabled() && IsMemCheck()) == false)
{
// Don't track in the release mode
return;
}
if (disable_tracking)
{ {
// Don't track in detail if the memory check option is not specified
return; return;
} }
@ -545,11 +522,7 @@ void TrackDeleteObj(UINT64 addr)
if (o == NULL) if (o == NULL)
{ {
UnlockTrackingList(); UnlockTrackingList();
Debug("TrackDeleteObj(): 0x%x not found in tracking list!\n", addr);
if (IsDebug())
{
printf("TrackDeleteObj: 0x%x is not Object!!\n", (void *)addr);
}
return; return;
} }
DeleteTrackingList(o, true); DeleteTrackingList(o, true);
@ -567,14 +540,9 @@ void TrackChangeObjSize(UINT64 addr, UINT size, UINT64 new_addr)
return; return;
} }
if (IsMemCheck() == false) if ((IsTrackingEnabled() && IsMemCheck()) == false)
{
// Don't track in the release mode
return;
}
if (disable_tracking)
{ {
// Don't track in detail if the memory check option is not specified
return; return;
} }

4
src/Mayaqua/Tracking.h
View File

@ -188,10 +188,6 @@ void PrintObjectList(TRACKING_OBJECT *o);
void PrintObjectInfo(TRACKING_OBJECT *o); void PrintObjectInfo(TRACKING_OBJECT *o);
void DebugPrintObjectInfo(UINT id); void DebugPrintObjectInfo(UINT id);
void TrackingEnable();
void TrackingDisable();
bool IsTrackingEnabled(); bool IsTrackingEnabled();
#endif // TRACKING_H #endif // TRACKING_H