cpuid not supported in sgx environment

Author: yihuang

sha2/Cargo.toml

@@ -21,7 +21,7 @@ opaque-debug = "0.3"
 cfg-if = "0.1"
 sha2-asm = { version = "0.5", optional = true }
 
-[target.'cfg(any(target_arch = "x86", target_arch = "x86_64"))'.dependencies]
+[target.'cfg(all(any(target_arch = "x86", target_arch = "x86_64"), not(target_env = "sgx")))'.dependencies]
 cpuid-bool = "0.1"
 
 [target.'cfg(all(target_arch = "aarch64", target_os = "linux"))'.dependencies]

sha2/src/sha256/x86.rs

@@ -95,6 +95,7 @@ unsafe fn digest_blocks(state: &mut [u32; 8], blocks: &[[u8; 64]]) {
     _mm_storeu_si128(state_ptr_mut.add(1), hgef);
 }
 
+#[cfg(feature = "cpuid_bool")]
 pub fn compress(state: &mut [u32; 8], blocks: &[[u8; 64]]) {
     // TODO: Replace with https://github.com/rust-lang/rfcs/pull/2725
     // after stabilization
@@ -106,3 +107,18 @@ pub fn compress(state: &mut [u32; 8], blocks: &[[u8; 64]]) {
         super::soft::compress(state, blocks);
     }
 }
+
+/// Just use `std::is_x86_feature_detected` in sgx environment
+#[cfg(not(feature = "cpuid_bool"))]
+pub fn compress(state: &mut [u32; 8], blocks: &[[u8; 64]]) {
+    if std::is_x86_feature_detected!("sse2")
+        || std::is_x86_feature_detected!("ssse3")
+        || std::is_x86_feature_detected!("sse4.1")
+    {
+        unsafe {
+            digest_blocks(state, blocks);
+        }
+    } else {
+        super::soft::compress(state, blocks);
+    }
+}