1af5136f9SWillem de Bruijn // SPDX-License-Identifier: GPL-2.0
2af5136f9SWillem de Bruijn /*
3af5136f9SWillem de Bruijn * Test the SO_TXTIME API
4af5136f9SWillem de Bruijn *
504080634SCarlos Llamas * Takes a stream of { payload, delivery time }[], to be sent across two
604080634SCarlos Llamas * processes. Start this program on two separate network namespaces or
704080634SCarlos Llamas * connected hosts, one instance in transmit mode and the other in receive
804080634SCarlos Llamas * mode using the '-r' option. Receiver will compare arrival timestamps to
904080634SCarlos Llamas * the expected stream. Sender will read transmit timestamps from the error
1004080634SCarlos Llamas * queue. The streams can differ due to out-of-order delivery and drops.
11af5136f9SWillem de Bruijn */
12af5136f9SWillem de Bruijn
13af5136f9SWillem de Bruijn #define _GNU_SOURCE
14af5136f9SWillem de Bruijn
15af5136f9SWillem de Bruijn #include <arpa/inet.h>
16af5136f9SWillem de Bruijn #include <error.h>
17af5136f9SWillem de Bruijn #include <errno.h>
18ea6a5476SWillem de Bruijn #include <inttypes.h>
19af5136f9SWillem de Bruijn #include <linux/net_tstamp.h>
20ea6a5476SWillem de Bruijn #include <linux/errqueue.h>
21ca882609SWillem de Bruijn #include <linux/if_ether.h>
22ea6a5476SWillem de Bruijn #include <linux/ipv6.h>
23ca882609SWillem de Bruijn #include <linux/udp.h>
24af5136f9SWillem de Bruijn #include <stdbool.h>
25af5136f9SWillem de Bruijn #include <stdlib.h>
26af5136f9SWillem de Bruijn #include <stdio.h>
27af5136f9SWillem de Bruijn #include <string.h>
28af5136f9SWillem de Bruijn #include <sys/socket.h>
29af5136f9SWillem de Bruijn #include <sys/stat.h>
30af5136f9SWillem de Bruijn #include <sys/time.h>
31af5136f9SWillem de Bruijn #include <sys/types.h>
32af5136f9SWillem de Bruijn #include <time.h>
33af5136f9SWillem de Bruijn #include <unistd.h>
3404080634SCarlos Llamas #include <poll.h>
35af5136f9SWillem de Bruijn
36af5136f9SWillem de Bruijn static int cfg_clockid = CLOCK_TAI;
37af5136f9SWillem de Bruijn static uint16_t cfg_port = 8000;
38ea6a5476SWillem de Bruijn static int cfg_variance_us = 4000;
3904080634SCarlos Llamas static uint64_t cfg_start_time_ns;
4004080634SCarlos Llamas static int cfg_mark;
4104080634SCarlos Llamas static bool cfg_rx;
42af5136f9SWillem de Bruijn
43af5136f9SWillem de Bruijn static uint64_t glob_tstart;
4404080634SCarlos Llamas static uint64_t tdeliver_max;
45af5136f9SWillem de Bruijn
46af5136f9SWillem de Bruijn /* encode one timed transmission (of a 1B payload) */
47af5136f9SWillem de Bruijn struct timed_send {
48af5136f9SWillem de Bruijn char data;
49af5136f9SWillem de Bruijn int64_t delay_us;
50af5136f9SWillem de Bruijn };
51af5136f9SWillem de Bruijn
52af5136f9SWillem de Bruijn #define MAX_NUM_PKT 8
5304080634SCarlos Llamas static struct timed_send cfg_buf[MAX_NUM_PKT];
54af5136f9SWillem de Bruijn static int cfg_num_pkt;
55af5136f9SWillem de Bruijn
56ea6a5476SWillem de Bruijn static int cfg_errq_level;
57ea6a5476SWillem de Bruijn static int cfg_errq_type;
58ea6a5476SWillem de Bruijn
5904080634SCarlos Llamas static struct sockaddr_storage cfg_dst_addr;
6004080634SCarlos Llamas static struct sockaddr_storage cfg_src_addr;
6104080634SCarlos Llamas static socklen_t cfg_alen;
6204080634SCarlos Llamas
gettime_ns(clockid_t clock)6304080634SCarlos Llamas static uint64_t gettime_ns(clockid_t clock)
64af5136f9SWillem de Bruijn {
65af5136f9SWillem de Bruijn struct timespec ts;
66af5136f9SWillem de Bruijn
6704080634SCarlos Llamas if (clock_gettime(clock, &ts))
68af5136f9SWillem de Bruijn error(1, errno, "gettime");
69af5136f9SWillem de Bruijn
70af5136f9SWillem de Bruijn return ts.tv_sec * (1000ULL * 1000 * 1000) + ts.tv_nsec;
71af5136f9SWillem de Bruijn }
72af5136f9SWillem de Bruijn
do_send_one(int fdt,struct timed_send * ts)73af5136f9SWillem de Bruijn static void do_send_one(int fdt, struct timed_send *ts)
74af5136f9SWillem de Bruijn {
75af5136f9SWillem de Bruijn char control[CMSG_SPACE(sizeof(uint64_t))];
76af5136f9SWillem de Bruijn struct msghdr msg = {0};
77af5136f9SWillem de Bruijn struct iovec iov = {0};
78af5136f9SWillem de Bruijn struct cmsghdr *cm;
79af5136f9SWillem de Bruijn uint64_t tdeliver;
80af5136f9SWillem de Bruijn int ret;
81af5136f9SWillem de Bruijn
82af5136f9SWillem de Bruijn iov.iov_base = &ts->data;
83af5136f9SWillem de Bruijn iov.iov_len = 1;
84af5136f9SWillem de Bruijn
85af5136f9SWillem de Bruijn msg.msg_iov = &iov;
86af5136f9SWillem de Bruijn msg.msg_iovlen = 1;
8704080634SCarlos Llamas msg.msg_name = (struct sockaddr *)&cfg_dst_addr;
8804080634SCarlos Llamas msg.msg_namelen = cfg_alen;
89af5136f9SWillem de Bruijn
90af5136f9SWillem de Bruijn if (ts->delay_us >= 0) {
91af5136f9SWillem de Bruijn memset(control, 0, sizeof(control));
92af5136f9SWillem de Bruijn msg.msg_control = &control;
93af5136f9SWillem de Bruijn msg.msg_controllen = sizeof(control);
94af5136f9SWillem de Bruijn
95af5136f9SWillem de Bruijn tdeliver = glob_tstart + ts->delay_us * 1000;
9604080634SCarlos Llamas tdeliver_max = tdeliver_max > tdeliver ?
9704080634SCarlos Llamas tdeliver_max : tdeliver;
98af5136f9SWillem de Bruijn
99af5136f9SWillem de Bruijn cm = CMSG_FIRSTHDR(&msg);
100af5136f9SWillem de Bruijn cm->cmsg_level = SOL_SOCKET;
101af5136f9SWillem de Bruijn cm->cmsg_type = SCM_TXTIME;
102af5136f9SWillem de Bruijn cm->cmsg_len = CMSG_LEN(sizeof(tdeliver));
103af5136f9SWillem de Bruijn memcpy(CMSG_DATA(cm), &tdeliver, sizeof(tdeliver));
104af5136f9SWillem de Bruijn }
105af5136f9SWillem de Bruijn
106af5136f9SWillem de Bruijn ret = sendmsg(fdt, &msg, 0);
107af5136f9SWillem de Bruijn if (ret == -1)
108af5136f9SWillem de Bruijn error(1, errno, "write");
109af5136f9SWillem de Bruijn if (ret == 0)
110af5136f9SWillem de Bruijn error(1, 0, "write: 0B");
111af5136f9SWillem de Bruijn
112af5136f9SWillem de Bruijn }
113af5136f9SWillem de Bruijn
do_recv_one(int fdr,struct timed_send * ts)11404080634SCarlos Llamas static void do_recv_one(int fdr, struct timed_send *ts)
115af5136f9SWillem de Bruijn {
116af5136f9SWillem de Bruijn int64_t tstop, texpect;
117af5136f9SWillem de Bruijn char rbuf[2];
118af5136f9SWillem de Bruijn int ret;
119af5136f9SWillem de Bruijn
120af5136f9SWillem de Bruijn ret = recv(fdr, rbuf, sizeof(rbuf), 0);
121ea6a5476SWillem de Bruijn if (ret == -1 && errno == EAGAIN)
12204080634SCarlos Llamas error(1, EAGAIN, "recv: timeout");
123af5136f9SWillem de Bruijn if (ret == -1)
124af5136f9SWillem de Bruijn error(1, errno, "read");
125af5136f9SWillem de Bruijn if (ret != 1)
126af5136f9SWillem de Bruijn error(1, 0, "read: %dB", ret);
127af5136f9SWillem de Bruijn
12804080634SCarlos Llamas tstop = (gettime_ns(cfg_clockid) - glob_tstart) / 1000;
129af5136f9SWillem de Bruijn texpect = ts->delay_us >= 0 ? ts->delay_us : 0;
130af5136f9SWillem de Bruijn
131670cd684SMasami Hiramatsu fprintf(stderr, "payload:%c delay:%lld expected:%lld (us)\n",
132670cd684SMasami Hiramatsu rbuf[0], (long long)tstop, (long long)texpect);
133af5136f9SWillem de Bruijn
134af5136f9SWillem de Bruijn if (rbuf[0] != ts->data)
135af5136f9SWillem de Bruijn error(1, 0, "payload mismatch. expected %c", ts->data);
136af5136f9SWillem de Bruijn
137b4da96ffSTanner Love if (llabs(tstop - texpect) > cfg_variance_us)
138af5136f9SWillem de Bruijn error(1, 0, "exceeds variance (%d us)", cfg_variance_us);
139af5136f9SWillem de Bruijn }
140af5136f9SWillem de Bruijn
do_recv_verify_empty(int fdr)141af5136f9SWillem de Bruijn static void do_recv_verify_empty(int fdr)
142af5136f9SWillem de Bruijn {
143af5136f9SWillem de Bruijn char rbuf[1];
144af5136f9SWillem de Bruijn int ret;
145af5136f9SWillem de Bruijn
146af5136f9SWillem de Bruijn ret = recv(fdr, rbuf, sizeof(rbuf), 0);
147af5136f9SWillem de Bruijn if (ret != -1 || errno != EAGAIN)
148af5136f9SWillem de Bruijn error(1, 0, "recv: not empty as expected (%d, %d)", ret, errno);
149af5136f9SWillem de Bruijn }
150af5136f9SWillem de Bruijn
do_recv_errqueue_timeout(int fdt)15104080634SCarlos Llamas static int do_recv_errqueue_timeout(int fdt)
152ea6a5476SWillem de Bruijn {
153ea6a5476SWillem de Bruijn char control[CMSG_SPACE(sizeof(struct sock_extended_err)) +
154ea6a5476SWillem de Bruijn CMSG_SPACE(sizeof(struct sockaddr_in6))] = {0};
155ca882609SWillem de Bruijn char data[sizeof(struct ethhdr) + sizeof(struct ipv6hdr) +
156ca882609SWillem de Bruijn sizeof(struct udphdr) + 1];
157ea6a5476SWillem de Bruijn struct sock_extended_err *err;
15804080634SCarlos Llamas int ret, num_tstamp = 0;
159ea6a5476SWillem de Bruijn struct msghdr msg = {0};
160ea6a5476SWillem de Bruijn struct iovec iov = {0};
161ea6a5476SWillem de Bruijn struct cmsghdr *cm;
162ea6a5476SWillem de Bruijn int64_t tstamp = 0;
163ea6a5476SWillem de Bruijn
164ea6a5476SWillem de Bruijn iov.iov_base = data;
165ea6a5476SWillem de Bruijn iov.iov_len = sizeof(data);
166ea6a5476SWillem de Bruijn
167ea6a5476SWillem de Bruijn msg.msg_iov = &iov;
168ea6a5476SWillem de Bruijn msg.msg_iovlen = 1;
169ea6a5476SWillem de Bruijn
170ea6a5476SWillem de Bruijn msg.msg_control = control;
171ea6a5476SWillem de Bruijn msg.msg_controllen = sizeof(control);
172ea6a5476SWillem de Bruijn
173ea6a5476SWillem de Bruijn while (1) {
174ca882609SWillem de Bruijn const char *reason;
175ca882609SWillem de Bruijn
176ea6a5476SWillem de Bruijn ret = recvmsg(fdt, &msg, MSG_ERRQUEUE);
177ea6a5476SWillem de Bruijn if (ret == -1 && errno == EAGAIN)
178ea6a5476SWillem de Bruijn break;
179ea6a5476SWillem de Bruijn if (ret == -1)
180ea6a5476SWillem de Bruijn error(1, errno, "errqueue");
181ea6a5476SWillem de Bruijn if (msg.msg_flags != MSG_ERRQUEUE)
182ea6a5476SWillem de Bruijn error(1, 0, "errqueue: flags 0x%x\n", msg.msg_flags);
183ea6a5476SWillem de Bruijn
184ea6a5476SWillem de Bruijn cm = CMSG_FIRSTHDR(&msg);
185ea6a5476SWillem de Bruijn if (cm->cmsg_level != cfg_errq_level ||
186ea6a5476SWillem de Bruijn cm->cmsg_type != cfg_errq_type)
187ea6a5476SWillem de Bruijn error(1, 0, "errqueue: type 0x%x.0x%x\n",
188ea6a5476SWillem de Bruijn cm->cmsg_level, cm->cmsg_type);
189ea6a5476SWillem de Bruijn
190ea6a5476SWillem de Bruijn err = (struct sock_extended_err *)CMSG_DATA(cm);
191ea6a5476SWillem de Bruijn if (err->ee_origin != SO_EE_ORIGIN_TXTIME)
192ea6a5476SWillem de Bruijn error(1, 0, "errqueue: origin 0x%x\n", err->ee_origin);
193ca882609SWillem de Bruijn
194ca882609SWillem de Bruijn switch (err->ee_errno) {
195ca882609SWillem de Bruijn case ECANCELED:
196ca882609SWillem de Bruijn if (err->ee_code != SO_EE_CODE_TXTIME_MISSED)
197ca882609SWillem de Bruijn error(1, 0, "errqueue: unknown ECANCELED %u\n",
198ca882609SWillem de Bruijn err->ee_code);
199ca882609SWillem de Bruijn reason = "missed txtime";
200ca882609SWillem de Bruijn break;
201ca882609SWillem de Bruijn case EINVAL:
202ca882609SWillem de Bruijn if (err->ee_code != SO_EE_CODE_TXTIME_INVALID_PARAM)
203ca882609SWillem de Bruijn error(1, 0, "errqueue: unknown EINVAL %u\n",
204ca882609SWillem de Bruijn err->ee_code);
205ca882609SWillem de Bruijn reason = "invalid txtime";
206ca882609SWillem de Bruijn break;
207ca882609SWillem de Bruijn default:
208ca882609SWillem de Bruijn error(1, 0, "errqueue: errno %u code %u\n",
209ca882609SWillem de Bruijn err->ee_errno, err->ee_code);
210c85b3bb7SYang Li }
211ea6a5476SWillem de Bruijn
212ea6a5476SWillem de Bruijn tstamp = ((int64_t) err->ee_data) << 32 | err->ee_info;
213ea6a5476SWillem de Bruijn tstamp -= (int64_t) glob_tstart;
214ea6a5476SWillem de Bruijn tstamp /= 1000 * 1000;
215ca882609SWillem de Bruijn fprintf(stderr, "send: pkt %c at %" PRId64 "ms dropped: %s\n",
216ca882609SWillem de Bruijn data[ret - 1], tstamp, reason);
217ea6a5476SWillem de Bruijn
218ea6a5476SWillem de Bruijn msg.msg_flags = 0;
219ea6a5476SWillem de Bruijn msg.msg_controllen = sizeof(control);
22004080634SCarlos Llamas num_tstamp++;
221ea6a5476SWillem de Bruijn }
222ea6a5476SWillem de Bruijn
22304080634SCarlos Llamas return num_tstamp;
22404080634SCarlos Llamas }
22504080634SCarlos Llamas
recv_errqueue_msgs(int fdt)22604080634SCarlos Llamas static void recv_errqueue_msgs(int fdt)
22704080634SCarlos Llamas {
22804080634SCarlos Llamas struct pollfd pfd = { .fd = fdt, .events = POLLERR };
22904080634SCarlos Llamas const int timeout_ms = 10;
23004080634SCarlos Llamas int ret, num_tstamp = 0;
23104080634SCarlos Llamas
23204080634SCarlos Llamas do {
23304080634SCarlos Llamas ret = poll(&pfd, 1, timeout_ms);
23404080634SCarlos Llamas if (ret == -1)
23504080634SCarlos Llamas error(1, errno, "poll");
23604080634SCarlos Llamas
23704080634SCarlos Llamas if (ret && (pfd.revents & POLLERR))
23804080634SCarlos Llamas num_tstamp += do_recv_errqueue_timeout(fdt);
23904080634SCarlos Llamas
24004080634SCarlos Llamas if (num_tstamp == cfg_num_pkt)
24104080634SCarlos Llamas break;
24204080634SCarlos Llamas
24304080634SCarlos Llamas } while (gettime_ns(cfg_clockid) < tdeliver_max);
24404080634SCarlos Llamas }
24504080634SCarlos Llamas
start_time_wait(void)24604080634SCarlos Llamas static void start_time_wait(void)
24704080634SCarlos Llamas {
24804080634SCarlos Llamas uint64_t now;
24904080634SCarlos Llamas int err;
25004080634SCarlos Llamas
25104080634SCarlos Llamas if (!cfg_start_time_ns)
25204080634SCarlos Llamas return;
25304080634SCarlos Llamas
25404080634SCarlos Llamas now = gettime_ns(CLOCK_REALTIME);
25504080634SCarlos Llamas if (cfg_start_time_ns < now)
25604080634SCarlos Llamas return;
25704080634SCarlos Llamas
25804080634SCarlos Llamas err = usleep((cfg_start_time_ns - now) / 1000);
25904080634SCarlos Llamas if (err)
26004080634SCarlos Llamas error(1, errno, "usleep");
261ea6a5476SWillem de Bruijn }
262ea6a5476SWillem de Bruijn
setsockopt_txtime(int fd)263af5136f9SWillem de Bruijn static void setsockopt_txtime(int fd)
264af5136f9SWillem de Bruijn {
265af5136f9SWillem de Bruijn struct sock_txtime so_txtime_val = { .clockid = cfg_clockid };
266af5136f9SWillem de Bruijn struct sock_txtime so_txtime_val_read = { 0 };
267af5136f9SWillem de Bruijn socklen_t vallen = sizeof(so_txtime_val);
268af5136f9SWillem de Bruijn
269ea6a5476SWillem de Bruijn so_txtime_val.flags = SOF_TXTIME_REPORT_ERRORS;
270ea6a5476SWillem de Bruijn
271af5136f9SWillem de Bruijn if (setsockopt(fd, SOL_SOCKET, SO_TXTIME,
272af5136f9SWillem de Bruijn &so_txtime_val, sizeof(so_txtime_val)))
273af5136f9SWillem de Bruijn error(1, errno, "setsockopt txtime");
274af5136f9SWillem de Bruijn
275af5136f9SWillem de Bruijn if (getsockopt(fd, SOL_SOCKET, SO_TXTIME,
276af5136f9SWillem de Bruijn &so_txtime_val_read, &vallen))
277af5136f9SWillem de Bruijn error(1, errno, "getsockopt txtime");
278af5136f9SWillem de Bruijn
279af5136f9SWillem de Bruijn if (vallen != sizeof(so_txtime_val) ||
280af5136f9SWillem de Bruijn memcmp(&so_txtime_val, &so_txtime_val_read, vallen))
281af5136f9SWillem de Bruijn error(1, 0, "getsockopt txtime: mismatch");
282af5136f9SWillem de Bruijn }
283af5136f9SWillem de Bruijn
setup_tx(struct sockaddr * addr,socklen_t alen)284af5136f9SWillem de Bruijn static int setup_tx(struct sockaddr *addr, socklen_t alen)
285af5136f9SWillem de Bruijn {
286af5136f9SWillem de Bruijn int fd;
287af5136f9SWillem de Bruijn
288af5136f9SWillem de Bruijn fd = socket(addr->sa_family, SOCK_DGRAM, 0);
289af5136f9SWillem de Bruijn if (fd == -1)
290af5136f9SWillem de Bruijn error(1, errno, "socket t");
291af5136f9SWillem de Bruijn
292af5136f9SWillem de Bruijn if (connect(fd, addr, alen))
293af5136f9SWillem de Bruijn error(1, errno, "connect");
294af5136f9SWillem de Bruijn
295af5136f9SWillem de Bruijn setsockopt_txtime(fd);
296af5136f9SWillem de Bruijn
29704080634SCarlos Llamas if (cfg_mark &&
29804080634SCarlos Llamas setsockopt(fd, SOL_SOCKET, SO_MARK, &cfg_mark, sizeof(cfg_mark)))
29904080634SCarlos Llamas error(1, errno, "setsockopt mark");
30004080634SCarlos Llamas
301af5136f9SWillem de Bruijn return fd;
302af5136f9SWillem de Bruijn }
303af5136f9SWillem de Bruijn
setup_rx(struct sockaddr * addr,socklen_t alen)304af5136f9SWillem de Bruijn static int setup_rx(struct sockaddr *addr, socklen_t alen)
305af5136f9SWillem de Bruijn {
306af5136f9SWillem de Bruijn struct timeval tv = { .tv_usec = 100 * 1000 };
307af5136f9SWillem de Bruijn int fd;
308af5136f9SWillem de Bruijn
309af5136f9SWillem de Bruijn fd = socket(addr->sa_family, SOCK_DGRAM, 0);
310af5136f9SWillem de Bruijn if (fd == -1)
311af5136f9SWillem de Bruijn error(1, errno, "socket r");
312af5136f9SWillem de Bruijn
313af5136f9SWillem de Bruijn if (bind(fd, addr, alen))
314af5136f9SWillem de Bruijn error(1, errno, "bind");
315af5136f9SWillem de Bruijn
316af5136f9SWillem de Bruijn if (setsockopt(fd, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv)))
317af5136f9SWillem de Bruijn error(1, errno, "setsockopt rcv timeout");
318af5136f9SWillem de Bruijn
319af5136f9SWillem de Bruijn return fd;
320af5136f9SWillem de Bruijn }
321af5136f9SWillem de Bruijn
do_test_tx(struct sockaddr * addr,socklen_t alen)32204080634SCarlos Llamas static void do_test_tx(struct sockaddr *addr, socklen_t alen)
323af5136f9SWillem de Bruijn {
32404080634SCarlos Llamas int fdt, i;
325af5136f9SWillem de Bruijn
326af5136f9SWillem de Bruijn fprintf(stderr, "\nSO_TXTIME ipv%c clock %s\n",
327af5136f9SWillem de Bruijn addr->sa_family == PF_INET ? '4' : '6',
328af5136f9SWillem de Bruijn cfg_clockid == CLOCK_TAI ? "tai" : "monotonic");
329af5136f9SWillem de Bruijn
330af5136f9SWillem de Bruijn fdt = setup_tx(addr, alen);
33104080634SCarlos Llamas
33204080634SCarlos Llamas start_time_wait();
33304080634SCarlos Llamas glob_tstart = gettime_ns(cfg_clockid);
33404080634SCarlos Llamas
33504080634SCarlos Llamas for (i = 0; i < cfg_num_pkt; i++)
33604080634SCarlos Llamas do_send_one(fdt, &cfg_buf[i]);
33704080634SCarlos Llamas
33804080634SCarlos Llamas recv_errqueue_msgs(fdt);
33904080634SCarlos Llamas
34004080634SCarlos Llamas if (close(fdt))
34104080634SCarlos Llamas error(1, errno, "close t");
34204080634SCarlos Llamas }
34304080634SCarlos Llamas
do_test_rx(struct sockaddr * addr,socklen_t alen)34404080634SCarlos Llamas static void do_test_rx(struct sockaddr *addr, socklen_t alen)
34504080634SCarlos Llamas {
34604080634SCarlos Llamas int fdr, i;
34704080634SCarlos Llamas
348af5136f9SWillem de Bruijn fdr = setup_rx(addr, alen);
349af5136f9SWillem de Bruijn
35004080634SCarlos Llamas start_time_wait();
35104080634SCarlos Llamas glob_tstart = gettime_ns(cfg_clockid);
352af5136f9SWillem de Bruijn
353af5136f9SWillem de Bruijn for (i = 0; i < cfg_num_pkt; i++)
35404080634SCarlos Llamas do_recv_one(fdr, &cfg_buf[i]);
355af5136f9SWillem de Bruijn
356af5136f9SWillem de Bruijn do_recv_verify_empty(fdr);
357af5136f9SWillem de Bruijn
358af5136f9SWillem de Bruijn if (close(fdr))
359af5136f9SWillem de Bruijn error(1, errno, "close r");
36004080634SCarlos Llamas }
36104080634SCarlos Llamas
setup_sockaddr(int domain,const char * str_addr,struct sockaddr_storage * sockaddr)36204080634SCarlos Llamas static void setup_sockaddr(int domain, const char *str_addr,
36304080634SCarlos Llamas struct sockaddr_storage *sockaddr)
36404080634SCarlos Llamas {
36504080634SCarlos Llamas struct sockaddr_in6 *addr6 = (void *) sockaddr;
36604080634SCarlos Llamas struct sockaddr_in *addr4 = (void *) sockaddr;
36704080634SCarlos Llamas
36804080634SCarlos Llamas switch (domain) {
36904080634SCarlos Llamas case PF_INET:
37004080634SCarlos Llamas memset(addr4, 0, sizeof(*addr4));
37104080634SCarlos Llamas addr4->sin_family = AF_INET;
37204080634SCarlos Llamas addr4->sin_port = htons(cfg_port);
37304080634SCarlos Llamas if (str_addr &&
37404080634SCarlos Llamas inet_pton(AF_INET, str_addr, &(addr4->sin_addr)) != 1)
37504080634SCarlos Llamas error(1, 0, "ipv4 parse error: %s", str_addr);
37604080634SCarlos Llamas break;
37704080634SCarlos Llamas case PF_INET6:
37804080634SCarlos Llamas memset(addr6, 0, sizeof(*addr6));
37904080634SCarlos Llamas addr6->sin6_family = AF_INET6;
38004080634SCarlos Llamas addr6->sin6_port = htons(cfg_port);
38104080634SCarlos Llamas if (str_addr &&
38204080634SCarlos Llamas inet_pton(AF_INET6, str_addr, &(addr6->sin6_addr)) != 1)
38304080634SCarlos Llamas error(1, 0, "ipv6 parse error: %s", str_addr);
38404080634SCarlos Llamas break;
38504080634SCarlos Llamas }
386af5136f9SWillem de Bruijn }
387af5136f9SWillem de Bruijn
parse_io(const char * optarg,struct timed_send * array)388af5136f9SWillem de Bruijn static int parse_io(const char *optarg, struct timed_send *array)
389af5136f9SWillem de Bruijn {
390af5136f9SWillem de Bruijn char *arg, *tok;
391af5136f9SWillem de Bruijn int aoff = 0;
392af5136f9SWillem de Bruijn
393af5136f9SWillem de Bruijn arg = strdup(optarg);
394af5136f9SWillem de Bruijn if (!arg)
395af5136f9SWillem de Bruijn error(1, errno, "strdup");
396af5136f9SWillem de Bruijn
397af5136f9SWillem de Bruijn while ((tok = strtok(arg, ","))) {
398af5136f9SWillem de Bruijn arg = NULL; /* only pass non-zero on first call */
399af5136f9SWillem de Bruijn
400af5136f9SWillem de Bruijn if (aoff / 2 == MAX_NUM_PKT)
401af5136f9SWillem de Bruijn error(1, 0, "exceeds max pkt count (%d)", MAX_NUM_PKT);
402af5136f9SWillem de Bruijn
403af5136f9SWillem de Bruijn if (aoff & 1) { /* parse delay */
404af5136f9SWillem de Bruijn array->delay_us = strtol(tok, NULL, 0) * 1000;
405af5136f9SWillem de Bruijn array++;
406af5136f9SWillem de Bruijn } else { /* parse character */
407af5136f9SWillem de Bruijn array->data = tok[0];
408af5136f9SWillem de Bruijn }
409af5136f9SWillem de Bruijn
410af5136f9SWillem de Bruijn aoff++;
411af5136f9SWillem de Bruijn }
412af5136f9SWillem de Bruijn
413af5136f9SWillem de Bruijn free(arg);
414af5136f9SWillem de Bruijn
415af5136f9SWillem de Bruijn return aoff / 2;
416af5136f9SWillem de Bruijn }
417af5136f9SWillem de Bruijn
usage(const char * progname)41804080634SCarlos Llamas static void usage(const char *progname)
41904080634SCarlos Llamas {
42004080634SCarlos Llamas fprintf(stderr, "\nUsage: %s [options] <payload>\n"
42104080634SCarlos Llamas "Options:\n"
42204080634SCarlos Llamas " -4 only IPv4\n"
42304080634SCarlos Llamas " -6 only IPv6\n"
424*e325bef5SMarc Kleine-Budde " -c <clock> monotonic or tai (default)\n"
42504080634SCarlos Llamas " -D <addr> destination IP address (server)\n"
42604080634SCarlos Llamas " -S <addr> source IP address (client)\n"
42704080634SCarlos Llamas " -r run rx mode\n"
42804080634SCarlos Llamas " -t <nsec> start time (UTC nanoseconds)\n"
42904080634SCarlos Llamas " -m <mark> socket mark\n"
43004080634SCarlos Llamas "\n",
43104080634SCarlos Llamas progname);
43204080634SCarlos Llamas exit(1);
43304080634SCarlos Llamas }
43404080634SCarlos Llamas
parse_opts(int argc,char ** argv)435af5136f9SWillem de Bruijn static void parse_opts(int argc, char **argv)
436af5136f9SWillem de Bruijn {
43704080634SCarlos Llamas char *daddr = NULL, *saddr = NULL;
43804080634SCarlos Llamas int domain = PF_UNSPEC;
43904080634SCarlos Llamas int c;
440af5136f9SWillem de Bruijn
44104080634SCarlos Llamas while ((c = getopt(argc, argv, "46c:S:D:rt:m:")) != -1) {
442af5136f9SWillem de Bruijn switch (c) {
443af5136f9SWillem de Bruijn case '4':
44404080634SCarlos Llamas if (domain != PF_UNSPEC)
44504080634SCarlos Llamas error(1, 0, "Pass one of -4 or -6");
44604080634SCarlos Llamas domain = PF_INET;
44704080634SCarlos Llamas cfg_alen = sizeof(struct sockaddr_in);
44804080634SCarlos Llamas cfg_errq_level = SOL_IP;
44904080634SCarlos Llamas cfg_errq_type = IP_RECVERR;
450af5136f9SWillem de Bruijn break;
451af5136f9SWillem de Bruijn case '6':
45204080634SCarlos Llamas if (domain != PF_UNSPEC)
45304080634SCarlos Llamas error(1, 0, "Pass one of -4 or -6");
45404080634SCarlos Llamas domain = PF_INET6;
45504080634SCarlos Llamas cfg_alen = sizeof(struct sockaddr_in6);
45604080634SCarlos Llamas cfg_errq_level = SOL_IPV6;
45704080634SCarlos Llamas cfg_errq_type = IPV6_RECVERR;
458af5136f9SWillem de Bruijn break;
459af5136f9SWillem de Bruijn case 'c':
460af5136f9SWillem de Bruijn if (!strcmp(optarg, "tai"))
461af5136f9SWillem de Bruijn cfg_clockid = CLOCK_TAI;
462af5136f9SWillem de Bruijn else if (!strcmp(optarg, "monotonic") ||
463af5136f9SWillem de Bruijn !strcmp(optarg, "mono"))
464af5136f9SWillem de Bruijn cfg_clockid = CLOCK_MONOTONIC;
465af5136f9SWillem de Bruijn else
466af5136f9SWillem de Bruijn error(1, 0, "unknown clock id %s", optarg);
467af5136f9SWillem de Bruijn break;
46804080634SCarlos Llamas case 'S':
46904080634SCarlos Llamas saddr = optarg;
47004080634SCarlos Llamas break;
47104080634SCarlos Llamas case 'D':
47204080634SCarlos Llamas daddr = optarg;
47304080634SCarlos Llamas break;
47404080634SCarlos Llamas case 'r':
47504080634SCarlos Llamas cfg_rx = true;
47604080634SCarlos Llamas break;
47704080634SCarlos Llamas case 't':
478ba5ece30SMarc Kleine-Budde cfg_start_time_ns = strtoll(optarg, NULL, 0);
47904080634SCarlos Llamas break;
48004080634SCarlos Llamas case 'm':
48104080634SCarlos Llamas cfg_mark = strtol(optarg, NULL, 0);
48204080634SCarlos Llamas break;
483af5136f9SWillem de Bruijn default:
48404080634SCarlos Llamas usage(argv[0]);
485af5136f9SWillem de Bruijn }
486af5136f9SWillem de Bruijn }
487af5136f9SWillem de Bruijn
48804080634SCarlos Llamas if (argc - optind != 1)
48904080634SCarlos Llamas usage(argv[0]);
490af5136f9SWillem de Bruijn
49104080634SCarlos Llamas if (domain == PF_UNSPEC)
49204080634SCarlos Llamas error(1, 0, "Pass one of -4 or -6");
49304080634SCarlos Llamas if (!daddr)
49404080634SCarlos Llamas error(1, 0, "-D <server addr> required\n");
49504080634SCarlos Llamas if (!cfg_rx && !saddr)
49604080634SCarlos Llamas error(1, 0, "-S <client addr> required\n");
49704080634SCarlos Llamas
49804080634SCarlos Llamas setup_sockaddr(domain, daddr, &cfg_dst_addr);
49904080634SCarlos Llamas setup_sockaddr(domain, saddr, &cfg_src_addr);
50004080634SCarlos Llamas
50104080634SCarlos Llamas cfg_num_pkt = parse_io(argv[optind], cfg_buf);
502af5136f9SWillem de Bruijn }
503af5136f9SWillem de Bruijn
main(int argc,char ** argv)504af5136f9SWillem de Bruijn int main(int argc, char **argv)
505af5136f9SWillem de Bruijn {
506af5136f9SWillem de Bruijn parse_opts(argc, argv);
507af5136f9SWillem de Bruijn
50804080634SCarlos Llamas if (cfg_rx)
50904080634SCarlos Llamas do_test_rx((void *)&cfg_dst_addr, cfg_alen);
51004080634SCarlos Llamas else
51104080634SCarlos Llamas do_test_tx((void *)&cfg_src_addr, cfg_alen);
512af5136f9SWillem de Bruijn
513af5136f9SWillem de Bruijn return 0;
514af5136f9SWillem de Bruijn }
515