cCipherRC4.h

Go to the documentation of this file.

//
//
#ifndef _INC_cCipherRC4_H
#define _INC_cCipherRC4_H
#ifndef NO_PRAGMA_ONCE
#pragma once
#endif
 
#include "cCipherBase.h"
#include "cCipherTypeDef.h"
 
namespace GrayLib
{
        class GRAYLIB_LINK cCipherRC4 : public cCipherBase
        {
 
        protected:
                int m_x;        
                int m_y;        
                BYTE m_Sbox[256];       
 
        public:
                cCipherRC4(bool bEncodeMode = true);
                cCipherRC4(bool bEncodeMode, const void* pKeyData, size_t nKeySize);
                virtual ~cCipherRC4(void);
 
                // cCipherBase support
                virtual HRESULT SetCipherKey(const void* pKeyData = nullptr, size_t nKeySize = 0) override;
 
                virtual HRESULT Cipher(BYTE* pOutput, const BYTE* pInput, size_t nSizeBytes) override;
 
                UNITTEST_FRIEND(cCipherRC4);
        };
 
        class GRAYLIB_LINK cCipherAlgRC4 : public cCipherAlgDef
        {
        public:
                cCipherAlgRC4() : cCipherAlgDef(CipherAlgorithm_RC4)
                {
                }
                virtual cCipherBase* AllocCipherAlg(bool bEncodeMode) const
                {
                        return new cCipherRC4(bEncodeMode);
                }
        };
};
 
#endif // _INC_cCipherRC4_H