; Copyright (c) 2015 Egor Tensin <Egor.Tensin@gmail.com>
; This file is part of the "AES tools" project.
; For details, see https://github.com/egor-tensin/aes-tools.
; Distributed under the MIT License.
.586
.xmm
.model flat
.code
@aes_AES192_encrypt_block_@20 proc
pxor xmm0, [ecx]
aesenc xmm0, [ecx + 10h]
aesenc xmm0, [ecx + 20h]
aesenc xmm0, [ecx + 30h]
aesenc xmm0, [ecx + 40h]
aesenc xmm0, [ecx + 50h]
aesenc xmm0, [ecx + 60h]
aesenc xmm0, [ecx + 70h]
aesenc xmm0, [ecx + 80h]
aesenc xmm0, [ecx + 90h]
aesenc xmm0, [ecx + 0A0h]
aesenc xmm0, [ecx + 0B0h]
aesenclast xmm0, [ecx + 0C0h]
ret
@aes_AES192_encrypt_block_@20 endp
@aes_AES192_decrypt_block_@20 proc
pxor xmm0, [ecx]
aesdec xmm0, [ecx + 10h]
aesdec xmm0, [ecx + 20h]
aesdec xmm0, [ecx + 30h]
aesdec xmm0, [ecx + 40h]
aesdec xmm0, [ecx + 50h]
aesdec xmm0, [ecx + 60h]
aesdec xmm0, [ecx + 70h]
aesdec xmm0, [ecx + 80h]
aesdec xmm0, [ecx + 90h]
aesdec xmm0, [ecx + 0A0h]
aesdec xmm0, [ecx + 0B0h]
aesdeclast xmm0, [ecx + 0C0h]
ret
@aes_AES192_decrypt_block_@20 endp
@aes_AES192_expand_key_@36 proc
; A "word" (in terms of the FIPS 187 standard) is a 32-bit block.
; Words are denoted by `w[N]`.
;
; A key schedule is composed of 12 "regular" keys and a dumb key for
; the "whitening" step.
;
; A key schedule is thus composed of 52 "words".
; The FIPS standard includes an algorithm to calculate these words via
; a simple loop:
;
; i = 6
; while i < 52:
; temp = w[i - 1]
; if i % 6 == 0:
; temp = SubWord(RotWord(temp))^Rcon
; w[i] = w[i - 6]^temp
; i = i + 1
;
; The loop above may be unrolled like this:
;
; w[6] = SubWord(RotWord(w[5]))^Rcon^w[0]
; w[7] = w[6]^w[1]
; = SubWord(RotWord(w[5]))^Rcon^w[0]^w[1]
; w[8] = w[7]^w[2]
; = SubWord(RotWord(w[5]))^Rcon^w[0]^w[1]^w[2]
; w[9] = w[8]^w[3]
; = SubWord(RotWord(w[5]))^Rcon^w[0]^w[1]^w[2]^w[3]
; w[10] = w[9]^w[4]
; = SubWord(RotWord(w[5]))^Rcon^w[0]^w[1]^w[2]^w[3]^w[4]
; w[11] = w[10]^w[5]
; = SubWord(RotWord(w[5]))^Rcon^w[0]^w[1]^w[2]^w[3]^w[4]^w[5]
; w[12] = SubWord(RotWord(w[11]))^Rcon^w[6]
; w[13] = w[12]^w[7]
; = SubWord(RotWord(w[11]))^Rcon^w[6]^w[7]
; w[14] = w[13]^w[8]
; = SubWord(RotWord(w[11]))^Rcon^w[6]^w[7]^w[8]
; w[15] = w[14]^w[9]
; = SubWord(RotWord(w[11]))^Rcon^w[6]^w[7]^w[8]^w[9]
; w[16] = w[15]^w[10]
; = SubWord(RotWord(w[11]))^Rcon^w[6]^w[7]^w[8]^w[9]^w[10]
; w[17] = w[16]^w[11]
; = SubWort(RotWord(w[11]))^Rcon^w[6]^w[7]^w[8]^w[9]^w[10]^w[11]
;
; ... and so on.
;
; The Intel AES-NI instruction set facilitates calculating SubWord
; and RotWord using `aeskeygenassist`, which is used in this routine.
;
; Preconditions:
; * xmm1[63:32] == w[5],
; * xmm1[31:0] == w[4],
; * xmm0[127:96] == w[3],
; * xmm0[95:64] == w[2],
; * xmm0[63:32] == w[1],
; * xmm0[31:0] == w[0].
movdqa [ecx], xmm0 ; sets w[0], w[1], w[2], w[3]
movq qword ptr [ecx + 10h], xmm1 ; sets w[4], w[5]
aeskeygenassist xmm5, xmm1, 1 ; xmm5[63:32] = RotWord(SubWord(w[5]))^Rcon,
call aes192_keygen_assist
movdqu [ecx + 18h], xmm0
movq qword ptr [ecx + 28h], xmm1
aeskeygenassist xmm5, xmm1, 2 ; xmm5[63:32] = RotWord(SubWord(w[11]))^Rcon
call aes192_keygen_assist
movdqa [ecx + 30h], xmm0
movq qword ptr [ecx + 40h], xmm1
aeskeygenassist xmm5, xmm1, 4 ; xmm5[63:32] = RotWord(SubWord(w[17]))^Rcon
call aes192_keygen_assist
movdqu [ecx + 48h], xmm0
movq qword ptr [ecx + 58h], xmm1
aeskeygenassist xmm5, xmm1, 8 ; xmm5[63:32] = RotWord(SubWord(w[23]))^Rcon
call aes192_keygen_assist
movdqa [ecx + 60h], xmm0
movq qword ptr [ecx + 70h], xmm1
aeskeygenassist xmm5, xmm1, 10h ; xmm5[63:32] = RotWord(SubWord(w[29]))^Rcon
call aes192_keygen_assist
movdqu [ecx + 78h], xmm0
movq qword ptr [ecx + 88h], xmm1
aeskeygenassist xmm5, xmm1, 20h ; xmm5[63:32] = RotWord(SubWord(w[35]))^Rcon
call aes192_keygen_assist
movdqa [ecx + 90h], xmm0
movq qword ptr [ecx + 0a0h], xmm1
aeskeygenassist xmm5, xmm1, 40h ; xmm5[63:32] = RotWord(SubWord(w[41]))^Rcon
call aes192_keygen_assist
movdqu [ecx + 0a8h], xmm0
movq qword ptr [ecx + 0b8h], xmm1
aeskeygenassist xmm5, xmm1, 80h ; xmm5[63:32] = RotWord(SubWord(w[49]))^Rcon
call aes192_keygen_assist
movdqa [ecx + 0c0h], xmm0
ret
aes192_keygen_assist:
; Preconditions:
; * xmm1[127:96] == 0,
; * xmm1[95:64] == 0,
; * xmm1[63:32] == w[i+5],
; * xmm1[31:0] == w[i+4],
; * xmm0[127:96] == w[i+3],
; * xmm0[95:64] == w[i+2],
; * xmm0[63:32] == w[i+1],
; * xmm0[31:0] == w[i],
; * xmm5[63:32] == RotWord(SubWord(w[i+5]))^Rcon.
;
; Postconditions:
; * xmm1[127:96] == 0,
; * xmm1[95:64] == 0,
; * xmm1[63:32] == w[i+11] == RotWord(SubWord(w[i+5]))^Rcon^w[i+5]^w[i+4]^w[i+3]^w[i+2]^w[i+1]^w[i],
; * xmm1[31:0] == w[i+10] == RotWord(SubWord(w[i+5]))^Rcon^w[i+4]^w[i+3]^w[i+2]^w[i+1]^w[i],
; * xmm0[127:96] == w[i+9] == RotWord(SubWord(w[i+5]))^Rcon^w[i+3]^w[i+2]^w[i+1]^w[i],
; * xmm0[95:64] == w[i+8] == RotWord(SubWord(w[i+5]))^Rcon^w[i+2]^w[i+1]^w[i],
; * xmm0[63:32] == w[i+7] == RotWord(SubWord(w[i+5]))^Rcon^w[i+1]^w[i],
; * xmm0[31:0] == w[i+6] == RotWord(SubWord(w[i+5]))^Rcon^w[i],
; * the value in xmm4 is also modified.
; Calculate
; w[i+3]^w[i+2]^w[i+1]^w[i],
; w[i+2]^w[i+1]^w[i],
; w[i+1]^w[i] and
; w[i].
movdqa xmm4, xmm0 ; xmm4 = xmm0
pslldq xmm4, 4 ; xmm4 <<= 32
pxor xmm0, xmm4 ; xmm0 ^= xmm4
pslldq xmm4, 4 ; xmm4 <<= 32
pxor xmm0, xmm4 ; xmm0 ^= xmm4
pslldq xmm4, 4 ; xmm4 <<= 32
pxor xmm0, xmm4 ; xmm0 ^= xmm4
; xmm0[127:96] == w[i+3]^w[i+2]^w[i+1]^w[i]
; xmm0[95:64] == w[i+2]^w[i+1]^w[i]
; xmm0[63:32] == w[i+1]^w[i]
; xmm0[31:0] == w[i]
; Calculate
; w[i+9] == RotWord(SubWord(w[i+5]))^Rcon^w[i+3]^w[i+2]^w[i+1]^w[i],
; w[i+8] == RotWord(SubWord(w[i+5]))^Rcon^w[i+2]^w[i+1]^w[i],
; w[i+7] == RotWord(SubWord(w[i+5]))^Rcon^w[i+1]^w[i] and
; w[i+6] == RotWord(SubWord(w[i+5]))^Rcon^w[i].
pshufd xmm4, xmm5, 55h ; xmm4[127:96] = xmm4[95:64] = xmm4[63:32] = xmm4[31:0] = xmm5[63:32]
pxor xmm0, xmm4 ; xmm0 ^= xmm4
; xmm0[127:96] == w[i+9] == RotWord(SubWord(w[i+5]))^Rcon^w[i+3]^w[i+2]^w[i+1]^w[i]
; xmm0[95:64] == w[i+8] == RotWord(SubWord(w[i+5]))^Rcon^w[i+2]^w[i+1]^w[i]
; xmm0[63:32] == w[i+7] == RotWord(SubWord(w[i+5]))^Rcon^w[i+1]^w[i]
; xmm0[31:0] == w[i+6] == RotWord(SubWord(w[i+5]))^Rcon^w[i]
; Calculate
; w[i+5]^w[i+4],
; w[i+4].
pshufd xmm4, xmm1, 0F3h ; xmm4 = xmm1[31:0] << 32
pxor xmm1, xmm4 ; xmm1 ^= xmm5
; xmm1[63:32] == w[i+5]^w[i+4]
; xmm1[31:0] == w[i+4]
; Calculate
; w[i+10] == RotWord(SubWord(w[i+5]))^Rcon^w[i+5]^w[i+4]^w[i+3]^w[i+2]^w[i+1]^w[i],
; w[i+11] == RotWord(SubWord(w[i+5]))^Rcon^w[i+4]^w[i+3]^w[i+2]^w[i+1]^w[i].
pshufd xmm4, xmm0, 0FFh ; xmm4[127:96] = xmm4[95:64] = xmm4[63:32] = xmm4[31:0] = xmm0[127:96]
psrldq xmm4, 8 ; xmm4 >>= 64
pxor xmm1, xmm4 ; xmm1 ^= xmm4
; xmm1[63:32] == w[i+11] == RotWord(SubWord(w[i+5]))^Rcon^w[i+5]^w[i+4]^w[i+3]^w[i+2]^w[i+1]^w[i]
; xmm1[31:0] == w[i+10] == RotWord(SubWord(w[i+5]))^Rcon^w[i+4]^w[i+3]^w[i+2]^w[i+1]^w[i]
ret
@aes_AES192_expand_key_@36 endp
@aes_AES192_derive_decryption_keys_@8 proc
movdqa xmm5, [ecx]
movdqa xmm4, [ecx + 0C0h]
movdqa [edx], xmm4
movdqa [edx + 0C0h], xmm5
aesimc xmm5, [ecx + 10h]
aesimc xmm4, [ecx + 0B0h]
movdqa [edx + 10h], xmm4
movdqa [edx + 0B0h], xmm5
aesimc xmm5, [ecx + 20h]
aesimc xmm4, [ecx + 0A0h]
movdqa [edx + 20h], xmm4
movdqa [edx + 0A0h], xmm5
aesimc xmm5, [ecx + 30h]
aesimc xmm4, [ecx + 90h]
movdqa [edx + 30h], xmm4
movdqa [edx + 90h], xmm5
aesimc xmm5, [ecx + 40h]
aesimc xmm4, [ecx + 80h]
movdqa [edx + 40h], xmm4
movdqa [edx + 80h], xmm5
aesimc xmm5, [ecx + 50h]
aesimc xmm4, [ecx + 70h]
movdqa [edx + 50h], xmm4
movdqa [edx + 70h], xmm5
aesimc xmm5, [ecx + 60h]
movdqa [edx + 60h], xmm5
ret
@aes_AES192_derive_decryption_keys_@8 endp
end