mirror of
https://github.com/Xevion/easy7zip.git
synced 2025-12-08 02:07:06 -06:00
21.02
This commit is contained in:
@@ -5,51 +5,73 @@
|
||||
|
||||
#include "../../../C/Aes.h"
|
||||
|
||||
#include "../../Common/MyBuffer2.h"
|
||||
#include "../../Common/MyCom.h"
|
||||
|
||||
#include "../ICoder.h"
|
||||
|
||||
namespace NCrypto {
|
||||
|
||||
class CAesCbcCoder:
|
||||
class CAesCoder:
|
||||
public ICompressFilter,
|
||||
public ICryptoProperties,
|
||||
#ifndef _SFX
|
||||
public ICompressSetCoderProperties,
|
||||
#endif
|
||||
public CMyUnknownImp
|
||||
{
|
||||
AES_CODE_FUNC _codeFunc;
|
||||
unsigned _offset;
|
||||
// unsigned _offset;
|
||||
unsigned _keySize;
|
||||
bool _keyIsSet;
|
||||
bool _encodeMode;
|
||||
UInt32 _aes[AES_NUM_IVMRK_WORDS + 3];
|
||||
bool _ctrMode;
|
||||
|
||||
// UInt32 _aes[AES_NUM_IVMRK_WORDS + 3];
|
||||
CAlignedBuffer _aes;
|
||||
|
||||
Byte _iv[AES_BLOCK_SIZE];
|
||||
|
||||
// UInt32 *Aes() { return _aes + _offset; }
|
||||
UInt32 *Aes() { return (UInt32 *)(void *)(Byte *)_aes; }
|
||||
|
||||
bool SetFunctions(UInt32 algo);
|
||||
|
||||
public:
|
||||
CAesCbcCoder(bool encodeMode, unsigned keySize);
|
||||
CAesCoder(bool encodeMode, unsigned keySize, bool ctrMode);
|
||||
|
||||
virtual ~CAesCbcCoder() {}; // we need virtual destructor for derived classes
|
||||
virtual ~CAesCoder() {}; // we need virtual destructor for derived classes
|
||||
|
||||
MY_UNKNOWN_IMP3(ICompressFilter, ICryptoProperties, ICompressSetCoderProperties)
|
||||
MY_QUERYINTERFACE_BEGIN2(ICompressFilter)
|
||||
MY_QUERYINTERFACE_ENTRY(ICryptoProperties)
|
||||
#ifndef _SFX
|
||||
MY_QUERYINTERFACE_ENTRY(ICompressSetCoderProperties)
|
||||
#endif
|
||||
MY_QUERYINTERFACE_END
|
||||
MY_ADDREF_RELEASE
|
||||
|
||||
INTERFACE_ICompressFilter(;)
|
||||
|
||||
void SetKeySize(unsigned size) { _keySize = size; }
|
||||
|
||||
STDMETHOD(SetKey)(const Byte *data, UInt32 size);
|
||||
STDMETHOD(SetInitVector)(const Byte *data, UInt32 size);
|
||||
|
||||
#ifndef _SFX
|
||||
STDMETHOD(SetCoderProperties)(const PROPID *propIDs, const PROPVARIANT *props, UInt32 numProps);
|
||||
#endif
|
||||
};
|
||||
|
||||
struct CAesCbcEncoder: public CAesCbcCoder
|
||||
#ifndef _SFX
|
||||
struct CAesCbcEncoder: public CAesCoder
|
||||
{
|
||||
CAesCbcEncoder(unsigned keySize = 0): CAesCbcCoder(true, keySize) {}
|
||||
CAesCbcEncoder(unsigned keySize = 0): CAesCoder(true, keySize, false) {}
|
||||
};
|
||||
#endif
|
||||
|
||||
struct CAesCbcDecoder: public CAesCbcCoder
|
||||
struct CAesCbcDecoder: public CAesCoder
|
||||
{
|
||||
CAesCbcDecoder(unsigned keySize = 0): CAesCbcCoder(false, keySize) {}
|
||||
CAesCbcDecoder(unsigned keySize = 0): CAesCoder(false, keySize, false) {}
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user