Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1 | /* BEGIN_HEADER */ |
Manuel Pégourié-Gonnard | 61ce13b | 2013-01-26 16:20:32 +0100 | [diff] [blame] | 2 | #include <polarssl/ecdh.h> |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 3 | /* END_HEADER */ |
Manuel Pégourié-Gonnard | 61ce13b | 2013-01-26 16:20:32 +0100 | [diff] [blame] | 4 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 5 | /* BEGIN_DEPENDENCIES |
Manuel Pégourié-Gonnard | a0f0747 | 2013-08-23 16:35:32 +0200 | [diff] [blame] | 6 | * depends_on:POLARSSL_ECDH_C |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 7 | * END_DEPENDENCIES |
| 8 | */ |
Manuel Pégourié-Gonnard | 61ce13b | 2013-01-26 16:20:32 +0100 | [diff] [blame] | 9 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 10 | /* BEGIN_CASE */ |
| 11 | void ecdh_primitive_random( int id ) |
Manuel Pégourié-Gonnard | 61ce13b | 2013-01-26 16:20:32 +0100 | [diff] [blame] | 12 | { |
| 13 | ecp_group grp; |
| 14 | ecp_point qA, qB; |
| 15 | mpi dA, dB, zA, zB; |
| 16 | rnd_pseudo_info rnd_info; |
| 17 | |
| 18 | ecp_group_init( &grp ); |
| 19 | ecp_point_init( &qA ); ecp_point_init( &qB ); |
| 20 | mpi_init( &dA ); mpi_init( &dB ); |
| 21 | mpi_init( &zA ); mpi_init( &zB ); |
| 22 | memset( &rnd_info, 0x00, sizeof( rnd_pseudo_info ) ); |
| 23 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 24 | TEST_ASSERT( ecp_use_known_dp( &grp, id ) == 0 ); |
Manuel Pégourié-Gonnard | 61ce13b | 2013-01-26 16:20:32 +0100 | [diff] [blame] | 25 | |
| 26 | TEST_ASSERT( ecdh_gen_public( &grp, &dA, &qA, &rnd_pseudo_rand, &rnd_info ) |
| 27 | == 0 ); |
| 28 | TEST_ASSERT( ecdh_gen_public( &grp, &dB, &qB, &rnd_pseudo_rand, &rnd_info ) |
| 29 | == 0 ); |
Manuel Pégourié-Gonnard | e09d2f8 | 2013-09-02 14:29:09 +0200 | [diff] [blame] | 30 | TEST_ASSERT( ecdh_compute_shared( &grp, &zA, &qB, &dA, |
| 31 | &rnd_pseudo_rand, &rnd_info ) == 0 ); |
| 32 | TEST_ASSERT( ecdh_compute_shared( &grp, &zB, &qA, &dB, |
| 33 | NULL, NULL ) == 0 ); |
Manuel Pégourié-Gonnard | 61ce13b | 2013-01-26 16:20:32 +0100 | [diff] [blame] | 34 | |
| 35 | TEST_ASSERT( mpi_cmp_mpi( &zA, &zB ) == 0 ); |
| 36 | |
| 37 | ecp_group_free( &grp ); |
| 38 | ecp_point_free( &qA ); ecp_point_free( &qB ); |
| 39 | mpi_free( &dA ); mpi_free( &dB ); |
| 40 | mpi_free( &zA ); mpi_free( &zB ); |
| 41 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 42 | /* END_CASE */ |
Manuel Pégourié-Gonnard | 007b717 | 2013-01-27 08:56:21 +0100 | [diff] [blame] | 43 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 44 | /* BEGIN_CASE */ |
| 45 | void ecdh_primitive_testvec( int id, char *dA_str, char *xA_str, char *yA_str, |
| 46 | char *dB_str, char *xB_str, char *yB_str, |
| 47 | char *z_str ) |
Manuel Pégourié-Gonnard | 007b717 | 2013-01-27 08:56:21 +0100 | [diff] [blame] | 48 | { |
| 49 | ecp_group grp; |
| 50 | ecp_point qA, qB; |
| 51 | mpi dA, dB, zA, zB, check; |
Manuel Pégourié-Gonnard | 544416a | 2014-01-23 16:55:18 +0100 | [diff] [blame] | 52 | unsigned char rnd_buf_A[POLARSSL_ECP_MAX_BYTES]; |
| 53 | unsigned char rnd_buf_B[POLARSSL_ECP_MAX_BYTES]; |
| 54 | rnd_buf_info rnd_info_A, rnd_info_B; |
Manuel Pégourié-Gonnard | 007b717 | 2013-01-27 08:56:21 +0100 | [diff] [blame] | 55 | |
| 56 | ecp_group_init( &grp ); |
| 57 | ecp_point_init( &qA ); ecp_point_init( &qB ); |
| 58 | mpi_init( &dA ); mpi_init( &dB ); |
| 59 | mpi_init( &zA ); mpi_init( &zB ); mpi_init( &check ); |
| 60 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 61 | TEST_ASSERT( ecp_use_known_dp( &grp, id ) == 0 ); |
Manuel Pégourié-Gonnard | 007b717 | 2013-01-27 08:56:21 +0100 | [diff] [blame] | 62 | |
Manuel Pégourié-Gonnard | 544416a | 2014-01-23 16:55:18 +0100 | [diff] [blame] | 63 | rnd_info_A.buf = rnd_buf_A; |
| 64 | rnd_info_A.length = unhexify( rnd_buf_A, dA_str ); |
| 65 | |
| 66 | /* Fix rnd_buf_A by shifting it left if necessary */ |
| 67 | if( grp.nbits % 8 != 0 ) |
| 68 | { |
| 69 | unsigned char shift = 8 - ( grp.nbits % 8 ); |
| 70 | size_t i; |
| 71 | |
| 72 | for( i = 0; i < rnd_info_A.length - 1; i++ ) |
| 73 | rnd_buf_A[i] = rnd_buf_A[i] << shift |
| 74 | | rnd_buf_A[i+1] >> ( 8 - shift ); |
| 75 | |
| 76 | rnd_buf_A[rnd_info_A.length-1] <<= shift; |
| 77 | } |
| 78 | |
| 79 | rnd_info_B.buf = rnd_buf_B; |
| 80 | rnd_info_B.length = unhexify( rnd_buf_B, dB_str ); |
| 81 | |
| 82 | /* Fix rnd_buf_B by shifting it left if necessary */ |
| 83 | if( grp.nbits % 8 != 0 ) |
| 84 | { |
| 85 | unsigned char shift = 8 - ( grp.nbits % 8 ); |
| 86 | size_t i; |
| 87 | |
| 88 | for( i = 0; i < rnd_info_B.length - 1; i++ ) |
| 89 | rnd_buf_B[i] = rnd_buf_B[i] << shift |
| 90 | | rnd_buf_B[i+1] >> ( 8 - shift ); |
| 91 | |
| 92 | rnd_buf_B[rnd_info_B.length-1] <<= shift; |
| 93 | } |
| 94 | |
| 95 | TEST_ASSERT( ecdh_gen_public( &grp, &dA, &qA, |
| 96 | rnd_buffer_rand, &rnd_info_A ) == 0 ); |
Manuel Pégourié-Gonnard | 007b717 | 2013-01-27 08:56:21 +0100 | [diff] [blame] | 97 | TEST_ASSERT( ! ecp_is_zero( &qA ) ); |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 98 | TEST_ASSERT( mpi_read_string( &check, 16, xA_str ) == 0 ); |
Manuel Pégourié-Gonnard | 007b717 | 2013-01-27 08:56:21 +0100 | [diff] [blame] | 99 | TEST_ASSERT( mpi_cmp_mpi( &qA.X, &check ) == 0 ); |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 100 | TEST_ASSERT( mpi_read_string( &check, 16, yA_str ) == 0 ); |
Manuel Pégourié-Gonnard | 007b717 | 2013-01-27 08:56:21 +0100 | [diff] [blame] | 101 | TEST_ASSERT( mpi_cmp_mpi( &qA.Y, &check ) == 0 ); |
| 102 | |
Manuel Pégourié-Gonnard | 544416a | 2014-01-23 16:55:18 +0100 | [diff] [blame] | 103 | TEST_ASSERT( ecdh_gen_public( &grp, &dB, &qB, |
| 104 | rnd_buffer_rand, &rnd_info_B ) == 0 ); |
Manuel Pégourié-Gonnard | 007b717 | 2013-01-27 08:56:21 +0100 | [diff] [blame] | 105 | TEST_ASSERT( ! ecp_is_zero( &qB ) ); |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 106 | TEST_ASSERT( mpi_read_string( &check, 16, xB_str ) == 0 ); |
Manuel Pégourié-Gonnard | 007b717 | 2013-01-27 08:56:21 +0100 | [diff] [blame] | 107 | TEST_ASSERT( mpi_cmp_mpi( &qB.X, &check ) == 0 ); |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 108 | TEST_ASSERT( mpi_read_string( &check, 16, yB_str ) == 0 ); |
Manuel Pégourié-Gonnard | 007b717 | 2013-01-27 08:56:21 +0100 | [diff] [blame] | 109 | TEST_ASSERT( mpi_cmp_mpi( &qB.Y, &check ) == 0 ); |
| 110 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 111 | TEST_ASSERT( mpi_read_string( &check, 16, z_str ) == 0 ); |
Manuel Pégourié-Gonnard | e09d2f8 | 2013-09-02 14:29:09 +0200 | [diff] [blame] | 112 | TEST_ASSERT( ecdh_compute_shared( &grp, &zA, &qB, &dA, NULL, NULL ) == 0 ); |
Manuel Pégourié-Gonnard | 007b717 | 2013-01-27 08:56:21 +0100 | [diff] [blame] | 113 | TEST_ASSERT( mpi_cmp_mpi( &zA, &check ) == 0 ); |
Manuel Pégourié-Gonnard | e09d2f8 | 2013-09-02 14:29:09 +0200 | [diff] [blame] | 114 | TEST_ASSERT( ecdh_compute_shared( &grp, &zB, &qA, &dB, NULL, NULL ) == 0 ); |
Manuel Pégourié-Gonnard | 007b717 | 2013-01-27 08:56:21 +0100 | [diff] [blame] | 115 | TEST_ASSERT( mpi_cmp_mpi( &zB, &check ) == 0 ); |
| 116 | |
| 117 | ecp_group_free( &grp ); |
| 118 | ecp_point_free( &qA ); ecp_point_free( &qB ); |
| 119 | mpi_free( &dA ); mpi_free( &dB ); |
| 120 | mpi_free( &zA ); mpi_free( &zB ); mpi_free( &check ); |
| 121 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 122 | /* END_CASE */ |
Manuel Pégourié-Gonnard | 854fbd7 | 2013-02-11 20:28:55 +0100 | [diff] [blame] | 123 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 124 | /* BEGIN_CASE */ |
| 125 | void ecdh_exchange( int id ) |
Manuel Pégourié-Gonnard | 854fbd7 | 2013-02-11 20:28:55 +0100 | [diff] [blame] | 126 | { |
| 127 | ecdh_context srv, cli; |
| 128 | unsigned char buf[1000]; |
| 129 | const unsigned char *vbuf; |
| 130 | size_t len; |
| 131 | rnd_pseudo_info rnd_info; |
| 132 | |
| 133 | ecdh_init( &srv ); |
| 134 | ecdh_init( &cli ); |
| 135 | memset( &rnd_info, 0x00, sizeof( rnd_pseudo_info ) ); |
| 136 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 137 | TEST_ASSERT( ecp_use_known_dp( &srv.grp, id ) == 0 ); |
Manuel Pégourié-Gonnard | 854fbd7 | 2013-02-11 20:28:55 +0100 | [diff] [blame] | 138 | |
| 139 | memset( buf, 0x00, sizeof( buf ) ); vbuf = buf; |
| 140 | TEST_ASSERT( ecdh_make_params( &srv, &len, buf, 1000, |
| 141 | &rnd_pseudo_rand, &rnd_info ) == 0 ); |
| 142 | TEST_ASSERT( ecdh_read_params( &cli, &vbuf, buf + len ) == 0 ); |
| 143 | |
Manuel Pégourié-Gonnard | 424fda5 | 2013-02-11 22:05:42 +0100 | [diff] [blame] | 144 | memset( buf, 0x00, sizeof( buf ) ); |
Manuel Pégourié-Gonnard | 5cceb41 | 2013-02-11 21:51:45 +0100 | [diff] [blame] | 145 | TEST_ASSERT( ecdh_make_public( &cli, &len, buf, 1000, |
| 146 | &rnd_pseudo_rand, &rnd_info ) == 0 ); |
| 147 | TEST_ASSERT( ecdh_read_public( &srv, buf, len ) == 0 ); |
| 148 | |
Manuel Pégourié-Gonnard | e09d2f8 | 2013-09-02 14:29:09 +0200 | [diff] [blame] | 149 | TEST_ASSERT( ecdh_calc_secret( &srv, &len, buf, 1000, |
| 150 | &rnd_pseudo_rand, &rnd_info ) == 0 ); |
| 151 | TEST_ASSERT( ecdh_calc_secret( &cli, &len, buf, 1000, NULL, NULL ) == 0 ); |
Manuel Pégourié-Gonnard | 424fda5 | 2013-02-11 22:05:42 +0100 | [diff] [blame] | 152 | TEST_ASSERT( mpi_cmp_mpi( &srv.z, &cli.z ) == 0 ); |
| 153 | |
Manuel Pégourié-Gonnard | 854fbd7 | 2013-02-11 20:28:55 +0100 | [diff] [blame] | 154 | ecdh_free( &srv ); |
| 155 | ecdh_free( &cli ); |
| 156 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 157 | /* END_CASE */ |