cRandomTest.h

Go to the documentation of this file.

//
  
#ifndef _INC_cRandomTest_H
#define _INC_cRandomTest_H
 
#include "GraySSLInt.h"
#include "GrayLib/include/Key/cRandomDSA.h"
#include "GrayLib/include/Cipher/cCipherAES.h"
 
namespace GraySSL
{
        class cRandomTest : public cRandomMixerBase
        {
 
        public:
                cCipherAES m_aes_ctx;        
                BYTE counter[16];                       
 
                static const int k_CTR_DRBG_MAX_SEED_INPUT = 384;     
 
                static const int k_CTR_DRBG_BLOCKSIZE = 16;     
                static const int k_CTR_DRBG_KEYSIZE = 32;      
                static const int k_CTR_DRBG_SEEDLEN = (k_CTR_DRBG_KEYSIZE + k_CTR_DRBG_BLOCKSIZE); 
 
        private:
                // Internal functions (do not call directly)
                HRESULT InitRandomTestCtr(IRandomNoise* pRand, const BYTE* pUserData, size_t, size_t);
                void ctr_drbg_update_internal(const BYTE data[k_CTR_DRBG_SEEDLEN]);
                static HRESULT block_cipher_df(BYTE* output, const BYTE* data, size_t data_len);
 
        public:
                cRandomTest() noexcept
                        : m_aes_ctx(true)
                {
                        cMem::Zero(counter, sizeof(counter));
                }
                ~cRandomTest() noexcept
                {
                        cMem::ZeroSecure(counter, sizeof(counter));
                }
 
                virtual HRESULT GetNoise(void* pData, size_t nSize) override;
 
                //
                //
                HRESULT InitRandomTest(IRandomNoise* pRand, const BYTE* pUserData, size_t len);
 
                //
 
                HRESULT ctr_drbg_reseed(const BYTE* pAddData, size_t len);
 
                //
                //
                //
                //
                void ctr_drbg_update(const BYTE* pAddData, size_t add_len);
 
                UNITTEST_FRIEND(cRandomTest);
        };
}
 
#endif // _INC_cRandomTest_H.h