00001 #include <polarssl/config.h>
00002
00003 #ifdef POLARSSL_CTR_DRBG_C
00004
00005 #include <polarssl/ctr_drbg.h>
00006
00007 int test_offset;
00008 int entropy_func( void *data, unsigned char *buf, size_t len )
00009 {
00010 unsigned char *p = (unsigned char *) data;
00011 memcpy( buf, p + test_offset, len );
00012 test_offset += 32;
00013 return( 0 );
00014 }
00015 #endif
00016
00017
00018 #if defined(POLARSSL_MEMORY_BUFFER_ALLOC_C)
00019 #include "polarssl/memory.h"
00020 #endif
00021
00022 #if defined(WANT_NOT_RND_MPI)
00023 #if defined(POLARSSL_BIGNUM_C)
00024 #include "polarssl/bignum.h"
00025 #else
00026 #error "not_rnd_mpi() need bignum.c"
00027 #endif
00028 #endif
00029
00030 #ifdef _MSC_VER
00031 #include <basetsd.h>
00032 typedef UINT32 uint32_t;
00033 #else
00034 #include <inttypes.h>
00035 #endif
00036
00037 #include <assert.h>
00038 #include <stdlib.h>
00039 #include <string.h>
00040
00041
00042
00043
00044 #ifndef GET_UINT32_BE
00045 #define GET_UINT32_BE(n,b,i) \
00046 { \
00047 (n) = ( (uint32_t) (b)[(i) ] << 24 ) \
00048 | ( (uint32_t) (b)[(i) + 1] << 16 ) \
00049 | ( (uint32_t) (b)[(i) + 2] << 8 ) \
00050 | ( (uint32_t) (b)[(i) + 3] ); \
00051 }
00052 #endif
00053
00054 #ifndef PUT_UINT32_BE
00055 #define PUT_UINT32_BE(n,b,i) \
00056 { \
00057 (b)[(i) ] = (unsigned char) ( (n) >> 24 ); \
00058 (b)[(i) + 1] = (unsigned char) ( (n) >> 16 ); \
00059 (b)[(i) + 2] = (unsigned char) ( (n) >> 8 ); \
00060 (b)[(i) + 3] = (unsigned char) ( (n) ); \
00061 }
00062 #endif
00063
00064 static int unhexify(unsigned char *obuf, const char *ibuf)
00065 {
00066 unsigned char c, c2;
00067 int len = strlen(ibuf) / 2;
00068 assert(!(strlen(ibuf) %1));
00069
00070 while (*ibuf != 0)
00071 {
00072 c = *ibuf++;
00073 if( c >= '0' && c <= '9' )
00074 c -= '0';
00075 else if( c >= 'a' && c <= 'f' )
00076 c -= 'a' - 10;
00077 else if( c >= 'A' && c <= 'F' )
00078 c -= 'A' - 10;
00079 else
00080 assert( 0 );
00081
00082 c2 = *ibuf++;
00083 if( c2 >= '0' && c2 <= '9' )
00084 c2 -= '0';
00085 else if( c2 >= 'a' && c2 <= 'f' )
00086 c2 -= 'a' - 10;
00087 else if( c2 >= 'A' && c2 <= 'F' )
00088 c2 -= 'A' - 10;
00089 else
00090 assert( 0 );
00091
00092 *obuf++ = ( c << 4 ) | c2;
00093 }
00094
00095 return len;
00096 }
00097
00098 static void hexify(unsigned char *obuf, const unsigned char *ibuf, int len)
00099 {
00100 unsigned char l, h;
00101
00102 while (len != 0)
00103 {
00104 h = (*ibuf) / 16;
00105 l = (*ibuf) % 16;
00106
00107 if( h < 10 )
00108 *obuf++ = '0' + h;
00109 else
00110 *obuf++ = 'a' + h - 10;
00111
00112 if( l < 10 )
00113 *obuf++ = '0' + l;
00114 else
00115 *obuf++ = 'a' + l - 10;
00116
00117 ++ibuf;
00118 len--;
00119 }
00120 }
00121
00131 static int rnd_std_rand( void *rng_state, unsigned char *output, size_t len )
00132 {
00133 size_t i;
00134
00135 if( rng_state != NULL )
00136 rng_state = NULL;
00137
00138 for( i = 0; i < len; ++i )
00139 output[i] = rand();
00140
00141 return( 0 );
00142 }
00143
00149 static int rnd_zero_rand( void *rng_state, unsigned char *output, size_t len )
00150 {
00151 if( rng_state != NULL )
00152 rng_state = NULL;
00153
00154 memset( output, 0, len );
00155
00156 return( 0 );
00157 }
00158
00159 typedef struct
00160 {
00161 unsigned char *buf;
00162 size_t length;
00163 } rnd_buf_info;
00164
00176 static int rnd_buffer_rand( void *rng_state, unsigned char *output, size_t len )
00177 {
00178 rnd_buf_info *info = (rnd_buf_info *) rng_state;
00179 size_t use_len;
00180
00181 if( rng_state == NULL )
00182 return( rnd_std_rand( NULL, output, len ) );
00183
00184 use_len = len;
00185 if( len > info->length )
00186 use_len = info->length;
00187
00188 if( use_len )
00189 {
00190 memcpy( output, info->buf, use_len );
00191 info->buf += use_len;
00192 info->length -= use_len;
00193 }
00194
00195 if( len - use_len > 0 )
00196 return( rnd_std_rand( NULL, output + use_len, len - use_len ) );
00197
00198 return( 0 );
00199 }
00200
00208 typedef struct
00209 {
00210 uint32_t key[16];
00211 uint32_t v0, v1;
00212 } rnd_pseudo_info;
00213
00222 static int rnd_pseudo_rand( void *rng_state, unsigned char *output, size_t len )
00223 {
00224 rnd_pseudo_info *info = (rnd_pseudo_info *) rng_state;
00225 uint32_t i, *k, sum, delta=0x9E3779B9;
00226 unsigned char result[4];
00227
00228 if( rng_state == NULL )
00229 return( rnd_std_rand( NULL, output, len ) );
00230
00231 k = info->key;
00232
00233 while( len > 0 )
00234 {
00235 size_t use_len = ( len > 4 ) ? 4 : len;
00236 sum = 0;
00237
00238 for( i = 0; i < 32; i++ )
00239 {
00240 info->v0 += (((info->v1 << 4) ^ (info->v1 >> 5)) + info->v1) ^ (sum + k[sum & 3]);
00241 sum += delta;
00242 info->v1 += (((info->v0 << 4) ^ (info->v0 >> 5)) + info->v0) ^ (sum + k[(sum>>11) & 3]);
00243 }
00244
00245 PUT_UINT32_BE( info->v0, result, 0 );
00246 memcpy( output, result, use_len );
00247 len -= use_len;
00248 }
00249
00250 return( 0 );
00251 }
00252
00253 #if defined(WANT_NOT_RND_MPI)
00254
00262 #define ciL (sizeof(t_uint))
00263 #define CHARS_TO_LIMBS(i) (((i) + ciL - 1) / ciL)
00264 static int not_rnd_mpi( void *in, unsigned char *out, size_t len )
00265 {
00266 char *str = (char *) in;
00267 mpi X;
00268
00269
00270
00271
00272
00273 X.s = 1;
00274 X.p = (t_uint *) out;
00275 X.n = CHARS_TO_LIMBS( len );
00276
00277
00278
00279
00280
00281 assert( strlen( str ) / 2 == len );
00282
00283 return( mpi_read_string( &X, 16, str ) );
00284 }
00285 #endif
00286
00287
00288 #include <stdio.h>
00289 #include <string.h>
00290
00291 static int test_errors = 0;
00292
00293 #ifdef POLARSSL_CTR_DRBG_C
00294
00295 #define TEST_SUITE_ACTIVE
00296
00297 static int test_assert( int correct, char *test )
00298 {
00299 if( correct )
00300 return( 0 );
00301
00302 test_errors++;
00303 if( test_errors == 1 )
00304 printf( "FAILED\n" );
00305 printf( " %s\n", test );
00306
00307 return( 1 );
00308 }
00309
00310 #define TEST_ASSERT( TEST ) \
00311 do { test_assert( (TEST) ? 1 : 0, #TEST ); \
00312 if( test_errors) return; \
00313 } while (0)
00314
00315 int verify_string( char **str )
00316 {
00317 if( (*str)[0] != '"' ||
00318 (*str)[strlen( *str ) - 1] != '"' )
00319 {
00320 printf( "Expected string (with \"\") for parameter and got: %s\n", *str );
00321 return( -1 );
00322 }
00323
00324 (*str)++;
00325 (*str)[strlen( *str ) - 1] = '\0';
00326
00327 return( 0 );
00328 }
00329
00330 int verify_int( char *str, int *value )
00331 {
00332 size_t i;
00333 int minus = 0;
00334 int digits = 1;
00335 int hex = 0;
00336
00337 for( i = 0; i < strlen( str ); i++ )
00338 {
00339 if( i == 0 && str[i] == '-' )
00340 {
00341 minus = 1;
00342 continue;
00343 }
00344
00345 if( ( ( minus && i == 2 ) || ( !minus && i == 1 ) ) &&
00346 str[i - 1] == '0' && str[i] == 'x' )
00347 {
00348 hex = 1;
00349 continue;
00350 }
00351
00352 if( str[i] < '0' || str[i] > '9' )
00353 {
00354 digits = 0;
00355 break;
00356 }
00357 }
00358
00359 if( digits )
00360 {
00361 if( hex )
00362 *value = strtol( str, NULL, 16 );
00363 else
00364 *value = strtol( str, NULL, 10 );
00365
00366 return( 0 );
00367 }
00368
00369
00370
00371 printf( "Expected integer for parameter and got: %s\n", str );
00372 return( -1 );
00373 }
00374
00375 void test_suite_ctr_drbg_validate_pr( char *add_init_string, char *entropy_string,
00376 char *add1_string, char *add2_string,
00377 char *result_str )
00378 {
00379 unsigned char entropy[512];
00380 unsigned char add_init[512];
00381 unsigned char add1[512];
00382 unsigned char add2[512];
00383 ctr_drbg_context ctx;
00384 unsigned char buf[512];
00385 unsigned char output_str[512];
00386 int add_init_len, add1_len, add2_len;
00387
00388 memset( output_str, 0, 512 );
00389
00390 unhexify( entropy, entropy_string );
00391 add_init_len = unhexify( add_init, add_init_string );
00392 add1_len = unhexify( add1, add1_string );
00393 add2_len = unhexify( add2, add2_string );
00394
00395 test_offset = 0;
00396 TEST_ASSERT( ctr_drbg_init_entropy_len( &ctx, entropy_func, entropy, add_init, add_init_len, 32 ) == 0 );
00397 ctr_drbg_set_prediction_resistance( &ctx, CTR_DRBG_PR_ON );
00398
00399 TEST_ASSERT( ctr_drbg_random_with_add( &ctx, buf, 16, add1, add1_len ) == 0 );
00400 TEST_ASSERT( ctr_drbg_random_with_add( &ctx, buf, 16, add2, add2_len ) == 0 );
00401 hexify( output_str, buf, 16 );
00402 TEST_ASSERT( strcmp( (char *) output_str, result_str ) == 0 );
00403 }
00404
00405 void test_suite_ctr_drbg_validate_nopr( char *add_init_string, char *entropy_string,
00406 char *add1_string, char *add_reseed_string,
00407 char *add2_string, char *result_str )
00408 {
00409 unsigned char entropy[512];
00410 unsigned char add_init[512];
00411 unsigned char add1[512];
00412 unsigned char add_reseed[512];
00413 unsigned char add2[512];
00414 ctr_drbg_context ctx;
00415 unsigned char buf[512];
00416 unsigned char output_str[512];
00417 int add_init_len, add1_len, add_reseed_len, add2_len;
00418
00419 memset( output_str, 0, 512 );
00420
00421 unhexify( entropy, entropy_string );
00422 add_init_len = unhexify( add_init, add_init_string );
00423 add1_len = unhexify( add1, add1_string );
00424 add_reseed_len = unhexify( add_reseed, add_reseed_string );
00425 add2_len = unhexify( add2, add2_string );
00426
00427 test_offset = 0;
00428 TEST_ASSERT( ctr_drbg_init_entropy_len( &ctx, entropy_func, entropy, add_init, add_init_len, 32 ) == 0 );
00429
00430 TEST_ASSERT( ctr_drbg_random_with_add( &ctx, buf, 16, add1, add1_len ) == 0 );
00431 TEST_ASSERT( ctr_drbg_reseed( &ctx, add_reseed, add_reseed_len ) == 0 );
00432 TEST_ASSERT( ctr_drbg_random_with_add( &ctx, buf, 16, add2, add2_len ) == 0 );
00433 hexify( output_str, buf, 16 );
00434 TEST_ASSERT( strcmp( (char *) output_str, result_str ) == 0 );
00435 }
00436
00437
00438 #endif
00439
00440
00441 int dep_check( char *str )
00442 {
00443 if( str == NULL )
00444 return( 1 );
00445
00446
00447
00448 return( 1 );
00449 }
00450
00451 int dispatch_test(int cnt, char *params[50])
00452 {
00453 int ret;
00454 ((void) cnt);
00455 ((void) params);
00456
00457 #if defined(TEST_SUITE_ACTIVE)
00458 if( strcmp( params[0], "ctr_drbg_validate_pr" ) == 0 )
00459 {
00460
00461 char *param1 = params[1];
00462 char *param2 = params[2];
00463 char *param3 = params[3];
00464 char *param4 = params[4];
00465 char *param5 = params[5];
00466
00467 if( cnt != 6 )
00468 {
00469 fprintf( stderr, "\nIncorrect argument count (%d != %d)\n", cnt, 6 );
00470 return( 2 );
00471 }
00472
00473 if( verify_string( ¶m1 ) != 0 ) return( 2 );
00474 if( verify_string( ¶m2 ) != 0 ) return( 2 );
00475 if( verify_string( ¶m3 ) != 0 ) return( 2 );
00476 if( verify_string( ¶m4 ) != 0 ) return( 2 );
00477 if( verify_string( ¶m5 ) != 0 ) return( 2 );
00478
00479 test_suite_ctr_drbg_validate_pr( param1, param2, param3, param4, param5 );
00480 return ( 0 );
00481
00482 return ( 3 );
00483 }
00484 else
00485 if( strcmp( params[0], "ctr_drbg_validate_nopr" ) == 0 )
00486 {
00487
00488 char *param1 = params[1];
00489 char *param2 = params[2];
00490 char *param3 = params[3];
00491 char *param4 = params[4];
00492 char *param5 = params[5];
00493 char *param6 = params[6];
00494
00495 if( cnt != 7 )
00496 {
00497 fprintf( stderr, "\nIncorrect argument count (%d != %d)\n", cnt, 7 );
00498 return( 2 );
00499 }
00500
00501 if( verify_string( ¶m1 ) != 0 ) return( 2 );
00502 if( verify_string( ¶m2 ) != 0 ) return( 2 );
00503 if( verify_string( ¶m3 ) != 0 ) return( 2 );
00504 if( verify_string( ¶m4 ) != 0 ) return( 2 );
00505 if( verify_string( ¶m5 ) != 0 ) return( 2 );
00506 if( verify_string( ¶m6 ) != 0 ) return( 2 );
00507
00508 test_suite_ctr_drbg_validate_nopr( param1, param2, param3, param4, param5, param6 );
00509 return ( 0 );
00510
00511 return ( 3 );
00512 }
00513 else
00514
00515 {
00516 fprintf( stdout, "FAILED\nSkipping unknown test function '%s'\n", params[0] );
00517 fflush( stdout );
00518 return( 1 );
00519 }
00520 #else
00521 return( 3 );
00522 #endif
00523 return( ret );
00524 }
00525
00526 int get_line( FILE *f, char *buf, size_t len )
00527 {
00528 char *ret;
00529
00530 ret = fgets( buf, len, f );
00531 if( ret == NULL )
00532 return( -1 );
00533
00534 if( strlen( buf ) && buf[strlen(buf) - 1] == '\n' )
00535 buf[strlen(buf) - 1] = '\0';
00536 if( strlen( buf ) && buf[strlen(buf) - 1] == '\r' )
00537 buf[strlen(buf) - 1] = '\0';
00538
00539 return( 0 );
00540 }
00541
00542 int parse_arguments( char *buf, size_t len, char *params[50] )
00543 {
00544 int cnt = 0, i;
00545 char *cur = buf;
00546 char *p = buf, *q;
00547
00548 params[cnt++] = cur;
00549
00550 while( *p != '\0' && p < buf + len )
00551 {
00552 if( *p == '\\' )
00553 {
00554 *p++;
00555 *p++;
00556 continue;
00557 }
00558 if( *p == ':' )
00559 {
00560 if( p + 1 < buf + len )
00561 {
00562 cur = p + 1;
00563 params[cnt++] = cur;
00564 }
00565 *p = '\0';
00566 }
00567
00568 *p++;
00569 }
00570
00571
00572 for( i = 0; i < cnt; i++ )
00573 {
00574 p = params[i];
00575 q = params[i];
00576
00577 while( *p != '\0' )
00578 {
00579 if( *p == '\\' && *(p + 1) == 'n' )
00580 {
00581 p += 2;
00582 *(q++) = '\n';
00583 }
00584 else if( *p == '\\' && *(p + 1) == ':' )
00585 {
00586 p += 2;
00587 *(q++) = ':';
00588 }
00589 else if( *p == '\\' && *(p + 1) == '?' )
00590 {
00591 p += 2;
00592 *(q++) = '?';
00593 }
00594 else
00595 *(q++) = *(p++);
00596 }
00597 *q = '\0';
00598 }
00599
00600 return( cnt );
00601 }
00602
00603 int main()
00604 {
00605 int ret, i, cnt, total_errors = 0, total_tests = 0, total_skipped = 0;
00606 const char *filename = "/home/abuild/rpmbuild/BUILD/polarssl-1.3.2/tests/suites/test_suite_ctr_drbg.data";
00607 FILE *file;
00608 char buf[5000];
00609 char *params[50];
00610
00611 #if defined(POLARSSL_MEMORY_BUFFER_ALLOC_C)
00612 unsigned char alloc_buf[1000000];
00613 memory_buffer_alloc_init( alloc_buf, sizeof(alloc_buf) );
00614 #endif
00615
00616 file = fopen( filename, "r" );
00617 if( file == NULL )
00618 {
00619 fprintf( stderr, "Failed to open\n" );
00620 return( 1 );
00621 }
00622
00623 while( !feof( file ) )
00624 {
00625 int skip = 0;
00626
00627 if( ( ret = get_line( file, buf, sizeof(buf) ) ) != 0 )
00628 break;
00629 fprintf( stdout, "%s%.66s", test_errors ? "\n" : "", buf );
00630 fprintf( stdout, " " );
00631 for( i = strlen( buf ) + 1; i < 67; i++ )
00632 fprintf( stdout, "." );
00633 fprintf( stdout, " " );
00634 fflush( stdout );
00635
00636 total_tests++;
00637
00638 if( ( ret = get_line( file, buf, sizeof(buf) ) ) != 0 )
00639 break;
00640 cnt = parse_arguments( buf, strlen(buf), params );
00641
00642 if( strcmp( params[0], "depends_on" ) == 0 )
00643 {
00644 for( i = 1; i < cnt; i++ )
00645 if( dep_check( params[i] ) != 0 )
00646 skip = 1;
00647
00648 if( ( ret = get_line( file, buf, sizeof(buf) ) ) != 0 )
00649 break;
00650 cnt = parse_arguments( buf, strlen(buf), params );
00651 }
00652
00653 if( skip == 0 )
00654 {
00655 test_errors = 0;
00656 ret = dispatch_test( cnt, params );
00657 }
00658
00659 if( skip == 1 || ret == 3 )
00660 {
00661 total_skipped++;
00662 fprintf( stdout, "----\n" );
00663 fflush( stdout );
00664 }
00665 else if( ret == 0 && test_errors == 0 )
00666 {
00667 fprintf( stdout, "PASS\n" );
00668 fflush( stdout );
00669 }
00670 else if( ret == 2 )
00671 {
00672 fprintf( stderr, "FAILED: FATAL PARSE ERROR\n" );
00673 fclose(file);
00674 exit( 2 );
00675 }
00676 else
00677 total_errors++;
00678
00679 if( ( ret = get_line( file, buf, sizeof(buf) ) ) != 0 )
00680 break;
00681 if( strlen(buf) != 0 )
00682 {
00683 fprintf( stderr, "Should be empty %d\n", (int) strlen(buf) );
00684 return( 1 );
00685 }
00686 }
00687 fclose(file);
00688
00689 fprintf( stdout, "\n----------------------------------------------------------------------------\n\n");
00690 if( total_errors == 0 )
00691 fprintf( stdout, "PASSED" );
00692 else
00693 fprintf( stdout, "FAILED" );
00694
00695 fprintf( stdout, " (%d / %d tests (%d skipped))\n",
00696 total_tests - total_errors, total_tests, total_skipped );
00697
00698 #if defined(POLARSSL_MEMORY_BUFFER_ALLOC_C)
00699 #if defined(POLARSSL_MEMORY_DEBUG)
00700 memory_buffer_alloc_status();
00701 #endif
00702 memory_buffer_alloc_free();
00703 #endif
00704
00705 return( total_errors != 0 );
00706 }
00707
00708