[Feature]add MT2731_MP2_MR2_SVN388 baseline version
Change-Id: Ief04314834b31e27effab435d3ca8ba33b499059
diff --git a/src/bsp/lk/lib/aes/test/aes_test.c b/src/bsp/lk/lib/aes/test/aes_test.c
new file mode 100644
index 0000000..c267c48
--- /dev/null
+++ b/src/bsp/lk/lib/aes/test/aes_test.c
@@ -0,0 +1,96 @@
+/*
+ * Unit test for HW AES encryption.
+ */
+
+#include <lib/aes.h>
+
+#include <stdint.h>
+#include <string.h>
+#include <platform.h>
+#include <debug.h>
+#include <trace.h>
+#include <arch/ops.h>
+#include <lib/console.h>
+
+/*
+ * These sample values come from publication "FIPS-197", Appendix C.1
+ * "AES-128" See
+ * http://csrc.nist.gov/publications/fips/fips197/fips-197.pdf
+ */
+static const uint8_t plaintext[] = {
+ 0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77,
+ 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff
+};
+
+static const uint8_t key[] = {
+ 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
+ 0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f
+};
+
+static const uint8_t expected_ciphertext[] = {
+ 0x69, 0xc4, 0xe0, 0xd8, 0x6a, 0x7b, 0x04, 0x30,
+ 0xd8, 0xcd, 0xb7, 0x80, 0x70, 0xb4, 0xc5, 0x5a
+};
+
+static int aes_command(int argc, const cmd_args *argv)
+{
+ AES_KEY aes_key;
+ uint8_t ciphertext[AES_BLOCK_SIZE];
+
+ TRACEF("Testing AES encryption.\n");
+ memset(ciphertext, 0, sizeof(ciphertext));
+ AES_set_encrypt_key(key, 128, &aes_key);
+ lk_bigtime_t start = current_time_hires();
+ AES_encrypt(plaintext, ciphertext, &aes_key);
+ lk_bigtime_t end = current_time_hires();
+ int elapsed = end - start;
+ TRACEF("Elapsed time: %d us for 16 bytes (%d.%03d us per byte)\n",
+ elapsed, elapsed / AES_BLOCK_SIZE,
+ ((elapsed * 1000) / AES_BLOCK_SIZE) % 1000);
+ int not_equal = memcmp(expected_ciphertext, ciphertext, AES_BLOCK_SIZE);
+ if (not_equal) {
+ TRACEF("Encryption failed. Expected:\n");
+ hexdump8(expected_ciphertext, sizeof(expected_ciphertext));
+ TRACEF("Actual:\n");
+ hexdump8(ciphertext, sizeof(ciphertext));
+ TRACEF("FAILED AES encryption\n");
+ } else {
+ TRACEF("PASSED AES encryption\n");
+ }
+ return 0;
+}
+
+static int aes_bench(int argc, const cmd_args *argv)
+{
+ uint32_t c;
+ int i;
+ AES_KEY aes_key;
+ uint8_t ciphertext[AES_BLOCK_SIZE];
+#define ITER 1000
+
+ memset(ciphertext, 0, sizeof(ciphertext));
+
+ c = arch_cycle_count();
+
+ for (i = 0; i < ITER; i++) {
+ AES_set_encrypt_key(key, 128, &aes_key);
+ }
+
+ c = arch_cycle_count() - c;
+ printf("%u cycles to set encryption key\n", c / ITER);
+
+ c = arch_cycle_count();
+ for (i = 0; i < ITER; i++) {
+ AES_encrypt(plaintext, ciphertext, &aes_key);
+ }
+ c = arch_cycle_count() - c;
+
+ printf("%u cycles to encrypt block of 16 bytes\n", c / ITER);
+
+ return 0;
+}
+
+STATIC_COMMAND_START
+STATIC_COMMAND("aes_test", "test AES encryption", &aes_command)
+STATIC_COMMAND("aes_bench", "bench AES encryption", &aes_bench)
+STATIC_COMMAND_END(aes_test);