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 <stdlib.h> 130b025033SStefan Hajnoczi #include <string.h> 140b025033SStefan Hajnoczi #include <errno.h> 150b025033SStefan Hajnoczi #include <unistd.h> 160b025033SStefan Hajnoczi #include <sys/stat.h> 170b025033SStefan Hajnoczi #include <sys/types.h> 180b025033SStefan Hajnoczi #include <linux/list.h> 190b025033SStefan Hajnoczi #include <linux/net.h> 200b025033SStefan Hajnoczi #include <linux/netlink.h> 210b025033SStefan Hajnoczi #include <linux/sock_diag.h> 2243985468SStefan Hajnoczi #include <linux/vm_sockets_diag.h> 230b025033SStefan Hajnoczi #include <netinet/tcp.h> 240b025033SStefan Hajnoczi 250b025033SStefan Hajnoczi #include "timeout.h" 260b025033SStefan Hajnoczi #include "control.h" 27df7e0e0dSStefan Hajnoczi #include "util.h" 280b025033SStefan Hajnoczi 290b025033SStefan Hajnoczi /* Per-socket status */ 300b025033SStefan Hajnoczi struct vsock_stat { 310b025033SStefan Hajnoczi struct list_head list; 320b025033SStefan Hajnoczi struct vsock_diag_msg msg; 330b025033SStefan Hajnoczi }; 340b025033SStefan Hajnoczi 350b025033SStefan Hajnoczi static const char *sock_type_str(int type) 360b025033SStefan Hajnoczi { 370b025033SStefan Hajnoczi switch (type) { 380b025033SStefan Hajnoczi case SOCK_DGRAM: 390b025033SStefan Hajnoczi return "DGRAM"; 400b025033SStefan Hajnoczi case SOCK_STREAM: 410b025033SStefan Hajnoczi return "STREAM"; 420b025033SStefan Hajnoczi default: 430b025033SStefan Hajnoczi return "INVALID TYPE"; 440b025033SStefan Hajnoczi } 450b025033SStefan Hajnoczi } 460b025033SStefan Hajnoczi 470b025033SStefan Hajnoczi static const char *sock_state_str(int state) 480b025033SStefan Hajnoczi { 490b025033SStefan Hajnoczi switch (state) { 500b025033SStefan Hajnoczi case TCP_CLOSE: 510b025033SStefan Hajnoczi return "UNCONNECTED"; 520b025033SStefan Hajnoczi case TCP_SYN_SENT: 530b025033SStefan Hajnoczi return "CONNECTING"; 540b025033SStefan Hajnoczi case TCP_ESTABLISHED: 550b025033SStefan Hajnoczi return "CONNECTED"; 560b025033SStefan Hajnoczi case TCP_CLOSING: 570b025033SStefan Hajnoczi return "DISCONNECTING"; 580b025033SStefan Hajnoczi case TCP_LISTEN: 590b025033SStefan Hajnoczi return "LISTEN"; 600b025033SStefan Hajnoczi default: 610b025033SStefan Hajnoczi return "INVALID STATE"; 620b025033SStefan Hajnoczi } 630b025033SStefan Hajnoczi } 640b025033SStefan Hajnoczi 650b025033SStefan Hajnoczi static const char *sock_shutdown_str(int shutdown) 660b025033SStefan Hajnoczi { 670b025033SStefan Hajnoczi switch (shutdown) { 680b025033SStefan Hajnoczi case 1: 690b025033SStefan Hajnoczi return "RCV_SHUTDOWN"; 700b025033SStefan Hajnoczi case 2: 710b025033SStefan Hajnoczi return "SEND_SHUTDOWN"; 720b025033SStefan Hajnoczi case 3: 730b025033SStefan Hajnoczi return "RCV_SHUTDOWN | SEND_SHUTDOWN"; 740b025033SStefan Hajnoczi default: 750b025033SStefan Hajnoczi return "0"; 760b025033SStefan Hajnoczi } 770b025033SStefan Hajnoczi } 780b025033SStefan Hajnoczi 790b025033SStefan Hajnoczi static void print_vsock_addr(FILE *fp, unsigned int cid, unsigned int port) 800b025033SStefan Hajnoczi { 810b025033SStefan Hajnoczi if (cid == VMADDR_CID_ANY) 820b025033SStefan Hajnoczi fprintf(fp, "*:"); 830b025033SStefan Hajnoczi else 840b025033SStefan Hajnoczi fprintf(fp, "%u:", cid); 850b025033SStefan Hajnoczi 860b025033SStefan Hajnoczi if (port == VMADDR_PORT_ANY) 870b025033SStefan Hajnoczi fprintf(fp, "*"); 880b025033SStefan Hajnoczi else 890b025033SStefan Hajnoczi fprintf(fp, "%u", port); 900b025033SStefan Hajnoczi } 910b025033SStefan Hajnoczi 920b025033SStefan Hajnoczi static void print_vsock_stat(FILE *fp, struct vsock_stat *st) 930b025033SStefan Hajnoczi { 940b025033SStefan Hajnoczi print_vsock_addr(fp, st->msg.vdiag_src_cid, st->msg.vdiag_src_port); 950b025033SStefan Hajnoczi fprintf(fp, " "); 960b025033SStefan Hajnoczi print_vsock_addr(fp, st->msg.vdiag_dst_cid, st->msg.vdiag_dst_port); 970b025033SStefan Hajnoczi fprintf(fp, " %s %s %s %u\n", 980b025033SStefan Hajnoczi sock_type_str(st->msg.vdiag_type), 990b025033SStefan Hajnoczi sock_state_str(st->msg.vdiag_state), 1000b025033SStefan Hajnoczi sock_shutdown_str(st->msg.vdiag_shutdown), 1010b025033SStefan Hajnoczi st->msg.vdiag_ino); 1020b025033SStefan Hajnoczi } 1030b025033SStefan Hajnoczi 1040b025033SStefan Hajnoczi static void print_vsock_stats(FILE *fp, struct list_head *head) 1050b025033SStefan Hajnoczi { 1060b025033SStefan Hajnoczi struct vsock_stat *st; 1070b025033SStefan Hajnoczi 1080b025033SStefan Hajnoczi list_for_each_entry(st, head, list) 1090b025033SStefan Hajnoczi print_vsock_stat(fp, st); 1100b025033SStefan Hajnoczi } 1110b025033SStefan Hajnoczi 1120b025033SStefan Hajnoczi static struct vsock_stat *find_vsock_stat(struct list_head *head, int fd) 1130b025033SStefan Hajnoczi { 1140b025033SStefan Hajnoczi struct vsock_stat *st; 1150b025033SStefan Hajnoczi struct stat stat; 1160b025033SStefan Hajnoczi 1170b025033SStefan Hajnoczi if (fstat(fd, &stat) < 0) { 1180b025033SStefan Hajnoczi perror("fstat"); 1190b025033SStefan Hajnoczi exit(EXIT_FAILURE); 1200b025033SStefan Hajnoczi } 1210b025033SStefan Hajnoczi 1220b025033SStefan Hajnoczi list_for_each_entry(st, head, list) 1230b025033SStefan Hajnoczi if (st->msg.vdiag_ino == stat.st_ino) 1240b025033SStefan Hajnoczi return st; 1250b025033SStefan Hajnoczi 1260b025033SStefan Hajnoczi fprintf(stderr, "cannot find fd %d\n", fd); 1270b025033SStefan Hajnoczi exit(EXIT_FAILURE); 1280b025033SStefan Hajnoczi } 1290b025033SStefan Hajnoczi 1300b025033SStefan Hajnoczi static void check_no_sockets(struct list_head *head) 1310b025033SStefan Hajnoczi { 1320b025033SStefan Hajnoczi if (!list_empty(head)) { 1330b025033SStefan Hajnoczi fprintf(stderr, "expected no sockets\n"); 1340b025033SStefan Hajnoczi print_vsock_stats(stderr, head); 1350b025033SStefan Hajnoczi exit(1); 1360b025033SStefan Hajnoczi } 1370b025033SStefan Hajnoczi } 1380b025033SStefan Hajnoczi 1390b025033SStefan Hajnoczi static void check_num_sockets(struct list_head *head, int expected) 1400b025033SStefan Hajnoczi { 1410b025033SStefan Hajnoczi struct list_head *node; 1420b025033SStefan Hajnoczi int n = 0; 1430b025033SStefan Hajnoczi 1440b025033SStefan Hajnoczi list_for_each(node, head) 1450b025033SStefan Hajnoczi n++; 1460b025033SStefan Hajnoczi 1470b025033SStefan Hajnoczi if (n != expected) { 1480b025033SStefan Hajnoczi fprintf(stderr, "expected %d sockets, found %d\n", 1490b025033SStefan Hajnoczi expected, n); 1500b025033SStefan Hajnoczi print_vsock_stats(stderr, head); 1510b025033SStefan Hajnoczi exit(EXIT_FAILURE); 1520b025033SStefan Hajnoczi } 1530b025033SStefan Hajnoczi } 1540b025033SStefan Hajnoczi 1550b025033SStefan Hajnoczi static void check_socket_state(struct vsock_stat *st, __u8 state) 1560b025033SStefan Hajnoczi { 1570b025033SStefan Hajnoczi if (st->msg.vdiag_state != state) { 1580b025033SStefan Hajnoczi fprintf(stderr, "expected socket state %#x, got %#x\n", 1590b025033SStefan Hajnoczi state, st->msg.vdiag_state); 1600b025033SStefan Hajnoczi exit(EXIT_FAILURE); 1610b025033SStefan Hajnoczi } 1620b025033SStefan Hajnoczi } 1630b025033SStefan Hajnoczi 1640b025033SStefan Hajnoczi static void send_req(int fd) 1650b025033SStefan Hajnoczi { 1660b025033SStefan Hajnoczi struct sockaddr_nl nladdr = { 1670b025033SStefan Hajnoczi .nl_family = AF_NETLINK, 1680b025033SStefan Hajnoczi }; 1690b025033SStefan Hajnoczi struct { 1700b025033SStefan Hajnoczi struct nlmsghdr nlh; 1710b025033SStefan Hajnoczi struct vsock_diag_req vreq; 1720b025033SStefan Hajnoczi } req = { 1730b025033SStefan Hajnoczi .nlh = { 1740b025033SStefan Hajnoczi .nlmsg_len = sizeof(req), 1750b025033SStefan Hajnoczi .nlmsg_type = SOCK_DIAG_BY_FAMILY, 1760b025033SStefan Hajnoczi .nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP, 1770b025033SStefan Hajnoczi }, 1780b025033SStefan Hajnoczi .vreq = { 1790b025033SStefan Hajnoczi .sdiag_family = AF_VSOCK, 1800b025033SStefan Hajnoczi .vdiag_states = ~(__u32)0, 1810b025033SStefan Hajnoczi }, 1820b025033SStefan Hajnoczi }; 1830b025033SStefan Hajnoczi struct iovec iov = { 1840b025033SStefan Hajnoczi .iov_base = &req, 1850b025033SStefan Hajnoczi .iov_len = sizeof(req), 1860b025033SStefan Hajnoczi }; 1870b025033SStefan Hajnoczi struct msghdr msg = { 1880b025033SStefan Hajnoczi .msg_name = &nladdr, 1890b025033SStefan Hajnoczi .msg_namelen = sizeof(nladdr), 1900b025033SStefan Hajnoczi .msg_iov = &iov, 1910b025033SStefan Hajnoczi .msg_iovlen = 1, 1920b025033SStefan Hajnoczi }; 1930b025033SStefan Hajnoczi 1940b025033SStefan Hajnoczi for (;;) { 1950b025033SStefan Hajnoczi if (sendmsg(fd, &msg, 0) < 0) { 1960b025033SStefan Hajnoczi if (errno == EINTR) 1970b025033SStefan Hajnoczi continue; 1980b025033SStefan Hajnoczi 1990b025033SStefan Hajnoczi perror("sendmsg"); 2000b025033SStefan Hajnoczi exit(EXIT_FAILURE); 2010b025033SStefan Hajnoczi } 2020b025033SStefan Hajnoczi 2030b025033SStefan Hajnoczi return; 2040b025033SStefan Hajnoczi } 2050b025033SStefan Hajnoczi } 2060b025033SStefan Hajnoczi 2070b025033SStefan Hajnoczi static ssize_t recv_resp(int fd, void *buf, size_t len) 2080b025033SStefan Hajnoczi { 2090b025033SStefan Hajnoczi struct sockaddr_nl nladdr = { 2100b025033SStefan Hajnoczi .nl_family = AF_NETLINK, 2110b025033SStefan Hajnoczi }; 2120b025033SStefan Hajnoczi struct iovec iov = { 2130b025033SStefan Hajnoczi .iov_base = buf, 2140b025033SStefan Hajnoczi .iov_len = len, 2150b025033SStefan Hajnoczi }; 2160b025033SStefan Hajnoczi struct msghdr msg = { 2170b025033SStefan Hajnoczi .msg_name = &nladdr, 2180b025033SStefan Hajnoczi .msg_namelen = sizeof(nladdr), 2190b025033SStefan Hajnoczi .msg_iov = &iov, 2200b025033SStefan Hajnoczi .msg_iovlen = 1, 2210b025033SStefan Hajnoczi }; 2220b025033SStefan Hajnoczi ssize_t ret; 2230b025033SStefan Hajnoczi 2240b025033SStefan Hajnoczi do { 2250b025033SStefan Hajnoczi ret = recvmsg(fd, &msg, 0); 2260b025033SStefan Hajnoczi } while (ret < 0 && errno == EINTR); 2270b025033SStefan Hajnoczi 2280b025033SStefan Hajnoczi if (ret < 0) { 2290b025033SStefan Hajnoczi perror("recvmsg"); 2300b025033SStefan Hajnoczi exit(EXIT_FAILURE); 2310b025033SStefan Hajnoczi } 2320b025033SStefan Hajnoczi 2330b025033SStefan Hajnoczi return ret; 2340b025033SStefan Hajnoczi } 2350b025033SStefan Hajnoczi 2360b025033SStefan Hajnoczi static void add_vsock_stat(struct list_head *sockets, 2370b025033SStefan Hajnoczi const struct vsock_diag_msg *resp) 2380b025033SStefan Hajnoczi { 2390b025033SStefan Hajnoczi struct vsock_stat *st; 2400b025033SStefan Hajnoczi 2410b025033SStefan Hajnoczi st = malloc(sizeof(*st)); 2420b025033SStefan Hajnoczi if (!st) { 2430b025033SStefan Hajnoczi perror("malloc"); 2440b025033SStefan Hajnoczi exit(EXIT_FAILURE); 2450b025033SStefan Hajnoczi } 2460b025033SStefan Hajnoczi 2470b025033SStefan Hajnoczi st->msg = *resp; 2480b025033SStefan Hajnoczi list_add_tail(&st->list, sockets); 2490b025033SStefan Hajnoczi } 2500b025033SStefan Hajnoczi 2510b025033SStefan Hajnoczi /* 2520b025033SStefan Hajnoczi * Read vsock stats into a list. 2530b025033SStefan Hajnoczi */ 2540b025033SStefan Hajnoczi static void read_vsock_stat(struct list_head *sockets) 2550b025033SStefan Hajnoczi { 2560b025033SStefan Hajnoczi long buf[8192 / sizeof(long)]; 2570b025033SStefan Hajnoczi int fd; 2580b025033SStefan Hajnoczi 2590b025033SStefan Hajnoczi fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_SOCK_DIAG); 2600b025033SStefan Hajnoczi if (fd < 0) { 2610b025033SStefan Hajnoczi perror("socket"); 2620b025033SStefan Hajnoczi exit(EXIT_FAILURE); 2630b025033SStefan Hajnoczi } 2640b025033SStefan Hajnoczi 2650b025033SStefan Hajnoczi send_req(fd); 2660b025033SStefan Hajnoczi 2670b025033SStefan Hajnoczi for (;;) { 2680b025033SStefan Hajnoczi const struct nlmsghdr *h; 2690b025033SStefan Hajnoczi ssize_t ret; 2700b025033SStefan Hajnoczi 2710b025033SStefan Hajnoczi ret = recv_resp(fd, buf, sizeof(buf)); 2720b025033SStefan Hajnoczi if (ret == 0) 2730b025033SStefan Hajnoczi goto done; 2740b025033SStefan Hajnoczi if (ret < sizeof(*h)) { 2750b025033SStefan Hajnoczi fprintf(stderr, "short read of %zd bytes\n", ret); 2760b025033SStefan Hajnoczi exit(EXIT_FAILURE); 2770b025033SStefan Hajnoczi } 2780b025033SStefan Hajnoczi 2790b025033SStefan Hajnoczi h = (struct nlmsghdr *)buf; 2800b025033SStefan Hajnoczi 2810b025033SStefan Hajnoczi while (NLMSG_OK(h, ret)) { 2820b025033SStefan Hajnoczi if (h->nlmsg_type == NLMSG_DONE) 2830b025033SStefan Hajnoczi goto done; 2840b025033SStefan Hajnoczi 2850b025033SStefan Hajnoczi if (h->nlmsg_type == NLMSG_ERROR) { 2860b025033SStefan Hajnoczi const struct nlmsgerr *err = NLMSG_DATA(h); 2870b025033SStefan Hajnoczi 2880b025033SStefan Hajnoczi if (h->nlmsg_len < NLMSG_LENGTH(sizeof(*err))) 2890b025033SStefan Hajnoczi fprintf(stderr, "NLMSG_ERROR\n"); 2900b025033SStefan Hajnoczi else { 2910b025033SStefan Hajnoczi errno = -err->error; 2920b025033SStefan Hajnoczi perror("NLMSG_ERROR"); 2930b025033SStefan Hajnoczi } 2940b025033SStefan Hajnoczi 2950b025033SStefan Hajnoczi exit(EXIT_FAILURE); 2960b025033SStefan Hajnoczi } 2970b025033SStefan Hajnoczi 2980b025033SStefan Hajnoczi if (h->nlmsg_type != SOCK_DIAG_BY_FAMILY) { 2990b025033SStefan Hajnoczi fprintf(stderr, "unexpected nlmsg_type %#x\n", 3000b025033SStefan Hajnoczi h->nlmsg_type); 3010b025033SStefan Hajnoczi exit(EXIT_FAILURE); 3020b025033SStefan Hajnoczi } 3030b025033SStefan Hajnoczi if (h->nlmsg_len < 3040b025033SStefan Hajnoczi NLMSG_LENGTH(sizeof(struct vsock_diag_msg))) { 3050b025033SStefan Hajnoczi fprintf(stderr, "short vsock_diag_msg\n"); 3060b025033SStefan Hajnoczi exit(EXIT_FAILURE); 3070b025033SStefan Hajnoczi } 3080b025033SStefan Hajnoczi 3090b025033SStefan Hajnoczi add_vsock_stat(sockets, NLMSG_DATA(h)); 3100b025033SStefan Hajnoczi 3110b025033SStefan Hajnoczi h = NLMSG_NEXT(h, ret); 3120b025033SStefan Hajnoczi } 3130b025033SStefan Hajnoczi } 3140b025033SStefan Hajnoczi 3150b025033SStefan Hajnoczi done: 3160b025033SStefan Hajnoczi close(fd); 3170b025033SStefan Hajnoczi } 3180b025033SStefan Hajnoczi 3190b025033SStefan Hajnoczi static void free_sock_stat(struct list_head *sockets) 3200b025033SStefan Hajnoczi { 3210b025033SStefan Hajnoczi struct vsock_stat *st; 3220b025033SStefan Hajnoczi struct vsock_stat *next; 3230b025033SStefan Hajnoczi 3240b025033SStefan Hajnoczi list_for_each_entry_safe(st, next, sockets, list) 3250b025033SStefan Hajnoczi free(st); 3260b025033SStefan Hajnoczi } 3270b025033SStefan Hajnoczi 328df7e0e0dSStefan Hajnoczi static void test_no_sockets(const struct test_opts *opts) 3290b025033SStefan Hajnoczi { 3300b025033SStefan Hajnoczi LIST_HEAD(sockets); 3310b025033SStefan Hajnoczi 3320b025033SStefan Hajnoczi read_vsock_stat(&sockets); 3330b025033SStefan Hajnoczi 3340b025033SStefan Hajnoczi check_no_sockets(&sockets); 3350b025033SStefan Hajnoczi 3360b025033SStefan Hajnoczi free_sock_stat(&sockets); 3370b025033SStefan Hajnoczi } 3380b025033SStefan Hajnoczi 339df7e0e0dSStefan Hajnoczi static void test_listen_socket_server(const struct test_opts *opts) 3400b025033SStefan Hajnoczi { 3410b025033SStefan Hajnoczi union { 3420b025033SStefan Hajnoczi struct sockaddr sa; 3430b025033SStefan Hajnoczi struct sockaddr_vm svm; 3440b025033SStefan Hajnoczi } addr = { 3450b025033SStefan Hajnoczi .svm = { 3460b025033SStefan Hajnoczi .svm_family = AF_VSOCK, 3470b025033SStefan Hajnoczi .svm_port = 1234, 3480b025033SStefan Hajnoczi .svm_cid = VMADDR_CID_ANY, 3490b025033SStefan Hajnoczi }, 3500b025033SStefan Hajnoczi }; 3510b025033SStefan Hajnoczi LIST_HEAD(sockets); 3520b025033SStefan Hajnoczi struct vsock_stat *st; 3530b025033SStefan Hajnoczi int fd; 3540b025033SStefan Hajnoczi 3550b025033SStefan Hajnoczi fd = socket(AF_VSOCK, SOCK_STREAM, 0); 3560b025033SStefan Hajnoczi 3570b025033SStefan Hajnoczi if (bind(fd, &addr.sa, sizeof(addr.svm)) < 0) { 3580b025033SStefan Hajnoczi perror("bind"); 3590b025033SStefan Hajnoczi exit(EXIT_FAILURE); 3600b025033SStefan Hajnoczi } 3610b025033SStefan Hajnoczi 3620b025033SStefan Hajnoczi if (listen(fd, 1) < 0) { 3630b025033SStefan Hajnoczi perror("listen"); 3640b025033SStefan Hajnoczi exit(EXIT_FAILURE); 3650b025033SStefan Hajnoczi } 3660b025033SStefan Hajnoczi 3670b025033SStefan Hajnoczi read_vsock_stat(&sockets); 3680b025033SStefan Hajnoczi 3690b025033SStefan Hajnoczi check_num_sockets(&sockets, 1); 3700b025033SStefan Hajnoczi st = find_vsock_stat(&sockets, fd); 3710b025033SStefan Hajnoczi check_socket_state(st, TCP_LISTEN); 3720b025033SStefan Hajnoczi 3730b025033SStefan Hajnoczi close(fd); 3740b025033SStefan Hajnoczi free_sock_stat(&sockets); 3750b025033SStefan Hajnoczi } 3760b025033SStefan Hajnoczi 377df7e0e0dSStefan Hajnoczi static void test_connect_client(const struct test_opts *opts) 3780b025033SStefan Hajnoczi { 3790b025033SStefan Hajnoczi int fd; 3800b025033SStefan Hajnoczi LIST_HEAD(sockets); 3810b025033SStefan Hajnoczi struct vsock_stat *st; 3820b025033SStefan Hajnoczi 3839bb8a29dSStefan Hajnoczi fd = vsock_stream_connect(opts->peer_cid, 1234); 3849bb8a29dSStefan Hajnoczi if (fd < 0) { 3850b025033SStefan Hajnoczi perror("connect"); 3860b025033SStefan Hajnoczi exit(EXIT_FAILURE); 3870b025033SStefan Hajnoczi } 3880b025033SStefan Hajnoczi 3890b025033SStefan Hajnoczi read_vsock_stat(&sockets); 3900b025033SStefan Hajnoczi 3910b025033SStefan Hajnoczi check_num_sockets(&sockets, 1); 3920b025033SStefan Hajnoczi st = find_vsock_stat(&sockets, fd); 3930b025033SStefan Hajnoczi check_socket_state(st, TCP_ESTABLISHED); 3940b025033SStefan Hajnoczi 3950b025033SStefan Hajnoczi control_expectln("DONE"); 3960b025033SStefan Hajnoczi control_writeln("DONE"); 3970b025033SStefan Hajnoczi 3980b025033SStefan Hajnoczi close(fd); 3990b025033SStefan Hajnoczi free_sock_stat(&sockets); 4000b025033SStefan Hajnoczi } 4010b025033SStefan Hajnoczi 402df7e0e0dSStefan Hajnoczi static void test_connect_server(const struct test_opts *opts) 4030b025033SStefan Hajnoczi { 4040b025033SStefan Hajnoczi struct vsock_stat *st; 4059bb8a29dSStefan Hajnoczi LIST_HEAD(sockets); 4060b025033SStefan Hajnoczi int client_fd; 4070b025033SStefan Hajnoczi 4089bb8a29dSStefan Hajnoczi client_fd = vsock_stream_accept(VMADDR_CID_ANY, 1234, NULL); 4090b025033SStefan Hajnoczi if (client_fd < 0) { 4100b025033SStefan Hajnoczi perror("accept"); 4110b025033SStefan Hajnoczi exit(EXIT_FAILURE); 4120b025033SStefan Hajnoczi } 4130b025033SStefan Hajnoczi 4140b025033SStefan Hajnoczi read_vsock_stat(&sockets); 4150b025033SStefan Hajnoczi 4169bb8a29dSStefan Hajnoczi check_num_sockets(&sockets, 1); 4170b025033SStefan Hajnoczi st = find_vsock_stat(&sockets, client_fd); 4180b025033SStefan Hajnoczi check_socket_state(st, TCP_ESTABLISHED); 4190b025033SStefan Hajnoczi 4200b025033SStefan Hajnoczi control_writeln("DONE"); 4210b025033SStefan Hajnoczi control_expectln("DONE"); 4220b025033SStefan Hajnoczi 4230b025033SStefan Hajnoczi close(client_fd); 4240b025033SStefan Hajnoczi free_sock_stat(&sockets); 4250b025033SStefan Hajnoczi } 4260b025033SStefan Hajnoczi 427df7e0e0dSStefan Hajnoczi static struct test_case test_cases[] = { 4280b025033SStefan Hajnoczi { 4290b025033SStefan Hajnoczi .name = "No sockets", 4300b025033SStefan Hajnoczi .run_server = test_no_sockets, 4310b025033SStefan Hajnoczi }, 4320b025033SStefan Hajnoczi { 4330b025033SStefan Hajnoczi .name = "Listen socket", 4340b025033SStefan Hajnoczi .run_server = test_listen_socket_server, 4350b025033SStefan Hajnoczi }, 4360b025033SStefan Hajnoczi { 4370b025033SStefan Hajnoczi .name = "Connect", 4380b025033SStefan Hajnoczi .run_client = test_connect_client, 4390b025033SStefan Hajnoczi .run_server = test_connect_server, 4400b025033SStefan Hajnoczi }, 4410b025033SStefan Hajnoczi {}, 4420b025033SStefan Hajnoczi }; 4430b025033SStefan Hajnoczi 4440b025033SStefan Hajnoczi static const char optstring[] = ""; 4450b025033SStefan Hajnoczi static const struct option longopts[] = { 4460b025033SStefan Hajnoczi { 4470b025033SStefan Hajnoczi .name = "control-host", 4480b025033SStefan Hajnoczi .has_arg = required_argument, 4490b025033SStefan Hajnoczi .val = 'H', 4500b025033SStefan Hajnoczi }, 4510b025033SStefan Hajnoczi { 4520b025033SStefan Hajnoczi .name = "control-port", 4530b025033SStefan Hajnoczi .has_arg = required_argument, 4540b025033SStefan Hajnoczi .val = 'P', 4550b025033SStefan Hajnoczi }, 4560b025033SStefan Hajnoczi { 4570b025033SStefan Hajnoczi .name = "mode", 4580b025033SStefan Hajnoczi .has_arg = required_argument, 4590b025033SStefan Hajnoczi .val = 'm', 4600b025033SStefan Hajnoczi }, 4610b025033SStefan Hajnoczi { 4620b025033SStefan Hajnoczi .name = "peer-cid", 4630b025033SStefan Hajnoczi .has_arg = required_argument, 4640b025033SStefan Hajnoczi .val = 'p', 4650b025033SStefan Hajnoczi }, 4660b025033SStefan Hajnoczi { 4675a2b2425SStefano Garzarella .name = "list", 4685a2b2425SStefano Garzarella .has_arg = no_argument, 4695a2b2425SStefano Garzarella .val = 'l', 4705a2b2425SStefano Garzarella }, 4715a2b2425SStefano Garzarella { 4725a2b2425SStefano Garzarella .name = "skip", 4735a2b2425SStefano Garzarella .has_arg = required_argument, 4745a2b2425SStefano Garzarella .val = 's', 4755a2b2425SStefano Garzarella }, 4765a2b2425SStefano Garzarella { 4770b025033SStefan Hajnoczi .name = "help", 4780b025033SStefan Hajnoczi .has_arg = no_argument, 4790b025033SStefan Hajnoczi .val = '?', 4800b025033SStefan Hajnoczi }, 4810b025033SStefan Hajnoczi {}, 4820b025033SStefan Hajnoczi }; 4830b025033SStefan Hajnoczi 4840b025033SStefan Hajnoczi static void usage(void) 4850b025033SStefan Hajnoczi { 4865a2b2425SStefano Garzarella fprintf(stderr, "Usage: vsock_diag_test [--help] [--control-host=<host>] --control-port=<port> --mode=client|server --peer-cid=<cid> [--list] [--skip=<test_id>]\n" 4870b025033SStefan Hajnoczi "\n" 4880b025033SStefan Hajnoczi " Server: vsock_diag_test --control-port=1234 --mode=server --peer-cid=3\n" 4890b025033SStefan Hajnoczi " Client: vsock_diag_test --control-host=192.168.0.1 --control-port=1234 --mode=client --peer-cid=2\n" 4900b025033SStefan Hajnoczi "\n" 4910b025033SStefan Hajnoczi "Run vsock_diag.ko tests. Must be launched in both\n" 4920b025033SStefan Hajnoczi "guest and host. One side must use --mode=client and\n" 4930b025033SStefan Hajnoczi "the other side must use --mode=server.\n" 4940b025033SStefan Hajnoczi "\n" 4950b025033SStefan Hajnoczi "A TCP control socket connection is used to coordinate tests\n" 4960b025033SStefan Hajnoczi "between the client and the server. The server requires a\n" 4970b025033SStefan Hajnoczi "listen address and the client requires an address to\n" 4980b025033SStefan Hajnoczi "connect to.\n" 4990b025033SStefan Hajnoczi "\n" 5000b025033SStefan Hajnoczi "The CID of the other side must be given with --peer-cid=<cid>.\n"); 5010b025033SStefan Hajnoczi exit(EXIT_FAILURE); 5020b025033SStefan Hajnoczi } 5030b025033SStefan Hajnoczi 5040b025033SStefan Hajnoczi int main(int argc, char **argv) 5050b025033SStefan Hajnoczi { 5060b025033SStefan Hajnoczi const char *control_host = NULL; 5070b025033SStefan Hajnoczi const char *control_port = NULL; 508df7e0e0dSStefan Hajnoczi struct test_opts opts = { 509df7e0e0dSStefan Hajnoczi .mode = TEST_MODE_UNSET, 510df7e0e0dSStefan Hajnoczi .peer_cid = VMADDR_CID_ANY, 511df7e0e0dSStefan Hajnoczi }; 5120b025033SStefan Hajnoczi 5130b025033SStefan Hajnoczi init_signals(); 5140b025033SStefan Hajnoczi 5150b025033SStefan Hajnoczi for (;;) { 5160b025033SStefan Hajnoczi int opt = getopt_long(argc, argv, optstring, longopts, NULL); 5170b025033SStefan Hajnoczi 5180b025033SStefan Hajnoczi if (opt == -1) 5190b025033SStefan Hajnoczi break; 5200b025033SStefan Hajnoczi 5210b025033SStefan Hajnoczi switch (opt) { 5220b025033SStefan Hajnoczi case 'H': 5230b025033SStefan Hajnoczi control_host = optarg; 5240b025033SStefan Hajnoczi break; 5250b025033SStefan Hajnoczi case 'm': 5260b025033SStefan Hajnoczi if (strcmp(optarg, "client") == 0) 527df7e0e0dSStefan Hajnoczi opts.mode = TEST_MODE_CLIENT; 5280b025033SStefan Hajnoczi else if (strcmp(optarg, "server") == 0) 529df7e0e0dSStefan Hajnoczi opts.mode = TEST_MODE_SERVER; 5300b025033SStefan Hajnoczi else { 5310b025033SStefan Hajnoczi fprintf(stderr, "--mode must be \"client\" or \"server\"\n"); 5320b025033SStefan Hajnoczi return EXIT_FAILURE; 5330b025033SStefan Hajnoczi } 5340b025033SStefan Hajnoczi break; 5350b025033SStefan Hajnoczi case 'p': 536df7e0e0dSStefan Hajnoczi opts.peer_cid = parse_cid(optarg); 5370b025033SStefan Hajnoczi break; 5380b025033SStefan Hajnoczi case 'P': 5390b025033SStefan Hajnoczi control_port = optarg; 5400b025033SStefan Hajnoczi break; 5415a2b2425SStefano Garzarella case 'l': 5425a2b2425SStefano Garzarella list_tests(test_cases); 5435a2b2425SStefano Garzarella break; 5445a2b2425SStefano Garzarella case 's': 5455a2b2425SStefano Garzarella skip_test(test_cases, ARRAY_SIZE(test_cases) - 1, 5465a2b2425SStefano Garzarella optarg); 5475a2b2425SStefano Garzarella break; 5480b025033SStefan Hajnoczi case '?': 5490b025033SStefan Hajnoczi default: 5500b025033SStefan Hajnoczi usage(); 5510b025033SStefan Hajnoczi } 5520b025033SStefan Hajnoczi } 5530b025033SStefan Hajnoczi 5540b025033SStefan Hajnoczi if (!control_port) 5550b025033SStefan Hajnoczi usage(); 556df7e0e0dSStefan Hajnoczi if (opts.mode == TEST_MODE_UNSET) 5570b025033SStefan Hajnoczi usage(); 558df7e0e0dSStefan Hajnoczi if (opts.peer_cid == VMADDR_CID_ANY) 5590b025033SStefan Hajnoczi usage(); 5600b025033SStefan Hajnoczi 5610b025033SStefan Hajnoczi if (!control_host) { 562df7e0e0dSStefan Hajnoczi if (opts.mode != TEST_MODE_SERVER) 5630b025033SStefan Hajnoczi usage(); 5640b025033SStefan Hajnoczi control_host = "0.0.0.0"; 5650b025033SStefan Hajnoczi } 5660b025033SStefan Hajnoczi 567df7e0e0dSStefan Hajnoczi control_init(control_host, control_port, 568df7e0e0dSStefan Hajnoczi opts.mode == TEST_MODE_SERVER); 5690b025033SStefan Hajnoczi 570df7e0e0dSStefan Hajnoczi run_tests(test_cases, &opts); 5710b025033SStefan Hajnoczi 5720b025033SStefan Hajnoczi control_cleanup(); 5730b025033SStefan Hajnoczi return EXIT_SUCCESS; 5740b025033SStefan Hajnoczi } 575