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
|
#include <cstring>
#include <cassert>
#include "aes.h"
using namespace std;
namespace AES{
//http://csrc.nist.gov/publications/fips/fips197/fips-197.pdf
void subWord(uint8_t *word);
void keyExpand(uint8_t *keysched,const uint8_t *key,int keylen,int numrounds);
void addRoundKey(uint8_t *state,const uint8_t *roundkey);
void subBytes(uint8_t *state){
subWord(state);
subWord(state+4);
subWord(state+8);
subWord(state+12);
}
void shiftRows(uint8_t *state);
void mixColumns(uint8_t *state);
void encryptBlock(uint8_t *state,const uint8_t *key,const uint8_t *data,int keylen,int numrounds){
uint8_t keysched[16*(numrounds+1)];
keyExpand(keysched,key,keylen,numrounds);
memcpy(state,data,16);
addRoundKey(state,keysched);
for(int round=0;round<numrounds-1;round++){
subBytes(state);
shiftRows(state);
mixColumns(state);
addRoundKey(state,keysched+16*(round+1));
}
subBytes(state);
shiftRows(state);
addRoundKey(state,keysched+16*numrounds);
}
string encrypt(const string &key,const string &data,int numrounds){
int sz=data.size();
if(sz==0)return {};
int blocks=sz/16;
int padding=sz%16==0?16:16-sz%16;
string res;
assert((sz+padding)%16==0);
res.reserve(sz+padding);
uint8_t buf[16];
for(int i=0;i<blocks;i++){
encryptBlock(buf,(const uint8_t*)key.data(),(const uint8_t*)data.data()+16*i,key.size(),numrounds);
res.insert(res.size(),(char*)buf,16);
}
uint8_t inbuf[16];
if(padding<16)memcpy(inbuf,data.data()+16*blocks,16-padding);
memset(inbuf+16-padding,padding,padding);
encryptBlock(buf,(const uint8_t*)key.data(),inbuf,key.size(),numrounds);
res.insert(res.size(),(char*)buf,16);
return res;
}
string decrypt(const string &key,const string &data,int numrounds);
string encrypt(const string &key,const string &data,Algorithm algo){
assert(key.size()==4+2*algo);
return encrypt(key,data,10+2*algo);
}
string decrypt(const string &key,const string &data,Algorithm algo)/*{
assert(key.size()==4+2*algo);
return decrypt(key,data,10+2*algo);
}*/;
}
|