sm4: ppc64x, encryptBlockAsm #249

This commit is contained in:
Sun Yimin 2024-09-12 08:14:06 +08:00 committed by GitHub
parent c279984922
commit 8de52628d2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 76 additions and 4 deletions

View File

@ -207,10 +207,65 @@ ksLoop:
RET RET
// func encryptBlockAsm(xk *uint32, dst, src *byte, inst int)
TEXT ·encryptBlockAsm(SB),NOSPLIT,$0
// prepare/load constants
VSPLTISW $4, V_FOUR;
#ifdef NEEDS_PERMW
MOVD $·rcon(SB), R4
LVX (R4), ESPERMW
#endif
MOVD $·rcon+0x50(SB), R4
LXVD2X (R4)(R0), REVERSE_WORDS
MOVD $16, R3
LXVD2X (R4)(R3), NIBBLE_MASK
MOVD $32, R3
LXVD2X (R4)(R3), INVERSE_SHIFT_ROWS
MOVD $48, R3
LXVD2X (R4)(R3), M1L
MOVD $64, R3
LXVD2X (R4)(R3), M1H
MOVD $80, R3
LXVD2X (R4)(R3), M2L
MOVD $96, R3
LXVD2X (R4)(R3), M2H
MOVD xk+0(FP), R3
MOVD dst+8(FP), R4
MOVD src+16(FP), R5
// load src
PPC64X_LXVW4X(R5, R0, V0)
VSLDOI $4, V0, V0, V1
VSLDOI $4, V1, V1, V2
VSLDOI $4, V2, V2, V3
// prepare counter
MOVD $8, R7
MOVD R7, CTR
encryptBlockLoop:
// load xk
LXVW4X (R3), V4
SM4_ROUND(V4, V5, V6, V7, V8, V0, V1, V2, V3)
VSLDOI $4, V4, V4, V4
SM4_ROUND(V4, V5, V6, V7, V8, V1, V2, V3, V0)
VSLDOI $4, V4, V4, V4
SM4_ROUND(V4, V5, V6, V7, V8, V2, V3, V0, V1)
VSLDOI $4, V4, V4, V4
SM4_ROUND(V4, V5, V6, V7, V8, V3, V0, V1, V2)
BDNZ encryptBlockLoop
VSLDOI $4, V3, V3, V3
VSLDOI $4, V3, V2, V2
VSLDOI $4, V2, V1, V1
VSLDOI $4, V1, V0, V0
PPC64X_STXVW4X(V0, R4, R0)
RET
// func encryptBlocksAsm(xk *uint32, dst, src []byte, inst int) // func encryptBlocksAsm(xk *uint32, dst, src []byte, inst int)
TEXT ·encryptBlocksAsm(SB),NOSPLIT,$0 TEXT ·encryptBlocksAsm(SB),NOSPLIT,$0
RET RET
// func encryptBlockAsm(xk *uint32, dst, src *byte, inst int)
TEXT ·encryptBlockAsm(SB),NOSPLIT,$0
RET

View File

@ -42,3 +42,20 @@ func TestExpandKey(t *testing.T) {
} }
} }
} }
func TestEncryptBlockAsm(t *testing.T) {
src := []byte{0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef, 0xfe, 0xdc, 0xba, 0x98, 0x76, 0x54, 0x32, 0x10}
expected := []byte{0x68, 0x1e, 0xdf, 0x34, 0xd2, 0x06, 0x96, 0x5e, 0x86, 0xb3, 0xe9, 0x4f, 0x53, 0x6e, 0x42, 0x46}
encRes2 := make([]uint32, 32)
decRes2 := make([]uint32, 32)
expandKeyAsm(&src[0], &ck[0], &encRes2[0], &decRes2[0], 0)
dst := make([]byte, 16)
encryptBlocksAsm(&encRes2[0], dst, src, 0)
if !reflect.DeepEqual(dst, expected) {
t.Errorf("expected=%x, result=%x\n", expected, dst)
}
encryptBlocksAsm(&decRes2[0], dst, expected, 0)
if !reflect.DeepEqual(dst, src) {
t.Errorf("expected=%x, result=%x\n", src, dst)
}
}