1 // SPDX-License-Identifier: GPL-2.0-only
5 * Copyright (C) 2017 Red Hat, Inc.
7 * Author: Stefan Hajnoczi <stefanha@redhat.com>
17 #include <sys/epoll.h>
23 /* Install signal handlers */
24 void init_signals(void)
26 struct sigaction act = {
27 .sa_handler = sigalrm,
30 sigaction(SIGALRM, &act, NULL);
31 signal(SIGPIPE, SIG_IGN);
34 /* Parse a CID in string representation */
35 unsigned int parse_cid(const char *str)
41 n = strtoul(str, &endptr, 10);
42 if (errno || *endptr != '\0') {
43 fprintf(stderr, "malformed CID \"%s\"\n", str);
49 /* Wait for the remote to close the connection */
50 void vsock_wait_remote_close(int fd)
52 struct epoll_event ev;
55 epollfd = epoll_create1(0);
57 perror("epoll_create1");
61 ev.events = EPOLLRDHUP | EPOLLHUP;
63 if (epoll_ctl(epollfd, EPOLL_CTL_ADD, fd, &ev) == -1) {
68 nfds = epoll_wait(epollfd, &ev, 1, TIMEOUT * 1000);
75 fprintf(stderr, "epoll_wait timed out\n");
80 assert(ev.events & (EPOLLRDHUP | EPOLLHUP));
81 assert(ev.data.fd == fd);
86 /* Connect to <cid, port> and return the file descriptor. */
87 static int vsock_connect(unsigned int cid, unsigned int port, int type)
91 struct sockaddr_vm svm;
94 .svm_family = AF_VSOCK,
102 control_expectln("LISTENING");
104 fd = socket(AF_VSOCK, type, 0);
106 timeout_begin(TIMEOUT);
108 ret = connect(fd, &addr.sa, sizeof(addr.svm));
109 timeout_check("connect");
110 } while (ret < 0 && errno == EINTR);
114 int old_errno = errno;
123 int vsock_stream_connect(unsigned int cid, unsigned int port)
125 return vsock_connect(cid, port, SOCK_STREAM);
128 int vsock_seqpacket_connect(unsigned int cid, unsigned int port)
130 return vsock_connect(cid, port, SOCK_SEQPACKET);
133 /* Listen on <cid, port> and return the first incoming connection. The remote
134 * address is stored to clientaddrp. clientaddrp may be NULL.
136 static int vsock_accept(unsigned int cid, unsigned int port,
137 struct sockaddr_vm *clientaddrp, int type)
141 struct sockaddr_vm svm;
144 .svm_family = AF_VSOCK,
151 struct sockaddr_vm svm;
153 socklen_t clientaddr_len = sizeof(clientaddr.svm);
158 fd = socket(AF_VSOCK, type, 0);
160 if (bind(fd, &addr.sa, sizeof(addr.svm)) < 0) {
165 if (listen(fd, 1) < 0) {
170 control_writeln("LISTENING");
172 timeout_begin(TIMEOUT);
174 client_fd = accept(fd, &clientaddr.sa, &clientaddr_len);
175 timeout_check("accept");
176 } while (client_fd < 0 && errno == EINTR);
186 if (clientaddr_len != sizeof(clientaddr.svm)) {
187 fprintf(stderr, "unexpected addrlen from accept(2), %zu\n",
188 (size_t)clientaddr_len);
191 if (clientaddr.sa.sa_family != AF_VSOCK) {
192 fprintf(stderr, "expected AF_VSOCK from accept(2), got %d\n",
193 clientaddr.sa.sa_family);
198 *clientaddrp = clientaddr.svm;
202 int vsock_stream_accept(unsigned int cid, unsigned int port,
203 struct sockaddr_vm *clientaddrp)
205 return vsock_accept(cid, port, clientaddrp, SOCK_STREAM);
208 int vsock_seqpacket_accept(unsigned int cid, unsigned int port,
209 struct sockaddr_vm *clientaddrp)
211 return vsock_accept(cid, port, clientaddrp, SOCK_SEQPACKET);
214 /* Transmit one byte and check the return value.
217 * <0 Negative errno (for testing errors)
221 void send_byte(int fd, int expected_ret, int flags)
223 const uint8_t byte = 'A';
226 timeout_begin(TIMEOUT);
228 nwritten = send(fd, &byte, sizeof(byte), flags);
229 timeout_check("write");
230 } while (nwritten < 0 && errno == EINTR);
233 if (expected_ret < 0) {
234 if (nwritten != -1) {
235 fprintf(stderr, "bogus send(2) return value %zd\n",
239 if (errno != -expected_ret) {
251 if (expected_ret == 0)
254 fprintf(stderr, "unexpected EOF while sending byte\n");
257 if (nwritten != sizeof(byte)) {
258 fprintf(stderr, "bogus send(2) return value %zd\n", nwritten);
263 /* Receive one byte and check the return value.
266 * <0 Negative errno (for testing errors)
270 void recv_byte(int fd, int expected_ret, int flags)
275 timeout_begin(TIMEOUT);
277 nread = recv(fd, &byte, sizeof(byte), flags);
278 timeout_check("read");
279 } while (nread < 0 && errno == EINTR);
282 if (expected_ret < 0) {
284 fprintf(stderr, "bogus recv(2) return value %zd\n",
288 if (errno != -expected_ret) {
300 if (expected_ret == 0)
303 fprintf(stderr, "unexpected EOF while receiving byte\n");
306 if (nread != sizeof(byte)) {
307 fprintf(stderr, "bogus recv(2) return value %zd\n", nread);
311 fprintf(stderr, "unexpected byte read %c\n", byte);
316 /* Run test cases. The program terminates if a failure occurs. */
317 void run_tests(const struct test_case *test_cases,
318 const struct test_opts *opts)
322 for (i = 0; test_cases[i].name; i++) {
323 void (*run)(const struct test_opts *opts);
326 printf("%d - %s...", i, test_cases[i].name);
329 /* Full barrier before executing the next test. This
330 * ensures that client and server are executing the
331 * same test case. In particular, it means whoever is
332 * faster will not see the peer still executing the
333 * last test. This is important because port numbers
334 * can be used by multiple test cases.
336 if (test_cases[i].skip)
337 control_writeln("SKIP");
339 control_writeln("NEXT");
341 line = control_readln();
342 if (control_cmpln(line, "SKIP", false) || test_cases[i].skip) {
350 control_cmpln(line, "NEXT", true);
353 if (opts->mode == TEST_MODE_CLIENT)
354 run = test_cases[i].run_client;
356 run = test_cases[i].run_server;
365 void list_tests(const struct test_case *test_cases)
369 printf("ID\tTest name\n");
371 for (i = 0; test_cases[i].name; i++)
372 printf("%d\t%s\n", i, test_cases[i].name);
377 void skip_test(struct test_case *test_cases, size_t test_cases_len,
378 const char *test_id_str)
380 unsigned long test_id;
384 test_id = strtoul(test_id_str, &endptr, 10);
385 if (errno || *endptr != '\0') {
386 fprintf(stderr, "malformed test ID \"%s\"\n", test_id_str);
390 if (test_id >= test_cases_len) {
391 fprintf(stderr, "test ID (%lu) larger than the max allowed (%lu)\n",
392 test_id, test_cases_len - 1);
396 test_cases[test_id].skip = true;