| 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]; |
| 10 | unsigned char dst_str[2000]; |
| Paul Bakker | 69998dd | 2009-07-11 19:15:20 +0000 | [diff] [blame^] | 11 | int src_len, key_len; |
| 12 | arc4_context ctx; |
| Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 13 | |
| 14 | memset(src_str, 0x00, 1000); |
| 15 | memset(key_str, 0x00, 1000); |
| 16 | memset(dst_str, 0x00, 2000); |
| 17 | |
| Paul Bakker | 69998dd | 2009-07-11 19:15:20 +0000 | [diff] [blame^] | 18 | src_len = unhexify( src_str, {hex_src_string} ); |
| 19 | key_len = unhexify( key_str, {hex_key_string} ); |
| Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 20 | |
| Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 21 | arc4_setup(&ctx, key_str, key_len); |
| 22 | arc4_crypt(&ctx, src_str, src_len); |
| 23 | hexify( dst_str, src_str, src_len ); |
| 24 | |
| 25 | TEST_ASSERT( strcmp( (char *) dst_str, {hex_dst_string} ) == 0 ); |
| 26 | } |
| 27 | END_CASE |
| Paul Bakker | 3d36082 | 2009-07-05 11:29:38 +0000 | [diff] [blame] | 28 | |
| 29 | BEGIN_CASE |
| 30 | arc4_selftest: |
| 31 | { |
| 32 | TEST_ASSERT( arc4_self_test( 0 ) == 0 ); |
| 33 | } |
| 34 | END_CASE |