/*      $OpenBSD: bytestringtest.c,v 1.17 2023/01/01 17:43:04 miod Exp $        */
/*
 * 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 <string.h>

#include <openssl/crypto.h>

#include "bytestring.h"

/* This is from <openssl/base.h> in boringssl */
#define OPENSSL_U64(x) x##ULL

#define PRINT_ERROR printf("Error in %s [%s:%d]\n", __func__, __FILE__, \
                __LINE__)

#define CHECK(a) do {                                                   \
        if (!(a)) {                                                     \
                PRINT_ERROR;                                            \
                return 0;                                               \
        }                                                               \
} while (0)

#define CHECK_GOTO(a) do {                                              \
        if (!(a)) {                                                     \
                PRINT_ERROR;                                            \
                goto err;                                               \
        }                                                               \
} while (0)

static int
test_skip(void)
{
        static const uint8_t kData[] = {1, 2, 3};
        CBS data;

        CBS_init(&data, kData, sizeof(kData));

        CHECK(CBS_len(&data) == 3);
        CHECK(CBS_skip(&data, 1));
        CHECK(CBS_len(&data) == 2);
        CHECK(CBS_skip(&data, 2));
        CHECK(CBS_len(&data) == 0);
        CHECK(!CBS_skip(&data, 1));

        return 1;
}

static int
test_get_u(void)
{
        static const uint8_t kData[] = {
                1, 2, 3, 4, 5, 6, 7, 8, 9, 10,
                11, 12, 13, 14, 15, 16, 17, 18, 19, 20,
        };
        uint8_t u8;
        uint16_t u16;
        uint32_t u32;
        uint64_t u64;
        CBS data;

        CBS_init(&data, kData, sizeof(kData));

        CHECK(CBS_get_u8(&data, &u8));
        CHECK(u8 == 1);
        CHECK(CBS_get_u16(&data, &u16));
        CHECK(u16 == 0x203);
        CHECK(CBS_get_u24(&data, &u32));
        CHECK(u32 == 0x40506);
        CHECK(CBS_get_u32(&data, &u32));
        CHECK(u32 == 0x708090a);
        CHECK(CBS_get_u64(&data, &u64));
        CHECK(u64 == 0x0b0c0d0e0f101112ULL);
        CHECK(CBS_get_last_u8(&data, &u8));
        CHECK(u8 == 20);
        CHECK(CBS_get_last_u8(&data, &u8));
        CHECK(u8 == 19);
        CHECK(!CBS_get_u8(&data, &u8));
        CHECK(!CBS_get_last_u8(&data, &u8));

        return 1;
}

static int
test_get_prefixed(void)
{
        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));

        CHECK(CBS_get_u8_length_prefixed(&data, &prefixed));
        CHECK(CBS_len(&prefixed) == 1);
        CHECK(CBS_get_u8(&prefixed, &u8));
        CHECK(u8 == 2);
        CHECK(CBS_get_u16_length_prefixed(&data, &prefixed));
        CHECK(CBS_len(&prefixed) == 2);
        CHECK(CBS_get_u16(&prefixed, &u16));
        CHECK(u16 == 0x304);
        CHECK(CBS_get_u24_length_prefixed(&data, &prefixed));
        CHECK(CBS_len(&prefixed) == 3);
        CHECK(CBS_get_u24(&prefixed, &u32));
        CHECK(u32 == 0x30201);

        return 1;
}

static int
test_get_prefixed_bad(void)
{
        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));
        CHECK(!CBS_get_u8_length_prefixed(&data, &prefixed));

        CBS_init(&data, kData2, sizeof(kData2));
        CHECK(!CBS_get_u16_length_prefixed(&data, &prefixed));

        CBS_init(&data, kData3, sizeof(kData3));
        CHECK(!CBS_get_u24_length_prefixed(&data, &prefixed));

        return 1;
}

static int
test_peek_u(void)
{
        static const uint8_t kData[] = {
                1, 2, 3, 4, 5, 6, 7, 8, 9,
        };
        uint8_t u8;
        uint16_t u16;
        uint32_t u32;
        CBS data;

        CBS_init(&data, kData, sizeof(kData));

        CHECK(CBS_peek_u8(&data, &u8));
        CHECK(u8 == 1);
        CHECK(CBS_peek_u16(&data, &u16));
        CHECK(u16 == 0x102);
        CHECK(CBS_peek_u24(&data, &u32));
        CHECK(u32 == 0x10203);
        CHECK(CBS_peek_u32(&data, &u32));
        CHECK(u32 == 0x1020304);
        CHECK(CBS_get_u32(&data, &u32));
        CHECK(u32 == 0x1020304);
        CHECK(CBS_peek_last_u8(&data, &u8));
        CHECK(u8 == 9);
        CHECK(CBS_peek_u32(&data, &u32));
        CHECK(u32 == 0x5060708);
        CHECK(CBS_get_u32(&data, &u32));
        CHECK(u32 == 0x5060708);
        CHECK(CBS_get_u8(&data, &u8));
        CHECK(u8 == 9);
        CHECK(!CBS_get_u8(&data, &u8));

        return 1;
}

static int
test_get_asn1(void)
{
        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[4 + 0x80] = {0x30, 0x82, 0, 0x80};
        static const uint8_t kData6[] = {0xa1, 3, 0x4, 1, 1};
        static const uint8_t kData7[] = {0xa1, 3, 0x4, 2, 1};
        static const uint8_t kData8[] = {0xa1, 3, 0x2, 1, 1};
        static const uint8_t kData9[] = {0xa1, 3, 0x2, 1, 0xff};

        CBS data, contents;
        int present;
        uint64_t value;

        CBS_init(&data, kData1, sizeof(kData1));

        CHECK(!CBS_peek_asn1_tag(&data, 0x1));
        CHECK(CBS_peek_asn1_tag(&data, 0x30));

        CHECK(CBS_get_asn1(&data, &contents, 0x30));
        CHECK(CBS_len(&contents) == 2);
        CHECK(memcmp(CBS_data(&contents), "\x01\x02", 2) == 0);

        CBS_init(&data, kData2, sizeof(kData2));
        /* data is truncated */
        CHECK(!CBS_get_asn1(&data, &contents, 0x30));

        CBS_init(&data, kData3, sizeof(kData3));
        /* zero byte length of length */
        CHECK(!CBS_get_asn1(&data, &contents, 0x30));

        CBS_init(&data, kData4, sizeof(kData4));
        /* long form mistakenly used. */
        CHECK(!CBS_get_asn1(&data, &contents, 0x30));

        CBS_init(&data, kData5, sizeof(kData5));
        /* length takes too many bytes. */
        CHECK(!CBS_get_asn1(&data, &contents, 0x30));

        CBS_init(&data, kData1, sizeof(kData1));
        /* wrong tag. */
        CHECK(!CBS_get_asn1(&data, &contents, 0x31));

        CBS_init(&data, NULL, 0);
        /* peek at empty data. */
        CHECK(!CBS_peek_asn1_tag(&data, 0x30));

        CBS_init(&data, NULL, 0);
        /* optional elements at empty data. */
        CHECK(CBS_get_optional_asn1(&data, &contents, &present, 0xa0));
        CHECK(!present);
        CHECK(CBS_get_optional_asn1_octet_string(&data, &contents, &present,
            0xa0));
        CHECK(!present);
        CHECK(CBS_len(&contents) == 0);
        CHECK(CBS_get_optional_asn1_octet_string(&data, &contents, NULL, 0xa0));
        CHECK(CBS_len(&contents) == 0);
        CHECK(CBS_get_optional_asn1_uint64(&data, &value, 0xa0, 42));
        CHECK(value == 42);

        CBS_init(&data, kData6, sizeof(kData6));
        /* optional element. */
        CHECK(CBS_get_optional_asn1(&data, &contents, &present, 0xa0));
        CHECK(!present);
        CHECK(CBS_get_optional_asn1(&data, &contents, &present, 0xa1));
        CHECK(present);
        CHECK(CBS_len(&contents) == 3);
        CHECK(memcmp(CBS_data(&contents), "\x04\x01\x01", 3) == 0);

        CBS_init(&data, kData6, sizeof(kData6));
        /* optional octet string. */
        CHECK(CBS_get_optional_asn1_octet_string(&data, &contents, &present,
            0xa0));
        CHECK(!present);
        CHECK(CBS_len(&contents) == 0);
        CHECK(CBS_get_optional_asn1_octet_string(&data, &contents, &present,
            0xa1));
        CHECK(present);
        CHECK(CBS_len(&contents) == 1);
        CHECK(CBS_data(&contents)[0] == 1);

        CBS_init(&data, kData7, sizeof(kData7));
        /* invalid optional octet string. */
        CHECK(!CBS_get_optional_asn1_octet_string(&data, &contents, &present,
            0xa1));

        CBS_init(&data, kData8, sizeof(kData8));
        /* optional octet string. */
        CHECK(CBS_get_optional_asn1_uint64(&data, &value, 0xa0, 42));
        CHECK(value == 42);
        CHECK(CBS_get_optional_asn1_uint64(&data, &value, 0xa1, 42));
        CHECK(value == 1);

        CBS_init(&data, kData9, sizeof(kData9));
        /* invalid optional integer. */
        CHECK(!CBS_get_optional_asn1_uint64(&data, &value, 0xa1, 42));

        return 1;
}

static int
test_get_optional_asn1_bool(void)
{
        CBS data;
        int val;

        static const uint8_t kTrue[] = {0x0a, 3, CBS_ASN1_BOOLEAN, 1, 0xff};
        static const uint8_t kFalse[] = {0x0a, 3, CBS_ASN1_BOOLEAN, 1, 0x00};
        static const uint8_t kInvalid[] = {0x0a, 3, CBS_ASN1_BOOLEAN, 1, 0x01};

        CBS_init(&data, NULL, 0);
        val = 2;
        CHECK(CBS_get_optional_asn1_bool(&data, &val, 0x0a, 0));
        CHECK(val == 0);

        CBS_init(&data, kTrue, sizeof(kTrue));
        val = 2;
        CHECK(CBS_get_optional_asn1_bool(&data, &val, 0x0a, 0));
        CHECK(val == 1);

        CBS_init(&data, kFalse, sizeof(kFalse));
        val = 2;
        CHECK(CBS_get_optional_asn1_bool(&data, &val, 0x0a, 1));
        CHECK(val == 0);

        CBS_init(&data, kInvalid, sizeof(kInvalid));
        CHECK(!CBS_get_optional_asn1_bool(&data, &val, 0x0a, 1));

        return 1;
}

static int
test_cbb_basic(void)
{
        static const uint8_t kExpected[] = {
            1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12,
            13, 14, 15, 16, 17, 18, 19, 20,
        };
        uint8_t *buf = NULL;
        size_t buf_len;
        int ret = 0;
        CBB cbb;

        CHECK(CBB_init(&cbb, 100));

        CBB_cleanup(&cbb);

        CHECK(CBB_init(&cbb, 0));
        CHECK_GOTO(CBB_add_u8(&cbb, 1));
        CHECK_GOTO(CBB_add_u16(&cbb, 0x203));
        CHECK_GOTO(CBB_add_u24(&cbb, 0x40506));
        CHECK_GOTO(CBB_add_u32(&cbb, 0x708090a));
        CHECK_GOTO(CBB_add_bytes(&cbb, (const uint8_t*) "\x0b\x0c", 2));
        CHECK_GOTO(CBB_add_u64(&cbb, 0xd0e0f1011121314LL));
        CHECK_GOTO(CBB_finish(&cbb, &buf, &buf_len));

        ret = (buf_len == sizeof(kExpected)
            && memcmp(buf, kExpected, buf_len) == 0);

        if (0) {
err:
                CBB_cleanup(&cbb);
        }
        free(buf);
        return ret;
}

static int
test_cbb_add_space(void)
{
        static const uint8_t kExpected[] = {1, 2, 0, 0, 0, 0, 7, 8};
        uint8_t *buf = NULL;
        size_t buf_len;
        uint8_t *data;
        int ret = 0;
        CBB cbb;

        CHECK(CBB_init(&cbb, 100));

        CHECK_GOTO(CBB_add_u16(&cbb, 0x102));
        CHECK_GOTO(CBB_add_space(&cbb, &data, 4));
        CHECK_GOTO(CBB_add_u16(&cbb, 0x708));
        CHECK_GOTO(CBB_finish(&cbb, &buf, &buf_len));

        ret |= (buf_len == sizeof(kExpected)
            && memcmp(buf, kExpected, buf_len) == 0);

        memset(buf, 0xa5, buf_len);
        CHECK(CBB_init_fixed(&cbb, buf, buf_len));

        CHECK_GOTO(CBB_add_u16(&cbb, 0x102));
        CHECK_GOTO(CBB_add_space(&cbb, &data, 4));
        CHECK_GOTO(CBB_add_u16(&cbb, 0x708));
        CHECK_GOTO(CBB_finish(&cbb, NULL, NULL));

        ret |= (buf_len == sizeof(kExpected)
            && memcmp(buf, kExpected, buf_len) == 0);

        if (0) {
err:
                CBB_cleanup(&cbb);
        }
        free(buf);
        return ret;
}

static int
test_cbb_fixed(void)
{
        CBB cbb;
        uint8_t buf[1];
        uint8_t *out_buf = NULL;
        size_t out_size;
        int ret = 0;

        CHECK(CBB_init_fixed(&cbb, NULL, 0));
        CHECK_GOTO(!CBB_add_u8(&cbb, 1));
        CHECK_GOTO(CBB_finish(&cbb, &out_buf, &out_size));
        CHECK(out_buf == NULL && out_size == 0);

        CHECK(CBB_init_fixed(&cbb, buf, 1));
        CHECK_GOTO(CBB_add_u8(&cbb, 1));
        CHECK_GOTO(!CBB_add_u8(&cbb, 2));
        CHECK_GOTO(CBB_finish(&cbb, &out_buf, &out_size));

        ret = (out_buf == buf && out_size == 1 && buf[0] == 1);

        if (0) {
err:
                CBB_cleanup(&cbb);
        }

        return ret;
}

static int
test_cbb_finish_child(void)
{
        CBB cbb, child;
        uint8_t *out_buf = NULL;
        size_t out_size;
        int ret = 0;

        CHECK(CBB_init(&cbb, 16));
        CHECK_GOTO(CBB_add_u8_length_prefixed(&cbb, &child));
        CHECK_GOTO(!CBB_finish(&child, &out_buf, &out_size));
        CHECK_GOTO(CBB_finish(&cbb, &out_buf, &out_size));

        ret = (out_size == 1 && out_buf[0] == 0);

err:
        free(out_buf);
        return ret;
}

static int
test_cbb_prefixed(void)
{
        static const uint8_t kExpected[] = {0, 1, 1, 0, 2, 2, 3, 0, 0, 3,
            4, 5, 6, 5, 4, 1, 0, 1, 2};
        CBB cbb, contents, inner_contents, inner_inner_contents;
        uint8_t *buf = NULL;
        size_t buf_len;
        int ret = 0;

        CHECK(CBB_init(&cbb, 0));
        CHECK_GOTO(CBB_add_u8_length_prefixed(&cbb, &contents));
        CHECK_GOTO(CBB_add_u8_length_prefixed(&cbb, &contents));
        CHECK_GOTO(CBB_add_u8(&contents, 1));
        CHECK_GOTO(CBB_add_u16_length_prefixed(&cbb, &contents));
        CHECK_GOTO(CBB_add_u16(&contents, 0x203));
        CHECK_GOTO(CBB_add_u24_length_prefixed(&cbb, &contents));
        CHECK_GOTO(CBB_add_u24(&contents, 0x40506));
        CHECK_GOTO(CBB_add_u8_length_prefixed(&cbb, &contents));
        CHECK_GOTO(CBB_add_u8_length_prefixed(&contents, &inner_contents));
        CHECK_GOTO(CBB_add_u8(&inner_contents, 1));
        CHECK_GOTO(CBB_add_u16_length_prefixed(&inner_contents,
            &inner_inner_contents));
        CHECK_GOTO(CBB_add_u8(&inner_inner_contents, 2));
        CHECK_GOTO(CBB_finish(&cbb, &buf, &buf_len));

        ret = (buf_len == sizeof(kExpected)
            && memcmp(buf, kExpected, buf_len) == 0);

        if (0) {
err:
                CBB_cleanup(&cbb);
        }
        free(buf);
        return ret;
}

static int
test_cbb_discard_child(void)
{
        static const uint8_t kExpected[] = {
                0xaa,
                0,
                1, 0xbb,
                0, 2, 0xcc, 0xcc,
                0, 0, 3, 0xdd, 0xdd, 0xdd,
                1, 0xff,
        };
        CBB cbb, contents, inner_contents, inner_inner_contents;
        uint8_t *buf = NULL;
        size_t buf_len;
        int ret = 0;

        CHECK(CBB_init(&cbb, 0));
        CHECK_GOTO(CBB_add_u8(&cbb, 0xaa));

        // Discarding |cbb|'s children preserves the byte written.
        CBB_discard_child(&cbb);

        CHECK_GOTO(CBB_add_u8_length_prefixed(&cbb, &contents));
        CHECK_GOTO(CBB_add_u8_length_prefixed(&cbb, &contents));
        CHECK_GOTO(CBB_add_u8(&contents, 0xbb));
        CHECK_GOTO(CBB_add_u16_length_prefixed(&cbb, &contents));
        CHECK_GOTO(CBB_add_u16(&contents, 0xcccc));
        CHECK_GOTO(CBB_add_u24_length_prefixed(&cbb, &contents));
        CHECK_GOTO(CBB_add_u24(&contents, 0xdddddd));
        CHECK_GOTO(CBB_add_u8_length_prefixed(&cbb, &contents));
        CHECK_GOTO(CBB_add_u8(&contents, 0xff));
        CHECK_GOTO(CBB_add_u8_length_prefixed(&contents, &inner_contents));
        CHECK_GOTO(CBB_add_u8(&inner_contents, 0x42));
        CHECK_GOTO(CBB_add_u16_length_prefixed(&inner_contents,
            &inner_inner_contents));
        CHECK_GOTO(CBB_add_u8(&inner_inner_contents, 0x99));

        // Discard everything from |inner_contents| down.
        CBB_discard_child(&contents);

        CHECK_GOTO(CBB_finish(&cbb, &buf, &buf_len));

        ret = (buf_len == sizeof(kExpected)
            && memcmp(buf, kExpected, buf_len) == 0);

        if (0) {
err:
                CBB_cleanup(&cbb);
        }
        free(buf);
        return ret;
}

static int
test_cbb_misuse(void)
{
        CBB cbb, child, contents;
        uint8_t *buf = NULL;
        size_t buf_len;
        int ret = 0;

        CHECK(CBB_init(&cbb, 0));
        CHECK_GOTO(CBB_add_u8_length_prefixed(&cbb, &child));
        CHECK_GOTO(CBB_add_u8(&child, 1));
        CHECK_GOTO(CBB_add_u8(&cbb, 2));

        /*
         * Since we wrote to |cbb|, |child| is now invalid and attempts to write
         * to it should fail.
         */
        CHECK_GOTO(!CBB_add_u8(&child, 1));
        CHECK_GOTO(!CBB_add_u16(&child, 1));
        CHECK_GOTO(!CBB_add_u24(&child, 1));
        CHECK_GOTO(!CBB_add_u8_length_prefixed(&child, &contents));
        CHECK_GOTO(!CBB_add_u16_length_prefixed(&child, &contents));
        CHECK_GOTO(!CBB_add_asn1(&child, &contents, 1));
        CHECK_GOTO(!CBB_add_bytes(&child, (const uint8_t*) "a", 1));
        CHECK_GOTO(CBB_finish(&cbb, &buf, &buf_len));

        ret = (buf_len == 3 && memcmp(buf, "\x01\x01\x02", 3) == 0);

        if (0) {
err:
                CBB_cleanup(&cbb);
        }
        free(buf);
        return ret;
}

static int
test_cbb_asn1(void)
{
        static const uint8_t kExpected[] = {0x30, 3, 1, 2, 3};
        uint8_t *buf = NULL, *test_data = NULL;
        size_t buf_len;
        CBB cbb, contents, inner_contents;
        int ret = 0;
        int alloc = 0;

        CHECK_GOTO(CBB_init(&cbb, 0));
        alloc = 1;
        CHECK_GOTO(CBB_add_asn1(&cbb, &contents, 0x30));
        CHECK_GOTO(CBB_add_bytes(&contents, (const uint8_t*) "\x01\x02\x03",
            3));
        CHECK_GOTO(CBB_finish(&cbb, &buf, &buf_len));
        alloc = 0;

        CHECK_GOTO(buf_len == sizeof(kExpected));
        CHECK_GOTO(memcmp(buf, kExpected, buf_len) == 0);

        free(buf);
        buf = NULL;

        CHECK_GOTO(((test_data = malloc(100000)) != NULL));
        memset(test_data, 0x42, 100000);

        CHECK_GOTO(CBB_init(&cbb, 0));
        alloc = 1;
        CHECK_GOTO(CBB_add_asn1(&cbb, &contents, 0x30));
        CHECK_GOTO(CBB_add_bytes(&contents, test_data, 130));
        CHECK_GOTO(CBB_finish(&cbb, &buf, &buf_len));
        alloc = 0;

        CHECK_GOTO(buf_len == 3 + 130);
        CHECK_GOTO(memcmp(buf, "\x30\x81\x82", 3) == 0);
        CHECK_GOTO(memcmp(buf + 3, test_data, 130) == 0);

        free(buf);
        buf = NULL;

        CHECK_GOTO(CBB_init(&cbb, 0));
        alloc = 1;
        CHECK_GOTO(CBB_add_asn1(&cbb, &contents, 0x30));
        CHECK_GOTO(CBB_add_bytes(&contents, test_data, 1000));
        CHECK_GOTO(CBB_finish(&cbb, &buf, &buf_len));
        alloc = 0;

        CHECK_GOTO(buf_len == 4 + 1000);
        CHECK_GOTO(memcmp(buf, "\x30\x82\x03\xe8", 4) == 0);
        CHECK_GOTO(!memcmp(buf + 4, test_data, 1000));

        free(buf);
        buf = NULL;

        CHECK_GOTO(CBB_init(&cbb, 0));
        alloc = 1;
        CHECK_GOTO(CBB_add_asn1(&cbb, &contents, 0x30));
        CHECK_GOTO(CBB_add_asn1(&contents, &inner_contents, 0x30));
        CHECK_GOTO(CBB_add_bytes(&inner_contents, test_data, 100000));
        CHECK_GOTO(CBB_finish(&cbb, &buf, &buf_len));
        alloc = 0;

        CHECK_GOTO(buf_len == 5 + 5 + 100000);
        CHECK_GOTO(memcmp(buf, "\x30\x83\x01\x86\xa5\x30\x83\x01\x86\xa0", 10)
            == 0);
        CHECK_GOTO(!memcmp(buf + 10, test_data, 100000));

        ret = 1;

        if (0) {
err:
                if (alloc)
                        CBB_cleanup(&cbb);
        }
        free(buf);
        free(test_data);
        return ret;
}

static int
do_indefinite_convert(const char *name, const uint8_t *definite_expected,
    size_t definite_len, const uint8_t *indefinite, size_t indefinite_len)
{
        CBS in;
        uint8_t *out = NULL;
        size_t out_len;
        int ret = 0;

        CBS_init(&in, indefinite, indefinite_len);

        CHECK_GOTO(CBS_asn1_indefinite_to_definite(&in, &out, &out_len));

        if (out == NULL) {

                if (indefinite_len != definite_len ||
                    memcmp(definite_expected, indefinite, indefinite_len) != 0) {
                        PRINT_ERROR;
                        goto err;
                }

                return 1;
        }

        if (out_len != definite_len ||
            memcmp(out, definite_expected, definite_len) != 0) {
                PRINT_ERROR;
                goto err;
        }

        ret = 1;
err:
        free(out);
        return ret;
}

static int
test_indefinite_convert(void)
{
        static const uint8_t kSimpleBER[] = {0x01, 0x01, 0x00};

        /* kIndefBER contains a SEQUENCE with an indefinite length. */
        static const uint8_t kIndefBER[] = {0x30, 0x80, 0x01, 0x01, 0x02, 0x00,
            0x00};
        static const uint8_t kIndefDER[] = {0x30, 0x03, 0x01, 0x01, 0x02};

        /*
         * kOctetStringBER contains an indefinite length OCTETSTRING with two
         * parts.  These parts need to be concatenated in DER form.
         */
        static const uint8_t kOctetStringBER[] = {0x24, 0x80, 0x04, 0x02, 0,
            1, 0x04, 0x02, 2,    3,    0x00, 0x00};
        static const uint8_t kOctetStringDER[] = {0x04, 0x04, 0, 1, 2, 3};

        /*
         * kNSSBER is part of a PKCS#12 message generated by NSS that uses
         * indefinite length elements extensively.
         */
        static const uint8_t kNSSBER[] = {
            0x30, 0x80, 0x02, 0x01, 0x03, 0x30, 0x80, 0x06, 0x09, 0x2a, 0x86,
            0x48, 0x86, 0xf7, 0x0d, 0x01, 0x07, 0x01, 0xa0, 0x80, 0x24, 0x80,
            0x04, 0x04, 0x01, 0x02, 0x03, 0x04, 0x00, 0x00, 0x00, 0x00, 0x00,
            0x00, 0x30, 0x39, 0x30, 0x21, 0x30, 0x09, 0x06, 0x05, 0x2b, 0x0e,
            0x03, 0x02, 0x1a, 0x05, 0x00, 0x04, 0x14, 0x84, 0x98, 0xfc, 0x66,
            0x33, 0xee, 0xba, 0xe7, 0x90, 0xc1, 0xb6, 0xe8, 0x8f, 0xfe, 0x1d,
            0xc5, 0xa5, 0x97, 0x93, 0x3e, 0x04, 0x10, 0x38, 0x62, 0xc6, 0x44,
            0x12, 0xd5, 0x30, 0x00, 0xf8, 0xf2, 0x1b, 0xf0, 0x6e, 0x10, 0x9b,
            0xb8, 0x02, 0x02, 0x07, 0xd0, 0x00, 0x00,
        };

        static const uint8_t kNSSDER[] = {
            0x30, 0x53, 0x02, 0x01, 0x03, 0x30, 0x13, 0x06, 0x09, 0x2a, 0x86,
            0x48, 0x86, 0xf7, 0x0d, 0x01, 0x07, 0x01, 0xa0, 0x06, 0x04, 0x04,
            0x01, 0x02, 0x03, 0x04, 0x30, 0x39, 0x30, 0x21, 0x30, 0x09, 0x06,
            0x05, 0x2b, 0x0e, 0x03, 0x02, 0x1a, 0x05, 0x00, 0x04, 0x14, 0x84,
            0x98, 0xfc, 0x66, 0x33, 0xee, 0xba, 0xe7, 0x90, 0xc1, 0xb6, 0xe8,
            0x8f, 0xfe, 0x1d, 0xc5, 0xa5, 0x97, 0x93, 0x3e, 0x04, 0x10, 0x38,
            0x62, 0xc6, 0x44, 0x12, 0xd5, 0x30, 0x00, 0xf8, 0xf2, 0x1b, 0xf0,
            0x6e, 0x10, 0x9b, 0xb8, 0x02, 0x02, 0x07, 0xd0,
        };

        CHECK(do_indefinite_convert("kSimpleBER", kSimpleBER, sizeof(kSimpleBER),
            kSimpleBER, sizeof(kSimpleBER)));
        CHECK(do_indefinite_convert("kIndefBER", kIndefDER, sizeof(kIndefDER),
            kIndefBER, sizeof(kIndefBER)));
        CHECK(do_indefinite_convert("kOctetStringBER", kOctetStringDER,
            sizeof(kOctetStringDER), kOctetStringBER,
            sizeof(kOctetStringBER)));
        CHECK(do_indefinite_convert("kNSSBER", kNSSDER, sizeof(kNSSDER), kNSSBER,
            sizeof(kNSSBER)));

        return 1;
}

typedef struct {
        uint64_t value;
        const char *encoding;
        size_t encoding_len;
} ASN1_UINT64_TEST;

static const ASN1_UINT64_TEST kAsn1Uint64Tests[] = {
        {0, "\x02\x01\x00", 3},
        {1, "\x02\x01\x01", 3},
        {127, "\x02\x01\x7f", 3},
        {128, "\x02\x02\x00\x80", 4},
        {0xdeadbeef, "\x02\x05\x00\xde\xad\xbe\xef", 7},
        {OPENSSL_U64(0x0102030405060708),
            "\x02\x08\x01\x02\x03\x04\x05\x06\x07\x08", 10},
        {OPENSSL_U64(0xffffffffffffffff),
            "\x02\x09\x00\xff\xff\xff\xff\xff\xff\xff\xff", 11},
};

typedef struct {
        const char *encoding;
        size_t encoding_len;
} ASN1_INVALID_UINT64_TEST;

static const ASN1_INVALID_UINT64_TEST kAsn1InvalidUint64Tests[] = {
        /* Bad tag. */
        {"\x03\x01\x00", 3},
        /* Empty contents. */
        {"\x02\x00", 2},
        /* Negative number. */
        {"\x02\x01\x80", 3},
        /* Overflow. */
        {"\x02\x09\x01\x00\x00\x00\x00\x00\x00\x00\x00", 11},
        /* Leading zeros. */
        {"\x02\x02\x00\x01", 4},
};

static int
test_asn1_uint64(void)
{
        CBB cbb;
        uint8_t *out = NULL;
        size_t i;
        int ret = 0;
        int alloc = 0;

        for (i = 0; i < sizeof(kAsn1Uint64Tests) / sizeof(kAsn1Uint64Tests[0]);
             i++) {
                const ASN1_UINT64_TEST *test = &kAsn1Uint64Tests[i];
                CBS cbs;
                uint64_t value;
                size_t len;

                CBS_init(&cbs, (const uint8_t *)test->encoding,
                    test->encoding_len);

                CHECK(CBS_get_asn1_uint64(&cbs, &value));
                CHECK(CBS_len(&cbs) == 0);
                CHECK(value == test->value);

                CHECK(CBB_init(&cbb, 0));
                alloc = 1;
                CHECK_GOTO(CBB_add_asn1_uint64(&cbb, test->value));
                CHECK_GOTO(CBB_finish(&cbb, &out, &len));
                alloc = 0;

                CHECK_GOTO(len == test->encoding_len);
                CHECK_GOTO(memcmp(out, test->encoding, len) == 0);
                free(out);
                out = NULL;
        }

        for (i = 0; i < sizeof(kAsn1InvalidUint64Tests)
            / sizeof(kAsn1InvalidUint64Tests[0]); i++) {
                const ASN1_INVALID_UINT64_TEST *test =
                    &kAsn1InvalidUint64Tests[i];
                CBS cbs;
                uint64_t value;

                CBS_init(&cbs, (const uint8_t *)test->encoding,
                    test->encoding_len);
                CHECK(!CBS_get_asn1_uint64(&cbs, &value));
        }

        ret = 1;

        if (0) {
err:
                if (alloc)
                        CBB_cleanup(&cbb);
        }
        free(out);

        return ret;
}

static int
test_offset(void)
{
        uint8_t v;
        static const uint8_t input[] = {1, 2, 3, 4, 5};
        CBS data;

        CBS_init(&data, input, sizeof(input));
        CHECK(sizeof(input) == 5);
        CHECK(CBS_len(&data) == 5);
        CHECK(CBS_offset(&data) == 0);
        CHECK(CBS_get_u8(&data, &v));
        CHECK(v == 1);
        CHECK(CBS_len(&data) == 4);
        CHECK(CBS_offset(&data) == 1);
        CHECK(CBS_skip(&data, 2));
        CHECK(CBS_len(&data) == 2);
        CHECK(CBS_offset(&data) == 3);
        CHECK(CBS_get_u8(&data, &v));
        CHECK(v == 4);
        CHECK(CBS_get_u8(&data, &v));
        CHECK(v == 5);
        CHECK(CBS_len(&data) == 0);
        CHECK(CBS_offset(&data) == 5);
        CHECK(!CBS_skip(&data, 1));

        CBS_init(&data, input, sizeof(input));
        CHECK(CBS_skip(&data, 2));
        CHECK(CBS_len(&data) == 3);
        CHECK(CBS_offset(&data) == 2);
        CHECK(CBS_skip(&data, 3));
        CHECK(CBS_len(&data) == 0);
        CHECK(CBS_offset(&data) == 5);
        CHECK(!CBS_get_u8(&data, &v));

        return 1;
}

static int
test_write_bytes(void)
{
        int ret = 0;
        uint8_t v;
        size_t len;
        static const uint8_t input[] = {'f', 'o', 'o', 'b', 'a', 'r'};
        CBS data;
        uint8_t *tmp = NULL;

        CHECK_GOTO((tmp = malloc(sizeof(input))) != NULL);
        memset(tmp, 100, sizeof(input));

        CBS_init(&data, input, sizeof(input));
        CHECK_GOTO(CBS_len(&data) == 6);
        CHECK_GOTO(CBS_offset(&data) == 0);
        CHECK_GOTO(CBS_get_u8(&data, &v));
        CHECK_GOTO(v == 102 /* f */);
        CHECK_GOTO(CBS_skip(&data, 1));
        CHECK_GOTO(!CBS_skip(&data, 15));
        CHECK_GOTO(CBS_write_bytes(&data, tmp, sizeof(input), &len));
        CHECK_GOTO(len == 4);
        CHECK_GOTO(memcmp(input + 2, tmp, len) == 0);
        CHECK_GOTO(tmp[4] == 100 && tmp[5] == 100);

        ret = 1;

err:
        free(tmp);
        return ret;
}

static int
test_cbs_dup(void)
{
        CBS data, check;
        static const uint8_t input[] = {'f', 'o', 'o', 'b', 'a', 'r'};

        CBS_init(&data, input, sizeof(input));
        CHECK(CBS_len(&data) == 6);
        CBS_dup(&data, &check);
        CHECK(CBS_len(&check) == 6);
        CHECK(CBS_data(&data) == CBS_data(&check));
        CHECK(CBS_skip(&data, 1));
        CHECK(CBS_len(&data) == 5);
        CHECK(CBS_len(&check) == 6);
        CHECK(CBS_data(&data) == CBS_data(&check) + 1);
        CHECK(CBS_skip(&check, 1));
        CHECK(CBS_len(&data) == 5);
        CHECK(CBS_len(&check) == 5);
        CHECK(CBS_data(&data) == CBS_data(&check));
        CHECK(CBS_offset(&data) == 1);
        CHECK(CBS_offset(&check) == 1);

        CBS_init(&data, input, sizeof(input));
        CHECK(CBS_skip(&data, 5));
        CBS_dup(&data, &check);
        CHECK(CBS_len(&data) == 1);
        CHECK(CBS_len(&check) == 1);
        CHECK(CBS_data(&data) == input + 5);
        CHECK(CBS_data(&data) == CBS_data(&check));
        CHECK(CBS_offset(&data) == 5);
        CHECK(CBS_offset(&check) == 5);

        return 1;
}

int
main(void)
{
        int failed = 0;

        failed |= !test_skip();
        failed |= !test_get_u();
        failed |= !test_get_prefixed();
        failed |= !test_get_prefixed_bad();
        failed |= !test_peek_u();
        failed |= !test_get_asn1();
        failed |= !test_cbb_basic();
        failed |= !test_cbb_add_space();
        failed |= !test_cbb_fixed();
        failed |= !test_cbb_finish_child();
        failed |= !test_cbb_discard_child();
        failed |= !test_cbb_misuse();
        failed |= !test_cbb_prefixed();
        failed |= !test_cbb_asn1();
        failed |= !test_indefinite_convert();
        failed |= !test_asn1_uint64();
        failed |= !test_get_optional_asn1_bool();
        failed |= !test_offset();
        failed |= !test_write_bytes();
        failed |= !test_cbs_dup();

        if (!failed)
                printf("PASS\n");
        return failed;
}