cStreamCipher.h

Go to the documentation of this file.

//
//
 
#ifndef _INC_CStreamCrypt_H
#define _INC_CStreamCrypt_H
#ifndef NO_PRAGMA_ONCE
#pragma once
#endif
 
#include "../Cipher/cCipherBase.h"
#include "GrayCore/include/cStreamStack.h"
 
namespace GrayLib
{
        class GRAYLIB_LINK cStreamCipher
        {
 
        public:
                ICipherBase* const m_pCipher;   
 
        public:
                cStreamCipher(ICipherBase* pCipher)
                : m_pCipher( pCipher)
                {
                        ASSERT(m_pCipher != nullptr);
                }
        };
 
        class GRAYLIB_LINK cStreamCipherEnc : public cStreamStackOut, public cStreamCipher
        {
 
        public:
                cStreamCipherEnc(ICipherBase* pCipher, size_t nSizeBuffer = cStream::k_FILE_BLOCK_SIZE, cStreamOutput* pStreamOut = nullptr);
                virtual ~cStreamCipherEnc();
 
                virtual HRESULT WriteX(const void* pData, size_t nDataSize) override; // cStreamOutput override
        };
 
        class GRAYLIB_LINK cStreamCipherDec : public cStreamStackInp, public cStreamCipher
        {
 
        public:
                cStreamCipherDec(ICipherBase* pCipher, size_t nSizeBuffer /*= cStream::k_FILE_BLOCK_SIZE*/, cStreamInput* pStreamInp = nullptr);
                cStreamCipherDec(ICipherBase* pCipher);
                virtual ~cStreamCipherDec();
 
                virtual HRESULT ReadX(void* pData, size_t nDataSize) override;  // cStreamInput override
        };
};
#endif