go-aes/aesni_amd64.s at main · jedisct1/go-aes · GitHub

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
// AES-NI hardware acceleration for AMD64 (Intel/AMD x86-64)
#include "textflag.h"

// func aesniRound(block *Block, roundKey *Block)
TEXT ·aesniRound(SB),NOSPLIT,$0
	MOVQ block+0(FP), AX
	MOVQ roundKey+8(FP), BX

	// Load the block into XMM0
	MOVOU (AX), X0

	// Load the round key into XMM1
	MOVOU (BX), X1

	// Perform AES round (AESENC = ShiftRows + SubBytes + MixColumns + XOR with key)
	AESENC X1, X0

	// Store result back to block
	MOVOU X0, (AX)
	RET

// func aesniFinalRound(block *Block, roundKey *Block)
TEXT ·aesniFinalRound(SB),NOSPLIT,$0
	MOVQ block+0(FP), AX
	MOVQ roundKey+8(FP), BX

	// Load the block into XMM0
	MOVOU (AX), X0

	// Load the round key into XMM1
	MOVOU (BX), X1

	// Perform final AES round (AESENCLAST = ShiftRows + SubBytes + XOR with key, no MixColumns)
	AESENCLAST X1, X0

	// Store result back to block
	MOVOU X0, (AX)
	RET

// func aesniInvRound(block *Block, roundKey *Block)
TEXT ·aesniInvRound(SB),NOSPLIT,$0
	MOVQ block+0(FP), AX
	MOVQ roundKey+8(FP), BX

	// Load the block into XMM0
	MOVOU (AX), X0

	// Load the round key into XMM1
	MOVOU (BX), X1

	// Perform inverse AES round (AESDEC = InvShiftRows + InvSubBytes + InvMixColumns + XOR with key)
	AESDEC X1, X0

	// Store result back to block
	MOVOU X0, (AX)
	RET

// func aesniInvFinalRound(block *Block, roundKey *Block)
TEXT ·aesniInvFinalRound(SB),NOSPLIT,$0
	MOVQ block+0(FP), AX
	MOVQ roundKey+8(FP), BX

	// Load the block into XMM0
	MOVOU (AX), X0

	// Load the round key into XMM1
	MOVOU (BX), X1

	// Perform final inverse AES round (AESDECLAST = InvShiftRows + InvSubBytes + XOR with key, no InvMixColumns)
	AESDECLAST X1, X0

	// Store result back to block
	MOVOU X0, (AX)
	RET

// func aesniInvMixColumns(block *Block)
TEXT ·aesniInvMixColumns(SB),NOSPLIT,$0
	MOVQ block+0(FP), AX

	// Load the block into XMM0
	MOVOU (AX), X0

	// Perform InvMixColumns transformation
	AESIMC X0, X0

	// Store result back to block
	MOVOU X0, (AX)
	RET