Diff: tests/suites/test_suite_arc4.function

Revision:

0:cdf462088d13

diff -r 000000000000 -r cdf462088d13 tests/suites/test_suite_arc4.function
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/suites/test_suite_arc4.function	Thu Jan 05 00:18:44 2017 +0000
@@ -0,0 +1,46 @@
+/* BEGIN_HEADER */
+#include "mbedtls/arc4.h"
+/* END_HEADER */
+
+/* BEGIN_DEPENDENCIES
+ * depends_on:MBEDTLS_ARC4_C
+ * END_DEPENDENCIES
+ */
+
+/* BEGIN_CASE */
+void mbedtls_arc4_crypt( char *hex_src_string, char *hex_key_string,
+                 char *hex_dst_string )
+{
+    unsigned char src_str[1000];
+    unsigned char key_str[1000];
+    unsigned char dst_str[1000];
+    unsigned char dst_hexstr[2000];
+    int src_len, key_len;
+    mbedtls_arc4_context ctx;
+
+    memset(src_str, 0x00, 1000);
+    memset(key_str, 0x00, 1000);
+    memset(dst_str, 0x00, 1000);
+    memset(dst_hexstr, 0x00, 2000);
+    mbedtls_arc4_init( &ctx );
+
+    src_len = unhexify( src_str, hex_src_string );
+    key_len = unhexify( key_str, hex_key_string );
+
+    mbedtls_arc4_setup(&ctx, key_str, key_len);
+    TEST_ASSERT( mbedtls_arc4_crypt(&ctx, src_len, src_str, dst_str ) == 0 );
+    hexify( dst_hexstr, dst_str, src_len );
+
+    TEST_ASSERT( strcmp( (char *) dst_hexstr, hex_dst_string ) == 0 );
+
+exit:
+    mbedtls_arc4_free( &ctx );
+}
+/* END_CASE */
+
+/* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */
+void arc4_selftest()
+{
+    TEST_ASSERT( mbedtls_arc4_self_test( 1 ) == 0 );
+}
+/* END_CASE */