TrustedFirmware Git Browser
Code Review Sign In
review.trustedfirmware.org / mirror / mbed-tls / 40ce79f1e6c57ba6e9c5fe5ea54c37d361f719a6 / . / tests / suites / test_suite_dhm.function
blob: dcf236312c9e91cee81dea96799bbb00fb27ec25 [file] [log] [blame]
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]1/* BEGIN_HEADER */
Paul Bakker5c60de22009-07-08 19:47:36 +0000[diff] [blame]2#include <polarssl/dhm.h>
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]3/* END_HEADER */
Paul Bakker5c60de22009-07-08 19:47:36 +0000[diff] [blame]4
Paul Bakker33b43f12013-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 Bakker5690efc2011-05-26 13:16:06 +0000[diff] [blame]9
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]10/* BEGIN_CASE */
Manuel Pégourié-Gonnarded8a02b2013-09-04 16:39:03 +0200[diff] [blame]11void dhm_do_dhm( int radix_P, char *input_P,
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]12 int radix_G, char *input_G )
Paul Bakker5c60de22009-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 Bakkerf4a3f302011-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 Bakker69998dd2009-07-11 19:15:20 +0000[diff] [blame]25 int x_size;
Paul Bakker997bbd12011-03-13 15:45:42 +0000[diff] [blame]26 rnd_pseudo_info rnd_info;
Paul Bakker5c60de22009-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 Bakker997bbd12011-03-13 15:45:42 +0000[diff] [blame]34 memset( &rnd_info, 0x00, sizeof( rnd_pseudo_info ) );
Paul Bakker5c60de22009-07-08 19:47:36 +0000[diff] [blame]35
Manuel Pégourié-Gonnard143b5022013-09-04 16:29:59 +0200[diff] [blame]36 /*
37 * Set params
38 */
Paul Bakker33b43f12013-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 Bakker69998dd2009-07-11 19:15:20 +0000[diff] [blame]41 x_size = mpi_size( &ctx_srv.P );
Manuel Pégourié-Gonnard143b5022013-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 Bakker5c60de22009-07-08 19:47:36 +0000[diff] [blame]76
Paul Bakker997bbd12011-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 Bakker5c60de22009-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 Bakker997bbd12011-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 Bakker5c60de22009-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é-Gonnard2d627642013-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 Bakker5c60de22009-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 Bakkerc43481a2011-02-20 16:34:26 +0000[diff] [blame]91
92 dhm_free( &ctx_srv );
93 dhm_free( &ctx_cli );
Paul Bakker5c60de22009-07-08 19:47:36 +0000[diff] [blame]94}
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]95/* END_CASE */
Paul Bakker40ce79f2013-09-15 17:43:54 +0200[diff] [blame^]96
97/* BEGIN_CASE */
98void dhm_selftest()
99{
100 TEST_ASSERT( dhm_self_test( 0 ) == 0 );
101}
102/* END_CASE */