3 #ifdef POLARSSL_CTR_DRBG_C
8 int entropy_func(
void *data,
unsigned char *buf,
size_t len )
10 unsigned char *p = (
unsigned char *) data;
11 memcpy( buf, p + test_offset, len );
18 #if defined(POLARSSL_MEMORY_BUFFER_ALLOC_C)
24 typedef UINT32 uint32_t;
37 #define GET_UINT32_BE(n,b,i) \
39 (n) = ( (uint32_t) (b)[(i) ] << 24 ) \
40 | ( (uint32_t) (b)[(i) + 1] << 16 ) \
41 | ( (uint32_t) (b)[(i) + 2] << 8 ) \
42 | ( (uint32_t) (b)[(i) + 3] ); \
47 #define PUT_UINT32_BE(n,b,i) \
49 (b)[(i) ] = (unsigned char) ( (n) >> 24 ); \
50 (b)[(i) + 1] = (unsigned char) ( (n) >> 16 ); \
51 (b)[(i) + 2] = (unsigned char) ( (n) >> 8 ); \
52 (b)[(i) + 3] = (unsigned char) ( (n) ); \
56 static int unhexify(
unsigned char *obuf,
const char *ibuf)
59 int len = strlen(ibuf) / 2;
60 assert(!(strlen(ibuf) %1));
65 if( c >=
'0' && c <=
'9' )
67 else if( c >=
'a' && c <=
'f' )
69 else if( c >=
'A' && c <=
'F' )
75 if( c2 >=
'0' && c2 <=
'9' )
77 else if( c2 >=
'a' && c2 <=
'f' )
79 else if( c2 >=
'A' && c2 <=
'F' )
84 *obuf++ = ( c << 4 ) | c2;
90 static void hexify(
unsigned char *obuf,
const unsigned char *ibuf,
int len)
102 *obuf++ =
'a' + h - 10;
107 *obuf++ =
'a' + l - 10;
123 static int rnd_std_rand(
void *rng_state,
unsigned char *output,
size_t len )
127 if( rng_state != NULL )
130 for( i = 0; i < len; ++i )
141 static int rnd_zero_rand(
void *rng_state,
unsigned char *output,
size_t len )
143 if( rng_state != NULL )
146 memset( output, 0, len );
173 if( rng_state == NULL )
182 memcpy( output, info->
buf, use_len );
183 info->
buf += use_len;
187 if( len - use_len > 0 )
188 return(
rnd_std_rand( NULL, output + use_len, len - use_len ) );
217 uint32_t i, *k, sum, delta=0x9E3779B9;
218 unsigned char result[4];
220 if( rng_state == NULL )
227 size_t use_len = ( len > 4 ) ? 4 : len;
230 for( i = 0; i < 32; i++ )
232 info->
v0 += (((info->
v1 << 4) ^ (info->
v1 >> 5)) + info->
v1) ^ (sum + k[sum & 3]);
234 info->
v1 += (((info->
v0 << 4) ^ (info->
v0 >> 5)) + info->
v0) ^ (sum + k[(sum>>11) & 3]);
238 memcpy( output, result, use_len );
254 static int not_rnd(
void *in,
unsigned char *out,
size_t len )
257 const char *ibuf = in;
259 assert( len == strlen(ibuf) / 2 );
260 assert(!(strlen(ibuf) %1));
262 obuf = out + (len - 1);
266 if( c >=
'0' && c <=
'9' )
268 else if( c >=
'a' && c <=
'f' )
270 else if( c >=
'A' && c <=
'F' )
276 if( c2 >=
'0' && c2 <=
'9' )
278 else if( c2 >=
'a' && c2 <=
'f' )
280 else if( c2 >=
'A' && c2 <=
'F' )
285 *obuf-- = ( c << 4 ) | c2;
297 #ifdef POLARSSL_CTR_DRBG_C
299 #define TEST_SUITE_ACTIVE
307 if( test_errors == 1 )
308 printf(
"FAILED\n" );
309 printf(
" %s\n", test );
314 #define TEST_ASSERT( TEST ) \
315 do { test_assert( (TEST) ? 1 : 0, #TEST ); \
316 if( test_errors) return; \
321 if( (*str)[0] !=
'"' ||
322 (*str)[strlen( *str ) - 1] !=
'"' )
324 printf(
"Expected string (with \"\") for parameter and got: %s\n", *str );
329 (*str)[strlen( *str ) - 1] =
'\0';
341 for( i = 0; i < strlen( str ); i++ )
343 if( i == 0 && str[i] ==
'-' )
349 if( ( ( minus && i == 2 ) || ( !minus && i == 1 ) ) &&
350 str[i - 1] ==
'0' && str[i] ==
'x' )
356 if( str[i] <
'0' || str[i] >
'9' )
366 *value = strtol( str, NULL, 16 );
368 *value = strtol( str, NULL, 10 );
375 printf(
"Expected integer for parameter and got: %s\n", str );
379 void test_suite_ctr_drbg_validate_pr(
char *add_init_string,
char *entropy_string,
380 char *add1_string,
char *add2_string,
383 unsigned char entropy[512];
384 unsigned char add_init[512];
385 unsigned char add1[512];
386 unsigned char add2[512];
388 unsigned char buf[512];
389 unsigned char output_str[512];
390 int add_init_len, add1_len, add2_len;
392 memset( output_str, 0, 512 );
394 unhexify( entropy, entropy_string );
395 add_init_len =
unhexify( add_init, add_init_string );
396 add1_len =
unhexify( add1, add1_string );
397 add2_len =
unhexify( add2, add2_string );
405 hexify( output_str, buf, 16 );
406 TEST_ASSERT( strcmp( (
char *) output_str, result_str ) == 0 );
409 void test_suite_ctr_drbg_validate_nopr(
char *add_init_string,
char *entropy_string,
410 char *add1_string,
char *add_reseed_string,
411 char *add2_string,
char *result_str )
413 unsigned char entropy[512];
414 unsigned char add_init[512];
415 unsigned char add1[512];
416 unsigned char add_reseed[512];
417 unsigned char add2[512];
419 unsigned char buf[512];
420 unsigned char output_str[512];
421 int add_init_len, add1_len, add_reseed_len, add2_len;
423 memset( output_str, 0, 512 );
425 unhexify( entropy, entropy_string );
426 add_init_len =
unhexify( add_init, add_init_string );
427 add1_len =
unhexify( add1, add1_string );
428 add_reseed_len =
unhexify( add_reseed, add_reseed_string );
429 add2_len =
unhexify( add2, add2_string );
437 hexify( output_str, buf, 16 );
438 TEST_ASSERT( strcmp( (
char *) output_str, result_str ) == 0 );
461 #if defined(TEST_SUITE_ACTIVE)
462 if( strcmp( params[0],
"ctr_drbg_validate_pr" ) == 0 )
465 char *param1 = params[1];
466 char *param2 = params[2];
467 char *param3 = params[3];
468 char *param4 = params[4];
469 char *param5 = params[5];
473 fprintf( stderr,
"\nIncorrect argument count (%d != %d)\n", cnt, 6 );
483 test_suite_ctr_drbg_validate_pr( param1, param2, param3, param4, param5 );
489 if( strcmp( params[0],
"ctr_drbg_validate_nopr" ) == 0 )
492 char *param1 = params[1];
493 char *param2 = params[2];
494 char *param3 = params[3];
495 char *param4 = params[4];
496 char *param5 = params[5];
497 char *param6 = params[6];
501 fprintf( stderr,
"\nIncorrect argument count (%d != %d)\n", cnt, 7 );
512 test_suite_ctr_drbg_validate_nopr( param1, param2, param3, param4, param5, param6 );
520 fprintf( stdout,
"FAILED\nSkipping unknown test function '%s'\n", params[0] );
534 ret = fgets( buf, len, f );
538 if( strlen( buf ) && buf[strlen(buf) - 1] ==
'\n' )
539 buf[strlen(buf) - 1] =
'\0';
540 if( strlen( buf ) && buf[strlen(buf) - 1] ==
'\r' )
541 buf[strlen(buf) - 1] =
'\0';
554 while( *p !=
'\0' && p < buf + len )
564 if( p + 1 < buf + len )
576 for( i = 0; i < cnt; i++ )
583 if( *p ==
'\\' && *(p + 1) ==
'n' )
588 else if( *p ==
'\\' && *(p + 1) ==
':' )
593 else if( *p ==
'\\' && *(p + 1) ==
'?' )
609 int ret, i, cnt, total_errors = 0, total_tests = 0, total_skipped = 0;
610 const char *filename =
"/home/iurt/rpmbuild/BUILD/polarssl-1.3.1/tests/suites/test_suite_ctr_drbg.data";
615 #if defined(POLARSSL_MEMORY_BUFFER_ALLOC_C)
616 unsigned char alloc_buf[1000000];
617 memory_buffer_alloc_init( alloc_buf,
sizeof(alloc_buf) );
620 file = fopen( filename,
"r" );
623 fprintf( stderr,
"Failed to open\n" );
627 while( !feof( file ) )
631 if( ( ret =
get_line( file, buf,
sizeof(buf) ) ) != 0 )
633 fprintf( stdout,
"%s%.66s", test_errors ?
"\n" :
"", buf );
634 fprintf( stdout,
" " );
635 for( i = strlen( buf ) + 1; i < 67; i++ )
636 fprintf( stdout,
"." );
637 fprintf( stdout,
" " );
642 if( ( ret =
get_line( file, buf,
sizeof(buf) ) ) != 0 )
646 if( strcmp( params[0],
"depends_on" ) == 0 )
648 for( i = 1; i < cnt; i++ )
652 if( ( ret =
get_line( file, buf,
sizeof(buf) ) ) != 0 )
663 if( skip == 1 || ret == 3 )
666 fprintf( stdout,
"----\n" );
669 else if( ret == 0 && test_errors == 0 )
671 fprintf( stdout,
"PASS\n" );
676 fprintf( stderr,
"FAILED: FATAL PARSE ERROR\n" );
683 if( ( ret =
get_line( file, buf,
sizeof(buf) ) ) != 0 )
685 if( strlen(buf) != 0 )
687 fprintf( stderr,
"Should be empty %d\n", (
int) strlen(buf) );
693 fprintf( stdout,
"\n----------------------------------------------------------------------------\n\n");
694 if( total_errors == 0 )
695 fprintf( stdout,
"PASSED" );
697 fprintf( stdout,
"FAILED" );
699 fprintf( stdout,
" (%d / %d tests (%d skipped))\n",
700 total_tests - total_errors, total_tests, total_skipped );
702 #if defined(POLARSSL_MEMORY_BUFFER_ALLOC_C)
703 #if defined(POLARSSL_MEMORY_DEBUG)
704 memory_buffer_alloc_status();
706 memory_buffer_alloc_free();
709 return( total_errors != 0 );
void ctr_drbg_set_prediction_resistance(ctr_drbg_context *ctx, int resistance)
Enable / disable prediction resistance (Default: Off)
#define CTR_DRBG_PR_ON
Prediction resistance enabled.
static int rnd_zero_rand(void *rng_state, unsigned char *output, size_t len)
This function only returns zeros.
Info structure for the pseudo random function.
Configuration options (set of defines)
static int rnd_std_rand(void *rng_state, unsigned char *output, size_t len)
This function just returns data from rand().
int ctr_drbg_init_entropy_len(ctr_drbg_context *, int(*)(void *, unsigned char *, size_t), void *, const unsigned char *, size_t, size_t)
static int not_rnd(void *in, unsigned char *out, size_t len)
This function returns a buffer given as a hex string.
static int test_assert(int correct, char *test)
int main(int argc, char *argv[])
#define TEST_ASSERT(TEST)
int ctr_drbg_random_with_add(void *p_rng, unsigned char *output, size_t output_len, const unsigned char *additional, size_t add_len)
CTR_DRBG generate random with additional update input.
static void hexify(unsigned char *obuf, const unsigned char *ibuf, int len)
int ctr_drbg_reseed(ctr_drbg_context *ctx, const unsigned char *additional, size_t len)
CTR_DRBG reseeding (extracts data from entropy source)
int parse_arguments(char *buf, size_t len, char *params[50])
CTR_DRBG context structure.
int verify_string(char **str)
int dispatch_test(int cnt, char *params[50])
static int rnd_buffer_rand(void *rng_state, unsigned char *output, size_t len)
This function returns random based on a buffer it receives.
static int unhexify(unsigned char *obuf, const char *ibuf)
static int rnd_pseudo_rand(void *rng_state, unsigned char *output, size_t len)
This function returns random based on a pseudo random function.
int verify_int(char *str, int *value)
int entropy_func(void *data, unsigned char *output, size_t len)
Retrieve entropy from the accumulator (Max ENTROPY_BLOCK_SIZE) (Thread-safe if POLARSSL_THREADING_C i...
#define PUT_UINT32_BE(n, b, i)
int get_line(FILE *f, char *buf, size_t len)
CTR_DRBG based on AES-256 (NIST SP 800-90)