Add tests for _mm_aes*_si128 intrinsics

These tests are based on the examples in Microsoft's documentation. Same
input should result in the same output in any case.

Author: ruuda

coresimd/src/x86/x86_64/aes.rs

@@ -66,3 +66,74 @@ pub unsafe fn _mm_aesimc_si128(a: __m128i) -> __m128i {
 pub unsafe fn _mm_aeskeygenassist_si128(a: __m128i, imm8: u8) -> __m128i {
     aeskeygenassist(a, imm8)
 }
+
+#[cfg(test)]
+mod tests {
+    use stdsimd_test::simd_test;
+
+    use x86::*;
+
+    #[simd_test = "aex"]
+    unsafe fn test_mm_aesdec_si128() {
+        // Constants taken from https://msdn.microsoft.com/en-us/library/cc664949.aspx.
+        let a = __m128i(0x0123456789abcdef, 0x8899aabbccddeeff);
+        let b = __m128i(0x1133557799bbddff, 0x0022446688aaccee);
+        // TODO(ruuda): These two parts probably need to be swapped.
+        let e = __m128i(0xb57ecfa381da39ee, 0x044e4f5176fec48f);
+        let r = _mm_aesdec_si128(a, key);
+        assert_eq_m128i(r, e);
+    }
+
+    #[simd_test = "aex"]
+    unsafe fn test_mm_aesdeclast_si128() {
+        // Constants taken from https://msdn.microsoft.com/en-us/library/cc714178.aspx.
+        let a = __m128i(0x0123456789abcdef, 0x8899aabbccddeeff);
+        let b = __m128i(0x1133557799bbddff, 0x0022446688aaccee);
+        // TODO(ruuda): These two parts probably need to be swapped.
+        let e = __m128i(0xf210dd981fa4a493, 0x36cad57d9072bf9e);
+        let r = _mm_aesdeclast_si128(a, key);
+        assert_eq_m128i(r, e);
+    }
+
+    #[simd_test = "aex"]
+    unsafe fn test_mm_aesenc_si128() {
+        // Constants taken from https://msdn.microsoft.com/en-us/library/cc664810.aspx.
+        let a = __m128i(0x0123456789abcdef, 0x8899aabbccddeeff);
+        let b = __m128i(0x1133557799bbddff, 0x0022446688aaccee);
+        // TODO(ruuda): These two parts probably need to be swapped.
+        let e = __m128i(0x28e4ee1884504333, 0x16ab0e57dfc442ed);
+        let r = _mm_aesenc_si128(a, key);
+        assert_eq_m128i(r, e);
+    }
+
+    #[simd_test = "aex"]
+    unsafe fn test_mm_aesenclast_si128() {
+        // Constants taken from https://msdn.microsoft.com/en-us/library/cc714136.aspx.
+        let a = __m128i(0x0123456789abcdef, 0x8899aabbccddeeff);
+        let b = __m128i(0x1133557799bbddff, 0x0022446688aaccee);
+        // TODO(ruuda): These two parts probably need to be swapped.
+        let e = __m128i(0x4b04f98cf4c860f8, 0xb6dd7df25d7ab320);
+        let r = _mm_aesenclast_si128(a, key);
+        assert_eq_m128i(r, e);
+    }
+
+    #[simd_test = "aex"]
+    unsafe fn test_mm_aesimc_si128() {
+        // Constants taken from https://msdn.microsoft.com/en-us/library/cc714195.aspx.
+        let a = __m128i(0x0123456789abcdef, 0x8899aabbccddeeff);
+        // TODO(ruuda): These two parts probably need to be swapped.
+        let e = __m128i(0x6633441122770055, 0xc66c82284ee40aa0);
+        let r = _mm_aesimc_si128(a);
+        assert_eq_m128i(r, e);
+    }
+
+    #[simd_test = "aex"]
+    unsafe fn test_mm_aeskeygenassist_si128() {
+        // Constants taken from https://msdn.microsoft.com/en-us/library/cc714138.aspx.
+        let a = __m128i(0x0123456789abcdef, 0x8899aabbccddeeff);
+        // TODO(ruuda): These two parts probably need to be swapped.
+        let e = __m128i(0xeac4eea9c4eeacea, 0x857c266b7c266e85);
+        let r = _mm_aeskeygenassist_si128(a, 5);
+        assert_eq_m128i(r, e);
+    }
+}