aes: autodetection support for AES-NI

Adds an off-by-default `autodetect` feature which the `cpuid-bool` crate
to detect whether AES-NI is available when compiling on i686/x86_64
architectures.

Author: tarcieri

.github/workflows/aes.yml

@@ -71,6 +71,7 @@ jobs:
       - run: ${{ matrix.deps }}
       - run: cargo check --target ${{ matrix.target }} --all-features
       - run: cargo test --release --target ${{ matrix.target }}
+      - run: cargo test --release --target ${{ matrix.target }} --features autodetect
       - run: cargo test --release --target ${{ matrix.target }} --features compact
       - run: cargo test --release --target ${{ matrix.target }} --features ctr
       - run: cargo test --release --target ${{ matrix.target }} --all-features
@@ -109,6 +110,7 @@ jobs:
       - run: ${{ matrix.deps }}
       - run: cargo check --target ${{ matrix.target }} --all-features
       - run: cargo test --release --target ${{ matrix.target }}
+      - run: cargo test --release --target ${{ matrix.target }} --features autodetect
       - run: cargo test --release --target ${{ matrix.target }} --features compact
       - run: cargo test --release --target ${{ matrix.target }} --features ctr
       - run: cargo test --release --target ${{ matrix.target }} --all-features

Cargo.lock

@@ -17,13 +17,15 @@ categories = ["cryptography", "no-std"]
 [dependencies]
 cfg-if = "1"
 cipher = "=0.3.0-pre"
+cpuid-bool = { version = "0.1", optional = true }
 ctr = { version = "=0.7.0-pre", optional = true }
 opaque-debug = "0.3"
 
 [dev-dependencies]
 cipher = { version = "=0.3.0-pre", features = ["dev"] }
 
 [features]
+autodetect = ["cpuid-bool"] # Detect hardware AES and fallback to "soft" if unavailable
 compact = [] # Reduce code size at the cost of performance
 
 [package.metadata.docs.rs]

aes/Cargo.toml

@@ -0,0 +1,107 @@
+//! Autodetection support for hardware accelerated AES backends with fallback
+//! to the fixsliced "soft" implementation.
+
+use crate::{Block, ParBlocks};
+use cipher::{
+    consts::{U16, U24, U32, U8},
+    generic_array::GenericArray,
+    BlockCipher, BlockDecrypt, BlockEncrypt, NewBlockCipher,
+};
+
+macro_rules! define_aes_impl {
+    (
+        $name:tt,
+        $module:tt,
+        $key_size:ty,
+        $doc:expr
+    ) => {
+        #[doc=$doc]
+        #[derive(Clone)]
+        pub struct $name($module::Inner);
+
+        mod $module {
+            #[derive(Clone)]
+            pub(super) enum Inner {
+                Ni(crate::ni::$name),
+                Soft(crate::soft::$name),
+            }
+        }
+
+        impl NewBlockCipher for $name {
+            type KeySize = $key_size;
+
+            #[inline]
+            fn new(key: &GenericArray<u8, $key_size>) -> Self {
+                if cpuid_bool::cpuid_bool!("aes") {
+                    $name($module::Inner::Ni(crate::ni::$name::new(key)))
+                } else {
+                    $name($module::Inner::Soft(crate::soft::$name::new(key)))
+                }
+            }
+        }
+
+        impl BlockCipher for $name {
+            type BlockSize = U16;
+            type ParBlocks = U8;
+        }
+
+        impl BlockEncrypt for $name {
+            #[inline]
+            fn encrypt_block(&self, block: &mut Block) {
+                match &self.0 {
+                    $module::Inner::Ni(aes) => aes.encrypt_block(block),
+                    $module::Inner::Soft(aes) => aes.encrypt_block(block),
+                }
+            }
+
+            #[inline]
+            fn encrypt_par_blocks(&self, blocks: &mut ParBlocks) {
+                match &self.0 {
+                    $module::Inner::Ni(aes) => aes.encrypt_par_blocks(blocks),
+                    $module::Inner::Soft(aes) => aes.encrypt_par_blocks(blocks),
+                }
+            }
+        }
+
+        impl BlockDecrypt for $name {
+            #[inline]
+            fn decrypt_block(&self, block: &mut Block) {
+                match &self.0 {
+                    $module::Inner::Ni(aes) => aes.decrypt_block(block),
+                    $module::Inner::Soft(aes) => aes.decrypt_block(block),
+                }
+            }
+
+            #[inline]
+            fn decrypt_par_blocks(&self, blocks: &mut ParBlocks) {
+                match &self.0 {
+                    $module::Inner::Ni(aes) => aes.decrypt_par_blocks(blocks),
+                    $module::Inner::Soft(aes) => aes.decrypt_par_blocks(blocks),
+                }
+            }
+        }
+
+        opaque_debug::implement!($name);
+    }
+}
+
+define_aes_impl!(
+    Aes128,
+    aes128,
+    U16,
+    "AES-128 block cipher instance"
+);
+
+define_aes_impl!(
+    Aes192,
+    aes192,
+    U24,
+    "AES-192 block cipher instance"
+);
+
+define_aes_impl!(
+    Aes256,
+    aes256,
+    U32,
+    "AES-256 block cipher instance"
+);

aes/src/autodetect.rs

@@ -63,12 +63,14 @@ use cfg_if::cfg_if;
 
 cfg_if! {
     if #[cfg(all(
-        target_feature = "aes",
-        target_feature = "sse2",
         any(target_arch = "x86_64", target_arch = "x86"),
+        feature = "autodetect"
     ))] {
         mod ni;
-        pub use ni::{Aes128, Aes192, Aes256};
+        mod autodetect;
+        mod soft;
+
+        pub use autodetect::{Aes128, Aes192, Aes256};
 
         #[cfg(feature = "ctr")]
         cfg_if! {

aes/src/lib.rs

@@ -1,5 +1,7 @@
 //! AES in counter mode (a.k.a. AES-CTR)
 
+#![cfg_attr(feature = "autodetect", allow(dead_code))]
+
 use super::{Aes128, Aes192, Aes256};
 
 /// AES-128 in CTR mode