Paul Bakker | 1630058 | 2014-04-11 13:28:43 +0200 | [diff] [blame] | 1 | /* BEGIN_HEADER */ |
Rich Evans | ce2f237 | 2015-02-06 13:57:42 +0000 | [diff] [blame] | 2 | #include "polarssl/base64.h" |
| 3 | #include "polarssl/pem.h" |
Paul Bakker | 1630058 | 2014-04-11 13:28:43 +0200 | [diff] [blame] | 4 | /* END_HEADER */ |
| 5 | |
Andres AG | 480f7e7 | 2016-10-24 14:31:54 +0100 | [diff] [blame] | 6 | /* BEGIN_CASE depends_on:POLARSSL_PEM_WRITE_C */ |
Paul Bakker | 1630058 | 2014-04-11 13:28:43 +0200 | [diff] [blame] | 7 | void pem_write_buffer( char *start, char *end, char *buf_str, char *result_str ) |
| 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 | |
| 18 | ret = pem_write_buffer( start, end, buf, buf_len, NULL, 0, &olen ); |
| 19 | TEST_ASSERT( ret == POLARSSL_ERR_BASE64_BUFFER_TOO_SMALL ); |
| 20 | |
| 21 | check_buf = (unsigned char *) polarssl_malloc( olen ); |
| 22 | TEST_ASSERT( check_buf != NULL ); |
| 23 | |
| 24 | memset( check_buf, 0, olen ); |
| 25 | ret = pem_write_buffer( start, end, buf, buf_len, check_buf, olen, &olen2 ); |
| 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: |
Paul Bakker | 1630058 | 2014-04-11 13:28:43 +0200 | [diff] [blame] | 33 | polarssl_free( check_buf ); |
| 34 | } |
| 35 | /* END_CASE */ |
Andres AG | 480f7e7 | 2016-10-24 14:31:54 +0100 | [diff] [blame] | 36 | |
| 37 | /* BEGIN_CASE depends_on:POLARSSL_PEM_PARSE_C:POLARSSL_AES_C:POLARSSL_DES_C:POLARSSL_MD5_C:POLARSSL_CIPHER_MODE_CBC */ |
| 38 | void pem_read_buffer( char *header, char *footer, char *data, int ret ) |
| 39 | { |
| 40 | pem_context ctx; |
| 41 | size_t use_len = 0; |
| 42 | |
| 43 | pem_init( &ctx ); |
| 44 | |
| 45 | TEST_ASSERT( pem_read_buffer( &ctx, header, footer, |
| 46 | (const unsigned char *)data, NULL, 0, |
| 47 | &use_len ) == ret ); |
| 48 | |
| 49 | exit: |
| 50 | pem_free( &ctx ); |
| 51 | } |
| 52 | /* END_CASE */ |