feedc0de/wolfssl
1
0
Fork 0
forked from wolfSSL/wolfssl
Code Pull Requests Activity

wolfssl/wolfcrypt/mem_track.h: refactor for linuxkm compatibility, mainly by supporting NO_STDIO_FILESYSTEM.

Browse Source
This commit is contained in:
Daniel Pouzzner
2023-05-17 01:13:36 -05:00
parent cc1ec8ace5
commit fc589d664e
1 changed files with 39 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
wolfssl/wolfcrypt/mem_track.h
View File

@ -62,15 +62,24 @@
#include "wolfssl/wolfcrypt/settings.h"
#include "wolfssl/wolfcrypt/logging.h"
#include "wolfssl/wolfcrypt/memory.h"
#if defined(WOLFSSL_TRACK_MEMORY) || defined(HAVE_STACK_SIZE) || \
defined(HAVE_STACK_SIZE_VERBOSE)
#include <stdio.h>
#ifdef NO_STDIO_FILESYSTEM
/* if wc_port.h/linuxkm_wc_port.h doesn't define printf, then the user
* needs to define it.
*/
#define wc_mem_printf(...) printf(__VA_ARGS__)
#else
#include <stdio.h>
#define wc_mem_printf(...) fprintf(stderr, __VA_ARGS__)
#endif
#endif
#if defined(WOLFSSL_TRACK_MEMORY)
#define DO_MEM_STATS
#if defined(__linux__) || defined(__MACH__)
#if (defined(__linux__) && !defined(WOLFSSL_LINUXKM)) || defined(__MACH__)
#define DO_MEM_LIST
#endif
#endif
@ -160,7 +169,7 @@ static WC_INLINE void* TrackMalloc(size_t sz)
#ifdef WOLFSSL_DEBUG_MEMORY
#ifdef WOLFSSL_DEBUG_MEMORY_PRINT
fprintf(stderr, "Alloc: %p -> %u at %s:%d\n", header->thisMemory, (word32)sz, func, line);
wc_mem_printf("Alloc: %p -> %u at %s:%d\n", header->thisMemory, (word32)sz, func, line);
#else
(void)func;
(void)line;
@ -268,7 +277,7 @@ static WC_INLINE void TrackFree(void* ptr)
#ifdef WOLFSSL_DEBUG_MEMORY
#ifdef WOLFSSL_DEBUG_MEMORY_PRINT
fprintf(stderr, "Free: %p -> %u at %s:%d\n", ptr, (word32)sz, func, line);
wc_mem_printf("Free: %p -> %u at %s:%d\n", ptr, (word32)sz, func, line);
#else
(void)func;
(void)line;
@ -334,11 +343,11 @@ static WC_INLINE int InitMemoryTracker(void)
ret = wolfSSL_GetAllocators(&mfDefault, &ffDefault, &rfDefault);
if (ret < 0) {
fprintf(stderr, "wolfSSL GetAllocators failed to get the defaults\n");
wc_mem_printf("wolfSSL GetAllocators failed to get the defaults\n");
}
ret = wolfSSL_SetAllocators(TrackMalloc, TrackFree, TrackRealloc);
if (ret < 0) {
fprintf(stderr, "wolfSSL SetAllocators failed for track memory\n");
wc_mem_printf("wolfSSL SetAllocators failed for track memory\n");
return ret;
}
@ -377,11 +386,11 @@ static WC_INLINE void ShowMemoryTracker(void)
#endif
#ifdef DO_MEM_STATS
fprintf(stderr, "total Allocs = %9ld\n", ourMemStats.totalAllocs);
fprintf(stderr, "total Deallocs = %9ld\n", ourMemStats.totalDeallocs);
fprintf(stderr, "total Bytes = %9ld\n", ourMemStats.totalBytes);
fprintf(stderr, "peak Bytes = %9ld\n", ourMemStats.peakBytes);
fprintf(stderr, "current Bytes = %9ld\n", ourMemStats.currentBytes);
wc_mem_printf("total Allocs = %9ld\n", ourMemStats.totalAllocs);
wc_mem_printf("total Deallocs = %9ld\n", ourMemStats.totalDeallocs);
wc_mem_printf("total Bytes = %9ld\n", ourMemStats.totalBytes);
wc_mem_printf("peak Bytes = %9ld\n", ourMemStats.peakBytes);
wc_mem_printf("current Bytes = %9ld\n", ourMemStats.currentBytes);
#endif
#ifdef DO_MEM_LIST
@ -389,16 +398,14 @@ static WC_INLINE void ShowMemoryTracker(void)
/* print list of allocations */
memHint* header;
for (header = ourMemList.head; header != NULL; header = header->next) {
fprintf(stderr, "Leak: Ptr %p, Size %u"
#ifdef WOLFSSL_DEBUG_MEMORY
", Func %s, Line %d"
#endif
"\n",
(byte*)header + sizeof(memHint), (unsigned int)header->thisSize
#ifdef WOLFSSL_DEBUG_MEMORY
, header->func, header->line
#endif
);
wc_mem_printf("Leak: Ptr %p, Size %u, Func %s, Line %d\n",
(byte*)header + sizeof(memHint), (unsigned int)header->thisSize,
header->func, header->line);
#else
wc_mem_printf("Leak: Ptr %p, Size %u\n",
(byte*)header + sizeof(memHint), (unsigned int)header->thisSize);
#endif
}
}
@ -538,7 +545,7 @@ int StackSizeHWMReset(void)
#define STACK_SIZE_CHECKPOINT_MSG(msg) ({ \
ssize_t HWM = StackSizeHWM_OffsetCorrected(); \
fprintf(stderr, "%ld\t%s\n", (long int)HWM, msg); \
wc_mem_printf("%ld\t%s\n", (long int)HWM, msg); \
StackSizeHWMReset(); \
})
@ -549,7 +556,7 @@ int StackSizeHWMReset(void)
printf(" relative stack peak usage = %ld bytes\n", (long int)HWM); \
_ret = StackSizeHWMReset(); \
if ((max >= 0) && (HWM > (ssize_t)(max))) { \
fprintf(stderr, \
wc_mem_printf( \
" relative stack usage at %s L%d exceeds designated max %ld bytes.\n", \
__FILE__, __LINE__, (long int)(max)); \
_ret = -1; \
@ -585,7 +592,7 @@ static WC_INLINE int StackSizeCheck(struct func_args* args, thread_func tf)
ret = posix_memalign((void**)&myStack, sysconf(_SC_PAGESIZE), stackSize);
if (ret != 0 || myStack == NULL) {
fprintf(stderr, "posix_memalign failed\n");
wc_mem_printf("posix_memalign failed\n");
return -1;
}
@ -593,13 +600,13 @@ static WC_INLINE int StackSizeCheck(struct func_args* args, thread_func tf)
ret = pthread_attr_init(&myAttr);
if (ret != 0) {
fprintf(stderr, "attr_init failed\n");
wc_mem_printf("attr_init failed\n");
return ret;
}
ret = pthread_attr_setstack(&myAttr, myStack, stackSize);
if (ret != 0) {
fprintf(stderr, "attr_setstackaddr failed\n");
wc_mem_printf("attr_setstackaddr failed\n");
return ret;
}
@ -623,7 +630,7 @@ static WC_INLINE int StackSizeCheck(struct func_args* args, thread_func tf)
ret = pthread_join(threadId, &status);
if (ret != 0) {
fprintf(stderr, "pthread_join failed\n");
wc_mem_printf("pthread_join failed\n");
return ret;
}
@ -672,7 +679,7 @@ static WC_INLINE int StackSizeCheck_launch(struct func_args* args,
ret = posix_memalign((void**)&myStack, sysconf(_SC_PAGESIZE), stackSize);
if (ret != 0 || myStack == NULL) {
fprintf(stderr, "posix_memalign failed\n");
wc_mem_printf("posix_memalign failed\n");
free(shim_args);
return -1;
}
@ -681,7 +688,7 @@ static WC_INLINE int StackSizeCheck_launch(struct func_args* args,
ret = pthread_attr_init(&myAttr);
if (ret != 0) {
fprintf(stderr, "attr_init failed\n");
wc_mem_printf("attr_init failed\n");
free(shim_args);
free(myStack);
return ret;
@ -689,7 +696,7 @@ static WC_INLINE int StackSizeCheck_launch(struct func_args* args,
ret = pthread_attr_setstack(&myAttr, myStack, stackSize);
if (ret != 0) {
fprintf(stderr, "attr_setstackaddr failed\n");
wc_mem_printf("attr_setstackaddr failed\n");
}
shim_args->myStack = myStack;
@ -721,7 +728,7 @@ static WC_INLINE int StackSizeCheck_reap(pthread_t threadId, void *stack_context
void *status;
int ret = pthread_join(threadId, &status);
if (ret != 0) {
fprintf(stderr, "pthread_join failed\n");
wc_mem_printf("pthread_join failed\n");
return ret;
}
@ -770,12 +777,12 @@ static WC_INLINE void StackTrap(void)
{
struct rlimit rl;
if (getrlimit(RLIMIT_STACK, &rl) != 0) {
fprintf(stderr, "getrlimit failed\n");
wc_mem_printf("getrlimit failed\n");
}
printf("rlim_cur = %llu\n", rl.rlim_cur);
rl.rlim_cur = 1024*21; /* adjust trap size here */
if (setrlimit(RLIMIT_STACK, &rl) != 0) {
fprintf(stderr, "setrlimit failed\n");
wc_mem_printf("setrlimit failed\n");
}
}