Update SM4 AES-NI implementation

guanzhi · May 24, 2024 · cde0065 · cde0065
1 parent 4076a30
commit cde0065
Show file tree

Hide file tree

Showing 2 changed files with 346 additions and 140 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -28,6 +28,7 @@ option(ENABLE_GMUL_ARM64 "Enable GF(2^128) Multiplication AArch64 assembly" OFF)
 
 
 option(ENABLE_SM4_AVX2 "Enable SM4 AVX2 8x implementation" OFF)
+option(ENABLE_SM4_AESNI "Enable SM4 AES-NI (4x) implementation" OFF)
 option(ENABLE_SM2_AMD64 "Enable SM2_Z256 X86_64 assembly" OFF)
 
 
@@ -319,6 +320,14 @@ if (ENABLE_SM4_AVX2)
 	set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -march=native")
 endif()
 
+if (ENABLE_SM4_AESNI)
+	message(STATUS "ENABLE_SM4_AESNI is ON")
+	list(FIND src src/sm4.c sm4_index)
+	list(REMOVE_AT src ${sm4_index})
+	list(INSERT src ${sm4_index} src/sm4_aesni.c)
+	set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -march=native")
+endif()
+