162 lines
4.1 KiB
C
162 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;
|
|
}
|