Paul Bakker | 1630058 | 2014-04-11 13:28:43 +0200 | [diff] [blame] | 1 | /* BEGIN_HEADER */ |
| 2 | #include <polarssl/base64.h> |
| 3 | #include <polarssl/pem.h> |
| 4 | /* END_HEADER */ |
| 5 | |
| 6 | /* BEGIN_DEPENDENCIES |
| 7 | * depends_on:POLARSSL_PEM_WRITE_C |
| 8 | * END_DEPENDENCIES |
| 9 | */ |
| 10 | |
| 11 | /* BEGIN_CASE */ |
| 12 | void pem_write_buffer( char *start, char *end, char *buf_str, char *result_str ) |
| 13 | { |
| 14 | unsigned char buf[5000]; |
| 15 | unsigned char *check_buf; |
| 16 | int ret; |
| 17 | size_t buf_len, olen = 0, olen2 = 0; |
| 18 | |
| 19 | memset( buf, 0, sizeof( buf ) ); |
| 20 | |
| 21 | buf_len = unhexify( buf, buf_str ); |
| 22 | |
| 23 | ret = pem_write_buffer( start, end, buf, buf_len, NULL, 0, &olen ); |
| 24 | TEST_ASSERT( ret == POLARSSL_ERR_BASE64_BUFFER_TOO_SMALL ); |
| 25 | |
| 26 | check_buf = (unsigned char *) polarssl_malloc( olen ); |
| 27 | TEST_ASSERT( check_buf != NULL ); |
| 28 | |
| 29 | memset( check_buf, 0, olen ); |
| 30 | ret = pem_write_buffer( start, end, buf, buf_len, check_buf, olen, &olen2 ); |
| 31 | |
| 32 | TEST_ASSERT( olen2 <= olen ); |
| 33 | TEST_ASSERT( olen > strlen( (char*) result_str ) ); |
| 34 | TEST_ASSERT( ret == 0 ); |
| 35 | TEST_ASSERT( strncmp( (char *) check_buf, (char *) result_str, olen ) == 0 ); |
| 36 | polarssl_free( check_buf ); |
| 37 | } |
| 38 | /* END_CASE */ |