Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +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" |
Paul Elliott | 448d546 | 2021-02-24 15:32:42 +0000 | [diff] [blame^] | 3 | #include <test/constant_flow.h> |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 4 | /* END_HEADER */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 5 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 6 | /* BEGIN_DEPENDENCIES |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 7 | * depends_on:MBEDTLS_BASE64_C |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 8 | * END_DEPENDENCIES |
| 9 | */ |
Paul Bakker | 5690efc | 2011-05-26 13:16:06 +0000 | [diff] [blame] | 10 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 11 | /* BEGIN_CASE */ |
Azim Khan | f1aaec9 | 2017-05-30 14:23:15 +0100 | [diff] [blame] | 12 | void mbedtls_base64_encode( char * src_string, char * dst_string, |
| 13 | int dst_buf_size, int result ) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 14 | { |
| 15 | unsigned char src_str[1000]; |
| 16 | unsigned char dst_str[1000]; |
Paul Elliott | 448d546 | 2021-02-24 15:32:42 +0000 | [diff] [blame^] | 17 | size_t len, src_len; |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 18 | |
| 19 | memset(src_str, 0x00, 1000); |
| 20 | memset(dst_str, 0x00, 1000); |
| 21 | |
Paul Bakker | dd0aae9 | 2014-04-17 16:06:37 +0200 | [diff] [blame] | 22 | strncpy( (char *) src_str, src_string, sizeof(src_str) - 1 ); |
Paul Elliott | 448d546 | 2021-02-24 15:32:42 +0000 | [diff] [blame^] | 23 | src_len = strlen( (char *) src_str ); |
| 24 | |
| 25 | TEST_CF_SECRET( src_str, sizeof( src_str ) ); |
| 26 | TEST_ASSERT( mbedtls_base64_encode( dst_str, dst_buf_size, &len, src_str, src_len) == result ); |
| 27 | TEST_CF_PUBLIC( src_str, sizeof( src_str ) ); |
| 28 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 29 | if( result == 0 ) |
Paul Bakker | 5946fd9 | 2009-07-11 15:29:30 +0000 | [diff] [blame] | 30 | { |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 31 | TEST_ASSERT( strcmp( (char *) dst_str, dst_string ) == 0 ); |
Paul Bakker | 5946fd9 | 2009-07-11 15:29:30 +0000 | [diff] [blame] | 32 | } |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 33 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 34 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 35 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 36 | /* BEGIN_CASE */ |
Azim Khan | f1aaec9 | 2017-05-30 14:23:15 +0100 | [diff] [blame] | 37 | void mbedtls_base64_decode( char * src_string, char * dst_string, int result ) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 38 | { |
| 39 | unsigned char src_str[1000]; |
| 40 | unsigned char dst_str[1000]; |
Manuel Pégourié-Gonnard | ba56136 | 2015-06-02 16:30:35 +0100 | [diff] [blame] | 41 | size_t len; |
Manuel Pégourié-Gonnard | 64938c6 | 2014-10-15 21:45:39 +0200 | [diff] [blame] | 42 | int res; |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 43 | |
| 44 | memset(src_str, 0x00, 1000); |
| 45 | memset(dst_str, 0x00, 1000); |
Paul Bakker | dd0aae9 | 2014-04-17 16:06:37 +0200 | [diff] [blame] | 46 | |
| 47 | strncpy( (char *) src_str, src_string, sizeof(src_str) - 1 ); |
Manuel Pégourié-Gonnard | ba56136 | 2015-06-02 16:30:35 +0100 | [diff] [blame] | 48 | res = mbedtls_base64_decode( dst_str, sizeof( dst_str ), &len, src_str, strlen( (char *) src_str ) ); |
Paul Bakker | 94b916c | 2014-04-17 16:07:20 +0200 | [diff] [blame] | 49 | TEST_ASSERT( res == result ); |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 50 | if( result == 0 ) |
Paul Bakker | 5946fd9 | 2009-07-11 15:29:30 +0000 | [diff] [blame] | 51 | { |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 52 | TEST_ASSERT( strcmp( (char *) dst_str, dst_string ) == 0 ); |
Paul Bakker | 5946fd9 | 2009-07-11 15:29:30 +0000 | [diff] [blame] | 53 | } |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 54 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 55 | /* END_CASE */ |
Paul Bakker | 3d36082 | 2009-07-05 11:29:38 +0000 | [diff] [blame] | 56 | |
Paul Bakker | d598318 | 2014-07-04 13:50:31 +0200 | [diff] [blame] | 57 | /* BEGIN_CASE */ |
Azim Khan | 5fcca46 | 2018-06-29 11:05:32 +0100 | [diff] [blame] | 58 | void base64_encode_hex( data_t * src, char * dst, int dst_buf_size, |
Paul Bakker | d598318 | 2014-07-04 13:50:31 +0200 | [diff] [blame] | 59 | int result ) |
| 60 | { |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame] | 61 | unsigned char *res = NULL; |
| 62 | size_t len; |
Paul Bakker | d598318 | 2014-07-04 13:50:31 +0200 | [diff] [blame] | 63 | |
Ronald Cron | 690f3eb | 2020-06-10 10:42:18 +0200 | [diff] [blame] | 64 | res = mbedtls_test_zero_alloc( dst_buf_size ); |
Paul Bakker | d598318 | 2014-07-04 13:50:31 +0200 | [diff] [blame] | 65 | |
Paul Elliott | 448d546 | 2021-02-24 15:32:42 +0000 | [diff] [blame^] | 66 | TEST_CF_SECRET( src->x, src->len ); |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame] | 67 | TEST_ASSERT( mbedtls_base64_encode( res, dst_buf_size, &len, src->x, src->len ) == result ); |
Paul Elliott | 448d546 | 2021-02-24 15:32:42 +0000 | [diff] [blame^] | 68 | TEST_CF_PUBLIC( src->x, src->len ); |
| 69 | |
Paul Bakker | d598318 | 2014-07-04 13:50:31 +0200 | [diff] [blame] | 70 | if( result == 0 ) |
| 71 | { |
| 72 | TEST_ASSERT( len == strlen( dst ) ); |
| 73 | TEST_ASSERT( memcmp( dst, res, len ) == 0 ); |
| 74 | } |
Paul Bakker | 6697b6c | 2014-07-04 18:35:50 +0200 | [diff] [blame] | 75 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 76 | exit: |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 77 | mbedtls_free( res ); |
Paul Bakker | d598318 | 2014-07-04 13:50:31 +0200 | [diff] [blame] | 78 | } |
| 79 | /* END_CASE */ |
| 80 | |
| 81 | /* BEGIN_CASE */ |
Azim Khan | 5fcca46 | 2018-06-29 11:05:32 +0100 | [diff] [blame] | 82 | void base64_decode_hex( char * src, data_t * dst, int dst_buf_size, |
Paul Bakker | d598318 | 2014-07-04 13:50:31 +0200 | [diff] [blame] | 83 | int result ) |
| 84 | { |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame] | 85 | unsigned char *res = NULL; |
| 86 | size_t len; |
Paul Bakker | d598318 | 2014-07-04 13:50:31 +0200 | [diff] [blame] | 87 | |
Ronald Cron | 690f3eb | 2020-06-10 10:42:18 +0200 | [diff] [blame] | 88 | res = mbedtls_test_zero_alloc( dst_buf_size ); |
Paul Bakker | d598318 | 2014-07-04 13:50:31 +0200 | [diff] [blame] | 89 | |
Manuel Pégourié-Gonnard | ba56136 | 2015-06-02 16:30:35 +0100 | [diff] [blame] | 90 | TEST_ASSERT( mbedtls_base64_decode( res, dst_buf_size, &len, (unsigned char *) src, |
Paul Bakker | d598318 | 2014-07-04 13:50:31 +0200 | [diff] [blame] | 91 | strlen( src ) ) == result ); |
| 92 | if( result == 0 ) |
| 93 | { |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame] | 94 | TEST_ASSERT( len == dst->len ); |
| 95 | TEST_ASSERT( memcmp( dst->x, res, len ) == 0 ); |
Paul Bakker | d598318 | 2014-07-04 13:50:31 +0200 | [diff] [blame] | 96 | } |
Paul Bakker | 6697b6c | 2014-07-04 18:35:50 +0200 | [diff] [blame] | 97 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 98 | exit: |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 99 | mbedtls_free( res ); |
Paul Bakker | d598318 | 2014-07-04 13:50:31 +0200 | [diff] [blame] | 100 | } |
| 101 | /* END_CASE */ |
| 102 | |
Manuel Pégourié-Gonnard | 64938c6 | 2014-10-15 21:45:39 +0200 | [diff] [blame] | 103 | /* BEGIN_CASE */ |
Azim Khan | 5fcca46 | 2018-06-29 11:05:32 +0100 | [diff] [blame] | 104 | void base64_decode_hex_src( data_t * src, char * dst_ref, int result ) |
Manuel Pégourié-Gonnard | 64938c6 | 2014-10-15 21:45:39 +0200 | [diff] [blame] | 105 | { |
| 106 | unsigned char dst[1000] = { 0 }; |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame] | 107 | size_t len; |
Manuel Pégourié-Gonnard | 64938c6 | 2014-10-15 21:45:39 +0200 | [diff] [blame] | 108 | |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame] | 109 | TEST_ASSERT( mbedtls_base64_decode( dst, sizeof( dst ), &len, src->x, src->len ) == result ); |
Manuel Pégourié-Gonnard | 64938c6 | 2014-10-15 21:45:39 +0200 | [diff] [blame] | 110 | if( result == 0 ) |
| 111 | { |
| 112 | TEST_ASSERT( len == strlen( dst_ref ) ); |
| 113 | TEST_ASSERT( memcmp( dst, dst_ref, len ) == 0 ); |
| 114 | } |
| 115 | |
| 116 | exit: |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame] | 117 | ;; |
Manuel Pégourié-Gonnard | 64938c6 | 2014-10-15 21:45:39 +0200 | [diff] [blame] | 118 | } |
| 119 | /* END_CASE */ |
| 120 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 121 | /* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */ |
Azim Khan | f1aaec9 | 2017-05-30 14:23:15 +0100 | [diff] [blame] | 122 | void base64_selftest( ) |
Paul Bakker | 3d36082 | 2009-07-05 11:29:38 +0000 | [diff] [blame] | 123 | { |
Andres AG | 93012e8 | 2016-09-09 09:10:28 +0100 | [diff] [blame] | 124 | TEST_ASSERT( mbedtls_base64_self_test( 1 ) == 0 ); |
Paul Bakker | 3d36082 | 2009-07-05 11:29:38 +0000 | [diff] [blame] | 125 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 126 | /* END_CASE */ |