1 files changed, 6 insertions, 4 deletions

diff --git a/src/PolarSSL++/CtrDrbgContext.cpp b/src/mbedTLS++/CtrDrbgContext.cpp
index c2243db38..bd4a55000 100644
--- a/src/PolarSSL++/CtrDrbgContext.cpp
+++ b/src/mbedTLS++/CtrDrbgContext.cpp
@@ -1,7 +1,7 @@
-
+
 // CtrDrbgContext.cpp
 
-// Implements the cCtrDrbgContext class representing a wrapper over CTR-DRBG implementation in PolarSSL
+// Implements the cCtrDrbgContext class representing a wrapper over CTR-DRBG implementation in mbedTLS
 
 #include "Globals.h"
 #include "CtrDrbgContext.h"
@@ -12,9 +12,10 @@
 
 
 cCtrDrbgContext::cCtrDrbgContext(void) :
-	m_EntropyContext(new cEntropyContext),
+	m_EntropyContext(std::make_shared<cEntropyContext>()),
 	m_IsValid(false)
 {
+	mbedtls_ctr_drbg_init(&m_CtrDrbg);
 }
 
 
@@ -25,6 +26,7 @@ cCtrDrbgContext::cCtrDrbgContext(const std::shared_ptr<cEntropyContext> & a_Entr
 	m_EntropyContext(a_EntropyContext),
 	m_IsValid(false)
 {
+	mbedtls_ctr_drbg_init(&m_CtrDrbg);
 }
 
 
@@ -39,7 +41,7 @@ int cCtrDrbgContext::Initialize(const void * a_Custom, size_t a_CustomSize)
 		return 0;
 	}
 
-	int res = ctr_drbg_init(&m_CtrDrbg, entropy_func, &(m_EntropyContext->m_Entropy), reinterpret_cast<const unsigned char *>(a_Custom), a_CustomSize);
+	int res = mbedtls_ctr_drbg_seed(&m_CtrDrbg, mbedtls_entropy_func, &(m_EntropyContext->m_Entropy), reinterpret_cast<const unsigned char *>(a_Custom), a_CustomSize);
 	m_IsValid = (res == 0);
 	return res;
 }