1229fce07SDmitri Tikhonov/* Copyright (c) 2017 - 2019 LiteSpeed Technologies Inc.  See LICENSE. */
250aadb33SDmitri Tikhonov#include <assert.h>
350aadb33SDmitri Tikhonov#include <stdio.h>
450aadb33SDmitri Tikhonov#include <stdlib.h>
550aadb33SDmitri Tikhonov#include <string.h>
6461e84d8SAmol Deshpande#ifndef WIN32
750aadb33SDmitri Tikhonov#include <sys/time.h>
8461e84d8SAmol Deshpande#endif
950aadb33SDmitri Tikhonov
1050aadb33SDmitri Tikhonov#include "lsquic.h"
1150aadb33SDmitri Tikhonov#include "lsquic_types.h"
1250aadb33SDmitri Tikhonov#include "lsquic_alarmset.h"
1350aadb33SDmitri Tikhonov#include "lsquic_parse.h"
1450aadb33SDmitri Tikhonov
15052a1c28SDmitri Tikhonovstatic const struct parse_funcs *const pf = select_pf_by_ver(LSQVER_035);
1650aadb33SDmitri Tikhonov
1750aadb33SDmitri Tikhonov
1850aadb33SDmitri Tikhonovstruct conn_close_parse_test {
1950aadb33SDmitri Tikhonov    int             lineno;
2050aadb33SDmitri Tikhonov    unsigned char   buf[0x100];
2150aadb33SDmitri Tikhonov    size_t          buf_len;
2250aadb33SDmitri Tikhonov    uint32_t        error_code;
2350aadb33SDmitri Tikhonov    uint16_t        reason_len;
2450aadb33SDmitri Tikhonov    uint8_t         reason_off;
2550aadb33SDmitri Tikhonov    int             retval;
2650aadb33SDmitri Tikhonov};
2750aadb33SDmitri Tikhonov
2850aadb33SDmitri Tikhonovstatic const struct conn_close_parse_test parse_tests[] = {
2950aadb33SDmitri Tikhonov
3050aadb33SDmitri Tikhonov    {
3150aadb33SDmitri Tikhonov        .lineno         = __LINE__,
3250aadb33SDmitri Tikhonov        .buf            = { 0x02, 0x31, 0x00, 0x00, 0x00, 0x05, 0x00, 'D', 'u', 'd', 'e', '!', },
3350aadb33SDmitri Tikhonov        .buf_len        = 0x100,
3450aadb33SDmitri Tikhonov        .error_code     = 0x31,
3550aadb33SDmitri Tikhonov        .reason_len     = 0x05,
3650aadb33SDmitri Tikhonov        .reason_off     = 7,
3750aadb33SDmitri Tikhonov        .retval         = 7 + 5,
3850aadb33SDmitri Tikhonov    },
3950aadb33SDmitri Tikhonov
4050aadb33SDmitri Tikhonov    {
4150aadb33SDmitri Tikhonov        .lineno         = __LINE__,
4250aadb33SDmitri Tikhonov        .buf            = { 0x02, 0x31, 0x00, 0x00, 0x00, 0x05, 0x00, 'D', 'u', 'd', 'e', '!', },
4350aadb33SDmitri Tikhonov        .buf_len        = 6,    /* Too short #1 */
4450aadb33SDmitri Tikhonov        .error_code     = 0x31,
4550aadb33SDmitri Tikhonov        .reason_len     = 0x05,
4650aadb33SDmitri Tikhonov        .reason_off     = 7,
4750aadb33SDmitri Tikhonov        .retval         = -1,
4850aadb33SDmitri Tikhonov    },
4950aadb33SDmitri Tikhonov
5050aadb33SDmitri Tikhonov    {
5150aadb33SDmitri Tikhonov        .lineno         = __LINE__,
5250aadb33SDmitri Tikhonov        .buf            = { 0x02, 0x31, 0x00, 0x00, 0x00, 0x05, 0x00, 'D', 'u', 'd', 'e', '!', },
5350aadb33SDmitri Tikhonov        .buf_len        = 9,    /* Too short #2 */
5450aadb33SDmitri Tikhonov        .error_code     = 0x31,
5550aadb33SDmitri Tikhonov        .reason_len     = 0x05,
5650aadb33SDmitri Tikhonov        .reason_off     = 7,
5750aadb33SDmitri Tikhonov        .retval         = -2,
5850aadb33SDmitri Tikhonov    },
5950aadb33SDmitri Tikhonov
6050aadb33SDmitri Tikhonov    {   .buf            = { 0 },    }
6150aadb33SDmitri Tikhonov};
6250aadb33SDmitri Tikhonov
6350aadb33SDmitri Tikhonov
6450aadb33SDmitri Tikhonovstruct conn_close_gen_test {
6550aadb33SDmitri Tikhonov    int             lineno;
6650aadb33SDmitri Tikhonov    uint32_t        error_code;
6750aadb33SDmitri Tikhonov    const char     *reason;
6850aadb33SDmitri Tikhonov    int             retval;
6950aadb33SDmitri Tikhonov    unsigned char   buf[0x100];
7050aadb33SDmitri Tikhonov    size_t          buf_len;
7150aadb33SDmitri Tikhonov};
7250aadb33SDmitri Tikhonov
7350aadb33SDmitri Tikhonov
7450aadb33SDmitri Tikhonovstatic const struct conn_close_gen_test gen_tests[] = {
7550aadb33SDmitri Tikhonov
7650aadb33SDmitri Tikhonov    {
7750aadb33SDmitri Tikhonov        .lineno         = __LINE__,
7850aadb33SDmitri Tikhonov        .error_code     = 0x12345678,
7950aadb33SDmitri Tikhonov        .reason         = "Dude, where is my car?",
8050aadb33SDmitri Tikhonov        .retval         = 7 + sizeof("Dude, where is my car?") - 1,
8150aadb33SDmitri Tikhonov        .buf_len        = 0x100,
8250aadb33SDmitri Tikhonov        .buf            = {
8350aadb33SDmitri Tikhonov            0x02,
8450aadb33SDmitri Tikhonov            0x78, 0x56, 0x34, 0x12,
8550aadb33SDmitri Tikhonov            sizeof("Dude, where is my car?") - 1, 0x00,
8650aadb33SDmitri Tikhonov            'D', 'u', 'd', 'e', ',', ' ', 'w', 'h', 'e', 'r', 'e', ' ', 'i', 's', ' ', 'm', 'y', ' ', 'c', 'a', 'r', '?',
8750aadb33SDmitri Tikhonov        },
8850aadb33SDmitri Tikhonov    },
8950aadb33SDmitri Tikhonov
9050aadb33SDmitri Tikhonov    {
9150aadb33SDmitri Tikhonov        .lineno         = __LINE__,
9250aadb33SDmitri Tikhonov        .error_code     = 0x12345678,
9350aadb33SDmitri Tikhonov        .reason         = NULL,
9450aadb33SDmitri Tikhonov        .retval         = 7,
9550aadb33SDmitri Tikhonov        .buf_len        = 0x100,
9650aadb33SDmitri Tikhonov        .buf            = {
9750aadb33SDmitri Tikhonov            0x02,
9850aadb33SDmitri Tikhonov            0x78, 0x56, 0x34, 0x12,
9950aadb33SDmitri Tikhonov            0x00, 0x00, /* Zero-sized string */
10050aadb33SDmitri Tikhonov        },
10150aadb33SDmitri Tikhonov    },
10250aadb33SDmitri Tikhonov
10350aadb33SDmitri Tikhonov    {
10450aadb33SDmitri Tikhonov        .lineno         = __LINE__,
10550aadb33SDmitri Tikhonov        .error_code     = 0x12345678,
10650aadb33SDmitri Tikhonov        .reason         = "Dude, where is my car?",
10750aadb33SDmitri Tikhonov        .retval         = -1,   /* Too short */
10850aadb33SDmitri Tikhonov        .buf_len        = 0x10,
10950aadb33SDmitri Tikhonov    },
11050aadb33SDmitri Tikhonov
11150aadb33SDmitri Tikhonov    {   .buf            = { 0 },    }
11250aadb33SDmitri Tikhonov
11350aadb33SDmitri Tikhonov};
11450aadb33SDmitri Tikhonov
11550aadb33SDmitri Tikhonov
11650aadb33SDmitri Tikhonovstatic void
11750aadb33SDmitri Tikhonovrun_parse_tests (void)
11850aadb33SDmitri Tikhonov{
11950aadb33SDmitri Tikhonov    const struct conn_close_parse_test *test;
12050aadb33SDmitri Tikhonov    for (test = parse_tests; test->buf[0]; ++test)
12150aadb33SDmitri Tikhonov    {
12250aadb33SDmitri Tikhonov        uint32_t error_code = ~0;
12350aadb33SDmitri Tikhonov        uint16_t reason_len = ~0;
12450aadb33SDmitri Tikhonov        uint8_t reason_off = ~0;
12550aadb33SDmitri Tikhonov        int sz = pf->pf_parse_connect_close_frame(test->buf, test->buf_len,
12650aadb33SDmitri Tikhonov                                        &error_code, &reason_len, &reason_off);
12750aadb33SDmitri Tikhonov        assert(sz == test->retval);
12850aadb33SDmitri Tikhonov        if (0 == sz)
12950aadb33SDmitri Tikhonov        {
13050aadb33SDmitri Tikhonov            assert(test->error_code == error_code);
13150aadb33SDmitri Tikhonov            assert(test->reason_len == reason_len);
13250aadb33SDmitri Tikhonov            assert(test->reason_off == reason_off);
13350aadb33SDmitri Tikhonov        }
13450aadb33SDmitri Tikhonov    }
13550aadb33SDmitri Tikhonov}
13650aadb33SDmitri Tikhonov
13750aadb33SDmitri Tikhonov
13850aadb33SDmitri Tikhonovstatic void
13950aadb33SDmitri Tikhonovrun_gen_tests (void)
14050aadb33SDmitri Tikhonov{
14150aadb33SDmitri Tikhonov    const struct conn_close_gen_test *test;
14250aadb33SDmitri Tikhonov    for (test = gen_tests; test->buf[0]; ++test)
14350aadb33SDmitri Tikhonov    {
14450aadb33SDmitri Tikhonov        unsigned char buf[0x100];
14550aadb33SDmitri Tikhonov        int sz = pf->pf_gen_connect_close_frame(buf, sizeof(buf),
14650aadb33SDmitri Tikhonov                    test->error_code, test->reason,
14750aadb33SDmitri Tikhonov                    test->reason ? strlen(test->reason) : 0);
14850aadb33SDmitri Tikhonov        assert(sz == test->retval);
14950aadb33SDmitri Tikhonov        if (0 == sz)
15050aadb33SDmitri Tikhonov            assert(0 == memcmp(test->buf, buf, sz));
15150aadb33SDmitri Tikhonov    }
15250aadb33SDmitri Tikhonov}
15350aadb33SDmitri Tikhonov
15450aadb33SDmitri Tikhonov
15550aadb33SDmitri Tikhonovint
15650aadb33SDmitri Tikhonovmain (void)
15750aadb33SDmitri Tikhonov{
15850aadb33SDmitri Tikhonov    run_parse_tests();
15950aadb33SDmitri Tikhonov    run_gen_tests();
16050aadb33SDmitri Tikhonov    return 0;
16150aadb33SDmitri Tikhonov}
162