1b886d83cSThomas Gleixner // SPDX-License-Identifier: GPL-2.0-only 20b025033SStefan Hajnoczi /* 30b025033SStefan Hajnoczi * vsock_diag_test - vsock_diag.ko test suite 40b025033SStefan Hajnoczi * 50b025033SStefan Hajnoczi * Copyright (C) 2017 Red Hat, Inc. 60b025033SStefan Hajnoczi * 70b025033SStefan Hajnoczi * Author: Stefan Hajnoczi <stefanha@redhat.com> 80b025033SStefan Hajnoczi */ 90b025033SStefan Hajnoczi 100b025033SStefan Hajnoczi #include <getopt.h> 110b025033SStefan Hajnoczi #include <stdio.h> 120b025033SStefan Hajnoczi #include <stdbool.h> 130b025033SStefan Hajnoczi #include <stdlib.h> 140b025033SStefan Hajnoczi #include <string.h> 150b025033SStefan Hajnoczi #include <errno.h> 160b025033SStefan Hajnoczi #include <unistd.h> 170b025033SStefan Hajnoczi #include <signal.h> 180b025033SStefan Hajnoczi #include <sys/socket.h> 190b025033SStefan Hajnoczi #include <sys/stat.h> 200b025033SStefan Hajnoczi #include <sys/types.h> 210b025033SStefan Hajnoczi #include <linux/list.h> 220b025033SStefan Hajnoczi #include <linux/net.h> 230b025033SStefan Hajnoczi #include <linux/netlink.h> 2443985468SStefan Hajnoczi #include <linux/vm_sockets.h> 250b025033SStefan Hajnoczi #include <linux/sock_diag.h> 2643985468SStefan Hajnoczi #include <linux/vm_sockets_diag.h> 270b025033SStefan Hajnoczi #include <netinet/tcp.h> 280b025033SStefan Hajnoczi 290b025033SStefan Hajnoczi #include "timeout.h" 300b025033SStefan Hajnoczi #include "control.h" 310b025033SStefan Hajnoczi 320b025033SStefan Hajnoczi enum test_mode { 330b025033SStefan Hajnoczi TEST_MODE_UNSET, 340b025033SStefan Hajnoczi TEST_MODE_CLIENT, 350b025033SStefan Hajnoczi TEST_MODE_SERVER 360b025033SStefan Hajnoczi }; 370b025033SStefan Hajnoczi 380b025033SStefan Hajnoczi /* Per-socket status */ 390b025033SStefan Hajnoczi struct vsock_stat { 400b025033SStefan Hajnoczi struct list_head list; 410b025033SStefan Hajnoczi struct vsock_diag_msg msg; 420b025033SStefan Hajnoczi }; 430b025033SStefan Hajnoczi 440b025033SStefan Hajnoczi static const char *sock_type_str(int type) 450b025033SStefan Hajnoczi { 460b025033SStefan Hajnoczi switch (type) { 470b025033SStefan Hajnoczi case SOCK_DGRAM: 480b025033SStefan Hajnoczi return "DGRAM"; 490b025033SStefan Hajnoczi case SOCK_STREAM: 500b025033SStefan Hajnoczi return "STREAM"; 510b025033SStefan Hajnoczi default: 520b025033SStefan Hajnoczi return "INVALID TYPE"; 530b025033SStefan Hajnoczi } 540b025033SStefan Hajnoczi } 550b025033SStefan Hajnoczi 560b025033SStefan Hajnoczi static const char *sock_state_str(int state) 570b025033SStefan Hajnoczi { 580b025033SStefan Hajnoczi switch (state) { 590b025033SStefan Hajnoczi case TCP_CLOSE: 600b025033SStefan Hajnoczi return "UNCONNECTED"; 610b025033SStefan Hajnoczi case TCP_SYN_SENT: 620b025033SStefan Hajnoczi return "CONNECTING"; 630b025033SStefan Hajnoczi case TCP_ESTABLISHED: 640b025033SStefan Hajnoczi return "CONNECTED"; 650b025033SStefan Hajnoczi case TCP_CLOSING: 660b025033SStefan Hajnoczi return "DISCONNECTING"; 670b025033SStefan Hajnoczi case TCP_LISTEN: 680b025033SStefan Hajnoczi return "LISTEN"; 690b025033SStefan Hajnoczi default: 700b025033SStefan Hajnoczi return "INVALID STATE"; 710b025033SStefan Hajnoczi } 720b025033SStefan Hajnoczi } 730b025033SStefan Hajnoczi 740b025033SStefan Hajnoczi static const char *sock_shutdown_str(int shutdown) 750b025033SStefan Hajnoczi { 760b025033SStefan Hajnoczi switch (shutdown) { 770b025033SStefan Hajnoczi case 1: 780b025033SStefan Hajnoczi return "RCV_SHUTDOWN"; 790b025033SStefan Hajnoczi case 2: 800b025033SStefan Hajnoczi return "SEND_SHUTDOWN"; 810b025033SStefan Hajnoczi case 3: 820b025033SStefan Hajnoczi return "RCV_SHUTDOWN | SEND_SHUTDOWN"; 830b025033SStefan Hajnoczi default: 840b025033SStefan Hajnoczi return "0"; 850b025033SStefan Hajnoczi } 860b025033SStefan Hajnoczi } 870b025033SStefan Hajnoczi 880b025033SStefan Hajnoczi static void print_vsock_addr(FILE *fp, unsigned int cid, unsigned int port) 890b025033SStefan Hajnoczi { 900b025033SStefan Hajnoczi if (cid == VMADDR_CID_ANY) 910b025033SStefan Hajnoczi fprintf(fp, "*:"); 920b025033SStefan Hajnoczi else 930b025033SStefan Hajnoczi fprintf(fp, "%u:", cid); 940b025033SStefan Hajnoczi 950b025033SStefan Hajnoczi if (port == VMADDR_PORT_ANY) 960b025033SStefan Hajnoczi fprintf(fp, "*"); 970b025033SStefan Hajnoczi else 980b025033SStefan Hajnoczi fprintf(fp, "%u", port); 990b025033SStefan Hajnoczi } 1000b025033SStefan Hajnoczi 1010b025033SStefan Hajnoczi static void print_vsock_stat(FILE *fp, struct vsock_stat *st) 1020b025033SStefan Hajnoczi { 1030b025033SStefan Hajnoczi print_vsock_addr(fp, st->msg.vdiag_src_cid, st->msg.vdiag_src_port); 1040b025033SStefan Hajnoczi fprintf(fp, " "); 1050b025033SStefan Hajnoczi print_vsock_addr(fp, st->msg.vdiag_dst_cid, st->msg.vdiag_dst_port); 1060b025033SStefan Hajnoczi fprintf(fp, " %s %s %s %u\n", 1070b025033SStefan Hajnoczi sock_type_str(st->msg.vdiag_type), 1080b025033SStefan Hajnoczi sock_state_str(st->msg.vdiag_state), 1090b025033SStefan Hajnoczi sock_shutdown_str(st->msg.vdiag_shutdown), 1100b025033SStefan Hajnoczi st->msg.vdiag_ino); 1110b025033SStefan Hajnoczi } 1120b025033SStefan Hajnoczi 1130b025033SStefan Hajnoczi static void print_vsock_stats(FILE *fp, struct list_head *head) 1140b025033SStefan Hajnoczi { 1150b025033SStefan Hajnoczi struct vsock_stat *st; 1160b025033SStefan Hajnoczi 1170b025033SStefan Hajnoczi list_for_each_entry(st, head, list) 1180b025033SStefan Hajnoczi print_vsock_stat(fp, st); 1190b025033SStefan Hajnoczi } 1200b025033SStefan Hajnoczi 1210b025033SStefan Hajnoczi static struct vsock_stat *find_vsock_stat(struct list_head *head, int fd) 1220b025033SStefan Hajnoczi { 1230b025033SStefan Hajnoczi struct vsock_stat *st; 1240b025033SStefan Hajnoczi struct stat stat; 1250b025033SStefan Hajnoczi 1260b025033SStefan Hajnoczi if (fstat(fd, &stat) < 0) { 1270b025033SStefan Hajnoczi perror("fstat"); 1280b025033SStefan Hajnoczi exit(EXIT_FAILURE); 1290b025033SStefan Hajnoczi } 1300b025033SStefan Hajnoczi 1310b025033SStefan Hajnoczi list_for_each_entry(st, head, list) 1320b025033SStefan Hajnoczi if (st->msg.vdiag_ino == stat.st_ino) 1330b025033SStefan Hajnoczi return st; 1340b025033SStefan Hajnoczi 1350b025033SStefan Hajnoczi fprintf(stderr, "cannot find fd %d\n", fd); 1360b025033SStefan Hajnoczi exit(EXIT_FAILURE); 1370b025033SStefan Hajnoczi } 1380b025033SStefan Hajnoczi 1390b025033SStefan Hajnoczi static void check_no_sockets(struct list_head *head) 1400b025033SStefan Hajnoczi { 1410b025033SStefan Hajnoczi if (!list_empty(head)) { 1420b025033SStefan Hajnoczi fprintf(stderr, "expected no sockets\n"); 1430b025033SStefan Hajnoczi print_vsock_stats(stderr, head); 1440b025033SStefan Hajnoczi exit(1); 1450b025033SStefan Hajnoczi } 1460b025033SStefan Hajnoczi } 1470b025033SStefan Hajnoczi 1480b025033SStefan Hajnoczi static void check_num_sockets(struct list_head *head, int expected) 1490b025033SStefan Hajnoczi { 1500b025033SStefan Hajnoczi struct list_head *node; 1510b025033SStefan Hajnoczi int n = 0; 1520b025033SStefan Hajnoczi 1530b025033SStefan Hajnoczi list_for_each(node, head) 1540b025033SStefan Hajnoczi n++; 1550b025033SStefan Hajnoczi 1560b025033SStefan Hajnoczi if (n != expected) { 1570b025033SStefan Hajnoczi fprintf(stderr, "expected %d sockets, found %d\n", 1580b025033SStefan Hajnoczi expected, n); 1590b025033SStefan Hajnoczi print_vsock_stats(stderr, head); 1600b025033SStefan Hajnoczi exit(EXIT_FAILURE); 1610b025033SStefan Hajnoczi } 1620b025033SStefan Hajnoczi } 1630b025033SStefan Hajnoczi 1640b025033SStefan Hajnoczi static void check_socket_state(struct vsock_stat *st, __u8 state) 1650b025033SStefan Hajnoczi { 1660b025033SStefan Hajnoczi if (st->msg.vdiag_state != state) { 1670b025033SStefan Hajnoczi fprintf(stderr, "expected socket state %#x, got %#x\n", 1680b025033SStefan Hajnoczi state, st->msg.vdiag_state); 1690b025033SStefan Hajnoczi exit(EXIT_FAILURE); 1700b025033SStefan Hajnoczi } 1710b025033SStefan Hajnoczi } 1720b025033SStefan Hajnoczi 1730b025033SStefan Hajnoczi static void send_req(int fd) 1740b025033SStefan Hajnoczi { 1750b025033SStefan Hajnoczi struct sockaddr_nl nladdr = { 1760b025033SStefan Hajnoczi .nl_family = AF_NETLINK, 1770b025033SStefan Hajnoczi }; 1780b025033SStefan Hajnoczi struct { 1790b025033SStefan Hajnoczi struct nlmsghdr nlh; 1800b025033SStefan Hajnoczi struct vsock_diag_req vreq; 1810b025033SStefan Hajnoczi } req = { 1820b025033SStefan Hajnoczi .nlh = { 1830b025033SStefan Hajnoczi .nlmsg_len = sizeof(req), 1840b025033SStefan Hajnoczi .nlmsg_type = SOCK_DIAG_BY_FAMILY, 1850b025033SStefan Hajnoczi .nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP, 1860b025033SStefan Hajnoczi }, 1870b025033SStefan Hajnoczi .vreq = { 1880b025033SStefan Hajnoczi .sdiag_family = AF_VSOCK, 1890b025033SStefan Hajnoczi .vdiag_states = ~(__u32)0, 1900b025033SStefan Hajnoczi }, 1910b025033SStefan Hajnoczi }; 1920b025033SStefan Hajnoczi struct iovec iov = { 1930b025033SStefan Hajnoczi .iov_base = &req, 1940b025033SStefan Hajnoczi .iov_len = sizeof(req), 1950b025033SStefan Hajnoczi }; 1960b025033SStefan Hajnoczi struct msghdr msg = { 1970b025033SStefan Hajnoczi .msg_name = &nladdr, 1980b025033SStefan Hajnoczi .msg_namelen = sizeof(nladdr), 1990b025033SStefan Hajnoczi .msg_iov = &iov, 2000b025033SStefan Hajnoczi .msg_iovlen = 1, 2010b025033SStefan Hajnoczi }; 2020b025033SStefan Hajnoczi 2030b025033SStefan Hajnoczi for (;;) { 2040b025033SStefan Hajnoczi if (sendmsg(fd, &msg, 0) < 0) { 2050b025033SStefan Hajnoczi if (errno == EINTR) 2060b025033SStefan Hajnoczi continue; 2070b025033SStefan Hajnoczi 2080b025033SStefan Hajnoczi perror("sendmsg"); 2090b025033SStefan Hajnoczi exit(EXIT_FAILURE); 2100b025033SStefan Hajnoczi } 2110b025033SStefan Hajnoczi 2120b025033SStefan Hajnoczi return; 2130b025033SStefan Hajnoczi } 2140b025033SStefan Hajnoczi } 2150b025033SStefan Hajnoczi 2160b025033SStefan Hajnoczi static ssize_t recv_resp(int fd, void *buf, size_t len) 2170b025033SStefan Hajnoczi { 2180b025033SStefan Hajnoczi struct sockaddr_nl nladdr = { 2190b025033SStefan Hajnoczi .nl_family = AF_NETLINK, 2200b025033SStefan Hajnoczi }; 2210b025033SStefan Hajnoczi struct iovec iov = { 2220b025033SStefan Hajnoczi .iov_base = buf, 2230b025033SStefan Hajnoczi .iov_len = len, 2240b025033SStefan Hajnoczi }; 2250b025033SStefan Hajnoczi struct msghdr msg = { 2260b025033SStefan Hajnoczi .msg_name = &nladdr, 2270b025033SStefan Hajnoczi .msg_namelen = sizeof(nladdr), 2280b025033SStefan Hajnoczi .msg_iov = &iov, 2290b025033SStefan Hajnoczi .msg_iovlen = 1, 2300b025033SStefan Hajnoczi }; 2310b025033SStefan Hajnoczi ssize_t ret; 2320b025033SStefan Hajnoczi 2330b025033SStefan Hajnoczi do { 2340b025033SStefan Hajnoczi ret = recvmsg(fd, &msg, 0); 2350b025033SStefan Hajnoczi } while (ret < 0 && errno == EINTR); 2360b025033SStefan Hajnoczi 2370b025033SStefan Hajnoczi if (ret < 0) { 2380b025033SStefan Hajnoczi perror("recvmsg"); 2390b025033SStefan Hajnoczi exit(EXIT_FAILURE); 2400b025033SStefan Hajnoczi } 2410b025033SStefan Hajnoczi 2420b025033SStefan Hajnoczi return ret; 2430b025033SStefan Hajnoczi } 2440b025033SStefan Hajnoczi 2450b025033SStefan Hajnoczi static void add_vsock_stat(struct list_head *sockets, 2460b025033SStefan Hajnoczi const struct vsock_diag_msg *resp) 2470b025033SStefan Hajnoczi { 2480b025033SStefan Hajnoczi struct vsock_stat *st; 2490b025033SStefan Hajnoczi 2500b025033SStefan Hajnoczi st = malloc(sizeof(*st)); 2510b025033SStefan Hajnoczi if (!st) { 2520b025033SStefan Hajnoczi perror("malloc"); 2530b025033SStefan Hajnoczi exit(EXIT_FAILURE); 2540b025033SStefan Hajnoczi } 2550b025033SStefan Hajnoczi 2560b025033SStefan Hajnoczi st->msg = *resp; 2570b025033SStefan Hajnoczi list_add_tail(&st->list, sockets); 2580b025033SStefan Hajnoczi } 2590b025033SStefan Hajnoczi 2600b025033SStefan Hajnoczi /* 2610b025033SStefan Hajnoczi * Read vsock stats into a list. 2620b025033SStefan Hajnoczi */ 2630b025033SStefan Hajnoczi static void read_vsock_stat(struct list_head *sockets) 2640b025033SStefan Hajnoczi { 2650b025033SStefan Hajnoczi long buf[8192 / sizeof(long)]; 2660b025033SStefan Hajnoczi int fd; 2670b025033SStefan Hajnoczi 2680b025033SStefan Hajnoczi fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_SOCK_DIAG); 2690b025033SStefan Hajnoczi if (fd < 0) { 2700b025033SStefan Hajnoczi perror("socket"); 2710b025033SStefan Hajnoczi exit(EXIT_FAILURE); 2720b025033SStefan Hajnoczi } 2730b025033SStefan Hajnoczi 2740b025033SStefan Hajnoczi send_req(fd); 2750b025033SStefan Hajnoczi 2760b025033SStefan Hajnoczi for (;;) { 2770b025033SStefan Hajnoczi const struct nlmsghdr *h; 2780b025033SStefan Hajnoczi ssize_t ret; 2790b025033SStefan Hajnoczi 2800b025033SStefan Hajnoczi ret = recv_resp(fd, buf, sizeof(buf)); 2810b025033SStefan Hajnoczi if (ret == 0) 2820b025033SStefan Hajnoczi goto done; 2830b025033SStefan Hajnoczi if (ret < sizeof(*h)) { 2840b025033SStefan Hajnoczi fprintf(stderr, "short read of %zd bytes\n", ret); 2850b025033SStefan Hajnoczi exit(EXIT_FAILURE); 2860b025033SStefan Hajnoczi } 2870b025033SStefan Hajnoczi 2880b025033SStefan Hajnoczi h = (struct nlmsghdr *)buf; 2890b025033SStefan Hajnoczi 2900b025033SStefan Hajnoczi while (NLMSG_OK(h, ret)) { 2910b025033SStefan Hajnoczi if (h->nlmsg_type == NLMSG_DONE) 2920b025033SStefan Hajnoczi goto done; 2930b025033SStefan Hajnoczi 2940b025033SStefan Hajnoczi if (h->nlmsg_type == NLMSG_ERROR) { 2950b025033SStefan Hajnoczi const struct nlmsgerr *err = NLMSG_DATA(h); 2960b025033SStefan Hajnoczi 2970b025033SStefan Hajnoczi if (h->nlmsg_len < NLMSG_LENGTH(sizeof(*err))) 2980b025033SStefan Hajnoczi fprintf(stderr, "NLMSG_ERROR\n"); 2990b025033SStefan Hajnoczi else { 3000b025033SStefan Hajnoczi errno = -err->error; 3010b025033SStefan Hajnoczi perror("NLMSG_ERROR"); 3020b025033SStefan Hajnoczi } 3030b025033SStefan Hajnoczi 3040b025033SStefan Hajnoczi exit(EXIT_FAILURE); 3050b025033SStefan Hajnoczi } 3060b025033SStefan Hajnoczi 3070b025033SStefan Hajnoczi if (h->nlmsg_type != SOCK_DIAG_BY_FAMILY) { 3080b025033SStefan Hajnoczi fprintf(stderr, "unexpected nlmsg_type %#x\n", 3090b025033SStefan Hajnoczi h->nlmsg_type); 3100b025033SStefan Hajnoczi exit(EXIT_FAILURE); 3110b025033SStefan Hajnoczi } 3120b025033SStefan Hajnoczi if (h->nlmsg_len < 3130b025033SStefan Hajnoczi NLMSG_LENGTH(sizeof(struct vsock_diag_msg))) { 3140b025033SStefan Hajnoczi fprintf(stderr, "short vsock_diag_msg\n"); 3150b025033SStefan Hajnoczi exit(EXIT_FAILURE); 3160b025033SStefan Hajnoczi } 3170b025033SStefan Hajnoczi 3180b025033SStefan Hajnoczi add_vsock_stat(sockets, NLMSG_DATA(h)); 3190b025033SStefan Hajnoczi 3200b025033SStefan Hajnoczi h = NLMSG_NEXT(h, ret); 3210b025033SStefan Hajnoczi } 3220b025033SStefan Hajnoczi } 3230b025033SStefan Hajnoczi 3240b025033SStefan Hajnoczi done: 3250b025033SStefan Hajnoczi close(fd); 3260b025033SStefan Hajnoczi } 3270b025033SStefan Hajnoczi 3280b025033SStefan Hajnoczi static void free_sock_stat(struct list_head *sockets) 3290b025033SStefan Hajnoczi { 3300b025033SStefan Hajnoczi struct vsock_stat *st; 3310b025033SStefan Hajnoczi struct vsock_stat *next; 3320b025033SStefan Hajnoczi 3330b025033SStefan Hajnoczi list_for_each_entry_safe(st, next, sockets, list) 3340b025033SStefan Hajnoczi free(st); 3350b025033SStefan Hajnoczi } 3360b025033SStefan Hajnoczi 3370b025033SStefan Hajnoczi static void test_no_sockets(unsigned int peer_cid) 3380b025033SStefan Hajnoczi { 3390b025033SStefan Hajnoczi LIST_HEAD(sockets); 3400b025033SStefan Hajnoczi 3410b025033SStefan Hajnoczi read_vsock_stat(&sockets); 3420b025033SStefan Hajnoczi 3430b025033SStefan Hajnoczi check_no_sockets(&sockets); 3440b025033SStefan Hajnoczi 3450b025033SStefan Hajnoczi free_sock_stat(&sockets); 3460b025033SStefan Hajnoczi } 3470b025033SStefan Hajnoczi 3480b025033SStefan Hajnoczi static void test_listen_socket_server(unsigned int peer_cid) 3490b025033SStefan Hajnoczi { 3500b025033SStefan Hajnoczi union { 3510b025033SStefan Hajnoczi struct sockaddr sa; 3520b025033SStefan Hajnoczi struct sockaddr_vm svm; 3530b025033SStefan Hajnoczi } addr = { 3540b025033SStefan Hajnoczi .svm = { 3550b025033SStefan Hajnoczi .svm_family = AF_VSOCK, 3560b025033SStefan Hajnoczi .svm_port = 1234, 3570b025033SStefan Hajnoczi .svm_cid = VMADDR_CID_ANY, 3580b025033SStefan Hajnoczi }, 3590b025033SStefan Hajnoczi }; 3600b025033SStefan Hajnoczi LIST_HEAD(sockets); 3610b025033SStefan Hajnoczi struct vsock_stat *st; 3620b025033SStefan Hajnoczi int fd; 3630b025033SStefan Hajnoczi 3640b025033SStefan Hajnoczi fd = socket(AF_VSOCK, SOCK_STREAM, 0); 3650b025033SStefan Hajnoczi 3660b025033SStefan Hajnoczi if (bind(fd, &addr.sa, sizeof(addr.svm)) < 0) { 3670b025033SStefan Hajnoczi perror("bind"); 3680b025033SStefan Hajnoczi exit(EXIT_FAILURE); 3690b025033SStefan Hajnoczi } 3700b025033SStefan Hajnoczi 3710b025033SStefan Hajnoczi if (listen(fd, 1) < 0) { 3720b025033SStefan Hajnoczi perror("listen"); 3730b025033SStefan Hajnoczi exit(EXIT_FAILURE); 3740b025033SStefan Hajnoczi } 3750b025033SStefan Hajnoczi 3760b025033SStefan Hajnoczi read_vsock_stat(&sockets); 3770b025033SStefan Hajnoczi 3780b025033SStefan Hajnoczi check_num_sockets(&sockets, 1); 3790b025033SStefan Hajnoczi st = find_vsock_stat(&sockets, fd); 3800b025033SStefan Hajnoczi check_socket_state(st, TCP_LISTEN); 3810b025033SStefan Hajnoczi 3820b025033SStefan Hajnoczi close(fd); 3830b025033SStefan Hajnoczi free_sock_stat(&sockets); 3840b025033SStefan Hajnoczi } 3850b025033SStefan Hajnoczi 3860b025033SStefan Hajnoczi static void test_connect_client(unsigned int peer_cid) 3870b025033SStefan Hajnoczi { 3880b025033SStefan Hajnoczi union { 3890b025033SStefan Hajnoczi struct sockaddr sa; 3900b025033SStefan Hajnoczi struct sockaddr_vm svm; 3910b025033SStefan Hajnoczi } addr = { 3920b025033SStefan Hajnoczi .svm = { 3930b025033SStefan Hajnoczi .svm_family = AF_VSOCK, 3940b025033SStefan Hajnoczi .svm_port = 1234, 3950b025033SStefan Hajnoczi .svm_cid = peer_cid, 3960b025033SStefan Hajnoczi }, 3970b025033SStefan Hajnoczi }; 3980b025033SStefan Hajnoczi int fd; 3990b025033SStefan Hajnoczi int ret; 4000b025033SStefan Hajnoczi LIST_HEAD(sockets); 4010b025033SStefan Hajnoczi struct vsock_stat *st; 4020b025033SStefan Hajnoczi 4030b025033SStefan Hajnoczi control_expectln("LISTENING"); 4040b025033SStefan Hajnoczi 4050b025033SStefan Hajnoczi fd = socket(AF_VSOCK, SOCK_STREAM, 0); 4060b025033SStefan Hajnoczi 4070b025033SStefan Hajnoczi timeout_begin(TIMEOUT); 4080b025033SStefan Hajnoczi do { 4090b025033SStefan Hajnoczi ret = connect(fd, &addr.sa, sizeof(addr.svm)); 4100b025033SStefan Hajnoczi timeout_check("connect"); 4110b025033SStefan Hajnoczi } while (ret < 0 && errno == EINTR); 4120b025033SStefan Hajnoczi timeout_end(); 4130b025033SStefan Hajnoczi 4140b025033SStefan Hajnoczi if (ret < 0) { 4150b025033SStefan Hajnoczi perror("connect"); 4160b025033SStefan Hajnoczi exit(EXIT_FAILURE); 4170b025033SStefan Hajnoczi } 4180b025033SStefan Hajnoczi 4190b025033SStefan Hajnoczi read_vsock_stat(&sockets); 4200b025033SStefan Hajnoczi 4210b025033SStefan Hajnoczi check_num_sockets(&sockets, 1); 4220b025033SStefan Hajnoczi st = find_vsock_stat(&sockets, fd); 4230b025033SStefan Hajnoczi check_socket_state(st, TCP_ESTABLISHED); 4240b025033SStefan Hajnoczi 4250b025033SStefan Hajnoczi control_expectln("DONE"); 4260b025033SStefan Hajnoczi control_writeln("DONE"); 4270b025033SStefan Hajnoczi 4280b025033SStefan Hajnoczi close(fd); 4290b025033SStefan Hajnoczi free_sock_stat(&sockets); 4300b025033SStefan Hajnoczi } 4310b025033SStefan Hajnoczi 4320b025033SStefan Hajnoczi static void test_connect_server(unsigned int peer_cid) 4330b025033SStefan Hajnoczi { 4340b025033SStefan Hajnoczi union { 4350b025033SStefan Hajnoczi struct sockaddr sa; 4360b025033SStefan Hajnoczi struct sockaddr_vm svm; 4370b025033SStefan Hajnoczi } addr = { 4380b025033SStefan Hajnoczi .svm = { 4390b025033SStefan Hajnoczi .svm_family = AF_VSOCK, 4400b025033SStefan Hajnoczi .svm_port = 1234, 4410b025033SStefan Hajnoczi .svm_cid = VMADDR_CID_ANY, 4420b025033SStefan Hajnoczi }, 4430b025033SStefan Hajnoczi }; 4440b025033SStefan Hajnoczi union { 4450b025033SStefan Hajnoczi struct sockaddr sa; 4460b025033SStefan Hajnoczi struct sockaddr_vm svm; 4470b025033SStefan Hajnoczi } clientaddr; 4480b025033SStefan Hajnoczi socklen_t clientaddr_len = sizeof(clientaddr.svm); 4490b025033SStefan Hajnoczi LIST_HEAD(sockets); 4500b025033SStefan Hajnoczi struct vsock_stat *st; 4510b025033SStefan Hajnoczi int fd; 4520b025033SStefan Hajnoczi int client_fd; 4530b025033SStefan Hajnoczi 4540b025033SStefan Hajnoczi fd = socket(AF_VSOCK, SOCK_STREAM, 0); 4550b025033SStefan Hajnoczi 4560b025033SStefan Hajnoczi if (bind(fd, &addr.sa, sizeof(addr.svm)) < 0) { 4570b025033SStefan Hajnoczi perror("bind"); 4580b025033SStefan Hajnoczi exit(EXIT_FAILURE); 4590b025033SStefan Hajnoczi } 4600b025033SStefan Hajnoczi 4610b025033SStefan Hajnoczi if (listen(fd, 1) < 0) { 4620b025033SStefan Hajnoczi perror("listen"); 4630b025033SStefan Hajnoczi exit(EXIT_FAILURE); 4640b025033SStefan Hajnoczi } 4650b025033SStefan Hajnoczi 4660b025033SStefan Hajnoczi control_writeln("LISTENING"); 4670b025033SStefan Hajnoczi 4680b025033SStefan Hajnoczi timeout_begin(TIMEOUT); 4690b025033SStefan Hajnoczi do { 4700b025033SStefan Hajnoczi client_fd = accept(fd, &clientaddr.sa, &clientaddr_len); 4710b025033SStefan Hajnoczi timeout_check("accept"); 4720b025033SStefan Hajnoczi } while (client_fd < 0 && errno == EINTR); 4730b025033SStefan Hajnoczi timeout_end(); 4740b025033SStefan Hajnoczi 4750b025033SStefan Hajnoczi if (client_fd < 0) { 4760b025033SStefan Hajnoczi perror("accept"); 4770b025033SStefan Hajnoczi exit(EXIT_FAILURE); 4780b025033SStefan Hajnoczi } 4790b025033SStefan Hajnoczi if (clientaddr.sa.sa_family != AF_VSOCK) { 4800b025033SStefan Hajnoczi fprintf(stderr, "expected AF_VSOCK from accept(2), got %d\n", 4810b025033SStefan Hajnoczi clientaddr.sa.sa_family); 4820b025033SStefan Hajnoczi exit(EXIT_FAILURE); 4830b025033SStefan Hajnoczi } 4840b025033SStefan Hajnoczi if (clientaddr.svm.svm_cid != peer_cid) { 4850b025033SStefan Hajnoczi fprintf(stderr, "expected peer CID %u from accept(2), got %u\n", 4860b025033SStefan Hajnoczi peer_cid, clientaddr.svm.svm_cid); 4870b025033SStefan Hajnoczi exit(EXIT_FAILURE); 4880b025033SStefan Hajnoczi } 4890b025033SStefan Hajnoczi 4900b025033SStefan Hajnoczi read_vsock_stat(&sockets); 4910b025033SStefan Hajnoczi 4920b025033SStefan Hajnoczi check_num_sockets(&sockets, 2); 4930b025033SStefan Hajnoczi find_vsock_stat(&sockets, fd); 4940b025033SStefan Hajnoczi st = find_vsock_stat(&sockets, client_fd); 4950b025033SStefan Hajnoczi check_socket_state(st, TCP_ESTABLISHED); 4960b025033SStefan Hajnoczi 4970b025033SStefan Hajnoczi control_writeln("DONE"); 4980b025033SStefan Hajnoczi control_expectln("DONE"); 4990b025033SStefan Hajnoczi 5000b025033SStefan Hajnoczi close(client_fd); 5010b025033SStefan Hajnoczi close(fd); 5020b025033SStefan Hajnoczi free_sock_stat(&sockets); 5030b025033SStefan Hajnoczi } 5040b025033SStefan Hajnoczi 5050b025033SStefan Hajnoczi static struct { 5060b025033SStefan Hajnoczi const char *name; 5070b025033SStefan Hajnoczi void (*run_client)(unsigned int peer_cid); 5080b025033SStefan Hajnoczi void (*run_server)(unsigned int peer_cid); 5090b025033SStefan Hajnoczi } test_cases[] = { 5100b025033SStefan Hajnoczi { 5110b025033SStefan Hajnoczi .name = "No sockets", 5120b025033SStefan Hajnoczi .run_server = test_no_sockets, 5130b025033SStefan Hajnoczi }, 5140b025033SStefan Hajnoczi { 5150b025033SStefan Hajnoczi .name = "Listen socket", 5160b025033SStefan Hajnoczi .run_server = test_listen_socket_server, 5170b025033SStefan Hajnoczi }, 5180b025033SStefan Hajnoczi { 5190b025033SStefan Hajnoczi .name = "Connect", 5200b025033SStefan Hajnoczi .run_client = test_connect_client, 5210b025033SStefan Hajnoczi .run_server = test_connect_server, 5220b025033SStefan Hajnoczi }, 5230b025033SStefan Hajnoczi {}, 5240b025033SStefan Hajnoczi }; 5250b025033SStefan Hajnoczi 5260b025033SStefan Hajnoczi static void init_signals(void) 5270b025033SStefan Hajnoczi { 5280b025033SStefan Hajnoczi struct sigaction act = { 5290b025033SStefan Hajnoczi .sa_handler = sigalrm, 5300b025033SStefan Hajnoczi }; 5310b025033SStefan Hajnoczi 5320b025033SStefan Hajnoczi sigaction(SIGALRM, &act, NULL); 5330b025033SStefan Hajnoczi signal(SIGPIPE, SIG_IGN); 5340b025033SStefan Hajnoczi } 5350b025033SStefan Hajnoczi 5360b025033SStefan Hajnoczi static unsigned int parse_cid(const char *str) 5370b025033SStefan Hajnoczi { 5380b025033SStefan Hajnoczi char *endptr = NULL; 5390b025033SStefan Hajnoczi unsigned long int n; 5400b025033SStefan Hajnoczi 5410b025033SStefan Hajnoczi errno = 0; 5420b025033SStefan Hajnoczi n = strtoul(str, &endptr, 10); 5430b025033SStefan Hajnoczi if (errno || *endptr != '\0') { 5440b025033SStefan Hajnoczi fprintf(stderr, "malformed CID \"%s\"\n", str); 5450b025033SStefan Hajnoczi exit(EXIT_FAILURE); 5460b025033SStefan Hajnoczi } 5470b025033SStefan Hajnoczi return n; 5480b025033SStefan Hajnoczi } 5490b025033SStefan Hajnoczi 5500b025033SStefan Hajnoczi static const char optstring[] = ""; 5510b025033SStefan Hajnoczi static const struct option longopts[] = { 5520b025033SStefan Hajnoczi { 5530b025033SStefan Hajnoczi .name = "control-host", 5540b025033SStefan Hajnoczi .has_arg = required_argument, 5550b025033SStefan Hajnoczi .val = 'H', 5560b025033SStefan Hajnoczi }, 5570b025033SStefan Hajnoczi { 5580b025033SStefan Hajnoczi .name = "control-port", 5590b025033SStefan Hajnoczi .has_arg = required_argument, 5600b025033SStefan Hajnoczi .val = 'P', 5610b025033SStefan Hajnoczi }, 5620b025033SStefan Hajnoczi { 5630b025033SStefan Hajnoczi .name = "mode", 5640b025033SStefan Hajnoczi .has_arg = required_argument, 5650b025033SStefan Hajnoczi .val = 'm', 5660b025033SStefan Hajnoczi }, 5670b025033SStefan Hajnoczi { 5680b025033SStefan Hajnoczi .name = "peer-cid", 5690b025033SStefan Hajnoczi .has_arg = required_argument, 5700b025033SStefan Hajnoczi .val = 'p', 5710b025033SStefan Hajnoczi }, 5720b025033SStefan Hajnoczi { 5730b025033SStefan Hajnoczi .name = "help", 5740b025033SStefan Hajnoczi .has_arg = no_argument, 5750b025033SStefan Hajnoczi .val = '?', 5760b025033SStefan Hajnoczi }, 5770b025033SStefan Hajnoczi {}, 5780b025033SStefan Hajnoczi }; 5790b025033SStefan Hajnoczi 5800b025033SStefan Hajnoczi static void usage(void) 5810b025033SStefan Hajnoczi { 5820b025033SStefan Hajnoczi fprintf(stderr, "Usage: vsock_diag_test [--help] [--control-host=<host>] --control-port=<port> --mode=client|server --peer-cid=<cid>\n" 5830b025033SStefan Hajnoczi "\n" 5840b025033SStefan Hajnoczi " Server: vsock_diag_test --control-port=1234 --mode=server --peer-cid=3\n" 5850b025033SStefan Hajnoczi " Client: vsock_diag_test --control-host=192.168.0.1 --control-port=1234 --mode=client --peer-cid=2\n" 5860b025033SStefan Hajnoczi "\n" 5870b025033SStefan Hajnoczi "Run vsock_diag.ko tests. Must be launched in both\n" 5880b025033SStefan Hajnoczi "guest and host. One side must use --mode=client and\n" 5890b025033SStefan Hajnoczi "the other side must use --mode=server.\n" 5900b025033SStefan Hajnoczi "\n" 5910b025033SStefan Hajnoczi "A TCP control socket connection is used to coordinate tests\n" 5920b025033SStefan Hajnoczi "between the client and the server. The server requires a\n" 5930b025033SStefan Hajnoczi "listen address and the client requires an address to\n" 5940b025033SStefan Hajnoczi "connect to.\n" 5950b025033SStefan Hajnoczi "\n" 5960b025033SStefan Hajnoczi "The CID of the other side must be given with --peer-cid=<cid>.\n"); 5970b025033SStefan Hajnoczi exit(EXIT_FAILURE); 5980b025033SStefan Hajnoczi } 5990b025033SStefan Hajnoczi 6000b025033SStefan Hajnoczi int main(int argc, char **argv) 6010b025033SStefan Hajnoczi { 6020b025033SStefan Hajnoczi const char *control_host = NULL; 6030b025033SStefan Hajnoczi const char *control_port = NULL; 6040b025033SStefan Hajnoczi int mode = TEST_MODE_UNSET; 6050b025033SStefan Hajnoczi unsigned int peer_cid = VMADDR_CID_ANY; 6060b025033SStefan Hajnoczi int i; 6070b025033SStefan Hajnoczi 6080b025033SStefan Hajnoczi init_signals(); 6090b025033SStefan Hajnoczi 6100b025033SStefan Hajnoczi for (;;) { 6110b025033SStefan Hajnoczi int opt = getopt_long(argc, argv, optstring, longopts, NULL); 6120b025033SStefan Hajnoczi 6130b025033SStefan Hajnoczi if (opt == -1) 6140b025033SStefan Hajnoczi break; 6150b025033SStefan Hajnoczi 6160b025033SStefan Hajnoczi switch (opt) { 6170b025033SStefan Hajnoczi case 'H': 6180b025033SStefan Hajnoczi control_host = optarg; 6190b025033SStefan Hajnoczi break; 6200b025033SStefan Hajnoczi case 'm': 6210b025033SStefan Hajnoczi if (strcmp(optarg, "client") == 0) 6220b025033SStefan Hajnoczi mode = TEST_MODE_CLIENT; 6230b025033SStefan Hajnoczi else if (strcmp(optarg, "server") == 0) 6240b025033SStefan Hajnoczi mode = TEST_MODE_SERVER; 6250b025033SStefan Hajnoczi else { 6260b025033SStefan Hajnoczi fprintf(stderr, "--mode must be \"client\" or \"server\"\n"); 6270b025033SStefan Hajnoczi return EXIT_FAILURE; 6280b025033SStefan Hajnoczi } 6290b025033SStefan Hajnoczi break; 6300b025033SStefan Hajnoczi case 'p': 6310b025033SStefan Hajnoczi peer_cid = parse_cid(optarg); 6320b025033SStefan Hajnoczi break; 6330b025033SStefan Hajnoczi case 'P': 6340b025033SStefan Hajnoczi control_port = optarg; 6350b025033SStefan Hajnoczi break; 6360b025033SStefan Hajnoczi case '?': 6370b025033SStefan Hajnoczi default: 6380b025033SStefan Hajnoczi usage(); 6390b025033SStefan Hajnoczi } 6400b025033SStefan Hajnoczi } 6410b025033SStefan Hajnoczi 6420b025033SStefan Hajnoczi if (!control_port) 6430b025033SStefan Hajnoczi usage(); 6440b025033SStefan Hajnoczi if (mode == TEST_MODE_UNSET) 6450b025033SStefan Hajnoczi usage(); 6460b025033SStefan Hajnoczi if (peer_cid == VMADDR_CID_ANY) 6470b025033SStefan Hajnoczi usage(); 6480b025033SStefan Hajnoczi 6490b025033SStefan Hajnoczi if (!control_host) { 6500b025033SStefan Hajnoczi if (mode != TEST_MODE_SERVER) 6510b025033SStefan Hajnoczi usage(); 6520b025033SStefan Hajnoczi control_host = "0.0.0.0"; 6530b025033SStefan Hajnoczi } 6540b025033SStefan Hajnoczi 6550b025033SStefan Hajnoczi control_init(control_host, control_port, mode == TEST_MODE_SERVER); 6560b025033SStefan Hajnoczi 6570b025033SStefan Hajnoczi for (i = 0; test_cases[i].name; i++) { 6580b025033SStefan Hajnoczi void (*run)(unsigned int peer_cid); 6590b025033SStefan Hajnoczi 6600b025033SStefan Hajnoczi printf("%s...", test_cases[i].name); 6610b025033SStefan Hajnoczi fflush(stdout); 6620b025033SStefan Hajnoczi 6630b025033SStefan Hajnoczi if (mode == TEST_MODE_CLIENT) 6640b025033SStefan Hajnoczi run = test_cases[i].run_client; 6650b025033SStefan Hajnoczi else 6660b025033SStefan Hajnoczi run = test_cases[i].run_server; 6670b025033SStefan Hajnoczi 6680b025033SStefan Hajnoczi if (run) 6690b025033SStefan Hajnoczi run(peer_cid); 6700b025033SStefan Hajnoczi 6710b025033SStefan Hajnoczi printf("ok\n"); 6720b025033SStefan Hajnoczi } 6730b025033SStefan Hajnoczi 6740b025033SStefan Hajnoczi control_cleanup(); 6750b025033SStefan Hajnoczi return EXIT_SUCCESS; 6760b025033SStefan Hajnoczi } 677