blob: 110eb8cef071a03d5b69ea77368ab8dc612630c0 [file] [log] [blame]
Paul Bakker0e04d0e2011-11-27 14:46:59 +00001BEGIN_HEADER
2#include <polarssl/ctr_drbg.h>
3
4int test_offset;
Paul Bakker69e095c2011-12-10 21:55:01 +00005int entropy_func( void *data, unsigned char *buf, size_t len )
Paul Bakker0e04d0e2011-11-27 14:46:59 +00006{
Paul Bakker69e095c2011-12-10 21:55:01 +00007 unsigned char *p = (unsigned char *) data;
Paul Bakker0e04d0e2011-11-27 14:46:59 +00008 memcpy( buf, p + test_offset, len );
9 test_offset += 32;
10 return( 0 );
11}
12END_HEADER
13
14BEGIN_DEPENDENCIES
15depends_on:POLARSSL_CTR_DRBG_C
16END_DEPENDENCIES
17
18BEGIN_CASE
19ctr_drbg_validate_pr:add_init_string:entropy_string:add1_string:add2_string:result_str
20{
21 unsigned char entropy[512];
22 unsigned char add_init[512];
23 unsigned char add1[512];
24 unsigned char add2[512];
25 ctr_drbg_context ctx;
26 unsigned char buf[512];
27 unsigned char output_str[512];
28 int add_init_len, add1_len, add2_len;
29
30 memset( output_str, 0, 512 );
31
32 unhexify( entropy, {entropy_string} );
33 add_init_len = unhexify( add_init, {add_init_string} );
34 add1_len = unhexify( add1, {add1_string} );
35 add2_len = unhexify( add2, {add2_string} );
36
37 test_offset = 0;
Paul Bakker18d32912011-12-10 21:42:49 +000038 TEST_ASSERT( ctr_drbg_init_entropy_len( &ctx, entropy_func, entropy, add_init, add_init_len, 32 ) == 0 );
Paul Bakker0e04d0e2011-11-27 14:46:59 +000039 ctr_drbg_set_prediction_resistance( &ctx, CTR_DRBG_PR_ON );
40
41 TEST_ASSERT( ctr_drbg_random_with_add( &ctx, buf, 16, add1, add1_len ) == 0 );
42 TEST_ASSERT( ctr_drbg_random_with_add( &ctx, buf, 16, add2, add2_len ) == 0 );
43 hexify( output_str, buf, 16 );
44 TEST_ASSERT( strcmp( (char *) output_str, {result_str} ) == 0 );
45}
46END_CASE
47
48BEGIN_CASE
49ctr_drbg_validate_nopr:add_init_string:entropy_string:add1_string:add_reseed_string:add2_string:result_str
50{
51 unsigned char entropy[512];
52 unsigned char add_init[512];
53 unsigned char add1[512];
54 unsigned char add_reseed[512];
55 unsigned char add2[512];
56 ctr_drbg_context ctx;
57 unsigned char buf[512];
58 unsigned char output_str[512];
59 int add_init_len, add1_len, add_reseed_len, add2_len;
60
61 memset( output_str, 0, 512 );
62
63 unhexify( entropy, {entropy_string} );
64 add_init_len = unhexify( add_init, {add_init_string} );
65 add1_len = unhexify( add1, {add1_string} );
66 add_reseed_len = unhexify( add_reseed, {add_reseed_string} );
67 add2_len = unhexify( add2, {add2_string} );
68
69 test_offset = 0;
Paul Bakker18d32912011-12-10 21:42:49 +000070 TEST_ASSERT( ctr_drbg_init_entropy_len( &ctx, entropy_func, entropy, add_init, add_init_len, 32 ) == 0 );
Paul Bakker0e04d0e2011-11-27 14:46:59 +000071
72 TEST_ASSERT( ctr_drbg_random_with_add( &ctx, buf, 16, add1, add1_len ) == 0 );
73 TEST_ASSERT( ctr_drbg_reseed( &ctx, add_reseed, add_reseed_len ) == 0 );
74 TEST_ASSERT( ctr_drbg_random_with_add( &ctx, buf, 16, add2, add2_len ) == 0 );
75 hexify( output_str, buf, 16 );
76 TEST_ASSERT( strcmp( (char *) output_str, {result_str} ) == 0 );
77}
78END_CASE