summaryrefslogtreecommitdiffstats
path: root/src/mbedTLS++/CtrDrbgContext.h
blob: e83b3494c59253269ede5d33a5b94a9c26d3086d (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

// 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; }
};