Manuel Pégourié-Gonnard | 36178ff | 2014-05-29 14:26:03 +0200 | [diff] [blame] | 1 | /* BEGIN_HEADER */ |
Manuel Pégourié-Gonnard | 7f80997 | 2015-03-09 17:05:11 +0000 | [diff] [blame] | 2 | #include "mbedtls/asn1write.h" |
Manuel Pégourié-Gonnard | 36178ff | 2014-05-29 14:26:03 +0200 | [diff] [blame] | 3 | |
| 4 | #define GUARD_LEN 4 |
| 5 | #define GUARD_VAL 0x2a |
| 6 | /* END_HEADER */ |
| 7 | |
| 8 | /* BEGIN_DEPENDENCIES |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 9 | * depends_on:MBEDTLS_ASN1_WRITE_C |
Manuel Pégourié-Gonnard | 36178ff | 2014-05-29 14:26:03 +0200 | [diff] [blame] | 10 | * END_DEPENDENCIES |
| 11 | */ |
| 12 | |
| 13 | /* BEGIN_CASE */ |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame^] | 14 | void mbedtls_asn1_write_octet_string( HexParam_t * str, HexParam_t * asn1, |
Azim Khan | f1aaec9 | 2017-05-30 14:23:15 +0100 | [diff] [blame] | 15 | int buf_len, int result ) |
Manuel Pégourié-Gonnard | 36178ff | 2014-05-29 14:26:03 +0200 | [diff] [blame] | 16 | { |
| 17 | int ret; |
| 18 | unsigned char buf[150]; |
Azim Khan | 9079170 | 2017-05-30 00:57:11 +0100 | [diff] [blame] | 19 | size_t i; |
Manuel Pégourié-Gonnard | 36178ff | 2014-05-29 14:26:03 +0200 | [diff] [blame] | 20 | unsigned char *p; |
| 21 | |
| 22 | memset( buf, GUARD_VAL, sizeof( buf ) ); |
| 23 | |
Manuel Pégourié-Gonnard | 36178ff | 2014-05-29 14:26:03 +0200 | [diff] [blame] | 24 | |
| 25 | p = buf + GUARD_LEN + buf_len; |
| 26 | |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame^] | 27 | ret = mbedtls_asn1_write_octet_string( &p, buf + GUARD_LEN, str->x, str->len ); |
Manuel Pégourié-Gonnard | 36178ff | 2014-05-29 14:26:03 +0200 | [diff] [blame] | 28 | |
| 29 | /* Check for buffer overwrite on both sides */ |
| 30 | for( i = 0; i < GUARD_LEN; i++ ) |
| 31 | { |
| 32 | TEST_ASSERT( buf[i] == GUARD_VAL ); |
| 33 | TEST_ASSERT( buf[GUARD_LEN + buf_len + i] == GUARD_VAL ); |
| 34 | } |
| 35 | |
| 36 | if( result >= 0 ) |
| 37 | { |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame^] | 38 | TEST_ASSERT( (size_t) ret == asn1->len ); |
| 39 | TEST_ASSERT( p + asn1->len == buf + GUARD_LEN + buf_len ); |
Manuel Pégourié-Gonnard | 36178ff | 2014-05-29 14:26:03 +0200 | [diff] [blame] | 40 | |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame^] | 41 | TEST_ASSERT( memcmp( p, asn1->x, asn1->len ) == 0 ); |
Manuel Pégourié-Gonnard | 36178ff | 2014-05-29 14:26:03 +0200 | [diff] [blame] | 42 | } |
| 43 | } |
| 44 | /* END_CASE */ |
Manuel Pégourié-Gonnard | c22bb49 | 2014-05-29 17:16:45 +0200 | [diff] [blame] | 45 | |
| 46 | /* BEGIN_CASE */ |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame^] | 47 | void mbedtls_asn1_write_ia5_string( char * str, HexParam_t * asn1, |
| 48 | int buf_len, int result ) |
Manuel Pégourié-Gonnard | c22bb49 | 2014-05-29 17:16:45 +0200 | [diff] [blame] | 49 | { |
| 50 | int ret; |
| 51 | unsigned char buf[150]; |
Azim Khan | 9079170 | 2017-05-30 00:57:11 +0100 | [diff] [blame] | 52 | size_t str_len; |
Azim Khan | 9079170 | 2017-05-30 00:57:11 +0100 | [diff] [blame] | 53 | size_t i; |
Manuel Pégourié-Gonnard | c22bb49 | 2014-05-29 17:16:45 +0200 | [diff] [blame] | 54 | unsigned char *p; |
| 55 | |
| 56 | memset( buf, GUARD_VAL, sizeof( buf ) ); |
| 57 | |
| 58 | str_len = strlen( str ); |
Manuel Pégourié-Gonnard | c22bb49 | 2014-05-29 17:16:45 +0200 | [diff] [blame] | 59 | |
| 60 | p = buf + GUARD_LEN + buf_len; |
| 61 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 62 | ret = mbedtls_asn1_write_ia5_string( &p, buf + GUARD_LEN, str, str_len ); |
Manuel Pégourié-Gonnard | c22bb49 | 2014-05-29 17:16:45 +0200 | [diff] [blame] | 63 | |
| 64 | /* Check for buffer overwrite on both sides */ |
| 65 | for( i = 0; i < GUARD_LEN; i++ ) |
| 66 | { |
| 67 | TEST_ASSERT( buf[i] == GUARD_VAL ); |
| 68 | TEST_ASSERT( buf[GUARD_LEN + buf_len + i] == GUARD_VAL ); |
| 69 | } |
| 70 | |
| 71 | if( result >= 0 ) |
| 72 | { |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame^] | 73 | TEST_ASSERT( (size_t) ret == asn1->len ); |
| 74 | TEST_ASSERT( p + asn1->len == buf + GUARD_LEN + buf_len ); |
Manuel Pégourié-Gonnard | c22bb49 | 2014-05-29 17:16:45 +0200 | [diff] [blame] | 75 | |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame^] | 76 | TEST_ASSERT( memcmp( p, asn1->x, asn1->len ) == 0 ); |
Manuel Pégourié-Gonnard | c22bb49 | 2014-05-29 17:16:45 +0200 | [diff] [blame] | 77 | } |
| 78 | } |
| 79 | /* END_CASE */ |
Paul Bakker | e325db9 | 2016-07-14 10:27:36 +0100 | [diff] [blame] | 80 | |
| 81 | /* BEGIN_CASE */ |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame^] | 82 | void mbedtls_asn1_write_len( int len, HexParam_t * asn1, int buf_len, |
| 83 | int result ) |
Paul Bakker | e325db9 | 2016-07-14 10:27:36 +0100 | [diff] [blame] | 84 | { |
| 85 | int ret; |
| 86 | unsigned char buf[150]; |
Paul Bakker | e325db9 | 2016-07-14 10:27:36 +0100 | [diff] [blame] | 87 | unsigned char *p; |
Azim Khan | 9079170 | 2017-05-30 00:57:11 +0100 | [diff] [blame] | 88 | size_t i; |
| 89 | size_t read_len; |
Paul Bakker | e325db9 | 2016-07-14 10:27:36 +0100 | [diff] [blame] | 90 | |
| 91 | memset( buf, GUARD_VAL, sizeof( buf ) ); |
Paul Bakker | e325db9 | 2016-07-14 10:27:36 +0100 | [diff] [blame] | 92 | |
Paul Bakker | 58bfb83 | 2016-07-14 11:02:31 +0100 | [diff] [blame] | 93 | p = buf + GUARD_LEN + buf_len; |
Paul Bakker | e325db9 | 2016-07-14 10:27:36 +0100 | [diff] [blame] | 94 | |
| 95 | ret = mbedtls_asn1_write_len( &p, buf + GUARD_LEN, (size_t) len ); |
| 96 | |
| 97 | TEST_ASSERT( ret == result ); |
| 98 | |
| 99 | /* Check for buffer overwrite on both sides */ |
| 100 | for( i = 0; i < GUARD_LEN; i++ ) |
| 101 | { |
| 102 | TEST_ASSERT( buf[i] == GUARD_VAL ); |
Paul Bakker | 58bfb83 | 2016-07-14 11:02:31 +0100 | [diff] [blame] | 103 | TEST_ASSERT( buf[GUARD_LEN + buf_len + i] == GUARD_VAL ); |
Paul Bakker | e325db9 | 2016-07-14 10:27:36 +0100 | [diff] [blame] | 104 | } |
| 105 | |
| 106 | if( result >= 0 ) |
| 107 | { |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame^] | 108 | TEST_ASSERT( p + asn1->len == buf + GUARD_LEN + buf_len ); |
Paul Bakker | e325db9 | 2016-07-14 10:27:36 +0100 | [diff] [blame] | 109 | |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame^] | 110 | TEST_ASSERT( memcmp( p, asn1->x, asn1->len ) == 0 ); |
Paul Bakker | 5e8b77c | 2016-07-14 11:14:54 +0100 | [diff] [blame] | 111 | |
| 112 | /* Read back with mbedtls_asn1_get_len() to check */ |
| 113 | ret = mbedtls_asn1_get_len( &p, buf + GUARD_LEN + buf_len, &read_len ); |
| 114 | |
| 115 | if( len == 0 ) |
| 116 | { |
| 117 | TEST_ASSERT( ret == 0 ); |
| 118 | } |
| 119 | else |
| 120 | { |
| 121 | /* Return will be MBEDTLS_ERR_ASN1_OUT_OF_DATA because the rest of |
| 122 | * the buffer is missing |
| 123 | */ |
| 124 | TEST_ASSERT( ret == MBEDTLS_ERR_ASN1_OUT_OF_DATA ); |
| 125 | } |
| 126 | TEST_ASSERT( read_len == (size_t) len ); |
| 127 | TEST_ASSERT( p == buf + GUARD_LEN + buf_len ); |
Paul Bakker | e325db9 | 2016-07-14 10:27:36 +0100 | [diff] [blame] | 128 | } |
| 129 | } |
| 130 | /* END_CASE */ |