Merge pull request #3 from zhaojuntao/aes-sample

[ADD] 增加 AES CBC 例程

SConscript

@@ -6,6 +6,9 @@ cwd = GetCurrentDir()
 src = Glob('src/*.c') + Glob('src/*.cpp')
 CPPPATH = [cwd + '/include']
 
+if GetDepend(['TINYCRYPT_USING_AES_SAMPLE']):
+    src += Glob('samples/*.c')
+
 group = DefineGroup('TinyCrypt', src, depend = ['PKG_USING_TINYCRYPT'], CPPPATH = CPPPATH)
 
 Return('group')

samples/aes_sample.c

@@ -0,0 +1,61 @@
+/*
+ * Copyright (c) 2006-2019, RT-Thread Development Team
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Change Logs:
+ * Date           Author       Notes
+ * 2019-03-22     MurphyZhao   the first version
+ */
+
+#include <rtthread.h>
+
+#include <tiny_aes.h>
+
+#define TEST_TINY_AES_IV  "0123456789ABCDEF"
+#define TEST_TINY_AES_KEY "0123456789ABCDEF0123456789ABCDEF"
+
+static rt_err_t test_tiny_aes(void)
+{
+    tiny_aes_context ctx;
+    uint8_t iv[16 + 1];
+    uint8_t private_key[32 + 1];
+
+    unsigned char data[] = "1234567890123456";
+    unsigned char data_encrypt[32];
+    unsigned char data_decrypt[32];
+
+    /* encrypt */
+    rt_memcpy(iv, TEST_TINY_AES_IV, rt_strlen(TEST_TINY_AES_IV));
+    iv[sizeof(iv) - 1] = '\0';
+    rt_memcpy(private_key, TEST_TINY_AES_KEY, rt_strlen(TEST_TINY_AES_KEY));
+    private_key[sizeof(private_key) - 1] = '\0';
+
+    rt_memset(data_encrypt, 0x0, sizeof(data_encrypt));
+    tiny_aes_setkey_enc(&ctx, (uint8_t *) private_key, 256);
+    tiny_aes_crypt_cbc(&ctx, AES_ENCRYPT, rt_strlen(data), iv, data, data_encrypt);
+
+    /* decrypt */
+    rt_memcpy(iv, TEST_TINY_AES_IV, rt_strlen(TEST_TINY_AES_IV));
+    iv[sizeof(iv) - 1] = '\0';
+    rt_memcpy(private_key, TEST_TINY_AES_KEY, rt_strlen(TEST_TINY_AES_KEY));
+    private_key[sizeof(private_key) - 1] = '\0';
+
+    rt_memset(data_decrypt, 0x0, sizeof(data_decrypt));
+    tiny_aes_setkey_dec(&ctx, (uint8_t *) private_key, 256);
+    tiny_aes_crypt_cbc(&ctx, AES_DECRYPT, rt_strlen(data), iv, data_encrypt, data_decrypt);
+
+    if(rt_memcmp(data, data_decrypt, rt_strlen(data)) == 0)
+    {
+        rt_kprintf("AES CBC mode passed!\n");
+        return RT_EOK;
+    }
+    else
+    {
+        rt_kprintf("AES CBC mode failed!");
+        return -RT_ERROR;
+    }
+}
+#ifdef FINSH_USING_MSH
+MSH_CMD_EXPORT_ALIAS(test_tiny_aes, tiny_aes_sample, type tiny_aes_sample cmd to run);
+#endif /* FINSH_USING_MSH */