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
|
#include "crypto/axtls/crypto.h"
#include <string.h>
#include <errno.h>
#include <gpxe/crypto.h>
#include <gpxe/aes.h>
static int aes_setkey ( void *ctx, const void *key, size_t keylen ) {
AES_CTX *aesctx = ctx;
AES_MODE mode;
switch ( keylen ) {
case ( 128 / 8 ):
mode = AES_MODE_128;
break;
case ( 256 / 8 ):
mode = AES_MODE_256;
break;
default:
return -EINVAL;
}
AES_set_key ( aesctx, key, aesctx->iv, mode );
return 0;
}
static void aes_setiv ( void *ctx, const void *iv ) {
AES_CTX *aesctx = ctx;
memcpy ( aesctx->iv, iv, sizeof ( aesctx->iv ) );
}
static void aes_encrypt ( void *ctx, const void *data, void *dst,
size_t len ) {
AES_CTX *aesctx = ctx;
AES_cbc_encrypt ( aesctx, data, dst, len );
}
static void aes_decrypt ( void *ctx, const void *data, void *dst,
size_t len ) {
AES_CTX *aesctx = ctx;
AES_cbc_decrypt ( aesctx, data, dst, len );
}
struct crypto_algorithm aes_algorithm = {
.name = "aes",
.ctxsize = sizeof ( AES_CTX ),
.blocksize = 16,
.setkey = aes_setkey,
.setiv = aes_setiv,
.encode = aes_encrypt,
.decode = aes_decrypt,
};
|