Paul Bakker | 1630058 | 2014-04-11 13:28:43 +0200 | [diff] [blame] | 1 | /* BEGIN_HEADER */ |
Manuel Pégourié-Gonnard | 7f80997 | 2015-03-09 17:05:11 +0000 | [diff] [blame] | 2 | #include "mbedtls/base64.h" |
| 3 | #include "mbedtls/pem.h" |
Paul Bakker | 1630058 | 2014-04-11 13:28:43 +0200 | [diff] [blame] | 4 | /* END_HEADER */ |
| 5 | |
Andres AG | 9c94b69 | 2016-10-24 14:31:54 +0100 | [diff] [blame^] | 6 | /* BEGIN_CASE depends_on:MBEDTLS_PEM_WRITE_C */ |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 7 | void mbedtls_pem_write_buffer( char *start, char *end, char *buf_str, char *result_str ) |
Paul Bakker | 1630058 | 2014-04-11 13:28:43 +0200 | [diff] [blame] | 8 | { |
| 9 | unsigned char buf[5000]; |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 10 | unsigned char *check_buf = NULL; |
Paul Bakker | 1630058 | 2014-04-11 13:28:43 +0200 | [diff] [blame] | 11 | int ret; |
| 12 | size_t buf_len, olen = 0, olen2 = 0; |
| 13 | |
| 14 | memset( buf, 0, sizeof( buf ) ); |
| 15 | |
| 16 | buf_len = unhexify( buf, buf_str ); |
| 17 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 18 | ret = mbedtls_pem_write_buffer( start, end, buf, buf_len, NULL, 0, &olen ); |
| 19 | TEST_ASSERT( ret == MBEDTLS_ERR_BASE64_BUFFER_TOO_SMALL ); |
Paul Bakker | 1630058 | 2014-04-11 13:28:43 +0200 | [diff] [blame] | 20 | |
Manuel Pégourié-Gonnard | 7551cb9 | 2015-05-26 16:04:06 +0200 | [diff] [blame] | 21 | check_buf = (unsigned char *) mbedtls_calloc( 1, olen ); |
Paul Bakker | 1630058 | 2014-04-11 13:28:43 +0200 | [diff] [blame] | 22 | TEST_ASSERT( check_buf != NULL ); |
| 23 | |
| 24 | memset( check_buf, 0, olen ); |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 25 | ret = mbedtls_pem_write_buffer( start, end, buf, buf_len, check_buf, olen, &olen2 ); |
Paul Bakker | 1630058 | 2014-04-11 13:28:43 +0200 | [diff] [blame] | 26 | |
| 27 | TEST_ASSERT( olen2 <= olen ); |
| 28 | TEST_ASSERT( olen > strlen( (char*) result_str ) ); |
| 29 | TEST_ASSERT( ret == 0 ); |
| 30 | TEST_ASSERT( strncmp( (char *) check_buf, (char *) result_str, olen ) == 0 ); |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 31 | |
| 32 | exit: |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 33 | mbedtls_free( check_buf ); |
Paul Bakker | 1630058 | 2014-04-11 13:28:43 +0200 | [diff] [blame] | 34 | } |
| 35 | /* END_CASE */ |
Andres AG | 9c94b69 | 2016-10-24 14:31:54 +0100 | [diff] [blame^] | 36 | |
| 37 | /* BEGIN_CASE depends_on:MBEDTLS_PEM_PARSE_C:MBEDTLS_AES_C:MBEDTLS_DES_C:MBEDTLS_MD5_C:MBEDTLS_CIPHER_MODE_CBC */ |
| 38 | void mbedtls_pem_read_buffer( char *header, char *footer, char *data, int ret ) |
| 39 | { |
| 40 | mbedtls_pem_context ctx; |
| 41 | size_t use_len = 0; |
| 42 | |
| 43 | mbedtls_pem_init( &ctx ); |
| 44 | |
| 45 | TEST_ASSERT( mbedtls_pem_read_buffer( &ctx, header, footer, |
| 46 | (const unsigned char *)data, NULL, 0, |
| 47 | &use_len ) == ret ); |
| 48 | |
| 49 | exit: |
| 50 | mbedtls_pem_free( &ctx ); |
| 51 | } |
| 52 | /* END_CASE */ |