| Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1 | BEGIN_HEADER |
| 2 | #include <polarssl/arc4.h> |
| 3 | END_HEADER |
| 4 | |
| 5 | BEGIN_CASE |
| 6 | arc4_crypt:hex_src_string:hex_key_string:hex_dst_string |
| 7 | { |
| 8 | unsigned char src_str[1000]; |
| 9 | unsigned char key_str[1000]; |
| Paul Bakker | baad650 | 2010-03-21 15:42:15 +0000 | [diff] [blame] | 10 | unsigned char dst_str[1000]; |
| 11 | unsigned char dst_hexstr[2000]; |
| Paul Bakker | 69998dd | 2009-07-11 19:15:20 +0000 | [diff] [blame] | 12 | int src_len, key_len; |
| 13 | arc4_context ctx; |
| Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 14 | |
| 15 | memset(src_str, 0x00, 1000); |
| 16 | memset(key_str, 0x00, 1000); |
| Paul Bakker | baad650 | 2010-03-21 15:42:15 +0000 | [diff] [blame] | 17 | memset(dst_str, 0x00, 1000); |
| 18 | memset(dst_hexstr, 0x00, 2000); |
| Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 19 | |
| Paul Bakker | 69998dd | 2009-07-11 19:15:20 +0000 | [diff] [blame] | 20 | src_len = unhexify( src_str, {hex_src_string} ); |
| 21 | key_len = unhexify( key_str, {hex_key_string} ); |
| Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 22 | |
| Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 23 | arc4_setup(&ctx, key_str, key_len); |
| Paul Bakker | baad650 | 2010-03-21 15:42:15 +0000 | [diff] [blame] | 24 | TEST_ASSERT( arc4_crypt(&ctx, src_len, src_str, dst_str ) == 0 ); |
| 25 | hexify( dst_hexstr, dst_str, src_len ); |
| Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 26 | |
| Paul Bakker | baad650 | 2010-03-21 15:42:15 +0000 | [diff] [blame] | 27 | TEST_ASSERT( strcmp( (char *) dst_hexstr, {hex_dst_string} ) == 0 ); |
| Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 28 | } |
| 29 | END_CASE |
| Paul Bakker | 3d36082 | 2009-07-05 11:29:38 +0000 | [diff] [blame] | 30 | |
| 31 | BEGIN_CASE |
| 32 | arc4_selftest: |
| 33 | { |
| 34 | TEST_ASSERT( arc4_self_test( 0 ) == 0 ); |
| 35 | } |
| 36 | END_CASE |