123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901 |
- // SPDX-License-Identifier: GPL-2.0-only
- /*
- * vsock_test - vsock.ko test suite
- *
- * Copyright (C) 2017 Red Hat, Inc.
- *
- * Author: Stefan Hajnoczi <[email protected]>
- */
- #include <getopt.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include <errno.h>
- #include <unistd.h>
- #include <linux/kernel.h>
- #include <sys/types.h>
- #include <sys/socket.h>
- #include <time.h>
- #include <sys/mman.h>
- #include <poll.h>
- #include "timeout.h"
- #include "control.h"
- #include "util.h"
- static void test_stream_connection_reset(const struct test_opts *opts)
- {
- union {
- struct sockaddr sa;
- struct sockaddr_vm svm;
- } addr = {
- .svm = {
- .svm_family = AF_VSOCK,
- .svm_port = 1234,
- .svm_cid = opts->peer_cid,
- },
- };
- int ret;
- int fd;
- fd = socket(AF_VSOCK, SOCK_STREAM, 0);
- timeout_begin(TIMEOUT);
- do {
- ret = connect(fd, &addr.sa, sizeof(addr.svm));
- timeout_check("connect");
- } while (ret < 0 && errno == EINTR);
- timeout_end();
- if (ret != -1) {
- fprintf(stderr, "expected connect(2) failure, got %d\n", ret);
- exit(EXIT_FAILURE);
- }
- if (errno != ECONNRESET) {
- fprintf(stderr, "unexpected connect(2) errno %d\n", errno);
- exit(EXIT_FAILURE);
- }
- close(fd);
- }
- static void test_stream_bind_only_client(const struct test_opts *opts)
- {
- union {
- struct sockaddr sa;
- struct sockaddr_vm svm;
- } addr = {
- .svm = {
- .svm_family = AF_VSOCK,
- .svm_port = 1234,
- .svm_cid = opts->peer_cid,
- },
- };
- int ret;
- int fd;
- /* Wait for the server to be ready */
- control_expectln("BIND");
- fd = socket(AF_VSOCK, SOCK_STREAM, 0);
- timeout_begin(TIMEOUT);
- do {
- ret = connect(fd, &addr.sa, sizeof(addr.svm));
- timeout_check("connect");
- } while (ret < 0 && errno == EINTR);
- timeout_end();
- if (ret != -1) {
- fprintf(stderr, "expected connect(2) failure, got %d\n", ret);
- exit(EXIT_FAILURE);
- }
- if (errno != ECONNRESET) {
- fprintf(stderr, "unexpected connect(2) errno %d\n", errno);
- exit(EXIT_FAILURE);
- }
- /* Notify the server that the client has finished */
- control_writeln("DONE");
- close(fd);
- }
- static void test_stream_bind_only_server(const struct test_opts *opts)
- {
- union {
- struct sockaddr sa;
- struct sockaddr_vm svm;
- } addr = {
- .svm = {
- .svm_family = AF_VSOCK,
- .svm_port = 1234,
- .svm_cid = VMADDR_CID_ANY,
- },
- };
- int fd;
- fd = socket(AF_VSOCK, SOCK_STREAM, 0);
- if (bind(fd, &addr.sa, sizeof(addr.svm)) < 0) {
- perror("bind");
- exit(EXIT_FAILURE);
- }
- /* Notify the client that the server is ready */
- control_writeln("BIND");
- /* Wait for the client to finish */
- control_expectln("DONE");
- close(fd);
- }
- static void test_stream_client_close_client(const struct test_opts *opts)
- {
- int fd;
- fd = vsock_stream_connect(opts->peer_cid, 1234);
- if (fd < 0) {
- perror("connect");
- exit(EXIT_FAILURE);
- }
- send_byte(fd, 1, 0);
- close(fd);
- }
- static void test_stream_client_close_server(const struct test_opts *opts)
- {
- int fd;
- fd = vsock_stream_accept(VMADDR_CID_ANY, 1234, NULL);
- if (fd < 0) {
- perror("accept");
- exit(EXIT_FAILURE);
- }
- /* Wait for the remote to close the connection, before check
- * -EPIPE error on send.
- */
- vsock_wait_remote_close(fd);
- send_byte(fd, -EPIPE, 0);
- recv_byte(fd, 1, 0);
- recv_byte(fd, 0, 0);
- close(fd);
- }
- static void test_stream_server_close_client(const struct test_opts *opts)
- {
- int fd;
- fd = vsock_stream_connect(opts->peer_cid, 1234);
- if (fd < 0) {
- perror("connect");
- exit(EXIT_FAILURE);
- }
- /* Wait for the remote to close the connection, before check
- * -EPIPE error on send.
- */
- vsock_wait_remote_close(fd);
- send_byte(fd, -EPIPE, 0);
- recv_byte(fd, 1, 0);
- recv_byte(fd, 0, 0);
- close(fd);
- }
- static void test_stream_server_close_server(const struct test_opts *opts)
- {
- int fd;
- fd = vsock_stream_accept(VMADDR_CID_ANY, 1234, NULL);
- if (fd < 0) {
- perror("accept");
- exit(EXIT_FAILURE);
- }
- send_byte(fd, 1, 0);
- close(fd);
- }
- /* With the standard socket sizes, VMCI is able to support about 100
- * concurrent stream connections.
- */
- #define MULTICONN_NFDS 100
- static void test_stream_multiconn_client(const struct test_opts *opts)
- {
- int fds[MULTICONN_NFDS];
- int i;
- for (i = 0; i < MULTICONN_NFDS; i++) {
- fds[i] = vsock_stream_connect(opts->peer_cid, 1234);
- if (fds[i] < 0) {
- perror("connect");
- exit(EXIT_FAILURE);
- }
- }
- for (i = 0; i < MULTICONN_NFDS; i++) {
- if (i % 2)
- recv_byte(fds[i], 1, 0);
- else
- send_byte(fds[i], 1, 0);
- }
- for (i = 0; i < MULTICONN_NFDS; i++)
- close(fds[i]);
- }
- static void test_stream_multiconn_server(const struct test_opts *opts)
- {
- int fds[MULTICONN_NFDS];
- int i;
- for (i = 0; i < MULTICONN_NFDS; i++) {
- fds[i] = vsock_stream_accept(VMADDR_CID_ANY, 1234, NULL);
- if (fds[i] < 0) {
- perror("accept");
- exit(EXIT_FAILURE);
- }
- }
- for (i = 0; i < MULTICONN_NFDS; i++) {
- if (i % 2)
- send_byte(fds[i], 1, 0);
- else
- recv_byte(fds[i], 1, 0);
- }
- for (i = 0; i < MULTICONN_NFDS; i++)
- close(fds[i]);
- }
- static void test_stream_msg_peek_client(const struct test_opts *opts)
- {
- int fd;
- fd = vsock_stream_connect(opts->peer_cid, 1234);
- if (fd < 0) {
- perror("connect");
- exit(EXIT_FAILURE);
- }
- send_byte(fd, 1, 0);
- close(fd);
- }
- static void test_stream_msg_peek_server(const struct test_opts *opts)
- {
- int fd;
- fd = vsock_stream_accept(VMADDR_CID_ANY, 1234, NULL);
- if (fd < 0) {
- perror("accept");
- exit(EXIT_FAILURE);
- }
- recv_byte(fd, 1, MSG_PEEK);
- recv_byte(fd, 1, 0);
- close(fd);
- }
- #define MESSAGES_CNT 7
- #define MSG_EOR_IDX (MESSAGES_CNT / 2)
- static void test_seqpacket_msg_bounds_client(const struct test_opts *opts)
- {
- int fd;
- fd = vsock_seqpacket_connect(opts->peer_cid, 1234);
- if (fd < 0) {
- perror("connect");
- exit(EXIT_FAILURE);
- }
- /* Send several messages, one with MSG_EOR flag */
- for (int i = 0; i < MESSAGES_CNT; i++)
- send_byte(fd, 1, (i == MSG_EOR_IDX) ? MSG_EOR : 0);
- control_writeln("SENDDONE");
- close(fd);
- }
- static void test_seqpacket_msg_bounds_server(const struct test_opts *opts)
- {
- int fd;
- char buf[16];
- struct msghdr msg = {0};
- struct iovec iov = {0};
- fd = vsock_seqpacket_accept(VMADDR_CID_ANY, 1234, NULL);
- if (fd < 0) {
- perror("accept");
- exit(EXIT_FAILURE);
- }
- control_expectln("SENDDONE");
- iov.iov_base = buf;
- iov.iov_len = sizeof(buf);
- msg.msg_iov = &iov;
- msg.msg_iovlen = 1;
- for (int i = 0; i < MESSAGES_CNT; i++) {
- if (recvmsg(fd, &msg, 0) != 1) {
- perror("message bound violated");
- exit(EXIT_FAILURE);
- }
- if ((i == MSG_EOR_IDX) ^ !!(msg.msg_flags & MSG_EOR)) {
- perror("MSG_EOR");
- exit(EXIT_FAILURE);
- }
- }
- close(fd);
- }
- #define MESSAGE_TRUNC_SZ 32
- static void test_seqpacket_msg_trunc_client(const struct test_opts *opts)
- {
- int fd;
- char buf[MESSAGE_TRUNC_SZ];
- fd = vsock_seqpacket_connect(opts->peer_cid, 1234);
- if (fd < 0) {
- perror("connect");
- exit(EXIT_FAILURE);
- }
- if (send(fd, buf, sizeof(buf), 0) != sizeof(buf)) {
- perror("send failed");
- exit(EXIT_FAILURE);
- }
- control_writeln("SENDDONE");
- close(fd);
- }
- static void test_seqpacket_msg_trunc_server(const struct test_opts *opts)
- {
- int fd;
- char buf[MESSAGE_TRUNC_SZ / 2];
- struct msghdr msg = {0};
- struct iovec iov = {0};
- fd = vsock_seqpacket_accept(VMADDR_CID_ANY, 1234, NULL);
- if (fd < 0) {
- perror("accept");
- exit(EXIT_FAILURE);
- }
- control_expectln("SENDDONE");
- iov.iov_base = buf;
- iov.iov_len = sizeof(buf);
- msg.msg_iov = &iov;
- msg.msg_iovlen = 1;
- ssize_t ret = recvmsg(fd, &msg, MSG_TRUNC);
- if (ret != MESSAGE_TRUNC_SZ) {
- printf("%zi\n", ret);
- perror("MSG_TRUNC doesn't work");
- exit(EXIT_FAILURE);
- }
- if (!(msg.msg_flags & MSG_TRUNC)) {
- fprintf(stderr, "MSG_TRUNC expected\n");
- exit(EXIT_FAILURE);
- }
- close(fd);
- }
- static time_t current_nsec(void)
- {
- struct timespec ts;
- if (clock_gettime(CLOCK_REALTIME, &ts)) {
- perror("clock_gettime(3) failed");
- exit(EXIT_FAILURE);
- }
- return (ts.tv_sec * 1000000000ULL) + ts.tv_nsec;
- }
- #define RCVTIMEO_TIMEOUT_SEC 1
- #define READ_OVERHEAD_NSEC 250000000 /* 0.25 sec */
- static void test_seqpacket_timeout_client(const struct test_opts *opts)
- {
- int fd;
- struct timeval tv;
- char dummy;
- time_t read_enter_ns;
- time_t read_overhead_ns;
- fd = vsock_seqpacket_connect(opts->peer_cid, 1234);
- if (fd < 0) {
- perror("connect");
- exit(EXIT_FAILURE);
- }
- tv.tv_sec = RCVTIMEO_TIMEOUT_SEC;
- tv.tv_usec = 0;
- if (setsockopt(fd, SOL_SOCKET, SO_RCVTIMEO, (void *)&tv, sizeof(tv)) == -1) {
- perror("setsockopt 'SO_RCVTIMEO'");
- exit(EXIT_FAILURE);
- }
- read_enter_ns = current_nsec();
- if (read(fd, &dummy, sizeof(dummy)) != -1) {
- fprintf(stderr,
- "expected 'dummy' read(2) failure\n");
- exit(EXIT_FAILURE);
- }
- if (errno != EAGAIN) {
- perror("EAGAIN expected");
- exit(EXIT_FAILURE);
- }
- read_overhead_ns = current_nsec() - read_enter_ns -
- 1000000000ULL * RCVTIMEO_TIMEOUT_SEC;
- if (read_overhead_ns > READ_OVERHEAD_NSEC) {
- fprintf(stderr,
- "too much time in read(2), %lu > %i ns\n",
- read_overhead_ns, READ_OVERHEAD_NSEC);
- exit(EXIT_FAILURE);
- }
- control_writeln("WAITDONE");
- close(fd);
- }
- static void test_seqpacket_timeout_server(const struct test_opts *opts)
- {
- int fd;
- fd = vsock_seqpacket_accept(VMADDR_CID_ANY, 1234, NULL);
- if (fd < 0) {
- perror("accept");
- exit(EXIT_FAILURE);
- }
- control_expectln("WAITDONE");
- close(fd);
- }
- #define BUF_PATTERN_1 'a'
- #define BUF_PATTERN_2 'b'
- static void test_seqpacket_invalid_rec_buffer_client(const struct test_opts *opts)
- {
- int fd;
- unsigned char *buf1;
- unsigned char *buf2;
- int buf_size = getpagesize() * 3;
- fd = vsock_seqpacket_connect(opts->peer_cid, 1234);
- if (fd < 0) {
- perror("connect");
- exit(EXIT_FAILURE);
- }
- buf1 = malloc(buf_size);
- if (!buf1) {
- perror("'malloc()' for 'buf1'");
- exit(EXIT_FAILURE);
- }
- buf2 = malloc(buf_size);
- if (!buf2) {
- perror("'malloc()' for 'buf2'");
- exit(EXIT_FAILURE);
- }
- memset(buf1, BUF_PATTERN_1, buf_size);
- memset(buf2, BUF_PATTERN_2, buf_size);
- if (send(fd, buf1, buf_size, 0) != buf_size) {
- perror("send failed");
- exit(EXIT_FAILURE);
- }
- if (send(fd, buf2, buf_size, 0) != buf_size) {
- perror("send failed");
- exit(EXIT_FAILURE);
- }
- close(fd);
- }
- static void test_seqpacket_invalid_rec_buffer_server(const struct test_opts *opts)
- {
- int fd;
- unsigned char *broken_buf;
- unsigned char *valid_buf;
- int page_size = getpagesize();
- int buf_size = page_size * 3;
- ssize_t res;
- int prot = PROT_READ | PROT_WRITE;
- int flags = MAP_PRIVATE | MAP_ANONYMOUS;
- int i;
- fd = vsock_seqpacket_accept(VMADDR_CID_ANY, 1234, NULL);
- if (fd < 0) {
- perror("accept");
- exit(EXIT_FAILURE);
- }
- /* Setup first buffer. */
- broken_buf = mmap(NULL, buf_size, prot, flags, -1, 0);
- if (broken_buf == MAP_FAILED) {
- perror("mmap for 'broken_buf'");
- exit(EXIT_FAILURE);
- }
- /* Unmap "hole" in buffer. */
- if (munmap(broken_buf + page_size, page_size)) {
- perror("'broken_buf' setup");
- exit(EXIT_FAILURE);
- }
- valid_buf = mmap(NULL, buf_size, prot, flags, -1, 0);
- if (valid_buf == MAP_FAILED) {
- perror("mmap for 'valid_buf'");
- exit(EXIT_FAILURE);
- }
- /* Try to fill buffer with unmapped middle. */
- res = read(fd, broken_buf, buf_size);
- if (res != -1) {
- fprintf(stderr,
- "expected 'broken_buf' read(2) failure, got %zi\n",
- res);
- exit(EXIT_FAILURE);
- }
- if (errno != ENOMEM) {
- perror("unexpected errno of 'broken_buf'");
- exit(EXIT_FAILURE);
- }
- /* Try to fill valid buffer. */
- res = read(fd, valid_buf, buf_size);
- if (res < 0) {
- perror("unexpected 'valid_buf' read(2) failure");
- exit(EXIT_FAILURE);
- }
- if (res != buf_size) {
- fprintf(stderr,
- "invalid 'valid_buf' read(2), expected %i, got %zi\n",
- buf_size, res);
- exit(EXIT_FAILURE);
- }
- for (i = 0; i < buf_size; i++) {
- if (valid_buf[i] != BUF_PATTERN_2) {
- fprintf(stderr,
- "invalid pattern for 'valid_buf' at %i, expected %hhX, got %hhX\n",
- i, BUF_PATTERN_2, valid_buf[i]);
- exit(EXIT_FAILURE);
- }
- }
- /* Unmap buffers. */
- munmap(broken_buf, page_size);
- munmap(broken_buf + page_size * 2, page_size);
- munmap(valid_buf, buf_size);
- close(fd);
- }
- #define RCVLOWAT_BUF_SIZE 128
- static void test_stream_poll_rcvlowat_server(const struct test_opts *opts)
- {
- int fd;
- int i;
- fd = vsock_stream_accept(VMADDR_CID_ANY, 1234, NULL);
- if (fd < 0) {
- perror("accept");
- exit(EXIT_FAILURE);
- }
- /* Send 1 byte. */
- send_byte(fd, 1, 0);
- control_writeln("SRVSENT");
- /* Wait until client is ready to receive rest of data. */
- control_expectln("CLNSENT");
- for (i = 0; i < RCVLOWAT_BUF_SIZE - 1; i++)
- send_byte(fd, 1, 0);
- /* Keep socket in active state. */
- control_expectln("POLLDONE");
- close(fd);
- }
- static void test_stream_poll_rcvlowat_client(const struct test_opts *opts)
- {
- unsigned long lowat_val = RCVLOWAT_BUF_SIZE;
- char buf[RCVLOWAT_BUF_SIZE];
- struct pollfd fds;
- ssize_t read_res;
- short poll_flags;
- int fd;
- fd = vsock_stream_connect(opts->peer_cid, 1234);
- if (fd < 0) {
- perror("connect");
- exit(EXIT_FAILURE);
- }
- if (setsockopt(fd, SOL_SOCKET, SO_RCVLOWAT,
- &lowat_val, sizeof(lowat_val))) {
- perror("setsockopt");
- exit(EXIT_FAILURE);
- }
- control_expectln("SRVSENT");
- /* At this point, server sent 1 byte. */
- fds.fd = fd;
- poll_flags = POLLIN | POLLRDNORM;
- fds.events = poll_flags;
- /* Try to wait for 1 sec. */
- if (poll(&fds, 1, 1000) < 0) {
- perror("poll");
- exit(EXIT_FAILURE);
- }
- /* poll() must return nothing. */
- if (fds.revents) {
- fprintf(stderr, "Unexpected poll result %hx\n",
- fds.revents);
- exit(EXIT_FAILURE);
- }
- /* Tell server to send rest of data. */
- control_writeln("CLNSENT");
- /* Poll for data. */
- if (poll(&fds, 1, 10000) < 0) {
- perror("poll");
- exit(EXIT_FAILURE);
- }
- /* Only these two bits are expected. */
- if (fds.revents != poll_flags) {
- fprintf(stderr, "Unexpected poll result %hx\n",
- fds.revents);
- exit(EXIT_FAILURE);
- }
- /* Use MSG_DONTWAIT, if call is going to wait, EAGAIN
- * will be returned.
- */
- read_res = recv(fd, buf, sizeof(buf), MSG_DONTWAIT);
- if (read_res != RCVLOWAT_BUF_SIZE) {
- fprintf(stderr, "Unexpected recv result %zi\n",
- read_res);
- exit(EXIT_FAILURE);
- }
- control_writeln("POLLDONE");
- close(fd);
- }
- static struct test_case test_cases[] = {
- {
- .name = "SOCK_STREAM connection reset",
- .run_client = test_stream_connection_reset,
- },
- {
- .name = "SOCK_STREAM bind only",
- .run_client = test_stream_bind_only_client,
- .run_server = test_stream_bind_only_server,
- },
- {
- .name = "SOCK_STREAM client close",
- .run_client = test_stream_client_close_client,
- .run_server = test_stream_client_close_server,
- },
- {
- .name = "SOCK_STREAM server close",
- .run_client = test_stream_server_close_client,
- .run_server = test_stream_server_close_server,
- },
- {
- .name = "SOCK_STREAM multiple connections",
- .run_client = test_stream_multiconn_client,
- .run_server = test_stream_multiconn_server,
- },
- {
- .name = "SOCK_STREAM MSG_PEEK",
- .run_client = test_stream_msg_peek_client,
- .run_server = test_stream_msg_peek_server,
- },
- {
- .name = "SOCK_SEQPACKET msg bounds",
- .run_client = test_seqpacket_msg_bounds_client,
- .run_server = test_seqpacket_msg_bounds_server,
- },
- {
- .name = "SOCK_SEQPACKET MSG_TRUNC flag",
- .run_client = test_seqpacket_msg_trunc_client,
- .run_server = test_seqpacket_msg_trunc_server,
- },
- {
- .name = "SOCK_SEQPACKET timeout",
- .run_client = test_seqpacket_timeout_client,
- .run_server = test_seqpacket_timeout_server,
- },
- {
- .name = "SOCK_SEQPACKET invalid receive buffer",
- .run_client = test_seqpacket_invalid_rec_buffer_client,
- .run_server = test_seqpacket_invalid_rec_buffer_server,
- },
- {
- .name = "SOCK_STREAM poll() + SO_RCVLOWAT",
- .run_client = test_stream_poll_rcvlowat_client,
- .run_server = test_stream_poll_rcvlowat_server,
- },
- {},
- };
- static const char optstring[] = "";
- static const struct option longopts[] = {
- {
- .name = "control-host",
- .has_arg = required_argument,
- .val = 'H',
- },
- {
- .name = "control-port",
- .has_arg = required_argument,
- .val = 'P',
- },
- {
- .name = "mode",
- .has_arg = required_argument,
- .val = 'm',
- },
- {
- .name = "peer-cid",
- .has_arg = required_argument,
- .val = 'p',
- },
- {
- .name = "list",
- .has_arg = no_argument,
- .val = 'l',
- },
- {
- .name = "skip",
- .has_arg = required_argument,
- .val = 's',
- },
- {
- .name = "help",
- .has_arg = no_argument,
- .val = '?',
- },
- {},
- };
- static void usage(void)
- {
- fprintf(stderr, "Usage: vsock_test [--help] [--control-host=<host>] --control-port=<port> --mode=client|server --peer-cid=<cid> [--list] [--skip=<test_id>]\n"
- "\n"
- " Server: vsock_test --control-port=1234 --mode=server --peer-cid=3\n"
- " Client: vsock_test --control-host=192.168.0.1 --control-port=1234 --mode=client --peer-cid=2\n"
- "\n"
- "Run vsock.ko tests. Must be launched in both guest\n"
- "and host. One side must use --mode=client and\n"
- "the other side must use --mode=server.\n"
- "\n"
- "A TCP control socket connection is used to coordinate tests\n"
- "between the client and the server. The server requires a\n"
- "listen address and the client requires an address to\n"
- "connect to.\n"
- "\n"
- "The CID of the other side must be given with --peer-cid=<cid>.\n"
- "\n"
- "Options:\n"
- " --help This help message\n"
- " --control-host <host> Server IP address to connect to\n"
- " --control-port <port> Server port to listen on/connect to\n"
- " --mode client|server Server or client mode\n"
- " --peer-cid <cid> CID of the other side\n"
- " --list List of tests that will be executed\n"
- " --skip <test_id> Test ID to skip;\n"
- " use multiple --skip options to skip more tests\n"
- );
- exit(EXIT_FAILURE);
- }
- int main(int argc, char **argv)
- {
- const char *control_host = NULL;
- const char *control_port = NULL;
- struct test_opts opts = {
- .mode = TEST_MODE_UNSET,
- .peer_cid = VMADDR_CID_ANY,
- };
- init_signals();
- for (;;) {
- int opt = getopt_long(argc, argv, optstring, longopts, NULL);
- if (opt == -1)
- break;
- switch (opt) {
- case 'H':
- control_host = optarg;
- break;
- case 'm':
- if (strcmp(optarg, "client") == 0)
- opts.mode = TEST_MODE_CLIENT;
- else if (strcmp(optarg, "server") == 0)
- opts.mode = TEST_MODE_SERVER;
- else {
- fprintf(stderr, "--mode must be \"client\" or \"server\"\n");
- return EXIT_FAILURE;
- }
- break;
- case 'p':
- opts.peer_cid = parse_cid(optarg);
- break;
- case 'P':
- control_port = optarg;
- break;
- case 'l':
- list_tests(test_cases);
- break;
- case 's':
- skip_test(test_cases, ARRAY_SIZE(test_cases) - 1,
- optarg);
- break;
- case '?':
- default:
- usage();
- }
- }
- if (!control_port)
- usage();
- if (opts.mode == TEST_MODE_UNSET)
- usage();
- if (opts.peer_cid == VMADDR_CID_ANY)
- usage();
- if (!control_host) {
- if (opts.mode != TEST_MODE_SERVER)
- usage();
- control_host = "0.0.0.0";
- }
- control_init(control_host, control_port,
- opts.mode == TEST_MODE_SERVER);
- run_tests(test_cases, &opts);
- control_cleanup();
- return EXIT_SUCCESS;
- }
|