tie in static memory debug callback

configure.ac

@@ -8077,6 +8077,10 @@ do
  ENABLED_STATICMEMORY=yes
  AM_CFLAGS="$AM_CFLAGS -DWOLFSSL_LEAN_STATIC_MEMORY"
  ;;
+ debug)
+ ENABLED_STATICMEMORY=yes
+ AM_CFLAGS="$AM_CFLAGS -DWOLFSSL_DEBUG_MEMORY_CALLBACK"
+ ;;
  *)
  AC_MSG_ERROR([Invalid choice for staticmemory.])
  break;;

examples/client/client.c

@@ -1865,6 +1865,38 @@ static int client_srtp_test(WOLFSSL *ssl, func_args *args)
 }
 #endif /* WOLFSSL_SRTP */
 
+#ifdef WOLFSSL_DEBUG_MEMORY_CALLBACK
+static void ExampleDebugMemoryCb(size_t sz, int bucketSz, byte st, int type) {
+ switch (st) {
+ case WOLFSSL_DEBUG_MEMORY_ALLOC:
+ if (type == DYNAMIC_TYPE_IN_BUFFER) {
+ printf("IN BUFFER: ");
+ }
+
+ if (type == DYNAMIC_TYPE_OUT_BUFFER) {
+ printf("OUT BUFFER: ");
+ }
+
+ printf("Alloc'd %d bytes using bucket size %d\n", (int)sz,
+ bucketSz);
+ break;
+
+ case WOLFSSL_DEBUG_MEMORY_FAIL:
+ printf("Failed when trying to allocate %d bytes\n", (int)sz);
+ break;
+
+ case WOLFSSL_DEBUG_MEMORY_FREE:
+ printf("Free'ing : %d\n", (int)sz);
+ break;
+
+ case WOLFSSL_DEBUG_MEMORY_INIT:
+ printf("Creating memory bucket of size : %d\n", bucketSz);
+ break;
+ }
+}
+#endif
+
+
 
 THREAD_RETURN WOLFSSL_THREAD client_test(void* args)
 {
@@ -3045,6 +3077,9 @@ THREAD_RETURN WOLFSSL_THREAD client_test(void* args)
  err_sys("unable to load static memory");
  }
 
+#ifdef WOLFSSL_DEBUG_MEMORY_CALLBACK
+ wolfSSL_SetDebugMemoryCb(ExampleDebugMemoryCb);
+#endif
  ctx = wolfSSL_CTX_new_ex(method(heap), heap);
  if (ctx == NULL)
  err_sys("unable to get ctx");

wolfcrypt/src/memory.c

@@ -531,9 +531,9 @@ static DebugMemoryCb DebugCb = NULL;
 
 /* Used to set a debug memory callback. Helpful in cases where
  * printf is not available. */
-void wolfSSL_SetDebugCallback(DebugMemoryCb in)
+void wolfSSL_SetDebugMemoryCb(DebugMemoryCb cb)
 {
- DebugCb = in;
+ DebugCb = cb;
 }
 #endif