| /* BEGIN_HEADER */ |
| #include "mbedtls/pk.h" |
| #include "mbedtls/pem.h" |
| #include "mbedtls/oid.h" |
| /* END_HEADER */ |
| |
| /* BEGIN_DEPENDENCIES |
| * depends_on:MBEDTLS_PK_PARSE_C:MBEDTLS_PK_WRITE_C:MBEDTLS_BIGNUM_C:MBEDTLS_FS_IO |
| * END_DEPENDENCIES |
| */ |
| |
| /* BEGIN_CASE depends_on:MBEDTLS_PEM_WRITE_C */ |
| void pk_write_pubkey_check( char * key_file ) |
| { |
| mbedtls_pk_context key; |
| unsigned char buf[5000]; |
| unsigned char check_buf[5000]; |
| int ret; |
| FILE *f; |
| size_t ilen, pem_len, buf_index; |
| |
| memset( buf, 0, sizeof( buf ) ); |
| memset( check_buf, 0, sizeof( check_buf ) ); |
| |
| mbedtls_pk_init( &key ); |
| TEST_ASSERT( mbedtls_pk_parse_public_keyfile( &key, key_file ) == 0 ); |
| |
| ret = mbedtls_pk_write_pubkey_pem( &key, buf, sizeof( buf )); |
| TEST_ASSERT( ret == 0 ); |
| |
| pem_len = strlen( (char *) buf ); |
| |
| // check that the rest of the buffer remains clear |
| for( buf_index = pem_len; buf_index < sizeof( buf ); ++buf_index ) |
| { |
| TEST_ASSERT( buf[buf_index] == 0 ); |
| } |
| |
| f = fopen( key_file, "r" ); |
| TEST_ASSERT( f != NULL ); |
| ilen = fread( check_buf, 1, sizeof( check_buf ), f ); |
| fclose( f ); |
| |
| TEST_ASSERT( ilen == pem_len ); |
| TEST_ASSERT( memcmp( (char *) buf, (char *) check_buf, ilen ) == 0 ); |
| |
| exit: |
| mbedtls_pk_free( &key ); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE depends_on:MBEDTLS_PEM_WRITE_C */ |
| void pk_write_key_check( char * key_file ) |
| { |
| mbedtls_pk_context key; |
| unsigned char buf[5000]; |
| unsigned char check_buf[5000]; |
| int ret; |
| FILE *f; |
| size_t ilen, pem_len, buf_index; |
| |
| memset( buf, 0, sizeof( buf ) ); |
| memset( check_buf, 0, sizeof( check_buf ) ); |
| |
| mbedtls_pk_init( &key ); |
| TEST_ASSERT( mbedtls_pk_parse_keyfile( &key, key_file, NULL, |
| mbedtls_test_rnd_std_rand, NULL ) == 0 ); |
| |
| ret = mbedtls_pk_write_key_pem( &key, buf, sizeof( buf )); |
| TEST_ASSERT( ret == 0 ); |
| |
| pem_len = strlen( (char *) buf ); |
| |
| // check that the rest of the buffer remains clear |
| for( buf_index = pem_len; buf_index < sizeof( buf ); ++buf_index ) |
| { |
| TEST_ASSERT( buf[buf_index] == 0 ); |
| } |
| |
| f = fopen( key_file, "r" ); |
| TEST_ASSERT( f != NULL ); |
| ilen = fread( check_buf, 1, sizeof( check_buf ), f ); |
| fclose( f ); |
| |
| TEST_ASSERT( ilen == strlen( (char *) buf ) ); |
| TEST_ASSERT( memcmp( (char *) buf, (char *) check_buf, ilen ) == 0 ); |
| |
| exit: |
| mbedtls_pk_free( &key ); |
| } |
| /* END_CASE */ |