^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 1) #include <errno.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 2) #include <error.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 3) #include <getopt.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 4) #include <stdbool.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 5) #include <stdio.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 6) #include <stdlib.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 7) #include <string.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 8) #include <unistd.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 9)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 10) #include <sys/time.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 11) #include <sys/socket.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 12) #include <sys/select.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 13) #include <sys/ioctl.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 14) #include <arpa/inet.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 15) #include <net/if.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 16)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 17) #include <asm/types.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 18) #include <linux/net_tstamp.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 19) #include <linux/errqueue.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 20)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 21) #define ARRAY_SIZE(arr) (sizeof(arr) / sizeof((arr)[0]))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 22)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 23) struct options {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 24) int so_timestamp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 25) int so_timestampns;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 26) int so_timestamping;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 27) };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 28)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 29) struct tstamps {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 30) bool tstamp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 31) bool tstampns;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 32) bool swtstamp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 33) bool hwtstamp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 34) };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 35)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 36) struct socket_type {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 37) char *friendly_name;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 38) int type;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 39) int protocol;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 40) bool enabled;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 41) };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 42)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 43) struct test_case {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 44) struct options sockopt;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 45) struct tstamps expected;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 46) bool enabled;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 47) bool warn_on_fail;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 48) };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 49)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 50) struct sof_flag {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 51) int mask;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 52) char *name;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 53) };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 54)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 55) static struct sof_flag sof_flags[] = {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 56) #define SOF_FLAG(f) { f, #f }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 57) SOF_FLAG(SOF_TIMESTAMPING_SOFTWARE),
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 58) SOF_FLAG(SOF_TIMESTAMPING_RX_SOFTWARE),
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 59) SOF_FLAG(SOF_TIMESTAMPING_RX_HARDWARE),
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 60) };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 61)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 62) static struct socket_type socket_types[] = {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 63) { "ip", SOCK_RAW, IPPROTO_EGP },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 64) { "udp", SOCK_DGRAM, IPPROTO_UDP },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 65) { "tcp", SOCK_STREAM, IPPROTO_TCP },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 66) };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 67)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 68) static struct test_case test_cases[] = {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 69) { {}, {} },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 70) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 71) { .so_timestamp = 1 },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 72) { .tstamp = true }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 73) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 74) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 75) { .so_timestampns = 1 },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 76) { .tstampns = true }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 77) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 78) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 79) { .so_timestamp = 1, .so_timestampns = 1 },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 80) { .tstampns = true }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 81) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 82) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 83) { .so_timestamping = SOF_TIMESTAMPING_RX_SOFTWARE },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 84) {}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 85) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 86) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 87) /* Loopback device does not support hw timestamps. */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 88) { .so_timestamping = SOF_TIMESTAMPING_RX_HARDWARE },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 89) {}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 90) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 91) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 92) { .so_timestamping = SOF_TIMESTAMPING_SOFTWARE },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 93) .warn_on_fail = true
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 94) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 95) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 96) { .so_timestamping = SOF_TIMESTAMPING_RX_SOFTWARE
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 97) | SOF_TIMESTAMPING_RX_HARDWARE },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 98) {}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 99) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 100) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 101) { .so_timestamping = SOF_TIMESTAMPING_SOFTWARE
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 102) | SOF_TIMESTAMPING_RX_SOFTWARE },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 103) { .swtstamp = true }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 104) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 105) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 106) { .so_timestamp = 1, .so_timestamping = SOF_TIMESTAMPING_SOFTWARE
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 107) | SOF_TIMESTAMPING_RX_SOFTWARE },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 108) { .tstamp = true, .swtstamp = true }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 109) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 110) };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 111)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 112) static struct option long_options[] = {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 113) { "list_tests", no_argument, 0, 'l' },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 114) { "test_num", required_argument, 0, 'n' },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 115) { "op_size", required_argument, 0, 's' },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 116) { "tcp", no_argument, 0, 't' },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 117) { "udp", no_argument, 0, 'u' },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 118) { "ip", no_argument, 0, 'i' },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 119) { "strict", no_argument, 0, 'S' },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 120) { "ipv4", no_argument, 0, '4' },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 121) { "ipv6", no_argument, 0, '6' },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 122) { NULL, 0, NULL, 0 },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 123) };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 124)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 125) static int next_port = 19999;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 126) static int op_size = 10 * 1024;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 127)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 128) void print_test_case(struct test_case *t)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 129) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 130) int f = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 131)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 132) printf("sockopts {");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 133) if (t->sockopt.so_timestamp)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 134) printf(" SO_TIMESTAMP ");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 135) if (t->sockopt.so_timestampns)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 136) printf(" SO_TIMESTAMPNS ");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 137) if (t->sockopt.so_timestamping) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 138) printf(" SO_TIMESTAMPING: {");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 139) for (f = 0; f < ARRAY_SIZE(sof_flags); f++)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 140) if (t->sockopt.so_timestamping & sof_flags[f].mask)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 141) printf(" %s |", sof_flags[f].name);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 142) printf("}");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 143) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 144) printf("} expected cmsgs: {");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 145) if (t->expected.tstamp)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 146) printf(" SCM_TIMESTAMP ");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 147) if (t->expected.tstampns)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 148) printf(" SCM_TIMESTAMPNS ");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 149) if (t->expected.swtstamp || t->expected.hwtstamp) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 150) printf(" SCM_TIMESTAMPING {");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 151) if (t->expected.swtstamp)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 152) printf("0");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 153) if (t->expected.swtstamp && t->expected.hwtstamp)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 154) printf(",");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 155) if (t->expected.hwtstamp)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 156) printf("2");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 157) printf("}");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 158) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 159) printf("}\n");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 160) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 161)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 162) void do_send(int src)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 163) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 164) int r;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 165) char *buf = malloc(op_size);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 166)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 167) memset(buf, 'z', op_size);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 168) r = write(src, buf, op_size);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 169) if (r < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 170) error(1, errno, "Failed to sendmsg");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 171)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 172) free(buf);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 173) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 174)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 175) bool do_recv(int rcv, int read_size, struct tstamps expected)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 176) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 177) const int CMSG_SIZE = 1024;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 178)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 179) struct scm_timestamping *ts;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 180) struct tstamps actual = {};
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 181) char cmsg_buf[CMSG_SIZE];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 182) struct iovec recv_iov;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 183) struct cmsghdr *cmsg;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 184) bool failed = false;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 185) struct msghdr hdr;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 186) int flags = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 187) int r;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 188)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 189) memset(&hdr, 0, sizeof(hdr));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 190) hdr.msg_iov = &recv_iov;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 191) hdr.msg_iovlen = 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 192) recv_iov.iov_base = malloc(read_size);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 193) recv_iov.iov_len = read_size;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 194)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 195) hdr.msg_control = cmsg_buf;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 196) hdr.msg_controllen = sizeof(cmsg_buf);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 197)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 198) r = recvmsg(rcv, &hdr, flags);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 199) if (r < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 200) error(1, errno, "Failed to recvmsg");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 201) if (r != read_size)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 202) error(1, 0, "Only received %d bytes of payload.", r);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 203)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 204) if (hdr.msg_flags & (MSG_TRUNC | MSG_CTRUNC))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 205) error(1, 0, "Message was truncated.");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 206)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 207) for (cmsg = CMSG_FIRSTHDR(&hdr); cmsg != NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 208) cmsg = CMSG_NXTHDR(&hdr, cmsg)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 209) if (cmsg->cmsg_level != SOL_SOCKET)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 210) error(1, 0, "Unexpected cmsg_level %d",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 211) cmsg->cmsg_level);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 212) switch (cmsg->cmsg_type) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 213) case SCM_TIMESTAMP:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 214) actual.tstamp = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 215) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 216) case SCM_TIMESTAMPNS:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 217) actual.tstampns = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 218) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 219) case SCM_TIMESTAMPING:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 220) ts = (struct scm_timestamping *)CMSG_DATA(cmsg);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 221) actual.swtstamp = !!ts->ts[0].tv_sec;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 222) if (ts->ts[1].tv_sec != 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 223) error(0, 0, "ts[1] should not be set.");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 224) actual.hwtstamp = !!ts->ts[2].tv_sec;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 225) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 226) default:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 227) error(1, 0, "Unexpected cmsg_type %d", cmsg->cmsg_type);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 228) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 229) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 230)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 231) #define VALIDATE(field) \
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 232) do { \
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 233) if (expected.field != actual.field) { \
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 234) if (expected.field) \
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 235) error(0, 0, "Expected " #field " to be set."); \
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 236) else \
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 237) error(0, 0, \
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 238) "Expected " #field " to not be set."); \
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 239) failed = true; \
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 240) } \
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 241) } while (0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 242)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 243) VALIDATE(tstamp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 244) VALIDATE(tstampns);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 245) VALIDATE(swtstamp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 246) VALIDATE(hwtstamp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 247) #undef VALIDATE
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 248)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 249) free(recv_iov.iov_base);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 250)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 251) return failed;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 252) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 253)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 254) void config_so_flags(int rcv, struct options o)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 255) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 256) int on = 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 257)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 258) if (setsockopt(rcv, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 259) error(1, errno, "Failed to enable SO_REUSEADDR");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 260)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 261) if (o.so_timestamp &&
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 262) setsockopt(rcv, SOL_SOCKET, SO_TIMESTAMP,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 263) &o.so_timestamp, sizeof(o.so_timestamp)) < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 264) error(1, errno, "Failed to enable SO_TIMESTAMP");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 265)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 266) if (o.so_timestampns &&
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 267) setsockopt(rcv, SOL_SOCKET, SO_TIMESTAMPNS,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 268) &o.so_timestampns, sizeof(o.so_timestampns)) < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 269) error(1, errno, "Failed to enable SO_TIMESTAMPNS");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 270)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 271) if (o.so_timestamping &&
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 272) setsockopt(rcv, SOL_SOCKET, SO_TIMESTAMPING,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 273) &o.so_timestamping, sizeof(o.so_timestamping)) < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 274) error(1, errno, "Failed to set SO_TIMESTAMPING");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 275) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 276)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 277) bool run_test_case(struct socket_type *s, int test_num, char ip_version,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 278) bool strict)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 279) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 280) union {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 281) struct sockaddr_in6 addr6;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 282) struct sockaddr_in addr4;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 283) struct sockaddr addr_un;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 284) } addr;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 285) int read_size = op_size;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 286) int src, dst, rcv, port;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 287) socklen_t addr_size;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 288) bool failed = false;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 289)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 290) port = (s->type == SOCK_RAW) ? 0 : next_port++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 291) memset(&addr, 0, sizeof(addr));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 292) if (ip_version == '4') {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 293) addr.addr4.sin_family = AF_INET;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 294) addr.addr4.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 295) addr.addr4.sin_port = htons(port);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 296) addr_size = sizeof(addr.addr4);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 297) if (s->type == SOCK_RAW)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 298) read_size += 20; /* for IPv4 header */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 299) } else {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 300) addr.addr6.sin6_family = AF_INET6;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 301) addr.addr6.sin6_addr = in6addr_loopback;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 302) addr.addr6.sin6_port = htons(port);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 303) addr_size = sizeof(addr.addr6);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 304) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 305) printf("Starting testcase %d over ipv%c...\n", test_num, ip_version);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 306) src = socket(addr.addr_un.sa_family, s->type,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 307) s->protocol);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 308) if (src < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 309) error(1, errno, "Failed to open src socket");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 310)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 311) dst = socket(addr.addr_un.sa_family, s->type,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 312) s->protocol);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 313) if (dst < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 314) error(1, errno, "Failed to open dst socket");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 315)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 316) if (bind(dst, &addr.addr_un, addr_size) < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 317) error(1, errno, "Failed to bind to port %d", port);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 318)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 319) if (s->type == SOCK_STREAM && (listen(dst, 1) < 0))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 320) error(1, errno, "Failed to listen");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 321)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 322) if (connect(src, &addr.addr_un, addr_size) < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 323) error(1, errno, "Failed to connect");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 324)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 325) if (s->type == SOCK_STREAM) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 326) rcv = accept(dst, NULL, NULL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 327) if (rcv < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 328) error(1, errno, "Failed to accept");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 329) close(dst);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 330) } else {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 331) rcv = dst;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 332) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 333)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 334) config_so_flags(rcv, test_cases[test_num].sockopt);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 335) usleep(20000); /* setsockopt for SO_TIMESTAMPING is asynchronous */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 336) do_send(src);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 337)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 338) failed = do_recv(rcv, read_size, test_cases[test_num].expected);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 339)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 340) close(rcv);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 341) close(src);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 342)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 343) if (failed) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 344) printf("FAILURE in testcase %d over ipv%c ", test_num,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 345) ip_version);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 346) print_test_case(&test_cases[test_num]);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 347) if (!strict && test_cases[test_num].warn_on_fail)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 348) failed = false;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 349) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 350) return failed;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 351) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 352)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 353) int main(int argc, char **argv)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 354) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 355) bool all_protocols = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 356) bool all_tests = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 357) bool cfg_ipv4 = false;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 358) bool cfg_ipv6 = false;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 359) bool strict = false;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 360) int arg_index = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 361) int failures = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 362) int s, t, opt;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 363)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 364) while ((opt = getopt_long(argc, argv, "", long_options,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 365) &arg_index)) != -1) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 366) switch (opt) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 367) case 'l':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 368) for (t = 0; t < ARRAY_SIZE(test_cases); t++) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 369) printf("%d\t", t);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 370) print_test_case(&test_cases[t]);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 371) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 372) return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 373) case 'n':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 374) t = atoi(optarg);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 375) if (t >= ARRAY_SIZE(test_cases))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 376) error(1, 0, "Invalid test case: %d", t);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 377) all_tests = false;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 378) test_cases[t].enabled = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 379) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 380) case 's':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 381) op_size = atoi(optarg);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 382) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 383) case 't':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 384) all_protocols = false;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 385) socket_types[2].enabled = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 386) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 387) case 'u':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 388) all_protocols = false;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 389) socket_types[1].enabled = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 390) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 391) case 'i':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 392) all_protocols = false;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 393) socket_types[0].enabled = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 394) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 395) case 'S':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 396) strict = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 397) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 398) case '4':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 399) cfg_ipv4 = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 400) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 401) case '6':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 402) cfg_ipv6 = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 403) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 404) default:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 405) error(1, 0, "Failed to parse parameters.");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 406) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 407) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 408)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 409) for (s = 0; s < ARRAY_SIZE(socket_types); s++) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 410) if (!all_protocols && !socket_types[s].enabled)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 411) continue;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 412)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 413) printf("Testing %s...\n", socket_types[s].friendly_name);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 414) for (t = 0; t < ARRAY_SIZE(test_cases); t++) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 415) if (!all_tests && !test_cases[t].enabled)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 416) continue;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 417) if (cfg_ipv4 || !cfg_ipv6)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 418) if (run_test_case(&socket_types[s], t, '4',
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 419) strict))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 420) failures++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 421) if (cfg_ipv6 || !cfg_ipv4)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 422) if (run_test_case(&socket_types[s], t, '6',
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 423) strict))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 424) failures++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 425) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 426) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 427) if (!failures)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 428) printf("PASSED.\n");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 429) return failures;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 430) }