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