1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3 * vsock test utilities
4 *
5 * Copyright (C) 2017 Red Hat, Inc.
6 *
7 * Author: Stefan Hajnoczi <stefanha@redhat.com>
8 */
9
10 #include <ctype.h>
11 #include <errno.h>
12 #include <stdio.h>
13 #include <stdint.h>
14 #include <stdlib.h>
15 #include <string.h>
16 #include <signal.h>
17 #include <unistd.h>
18 #include <assert.h>
19 #include <sys/epoll.h>
20 #include <sys/ioctl.h>
21 #include <sys/mman.h>
22 #include <linux/sockios.h>
23
24 #include "timeout.h"
25 #include "control.h"
26 #include "util.h"
27
28 #define KALLSYMS_PATH "/proc/kallsyms"
29 #define KALLSYMS_LINE_LEN 512
30
31 /* Install signal handlers */
init_signals(void)32 void init_signals(void)
33 {
34 struct sigaction act = {
35 .sa_handler = sigalrm,
36 };
37
38 sigaction(SIGALRM, &act, NULL);
39 signal(SIGPIPE, SIG_IGN);
40 }
41
parse_uint(const char * str,const char * err_str)42 static unsigned int parse_uint(const char *str, const char *err_str)
43 {
44 char *endptr = NULL;
45 unsigned long n;
46
47 errno = 0;
48 n = strtoul(str, &endptr, 10);
49 if (errno || *endptr != '\0') {
50 fprintf(stderr, "malformed %s \"%s\"\n", err_str, str);
51 exit(EXIT_FAILURE);
52 }
53 return n;
54 }
55
56 /* Parse a CID in string representation */
parse_cid(const char * str)57 unsigned int parse_cid(const char *str)
58 {
59 return parse_uint(str, "CID");
60 }
61
62 /* Parse a port in string representation */
parse_port(const char * str)63 unsigned int parse_port(const char *str)
64 {
65 return parse_uint(str, "port");
66 }
67
68 /* Wait for the remote to close the connection */
vsock_wait_remote_close(int fd)69 void vsock_wait_remote_close(int fd)
70 {
71 struct epoll_event ev;
72 int epollfd, nfds;
73
74 epollfd = epoll_create1(0);
75 if (epollfd == -1) {
76 perror("epoll_create1");
77 exit(EXIT_FAILURE);
78 }
79
80 ev.events = EPOLLRDHUP | EPOLLHUP;
81 ev.data.fd = fd;
82 if (epoll_ctl(epollfd, EPOLL_CTL_ADD, fd, &ev) == -1) {
83 perror("epoll_ctl");
84 exit(EXIT_FAILURE);
85 }
86
87 nfds = epoll_wait(epollfd, &ev, 1, TIMEOUT * 1000);
88 if (nfds == -1) {
89 perror("epoll_wait");
90 exit(EXIT_FAILURE);
91 }
92
93 if (nfds == 0) {
94 fprintf(stderr, "epoll_wait timed out\n");
95 exit(EXIT_FAILURE);
96 }
97
98 assert(nfds == 1);
99 assert(ev.events & (EPOLLRDHUP | EPOLLHUP));
100 assert(ev.data.fd == fd);
101
102 close(epollfd);
103 }
104
105 /* Wait until ioctl gives an expected int value.
106 * Return false if the op is not supported.
107 */
vsock_ioctl_int(int fd,unsigned long op,int expected)108 bool vsock_ioctl_int(int fd, unsigned long op, int expected)
109 {
110 int actual, ret;
111 char name[32];
112
113 snprintf(name, sizeof(name), "ioctl(%lu)", op);
114
115 timeout_begin(TIMEOUT);
116 do {
117 ret = ioctl(fd, op, &actual);
118 if (ret < 0) {
119 if (errno == EOPNOTSUPP || errno == ENOTTY)
120 break;
121
122 perror(name);
123 exit(EXIT_FAILURE);
124 }
125 timeout_check(name);
126 } while (actual != expected);
127 timeout_end();
128
129 return ret >= 0;
130 }
131
132 /* Wait until transport reports no data left to be sent.
133 * Return false if transport does not implement the unsent_bytes() callback.
134 */
vsock_wait_sent(int fd)135 bool vsock_wait_sent(int fd)
136 {
137 return vsock_ioctl_int(fd, SIOCOUTQ, 0);
138 }
139
140 /* Create socket <type>, bind to <cid, port>.
141 * Return the file descriptor, or -1 on error.
142 */
vsock_bind_try(unsigned int cid,unsigned int port,int type)143 int vsock_bind_try(unsigned int cid, unsigned int port, int type)
144 {
145 struct sockaddr_vm sa = {
146 .svm_family = AF_VSOCK,
147 .svm_cid = cid,
148 .svm_port = port,
149 };
150 int fd, saved_errno;
151
152 fd = socket(AF_VSOCK, type, 0);
153 if (fd < 0) {
154 perror("socket");
155 exit(EXIT_FAILURE);
156 }
157
158 if (bind(fd, (struct sockaddr *)&sa, sizeof(sa))) {
159 saved_errno = errno;
160 close(fd);
161 errno = saved_errno;
162 fd = -1;
163 }
164
165 return fd;
166 }
167
168 /* Create socket <type>, bind to <cid, port> and return the file descriptor. */
vsock_bind(unsigned int cid,unsigned int port,int type)169 int vsock_bind(unsigned int cid, unsigned int port, int type)
170 {
171 int fd;
172
173 fd = vsock_bind_try(cid, port, type);
174 if (fd < 0) {
175 perror("bind");
176 exit(EXIT_FAILURE);
177 }
178
179 return fd;
180 }
181
vsock_connect_fd(int fd,unsigned int cid,unsigned int port)182 int vsock_connect_fd(int fd, unsigned int cid, unsigned int port)
183 {
184 struct sockaddr_vm sa = {
185 .svm_family = AF_VSOCK,
186 .svm_cid = cid,
187 .svm_port = port,
188 };
189 int ret;
190
191 timeout_begin(TIMEOUT);
192 do {
193 ret = connect(fd, (struct sockaddr *)&sa, sizeof(sa));
194 timeout_check("connect");
195 } while (ret < 0 && errno == EINTR);
196 timeout_end();
197
198 return ret;
199 }
200
201 /* Bind to <bind_port>, connect to <cid, port> and return the file descriptor. */
vsock_bind_connect(unsigned int cid,unsigned int port,unsigned int bind_port,int type)202 int vsock_bind_connect(unsigned int cid, unsigned int port, unsigned int bind_port, int type)
203 {
204 int client_fd;
205
206 client_fd = vsock_bind(VMADDR_CID_ANY, bind_port, type);
207
208 if (vsock_connect_fd(client_fd, cid, port)) {
209 perror("connect");
210 exit(EXIT_FAILURE);
211 }
212
213 return client_fd;
214 }
215
216 /* Connect to <cid, port> and return the file descriptor. */
vsock_connect(unsigned int cid,unsigned int port,int type)217 int vsock_connect(unsigned int cid, unsigned int port, int type)
218 {
219 int fd;
220
221 control_expectln("LISTENING");
222
223 fd = socket(AF_VSOCK, type, 0);
224 if (fd < 0) {
225 perror("socket");
226 exit(EXIT_FAILURE);
227 }
228
229 if (vsock_connect_fd(fd, cid, port)) {
230 int old_errno = errno;
231
232 close(fd);
233 fd = -1;
234 errno = old_errno;
235 }
236
237 return fd;
238 }
239
vsock_stream_connect(unsigned int cid,unsigned int port)240 int vsock_stream_connect(unsigned int cid, unsigned int port)
241 {
242 return vsock_connect(cid, port, SOCK_STREAM);
243 }
244
vsock_seqpacket_connect(unsigned int cid,unsigned int port)245 int vsock_seqpacket_connect(unsigned int cid, unsigned int port)
246 {
247 return vsock_connect(cid, port, SOCK_SEQPACKET);
248 }
249
250 /* Listen on <cid, port> and return the file descriptor. */
vsock_listen(unsigned int cid,unsigned int port,int type)251 static int vsock_listen(unsigned int cid, unsigned int port, int type)
252 {
253 int fd;
254
255 fd = vsock_bind(cid, port, type);
256
257 if (listen(fd, 1) < 0) {
258 perror("listen");
259 exit(EXIT_FAILURE);
260 }
261
262 return fd;
263 }
264
265 /* Listen on <cid, port> and return the first incoming connection. The remote
266 * address is stored to clientaddrp. clientaddrp may be NULL.
267 */
vsock_accept(unsigned int cid,unsigned int port,struct sockaddr_vm * clientaddrp,int type)268 int vsock_accept(unsigned int cid, unsigned int port,
269 struct sockaddr_vm *clientaddrp, int type)
270 {
271 union {
272 struct sockaddr sa;
273 struct sockaddr_vm svm;
274 } clientaddr;
275 socklen_t clientaddr_len = sizeof(clientaddr.svm);
276 int fd, client_fd, old_errno;
277
278 fd = vsock_listen(cid, port, type);
279
280 control_writeln("LISTENING");
281
282 timeout_begin(TIMEOUT);
283 do {
284 client_fd = accept(fd, &clientaddr.sa, &clientaddr_len);
285 timeout_check("accept");
286 } while (client_fd < 0 && errno == EINTR);
287 timeout_end();
288
289 old_errno = errno;
290 close(fd);
291 errno = old_errno;
292
293 if (client_fd < 0)
294 return client_fd;
295
296 if (clientaddr_len != sizeof(clientaddr.svm)) {
297 fprintf(stderr, "unexpected addrlen from accept(2), %zu\n",
298 (size_t)clientaddr_len);
299 exit(EXIT_FAILURE);
300 }
301 if (clientaddr.sa.sa_family != AF_VSOCK) {
302 fprintf(stderr, "expected AF_VSOCK from accept(2), got %d\n",
303 clientaddr.sa.sa_family);
304 exit(EXIT_FAILURE);
305 }
306
307 if (clientaddrp)
308 *clientaddrp = clientaddr.svm;
309 return client_fd;
310 }
311
vsock_stream_accept(unsigned int cid,unsigned int port,struct sockaddr_vm * clientaddrp)312 int vsock_stream_accept(unsigned int cid, unsigned int port,
313 struct sockaddr_vm *clientaddrp)
314 {
315 return vsock_accept(cid, port, clientaddrp, SOCK_STREAM);
316 }
317
vsock_stream_listen(unsigned int cid,unsigned int port)318 int vsock_stream_listen(unsigned int cid, unsigned int port)
319 {
320 return vsock_listen(cid, port, SOCK_STREAM);
321 }
322
vsock_seqpacket_accept(unsigned int cid,unsigned int port,struct sockaddr_vm * clientaddrp)323 int vsock_seqpacket_accept(unsigned int cid, unsigned int port,
324 struct sockaddr_vm *clientaddrp)
325 {
326 return vsock_accept(cid, port, clientaddrp, SOCK_SEQPACKET);
327 }
328
329 /* Transmit bytes from a buffer and check the return value.
330 *
331 * expected_ret:
332 * <0 Negative errno (for testing errors)
333 * 0 End-of-file
334 * >0 Success (bytes successfully written)
335 */
send_buf(int fd,const void * buf,size_t len,int flags,ssize_t expected_ret)336 void send_buf(int fd, const void *buf, size_t len, int flags,
337 ssize_t expected_ret)
338 {
339 ssize_t nwritten = 0;
340 ssize_t ret;
341
342 timeout_begin(TIMEOUT);
343 do {
344 ret = send(fd, buf + nwritten, len - nwritten, flags);
345 timeout_check("send");
346
347 if (ret == 0 || (ret < 0 && errno != EINTR))
348 break;
349
350 nwritten += ret;
351 } while (nwritten < len);
352 timeout_end();
353
354 if (expected_ret < 0) {
355 if (ret != -1) {
356 fprintf(stderr, "bogus send(2) return value %zd (expected %zd)\n",
357 ret, expected_ret);
358 exit(EXIT_FAILURE);
359 }
360 if (errno != -expected_ret) {
361 perror("send");
362 exit(EXIT_FAILURE);
363 }
364 return;
365 }
366
367 if (ret < 0) {
368 perror("send");
369 exit(EXIT_FAILURE);
370 }
371
372 if (nwritten != expected_ret) {
373 if (ret == 0)
374 fprintf(stderr, "unexpected EOF while sending bytes\n");
375
376 fprintf(stderr, "bogus send(2) bytes written %zd (expected %zd)\n",
377 nwritten, expected_ret);
378 exit(EXIT_FAILURE);
379 }
380 }
381
382 /* Receive bytes in a buffer and check the return value.
383 *
384 * expected_ret:
385 * <0 Negative errno (for testing errors)
386 * 0 End-of-file
387 * >0 Success (bytes successfully read)
388 */
recv_buf(int fd,void * buf,size_t len,int flags,ssize_t expected_ret)389 void recv_buf(int fd, void *buf, size_t len, int flags, ssize_t expected_ret)
390 {
391 ssize_t nread = 0;
392 ssize_t ret;
393
394 timeout_begin(TIMEOUT);
395 do {
396 ret = recv(fd, buf + nread, len - nread, flags);
397 timeout_check("recv");
398
399 if (ret == 0 || (ret < 0 && errno != EINTR))
400 break;
401
402 nread += ret;
403 } while (nread < len);
404 timeout_end();
405
406 if (expected_ret < 0) {
407 if (ret != -1) {
408 fprintf(stderr, "bogus recv(2) return value %zd (expected %zd)\n",
409 ret, expected_ret);
410 exit(EXIT_FAILURE);
411 }
412 if (errno != -expected_ret) {
413 perror("recv");
414 exit(EXIT_FAILURE);
415 }
416 return;
417 }
418
419 if (ret < 0) {
420 perror("recv");
421 exit(EXIT_FAILURE);
422 }
423
424 if (nread != expected_ret) {
425 if (ret == 0)
426 fprintf(stderr, "unexpected EOF while receiving bytes\n");
427
428 fprintf(stderr, "bogus recv(2) bytes read %zd (expected %zd)\n",
429 nread, expected_ret);
430 exit(EXIT_FAILURE);
431 }
432 }
433
434 /* Transmit one byte and check the return value.
435 *
436 * expected_ret:
437 * <0 Negative errno (for testing errors)
438 * 0 End-of-file
439 * 1 Success
440 */
send_byte(int fd,int expected_ret,int flags)441 void send_byte(int fd, int expected_ret, int flags)
442 {
443 static const uint8_t byte = 'A';
444
445 send_buf(fd, &byte, sizeof(byte), flags, expected_ret);
446 }
447
448 /* Receive one byte and check the return value.
449 *
450 * expected_ret:
451 * <0 Negative errno (for testing errors)
452 * 0 End-of-file
453 * 1 Success
454 */
recv_byte(int fd,int expected_ret,int flags)455 void recv_byte(int fd, int expected_ret, int flags)
456 {
457 uint8_t byte;
458
459 recv_buf(fd, &byte, sizeof(byte), flags, expected_ret);
460
461 if (byte != 'A') {
462 fprintf(stderr, "unexpected byte read 0x%02x\n", byte);
463 exit(EXIT_FAILURE);
464 }
465 }
466
467 /* Run test cases. The program terminates if a failure occurs. */
run_tests(const struct test_case * test_cases,const struct test_opts * opts)468 void run_tests(const struct test_case *test_cases,
469 const struct test_opts *opts)
470 {
471 int i;
472
473 for (i = 0; test_cases[i].name; i++) {
474 void (*run)(const struct test_opts *opts);
475 char *line;
476
477 printf("%d - %s...", i, test_cases[i].name);
478 fflush(stdout);
479
480 /* Full barrier before executing the next test. This
481 * ensures that client and server are executing the
482 * same test case. In particular, it means whoever is
483 * faster will not see the peer still executing the
484 * last test. This is important because port numbers
485 * can be used by multiple test cases.
486 */
487 if (test_cases[i].skip)
488 control_writeln("SKIP");
489 else
490 control_writeln("NEXT");
491
492 line = control_readln();
493 if (control_cmpln(line, "SKIP", false) || test_cases[i].skip) {
494
495 printf("skipped\n");
496
497 free(line);
498 continue;
499 }
500
501 control_cmpln(line, "NEXT", true);
502 free(line);
503
504 if (opts->mode == TEST_MODE_CLIENT)
505 run = test_cases[i].run_client;
506 else
507 run = test_cases[i].run_server;
508
509 if (run)
510 run(opts);
511
512 printf("ok\n");
513 }
514
515 printf("All tests have been executed. Waiting other peer...");
516 fflush(stdout);
517
518 /*
519 * Final full barrier, to ensure that all tests have been run and
520 * that even the last one has been successful on both sides.
521 */
522 control_writeln("COMPLETED");
523 control_expectln("COMPLETED");
524
525 printf("ok\n");
526 }
527
list_tests(const struct test_case * test_cases)528 void list_tests(const struct test_case *test_cases)
529 {
530 int i;
531
532 printf("ID\tTest name\n");
533
534 for (i = 0; test_cases[i].name; i++)
535 printf("%d\t%s\n", i, test_cases[i].name);
536
537 exit(EXIT_FAILURE);
538 }
539
parse_test_id(const char * test_id_str,size_t test_cases_len)540 static unsigned long parse_test_id(const char *test_id_str, size_t test_cases_len)
541 {
542 unsigned long test_id;
543 char *endptr = NULL;
544
545 errno = 0;
546 test_id = strtoul(test_id_str, &endptr, 10);
547 if (errno || *endptr != '\0') {
548 fprintf(stderr, "malformed test ID \"%s\"\n", test_id_str);
549 exit(EXIT_FAILURE);
550 }
551
552 if (test_id >= test_cases_len) {
553 fprintf(stderr, "test ID (%lu) larger than the max allowed (%lu)\n",
554 test_id, test_cases_len - 1);
555 exit(EXIT_FAILURE);
556 }
557
558 return test_id;
559 }
560
skip_test(struct test_case * test_cases,size_t test_cases_len,const char * test_id_str)561 void skip_test(struct test_case *test_cases, size_t test_cases_len,
562 const char *test_id_str)
563 {
564 unsigned long test_id = parse_test_id(test_id_str, test_cases_len);
565 test_cases[test_id].skip = true;
566 }
567
pick_test(struct test_case * test_cases,size_t test_cases_len,const char * test_id_str)568 void pick_test(struct test_case *test_cases, size_t test_cases_len,
569 const char *test_id_str)
570 {
571 static bool skip_all = true;
572 unsigned long test_id;
573
574 if (skip_all) {
575 unsigned long i;
576
577 for (i = 0; i < test_cases_len; ++i)
578 test_cases[i].skip = true;
579
580 skip_all = false;
581 }
582
583 test_id = parse_test_id(test_id_str, test_cases_len);
584 test_cases[test_id].skip = false;
585 }
586
hash_djb2(const void * data,size_t len)587 unsigned long hash_djb2(const void *data, size_t len)
588 {
589 unsigned long hash = 5381;
590 int i = 0;
591
592 while (i < len) {
593 hash = ((hash << 5) + hash) + ((unsigned char *)data)[i];
594 i++;
595 }
596
597 return hash;
598 }
599
iovec_bytes(const struct iovec * iov,size_t iovnum)600 size_t iovec_bytes(const struct iovec *iov, size_t iovnum)
601 {
602 size_t bytes;
603 int i;
604
605 for (bytes = 0, i = 0; i < iovnum; i++)
606 bytes += iov[i].iov_len;
607
608 return bytes;
609 }
610
iovec_hash_djb2(const struct iovec * iov,size_t iovnum)611 unsigned long iovec_hash_djb2(const struct iovec *iov, size_t iovnum)
612 {
613 unsigned long hash;
614 size_t iov_bytes;
615 size_t offs;
616 void *tmp;
617 int i;
618
619 iov_bytes = iovec_bytes(iov, iovnum);
620
621 tmp = malloc(iov_bytes);
622 if (!tmp) {
623 perror("malloc");
624 exit(EXIT_FAILURE);
625 }
626
627 for (offs = 0, i = 0; i < iovnum; i++) {
628 memcpy(tmp + offs, iov[i].iov_base, iov[i].iov_len);
629 offs += iov[i].iov_len;
630 }
631
632 hash = hash_djb2(tmp, iov_bytes);
633 free(tmp);
634
635 return hash;
636 }
637
638 /* Allocates and returns new 'struct iovec *' according pattern
639 * in the 'test_iovec'. For each element in the 'test_iovec' it
640 * allocates new element in the resulting 'iovec'. 'iov_len'
641 * of the new element is copied from 'test_iovec'. 'iov_base' is
642 * allocated depending on the 'iov_base' of 'test_iovec':
643 *
644 * 'iov_base' == NULL -> valid buf: mmap('iov_len').
645 *
646 * 'iov_base' == MAP_FAILED -> invalid buf:
647 * mmap('iov_len'), then munmap('iov_len').
648 * 'iov_base' still contains result of
649 * mmap().
650 *
651 * 'iov_base' == number -> unaligned valid buf:
652 * mmap('iov_len') + number.
653 *
654 * 'iovnum' is number of elements in 'test_iovec'.
655 *
656 * Returns new 'iovec' or calls 'exit()' on error.
657 */
alloc_test_iovec(const struct iovec * test_iovec,int iovnum)658 struct iovec *alloc_test_iovec(const struct iovec *test_iovec, int iovnum)
659 {
660 struct iovec *iovec;
661 int i;
662
663 iovec = malloc(sizeof(*iovec) * iovnum);
664 if (!iovec) {
665 perror("malloc");
666 exit(EXIT_FAILURE);
667 }
668
669 for (i = 0; i < iovnum; i++) {
670 iovec[i].iov_len = test_iovec[i].iov_len;
671
672 iovec[i].iov_base = mmap(NULL, iovec[i].iov_len,
673 PROT_READ | PROT_WRITE,
674 MAP_PRIVATE | MAP_ANONYMOUS | MAP_POPULATE,
675 -1, 0);
676 if (iovec[i].iov_base == MAP_FAILED) {
677 perror("mmap");
678 exit(EXIT_FAILURE);
679 }
680
681 if (test_iovec[i].iov_base != MAP_FAILED)
682 iovec[i].iov_base += (uintptr_t)test_iovec[i].iov_base;
683 }
684
685 /* Unmap "invalid" elements. */
686 for (i = 0; i < iovnum; i++) {
687 if (test_iovec[i].iov_base == MAP_FAILED) {
688 if (munmap(iovec[i].iov_base, iovec[i].iov_len)) {
689 perror("munmap");
690 exit(EXIT_FAILURE);
691 }
692 }
693 }
694
695 for (i = 0; i < iovnum; i++) {
696 int j;
697
698 if (test_iovec[i].iov_base == MAP_FAILED)
699 continue;
700
701 for (j = 0; j < iovec[i].iov_len; j++)
702 ((uint8_t *)iovec[i].iov_base)[j] = rand() & 0xff;
703 }
704
705 return iovec;
706 }
707
708 /* Frees 'iovec *', previously allocated by 'alloc_test_iovec()'.
709 * On error calls 'exit()'.
710 */
free_test_iovec(const struct iovec * test_iovec,struct iovec * iovec,int iovnum)711 void free_test_iovec(const struct iovec *test_iovec,
712 struct iovec *iovec, int iovnum)
713 {
714 int i;
715
716 for (i = 0; i < iovnum; i++) {
717 if (test_iovec[i].iov_base != MAP_FAILED) {
718 if (test_iovec[i].iov_base)
719 iovec[i].iov_base -= (uintptr_t)test_iovec[i].iov_base;
720
721 if (munmap(iovec[i].iov_base, iovec[i].iov_len)) {
722 perror("munmap");
723 exit(EXIT_FAILURE);
724 }
725 }
726 }
727
728 free(iovec);
729 }
730
731 /* Set "unsigned long long" socket option and check that it's indeed set */
setsockopt_ull_check(int fd,int level,int optname,unsigned long long val,char const * errmsg)732 void setsockopt_ull_check(int fd, int level, int optname,
733 unsigned long long val, char const *errmsg)
734 {
735 unsigned long long chkval;
736 socklen_t chklen;
737 int err;
738
739 err = setsockopt(fd, level, optname, &val, sizeof(val));
740 if (err) {
741 fprintf(stderr, "setsockopt err: %s (%d)\n",
742 strerror(errno), errno);
743 goto fail;
744 }
745
746 chkval = ~val; /* just make storage != val */
747 chklen = sizeof(chkval);
748
749 err = getsockopt(fd, level, optname, &chkval, &chklen);
750 if (err) {
751 fprintf(stderr, "getsockopt err: %s (%d)\n",
752 strerror(errno), errno);
753 goto fail;
754 }
755
756 if (chklen != sizeof(chkval)) {
757 fprintf(stderr, "size mismatch: set %zu got %d\n", sizeof(val),
758 chklen);
759 goto fail;
760 }
761
762 if (chkval != val) {
763 fprintf(stderr, "value mismatch: set %llu got %llu\n", val,
764 chkval);
765 goto fail;
766 }
767 return;
768 fail:
769 fprintf(stderr, "%s val %llu\n", errmsg, val);
770 exit(EXIT_FAILURE);
771 }
772
773 /* Set "int" socket option and check that it's indeed set */
setsockopt_int_check(int fd,int level,int optname,int val,char const * errmsg)774 void setsockopt_int_check(int fd, int level, int optname, int val,
775 char const *errmsg)
776 {
777 int chkval;
778 socklen_t chklen;
779 int err;
780
781 err = setsockopt(fd, level, optname, &val, sizeof(val));
782 if (err) {
783 fprintf(stderr, "setsockopt err: %s (%d)\n",
784 strerror(errno), errno);
785 goto fail;
786 }
787
788 chkval = ~val; /* just make storage != val */
789 chklen = sizeof(chkval);
790
791 err = getsockopt(fd, level, optname, &chkval, &chklen);
792 if (err) {
793 fprintf(stderr, "getsockopt err: %s (%d)\n",
794 strerror(errno), errno);
795 goto fail;
796 }
797
798 if (chklen != sizeof(chkval)) {
799 fprintf(stderr, "size mismatch: set %zu got %d\n", sizeof(val),
800 chklen);
801 goto fail;
802 }
803
804 if (chkval != val) {
805 fprintf(stderr, "value mismatch: set %d got %d\n", val, chkval);
806 goto fail;
807 }
808 return;
809 fail:
810 fprintf(stderr, "%s val %d\n", errmsg, val);
811 exit(EXIT_FAILURE);
812 }
813
mem_invert(unsigned char * mem,size_t size)814 static void mem_invert(unsigned char *mem, size_t size)
815 {
816 size_t i;
817
818 for (i = 0; i < size; i++)
819 mem[i] = ~mem[i];
820 }
821
822 /* Set "timeval" socket option and check that it's indeed set */
setsockopt_timeval_check(int fd,int level,int optname,struct timeval val,char const * errmsg)823 void setsockopt_timeval_check(int fd, int level, int optname,
824 struct timeval val, char const *errmsg)
825 {
826 struct timeval chkval;
827 socklen_t chklen;
828 int err;
829
830 err = setsockopt(fd, level, optname, &val, sizeof(val));
831 if (err) {
832 fprintf(stderr, "setsockopt err: %s (%d)\n",
833 strerror(errno), errno);
834 goto fail;
835 }
836
837 /* just make storage != val */
838 chkval = val;
839 mem_invert((unsigned char *)&chkval, sizeof(chkval));
840 chklen = sizeof(chkval);
841
842 err = getsockopt(fd, level, optname, &chkval, &chklen);
843 if (err) {
844 fprintf(stderr, "getsockopt err: %s (%d)\n",
845 strerror(errno), errno);
846 goto fail;
847 }
848
849 if (chklen != sizeof(chkval)) {
850 fprintf(stderr, "size mismatch: set %zu got %d\n", sizeof(val),
851 chklen);
852 goto fail;
853 }
854
855 if (memcmp(&chkval, &val, sizeof(val)) != 0) {
856 fprintf(stderr, "value mismatch: set %ld:%ld got %ld:%ld\n",
857 val.tv_sec, val.tv_usec, chkval.tv_sec, chkval.tv_usec);
858 goto fail;
859 }
860 return;
861 fail:
862 fprintf(stderr, "%s val %ld:%ld\n", errmsg, val.tv_sec, val.tv_usec);
863 exit(EXIT_FAILURE);
864 }
865
enable_so_zerocopy_check(int fd)866 void enable_so_zerocopy_check(int fd)
867 {
868 setsockopt_int_check(fd, SOL_SOCKET, SO_ZEROCOPY, 1,
869 "setsockopt SO_ZEROCOPY");
870 }
871
enable_so_linger(int fd,int timeout)872 void enable_so_linger(int fd, int timeout)
873 {
874 struct linger optval = {
875 .l_onoff = 1,
876 .l_linger = timeout
877 };
878
879 if (setsockopt(fd, SOL_SOCKET, SO_LINGER, &optval, sizeof(optval))) {
880 perror("setsockopt(SO_LINGER)");
881 exit(EXIT_FAILURE);
882 }
883 }
884
__get_transports(void)885 static int __get_transports(void)
886 {
887 char buf[KALLSYMS_LINE_LEN];
888 const char *ksym;
889 int ret = 0;
890 FILE *f;
891
892 f = fopen(KALLSYMS_PATH, "r");
893 if (!f) {
894 perror("Can't open " KALLSYMS_PATH);
895 exit(EXIT_FAILURE);
896 }
897
898 while (fgets(buf, sizeof(buf), f)) {
899 char *match;
900 int i;
901
902 assert(buf[strlen(buf) - 1] == '\n');
903
904 for (i = 0; i < TRANSPORT_NUM; ++i) {
905 if (ret & BIT(i))
906 continue;
907
908 /* Match should be followed by '\t' or '\n'.
909 * See kallsyms.c:s_show().
910 */
911 ksym = transport_ksyms[i];
912 match = strstr(buf, ksym);
913 if (match && isspace(match[strlen(ksym)])) {
914 ret |= BIT(i);
915 break;
916 }
917 }
918 }
919
920 fclose(f);
921 return ret;
922 }
923
924 /* Return integer with TRANSPORT_* bit set for every (known) registered vsock
925 * transport.
926 */
get_transports(void)927 int get_transports(void)
928 {
929 static int tr = -1;
930
931 if (tr == -1)
932 tr = __get_transports();
933
934 return tr;
935 }
936