| /* Copyright (c) 2014, Google Inc. |
| * |
| * Permission to use, copy, modify, and/or distribute this software for any |
| * purpose with or without fee is hereby granted, provided that the above |
| * copyright notice and this permission notice appear in all copies. |
| * |
| * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES |
| * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF |
| * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY |
| * SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES |
| * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION |
| * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN |
| * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */ |
| |
| #include <stdio.h> |
| #include <stdlib.h> |
| |
| #include <openssl/bytestring.h> |
| |
| |
| static int test_skip() { |
| static const uint8_t kData[] = {1, 2, 3}; |
| CBS data; |
| |
| CBS_init(&data, kData, sizeof(kData)); |
| return CBS_len(&data) == 3 && |
| CBS_skip(&data, 1) && |
| CBS_len(&data) == 2 && |
| CBS_skip(&data, 2) && |
| CBS_len(&data) == 0 && |
| !CBS_skip(&data, 1); |
| } |
| |
| static int test_get_u() { |
| static const uint8_t kData[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}; |
| uint8_t u8; |
| uint16_t u16; |
| uint32_t u32; |
| CBS data; |
| |
| CBS_init(&data, kData, sizeof(kData)); |
| return CBS_get_u8(&data, &u8) && |
| u8 == 1 && |
| CBS_get_u16(&data, &u16) && |
| u16 == 0x203 && |
| CBS_get_u24(&data, &u32) && |
| u32 == 0x40506 && |
| CBS_get_u32(&data, &u32) && |
| u32 == 0x708090a && |
| !CBS_get_u8(&data, &u8); |
| } |
| |
| static int test_get_prefixed() { |
| static const uint8_t kData[] = {1, 2, 0, 2, 3, 4, 0, 0, 3, 3, 2, 1}; |
| uint8_t u8; |
| uint16_t u16; |
| uint32_t u32; |
| CBS data, prefixed; |
| |
| CBS_init(&data, kData, sizeof(kData)); |
| return CBS_get_u8_length_prefixed(&data, &prefixed) && |
| CBS_len(&prefixed) == 1 && |
| CBS_get_u8(&prefixed, &u8) && |
| u8 == 2 && |
| CBS_get_u16_length_prefixed(&data, &prefixed) && |
| CBS_len(&prefixed) == 2 && |
| CBS_get_u16(&prefixed, &u16) && |
| u16 == 0x304 && |
| CBS_get_u24_length_prefixed(&data, &prefixed) && |
| CBS_len(&prefixed) == 3 && |
| CBS_get_u24(&prefixed, &u32) && |
| u32 == 0x30201; |
| } |
| |
| static int test_get_prefixed_bad() { |
| static const uint8_t kData1[] = {2, 1}; |
| static const uint8_t kData2[] = {0, 2, 1}; |
| static const uint8_t kData3[] = {0, 0, 2, 1}; |
| CBS data, prefixed; |
| |
| CBS_init(&data, kData1, sizeof(kData1)); |
| if (CBS_get_u8_length_prefixed(&data, &prefixed)) { |
| return 0; |
| } |
| |
| CBS_init(&data, kData2, sizeof(kData2)); |
| if (CBS_get_u16_length_prefixed(&data, &prefixed)) { |
| return 0; |
| } |
| |
| CBS_init(&data, kData3, sizeof(kData3)); |
| if (CBS_get_u24_length_prefixed(&data, &prefixed)) { |
| return 0; |
| } |
| |
| return 1; |
| } |
| |
| static int test_get_asn1() { |
| static const uint8_t kData1[] = {0x30, 2, 1, 2}; |
| static const uint8_t kData2[] = {0x30, 3, 1, 2}; |
| static const uint8_t kData3[] = {0x30, 0x80}; |
| static const uint8_t kData4[] = {0x30, 0x81, 1, 1}; |
| static const uint8_t kData5[] = {0x30, 0x82, 0, 1, 1}; |
| |
| CBS data, contents; |
| |
| CBS_init(&data, kData1, sizeof(kData1)); |
| if (!CBS_get_asn1(&data, &contents, 0x30) || |
| CBS_len(&contents) != 2 || |
| memcmp(CBS_data(&contents), "\x01\x02", 2) != 0) { |
| return 0; |
| } |
| |
| CBS_init(&data, kData2, sizeof(kData2)); |
| /* data is truncated */ |
| if (CBS_get_asn1(&data, &contents, 0x30)) { |
| return 0; |
| } |
| |
| CBS_init(&data, kData3, sizeof(kData3)); |
| /* zero byte length of length */ |
| if (CBS_get_asn1(&data, &contents, 0x30)) { |
| return 0; |
| } |
| |
| CBS_init(&data, kData4, sizeof(kData4)); |
| /* long form mistakenly used. */ |
| if (CBS_get_asn1(&data, &contents, 0x30)) { |
| return 0; |
| } |
| |
| CBS_init(&data, kData5, sizeof(kData5)); |
| /* length takes too many bytes. */ |
| if (CBS_get_asn1(&data, &contents, 0x30)) { |
| return 0; |
| } |
| |
| CBS_init(&data, kData1, sizeof(kData1)); |
| /* wrong tag. */ |
| if (CBS_get_asn1(&data, &contents, 0x31)) { |
| return 0; |
| } |
| |
| return 1; |
| } |
| |
| static int test_get_indef() { |
| static const uint8_t kData1[] = {0x30, 0x80, 0x00, 0x00}; |
| static const uint8_t kDataWithoutEOC[] = {0x30, 0x80, 0x01, 0x00}; |
| static const uint8_t kDataWithBadInternalLength[] = {0x30, 0x80, 0x01, 0x01}; |
| static const uint8_t kDataNested[] = {0x30, 0x80, 0x30, 0x80, 0x30, 0x80, |
| 0x00, 0x00, 0x00, 0x00, 0x00, 0x00}; |
| static const uint8_t kDataPrimitive[] = {0x02, 0x80, 0x00, 0x00}; |
| |
| CBS data, contents; |
| CBS_init(&data, kData1, sizeof(kData1)); |
| if (CBS_get_asn1(&data, &contents, 0x30)) { |
| /* Indefinite lengths should not be supported in DER mode. */ |
| fprintf(stderr, "Indefinite length parsed by CBS_get_asn1.\n"); |
| return 0; |
| } |
| |
| if (!CBS_get_asn1_ber(&data, &contents, 0x30) || |
| CBS_len(&contents) != 0 || |
| CBS_len(&data) != 0) { |
| fprintf(stderr, "Simple indefinite length failed.\n"); |
| return 0; |
| } |
| |
| CBS_init(&data, kDataWithoutEOC, sizeof(kDataWithoutEOC)); |
| if (CBS_get_asn1_ber(&data, &contents, 0x30)) { |
| fprintf(stderr, "Parsed without EOC.\n"); |
| return 0; |
| } |
| |
| CBS_init(&data, kDataWithBadInternalLength, |
| sizeof(kDataWithBadInternalLength)); |
| if (CBS_get_asn1_ber(&data, &contents, 0x30)) { |
| fprintf(stderr, "Parsed with internal length.\n"); |
| return 0; |
| } |
| |
| CBS_init(&data, kDataNested, sizeof(kDataNested)); |
| if (!CBS_get_asn1_ber(&data, &contents, 0x30) || |
| CBS_len(&contents) != 8 || |
| CBS_len(&data) != 0) { |
| fprintf(stderr, "Nested indefinite lengths failed.\n"); |
| return 0; |
| } |
| |
| CBS_init(&data, kDataPrimitive, sizeof(kDataPrimitive)); |
| if (CBS_get_asn1_ber(&data, &contents, 0x02)) { |
| /* Indefinite lengths should not be supported for non-constructed |
| * elements. */ |
| fprintf(stderr, "Parsed non-constructed element with indefinite length\n"); |
| return 0; |
| } |
| |
| return 1; |
| } |
| |
| static int test_cbb_basic() { |
| static const uint8_t kExpected[] = {1, 2, 3, 4, 5, 6, 7, 8}; |
| uint8_t *buf; |
| size_t buf_len; |
| int ok; |
| CBB cbb; |
| |
| if (!CBB_init(&cbb, 100)) { |
| return 0; |
| } |
| CBB_cleanup(&cbb); |
| |
| if (!CBB_init(&cbb, 0) || |
| !CBB_add_u8(&cbb, 1) || |
| !CBB_add_u16(&cbb, 0x203) || |
| !CBB_add_u24(&cbb, 0x40506) || |
| !CBB_add_bytes(&cbb, (const uint8_t*) "\x07\x08", 2) || |
| !CBB_finish(&cbb, &buf, &buf_len)) { |
| return 0; |
| } |
| |
| ok = buf_len == sizeof(kExpected) && memcmp(buf, kExpected, buf_len) == 0; |
| free(buf); |
| return ok; |
| } |
| |
| static int test_cbb_fixed() { |
| CBB cbb; |
| uint8_t buf[1]; |
| uint8_t *out_buf; |
| size_t out_size; |
| |
| if (!CBB_init_fixed(&cbb, NULL, 0) || |
| CBB_add_u8(&cbb, 1) || |
| !CBB_finish(&cbb, &out_buf, &out_size) || |
| out_buf != NULL || |
| out_size != 0) { |
| return 0; |
| } |
| |
| if (!CBB_init_fixed(&cbb, buf, 1) || |
| !CBB_add_u8(&cbb, 1) || |
| CBB_add_u8(&cbb, 2) || |
| !CBB_finish(&cbb, &out_buf, &out_size) || |
| out_buf != buf || |
| out_size != 1 || |
| buf[0] != 1) { |
| return 0; |
| } |
| |
| return 1; |
| } |
| |
| static int test_cbb_finish_child() { |
| CBB cbb, child; |
| uint8_t *out_buf; |
| size_t out_size; |
| |
| if (!CBB_init(&cbb, 16) || |
| !CBB_add_u8_length_prefixed(&cbb, &child) || |
| CBB_finish(&child, &out_buf, &out_size) || |
| !CBB_finish(&cbb, &out_buf, &out_size) || |
| out_size != 1 || |
| out_buf[0] != 0) { |
| return 0; |
| } |
| |
| free(out_buf); |
| return 1; |
| } |
| |
| static int test_cbb_prefixed() { |
| static const uint8_t kExpected[] = {0, 1, 1, 0, 2, 2, 3, 0, 0, 3, |
| 4, 5, 6, 5, 4, 1, 0, 1, 2}; |
| uint8_t *buf; |
| size_t buf_len; |
| CBB cbb, contents, inner_contents, inner_inner_contents; |
| int ok; |
| |
| if (!CBB_init(&cbb, 0) || |
| !CBB_add_u8_length_prefixed(&cbb, &contents) || |
| !CBB_add_u8_length_prefixed(&cbb, &contents) || |
| !CBB_add_u8(&contents, 1) || |
| !CBB_add_u16_length_prefixed(&cbb, &contents) || |
| !CBB_add_u16(&contents, 0x203) || |
| !CBB_add_u24_length_prefixed(&cbb, &contents) || |
| !CBB_add_u24(&contents, 0x40506) || |
| !CBB_add_u8_length_prefixed(&cbb, &contents) || |
| !CBB_add_u8_length_prefixed(&contents, &inner_contents) || |
| !CBB_add_u8(&inner_contents, 1) || |
| !CBB_add_u16_length_prefixed(&inner_contents, &inner_inner_contents) || |
| !CBB_add_u8(&inner_inner_contents, 2) || |
| !CBB_finish(&cbb, &buf, &buf_len)) { |
| return 0; |
| } |
| |
| ok = buf_len == sizeof(kExpected) && memcmp(buf, kExpected, buf_len) == 0; |
| free(buf); |
| return ok; |
| } |
| |
| static int test_cbb_misuse() { |
| CBB cbb, child, contents; |
| uint8_t *buf; |
| size_t buf_len; |
| |
| if (!CBB_init(&cbb, 0) || |
| !CBB_add_u8_length_prefixed(&cbb, &child) || |
| !CBB_add_u8(&child, 1) || |
| !CBB_add_u8(&cbb, 2)) { |
| return 0; |
| } |
| |
| /* Since we wrote to |cbb|, |child| is now invalid and attempts to write to |
| * it should fail. */ |
| if (CBB_add_u8(&child, 1) || |
| CBB_add_u16(&child, 1) || |
| CBB_add_u24(&child, 1) || |
| CBB_add_u8_length_prefixed(&child, &contents) || |
| CBB_add_u16_length_prefixed(&child, &contents) || |
| CBB_add_asn1(&child, &contents, 1) || |
| CBB_add_bytes(&child, (const uint8_t*) "a", 1)) { |
| fprintf(stderr, "CBB operation on invalid CBB did not fail.\n"); |
| return 0; |
| } |
| |
| if (!CBB_finish(&cbb, &buf, &buf_len) || |
| buf_len != 3 || |
| memcmp(buf, "\x01\x01\x02", 3) != 0) { |
| return 0; |
| } |
| |
| free(buf); |
| |
| return 1; |
| } |
| |
| static int test_cbb_asn1() { |
| static const uint8_t kExpected[] = {0x30, 3, 1, 2, 3}; |
| uint8_t *buf, *test_data; |
| size_t buf_len; |
| CBB cbb, contents, inner_contents; |
| |
| if (!CBB_init(&cbb, 0) || |
| !CBB_add_asn1(&cbb, &contents, 0x30) || |
| !CBB_add_bytes(&contents, (const uint8_t*) "\x01\x02\x03", 3) || |
| !CBB_finish(&cbb, &buf, &buf_len)) { |
| return 0; |
| } |
| |
| if (buf_len != sizeof(kExpected) || memcmp(buf, kExpected, buf_len) != 0) { |
| return 0; |
| } |
| free(buf); |
| |
| test_data = malloc(100000); |
| memset(test_data, 0x42, 100000); |
| |
| if (!CBB_init(&cbb, 0) || |
| !CBB_add_asn1(&cbb, &contents, 0x30) || |
| !CBB_add_bytes(&contents, test_data, 130) || |
| !CBB_finish(&cbb, &buf, &buf_len)) { |
| return 0; |
| } |
| |
| if (buf_len != 3 + 130 || |
| memcmp(buf, "\x30\x81\x82", 3) != 0 || |
| memcmp(buf + 3, test_data, 130) != 0) { |
| return 0; |
| } |
| free(buf); |
| |
| if (!CBB_init(&cbb, 0) || |
| !CBB_add_asn1(&cbb, &contents, 0x30) || |
| !CBB_add_bytes(&contents, test_data, 1000) || |
| !CBB_finish(&cbb, &buf, &buf_len)) { |
| return 0; |
| } |
| |
| if (buf_len != 4 + 1000 || |
| memcmp(buf, "\x30\x82\x03\xe8", 4) != 0 || |
| memcmp(buf + 4, test_data, 1000)) { |
| return 0; |
| } |
| free(buf); |
| |
| if (!CBB_init(&cbb, 0) || |
| !CBB_add_asn1(&cbb, &contents, 0x30) || |
| !CBB_add_asn1(&contents, &inner_contents, 0x30) || |
| !CBB_add_bytes(&inner_contents, test_data, 100000) || |
| !CBB_finish(&cbb, &buf, &buf_len)) { |
| return 0; |
| } |
| |
| if (buf_len != 5 + 5 + 100000 || |
| memcmp(buf, "\x30\x83\x01\x86\xa5\x30\x83\x01\x86\xa0", 10) != 0 || |
| memcmp(buf + 10, test_data, 100000)) { |
| return 0; |
| } |
| free(buf); |
| |
| free(test_data); |
| return 1; |
| } |
| |
| int main() { |
| if (!test_skip() || |
| !test_get_u() || |
| !test_get_prefixed() || |
| !test_get_prefixed_bad() || |
| !test_get_asn1() || |
| !test_get_indef() || |
| !test_cbb_basic() || |
| !test_cbb_fixed() || |
| !test_cbb_finish_child() || |
| !test_cbb_misuse() || |
| !test_cbb_prefixed() || |
| !test_cbb_asn1()) { |
| return 1; |
| } |
| |
| printf("PASS\n"); |
| return 0; |
| } |