From f2db01bef20a34ccfbe6e94298290ae88f5698d2 Mon Sep 17 00:00:00 2001
From: toddouska <todd@wolfssl.com>
Date: Wed, 26 Aug 2015 12:10:10 -0700
Subject: [PATCH] fix defrag memory errors

---
 src/internal.c | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/internal.c b/src/internal.c
index b7c6e912b..374e2bd68 100644
--- a/src/internal.c
+++ b/src/internal.c
@@ -1832,8 +1832,10 @@ void FreeArrays(WOLFSSL* ssl, int keep)
         XMEMCPY(ssl->session.sessionID, ssl->arrays->sessionID, ID_LEN);
         ssl->session.sessionIDSz = ssl->arrays->sessionIDSz;
     }
-    if (ssl->arrays)
+    if (ssl->arrays) {
         XFREE(ssl->arrays->pendingMsg, ssl->heap, DYNAMIC_TYPE_ARRAYS);
+        ssl->arrays->pendingMsg = NULL;
+    }
     XFREE(ssl->arrays, ssl->heap, DYNAMIC_TYPE_CERT);
     ssl->arrays = NULL;
 }
@@ -5131,6 +5133,8 @@ static int DoHandShakeMsg(WOLFSSL* ssl, byte* input, word32* inOutIdx,
             ssl->arrays->pendingMsg = (byte*)XMALLOC(size + HANDSHAKE_HEADER_SZ,
                                                      ssl->heap,
                                                      DYNAMIC_TYPE_ARRAYS);
+            if (ssl->arrays->pendingMsg == NULL)
+                return MEMORY_E;
             XMEMCPY(ssl->arrays->pendingMsg,
                     input + *inOutIdx - HANDSHAKE_HEADER_SZ, totalSz);
             ssl->arrays->pendingMsgOffset = totalSz;