aboutsummaryrefslogblamecommitdiffstatshomepage
path: root/src/asm/aes192.asm
blob: 6a41243abd548069f66326ad838b3f5baf76b7ff (plain) (tree)
1
2
3
4
5
6
7
8
9



                                                            




           

     
                                  












                                 
       
                                  
 
                                  












                                 
       
                                  
 
                               




                                                                        




                                                                          
           
                   
                         


                                                
                   































                                                                          





                             
 

                                                                     
 
                                                                                   


                                    
                                                                                   


                                    
                                                                                   


                                    
                                                                                   


                                    
                                                                                   


                                    
                                                                                   


                                     
                                                                                   


                                     
                                                                                   

                             
 

       
                     
                    







                               
                                                      

                     







                                                                                                         
                                           





                                    






                                       



                                                                    





                                                                            

                                                                                                   



                                                                                                                 
 


                        

                                                        

                                                             



                                                                                           


                                                                                                     

                                                                                                                                
 
       
                               
 
                                          































                             
 
       
                                          

   
; 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