| /* BEGIN_HEADER */ |
| #include "mbedtls/asn1write.h" |
| |
| #define GUARD_LEN 4 |
| #define GUARD_VAL 0x2a |
| |
| typedef struct { |
| unsigned char *output; |
| unsigned char *start; |
| unsigned char *end; |
| unsigned char *p; |
| size_t size; |
| } generic_write_data_t; |
| |
| int generic_write_start_step(generic_write_data_t *data) |
| { |
| mbedtls_test_set_step(data->size); |
| ASSERT_ALLOC(data->output, data->size == 0 ? 1 : data->size); |
| data->end = data->output + data->size; |
| data->p = data->end; |
| data->start = data->end - data->size; |
| return 1; |
| exit: |
| return 0; |
| } |
| |
| int generic_write_finish_step(generic_write_data_t *data, |
| const data_t *expected, |
| int ret) |
| { |
| int ok = 0; |
| |
| if (data->size < expected->len) { |
| TEST_EQUAL(ret, MBEDTLS_ERR_ASN1_BUF_TOO_SMALL); |
| } else { |
| TEST_EQUAL(ret, data->end - data->p); |
| TEST_ASSERT(data->p >= data->start); |
| TEST_ASSERT(data->p <= data->end); |
| ASSERT_COMPARE(data->p, (size_t)(data->end - data->p), expected->x, |
| expected->len); |
| } |
| ok = 1; |
| |
| exit: |
| mbedtls_free(data->output); |
| data->output = NULL; |
| return ok; |
| } |
| |
| /* END_HEADER */ |
| |
| /* BEGIN_DEPENDENCIES |
| * depends_on:MBEDTLS_ASN1_WRITE_C |
| * END_DEPENDENCIES |
| */ |
| |
| /* BEGIN_CASE */ |
| void mbedtls_asn1_write_null(data_t *expected) |
| { |
| generic_write_data_t data = { NULL, NULL, NULL, NULL, 0 }; |
| int ret; |
| |
| for (data.size = 0; data.size < expected->len + 1; data.size++) { |
| if (!generic_write_start_step(&data)) |
| goto exit; |
| ret = mbedtls_asn1_write_null(&data.p, data.start); |
| if (!generic_write_finish_step(&data, expected, ret)) |
| goto exit; |
| } |
| |
| exit: |
| mbedtls_free(data.output); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE */ |
| void mbedtls_asn1_write_bool(int val, data_t *expected) |
| { |
| generic_write_data_t data = { NULL, NULL, NULL, NULL, 0 }; |
| int ret; |
| |
| for (data.size = 0; data.size < expected->len + 1; data.size++) { |
| if (!generic_write_start_step(&data)) |
| goto exit; |
| ret = mbedtls_asn1_write_bool(&data.p, data.start, val); |
| if (!generic_write_finish_step(&data, expected, ret)) |
| goto exit; |
| } |
| |
| exit: |
| mbedtls_free(data.output); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE */ |
| void mbedtls_asn1_write_int(int val, data_t *expected) |
| { |
| generic_write_data_t data = { NULL, NULL, NULL, NULL, 0 }; |
| int ret; |
| |
| for (data.size = 0; data.size < expected->len + 1; data.size++) { |
| if (!generic_write_start_step(&data)) |
| goto exit; |
| ret = mbedtls_asn1_write_int(&data.p, data.start, val); |
| if (!generic_write_finish_step(&data, expected, ret)) |
| goto exit; |
| } |
| |
| exit: |
| mbedtls_free(data.output); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE */ |
| void mbedtls_asn1_write_enum(int val, data_t *expected) |
| { |
| generic_write_data_t data = { NULL, NULL, NULL, NULL, 0 }; |
| int ret; |
| |
| for (data.size = 0; data.size < expected->len + 1; data.size++) { |
| if (!generic_write_start_step(&data)) |
| goto exit; |
| ret = mbedtls_asn1_write_enum(&data.p, data.start, val); |
| if (!generic_write_finish_step(&data, expected, ret)) |
| goto exit; |
| } |
| |
| exit: |
| mbedtls_free(data.output); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE depends_on:MBEDTLS_BIGNUM_C */ |
| void mbedtls_asn1_write_mpi(data_t *val, data_t *expected) |
| { |
| generic_write_data_t data = { NULL, NULL, NULL, NULL, 0 }; |
| mbedtls_mpi mpi; |
| int ret; |
| |
| mbedtls_mpi_init(&mpi); |
| TEST_ASSERT(mbedtls_mpi_read_binary(&mpi, val->x, val->len) == 0); |
| |
| for (data.size = 0; data.size < expected->len + 1; data.size++) { |
| if (!generic_write_start_step(&data)) |
| goto exit; |
| ret = mbedtls_asn1_write_mpi(&data.p, data.start, &mpi); |
| if (!generic_write_finish_step(&data, expected, ret)) |
| goto exit; |
| if (expected->len > 10 && data.size == 8) |
| data.size = expected->len - 2; |
| } |
| |
| exit: |
| mbedtls_mpi_free(&mpi); |
| mbedtls_free(data.output); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE */ |
| void mbedtls_asn1_write_string(int tag, data_t *content, data_t *expected) |
| { |
| generic_write_data_t data = { NULL, NULL, NULL, NULL, 0 }; |
| int ret; |
| |
| for (data.size = 0; data.size < expected->len + 1; data.size++) { |
| if (!generic_write_start_step(&data)) |
| goto exit; |
| switch (tag) { |
| case MBEDTLS_ASN1_OCTET_STRING: |
| ret = mbedtls_asn1_write_octet_string(&data.p, data.start, |
| content->x, content->len); |
| break; |
| case MBEDTLS_ASN1_OID: |
| ret = mbedtls_asn1_write_oid(&data.p, data.start, |
| (const char *)content->x, |
| content->len); |
| break; |
| case MBEDTLS_ASN1_UTF8_STRING: |
| ret = mbedtls_asn1_write_utf8_string(&data.p, data.start, |
| (const char *)content->x, |
| content->len); |
| break; |
| case MBEDTLS_ASN1_PRINTABLE_STRING: |
| ret = mbedtls_asn1_write_printable_string( |
| &data.p, data.start, (const char *)content->x, |
| content->len); |
| break; |
| case MBEDTLS_ASN1_IA5_STRING: |
| ret = mbedtls_asn1_write_ia5_string(&data.p, data.start, |
| (const char *)content->x, |
| content->len); |
| break; |
| default: |
| ret = mbedtls_asn1_write_tagged_string(&data.p, data.start, tag, |
| (const char *)content->x, |
| content->len); |
| } |
| if (!generic_write_finish_step(&data, expected, ret)) |
| goto exit; |
| if (expected->len > 10 && data.size == 8) |
| data.size = expected->len - 2; |
| } |
| |
| exit: |
| mbedtls_free(data.output); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE */ |
| void mbedtls_asn1_write_algorithm_identifier(data_t *oid, |
| int par_len, |
| data_t *expected) |
| { |
| generic_write_data_t data = { NULL, NULL, NULL, NULL, 0 }; |
| int ret; |
| |
| for (data.size = 0; data.size < expected->len + 1; data.size++) { |
| if (!generic_write_start_step(&data)) |
| goto exit; |
| ret = mbedtls_asn1_write_algorithm_identifier( |
| &data.p, data.start, (const char *)oid->x, oid->len, par_len); |
| /* If params_len != 0, mbedtls_asn1_write_algorithm_identifier() |
| * assumes that the parameters are already present in the buffer |
| * and returns a length that accounts for this, but our test |
| * data omits the parameters. */ |
| if (ret >= 0) |
| ret -= par_len; |
| if (!generic_write_finish_step(&data, expected, ret)) |
| goto exit; |
| } |
| |
| exit: |
| mbedtls_free(data.output); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE depends_on:MBEDTLS_ASN1_PARSE_C */ |
| void mbedtls_asn1_write_len(int len, data_t *asn1, int buf_len, int result) |
| { |
| int ret; |
| unsigned char buf[150]; |
| unsigned char *p; |
| size_t i; |
| size_t read_len; |
| |
| memset(buf, GUARD_VAL, sizeof(buf)); |
| |
| p = buf + GUARD_LEN + buf_len; |
| |
| ret = mbedtls_asn1_write_len(&p, buf + GUARD_LEN, (size_t)len); |
| |
| TEST_ASSERT(ret == result); |
| |
| /* Check for buffer overwrite on both sides */ |
| for (i = 0; i < GUARD_LEN; i++) { |
| TEST_ASSERT(buf[i] == GUARD_VAL); |
| TEST_ASSERT(buf[GUARD_LEN + buf_len + i] == GUARD_VAL); |
| } |
| |
| if (result >= 0) { |
| TEST_ASSERT(p + asn1->len == buf + GUARD_LEN + buf_len); |
| |
| TEST_ASSERT(memcmp(p, asn1->x, asn1->len) == 0); |
| |
| /* Read back with mbedtls_asn1_get_len() to check */ |
| ret = mbedtls_asn1_get_len(&p, buf + GUARD_LEN + buf_len, &read_len); |
| |
| if (len == 0) { |
| TEST_ASSERT(ret == 0); |
| } else { |
| /* Return will be MBEDTLS_ERR_ASN1_OUT_OF_DATA because the rest of |
| * the buffer is missing |
| */ |
| TEST_ASSERT(ret == MBEDTLS_ERR_ASN1_OUT_OF_DATA); |
| } |
| TEST_ASSERT(read_len == (size_t)len); |
| TEST_ASSERT(p == buf + GUARD_LEN + buf_len); |
| } |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE */ |
| void test_asn1_write_bitstrings(data_t *bitstring, |
| int bits, |
| data_t *expected, |
| int is_named) |
| { |
| generic_write_data_t data = { NULL, NULL, NULL, NULL, 0 }; |
| int ret; |
| int (*func)(unsigned char **p, const unsigned char *start, |
| const unsigned char *buf, |
| size_t bits) = (is_named ? mbedtls_asn1_write_named_bitstring : |
| mbedtls_asn1_write_bitstring); |
| |
| for (data.size = 0; data.size < expected->len + 1; data.size++) { |
| if (!generic_write_start_step(&data)) |
| goto exit; |
| ret = (*func)(&data.p, data.start, bitstring->x, bits); |
| if (!generic_write_finish_step(&data, expected, ret)) |
| goto exit; |
| } |
| |
| exit: |
| mbedtls_free(data.output); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE */ |
| void store_named_data_find(data_t *oid0, |
| data_t *oid1, |
| data_t *oid2, |
| data_t *oid3, |
| data_t *needle, |
| int from, |
| int position) |
| { |
| data_t *oid[4] = { oid0, oid1, oid2, oid3 }; |
| mbedtls_asn1_named_data nd[] = { |
| { { 0x06, 0, NULL }, { 0, 0, NULL }, NULL, 0 }, |
| { { 0x06, 0, NULL }, { 0, 0, NULL }, NULL, 0 }, |
| { { 0x06, 0, NULL }, { 0, 0, NULL }, NULL, 0 }, |
| { { 0x06, 0, NULL }, { 0, 0, NULL }, NULL, 0 }, |
| }; |
| mbedtls_asn1_named_data *pointers[ARRAY_LENGTH(nd) + 1]; |
| size_t i; |
| mbedtls_asn1_named_data *head = NULL; |
| mbedtls_asn1_named_data *found = NULL; |
| |
| for (i = 0; i < ARRAY_LENGTH(nd); i++) |
| pointers[i] = &nd[i]; |
| pointers[ARRAY_LENGTH(nd)] = NULL; |
| for (i = 0; i < ARRAY_LENGTH(nd); i++) { |
| ASSERT_ALLOC(nd[i].oid.p, oid[i]->len); |
| memcpy(nd[i].oid.p, oid[i]->x, oid[i]->len); |
| nd[i].oid.len = oid[i]->len; |
| nd[i].next = pointers[i + 1]; |
| } |
| |
| head = pointers[from]; |
| found = mbedtls_asn1_store_named_data(&head, (const char *)needle->x, |
| needle->len, NULL, 0); |
| |
| /* In any case, the existing list structure must be unchanged. */ |
| for (i = 0; i < ARRAY_LENGTH(nd); i++) |
| TEST_ASSERT(nd[i].next == pointers[i + 1]); |
| |
| if (position >= 0) { |
| /* position should have been found and modified. */ |
| TEST_ASSERT(head == pointers[from]); |
| TEST_ASSERT(found == pointers[position]); |
| } else { |
| /* A new entry should have been created. */ |
| TEST_ASSERT(found == head); |
| TEST_ASSERT(head->next == pointers[from]); |
| for (i = 0; i < ARRAY_LENGTH(nd); i++) |
| TEST_ASSERT(found != &nd[i]); |
| } |
| |
| exit: |
| if (found != NULL && found == head && found != pointers[from]) { |
| mbedtls_free(found->oid.p); |
| mbedtls_free(found); |
| } |
| for (i = 0; i < ARRAY_LENGTH(nd); i++) |
| mbedtls_free(nd[i].oid.p); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE */ |
| void store_named_data_val_found(int old_len, int new_len) |
| { |
| mbedtls_asn1_named_data nd = { { 0x06, 3, (unsigned char *)"OID" }, |
| { 0, 0, NULL }, |
| NULL, |
| 0 }; |
| mbedtls_asn1_named_data *head = &nd; |
| mbedtls_asn1_named_data *found = NULL; |
| unsigned char *old_val = NULL; |
| unsigned char *new_val = (unsigned char *)"new value"; |
| |
| if (old_len != 0) { |
| ASSERT_ALLOC(nd.val.p, (size_t)old_len); |
| old_val = nd.val.p; |
| nd.val.len = old_len; |
| memset(old_val, 'x', old_len); |
| } |
| if (new_len <= 0) { |
| new_len = -new_len; |
| new_val = NULL; |
| } |
| |
| found = mbedtls_asn1_store_named_data(&head, "OID", 3, new_val, new_len); |
| TEST_ASSERT(head == &nd); |
| TEST_ASSERT(found == head); |
| |
| if (new_val != NULL) |
| ASSERT_COMPARE(found->val.p, found->val.len, new_val, (size_t)new_len); |
| if (new_len == 0) |
| TEST_ASSERT(found->val.p == NULL); |
| else if (new_len == old_len) |
| TEST_ASSERT(found->val.p == old_val); |
| else |
| TEST_ASSERT(found->val.p != old_val); |
| |
| exit: |
| mbedtls_free(nd.val.p); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE */ |
| void store_named_data_val_new(int new_len) |
| { |
| mbedtls_asn1_named_data *head = NULL; |
| mbedtls_asn1_named_data *found = NULL; |
| const unsigned char *oid = (unsigned char *)"OID"; |
| size_t oid_len = strlen((const char *)oid); |
| const unsigned char *new_val = (unsigned char *)"new value"; |
| |
| if (new_len <= 0) |
| new_val = NULL; |
| if (new_len < 0) |
| new_len = -new_len; |
| |
| found = mbedtls_asn1_store_named_data(&head, (const char *)oid, oid_len, |
| new_val, (size_t)new_len); |
| TEST_ASSERT(found != NULL); |
| TEST_ASSERT(found == head); |
| TEST_ASSERT(found->oid.p != oid); |
| ASSERT_COMPARE(found->oid.p, found->oid.len, oid, oid_len); |
| if (new_len == 0) |
| TEST_ASSERT(found->val.p == NULL); |
| else if (new_val == NULL) |
| TEST_ASSERT(found->val.p != NULL); |
| else { |
| TEST_ASSERT(found->val.p != new_val); |
| ASSERT_COMPARE(found->val.p, found->val.len, new_val, (size_t)new_len); |
| } |
| |
| exit: |
| if (found != NULL) { |
| mbedtls_free(found->oid.p); |
| mbedtls_free(found->val.p); |
| } |
| mbedtls_free(found); |
| } |
| /* END_CASE */ |