initial commit

1 files changed, 181 insertions, 0 deletions

diff --git a/src/aes256ecb.asm b/src/aes256ecb.asm
new file mode 100644
index 0000000..4246dee
--- /dev/null
+++ b/src/aes256ecb.asm
@@ -0,0 +1,181 @@
+; Copyright 2015 Egor Tensin <Egor.Tensin@gmail.com>
+; This file is licensed under the terms of the MIT License.
+; See LICENSE.txt for details.
+
+.586
+.xmm
+.model flat
+
+.data
+
+align 10h
+key_schedule oword 15 dup(0)
+
+align 10h
+inverse_key_schedule oword 15 dup(0)
+
+.code
+
+@aes256ecb_encrypt@48 proc
+    call expand_keys_256ecb
+    pxor xmm0, [key_schedule]
+    aesenc xmm0, [key_schedule + 10h]
+    aesenc xmm0, [key_schedule + 20h]
+    aesenc xmm0, [key_schedule + 30h]
+    aesenc xmm0, [key_schedule + 40h]
+    aesenc xmm0, [key_schedule + 50h]
+    aesenc xmm0, [key_schedule + 60h]
+    aesenc xmm0, [key_schedule + 70h]
+    aesenc xmm0, [key_schedule + 80h]
+    aesenc xmm0, [key_schedule + 90h]
+    aesenc xmm0, [key_schedule + 0A0h]
+    aesenc xmm0, [key_schedule + 0B0h]
+    aesenc xmm0, [key_schedule + 0C0h]
+    aesenc xmm0, [key_schedule + 0D0h]
+    aesenclast xmm0, [key_schedule + 0E0h]
+    ret
+@aes256ecb_encrypt@48 endp
+
+expand_keys_256ecb proc
+    lea edx, [key_schedule + 20h]
+    movdqa [key_schedule], xmm1
+    movdqa [key_schedule + 10h], xmm2
+
+    aeskeygenassist xmm7, xmm2, 1h
+    pshufd xmm7, xmm7, 0FFh
+    call gen_round_key
+
+    aeskeygenassist xmm7, xmm2, 0
+    pshufd xmm7, xmm7, 0AAh
+    call gen_round_key
+
+    aeskeygenassist xmm7, xmm2, 2h
+    pshufd xmm7, xmm7, 0FFh
+    call gen_round_key
+
+    aeskeygenassist xmm7, xmm2, 0
+    pshufd xmm7, xmm7, 0AAh
+    call gen_round_key
+
+    aeskeygenassist xmm7, xmm2, 4h
+    pshufd xmm7, xmm7, 0FFh
+    call gen_round_key
+
+    aeskeygenassist xmm7, xmm2, 0
+    pshufd xmm7, xmm7, 0AAh
+    call gen_round_key
+
+    aeskeygenassist xmm7, xmm2, 8h
+    pshufd xmm7, xmm7, 0FFh
+    call gen_round_key
+
+    aeskeygenassist xmm7, xmm2, 0
+    pshufd xmm7, xmm7, 0AAh
+    call gen_round_key
+
+    aeskeygenassist xmm7, xmm2, 10h
+    pshufd xmm7, xmm7, 0FFh
+    call gen_round_key
+
+    aeskeygenassist xmm7, xmm2, 0
+    pshufd xmm7, xmm7, 0AAh
+    call gen_round_key
+
+    aeskeygenassist xmm7, xmm2, 20h
+    pshufd xmm7, xmm7, 0FFh
+    call gen_round_key
+
+    aeskeygenassist xmm7, xmm2, 0
+    pshufd xmm7, xmm7, 0AAh
+    call gen_round_key
+
+    aeskeygenassist xmm7, xmm2, 40h
+    pshufd xmm7, xmm7, 0FFh
+    call gen_round_key
+
+    call invert_key_schedule
+    ret
+
+gen_round_key:
+    movdqa xmm6, xmm1
+
+    pslldq xmm6, 4
+    pxor xmm1, xmm6
+    pslldq xmm6, 4
+    pxor xmm1, xmm6
+    pslldq xmm6, 4
+    pxor xmm1, xmm6
+
+    pxor xmm1, xmm7
+
+    movdqa [edx], xmm1
+    add edx, 10h
+
+    movdqa xmm7, xmm1
+    movdqa xmm1, xmm2
+    movdqa xmm2, xmm7
+    ret
+
+invert_key_schedule:
+    movdqa xmm7, [key_schedule       ]
+    movdqa xmm6, [key_schedule + 0E0h]
+    movdqa [inverse_key_schedule       ], xmm6
+    movdqa [inverse_key_schedule + 0E0h], xmm7
+
+    aesimc xmm7, [key_schedule +  10h]
+    aesimc xmm6, [key_schedule + 0D0h]
+    movdqa [inverse_key_schedule +  10h], xmm6
+    movdqa [inverse_key_schedule + 0D0h], xmm7
+
+    aesimc xmm7, [key_schedule +  20h]
+    aesimc xmm6, [key_schedule + 0C0h]
+    movdqa [inverse_key_schedule +  20h], xmm6
+    movdqa [inverse_key_schedule + 0C0h], xmm7
+
+    aesimc xmm7, [key_schedule +  30h]
+    aesimc xmm6, [key_schedule + 0B0h]
+    movdqa [inverse_key_schedule +  30h], xmm6
+    movdqa [inverse_key_schedule + 0B0h], xmm7
+
+    aesimc xmm7, [key_schedule +  40h]
+    aesimc xmm6, [key_schedule + 0A0h]
+    movdqa [inverse_key_schedule +  40h], xmm6
+    movdqa [inverse_key_schedule + 0A0h], xmm7
+
+    aesimc xmm7, [key_schedule + 50h]
+    aesimc xmm6, [key_schedule + 90h]
+    movdqa [inverse_key_schedule + 50h], xmm6
+    movdqa [inverse_key_schedule + 90h], xmm7
+
+    aesimc xmm7, [key_schedule + 60h]
+    aesimc xmm6, [key_schedule + 80h]
+    movdqa [inverse_key_schedule + 60h], xmm6
+    movdqa [inverse_key_schedule + 80h], xmm7
+
+    aesimc xmm7, [key_schedule + 70h]
+    movdqa [inverse_key_schedule + 70h], xmm7
+
+    ret
+expand_keys_256ecb endp
+
+@aes256ecb_decrypt@48 proc
+    call expand_keys_256ecb
+    pxor xmm0, [inverse_key_schedule]
+    aesdec xmm0, [inverse_key_schedule + 10h]
+    aesdec xmm0, [inverse_key_schedule + 20h]
+    aesdec xmm0, [inverse_key_schedule + 30h]
+    aesdec xmm0, [inverse_key_schedule + 40h]
+    aesdec xmm0, [inverse_key_schedule + 50h]
+    aesdec xmm0, [inverse_key_schedule + 60h]
+    aesdec xmm0, [inverse_key_schedule + 70h]
+    aesdec xmm0, [inverse_key_schedule + 80h]
+    aesdec xmm0, [inverse_key_schedule + 90h]
+    aesdec xmm0, [inverse_key_schedule + 0A0h]
+    aesdec xmm0, [inverse_key_schedule + 0B0h]
+    aesdec xmm0, [inverse_key_schedule + 0C0h]
+    aesdec xmm0, [inverse_key_schedule + 0D0h]
+    aesdeclast xmm0, [inverse_key_schedule + 0E0h]    
+    ret
+@aes256ecb_decrypt@48 endp
+
+end