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 | f1aaec9 | 2017-05-30 14:23:15 +0100 | [diff] [blame] | 14 | void mbedtls_asn1_write_octet_string( uint8_t * str, uint32_t str_len, |
| 15 | uint8_t * asn1, uint32_t asn1_len, |
| 16 | int buf_len, int result ) |
Manuel Pégourié-Gonnard | 36178ff | 2014-05-29 14:26:03 +0200 | [diff] [blame] | 17 | { |
| 18 | int ret; |
| 19 | unsigned char buf[150]; |
Azim Khan | 9079170 | 2017-05-30 00:57:11 +0100 | [diff] [blame] | 20 | size_t i; |
Manuel Pégourié-Gonnard | 36178ff | 2014-05-29 14:26:03 +0200 | [diff] [blame] | 21 | unsigned char *p; |
| 22 | |
| 23 | memset( buf, GUARD_VAL, sizeof( buf ) ); |
| 24 | |
Manuel Pégourié-Gonnard | 36178ff | 2014-05-29 14:26:03 +0200 | [diff] [blame] | 25 | |
| 26 | p = buf + GUARD_LEN + buf_len; |
| 27 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 28 | ret = mbedtls_asn1_write_octet_string( &p, buf + GUARD_LEN, str, str_len ); |
Manuel Pégourié-Gonnard | 36178ff | 2014-05-29 14:26:03 +0200 | [diff] [blame] | 29 | |
| 30 | /* Check for buffer overwrite on both sides */ |
| 31 | for( i = 0; i < GUARD_LEN; i++ ) |
| 32 | { |
| 33 | TEST_ASSERT( buf[i] == GUARD_VAL ); |
| 34 | TEST_ASSERT( buf[GUARD_LEN + buf_len + i] == GUARD_VAL ); |
| 35 | } |
| 36 | |
| 37 | if( result >= 0 ) |
| 38 | { |
Manuel Pégourié-Gonnard | 36178ff | 2014-05-29 14:26:03 +0200 | [diff] [blame] | 39 | TEST_ASSERT( p + asn1_len == buf + GUARD_LEN + buf_len ); |
| 40 | |
| 41 | TEST_ASSERT( memcmp( p, asn1, asn1_len ) == 0 ); |
| 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 | f1aaec9 | 2017-05-30 14:23:15 +0100 | [diff] [blame] | 47 | void mbedtls_asn1_write_ia5_string( char * str, uint8_t * asn1, |
| 48 | uint32_t asn1_len, int buf_len, int result |
| 49 | ) |
Manuel Pégourié-Gonnard | c22bb49 | 2014-05-29 17:16:45 +0200 | [diff] [blame] | 50 | { |
| 51 | int ret; |
| 52 | unsigned char buf[150]; |
Azim Khan | 9079170 | 2017-05-30 00:57:11 +0100 | [diff] [blame] | 53 | size_t str_len; |
Azim Khan | 9079170 | 2017-05-30 00:57:11 +0100 | [diff] [blame] | 54 | size_t i; |
Manuel Pégourié-Gonnard | c22bb49 | 2014-05-29 17:16:45 +0200 | [diff] [blame] | 55 | unsigned char *p; |
| 56 | |
| 57 | memset( buf, GUARD_VAL, sizeof( buf ) ); |
| 58 | |
| 59 | str_len = strlen( str ); |
Manuel Pégourié-Gonnard | c22bb49 | 2014-05-29 17:16:45 +0200 | [diff] [blame] | 60 | |
| 61 | p = buf + GUARD_LEN + buf_len; |
| 62 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 63 | 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] | 64 | |
| 65 | /* Check for buffer overwrite on both sides */ |
| 66 | for( i = 0; i < GUARD_LEN; i++ ) |
| 67 | { |
| 68 | TEST_ASSERT( buf[i] == GUARD_VAL ); |
| 69 | TEST_ASSERT( buf[GUARD_LEN + buf_len + i] == GUARD_VAL ); |
| 70 | } |
| 71 | |
| 72 | if( result >= 0 ) |
| 73 | { |
Manuel Pégourié-Gonnard | c22bb49 | 2014-05-29 17:16:45 +0200 | [diff] [blame] | 74 | TEST_ASSERT( p + asn1_len == buf + GUARD_LEN + buf_len ); |
| 75 | |
| 76 | TEST_ASSERT( memcmp( p, asn1, asn1_len ) == 0 ); |
| 77 | } |
| 78 | } |
| 79 | /* END_CASE */ |
Paul Bakker | e325db9 | 2016-07-14 10:27:36 +0100 | [diff] [blame] | 80 | |
| 81 | /* BEGIN_CASE */ |
Azim Khan | f1aaec9 | 2017-05-30 14:23:15 +0100 | [diff] [blame] | 82 | void mbedtls_asn1_write_len( int len, uint8_t * asn1, uint32_t asn1_len, |
| 83 | int buf_len, 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 | { |
Paul Bakker | 58bfb83 | 2016-07-14 11:02:31 +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 | |
| 110 | TEST_ASSERT( memcmp( p, asn1, 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 */ |