1 files changed, 51 insertions, 0 deletions

diff --git a/src/mbedTLS++/CtrDrbgContext.cpp b/src/mbedTLS++/CtrDrbgContext.cpp
new file mode 100644
index 000000000..bd4a55000
--- /dev/null
+++ b/src/mbedTLS++/CtrDrbgContext.cpp
@@ -0,0 +1,51 @@
+
+// CtrDrbgContext.cpp
+
+// Implements the cCtrDrbgContext class representing a wrapper over CTR-DRBG implementation in mbedTLS
+
+#include "Globals.h"
+#include "CtrDrbgContext.h"
+#include "EntropyContext.h"
+
+
+
+
+
+cCtrDrbgContext::cCtrDrbgContext(void) :
+	m_EntropyContext(std::make_shared<cEntropyContext>()),
+	m_IsValid(false)
+{
+	mbedtls_ctr_drbg_init(&m_CtrDrbg);
+}
+
+
+
+
+
+cCtrDrbgContext::cCtrDrbgContext(const std::shared_ptr<cEntropyContext> & a_EntropyContext) :
+	m_EntropyContext(a_EntropyContext),
+	m_IsValid(false)
+{
+	mbedtls_ctr_drbg_init(&m_CtrDrbg);
+}
+
+
+
+
+
+int cCtrDrbgContext::Initialize(const void * a_Custom, size_t a_CustomSize)
+{
+	if (m_IsValid)
+	{
+		// Already initialized
+		return 0;
+	}
+
+	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;
+}
+
+
+
+