1a8e1175bSopenharmony_ci/* BEGIN_HEADER */
2a8e1175bSopenharmony_ci#include <errno.h>
3a8e1175bSopenharmony_ci#include <stdlib.h>
4a8e1175bSopenharmony_ci#include <limits.h>
5a8e1175bSopenharmony_ci
6a8e1175bSopenharmony_ci#include "mbedtls/bignum.h"
7a8e1175bSopenharmony_ci#include "mbedtls/asn1.h"
8a8e1175bSopenharmony_ci#if defined(MBEDTLS_ASN1_WRITE_C)
9a8e1175bSopenharmony_ci#include "mbedtls/asn1write.h"
10a8e1175bSopenharmony_ci#endif
11a8e1175bSopenharmony_ci
12a8e1175bSopenharmony_ci/* Used internally to report an error that indicates a bug in a parsing function. */
13a8e1175bSopenharmony_ci#define ERR_PARSE_INCONSISTENCY INT_MAX
14a8e1175bSopenharmony_ci
15a8e1175bSopenharmony_ci/* Use this magic value in some tests to indicate that the expected result
16a8e1175bSopenharmony_ci * should not be checked. */
17a8e1175bSopenharmony_ci#define UNPREDICTABLE_RESULT 0x5552
18a8e1175bSopenharmony_ci
19a8e1175bSopenharmony_cistatic int nested_parse(unsigned char **const p,
20a8e1175bSopenharmony_ci                        const unsigned char *const end)
21a8e1175bSopenharmony_ci{
22a8e1175bSopenharmony_ci    int ret;
23a8e1175bSopenharmony_ci    size_t len = 0;
24a8e1175bSopenharmony_ci    size_t len2 = 0;
25a8e1175bSopenharmony_ci    unsigned char *const start = *p;
26a8e1175bSopenharmony_ci    unsigned char *content_start;
27a8e1175bSopenharmony_ci    unsigned char tag;
28a8e1175bSopenharmony_ci
29a8e1175bSopenharmony_ci    /* First get the length, skipping over the tag. */
30a8e1175bSopenharmony_ci    content_start = start + 1;
31a8e1175bSopenharmony_ci    ret = mbedtls_asn1_get_len(&content_start, end, &len);
32a8e1175bSopenharmony_ci    TEST_ASSERT(content_start <= end);
33a8e1175bSopenharmony_ci    if (ret != 0) {
34a8e1175bSopenharmony_ci        return ret;
35a8e1175bSopenharmony_ci    }
36a8e1175bSopenharmony_ci
37a8e1175bSopenharmony_ci    /* Since we have a valid element start (tag and length), retrieve and
38a8e1175bSopenharmony_ci     * check the tag. */
39a8e1175bSopenharmony_ci    tag = start[0];
40a8e1175bSopenharmony_ci    TEST_EQUAL(mbedtls_asn1_get_tag(p, end, &len2, tag ^ 1),
41a8e1175bSopenharmony_ci               MBEDTLS_ERR_ASN1_UNEXPECTED_TAG);
42a8e1175bSopenharmony_ci    *p = start;
43a8e1175bSopenharmony_ci    TEST_EQUAL(mbedtls_asn1_get_tag(p, end, &len2, tag), 0);
44a8e1175bSopenharmony_ci    TEST_EQUAL(len, len2);
45a8e1175bSopenharmony_ci    TEST_ASSERT(*p == content_start);
46a8e1175bSopenharmony_ci    *p = content_start;
47a8e1175bSopenharmony_ci
48a8e1175bSopenharmony_ci    switch (tag & 0x1f) {
49a8e1175bSopenharmony_ci        case MBEDTLS_ASN1_BOOLEAN:
50a8e1175bSopenharmony_ci        {
51a8e1175bSopenharmony_ci            int val = -257;
52a8e1175bSopenharmony_ci            *p = start;
53a8e1175bSopenharmony_ci            ret = mbedtls_asn1_get_bool(p, end, &val);
54a8e1175bSopenharmony_ci            if (ret == 0) {
55a8e1175bSopenharmony_ci                TEST_ASSERT(val == 0 || val == 1);
56a8e1175bSopenharmony_ci            }
57a8e1175bSopenharmony_ci            break;
58a8e1175bSopenharmony_ci        }
59a8e1175bSopenharmony_ci
60a8e1175bSopenharmony_ci        case MBEDTLS_ASN1_INTEGER:
61a8e1175bSopenharmony_ci        {
62a8e1175bSopenharmony_ci#if defined(MBEDTLS_BIGNUM_C)
63a8e1175bSopenharmony_ci            mbedtls_mpi mpi;
64a8e1175bSopenharmony_ci            mbedtls_mpi_init(&mpi);
65a8e1175bSopenharmony_ci            *p = start;
66a8e1175bSopenharmony_ci            ret = mbedtls_asn1_get_mpi(p, end, &mpi);
67a8e1175bSopenharmony_ci            mbedtls_mpi_free(&mpi);
68a8e1175bSopenharmony_ci#else
69a8e1175bSopenharmony_ci            *p = start + 1;
70a8e1175bSopenharmony_ci            ret = mbedtls_asn1_get_len(p, end, &len);
71a8e1175bSopenharmony_ci            *p += len;
72a8e1175bSopenharmony_ci#endif
73a8e1175bSopenharmony_ci            /* If we're sure that the number fits in an int, also
74a8e1175bSopenharmony_ci             * call mbedtls_asn1_get_int(). */
75a8e1175bSopenharmony_ci            if (ret == 0 && len < sizeof(int)) {
76a8e1175bSopenharmony_ci                int val = -257;
77a8e1175bSopenharmony_ci                unsigned char *q = start;
78a8e1175bSopenharmony_ci                ret = mbedtls_asn1_get_int(&q, end, &val);
79a8e1175bSopenharmony_ci                TEST_ASSERT(*p == q);
80a8e1175bSopenharmony_ci            }
81a8e1175bSopenharmony_ci            break;
82a8e1175bSopenharmony_ci        }
83a8e1175bSopenharmony_ci
84a8e1175bSopenharmony_ci        case MBEDTLS_ASN1_BIT_STRING:
85a8e1175bSopenharmony_ci        {
86a8e1175bSopenharmony_ci            mbedtls_asn1_bitstring bs;
87a8e1175bSopenharmony_ci            *p = start;
88a8e1175bSopenharmony_ci            ret = mbedtls_asn1_get_bitstring(p, end, &bs);
89a8e1175bSopenharmony_ci            break;
90a8e1175bSopenharmony_ci        }
91a8e1175bSopenharmony_ci
92a8e1175bSopenharmony_ci        case MBEDTLS_ASN1_SEQUENCE:
93a8e1175bSopenharmony_ci        {
94a8e1175bSopenharmony_ci            while (*p <= end && *p < content_start + len && ret == 0) {
95a8e1175bSopenharmony_ci                ret = nested_parse(p, content_start + len);
96a8e1175bSopenharmony_ci            }
97a8e1175bSopenharmony_ci            break;
98a8e1175bSopenharmony_ci        }
99a8e1175bSopenharmony_ci
100a8e1175bSopenharmony_ci        case MBEDTLS_ASN1_OCTET_STRING:
101a8e1175bSopenharmony_ci        case MBEDTLS_ASN1_NULL:
102a8e1175bSopenharmony_ci        case MBEDTLS_ASN1_OID:
103a8e1175bSopenharmony_ci        case MBEDTLS_ASN1_UTF8_STRING:
104a8e1175bSopenharmony_ci        case MBEDTLS_ASN1_SET:
105a8e1175bSopenharmony_ci        case MBEDTLS_ASN1_PRINTABLE_STRING:
106a8e1175bSopenharmony_ci        case MBEDTLS_ASN1_T61_STRING:
107a8e1175bSopenharmony_ci        case MBEDTLS_ASN1_IA5_STRING:
108a8e1175bSopenharmony_ci        case MBEDTLS_ASN1_UTC_TIME:
109a8e1175bSopenharmony_ci        case MBEDTLS_ASN1_GENERALIZED_TIME:
110a8e1175bSopenharmony_ci        case MBEDTLS_ASN1_UNIVERSAL_STRING:
111a8e1175bSopenharmony_ci        case MBEDTLS_ASN1_BMP_STRING:
112a8e1175bSopenharmony_ci        default:
113a8e1175bSopenharmony_ci            /* No further testing implemented for this tag. */
114a8e1175bSopenharmony_ci            *p += len;
115a8e1175bSopenharmony_ci            return 0;
116a8e1175bSopenharmony_ci    }
117a8e1175bSopenharmony_ci
118a8e1175bSopenharmony_ci    TEST_ASSERT(*p <= end);
119a8e1175bSopenharmony_ci    return ret;
120a8e1175bSopenharmony_ci
121a8e1175bSopenharmony_ciexit:
122a8e1175bSopenharmony_ci    return ERR_PARSE_INCONSISTENCY;
123a8e1175bSopenharmony_ci}
124a8e1175bSopenharmony_ci
125a8e1175bSopenharmony_ciint get_len_step(const data_t *input, size_t buffer_size,
126a8e1175bSopenharmony_ci                 size_t actual_length)
127a8e1175bSopenharmony_ci{
128a8e1175bSopenharmony_ci    unsigned char *buf = NULL;
129a8e1175bSopenharmony_ci    unsigned char *p = NULL;
130a8e1175bSopenharmony_ci    unsigned char *end;
131a8e1175bSopenharmony_ci    size_t parsed_length;
132a8e1175bSopenharmony_ci    int ret;
133a8e1175bSopenharmony_ci
134a8e1175bSopenharmony_ci    mbedtls_test_set_step(buffer_size);
135a8e1175bSopenharmony_ci    /* Allocate a new buffer of exactly the length to parse each time.
136a8e1175bSopenharmony_ci     * This gives memory sanitizers a chance to catch buffer overreads. */
137a8e1175bSopenharmony_ci    if (buffer_size == 0) {
138a8e1175bSopenharmony_ci        TEST_CALLOC(buf, 1);
139a8e1175bSopenharmony_ci        end = buf + 1;
140a8e1175bSopenharmony_ci        p = end;
141a8e1175bSopenharmony_ci    } else {
142a8e1175bSopenharmony_ci        TEST_CALLOC_OR_SKIP(buf, buffer_size);
143a8e1175bSopenharmony_ci        if (buffer_size > input->len) {
144a8e1175bSopenharmony_ci            memcpy(buf, input->x, input->len);
145a8e1175bSopenharmony_ci            memset(buf + input->len, 'A', buffer_size - input->len);
146a8e1175bSopenharmony_ci        } else {
147a8e1175bSopenharmony_ci            memcpy(buf, input->x, buffer_size);
148a8e1175bSopenharmony_ci        }
149a8e1175bSopenharmony_ci        p = buf;
150a8e1175bSopenharmony_ci        end = buf + buffer_size;
151a8e1175bSopenharmony_ci    }
152a8e1175bSopenharmony_ci
153a8e1175bSopenharmony_ci    ret = mbedtls_asn1_get_len(&p, end, &parsed_length);
154a8e1175bSopenharmony_ci
155a8e1175bSopenharmony_ci    if (buffer_size >= input->len + actual_length) {
156a8e1175bSopenharmony_ci        TEST_EQUAL(ret, 0);
157a8e1175bSopenharmony_ci        TEST_ASSERT(p == buf + input->len);
158a8e1175bSopenharmony_ci        TEST_EQUAL(parsed_length, actual_length);
159a8e1175bSopenharmony_ci    } else {
160a8e1175bSopenharmony_ci        TEST_EQUAL(ret, MBEDTLS_ERR_ASN1_OUT_OF_DATA);
161a8e1175bSopenharmony_ci    }
162a8e1175bSopenharmony_ci    mbedtls_free(buf);
163a8e1175bSopenharmony_ci    return 1;
164a8e1175bSopenharmony_ci
165a8e1175bSopenharmony_ciexit:
166a8e1175bSopenharmony_ci    mbedtls_free(buf);
167a8e1175bSopenharmony_ci    return 0;
168a8e1175bSopenharmony_ci}
169a8e1175bSopenharmony_ci
170a8e1175bSopenharmony_citypedef struct {
171a8e1175bSopenharmony_ci    const unsigned char *input_start;
172a8e1175bSopenharmony_ci    const char *description;
173a8e1175bSopenharmony_ci} traverse_state_t;
174a8e1175bSopenharmony_ci
175a8e1175bSopenharmony_ci/* Value returned by traverse_callback if description runs out. */
176a8e1175bSopenharmony_ci#define RET_TRAVERSE_STOP 1
177a8e1175bSopenharmony_ci/* Value returned by traverse_callback if description has an invalid format
178a8e1175bSopenharmony_ci * (see traverse_sequence_of). */
179a8e1175bSopenharmony_ci#define RET_TRAVERSE_ERROR 2
180a8e1175bSopenharmony_ci
181a8e1175bSopenharmony_ci
182a8e1175bSopenharmony_cistatic int traverse_callback(void *ctx, int tag,
183a8e1175bSopenharmony_ci                             unsigned char *content, size_t len)
184a8e1175bSopenharmony_ci{
185a8e1175bSopenharmony_ci    traverse_state_t *state = ctx;
186a8e1175bSopenharmony_ci    size_t offset;
187a8e1175bSopenharmony_ci    const char *rest = state->description;
188a8e1175bSopenharmony_ci    unsigned long n;
189a8e1175bSopenharmony_ci
190a8e1175bSopenharmony_ci    TEST_ASSERT(content > state->input_start);
191a8e1175bSopenharmony_ci    offset = content - state->input_start;
192a8e1175bSopenharmony_ci    mbedtls_test_set_step(offset);
193a8e1175bSopenharmony_ci
194a8e1175bSopenharmony_ci    if (*rest == 0) {
195a8e1175bSopenharmony_ci        return RET_TRAVERSE_STOP;
196a8e1175bSopenharmony_ci    }
197a8e1175bSopenharmony_ci    n = strtoul(rest, (char **) &rest, 0);
198a8e1175bSopenharmony_ci    TEST_EQUAL(n, offset);
199a8e1175bSopenharmony_ci    TEST_EQUAL(*rest, ',');
200a8e1175bSopenharmony_ci    ++rest;
201a8e1175bSopenharmony_ci    n = strtoul(rest, (char **) &rest, 0);
202a8e1175bSopenharmony_ci    TEST_EQUAL(n, (unsigned) tag);
203a8e1175bSopenharmony_ci    TEST_EQUAL(*rest, ',');
204a8e1175bSopenharmony_ci    ++rest;
205a8e1175bSopenharmony_ci    n = strtoul(rest, (char **) &rest, 0);
206a8e1175bSopenharmony_ci    TEST_EQUAL(n, len);
207a8e1175bSopenharmony_ci    if (*rest == ',') {
208a8e1175bSopenharmony_ci        ++rest;
209a8e1175bSopenharmony_ci    }
210a8e1175bSopenharmony_ci
211a8e1175bSopenharmony_ci    state->description = rest;
212a8e1175bSopenharmony_ci    return 0;
213a8e1175bSopenharmony_ci
214a8e1175bSopenharmony_ciexit:
215a8e1175bSopenharmony_ci    return RET_TRAVERSE_ERROR;
216a8e1175bSopenharmony_ci}
217a8e1175bSopenharmony_ci
218a8e1175bSopenharmony_ci/* END_HEADER */
219a8e1175bSopenharmony_ci
220a8e1175bSopenharmony_ci/* BEGIN_DEPENDENCIES
221a8e1175bSopenharmony_ci * depends_on:MBEDTLS_ASN1_PARSE_C
222a8e1175bSopenharmony_ci * END_DEPENDENCIES
223a8e1175bSopenharmony_ci */
224a8e1175bSopenharmony_ci
225a8e1175bSopenharmony_ci/* BEGIN_CASE */
226a8e1175bSopenharmony_civoid parse_prefixes(const data_t *input,
227a8e1175bSopenharmony_ci                    int full_result,
228a8e1175bSopenharmony_ci                    int overfull_result)
229a8e1175bSopenharmony_ci{
230a8e1175bSopenharmony_ci    /* full_result: expected result from parsing the given string. */
231a8e1175bSopenharmony_ci    /* overfull_result: expected_result from parsing the given string plus
232a8e1175bSopenharmony_ci     * some trailing garbage. This may be UNPREDICTABLE_RESULT to accept
233a8e1175bSopenharmony_ci     * any result: use this for invalid inputs that may or may not become
234a8e1175bSopenharmony_ci     * valid depending on what the trailing garbage is. */
235a8e1175bSopenharmony_ci
236a8e1175bSopenharmony_ci    unsigned char *buf = NULL;
237a8e1175bSopenharmony_ci    unsigned char *p = NULL;
238a8e1175bSopenharmony_ci    size_t buffer_size;
239a8e1175bSopenharmony_ci    int ret;
240a8e1175bSopenharmony_ci
241a8e1175bSopenharmony_ci    /* Test every prefix of the input, except the empty string.
242a8e1175bSopenharmony_ci     * The first byte of the string is the tag. Without a tag byte,
243a8e1175bSopenharmony_ci     * we wouldn't know what to parse the input as.
244a8e1175bSopenharmony_ci     * Also test the input followed by an extra byte.
245a8e1175bSopenharmony_ci     */
246a8e1175bSopenharmony_ci    for (buffer_size = 1; buffer_size <= input->len + 1; buffer_size++) {
247a8e1175bSopenharmony_ci        mbedtls_test_set_step(buffer_size);
248a8e1175bSopenharmony_ci        /* Allocate a new buffer of exactly the length to parse each time.
249a8e1175bSopenharmony_ci         * This gives memory sanitizers a chance to catch buffer overreads. */
250a8e1175bSopenharmony_ci        TEST_CALLOC(buf, buffer_size);
251a8e1175bSopenharmony_ci        memcpy(buf, input->x, buffer_size);
252a8e1175bSopenharmony_ci        p = buf;
253a8e1175bSopenharmony_ci        ret = nested_parse(&p, buf + buffer_size);
254a8e1175bSopenharmony_ci
255a8e1175bSopenharmony_ci        if (ret == ERR_PARSE_INCONSISTENCY) {
256a8e1175bSopenharmony_ci            goto exit;
257a8e1175bSopenharmony_ci        }
258a8e1175bSopenharmony_ci        if (buffer_size < input->len) {
259a8e1175bSopenharmony_ci            TEST_EQUAL(ret, MBEDTLS_ERR_ASN1_OUT_OF_DATA);
260a8e1175bSopenharmony_ci        } else if (buffer_size == input->len) {
261a8e1175bSopenharmony_ci            TEST_EQUAL(ret, full_result);
262a8e1175bSopenharmony_ci        } else { /* ( buffer_size > input->len ) */
263a8e1175bSopenharmony_ci            if (overfull_result != UNPREDICTABLE_RESULT) {
264a8e1175bSopenharmony_ci                TEST_EQUAL(ret, overfull_result);
265a8e1175bSopenharmony_ci            }
266a8e1175bSopenharmony_ci        }
267a8e1175bSopenharmony_ci        if (ret == 0) {
268a8e1175bSopenharmony_ci            TEST_ASSERT(p == buf + input->len);
269a8e1175bSopenharmony_ci        }
270a8e1175bSopenharmony_ci
271a8e1175bSopenharmony_ci        mbedtls_free(buf);
272a8e1175bSopenharmony_ci        buf = NULL;
273a8e1175bSopenharmony_ci    }
274a8e1175bSopenharmony_ci
275a8e1175bSopenharmony_ciexit:
276a8e1175bSopenharmony_ci    mbedtls_free(buf);
277a8e1175bSopenharmony_ci}
278a8e1175bSopenharmony_ci/* END_CASE */
279a8e1175bSopenharmony_ci
280a8e1175bSopenharmony_ci/* BEGIN_CASE */
281a8e1175bSopenharmony_civoid get_len(const data_t *input, int actual_length_arg)
282a8e1175bSopenharmony_ci{
283a8e1175bSopenharmony_ci    size_t actual_length = actual_length_arg;
284a8e1175bSopenharmony_ci    size_t buffer_size;
285a8e1175bSopenharmony_ci
286a8e1175bSopenharmony_ci    /* Test prefixes of a buffer containing the given length string
287a8e1175bSopenharmony_ci     * followed by `actual_length` bytes of payload. To save a bit of
288a8e1175bSopenharmony_ci     * time, we skip some "boring" prefixes: we don't test prefixes where
289a8e1175bSopenharmony_ci     * the payload is truncated more than one byte away from either end,
290a8e1175bSopenharmony_ci     * and we only test the empty string on a 1-byte input.
291a8e1175bSopenharmony_ci     */
292a8e1175bSopenharmony_ci    for (buffer_size = 1; buffer_size <= input->len + 1; buffer_size++) {
293a8e1175bSopenharmony_ci        if (!get_len_step(input, buffer_size, actual_length)) {
294a8e1175bSopenharmony_ci            goto exit;
295a8e1175bSopenharmony_ci        }
296a8e1175bSopenharmony_ci    }
297a8e1175bSopenharmony_ci    if (!get_len_step(input, input->len + actual_length - 1, actual_length)) {
298a8e1175bSopenharmony_ci        goto exit;
299a8e1175bSopenharmony_ci    }
300a8e1175bSopenharmony_ci    if (!get_len_step(input, input->len + actual_length, actual_length)) {
301a8e1175bSopenharmony_ci        goto exit;
302a8e1175bSopenharmony_ci    }
303a8e1175bSopenharmony_ci}
304a8e1175bSopenharmony_ci/* END_CASE */
305a8e1175bSopenharmony_ci
306a8e1175bSopenharmony_ci/* BEGIN_CASE */
307a8e1175bSopenharmony_civoid get_boolean(const data_t *input,
308a8e1175bSopenharmony_ci                 int expected_value, int expected_result)
309a8e1175bSopenharmony_ci{
310a8e1175bSopenharmony_ci    unsigned char *p = input->x;
311a8e1175bSopenharmony_ci    int val;
312a8e1175bSopenharmony_ci    int ret;
313a8e1175bSopenharmony_ci    ret = mbedtls_asn1_get_bool(&p, input->x + input->len, &val);
314a8e1175bSopenharmony_ci    TEST_EQUAL(ret, expected_result);
315a8e1175bSopenharmony_ci    if (expected_result == 0) {
316a8e1175bSopenharmony_ci        TEST_EQUAL(val, expected_value);
317a8e1175bSopenharmony_ci        TEST_ASSERT(p == input->x + input->len);
318a8e1175bSopenharmony_ci    }
319a8e1175bSopenharmony_ci}
320a8e1175bSopenharmony_ci/* END_CASE */
321a8e1175bSopenharmony_ci
322a8e1175bSopenharmony_ci/* BEGIN_CASE */
323a8e1175bSopenharmony_civoid empty_integer(const data_t *input)
324a8e1175bSopenharmony_ci{
325a8e1175bSopenharmony_ci    unsigned char *p;
326a8e1175bSopenharmony_ci#if defined(MBEDTLS_BIGNUM_C)
327a8e1175bSopenharmony_ci    mbedtls_mpi actual_mpi;
328a8e1175bSopenharmony_ci#endif
329a8e1175bSopenharmony_ci    int val;
330a8e1175bSopenharmony_ci
331a8e1175bSopenharmony_ci#if defined(MBEDTLS_BIGNUM_C)
332a8e1175bSopenharmony_ci    mbedtls_mpi_init(&actual_mpi);
333a8e1175bSopenharmony_ci#endif
334a8e1175bSopenharmony_ci
335a8e1175bSopenharmony_ci    /* An INTEGER with no content is not valid. */
336a8e1175bSopenharmony_ci    p = input->x;
337a8e1175bSopenharmony_ci    TEST_EQUAL(mbedtls_asn1_get_int(&p, input->x + input->len, &val),
338a8e1175bSopenharmony_ci               MBEDTLS_ERR_ASN1_INVALID_LENGTH);
339a8e1175bSopenharmony_ci
340a8e1175bSopenharmony_ci#if defined(MBEDTLS_BIGNUM_C)
341a8e1175bSopenharmony_ci    /* INTEGERs are sometimes abused as bitstrings, so the library accepts
342a8e1175bSopenharmony_ci     * an INTEGER with empty content and gives it the value 0. */
343a8e1175bSopenharmony_ci    p = input->x;
344a8e1175bSopenharmony_ci    TEST_EQUAL(mbedtls_asn1_get_mpi(&p, input->x + input->len, &actual_mpi),
345a8e1175bSopenharmony_ci               0);
346a8e1175bSopenharmony_ci    TEST_EQUAL(mbedtls_mpi_cmp_int(&actual_mpi, 0), 0);
347a8e1175bSopenharmony_ci#endif
348a8e1175bSopenharmony_ci
349a8e1175bSopenharmony_ciexit:
350a8e1175bSopenharmony_ci#if defined(MBEDTLS_BIGNUM_C)
351a8e1175bSopenharmony_ci    mbedtls_mpi_free(&actual_mpi);
352a8e1175bSopenharmony_ci#endif
353a8e1175bSopenharmony_ci    /*empty cleanup in some configurations*/;
354a8e1175bSopenharmony_ci}
355a8e1175bSopenharmony_ci/* END_CASE */
356a8e1175bSopenharmony_ci
357a8e1175bSopenharmony_ci/* BEGIN_CASE */
358a8e1175bSopenharmony_civoid get_integer(const data_t *input,
359a8e1175bSopenharmony_ci                 const char *expected_hex, int expected_result)
360a8e1175bSopenharmony_ci{
361a8e1175bSopenharmony_ci    unsigned char *p;
362a8e1175bSopenharmony_ci#if defined(MBEDTLS_BIGNUM_C)
363a8e1175bSopenharmony_ci    mbedtls_mpi expected_mpi;
364a8e1175bSopenharmony_ci    mbedtls_mpi actual_mpi;
365a8e1175bSopenharmony_ci    mbedtls_mpi complement;
366a8e1175bSopenharmony_ci    int expected_result_for_mpi = expected_result;
367a8e1175bSopenharmony_ci#endif
368a8e1175bSopenharmony_ci    long expected_value;
369a8e1175bSopenharmony_ci    int expected_result_for_int = expected_result;
370a8e1175bSopenharmony_ci    int val;
371a8e1175bSopenharmony_ci    int ret;
372a8e1175bSopenharmony_ci
373a8e1175bSopenharmony_ci#if defined(MBEDTLS_BIGNUM_C)
374a8e1175bSopenharmony_ci    mbedtls_mpi_init(&expected_mpi);
375a8e1175bSopenharmony_ci    mbedtls_mpi_init(&actual_mpi);
376a8e1175bSopenharmony_ci    mbedtls_mpi_init(&complement);
377a8e1175bSopenharmony_ci#endif
378a8e1175bSopenharmony_ci
379a8e1175bSopenharmony_ci    errno = 0;
380a8e1175bSopenharmony_ci    expected_value = strtol(expected_hex, NULL, 16);
381a8e1175bSopenharmony_ci    if (expected_result == 0 &&
382a8e1175bSopenharmony_ci        (errno == ERANGE
383a8e1175bSopenharmony_ci#if LONG_MAX > INT_MAX
384a8e1175bSopenharmony_ci         || expected_value > INT_MAX || expected_value < INT_MIN
385a8e1175bSopenharmony_ci#endif
386a8e1175bSopenharmony_ci        )) {
387a8e1175bSopenharmony_ci        /* The library returns the dubious error code INVALID_LENGTH
388a8e1175bSopenharmony_ci         * for integers that are out of range. */
389a8e1175bSopenharmony_ci        expected_result_for_int = MBEDTLS_ERR_ASN1_INVALID_LENGTH;
390a8e1175bSopenharmony_ci    }
391a8e1175bSopenharmony_ci    if (expected_result == 0 && expected_value < 0) {
392a8e1175bSopenharmony_ci        /* The library does not support negative INTEGERs and
393a8e1175bSopenharmony_ci         * returns the dubious error code INVALID_LENGTH.
394a8e1175bSopenharmony_ci         * Test that we preserve the historical behavior. If we
395a8e1175bSopenharmony_ci         * decide to change the behavior, we'll also change this test. */
396a8e1175bSopenharmony_ci        expected_result_for_int = MBEDTLS_ERR_ASN1_INVALID_LENGTH;
397a8e1175bSopenharmony_ci    }
398a8e1175bSopenharmony_ci
399a8e1175bSopenharmony_ci    p = input->x;
400a8e1175bSopenharmony_ci    ret = mbedtls_asn1_get_int(&p, input->x + input->len, &val);
401a8e1175bSopenharmony_ci    TEST_EQUAL(ret, expected_result_for_int);
402a8e1175bSopenharmony_ci    if (ret == 0) {
403a8e1175bSopenharmony_ci        TEST_EQUAL(val, expected_value);
404a8e1175bSopenharmony_ci        TEST_ASSERT(p == input->x + input->len);
405a8e1175bSopenharmony_ci    }
406a8e1175bSopenharmony_ci
407a8e1175bSopenharmony_ci#if defined(MBEDTLS_BIGNUM_C)
408a8e1175bSopenharmony_ci    ret = mbedtls_test_read_mpi(&expected_mpi, expected_hex);
409a8e1175bSopenharmony_ci    TEST_ASSERT(ret == 0 || ret == MBEDTLS_ERR_MPI_BAD_INPUT_DATA);
410a8e1175bSopenharmony_ci    if (ret == MBEDTLS_ERR_MPI_BAD_INPUT_DATA) {
411a8e1175bSopenharmony_ci        /* The data overflows the maximum MPI size. */
412a8e1175bSopenharmony_ci        expected_result_for_mpi = MBEDTLS_ERR_MPI_BAD_INPUT_DATA;
413a8e1175bSopenharmony_ci    }
414a8e1175bSopenharmony_ci    p = input->x;
415a8e1175bSopenharmony_ci    ret = mbedtls_asn1_get_mpi(&p, input->x + input->len, &actual_mpi);
416a8e1175bSopenharmony_ci    TEST_EQUAL(ret, expected_result_for_mpi);
417a8e1175bSopenharmony_ci    if (ret == 0) {
418a8e1175bSopenharmony_ci        if (expected_value >= 0) {
419a8e1175bSopenharmony_ci            TEST_ASSERT(mbedtls_mpi_cmp_mpi(&actual_mpi,
420a8e1175bSopenharmony_ci                                            &expected_mpi) == 0);
421a8e1175bSopenharmony_ci        } else {
422a8e1175bSopenharmony_ci            /* The library ignores the sign bit in ASN.1 INTEGERs
423a8e1175bSopenharmony_ci             * (which makes sense insofar as INTEGERs are sometimes
424a8e1175bSopenharmony_ci             * abused as bit strings), so the result of parsing them
425a8e1175bSopenharmony_ci             * is a positive integer such that expected_mpi +
426a8e1175bSopenharmony_ci             * actual_mpi = 2^n where n is the length of the content
427a8e1175bSopenharmony_ci             * of the INTEGER. (Leading ff octets don't matter for the
428a8e1175bSopenharmony_ci             * expected value, but they matter for the actual value.)
429a8e1175bSopenharmony_ci             * Test that we don't change from this behavior. If we
430a8e1175bSopenharmony_ci             * decide to fix the library to change the behavior on
431a8e1175bSopenharmony_ci             * negative INTEGERs, we'll fix this test code. */
432a8e1175bSopenharmony_ci            unsigned char *q = input->x + 1;
433a8e1175bSopenharmony_ci            size_t len;
434a8e1175bSopenharmony_ci            TEST_ASSERT(mbedtls_asn1_get_len(&q, input->x + input->len,
435a8e1175bSopenharmony_ci                                             &len) == 0);
436a8e1175bSopenharmony_ci            TEST_ASSERT(mbedtls_mpi_lset(&complement, 1) == 0);
437a8e1175bSopenharmony_ci            TEST_ASSERT(mbedtls_mpi_shift_l(&complement, len * 8) == 0);
438a8e1175bSopenharmony_ci            TEST_ASSERT(mbedtls_mpi_add_mpi(&complement, &complement,
439a8e1175bSopenharmony_ci                                            &expected_mpi) == 0);
440a8e1175bSopenharmony_ci            TEST_ASSERT(mbedtls_mpi_cmp_mpi(&complement,
441a8e1175bSopenharmony_ci                                            &actual_mpi) == 0);
442a8e1175bSopenharmony_ci        }
443a8e1175bSopenharmony_ci        TEST_ASSERT(p == input->x + input->len);
444a8e1175bSopenharmony_ci    }
445a8e1175bSopenharmony_ci#endif
446a8e1175bSopenharmony_ci
447a8e1175bSopenharmony_ciexit:
448a8e1175bSopenharmony_ci#if defined(MBEDTLS_BIGNUM_C)
449a8e1175bSopenharmony_ci    mbedtls_mpi_free(&expected_mpi);
450a8e1175bSopenharmony_ci    mbedtls_mpi_free(&actual_mpi);
451a8e1175bSopenharmony_ci    mbedtls_mpi_free(&complement);
452a8e1175bSopenharmony_ci#endif
453a8e1175bSopenharmony_ci    /*empty cleanup in some configurations*/;
454a8e1175bSopenharmony_ci}
455a8e1175bSopenharmony_ci/* END_CASE */
456a8e1175bSopenharmony_ci
457a8e1175bSopenharmony_ci/* BEGIN_CASE */
458a8e1175bSopenharmony_civoid get_enum(const data_t *input,
459a8e1175bSopenharmony_ci              const char *expected_hex, int expected_result)
460a8e1175bSopenharmony_ci{
461a8e1175bSopenharmony_ci    unsigned char *p;
462a8e1175bSopenharmony_ci    long expected_value;
463a8e1175bSopenharmony_ci    int expected_result_for_enum = expected_result;
464a8e1175bSopenharmony_ci    int val;
465a8e1175bSopenharmony_ci    int ret;
466a8e1175bSopenharmony_ci
467a8e1175bSopenharmony_ci    errno = 0;
468a8e1175bSopenharmony_ci    expected_value = strtol(expected_hex, NULL, 16);
469a8e1175bSopenharmony_ci    if (expected_result == 0 &&
470a8e1175bSopenharmony_ci        (errno == ERANGE
471a8e1175bSopenharmony_ci#if LONG_MAX > INT_MAX
472a8e1175bSopenharmony_ci         || expected_value > INT_MAX || expected_value < INT_MIN
473a8e1175bSopenharmony_ci#endif
474a8e1175bSopenharmony_ci        )) {
475a8e1175bSopenharmony_ci        /* The library returns the dubious error code INVALID_LENGTH
476a8e1175bSopenharmony_ci         * for integers that are out of range. */
477a8e1175bSopenharmony_ci        expected_result_for_enum = MBEDTLS_ERR_ASN1_INVALID_LENGTH;
478a8e1175bSopenharmony_ci    }
479a8e1175bSopenharmony_ci    if (expected_result == 0 && expected_value < 0) {
480a8e1175bSopenharmony_ci        /* The library does not support negative INTEGERs and
481a8e1175bSopenharmony_ci         * returns the dubious error code INVALID_LENGTH.
482a8e1175bSopenharmony_ci         * Test that we preserve the historical behavior. If we
483a8e1175bSopenharmony_ci         * decide to change the behavior, we'll also change this test. */
484a8e1175bSopenharmony_ci        expected_result_for_enum = MBEDTLS_ERR_ASN1_INVALID_LENGTH;
485a8e1175bSopenharmony_ci    }
486a8e1175bSopenharmony_ci
487a8e1175bSopenharmony_ci    p = input->x;
488a8e1175bSopenharmony_ci    ret = mbedtls_asn1_get_enum(&p, input->x + input->len, &val);
489a8e1175bSopenharmony_ci    TEST_EQUAL(ret, expected_result_for_enum);
490a8e1175bSopenharmony_ci    if (ret == 0) {
491a8e1175bSopenharmony_ci        TEST_EQUAL(val, expected_value);
492a8e1175bSopenharmony_ci        TEST_ASSERT(p == input->x + input->len);
493a8e1175bSopenharmony_ci    }
494a8e1175bSopenharmony_ci}
495a8e1175bSopenharmony_ci/* END_CASE */
496a8e1175bSopenharmony_ci
497a8e1175bSopenharmony_ci/* BEGIN_CASE depends_on:MBEDTLS_BIGNUM_C */
498a8e1175bSopenharmony_civoid get_mpi_too_large()
499a8e1175bSopenharmony_ci{
500a8e1175bSopenharmony_ci    unsigned char *buf = NULL;
501a8e1175bSopenharmony_ci    unsigned char *p;
502a8e1175bSopenharmony_ci    mbedtls_mpi actual_mpi;
503a8e1175bSopenharmony_ci    size_t too_many_octets =
504a8e1175bSopenharmony_ci        MBEDTLS_MPI_MAX_LIMBS * sizeof(mbedtls_mpi_uint) + 1;
505a8e1175bSopenharmony_ci    size_t size = too_many_octets + 6;
506a8e1175bSopenharmony_ci
507a8e1175bSopenharmony_ci    mbedtls_mpi_init(&actual_mpi);
508a8e1175bSopenharmony_ci
509a8e1175bSopenharmony_ci    TEST_CALLOC(buf, size);
510a8e1175bSopenharmony_ci    buf[0] = 0x02; /* tag: INTEGER */
511a8e1175bSopenharmony_ci    buf[1] = 0x84; /* 4-octet length */
512a8e1175bSopenharmony_ci    buf[2] = (too_many_octets >> 24) & 0xff;
513a8e1175bSopenharmony_ci    buf[3] = (too_many_octets >> 16) & 0xff;
514a8e1175bSopenharmony_ci    buf[4] = (too_many_octets >> 8) & 0xff;
515a8e1175bSopenharmony_ci    buf[5] = too_many_octets & 0xff;
516a8e1175bSopenharmony_ci    buf[6] = 0x01; /* most significant octet */
517a8e1175bSopenharmony_ci
518a8e1175bSopenharmony_ci    p = buf;
519a8e1175bSopenharmony_ci    TEST_EQUAL(mbedtls_asn1_get_mpi(&p, buf + size, &actual_mpi),
520a8e1175bSopenharmony_ci               MBEDTLS_ERR_MPI_ALLOC_FAILED);
521a8e1175bSopenharmony_ci
522a8e1175bSopenharmony_ciexit:
523a8e1175bSopenharmony_ci    mbedtls_mpi_free(&actual_mpi);
524a8e1175bSopenharmony_ci    mbedtls_free(buf);
525a8e1175bSopenharmony_ci}
526a8e1175bSopenharmony_ci/* END_CASE */
527a8e1175bSopenharmony_ci
528a8e1175bSopenharmony_ci/* BEGIN_CASE */
529a8e1175bSopenharmony_civoid get_bitstring(const data_t *input,
530a8e1175bSopenharmony_ci                   int expected_length, int expected_unused_bits,
531a8e1175bSopenharmony_ci                   int expected_result, int expected_result_null)
532a8e1175bSopenharmony_ci{
533a8e1175bSopenharmony_ci    mbedtls_asn1_bitstring bs = { 0xdead, 0x21, NULL };
534a8e1175bSopenharmony_ci    unsigned char *p = input->x;
535a8e1175bSopenharmony_ci
536a8e1175bSopenharmony_ci    TEST_EQUAL(mbedtls_asn1_get_bitstring(&p, input->x + input->len, &bs),
537a8e1175bSopenharmony_ci               expected_result);
538a8e1175bSopenharmony_ci    if (expected_result == 0) {
539a8e1175bSopenharmony_ci        TEST_EQUAL(bs.len, (size_t) expected_length);
540a8e1175bSopenharmony_ci        TEST_EQUAL(bs.unused_bits, expected_unused_bits);
541a8e1175bSopenharmony_ci        TEST_ASSERT(bs.p != NULL);
542a8e1175bSopenharmony_ci        TEST_EQUAL(bs.p - input->x + bs.len, input->len);
543a8e1175bSopenharmony_ci        TEST_ASSERT(p == input->x + input->len);
544a8e1175bSopenharmony_ci    }
545a8e1175bSopenharmony_ci
546a8e1175bSopenharmony_ci    p = input->x;
547a8e1175bSopenharmony_ci    TEST_EQUAL(mbedtls_asn1_get_bitstring_null(&p, input->x + input->len,
548a8e1175bSopenharmony_ci                                               &bs.len),
549a8e1175bSopenharmony_ci               expected_result_null);
550a8e1175bSopenharmony_ci    if (expected_result_null == 0) {
551a8e1175bSopenharmony_ci        TEST_EQUAL(bs.len, (size_t) expected_length);
552a8e1175bSopenharmony_ci        if (expected_result == 0) {
553a8e1175bSopenharmony_ci            TEST_ASSERT(p == input->x + input->len - bs.len);
554a8e1175bSopenharmony_ci        }
555a8e1175bSopenharmony_ci    }
556a8e1175bSopenharmony_ci}
557a8e1175bSopenharmony_ci/* END_CASE */
558a8e1175bSopenharmony_ci
559a8e1175bSopenharmony_ci/* BEGIN_CASE */
560a8e1175bSopenharmony_civoid get_sequence_of(const data_t *input, int tag,
561a8e1175bSopenharmony_ci                     const char *description,
562a8e1175bSopenharmony_ci                     int expected_result)
563a8e1175bSopenharmony_ci{
564a8e1175bSopenharmony_ci    /* The description string is a comma-separated list of integers.
565a8e1175bSopenharmony_ci     * For each element in the SEQUENCE in input, description contains
566a8e1175bSopenharmony_ci     * two integers: the offset of the element (offset from the start
567a8e1175bSopenharmony_ci     * of input to the tag of the element) and the length of the
568a8e1175bSopenharmony_ci     * element's contents.
569a8e1175bSopenharmony_ci     * "offset1,length1,..." */
570a8e1175bSopenharmony_ci
571a8e1175bSopenharmony_ci    mbedtls_asn1_sequence head = { { 0, 0, NULL }, NULL };
572a8e1175bSopenharmony_ci    mbedtls_asn1_sequence *cur;
573a8e1175bSopenharmony_ci    unsigned char *p = input->x;
574a8e1175bSopenharmony_ci    const char *rest = description;
575a8e1175bSopenharmony_ci    unsigned long n;
576a8e1175bSopenharmony_ci    unsigned int step = 0;
577a8e1175bSopenharmony_ci
578a8e1175bSopenharmony_ci    TEST_EQUAL(mbedtls_asn1_get_sequence_of(&p, input->x + input->len,
579a8e1175bSopenharmony_ci                                            &head, tag),
580a8e1175bSopenharmony_ci               expected_result);
581a8e1175bSopenharmony_ci    if (expected_result == 0) {
582a8e1175bSopenharmony_ci        TEST_ASSERT(p == input->x + input->len);
583a8e1175bSopenharmony_ci
584a8e1175bSopenharmony_ci        if (!*rest) {
585a8e1175bSopenharmony_ci            TEST_EQUAL(head.buf.tag, 0);
586a8e1175bSopenharmony_ci            TEST_ASSERT(head.buf.p == NULL);
587a8e1175bSopenharmony_ci            TEST_EQUAL(head.buf.len, 0);
588a8e1175bSopenharmony_ci            TEST_ASSERT(head.next == NULL);
589a8e1175bSopenharmony_ci        } else {
590a8e1175bSopenharmony_ci            cur = &head;
591a8e1175bSopenharmony_ci            while (*rest) {
592a8e1175bSopenharmony_ci                mbedtls_test_set_step(step);
593a8e1175bSopenharmony_ci                TEST_ASSERT(cur != NULL);
594a8e1175bSopenharmony_ci                TEST_EQUAL(cur->buf.tag, tag);
595a8e1175bSopenharmony_ci                n = strtoul(rest, (char **) &rest, 0);
596a8e1175bSopenharmony_ci                TEST_EQUAL(n, (size_t) (cur->buf.p - input->x));
597a8e1175bSopenharmony_ci                ++rest;
598a8e1175bSopenharmony_ci                n = strtoul(rest, (char **) &rest, 0);
599a8e1175bSopenharmony_ci                TEST_EQUAL(n, cur->buf.len);
600a8e1175bSopenharmony_ci                if (*rest) {
601a8e1175bSopenharmony_ci                    ++rest;
602a8e1175bSopenharmony_ci                }
603a8e1175bSopenharmony_ci                cur = cur->next;
604a8e1175bSopenharmony_ci                ++step;
605a8e1175bSopenharmony_ci            }
606a8e1175bSopenharmony_ci            TEST_ASSERT(cur == NULL);
607a8e1175bSopenharmony_ci        }
608a8e1175bSopenharmony_ci    }
609a8e1175bSopenharmony_ci
610a8e1175bSopenharmony_ciexit:
611a8e1175bSopenharmony_ci    mbedtls_asn1_sequence_free(head.next);
612a8e1175bSopenharmony_ci}
613a8e1175bSopenharmony_ci/* END_CASE */
614a8e1175bSopenharmony_ci
615a8e1175bSopenharmony_ci/* BEGIN_CASE */
616a8e1175bSopenharmony_civoid traverse_sequence_of(const data_t *input,
617a8e1175bSopenharmony_ci                          int tag_must_mask, int tag_must_val,
618a8e1175bSopenharmony_ci                          int tag_may_mask, int tag_may_val,
619a8e1175bSopenharmony_ci                          const char *description,
620a8e1175bSopenharmony_ci                          int expected_result)
621a8e1175bSopenharmony_ci{
622a8e1175bSopenharmony_ci    /* The description string is a comma-separated list of integers.
623a8e1175bSopenharmony_ci     * For each element in the SEQUENCE in input, description contains
624a8e1175bSopenharmony_ci     * three integers: the offset of the element's content (offset from
625a8e1175bSopenharmony_ci     * the start of input to the content of the element), the element's tag,
626a8e1175bSopenharmony_ci     * and the length of the element's contents.
627a8e1175bSopenharmony_ci     * "offset1,tag1,length1,..." */
628a8e1175bSopenharmony_ci
629a8e1175bSopenharmony_ci    unsigned char *p = input->x;
630a8e1175bSopenharmony_ci    traverse_state_t traverse_state = { input->x, description };
631a8e1175bSopenharmony_ci    int ret;
632a8e1175bSopenharmony_ci
633a8e1175bSopenharmony_ci    ret = mbedtls_asn1_traverse_sequence_of(&p, input->x + input->len,
634a8e1175bSopenharmony_ci                                            (uint8_t) tag_must_mask, (uint8_t) tag_must_val,
635a8e1175bSopenharmony_ci                                            (uint8_t) tag_may_mask, (uint8_t) tag_may_val,
636a8e1175bSopenharmony_ci                                            traverse_callback, &traverse_state);
637a8e1175bSopenharmony_ci    if (ret == RET_TRAVERSE_ERROR) {
638a8e1175bSopenharmony_ci        goto exit;
639a8e1175bSopenharmony_ci    }
640a8e1175bSopenharmony_ci    TEST_EQUAL(ret, expected_result);
641a8e1175bSopenharmony_ci    TEST_EQUAL(*traverse_state.description, 0);
642a8e1175bSopenharmony_ci}
643a8e1175bSopenharmony_ci/* END_CASE */
644a8e1175bSopenharmony_ci
645a8e1175bSopenharmony_ci/* BEGIN_CASE */
646a8e1175bSopenharmony_civoid get_alg(const data_t *input,
647a8e1175bSopenharmony_ci             int oid_offset, int oid_length,
648a8e1175bSopenharmony_ci             int params_tag, int params_offset, int params_length,
649a8e1175bSopenharmony_ci             int total_length,
650a8e1175bSopenharmony_ci             int expected_result)
651a8e1175bSopenharmony_ci{
652a8e1175bSopenharmony_ci    mbedtls_asn1_buf oid = { -1, 0, NULL };
653a8e1175bSopenharmony_ci    mbedtls_asn1_buf params = { -1, 0, NULL };
654a8e1175bSopenharmony_ci    unsigned char *p = input->x;
655a8e1175bSopenharmony_ci    int ret;
656a8e1175bSopenharmony_ci
657a8e1175bSopenharmony_ci    TEST_EQUAL(mbedtls_asn1_get_alg(&p, input->x + input->len,
658a8e1175bSopenharmony_ci                                    &oid, &params),
659a8e1175bSopenharmony_ci               expected_result);
660a8e1175bSopenharmony_ci    if (expected_result == 0) {
661a8e1175bSopenharmony_ci        TEST_EQUAL(oid.tag, MBEDTLS_ASN1_OID);
662a8e1175bSopenharmony_ci        TEST_EQUAL(oid.p - input->x, oid_offset);
663a8e1175bSopenharmony_ci        TEST_EQUAL(oid.len, (size_t) oid_length);
664a8e1175bSopenharmony_ci        TEST_EQUAL(params.tag, params_tag);
665a8e1175bSopenharmony_ci        if (params_offset != 0) {
666a8e1175bSopenharmony_ci            TEST_EQUAL(params.p - input->x, params_offset);
667a8e1175bSopenharmony_ci        } else {
668a8e1175bSopenharmony_ci            TEST_ASSERT(params.p == NULL);
669a8e1175bSopenharmony_ci        }
670a8e1175bSopenharmony_ci        TEST_EQUAL(params.len, (size_t) params_length);
671a8e1175bSopenharmony_ci        TEST_EQUAL(p - input->x, total_length);
672a8e1175bSopenharmony_ci    }
673a8e1175bSopenharmony_ci
674a8e1175bSopenharmony_ci    ret = mbedtls_asn1_get_alg_null(&p, input->x + input->len, &oid);
675a8e1175bSopenharmony_ci    if (expected_result == 0 && params_offset == 0) {
676a8e1175bSopenharmony_ci        TEST_EQUAL(oid.tag, MBEDTLS_ASN1_OID);
677a8e1175bSopenharmony_ci        TEST_EQUAL(oid.p - input->x, oid_offset);
678a8e1175bSopenharmony_ci        TEST_EQUAL(oid.len, (size_t) oid_length);
679a8e1175bSopenharmony_ci        TEST_EQUAL(p - input->x, total_length);
680a8e1175bSopenharmony_ci    } else {
681a8e1175bSopenharmony_ci        TEST_ASSERT(ret != 0);
682a8e1175bSopenharmony_ci    }
683a8e1175bSopenharmony_ci}
684a8e1175bSopenharmony_ci/* END_CASE */
685a8e1175bSopenharmony_ci
686a8e1175bSopenharmony_ci/* BEGIN_CASE */
687a8e1175bSopenharmony_civoid find_named_data(data_t *oid0, data_t *oid1, data_t *oid2, data_t *oid3,
688a8e1175bSopenharmony_ci                     data_t *needle, int from, int position)
689a8e1175bSopenharmony_ci{
690a8e1175bSopenharmony_ci    mbedtls_asn1_named_data nd[] = {
691a8e1175bSopenharmony_ci        { { 0x06, oid0->len, oid0->x }, { 0, 0, NULL }, NULL, 0 },
692a8e1175bSopenharmony_ci        { { 0x06, oid1->len, oid1->x }, { 0, 0, NULL }, NULL, 0 },
693a8e1175bSopenharmony_ci        { { 0x06, oid2->len, oid2->x }, { 0, 0, NULL }, NULL, 0 },
694a8e1175bSopenharmony_ci        { { 0x06, oid3->len, oid3->x }, { 0, 0, NULL }, NULL, 0 },
695a8e1175bSopenharmony_ci    };
696a8e1175bSopenharmony_ci    mbedtls_asn1_named_data *pointers[ARRAY_LENGTH(nd) + 1];
697a8e1175bSopenharmony_ci    size_t i;
698a8e1175bSopenharmony_ci    const mbedtls_asn1_named_data *found;
699a8e1175bSopenharmony_ci
700a8e1175bSopenharmony_ci    for (i = 0; i < ARRAY_LENGTH(nd); i++) {
701a8e1175bSopenharmony_ci        pointers[i] = &nd[i];
702a8e1175bSopenharmony_ci    }
703a8e1175bSopenharmony_ci    pointers[ARRAY_LENGTH(nd)] = NULL;
704a8e1175bSopenharmony_ci    for (i = 0; i < ARRAY_LENGTH(nd); i++) {
705a8e1175bSopenharmony_ci        nd[i].next = pointers[i+1];
706a8e1175bSopenharmony_ci    }
707a8e1175bSopenharmony_ci
708a8e1175bSopenharmony_ci    found = mbedtls_asn1_find_named_data((const mbedtls_asn1_named_data *) pointers[from],
709a8e1175bSopenharmony_ci                                         (const char *) needle->x,
710a8e1175bSopenharmony_ci                                         needle->len);
711a8e1175bSopenharmony_ci    TEST_ASSERT(found == pointers[position]);
712a8e1175bSopenharmony_ci}
713a8e1175bSopenharmony_ci/* END_CASE */
714a8e1175bSopenharmony_ci
715a8e1175bSopenharmony_ci/* BEGIN_CASE depends_on:!MBEDTLS_DEPRECATED_REMOVED:!MBEDTLS_DEPRECATED_WARNING */
716a8e1175bSopenharmony_civoid free_named_data_null()
717a8e1175bSopenharmony_ci{
718a8e1175bSopenharmony_ci    mbedtls_asn1_free_named_data(NULL);
719a8e1175bSopenharmony_ci    goto exit; /* Silence unused label warning */
720a8e1175bSopenharmony_ci}
721a8e1175bSopenharmony_ci/* END_CASE */
722a8e1175bSopenharmony_ci
723a8e1175bSopenharmony_ci/* BEGIN_CASE depends_on:!MBEDTLS_DEPRECATED_REMOVED:!MBEDTLS_DEPRECATED_WARNING */
724a8e1175bSopenharmony_civoid free_named_data(int with_oid, int with_val, int with_next)
725a8e1175bSopenharmony_ci{
726a8e1175bSopenharmony_ci    mbedtls_asn1_named_data next =
727a8e1175bSopenharmony_ci    { { 0x06, 0, NULL }, { 0, 0xcafe, NULL }, NULL, 0 };
728a8e1175bSopenharmony_ci    mbedtls_asn1_named_data head =
729a8e1175bSopenharmony_ci    { { 0x06, 0, NULL }, { 0, 0, NULL }, NULL, 0 };
730a8e1175bSopenharmony_ci
731a8e1175bSopenharmony_ci    if (with_oid) {
732a8e1175bSopenharmony_ci        TEST_CALLOC(head.oid.p, 1);
733a8e1175bSopenharmony_ci    }
734a8e1175bSopenharmony_ci    if (with_val) {
735a8e1175bSopenharmony_ci        TEST_CALLOC(head.val.p, 1);
736a8e1175bSopenharmony_ci    }
737a8e1175bSopenharmony_ci    if (with_next) {
738a8e1175bSopenharmony_ci        head.next = &next;
739a8e1175bSopenharmony_ci    }
740a8e1175bSopenharmony_ci
741a8e1175bSopenharmony_ci    mbedtls_asn1_free_named_data(&head);
742a8e1175bSopenharmony_ci    TEST_ASSERT(head.oid.p == NULL);
743a8e1175bSopenharmony_ci    TEST_ASSERT(head.val.p == NULL);
744a8e1175bSopenharmony_ci    TEST_ASSERT(head.next == NULL);
745a8e1175bSopenharmony_ci    TEST_ASSERT(next.val.len == 0xcafe);
746a8e1175bSopenharmony_ci
747a8e1175bSopenharmony_ciexit:
748a8e1175bSopenharmony_ci    mbedtls_free(head.oid.p);
749a8e1175bSopenharmony_ci    mbedtls_free(head.val.p);
750a8e1175bSopenharmony_ci}
751a8e1175bSopenharmony_ci/* END_CASE */
752a8e1175bSopenharmony_ci
753a8e1175bSopenharmony_ci/* BEGIN_CASE */
754a8e1175bSopenharmony_civoid free_named_data_list(int length)
755a8e1175bSopenharmony_ci{
756a8e1175bSopenharmony_ci    mbedtls_asn1_named_data *head = NULL;
757a8e1175bSopenharmony_ci    int i;
758a8e1175bSopenharmony_ci
759a8e1175bSopenharmony_ci    for (i = 0; i < length; i++) {
760a8e1175bSopenharmony_ci        mbedtls_asn1_named_data *new = NULL;
761a8e1175bSopenharmony_ci        TEST_CALLOC(new, 1);
762a8e1175bSopenharmony_ci        new->next = head;
763a8e1175bSopenharmony_ci        head = new;
764a8e1175bSopenharmony_ci    }
765a8e1175bSopenharmony_ci
766a8e1175bSopenharmony_ci    mbedtls_asn1_free_named_data_list(&head);
767a8e1175bSopenharmony_ci    TEST_ASSERT(head == NULL);
768a8e1175bSopenharmony_ci    /* Most of the point of the test is that it doesn't leak memory.
769a8e1175bSopenharmony_ci     * So this test is only really useful under a memory leak detection
770a8e1175bSopenharmony_ci     * framework. */
771a8e1175bSopenharmony_ciexit:
772a8e1175bSopenharmony_ci    mbedtls_asn1_free_named_data_list(&head);
773a8e1175bSopenharmony_ci}
774a8e1175bSopenharmony_ci/* END_CASE */
775