158 lines
3.9 KiB
C
158 lines
3.9 KiB
C
/* Copyright (c) 2017 - 2022 LiteSpeed Technologies Inc. See LICENSE. */
|
|
/*
|
|
* Test client's and server's common components.
|
|
*/
|
|
|
|
#ifndef TEST_COMMON_H
|
|
#define TEST_COMMON_H 1
|
|
|
|
#if __linux__
|
|
# include <net/if.h> /* For IFNAMSIZ */
|
|
#endif
|
|
|
|
#ifdef WIN32
|
|
#include <winsock2.h>
|
|
#include <ws2tcpip.h>
|
|
#endif
|
|
|
|
struct lsquic_engine;
|
|
struct lsquic_engine_settings;
|
|
struct lsquic_out_spec;
|
|
struct event_base;
|
|
struct event;
|
|
struct packets_in;
|
|
struct lsquic_conn;
|
|
struct prog;
|
|
struct reader_ctx;
|
|
struct lsxpack_header;
|
|
|
|
#ifndef WIN32
|
|
# define SOCKOPT_VAL int
|
|
# define SOCKET_TYPE int
|
|
# define CLOSE_SOCKET close
|
|
# define CHAR_CAST
|
|
#else
|
|
# define SOCKOPT_VAL DWORD
|
|
# define SOCKET_TYPE SOCKET
|
|
# define CLOSE_SOCKET closesocket
|
|
# define CHAR_CAST (char *)
|
|
#endif
|
|
|
|
enum sport_flags
|
|
{
|
|
#if LSQUIC_DONTFRAG_SUPPORTED
|
|
SPORT_FRAGMENT_OK = (1 << 0),
|
|
#endif
|
|
SPORT_SET_SNDBUF = (1 << 1), /* SO_SNDBUF */
|
|
SPORT_SET_RCVBUF = (1 << 2), /* SO_RCVBUF */
|
|
SPORT_SERVER = (1 << 3),
|
|
SPORT_CONNECT = (1 << 4),
|
|
};
|
|
|
|
struct service_port {
|
|
TAILQ_ENTRY(service_port) next_sport;
|
|
#ifndef WIN32
|
|
int fd;
|
|
#else
|
|
SOCKET fd;
|
|
#endif
|
|
#if __linux__
|
|
uint32_t n_dropped;
|
|
int drop_init;
|
|
char if_name[IFNAMSIZ];
|
|
#endif
|
|
struct event *ev;
|
|
struct lsquic_engine *engine;
|
|
void *conn_ctx;
|
|
char host[80];
|
|
struct sockaddr_storage sas;
|
|
struct sockaddr_storage sp_local_addr;
|
|
struct packets_in *packs_in;
|
|
enum sport_flags sp_flags;
|
|
SOCKOPT_VAL sp_sndbuf; /* If SPORT_SET_SNDBUF is set */
|
|
SOCKOPT_VAL sp_rcvbuf; /* If SPORT_SET_RCVBUF is set */
|
|
struct prog *sp_prog;
|
|
unsigned char *sp_token_buf;
|
|
size_t sp_token_sz;
|
|
};
|
|
|
|
TAILQ_HEAD(sport_head, service_port);
|
|
|
|
struct service_port *
|
|
sport_new (const char *optarg, struct prog *);
|
|
|
|
void
|
|
sport_destroy (struct service_port *);
|
|
|
|
int
|
|
sport_init_server (struct service_port *, struct lsquic_engine *,
|
|
struct event_base *);
|
|
|
|
int
|
|
sport_init_client (struct service_port *, struct lsquic_engine *,
|
|
struct event_base *);
|
|
|
|
int
|
|
sport_packets_out (void *ctx, const struct lsquic_out_spec *, unsigned count);
|
|
|
|
int
|
|
sport_set_token (struct service_port *, const char *);
|
|
|
|
int
|
|
set_engine_option (struct lsquic_engine_settings *,
|
|
int *version_cleared, const char *name_value);
|
|
|
|
struct packout_buf;
|
|
|
|
struct packout_buf_allocator
|
|
{
|
|
unsigned n_out, /* Number of buffers outstanding */
|
|
max; /* Maximum outstanding. Zero mean no limit */
|
|
SLIST_HEAD(, packout_buf) free_packout_bufs;
|
|
};
|
|
|
|
void
|
|
pba_init (struct packout_buf_allocator *, unsigned max);
|
|
|
|
void *
|
|
pba_allocate (void *packout_buf_allocator, void*, lsquic_conn_ctx_t *, unsigned short, char);
|
|
|
|
void
|
|
pba_release (void *packout_buf_allocator, void *, void *obj, char);
|
|
|
|
void
|
|
pba_cleanup (struct packout_buf_allocator *);
|
|
|
|
void
|
|
print_conn_info (const struct lsquic_conn *conn);
|
|
|
|
size_t
|
|
test_reader_size (void *void_ctx);
|
|
|
|
size_t
|
|
test_reader_read (void *void_ctx, void *buf, size_t count);
|
|
|
|
struct reader_ctx *
|
|
create_lsquic_reader_ctx (const char *filename);
|
|
|
|
void
|
|
destroy_lsquic_reader_ctx (struct reader_ctx *ctx);
|
|
|
|
#define STRINGIFY(x) #x
|
|
#define TOSTRING(x) STRINGIFY(x)
|
|
#define LITESPEED_ID "lsquic" "/" TOSTRING(LSQUIC_MAJOR_VERSION) "." \
|
|
TOSTRING(LSQUIC_MINOR_VERSION) "." TOSTRING(LSQUIC_PATCH_VERSION)
|
|
|
|
struct header_buf
|
|
{
|
|
unsigned off;
|
|
char buf[UINT16_MAX];
|
|
};
|
|
|
|
int
|
|
header_set_ptr (struct lsxpack_header *hdr, struct header_buf *header_buf,
|
|
const char *name, size_t name_len,
|
|
const char *val, size_t val_len);
|
|
|
|
#endif
|