From a0874f58b67af45c9c219db7195bcab0f479cfee Mon Sep 17 00:00:00 2001
From: ty <zonyitoo@users.noreply.github.com>
Date: Fri, 27 Oct 2023 01:15:15 +0800
Subject: [PATCH] cpufeatures: add support for aarch64's sm4 target feature
 (#972)

---
 cpufeatures/src/aarch64.rs   | 2 ++
 cpufeatures/src/lib.rs       | 4 ++++
 cpufeatures/tests/aarch64.rs | 2 +-
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/cpufeatures/src/aarch64.rs b/cpufeatures/src/aarch64.rs
index 2c9a044b..2305d898 100644
--- a/cpufeatures/src/aarch64.rs
+++ b/cpufeatures/src/aarch64.rs
@@ -69,6 +69,7 @@ __expand_check_macro! {
     ("aes",    AES),    // Enable AES support.
     ("sha2",   SHA2),   // Enable SHA1 and SHA256 support.
     ("sha3",   SHA3),   // Enable SHA512 and SHA3 support.
+    ("sm4",    SM4),    // Enable SM3 and SM4 support.
 }
 
 /// Linux hardware capabilities mapped to target features.
@@ -85,6 +86,7 @@ pub mod hwcaps {
     pub const AES: c_ulong = libc::HWCAP_AES | libc::HWCAP_PMULL;
     pub const SHA2: c_ulong = libc::HWCAP_SHA2;
     pub const SHA3: c_ulong = libc::HWCAP_SHA3 | libc::HWCAP_SHA512;
+    pub const SM4: c_ulong = libc::HWCAP_SM3 | libc::HWCAP_SM4;
 }
 
 // Apple OS (macOS, iOS, watchOS, and tvOS) `check!` macro.
diff --git a/cpufeatures/src/lib.rs b/cpufeatures/src/lib.rs
index 4c916f45..ba9aa25d 100644
--- a/cpufeatures/src/lib.rs
+++ b/cpufeatures/src/lib.rs
@@ -18,6 +18,10 @@
 //! - `sha2`*
 //! - `sha3`*
 //!
+//! Linux only
+//!
+//! - `sm4`*
+//!
 //! ## `loongarch64`
 //!
 //! Linux only (LoongArch64 does not support OS-independent feature detection)
diff --git a/cpufeatures/tests/aarch64.rs b/cpufeatures/tests/aarch64.rs
index 949669be..41a61233 100644
--- a/cpufeatures/tests/aarch64.rs
+++ b/cpufeatures/tests/aarch64.rs
@@ -2,7 +2,7 @@
 
 #![cfg(target_arch = "aarch64")]
 
-cpufeatures::new!(armcaps, "aes", "sha2", "sha3");
+cpufeatures::new!(armcaps, "aes", "sha2", "sha3", "sm4");
 
 #[test]
 fn init() {