Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 1 | /* BEGIN_HEADER */ |
Manuel Pégourié-Gonnard | 7f80997 | 2015-03-09 17:05:11 +0000 | [diff] [blame] | 2 | #include "mbedtls/pk.h" |
| 3 | #include "mbedtls/pem.h" |
| 4 | #include "mbedtls/oid.h" |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 5 | /* END_HEADER */ |
| 6 | |
| 7 | /* BEGIN_DEPENDENCIES |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 8 | * depends_on:MBEDTLS_PK_PARSE_C:MBEDTLS_BIGNUM_C |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 9 | * END_DEPENDENCIES |
| 10 | */ |
| 11 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 12 | /* BEGIN_CASE depends_on:MBEDTLS_RSA_C:MBEDTLS_FS_IO */ |
Azim Khan | f1aaec9 | 2017-05-30 14:23:15 +0100 | [diff] [blame] | 13 | void pk_parse_keyfile_rsa( char * key_file, char * password, int result ) |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 14 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 15 | mbedtls_pk_context ctx; |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 16 | int res; |
| 17 | char *pwd = password; |
| 18 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 19 | mbedtls_pk_init( &ctx ); |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 20 | |
| 21 | if( strcmp( pwd, "NULL" ) == 0 ) |
| 22 | pwd = NULL; |
| 23 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 24 | res = mbedtls_pk_parse_keyfile( &ctx, key_file, pwd ); |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 25 | |
| 26 | TEST_ASSERT( res == result ); |
| 27 | |
| 28 | if( res == 0 ) |
| 29 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 30 | mbedtls_rsa_context *rsa; |
| 31 | TEST_ASSERT( mbedtls_pk_can_do( &ctx, MBEDTLS_PK_RSA ) ); |
| 32 | rsa = mbedtls_pk_rsa( ctx ); |
| 33 | TEST_ASSERT( mbedtls_rsa_check_privkey( rsa ) == 0 ); |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 34 | } |
| 35 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 36 | exit: |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 37 | mbedtls_pk_free( &ctx ); |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 38 | } |
| 39 | /* END_CASE */ |
| 40 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 41 | /* BEGIN_CASE depends_on:MBEDTLS_RSA_C:MBEDTLS_FS_IO */ |
Azim Khan | f1aaec9 | 2017-05-30 14:23:15 +0100 | [diff] [blame] | 42 | void pk_parse_public_keyfile_rsa( char * key_file, int result ) |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 43 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 44 | mbedtls_pk_context ctx; |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 45 | int res; |
| 46 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 47 | mbedtls_pk_init( &ctx ); |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 48 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 49 | res = mbedtls_pk_parse_public_keyfile( &ctx, key_file ); |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 50 | |
| 51 | TEST_ASSERT( res == result ); |
| 52 | |
| 53 | if( res == 0 ) |
| 54 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 55 | mbedtls_rsa_context *rsa; |
| 56 | TEST_ASSERT( mbedtls_pk_can_do( &ctx, MBEDTLS_PK_RSA ) ); |
| 57 | rsa = mbedtls_pk_rsa( ctx ); |
| 58 | TEST_ASSERT( mbedtls_rsa_check_pubkey( rsa ) == 0 ); |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 59 | } |
| 60 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 61 | exit: |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 62 | mbedtls_pk_free( &ctx ); |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 63 | } |
| 64 | /* END_CASE */ |
| 65 | |
Hanno Becker | d84dbe5 | 2019-08-21 14:31:52 +0100 | [diff] [blame] | 66 | /* BEGIN_CASE depends_on:MBEDTLS_FS_IO */ |
Azim Khan | f1aaec9 | 2017-05-30 14:23:15 +0100 | [diff] [blame] | 67 | void pk_parse_public_keyfile_ec( char * key_file, int result ) |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 68 | { |
Hanno Becker | d84dbe5 | 2019-08-21 14:31:52 +0100 | [diff] [blame] | 69 | #if !defined(MBEDTLS_ECP_C) && !defined(MBEDTLS_USE_TINYCRYPT) |
| 70 | ((void) key_file); |
| 71 | ((void) result); |
| 72 | #else |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 73 | mbedtls_pk_context ctx; |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 74 | int res; |
| 75 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 76 | mbedtls_pk_init( &ctx ); |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 77 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 78 | res = mbedtls_pk_parse_public_keyfile( &ctx, key_file ); |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 79 | |
| 80 | TEST_ASSERT( res == result ); |
| 81 | |
| 82 | if( res == 0 ) |
| 83 | { |
Hanno Becker | d84dbe5 | 2019-08-21 14:31:52 +0100 | [diff] [blame] | 84 | #if !defined(MBEDTLS_USE_TINYCRYPT) |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 85 | mbedtls_ecp_keypair *eckey; |
Hanno Becker | d84dbe5 | 2019-08-21 14:31:52 +0100 | [diff] [blame] | 86 | #else |
| 87 | mbedtls_uecc_keypair *uecckey; |
| 88 | #endif |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 89 | TEST_ASSERT( mbedtls_pk_can_do( &ctx, MBEDTLS_PK_ECKEY ) ); |
Hanno Becker | d84dbe5 | 2019-08-21 14:31:52 +0100 | [diff] [blame] | 90 | |
| 91 | #if !defined(MBEDTLS_USE_TINYCRYPT) |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 92 | eckey = mbedtls_pk_ec( ctx ); |
| 93 | TEST_ASSERT( mbedtls_ecp_check_pubkey( &eckey->grp, &eckey->Q ) == 0 ); |
Hanno Becker | d84dbe5 | 2019-08-21 14:31:52 +0100 | [diff] [blame] | 94 | #else |
| 95 | uecckey = mbedtls_pk_uecc( ctx ); |
| 96 | TEST_ASSERT( uECC_valid_public_key( uecckey->public_key, |
| 97 | uECC_secp256r1() ) == 0 ); |
| 98 | #endif /* MBEDTLS_USE_TINYCRYPT */ |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 99 | } |
| 100 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 101 | exit: |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 102 | mbedtls_pk_free( &ctx ); |
Hanno Becker | d84dbe5 | 2019-08-21 14:31:52 +0100 | [diff] [blame] | 103 | #endif /* !MBEDTLS_ECP_C && !MBEDTLS_USE_TINYCRYPT */ |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 104 | } |
| 105 | /* END_CASE */ |
| 106 | |
Hanno Becker | d84dbe5 | 2019-08-21 14:31:52 +0100 | [diff] [blame] | 107 | /* BEGIN_CASE depends_on:MBEDTLS_FS_IO */ |
Azim Khan | f1aaec9 | 2017-05-30 14:23:15 +0100 | [diff] [blame] | 108 | void pk_parse_keyfile_ec( char * key_file, char * password, int result ) |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 109 | { |
Hanno Becker | d84dbe5 | 2019-08-21 14:31:52 +0100 | [diff] [blame] | 110 | #if !defined(MBEDTLS_ECP_C) && !defined(MBEDTLS_USE_TINYCRYPT) |
| 111 | ((void) key_file); |
| 112 | ((void) password); |
| 113 | ((void) result); |
| 114 | #else |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 115 | mbedtls_pk_context ctx; |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 116 | int res; |
| 117 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 118 | mbedtls_pk_init( &ctx ); |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 119 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 120 | res = mbedtls_pk_parse_keyfile( &ctx, key_file, password ); |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 121 | |
| 122 | TEST_ASSERT( res == result ); |
| 123 | |
| 124 | if( res == 0 ) |
| 125 | { |
Hanno Becker | d84dbe5 | 2019-08-21 14:31:52 +0100 | [diff] [blame] | 126 | #if !defined(MBEDTLS_USE_TINYCRYPT) |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 127 | mbedtls_ecp_keypair *eckey; |
Hanno Becker | d84dbe5 | 2019-08-21 14:31:52 +0100 | [diff] [blame] | 128 | #else |
| 129 | mbedtls_uecc_keypair *uecckey; |
| 130 | unsigned char tmp_pubkey[ 2 * NUM_ECC_BYTES ]; |
| 131 | #endif |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 132 | TEST_ASSERT( mbedtls_pk_can_do( &ctx, MBEDTLS_PK_ECKEY ) ); |
Hanno Becker | d84dbe5 | 2019-08-21 14:31:52 +0100 | [diff] [blame] | 133 | |
| 134 | #if !defined(MBEDTLS_USE_TINYCRYPT) |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 135 | eckey = mbedtls_pk_ec( ctx ); |
| 136 | TEST_ASSERT( mbedtls_ecp_check_privkey( &eckey->grp, &eckey->d ) == 0 ); |
Hanno Becker | d84dbe5 | 2019-08-21 14:31:52 +0100 | [diff] [blame] | 137 | #else |
| 138 | uecckey = mbedtls_pk_uecc( ctx ); |
| 139 | TEST_ASSERT( uECC_valid_public_key( uecckey->public_key, |
| 140 | uECC_secp256r1() ) == 0 ); |
| 141 | TEST_ASSERT( uECC_compute_public_key( uecckey->private_key, |
| 142 | tmp_pubkey, |
| 143 | uECC_secp256r1() ) != 0 ); |
| 144 | TEST_ASSERT( memcmp( tmp_pubkey, uecckey->public_key, |
| 145 | sizeof( tmp_pubkey ) ) == 0 ); |
| 146 | #endif /* MBEDTLS_USE_TINYCRYPT */ |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 147 | } |
| 148 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 149 | exit: |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 150 | mbedtls_pk_free( &ctx ); |
Hanno Becker | d84dbe5 | 2019-08-21 14:31:52 +0100 | [diff] [blame] | 151 | #endif /* !MBEDTLS_ECP_C && !MBEDTLS_USE_TINYCRYPT */ |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 152 | } |
| 153 | /* END_CASE */ |
| 154 | |
Azim Khan | f1aaec9 | 2017-05-30 14:23:15 +0100 | [diff] [blame] | 155 | /* BEGIN_CASE depends_on:MBEDTLS_RSA_C */ |
Azim Khan | 5fcca46 | 2018-06-29 11:05:32 +0100 | [diff] [blame] | 156 | void pk_parse_key( data_t * buf, char * result_str, int result ) |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 157 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 158 | mbedtls_pk_context pk; |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 159 | unsigned char output[2000]; |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 160 | ((void) result_str); |
| 161 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 162 | mbedtls_pk_init( &pk ); |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 163 | |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 164 | memset( output, 0, 2000 ); |
| 165 | |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 166 | |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame] | 167 | TEST_ASSERT( mbedtls_pk_parse_key( &pk, buf->x, buf->len, NULL, 0 ) == ( result ) ); |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 168 | if( ( result ) == 0 ) |
| 169 | { |
| 170 | TEST_ASSERT( 1 ); |
| 171 | } |
| 172 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 173 | exit: |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 174 | mbedtls_pk_free( &pk ); |
Paul Bakker | 1a7550a | 2013-09-15 13:01:22 +0200 | [diff] [blame] | 175 | } |
| 176 | /* END_CASE */ |