echo_client.c revision 4429f8ea
17d09751dSDmitri Tikhonov/* Copyright (c) 2017 - 2020 LiteSpeed Technologies Inc.  See LICENSE. */
20adf085aSDmitri Tikhonov/*
30adf085aSDmitri Tikhonov * echo_client.c -- This is really a "line client:" it connects to QUIC server
40adf085aSDmitri Tikhonov * and sends it stuff, line by line.  It works in tandem with echo_server.
50adf085aSDmitri Tikhonov */
60adf085aSDmitri Tikhonov
70adf085aSDmitri Tikhonov#include <assert.h>
80adf085aSDmitri Tikhonov#include <errno.h>
90adf085aSDmitri Tikhonov#include <stdio.h>
100adf085aSDmitri Tikhonov#include <stdlib.h>
110adf085aSDmitri Tikhonov#include <string.h>
120adf085aSDmitri Tikhonov#include <sys/queue.h>
130adf085aSDmitri Tikhonov#include <sys/types.h>
140adf085aSDmitri Tikhonov#include <sys/stat.h>
15fb3e20e0SDmitri Tikhonov
16fb3e20e0SDmitri Tikhonov#ifndef WIN32
170adf085aSDmitri Tikhonov#include <fcntl.h>
18fb3e20e0SDmitri Tikhonov#include <unistd.h>
19fb3e20e0SDmitri Tikhonov#define Read read
20fb3e20e0SDmitri Tikhonov#else
21fb3e20e0SDmitri Tikhonov#include "vc_compat.h"
22fb3e20e0SDmitri Tikhonov#include "getopt.h"
23fb3e20e0SDmitri Tikhonov#include <io.h>
24fb3e20e0SDmitri Tikhonov#define Read _read
25fb3e20e0SDmitri Tikhonov#define STDIN_FILENO 0
26fb3e20e0SDmitri Tikhonov#endif
270adf085aSDmitri Tikhonov
280adf085aSDmitri Tikhonov#include <event2/event.h>
290adf085aSDmitri Tikhonov
300adf085aSDmitri Tikhonov#include "lsquic.h"
310adf085aSDmitri Tikhonov#include "test_common.h"
320adf085aSDmitri Tikhonov#include "prog.h"
330adf085aSDmitri Tikhonov
340adf085aSDmitri Tikhonov#include "../src/liblsquic/lsquic_logger.h"
350adf085aSDmitri Tikhonov
360adf085aSDmitri Tikhonovstruct lsquic_conn_ctx;
370adf085aSDmitri Tikhonov
380adf085aSDmitri Tikhonovstruct echo_client_ctx {
390adf085aSDmitri Tikhonov    struct lsquic_conn_ctx  *conn_h;
400adf085aSDmitri Tikhonov    struct prog                 *prog;
410adf085aSDmitri Tikhonov};
420adf085aSDmitri Tikhonov
430adf085aSDmitri Tikhonovstruct lsquic_conn_ctx {
440adf085aSDmitri Tikhonov    lsquic_conn_t       *conn;
450adf085aSDmitri Tikhonov    struct echo_client_ctx   *client_ctx;
460adf085aSDmitri Tikhonov};
470adf085aSDmitri Tikhonov
480adf085aSDmitri Tikhonov
490adf085aSDmitri Tikhonovstatic lsquic_conn_ctx_t *
500adf085aSDmitri Tikhonovecho_client_on_new_conn (void *stream_if_ctx, lsquic_conn_t *conn)
510adf085aSDmitri Tikhonov{
520adf085aSDmitri Tikhonov    struct echo_client_ctx *client_ctx = stream_if_ctx;
530adf085aSDmitri Tikhonov    lsquic_conn_ctx_t *conn_h = malloc(sizeof(*conn_h));
540adf085aSDmitri Tikhonov    conn_h->conn = conn;
550adf085aSDmitri Tikhonov    conn_h->client_ctx = client_ctx;
560adf085aSDmitri Tikhonov    client_ctx->conn_h = conn_h;
570adf085aSDmitri Tikhonov    lsquic_conn_make_stream(conn);
580adf085aSDmitri Tikhonov    return conn_h;
590adf085aSDmitri Tikhonov}
600adf085aSDmitri Tikhonov
610adf085aSDmitri Tikhonov
620adf085aSDmitri Tikhonovstatic void
630adf085aSDmitri Tikhonovecho_client_on_conn_closed (lsquic_conn_t *conn)
640adf085aSDmitri Tikhonov{
650adf085aSDmitri Tikhonov    lsquic_conn_ctx_t *conn_h = lsquic_conn_get_ctx(conn);
660adf085aSDmitri Tikhonov    LSQ_NOTICE("Connection closed");
670adf085aSDmitri Tikhonov    prog_stop(conn_h->client_ctx->prog);
680adf085aSDmitri Tikhonov    free(conn_h);
690adf085aSDmitri Tikhonov}
700adf085aSDmitri Tikhonov
710adf085aSDmitri Tikhonov
720adf085aSDmitri Tikhonovstruct lsquic_stream_ctx {
730adf085aSDmitri Tikhonov    lsquic_stream_t     *stream;
740adf085aSDmitri Tikhonov    struct echo_client_ctx   *client_ctx;
750adf085aSDmitri Tikhonov    struct event        *read_stdin_ev;
760adf085aSDmitri Tikhonov    char                 buf[0x100];
770adf085aSDmitri Tikhonov    size_t               buf_off;
780adf085aSDmitri Tikhonov};
790adf085aSDmitri Tikhonov
800adf085aSDmitri Tikhonov
810adf085aSDmitri Tikhonovstatic void
82fb3e20e0SDmitri Tikhonovread_stdin (evutil_socket_t fd, short what, void *ctx)
830adf085aSDmitri Tikhonov{
840adf085aSDmitri Tikhonov    ssize_t nr;
850adf085aSDmitri Tikhonov    lsquic_stream_ctx_t *st_h = ctx;
860adf085aSDmitri Tikhonov
87fb3e20e0SDmitri Tikhonov    nr = Read(fd, st_h->buf + st_h->buf_off++, 1);
880adf085aSDmitri Tikhonov    LSQ_DEBUG("read %zd bytes from stdin", nr);
890adf085aSDmitri Tikhonov    if (0 == nr)
900adf085aSDmitri Tikhonov    {
910adf085aSDmitri Tikhonov        lsquic_stream_shutdown(st_h->stream, 2);
920adf085aSDmitri Tikhonov    }
930adf085aSDmitri Tikhonov    else if (-1 == nr)
940adf085aSDmitri Tikhonov    {
950adf085aSDmitri Tikhonov        perror("read");
960adf085aSDmitri Tikhonov        exit(1);
970adf085aSDmitri Tikhonov    }
980adf085aSDmitri Tikhonov    else if ('\n' == st_h->buf[ st_h->buf_off - 1 ])
990adf085aSDmitri Tikhonov    {
1000adf085aSDmitri Tikhonov        LSQ_DEBUG("read newline: wantwrite");
1010adf085aSDmitri Tikhonov        lsquic_stream_wantwrite(st_h->stream, 1);
1020adf085aSDmitri Tikhonov        lsquic_engine_process_conns(st_h->client_ctx->prog->prog_engine);
1030adf085aSDmitri Tikhonov    }
1040adf085aSDmitri Tikhonov    else if (st_h->buf_off == sizeof(st_h->buf))
1050adf085aSDmitri Tikhonov    {
1060adf085aSDmitri Tikhonov        LSQ_NOTICE("line too long");
1070adf085aSDmitri Tikhonov        exit(2);
1080adf085aSDmitri Tikhonov    }
1090adf085aSDmitri Tikhonov    else
1100adf085aSDmitri Tikhonov        event_add(st_h->read_stdin_ev, NULL);
1110adf085aSDmitri Tikhonov}
1120adf085aSDmitri Tikhonov
1130adf085aSDmitri Tikhonov
1140adf085aSDmitri Tikhonovstatic lsquic_stream_ctx_t *
1150adf085aSDmitri Tikhonovecho_client_on_new_stream (void *stream_if_ctx, lsquic_stream_t *stream)
1160adf085aSDmitri Tikhonov{
1170adf085aSDmitri Tikhonov    lsquic_stream_ctx_t *st_h = calloc(1, sizeof(*st_h));
1180adf085aSDmitri Tikhonov    st_h->stream = stream;
1190adf085aSDmitri Tikhonov    st_h->client_ctx = stream_if_ctx;
1200adf085aSDmitri Tikhonov    st_h->buf_off = 0;
1210adf085aSDmitri Tikhonov    st_h->read_stdin_ev = event_new(prog_eb(st_h->client_ctx->prog),
1220adf085aSDmitri Tikhonov                                    STDIN_FILENO, EV_READ, read_stdin, st_h);
1230adf085aSDmitri Tikhonov    event_add(st_h->read_stdin_ev, NULL);
1240adf085aSDmitri Tikhonov    return st_h;
1250adf085aSDmitri Tikhonov}
1260adf085aSDmitri Tikhonov
1270adf085aSDmitri Tikhonov
1280adf085aSDmitri Tikhonovstatic void
1290adf085aSDmitri Tikhonovecho_client_on_read (lsquic_stream_t *stream, lsquic_stream_ctx_t *st_h)
1300adf085aSDmitri Tikhonov{
1310adf085aSDmitri Tikhonov    char c;
1320adf085aSDmitri Tikhonov    size_t nr;
1330adf085aSDmitri Tikhonov
1340adf085aSDmitri Tikhonov    nr = lsquic_stream_read(stream, &c, 1);
1350adf085aSDmitri Tikhonov    if (0 == nr)
1360adf085aSDmitri Tikhonov    {
1370adf085aSDmitri Tikhonov        lsquic_stream_shutdown(stream, 2);
1380adf085aSDmitri Tikhonov        return;
1390adf085aSDmitri Tikhonov    }
1400adf085aSDmitri Tikhonov    printf("%c", c);
1410adf085aSDmitri Tikhonov    fflush(stdout);
1420adf085aSDmitri Tikhonov    if ('\n' == c)
1430adf085aSDmitri Tikhonov    {
1440adf085aSDmitri Tikhonov        event_add(st_h->read_stdin_ev, NULL);
1450adf085aSDmitri Tikhonov        lsquic_stream_wantread(stream, 0);
1460adf085aSDmitri Tikhonov    }
1470adf085aSDmitri Tikhonov}
1480adf085aSDmitri Tikhonov
1490adf085aSDmitri Tikhonov
1500adf085aSDmitri Tikhonovstatic void
1510adf085aSDmitri Tikhonovecho_client_on_write (lsquic_stream_t *stream, lsquic_stream_ctx_t *st_h)
1520adf085aSDmitri Tikhonov{
1530adf085aSDmitri Tikhonov    /* Here we make an assumption that we can write the whole buffer.
1540adf085aSDmitri Tikhonov     * Don't do it in a real program.
1550adf085aSDmitri Tikhonov     */
1560adf085aSDmitri Tikhonov    lsquic_stream_write(stream, st_h->buf, st_h->buf_off);
1570adf085aSDmitri Tikhonov    st_h->buf_off = 0;
1580adf085aSDmitri Tikhonov
1590adf085aSDmitri Tikhonov    lsquic_stream_flush(stream);
1600adf085aSDmitri Tikhonov    lsquic_stream_wantwrite(stream, 0);
1610adf085aSDmitri Tikhonov    lsquic_stream_wantread(stream, 1);
1620adf085aSDmitri Tikhonov}
1630adf085aSDmitri Tikhonov
1640adf085aSDmitri Tikhonov
1650adf085aSDmitri Tikhonovstatic void
1660adf085aSDmitri Tikhonovecho_client_on_close (lsquic_stream_t *stream, lsquic_stream_ctx_t *st_h)
1670adf085aSDmitri Tikhonov{
1680adf085aSDmitri Tikhonov    LSQ_NOTICE("%s called", __func__);
1690adf085aSDmitri Tikhonov    if (st_h->read_stdin_ev)
1700adf085aSDmitri Tikhonov    {
1710adf085aSDmitri Tikhonov        event_del(st_h->read_stdin_ev);
1720adf085aSDmitri Tikhonov        event_free(st_h->read_stdin_ev);
1730adf085aSDmitri Tikhonov    }
1740adf085aSDmitri Tikhonov    free(st_h);
1750adf085aSDmitri Tikhonov    lsquic_conn_close(lsquic_stream_conn(stream));
1760adf085aSDmitri Tikhonov}
1770adf085aSDmitri Tikhonov
1780adf085aSDmitri Tikhonov
1790adf085aSDmitri Tikhonovconst struct lsquic_stream_if client_echo_stream_if = {
1800adf085aSDmitri Tikhonov    .on_new_conn            = echo_client_on_new_conn,
1810adf085aSDmitri Tikhonov    .on_conn_closed         = echo_client_on_conn_closed,
1820adf085aSDmitri Tikhonov    .on_new_stream          = echo_client_on_new_stream,
1830adf085aSDmitri Tikhonov    .on_read                = echo_client_on_read,
1840adf085aSDmitri Tikhonov    .on_write               = echo_client_on_write,
1850adf085aSDmitri Tikhonov    .on_close               = echo_client_on_close,
1860adf085aSDmitri Tikhonov};
1870adf085aSDmitri Tikhonov
1880adf085aSDmitri Tikhonov
1890adf085aSDmitri Tikhonovstatic void
1900adf085aSDmitri Tikhonovusage (const char *prog)
1910adf085aSDmitri Tikhonov{
1920adf085aSDmitri Tikhonov    const char *const slash = strrchr(prog, '/');
1930adf085aSDmitri Tikhonov    if (slash)
1940adf085aSDmitri Tikhonov        prog = slash + 1;
1950adf085aSDmitri Tikhonov    LSQ_NOTICE(
1960adf085aSDmitri Tikhonov"Usage: %s [opts]\n"
1970adf085aSDmitri Tikhonov"\n"
1980adf085aSDmitri Tikhonov"Options:\n"
1990adf085aSDmitri Tikhonov            , prog);
2000adf085aSDmitri Tikhonov}
2010adf085aSDmitri Tikhonov
2020adf085aSDmitri Tikhonov
2030adf085aSDmitri Tikhonovint
2040adf085aSDmitri Tikhonovmain (int argc, char **argv)
2050adf085aSDmitri Tikhonov{
2060adf085aSDmitri Tikhonov    int opt, s;
2070adf085aSDmitri Tikhonov    struct sport_head sports;
2080adf085aSDmitri Tikhonov    struct prog prog;
2090adf085aSDmitri Tikhonov    struct echo_client_ctx client_ctx;
2100adf085aSDmitri Tikhonov
2110adf085aSDmitri Tikhonov    memset(&client_ctx, 0, sizeof(client_ctx));
2120adf085aSDmitri Tikhonov    client_ctx.prog = &prog;
2130adf085aSDmitri Tikhonov
2140adf085aSDmitri Tikhonov    TAILQ_INIT(&sports);
2150adf085aSDmitri Tikhonov    prog_init(&prog, 0, &sports, &client_echo_stream_if, &client_ctx);
2164429f8eaSDmitri Tikhonov    prog.prog_api.ea_alpn = "echo";
2170adf085aSDmitri Tikhonov
2180adf085aSDmitri Tikhonov    while (-1 != (opt = getopt(argc, argv, PROG_OPTS "h")))
2190adf085aSDmitri Tikhonov    {
2200adf085aSDmitri Tikhonov        switch (opt) {
2210adf085aSDmitri Tikhonov        case 'h':
2220adf085aSDmitri Tikhonov            usage(argv[0]);
2230adf085aSDmitri Tikhonov            prog_print_common_options(&prog, stdout);
2240adf085aSDmitri Tikhonov            exit(0);
2250adf085aSDmitri Tikhonov        default:
2260adf085aSDmitri Tikhonov            if (0 != prog_set_opt(&prog, opt, optarg))
2270adf085aSDmitri Tikhonov                exit(1);
2280adf085aSDmitri Tikhonov        }
2290adf085aSDmitri Tikhonov    }
2300adf085aSDmitri Tikhonov
231fb3e20e0SDmitri Tikhonov#ifndef WIN32
2320adf085aSDmitri Tikhonov    int flags = fcntl(STDIN_FILENO, F_GETFL);
2330adf085aSDmitri Tikhonov    flags |= O_NONBLOCK;
2340adf085aSDmitri Tikhonov    if (0 != fcntl(STDIN_FILENO, F_SETFL, flags))
2350adf085aSDmitri Tikhonov    {
2360adf085aSDmitri Tikhonov        perror("fcntl");
2370adf085aSDmitri Tikhonov        exit(1);
2380adf085aSDmitri Tikhonov    }
239fb3e20e0SDmitri Tikhonov#else
240fb3e20e0SDmitri Tikhonov    {
241fb3e20e0SDmitri Tikhonov        u_long on = 1;
242fb3e20e0SDmitri Tikhonov        ioctlsocket(STDIN_FILENO, FIONBIO, &on);
243fb3e20e0SDmitri Tikhonov    }
244fb3e20e0SDmitri Tikhonov#endif
2450adf085aSDmitri Tikhonov
2460adf085aSDmitri Tikhonov    if (0 != prog_prep(&prog))
2470adf085aSDmitri Tikhonov    {
2480adf085aSDmitri Tikhonov        LSQ_ERROR("could not prep");
2490adf085aSDmitri Tikhonov        exit(EXIT_FAILURE);
2500adf085aSDmitri Tikhonov    }
2510adf085aSDmitri Tikhonov    if (0 != prog_connect(&prog, NULL, 0))
2520adf085aSDmitri Tikhonov    {
2530adf085aSDmitri Tikhonov        LSQ_ERROR("could not connect");
2540adf085aSDmitri Tikhonov        exit(EXIT_FAILURE);
2550adf085aSDmitri Tikhonov    }
2560adf085aSDmitri Tikhonov
2570adf085aSDmitri Tikhonov    LSQ_DEBUG("entering event loop");
2580adf085aSDmitri Tikhonov
2590adf085aSDmitri Tikhonov    s = prog_run(&prog);
2600adf085aSDmitri Tikhonov    prog_cleanup(&prog);
2610adf085aSDmitri Tikhonov
2620adf085aSDmitri Tikhonov    exit(0 == s ? EXIT_SUCCESS : EXIT_FAILURE);
2630adf085aSDmitri Tikhonov}
264