Merge pull request #1870 from shakespark/fix-sm2-neon-support

fix(sm2): fix ENABLE_SM2_NEON macro mismatch and add missing functions
This commit is contained in:
Zhi Guan
2026-06-21 09:58:09 +08:00
committed by GitHub
2 changed files with 23 additions and 1 deletions

View File

@@ -260,7 +260,7 @@ endif()
if (ENABLE_SM2_NEON)
message(STATUS "ENABLE_SM2_NEON is ON")
add_definitions(-DENABLE_SM2_NEON)
add_definitions(-DENABLE_SM2_Z256_NEON)
endif()
if (ENABLE_SM9_ARM64)

View File

@@ -596,6 +596,28 @@ void sm2_z256_modp_mont_mul(sm2_z256_t r, const sm2_z256_t a, const sm2_z256_t b
}
}
void sm2_z256_modp_mont_sqr(sm2_z256_t r, const sm2_z256_t a)
{
sm2_z256_modp_mont_mul(r, a, a);
}
// mont(mont(a), 1) = aR * 1 * R^-1 (mod p) = a (mod p)
void sm2_z256_modp_from_mont(sm2_z256_t r, const sm2_z256_t a)
{
sm2_z256_modp_mont_mul(r, a, SM2_Z256_ONE);
}
// 2^512 (mod p)
static const uint64_t SM2_Z256_2e512modp_neon[4] = {
0x0000000200000003, 0x00000002ffffffff, 0x0000000100000001, 0x0000000400000002
};
// mont(a) = a * 2^256 (mod p) = mont_mul(a, 2^512 mod p)
void sm2_z256_modp_to_mont(const sm2_z256_t a, uint64_t r[4])
{
sm2_z256_modp_mont_mul(r, a, SM2_Z256_2e512modp_neon);
}
#else // ENABLE_SM2_Z256_NEON
// z = a*b