litespeed-quic/tests/test_conn_close_gquic_le.c
2020-05-17 12:42:32 -04:00

161 lines
4.1 KiB
C

/* Copyright (c) 2017 - 2019 LiteSpeed Technologies Inc. See LICENSE. */
#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#ifndef WIN32
#include <sys/time.h>
#endif
#include "lsquic.h"
#include "lsquic_types.h"
#include "lsquic_alarmset.h"
#include "lsquic_parse.h"
static const struct parse_funcs *const pf = select_pf_by_ver(LSQVER_035);
struct conn_close_parse_test {
int lineno;
unsigned char buf[0x100];
size_t buf_len;
uint32_t error_code;
uint16_t reason_len;
uint8_t reason_off;
int retval;
};
static const struct conn_close_parse_test parse_tests[] = {
{
.lineno = __LINE__,
.buf = { 0x02, 0x31, 0x00, 0x00, 0x00, 0x05, 0x00, 'D', 'u', 'd', 'e', '!', },
.buf_len = 0x100,
.error_code = 0x31,
.reason_len = 0x05,
.reason_off = 7,
.retval = 7 + 5,
},
{
.lineno = __LINE__,
.buf = { 0x02, 0x31, 0x00, 0x00, 0x00, 0x05, 0x00, 'D', 'u', 'd', 'e', '!', },
.buf_len = 6, /* Too short #1 */
.error_code = 0x31,
.reason_len = 0x05,
.reason_off = 7,
.retval = -1,
},
{
.lineno = __LINE__,
.buf = { 0x02, 0x31, 0x00, 0x00, 0x00, 0x05, 0x00, 'D', 'u', 'd', 'e', '!', },
.buf_len = 9, /* Too short #2 */
.error_code = 0x31,
.reason_len = 0x05,
.reason_off = 7,
.retval = -2,
},
{ .buf = { 0 }, }
};
struct conn_close_gen_test {
int lineno;
uint32_t error_code;
const char *reason;
int retval;
unsigned char buf[0x100];
size_t buf_len;
};
static const struct conn_close_gen_test gen_tests[] = {
{
.lineno = __LINE__,
.error_code = 0x12345678,
.reason = "Dude, where is my car?",
.retval = 7 + sizeof("Dude, where is my car?") - 1,
.buf_len = 0x100,
.buf = {
0x02,
0x78, 0x56, 0x34, 0x12,
sizeof("Dude, where is my car?") - 1, 0x00,
'D', 'u', 'd', 'e', ',', ' ', 'w', 'h', 'e', 'r', 'e', ' ', 'i', 's', ' ', 'm', 'y', ' ', 'c', 'a', 'r', '?',
},
},
{
.lineno = __LINE__,
.error_code = 0x12345678,
.reason = NULL,
.retval = 7,
.buf_len = 0x100,
.buf = {
0x02,
0x78, 0x56, 0x34, 0x12,
0x00, 0x00, /* Zero-sized string */
},
},
{
.lineno = __LINE__,
.error_code = 0x12345678,
.reason = "Dude, where is my car?",
.retval = -1, /* Too short */
.buf_len = 0x10,
},
{ .buf = { 0 }, }
};
static void
run_parse_tests (void)
{
const struct conn_close_parse_test *test;
for (test = parse_tests; test->buf[0]; ++test)
{
uint32_t error_code = ~0;
uint16_t reason_len = ~0;
uint8_t reason_off = ~0;
int sz = pf->pf_parse_connect_close_frame(test->buf, test->buf_len,
&error_code, &reason_len, &reason_off);
assert(sz == test->retval);
if (0 == sz)
{
assert(test->error_code == error_code);
assert(test->reason_len == reason_len);
assert(test->reason_off == reason_off);
}
}
}
static void
run_gen_tests (void)
{
const struct conn_close_gen_test *test;
for (test = gen_tests; test->buf[0]; ++test)
{
unsigned char buf[0x100];
int sz = pf->pf_gen_connect_close_frame(buf, sizeof(buf),
test->error_code, test->reason,
test->reason ? strlen(test->reason) : 0);
assert(sz == test->retval);
if (0 == sz)
assert(0 == memcmp(test->buf, buf, sz));
}
}
int
main (void)
{
run_parse_tests();
run_gen_tests();
return 0;
}