1 files changed, 63 insertions, 0 deletions

diff --git a/src/mbedTLS++/CtrDrbgContext.h b/src/mbedTLS++/CtrDrbgContext.h
new file mode 100644
index 000000000..21d786c2e
--- /dev/null
+++ b/src/mbedTLS++/CtrDrbgContext.h
@@ -0,0 +1,63 @@
+
+// CtrDrbgContext.h
+
+// Declares the cCtrDrbgContext class representing a wrapper over CTR-DRBG implementation in mbedTLS
+
+
+
+
+
+#pragma once
+
+#include "mbedtls/ctr_drbg.h"
+
+
+
+
+
+// fwd: EntropyContext.h
+class cEntropyContext;
+
+
+
+
+
+class cCtrDrbgContext
+{
+	friend class cSslConfig;
+	friend class cRsaPrivateKey;
+	friend class cCryptoKey;
+
+public:
+	/** Constructs the context with a new entropy context. */
+	cCtrDrbgContext(void);
+
+	/** Constructs the context with the specified entropy context. */
+	cCtrDrbgContext(const std::shared_ptr<cEntropyContext> & a_EntropyContext);
+
+	/** Initializes the context.
+	a_Custom is optional additional data to use for entropy, nullptr is accepted.
+	Returns 0 if successful, mbedTLS error code on failure. */
+	int Initialize(const void * a_Custom, size_t a_CustomSize);
+
+	/** Returns true if the object is valid (has been initialized properly) */
+	bool IsValid(void) const { return m_IsValid; }
+
+protected:
+	/** The entropy source used for generating the random */
+	std::shared_ptr<cEntropyContext> m_EntropyContext;
+
+	/** The random generator context */
+	mbedtls_ctr_drbg_context m_CtrDrbg;
+
+	/** Set to true if the object is valid (has been initialized properly) */
+	bool m_IsValid;
+
+
+	/** Returns the internal context ptr. Only use in mbedTLS API calls. */
+	mbedtls_ctr_drbg_context * GetInternal(void) { return &m_CtrDrbg; }
+} ;
+
+
+
+