fix bug with DTLS and IO Pools

This commit is contained in:
John Safranek
2014-06-15 17:26:18 -07:00
parent 33fb679334
commit acb3b1afb2

View File

@@ -2082,7 +2082,7 @@ int DtlsPoolSave(CYASSL* ssl, const byte *src, int sz)
DtlsPool *pool = ssl->dtls_pool; DtlsPool *pool = ssl->dtls_pool;
if (pool != NULL && pool->used < DTLS_POOL_SZ) { if (pool != NULL && pool->used < DTLS_POOL_SZ) {
buffer *pBuf = &pool->buf[pool->used]; buffer *pBuf = &pool->buf[pool->used];
pBuf->buffer = (byte*)XMALLOC(sz, ssl->heap, DYNAMIC_TYPE_OUT_BUFFER); pBuf->buffer = (byte*)XMALLOC(sz, ssl->heap, DYNAMIC_TYPE_DTLS_POOL);
if (pBuf->buffer == NULL) { if (pBuf->buffer == NULL) {
CYASSL_MSG("DTLS Buffer Memory error"); CYASSL_MSG("DTLS Buffer Memory error");
return MEMORY_ERROR; return MEMORY_ERROR;
@@ -2104,7 +2104,7 @@ void DtlsPoolReset(CYASSL* ssl)
used = pool->used; used = pool->used;
for (i = 0, pBuf = &pool->buf[0]; i < used; i++, pBuf++) { for (i = 0, pBuf = &pool->buf[0]; i < used; i++, pBuf++) {
XFREE(pBuf->buffer, ssl->heap, DYNAMIC_TYPE_OUT_BUFFER); XFREE(pBuf->buffer, ssl->heap, DYNAMIC_TYPE_DTLS_POOL);
pBuf->buffer = NULL; pBuf->buffer = NULL;
pBuf->length = 0; pBuf->length = 0;
} }