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

Merge pull request #771 from JacobBarthelmeh/master

Browse Source 
account for static memory IO_POOL free when general memory was used
This commit is contained in:
toddouska
2017-03-03 08:31:55 -08:00
committed by GitHub
parent 0608cd0dd3 9ab28f9756
commit 2b937e2f23
1 changed files with 20 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
wolfcrypt/src/memory.c
View File

@ -539,7 +539,9 @@ void* wolfSSL_Malloc(size_t size, void* heap, int type)
}
/* case of using fixed IO buffers */
if (mem->flag & WOLFMEM_IO_POOL_FIXED) {
if (mem->flag & WOLFMEM_IO_POOL_FIXED &&
(type == DYNAMIC_TYPE_OUT_BUFFER ||
type == DYNAMIC_TYPE_IN_BUFFER)) {
if (type == DYNAMIC_TYPE_OUT_BUFFER) {
pt = hint->outBuf;
}
@ -547,9 +549,9 @@ void* wolfSSL_Malloc(size_t size, void* heap, int type)
pt = hint->inBuf;
}
}
else {
/* check if using IO pool flag */
if (mem->flag & WOLFMEM_IO_POOL && pt == NULL &&
if (mem->flag & WOLFMEM_IO_POOL &&
(type == DYNAMIC_TYPE_OUT_BUFFER ||
type == DYNAMIC_TYPE_IN_BUFFER)) {
if (mem->io != NULL) {
@ -570,6 +572,7 @@ void* wolfSSL_Malloc(size_t size, void* heap, int type)
}
}
}
}
if (pt != NULL) {
mem->inUse += pt->sz;
@ -672,7 +675,7 @@ void wolfSSL_Free(void *ptr, void* heap, int type)
/* fixed IO pools are free'd at the end of SSL lifetime
using FreeFixedIO(WOLFSSL_HEAP* heap, wc_Memory** io) */
}
else if (mem->flag & WOLFMEM_IO_POOL &&
else if (mem->flag & WOLFMEM_IO_POOL && pt->sz == WOLFMEM_IO_SZ &&
(type == DYNAMIC_TYPE_OUT_BUFFER ||
type == DYNAMIC_TYPE_IN_BUFFER)) {
pt->next = mem->io;