From d648d4f6c770adb2eafec21c77d502641e8e0f2e Mon Sep 17 00:00:00 2001
From: David Garske <david@wolfssl.com>
Date: Wed, 5 Apr 2017 14:24:55 -0700
Subject: [PATCH] Fix leak in StackSizeCheck. Fix build error with debug
 enabled and stack size check.

---
 wolfcrypt/benchmark/benchmark.c | 3 ++-
 wolfcrypt/test/test.c           | 3 ++-
 wolfssl/test.h                  | 2 ++
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/wolfcrypt/benchmark/benchmark.c b/wolfcrypt/benchmark/benchmark.c
index 794ad6392..140865bfa 100644
--- a/wolfcrypt/benchmark/benchmark.c
+++ b/wolfcrypt/benchmark/benchmark.c
@@ -250,7 +250,8 @@ void bench_rng(void);
     double current_time(int);
 #endif
 
-#if defined(DEBUG_WOLFSSL) && !defined(HAVE_VALGRIND)
+#if defined(DEBUG_WOLFSSL) && !defined(HAVE_VALGRIND) && \
+        !defined(HAVE_STACK_SIZE)
     WOLFSSL_API int wolfSSL_Debugging_ON();
 #endif
 
diff --git a/wolfcrypt/test/test.c b/wolfcrypt/test/test.c
index 731271310..71c1a15c9 100644
--- a/wolfcrypt/test/test.c
+++ b/wolfcrypt/test/test.c
@@ -293,7 +293,8 @@ int mutex_test(void);
 int memcb_test(void);
 #endif
 
-#if defined(DEBUG_WOLFSSL) && !defined(HAVE_VALGRIND) && !defined(OPENSSL_EXTRA)
+#if defined(DEBUG_WOLFSSL) && !defined(HAVE_VALGRIND) && \
+        !defined(OPENSSL_EXTRA) && !defined(HAVE_STACK_SIZE)
     int  wolfSSL_Debugging_ON(void);
 #endif
 
diff --git a/wolfssl/test.h b/wolfssl/test.h
index db73f406c..f00bfe619 100644
--- a/wolfssl/test.h
+++ b/wolfssl/test.h
@@ -1383,6 +1383,8 @@ static INLINE int StackSizeCheck(func_args* args, thread_func tf)
         }
     }
 
+    free(myStack);
+
     used = stackSize - i;
     printf("stack used = %d\n", used);