TrustedFirmware Git Browser
Code Review Sign In
review.trustedfirmware.org / mirror / mbed-tls / 07de4b1d084ab68587066b68d29befb91ccc88c9 / . / tests / suites / test_suite_dhm.function
blob: e6524cc14627e0934f04a70d525dabc6b92e982b [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 */
11void dhm_do_dhm( int NOTUSED, int radix_P, char *input_P,
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
Paul Bakkerdbd443d2013-08-16 13:38:47 +0200[diff] [blame]28 ((void)NOTUSED);
29
Paul Bakker5c60de22009-07-08 19:47:36 +0000[diff] [blame]30 memset( &ctx_srv, 0x00, sizeof( dhm_context ) );
31 memset( &ctx_cli, 0x00, sizeof( dhm_context ) );
32 memset( ske, 0x00, 1000 );
33 memset( pub_cli, 0x00, 1000 );
34 memset( sec_srv, 0x00, 1000 );
35 memset( sec_cli, 0x00, 1000 );
Paul Bakker997bbd12011-03-13 15:45:42 +0000[diff] [blame]36 memset( &rnd_info, 0x00, sizeof( rnd_pseudo_info ) );
Paul Bakker5c60de22009-07-08 19:47:36 +0000[diff] [blame]37
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]38 TEST_ASSERT( mpi_read_string( &ctx_srv.P, radix_P, input_P ) == 0 );
39 TEST_ASSERT( mpi_read_string( &ctx_srv.G, radix_G, input_G ) == 0 );
Paul Bakker69998dd2009-07-11 19:15:20 +0000[diff] [blame]40 x_size = mpi_size( &ctx_srv.P );
Paul Bakker5c60de22009-07-08 19:47:36 +0000[diff] [blame]41
Paul Bakker997bbd12011-03-13 15:45:42 +0000[diff] [blame]42 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]43 ske[ske_len++] = 0;
44 ske[ske_len++] = 0;
45 TEST_ASSERT( dhm_read_params( &ctx_cli, &p, ske + ske_len ) == 0 );
46
47 pub_cli_len = x_size;
Paul Bakker997bbd12011-03-13 15:45:42 +0000[diff] [blame]48 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]49
50 TEST_ASSERT( dhm_read_public( &ctx_srv, pub_cli, pub_cli_len ) == 0 );
51
52 TEST_ASSERT( dhm_calc_secret( &ctx_srv, sec_srv, &sec_srv_len ) == 0 );
53 TEST_ASSERT( dhm_calc_secret( &ctx_cli, sec_cli, &sec_cli_len ) == 0 );
54
55 TEST_ASSERT( sec_srv_len == sec_cli_len );
56 TEST_ASSERT( sec_srv_len != 0 );
57 TEST_ASSERT( memcmp( sec_srv, sec_cli, sec_srv_len ) == 0 );
Paul Bakkerc43481a2011-02-20 16:34:26 +0000[diff] [blame]58
59 dhm_free( &ctx_srv );
60 dhm_free( &ctx_cli );
Paul Bakker5c60de22009-07-08 19:47:36 +0000[diff] [blame]61}
Paul Bakker33b43f12013-08-20 11:48:36 +0200[diff] [blame]62/* END_CASE */