mirror/gmsm
1
0
Fork 0
mirror of https://github.com/emmansun/gmsm.git synced 2025-04-22 10:16:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

sm9/bn256: ppc64x gfp sqr

Browse Source
This commit is contained in:
Sun Yimin 2024-10-05 17:17:51 +08:00 committed by GitHub
parent 9477b78fec
commit ed4eafdc3e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 113 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
sm9/bn256/gfp_ppc64x.go
View File

@ -36,3 +36,18 @@ func gfpSubAsm(c, a, b *gfP)
//
//go:noescape
func gfpMulAsm(c, a, b *gfP)
// Montgomery square, repeated n times (n >= 1).
//
//go:noescape
func gfpSqrAsm(res, in *gfP, n int)
// Marshal gfP into big endian form
//
//go:noescape
func gfpMarshalAsm(out *[32]byte, in *gfP)
// Unmarshal the bytes into little endian form
//
//go:noescape
func gfpUnmarshalAsm(out *gfP, in *[32]byte)

91
sm9/bn256/gfp_ppc64x.s
View File

@ -6,6 +6,36 @@
#include "textflag.h"
//func gfpUnmarshal(out *gfP, in *[32]byte)
TEXT ·gfpUnmarshalAsm(SB), NOSPLIT, $0-16
MOVD res+0(FP), R3
MOVD in+8(FP), R4
BR gfpInternalEndianSwap<>(SB)
// func gfpMarshal(out *[32]byte, in *gfP)
TEXT ·gfpMarshalAsm(SB), NOSPLIT, $0-16
MOVD res+0(FP), R3
MOVD in+8(FP), R4
BR gfpInternalEndianSwap<>(SB)
TEXT gfpInternalEndianSwap<>(SB), NOSPLIT, $0-0
// Index registers needed for BR movs
MOVD $8, R9
MOVD $16, R10
MOVD $24, R14
MOVDBR (R0)(R4), R5
MOVDBR (R9)(R4), R6
MOVDBR (R10)(R4), R7
MOVDBR (R14)(R4), R8
MOVD R8, 0(R3)
MOVD R7, 8(R3)
MOVD R6, 16(R3)
MOVD R5, 24(R3)
RET
#define X1L V0
#define X1H V1
#define Y1L V2
@ -734,3 +764,64 @@ TEXT ·gfpMulAsm(SB),NOSPLIT,$0
STXVD2X T1, (R16)(res_ptr)
RET
// func gfpSqr(res, in *gfP, n int)
TEXT ·gfpSqrAsm(SB),NOSPLIT,$0
MOVD res+0(FP), res_ptr
MOVD in+8(FP), x_ptr
MOVD n+16(FP), N
MOVD $16, R16
LXVD2X (R0)(x_ptr), X0
LXVD2X (R16)(x_ptr), X1
XXPERMDI X0, X0, $2, X0
XXPERMDI X1, X1, $2, X1
MOVD $·p2+0(SB), CPOOL
LXVD2X (CPOOL)(R0), M0
LXVD2X (CPOOL)(R16), M1
XXPERMDI M0, M0, $2, M0
XXPERMDI M1, M1, $2, M1
MOVD $·np+0(SB), CPOOL
LXVD2X (CPOOL)(R0), K0
VSPLTW $1, K0, K0
sqrLoop:
// Sqr uses same value for both
VOR X0, X0, Y0
VOR X1, X1, Y1
CALL gfpMulInternal<>(SB)
ADD $-1, N
CMP $0, N
BEQ done
VOR T0, T0, X0
VOR T1, T1, X1
BR sqrLoop
done:
XXPERMDI T0, T0, $2, T0
XXPERMDI T1, T1, $2, T1
STXVD2X T0, (R0)(res_ptr)
STXVD2X T1, (R16)(res_ptr)
RET
#undef res_ptr
#undef x_ptr
#undef y_ptr
#undef CPOOL
#undef N
#undef X0
#undef X1
#undef Y0
#undef Y1
#undef M0
#undef M1
#undef T0
#undef T1
#undef K0

15
sm9/bn256/gfp_ppc64x_test.go
View File

@ -32,7 +32,7 @@ func TestGfpAsmBasicOperations(t *testing.T) {
expectedSub := fromBigInt(bigFromHex("67b381821c52a5624f3304a8149be8461e3bc07adcb872c38aa65051ba53ba97"))
expectedNeg := fromBigInt(bigFromHex("7f1d8aad70909be90358f1d02240062433cc3a0248ded72febb879ec33ce6f22"))
expectedMul := fromBigInt(bigFromHex("3d08bbad376584e4f74bd31f78f716372b96ba8c3f939c12b8d54e79b6489e76"))
//expectedMul2 := fromBigInt(bigFromHex("1df94a9e05a559ff38e0ab50cece734dc058d33738ceacaa15986a67cbff1ef6"))
expectedMul2 := fromBigInt(bigFromHex("1df94a9e05a559ff38e0ab50cece734dc058d33738ceacaa15986a67cbff1ef6"))
t.Parallel()
t.Run("add", func(t *testing.T) {
@ -116,24 +116,23 @@ func TestGfpAsmBasicOperations(t *testing.T) {
t.Errorf("mul not same when mul self")
}
})
/*
t.Run("square", func(t *testing.T) {
ret, ret1, ret2 := &gfP{}, &gfP{}, &gfP{}
gfpMul(ret, x, y)
gfpMul(ret1, ret, ret)
gfpMulAsm(ret, x, y)
gfpMulAsm(ret1, ret, ret)
if *ret1 != *expectedMul2 {
t.Errorf("mul not same")
}
gfpMul(ret1, ret1, ret1)
gfpSqr(ret2, ret, 2)
gfpMulAsm(ret1, ret1, ret1)
gfpSqrAsm(ret2, ret, 2)
if *ret1 != *ret2 {
t.Errorf("mul/sqr not same")
}
ret2.Set(ret)
gfpSqr(ret2, ret2, 2)
gfpSqrAsm(ret2, ret2, 2)
if *ret1 != *ret2 {
t.Errorf("mul/sqr not same when square self")
}
})
*/
}