Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1 | /* BEGIN_HEADER */ |
Paul Bakker | 5c60de2 | 2009-07-08 19:47:36 +0000 | [diff] [blame] | 2 | #include <polarssl/dhm.h> |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 3 | /* END_HEADER */ |
Paul Bakker | 5c60de2 | 2009-07-08 19:47:36 +0000 | [diff] [blame] | 4 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 5 | /* BEGIN_DEPENDENCIES |
| 6 | * depends_on:POLARSSL_DHM_C:POLARSSL_BIGNUM_C |
| 7 | * END_DEPENDENCIES |
| 8 | */ |
Paul Bakker | 5690efc | 2011-05-26 13:16:06 +0000 | [diff] [blame] | 9 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 10 | /* BEGIN_CASE */ |
Manuel Pégourié-Gonnard | ed8a02b | 2013-09-04 16:39:03 +0200 | [diff] [blame] | 11 | void dhm_do_dhm( int radix_P, char *input_P, |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 12 | int radix_G, char *input_G ) |
Paul Bakker | 5c60de2 | 2009-07-08 19:47:36 +0000 | [diff] [blame] | 13 | { |
| 14 | dhm_context ctx_srv; |
| 15 | dhm_context ctx_cli; |
| 16 | unsigned char ske[1000]; |
| 17 | unsigned char *p = ske; |
| 18 | unsigned char pub_cli[1000]; |
| 19 | unsigned char sec_srv[1000]; |
| 20 | unsigned char sec_cli[1000]; |
Paul Bakker | f4a3f30 | 2011-04-24 15:53:29 +0000 | [diff] [blame] | 21 | size_t ske_len = 0; |
| 22 | size_t pub_cli_len = 0; |
| 23 | size_t sec_srv_len = 1000; |
| 24 | size_t sec_cli_len = 1000; |
Paul Bakker | 69998dd | 2009-07-11 19:15:20 +0000 | [diff] [blame] | 25 | int x_size; |
Paul Bakker | 997bbd1 | 2011-03-13 15:45:42 +0000 | [diff] [blame] | 26 | rnd_pseudo_info rnd_info; |
Paul Bakker | 5c60de2 | 2009-07-08 19:47:36 +0000 | [diff] [blame] | 27 | |
| 28 | memset( &ctx_srv, 0x00, sizeof( dhm_context ) ); |
| 29 | memset( &ctx_cli, 0x00, sizeof( dhm_context ) ); |
| 30 | memset( ske, 0x00, 1000 ); |
| 31 | memset( pub_cli, 0x00, 1000 ); |
| 32 | memset( sec_srv, 0x00, 1000 ); |
| 33 | memset( sec_cli, 0x00, 1000 ); |
Paul Bakker | 997bbd1 | 2011-03-13 15:45:42 +0000 | [diff] [blame] | 34 | memset( &rnd_info, 0x00, sizeof( rnd_pseudo_info ) ); |
Paul Bakker | 5c60de2 | 2009-07-08 19:47:36 +0000 | [diff] [blame] | 35 | |
Manuel Pégourié-Gonnard | 143b502 | 2013-09-04 16:29:59 +0200 | [diff] [blame] | 36 | /* |
| 37 | * Set params |
| 38 | */ |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 39 | TEST_ASSERT( mpi_read_string( &ctx_srv.P, radix_P, input_P ) == 0 ); |
| 40 | TEST_ASSERT( mpi_read_string( &ctx_srv.G, radix_G, input_G ) == 0 ); |
Paul Bakker | 69998dd | 2009-07-11 19:15:20 +0000 | [diff] [blame] | 41 | x_size = mpi_size( &ctx_srv.P ); |
Manuel Pégourié-Gonnard | 143b502 | 2013-09-04 16:29:59 +0200 | [diff] [blame] | 42 | pub_cli_len = x_size; |
| 43 | |
| 44 | /* |
| 45 | * First key exchange |
| 46 | */ |
| 47 | TEST_ASSERT( dhm_make_params( &ctx_srv, x_size, ske, &ske_len, &rnd_pseudo_rand, &rnd_info ) == 0 ); |
| 48 | ske[ske_len++] = 0; |
| 49 | ske[ske_len++] = 0; |
| 50 | TEST_ASSERT( dhm_read_params( &ctx_cli, &p, ske + ske_len ) == 0 ); |
| 51 | |
| 52 | TEST_ASSERT( dhm_make_public( &ctx_cli, x_size, pub_cli, pub_cli_len, &rnd_pseudo_rand, &rnd_info ) == 0 ); |
| 53 | TEST_ASSERT( dhm_read_public( &ctx_srv, pub_cli, pub_cli_len ) == 0 ); |
| 54 | |
| 55 | TEST_ASSERT( dhm_calc_secret( &ctx_srv, sec_srv, &sec_srv_len, &rnd_pseudo_rand, &rnd_info ) == 0 ); |
| 56 | TEST_ASSERT( dhm_calc_secret( &ctx_cli, sec_cli, &sec_cli_len, NULL, NULL ) == 0 ); |
| 57 | |
| 58 | TEST_ASSERT( sec_srv_len == sec_cli_len ); |
| 59 | TEST_ASSERT( sec_srv_len != 0 ); |
| 60 | TEST_ASSERT( memcmp( sec_srv, sec_cli, sec_srv_len ) == 0 ); |
| 61 | |
| 62 | /* Re-do calc_secret on server to test update of blinding values */ |
| 63 | sec_srv_len = 1000; |
| 64 | TEST_ASSERT( dhm_calc_secret( &ctx_srv, sec_srv, &sec_srv_len, &rnd_pseudo_rand, &rnd_info ) == 0 ); |
| 65 | |
| 66 | TEST_ASSERT( sec_srv_len == sec_cli_len ); |
| 67 | TEST_ASSERT( sec_srv_len != 0 ); |
| 68 | TEST_ASSERT( memcmp( sec_srv, sec_cli, sec_srv_len ) == 0 ); |
| 69 | |
| 70 | /* |
| 71 | * Second key exchange to test change of blinding values on server |
| 72 | */ |
| 73 | sec_cli_len = 1000; |
| 74 | sec_srv_len = 1000; |
| 75 | p = ske; |
Paul Bakker | 5c60de2 | 2009-07-08 19:47:36 +0000 | [diff] [blame] | 76 | |
Paul Bakker | 997bbd1 | 2011-03-13 15:45:42 +0000 | [diff] [blame] | 77 | TEST_ASSERT( dhm_make_params( &ctx_srv, x_size, ske, &ske_len, &rnd_pseudo_rand, &rnd_info ) == 0 ); |
Paul Bakker | 5c60de2 | 2009-07-08 19:47:36 +0000 | [diff] [blame] | 78 | ske[ske_len++] = 0; |
| 79 | ske[ske_len++] = 0; |
| 80 | TEST_ASSERT( dhm_read_params( &ctx_cli, &p, ske + ske_len ) == 0 ); |
| 81 | |
Paul Bakker | 997bbd1 | 2011-03-13 15:45:42 +0000 | [diff] [blame] | 82 | TEST_ASSERT( dhm_make_public( &ctx_cli, x_size, pub_cli, pub_cli_len, &rnd_pseudo_rand, &rnd_info ) == 0 ); |
Paul Bakker | 5c60de2 | 2009-07-08 19:47:36 +0000 | [diff] [blame] | 83 | TEST_ASSERT( dhm_read_public( &ctx_srv, pub_cli, pub_cli_len ) == 0 ); |
| 84 | |
Manuel Pégourié-Gonnard | 2d62764 | 2013-09-04 14:22:07 +0200 | [diff] [blame] | 85 | TEST_ASSERT( dhm_calc_secret( &ctx_srv, sec_srv, &sec_srv_len, &rnd_pseudo_rand, &rnd_info ) == 0 ); |
| 86 | TEST_ASSERT( dhm_calc_secret( &ctx_cli, sec_cli, &sec_cli_len, NULL, NULL ) == 0 ); |
Paul Bakker | 5c60de2 | 2009-07-08 19:47:36 +0000 | [diff] [blame] | 87 | |
| 88 | TEST_ASSERT( sec_srv_len == sec_cli_len ); |
| 89 | TEST_ASSERT( sec_srv_len != 0 ); |
| 90 | TEST_ASSERT( memcmp( sec_srv, sec_cli, sec_srv_len ) == 0 ); |
Paul Bakker | c43481a | 2011-02-20 16:34:26 +0000 | [diff] [blame] | 91 | |
| 92 | dhm_free( &ctx_srv ); |
| 93 | dhm_free( &ctx_cli ); |
Paul Bakker | 5c60de2 | 2009-07-08 19:47:36 +0000 | [diff] [blame] | 94 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 95 | /* END_CASE */ |
Paul Bakker | 40ce79f | 2013-09-15 17:43:54 +0200 | [diff] [blame^] | 96 | |
| 97 | /* BEGIN_CASE */ |
| 98 | void dhm_selftest() |
| 99 | { |
| 100 | TEST_ASSERT( dhm_self_test( 0 ) == 0 ); |
| 101 | } |
| 102 | /* END_CASE */ |