1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
// Crypto/AES/MyAES.cpp
#include "StdAfx.h"
#include "windows.h"
#include "MyAES.h"
#include "Windows/Defs.h"
#include "AES_CBC.h"
static const int kAESBlockSize = 16;
extern "C"
{
#include "aesopt.h"
}
class CTabInit
{
public:
CTabInit() { gen_tabs();}
} g_TabInit;
STDMETHODIMP CAESFilter::Init() { return S_OK; }
STDMETHODIMP_(UInt32) CAESFilter::Filter(Byte *data, UInt32 size)
{
if (size > 0 && size < kAESBlockSize)
return kAESBlockSize;
UInt32 i;
for (i = 0; i + kAESBlockSize <= size; i += kAESBlockSize)
{
Byte outBlock[kAESBlockSize];
SubFilter(data + i, outBlock);
for (int j = 0; j < kAESBlockSize; j++)
data[i + j] = outBlock[j];
}
return i;
}
STDMETHODIMP CAESFilter::SetInitVector(const Byte *data, UInt32 size)
{
if (size != 16)
return E_INVALIDARG;
AES.Init(data);
return S_OK;
}
STDMETHODIMP CAESEncoder::SetKey(const Byte *data, UInt32 size)
{ return (AES.enc_key(data, size) == aes_good) ? S_OK: E_FAIL; }
void CAESEncoder::SubFilter(const Byte *inBlock, Byte *outBlock)
{ AES.Encode(inBlock, outBlock); }
STDMETHODIMP CAESDecoder::SetKey(const Byte *data, UInt32 size)
{ return (AES.dec_key(data, size) == aes_good) ? S_OK: E_FAIL; }
void CAESDecoder::SubFilter(const Byte *inBlock, Byte *outBlock)
{ AES.Decode(inBlock, outBlock); }
////////////////////////////
// ECB mode
STDMETHODIMP CAesEcbFilter::Init() { return S_OK; }
STDMETHODIMP CAesEcbFilter::SetInitVector(const Byte * /* data */, UInt32 size)
{ return (size == 0) ? S_OK: E_INVALIDARG; }
STDMETHODIMP_(UInt32) CAesEcbFilter::Filter(Byte *data, UInt32 size)
{
if (size > 0 && size < kAESBlockSize)
return kAESBlockSize;
UInt32 i;
for (i = 0; i + kAESBlockSize <= size; i += kAESBlockSize)
{
Byte outBlock[kAESBlockSize];
SubFilter(data + i, outBlock);
for (int j = 0; j < kAESBlockSize; j++)
data[i + j] = outBlock[j];
}
return i;
}
STDMETHODIMP CAesEcbEncoder::SetKey(const Byte *data, UInt32 size)
{ return (AES.enc_key(data, size) == aes_good) ? S_OK: E_FAIL; }
void CAesEcbEncoder::SubFilter(const Byte *inBlock, Byte *outBlock)
{ AES.enc_blk(inBlock, outBlock); }
STDMETHODIMP CAesEcbDecoder::SetKey(const Byte *data, UInt32 size)
{ return (AES.dec_key(data, size) == aes_good) ? S_OK: E_FAIL; }
void CAesEcbDecoder::SubFilter(const Byte *inBlock, Byte *outBlock)
{ AES.dec_blk(inBlock, outBlock); }
|