TnbBytesCipher.h

[詳解]

#pragma once
#include "TnbStr.h"
#include "TnbSlideCompressor.h"
 
 
 
//TNB Library
namespace TNB
{
 
 
 
class CBytesCipher
{
    CAscii      m_ascKey;                   
    bool        m_boIsCompressionEnable;    
 
    bool m_Add(ICollectionT<BYTE>& _out, const IConstCollectionT<BYTE>& in, bool boIsAdd = true)
    {
        if ( m_ascKey.IsEmpty() )
        {
            ASSERT0(false, "CBytesCipher", "暗号、復号キーが指定されていません。");
            throw CEmptyException();
        }
        bool r = true;
        const BYTE* B = reinterpret_cast<const BYTE*>(LPCSTR(m_ascKey));
        size_t iKeySize = m_ascKey.GetLength();
        size_t base = _out.GetSize();
        _out.SetSize(base + in.GetSize());
        if ( boIsAdd )
        {
            loop ( i, in.GetSize() )
            {
                r &= _out.Set(base + i, static_cast<BYTE>(in.At(i) + B[i % iKeySize]));
            }
        }
        else
        {
            loop ( i, in.GetSize() )
            {
                r &= _out.Set(base + i, static_cast<BYTE>(in.At(i) - B[i % iKeySize]));
            }
        }
        return r;
    }
 
public:
 
    CBytesCipher(void) : m_boIsCompressionEnable(false)
    {
    }
 
    void SetKey(const CAscii& ascKey)
    {
        m_ascKey = ascKey;
    }
 
    void Compression(bool boIsEnable)
    {
        m_boIsCompressionEnable = boIsEnable;
    }
 
    bool Encode(ICollectionT<BYTE>& _out, const IConstCollectionT<BYTE>& in)
    {
        EXCLUSIVE2(&_out, &in);
        bool r = false;
        if ( m_boIsCompressionEnable )
        {
            CSlideCompressor comp;
            CByteVector vb;
            if ( comp.Encode(vb, in) )
            {
                r = m_Add(_out, vb);
            }
        }
        else
        {
            r = m_Add(_out, in);
        }
        return r;
    }
 
    CByteVector Encode(const IConstCollectionT<BYTE>& c)
    {
        CByteVector vb;
        if ( ! Encode(vb, c) )
        {
            vb.Invalid();
        }
        return vb;
    }
 
    bool Decode(ICollectionT<BYTE>& _out, const IConstCollectionT<BYTE>& in)
    {
        EXCLUSIVE2(&_out, &in);
        bool r = false;
        if ( m_boIsCompressionEnable )
        {
            CSlideCompressor comp;
            CByteVector vb;
            if ( m_Add(vb, in, false) )
            {
                r = comp.Decode(_out, vb);
            }
        }
        else
        {
            r = m_Add(_out, in, false);
        }
        return r;
    }
 
    CByteVector Decode(const IConstCollectionT<BYTE>& c)
    {
        CByteVector vb;
        if ( ! Decode(vb, c) )
        {
            vb.Invalid();
        }
        return vb;
    }
};
 
 
 
};//TNB