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
95
96
97
98
|
/*
* Copyright (c) 2015 Egor Tensin <egor@tensin.name>
* This file is part of the "AES tools" project.
* For details, see https://github.com/egor-tensin/aes-tools.
* Distributed under the MIT License.
*/
#include <aes/all.h>
#include <emmintrin.h>
#include <wmmintrin.h>
AES_Block AES_ASM_ATTR
aes128_encrypt_block_internal(AES_Block plaintext, const AES128_RoundKeys* encryption_keys) {
plaintext = _mm_xor_si128(plaintext, encryption_keys->keys[0]);
plaintext = _mm_aesenc_si128(plaintext, encryption_keys->keys[1]);
plaintext = _mm_aesenc_si128(plaintext, encryption_keys->keys[2]);
plaintext = _mm_aesenc_si128(plaintext, encryption_keys->keys[3]);
plaintext = _mm_aesenc_si128(plaintext, encryption_keys->keys[4]);
plaintext = _mm_aesenc_si128(plaintext, encryption_keys->keys[5]);
plaintext = _mm_aesenc_si128(plaintext, encryption_keys->keys[6]);
plaintext = _mm_aesenc_si128(plaintext, encryption_keys->keys[7]);
plaintext = _mm_aesenc_si128(plaintext, encryption_keys->keys[8]);
plaintext = _mm_aesenc_si128(plaintext, encryption_keys->keys[9]);
return _mm_aesenclast_si128(plaintext, encryption_keys->keys[10]);
}
AES_Block AES_ASM_ATTR
aes128_decrypt_block_internal(AES_Block ciphertext, const AES128_RoundKeys* decryption_keys) {
ciphertext = _mm_xor_si128(ciphertext, decryption_keys->keys[0]);
ciphertext = _mm_aesdec_si128(ciphertext, decryption_keys->keys[1]);
ciphertext = _mm_aesdec_si128(ciphertext, decryption_keys->keys[2]);
ciphertext = _mm_aesdec_si128(ciphertext, decryption_keys->keys[3]);
ciphertext = _mm_aesdec_si128(ciphertext, decryption_keys->keys[4]);
ciphertext = _mm_aesdec_si128(ciphertext, decryption_keys->keys[5]);
ciphertext = _mm_aesdec_si128(ciphertext, decryption_keys->keys[6]);
ciphertext = _mm_aesdec_si128(ciphertext, decryption_keys->keys[7]);
ciphertext = _mm_aesdec_si128(ciphertext, decryption_keys->keys[8]);
ciphertext = _mm_aesdec_si128(ciphertext, decryption_keys->keys[9]);
return _mm_aesdeclast_si128(ciphertext, decryption_keys->keys[10]);
}
static AES_Block AES_ASM_ATTR aes128_expand_key_assist(AES_Block prev, AES_Block hwgen) {
AES_Block tmp = prev;
tmp = _mm_slli_si128(tmp, 4);
prev = _mm_xor_si128(prev, tmp);
tmp = _mm_slli_si128(tmp, 4);
prev = _mm_xor_si128(prev, tmp);
tmp = _mm_slli_si128(tmp, 4);
prev = _mm_xor_si128(prev, tmp);
hwgen = _mm_shuffle_epi32(hwgen, 0xff);
prev = _mm_xor_si128(prev, hwgen);
return prev;
}
void AES_ASM_ATTR aes128_expand_key_internal(AES_Block key, AES128_RoundKeys* encryption_keys) {
AES_Block prev = encryption_keys->keys[0] = key;
prev = encryption_keys->keys[1] =
aes128_expand_key_assist(prev, _mm_aeskeygenassist_si128(prev, 0x01));
prev = encryption_keys->keys[2] =
aes128_expand_key_assist(prev, _mm_aeskeygenassist_si128(prev, 0x02));
prev = encryption_keys->keys[3] =
aes128_expand_key_assist(prev, _mm_aeskeygenassist_si128(prev, 0x04));
prev = encryption_keys->keys[4] =
aes128_expand_key_assist(prev, _mm_aeskeygenassist_si128(prev, 0x08));
prev = encryption_keys->keys[5] =
aes128_expand_key_assist(prev, _mm_aeskeygenassist_si128(prev, 0x10));
prev = encryption_keys->keys[6] =
aes128_expand_key_assist(prev, _mm_aeskeygenassist_si128(prev, 0x20));
prev = encryption_keys->keys[7] =
aes128_expand_key_assist(prev, _mm_aeskeygenassist_si128(prev, 0x40));
prev = encryption_keys->keys[8] =
aes128_expand_key_assist(prev, _mm_aeskeygenassist_si128(prev, 0x80));
prev = encryption_keys->keys[9] =
aes128_expand_key_assist(prev, _mm_aeskeygenassist_si128(prev, 0x1b));
prev = encryption_keys->keys[10] =
aes128_expand_key_assist(prev, _mm_aeskeygenassist_si128(prev, 0x36));
}
void AES_ASM_ATTR aes128_derive_decryption_keys_internal(
const AES128_RoundKeys* encryption_keys,
AES128_RoundKeys* decryption_keys
) {
decryption_keys->keys[0] = encryption_keys->keys[10];
decryption_keys->keys[1] = _mm_aesimc_si128(encryption_keys->keys[9]);
decryption_keys->keys[2] = _mm_aesimc_si128(encryption_keys->keys[8]);
decryption_keys->keys[3] = _mm_aesimc_si128(encryption_keys->keys[7]);
decryption_keys->keys[4] = _mm_aesimc_si128(encryption_keys->keys[6]);
decryption_keys->keys[5] = _mm_aesimc_si128(encryption_keys->keys[5]);
decryption_keys->keys[6] = _mm_aesimc_si128(encryption_keys->keys[4]);
decryption_keys->keys[7] = _mm_aesimc_si128(encryption_keys->keys[3]);
decryption_keys->keys[8] = _mm_aesimc_si128(encryption_keys->keys[2]);
decryption_keys->keys[9] = _mm_aesimc_si128(encryption_keys->keys[1]);
decryption_keys->keys[10] = encryption_keys->keys[0];
}
|
