1 // SPDX-License-Identifier: GPL-2.0+
3 * Author: Justin Iurman (justin.iurman@uliege.be)
5 * IOAM tester for IPv6, see ioam6.sh for details on each test case.
10 #include <linux/const.h>
11 #include <linux/ioam6.h>
12 #include <linux/ipv6.h>
32 * Be careful if you modify structs below - everything MUST be kept synchronized
33 * with configurations inside ioam6.sh and always reflect the same.
36 static struct ioam_config node1 = {
39 .ingr_id = 0xffff, /* default value */
41 .ingr_wide = 0xffffffff, /* default value */
43 .ns_data = 0xdeadbee0,
44 .ns_wide = 0xcafec0caf00dc0de,
46 .sc_data = "something that will be 4n-aligned",
50 static struct ioam_config node2 = {
57 .ns_data = 0xdeadbee1,
58 .ns_wide = 0xcafec0caf11dc0de,
60 .sc_data = "Hello there -Obi",
64 static struct ioam_config node3 = {
68 .egr_id = 0xffff, /* default value */
70 .egr_wide = 0xffffffff, /* default value */
71 .ns_data = 0xdeadbee2,
72 .ns_wide = 0xcafec0caf22dc0de,
73 .sc_id = 0xffffff, /* default value */
97 TEST_OUT_FULL_SUPP_TRACE,
118 TEST_IN_FULL_SUPP_TRACE,
123 TEST_FWD_FULL_SUPP_TRACE,
128 static int check_ioam_header(int tid, struct ioam6_trace_hdr *ioam6h,
129 __u32 trace_type, __u16 ioam_ns)
131 if (__be16_to_cpu(ioam6h->namespace_id) != ioam_ns ||
132 __be32_to_cpu(ioam6h->type_be32) != (trace_type << 8))
136 case TEST_OUT_UNDEF_NS:
137 case TEST_IN_UNDEF_NS:
138 return ioam6h->overflow ||
139 ioam6h->nodelen != 1 ||
142 case TEST_OUT_NO_ROOM:
143 case TEST_IN_NO_ROOM:
145 return !ioam6h->overflow ||
146 ioam6h->nodelen != 2 ||
167 return ioam6h->overflow ||
168 ioam6h->nodelen != 1 ||
177 return ioam6h->overflow ||
178 ioam6h->nodelen != 2 ||
183 return ioam6h->overflow ||
187 case TEST_OUT_FULL_SUPP_TRACE:
188 case TEST_IN_FULL_SUPP_TRACE:
189 case TEST_FWD_FULL_SUPP_TRACE:
190 return ioam6h->overflow ||
191 ioam6h->nodelen != 15 ||
201 static int check_ioam6_data(__u8 **p, struct ioam6_trace_hdr *ioam6h,
202 const struct ioam_config cnf)
209 if (ioam6h->type.bit0) {
210 raw32 = __be32_to_cpu(*((__u32 *)*p));
211 if (cnf.hlim != (raw32 >> 24) || cnf.id != (raw32 & 0xffffff))
216 if (ioam6h->type.bit1) {
217 raw32 = __be32_to_cpu(*((__u32 *)*p));
218 if (cnf.ingr_id != (raw32 >> 16) ||
219 cnf.egr_id != (raw32 & 0xffff))
224 if (ioam6h->type.bit2)
227 if (ioam6h->type.bit3)
230 if (ioam6h->type.bit4) {
231 if (__be32_to_cpu(*((__u32 *)*p)) != 0xffffffff)
236 if (ioam6h->type.bit5) {
237 if (__be32_to_cpu(*((__u32 *)*p)) != cnf.ns_data)
242 if (ioam6h->type.bit6)
245 if (ioam6h->type.bit7) {
246 if (__be32_to_cpu(*((__u32 *)*p)) != 0xffffffff)
251 if (ioam6h->type.bit8) {
252 raw64 = __be64_to_cpu(*((__u64 *)*p));
253 if (cnf.hlim != (raw64 >> 56) ||
254 cnf.wide != (raw64 & 0xffffffffffffff))
259 if (ioam6h->type.bit9) {
260 if (__be32_to_cpu(*((__u32 *)*p)) != cnf.ingr_wide)
264 if (__be32_to_cpu(*((__u32 *)*p)) != cnf.egr_wide)
269 if (ioam6h->type.bit10) {
270 if (__be64_to_cpu(*((__u64 *)*p)) != cnf.ns_wide)
275 if (ioam6h->type.bit11) {
276 if (__be32_to_cpu(*((__u32 *)*p)) != 0xffffffff)
281 if (ioam6h->type.bit12) {
282 if (__be32_to_cpu(*((__u32 *)*p)) != 0xffffffff)
287 if (ioam6h->type.bit13) {
288 if (__be32_to_cpu(*((__u32 *)*p)) != 0xffffffff)
293 if (ioam6h->type.bit14) {
294 if (__be32_to_cpu(*((__u32 *)*p)) != 0xffffffff)
299 if (ioam6h->type.bit15) {
300 if (__be32_to_cpu(*((__u32 *)*p)) != 0xffffffff)
305 if (ioam6h->type.bit16) {
306 if (__be32_to_cpu(*((__u32 *)*p)) != 0xffffffff)
311 if (ioam6h->type.bit17) {
312 if (__be32_to_cpu(*((__u32 *)*p)) != 0xffffffff)
317 if (ioam6h->type.bit18) {
318 if (__be32_to_cpu(*((__u32 *)*p)) != 0xffffffff)
323 if (ioam6h->type.bit19) {
324 if (__be32_to_cpu(*((__u32 *)*p)) != 0xffffffff)
329 if (ioam6h->type.bit20) {
330 if (__be32_to_cpu(*((__u32 *)*p)) != 0xffffffff)
335 if (ioam6h->type.bit21) {
336 if (__be32_to_cpu(*((__u32 *)*p)) != 0xffffffff)
341 if (ioam6h->type.bit22) {
342 len = cnf.sc_data ? strlen(cnf.sc_data) : 0;
343 aligned = cnf.sc_data ? __ALIGN_KERNEL(len, 4) : 0;
345 raw32 = __be32_to_cpu(*((__u32 *)*p));
346 if (aligned != (raw32 >> 24) * 4 ||
347 cnf.sc_id != (raw32 & 0xffffff))
352 if (strncmp((char *)*p, cnf.sc_data, len))
369 static int check_ioam_header_and_data(int tid, struct ioam6_trace_hdr *ioam6h,
370 __u32 trace_type, __u16 ioam_ns)
374 if (check_ioam_header(tid, ioam6h, trace_type, ioam_ns))
377 p = ioam6h->data + ioam6h->remlen * 4;
393 case TEST_OUT_FULL_SUPP_TRACE:
394 return check_ioam6_data(&p, ioam6h, node1);
409 case TEST_IN_FULL_SUPP_TRACE:
411 __u32 tmp32 = node2.egr_wide;
412 __u16 tmp16 = node2.egr_id;
415 node2.egr_id = 0xffff;
416 node2.egr_wide = 0xffffffff;
418 res = check_ioam6_data(&p, ioam6h, node2);
420 node2.egr_id = tmp16;
421 node2.egr_wide = tmp32;
426 case TEST_FWD_FULL_SUPP_TRACE:
427 if (check_ioam6_data(&p, ioam6h, node3))
429 if (check_ioam6_data(&p, ioam6h, node2))
431 return check_ioam6_data(&p, ioam6h, node1);
440 static int str2id(const char *tname)
442 if (!strcmp("out_undef_ns", tname))
443 return TEST_OUT_UNDEF_NS;
444 if (!strcmp("out_no_room", tname))
445 return TEST_OUT_NO_ROOM;
446 if (!strcmp("out_bit0", tname))
447 return TEST_OUT_BIT0;
448 if (!strcmp("out_bit1", tname))
449 return TEST_OUT_BIT1;
450 if (!strcmp("out_bit2", tname))
451 return TEST_OUT_BIT2;
452 if (!strcmp("out_bit3", tname))
453 return TEST_OUT_BIT3;
454 if (!strcmp("out_bit4", tname))
455 return TEST_OUT_BIT4;
456 if (!strcmp("out_bit5", tname))
457 return TEST_OUT_BIT5;
458 if (!strcmp("out_bit6", tname))
459 return TEST_OUT_BIT6;
460 if (!strcmp("out_bit7", tname))
461 return TEST_OUT_BIT7;
462 if (!strcmp("out_bit8", tname))
463 return TEST_OUT_BIT8;
464 if (!strcmp("out_bit9", tname))
465 return TEST_OUT_BIT9;
466 if (!strcmp("out_bit10", tname))
467 return TEST_OUT_BIT10;
468 if (!strcmp("out_bit11", tname))
469 return TEST_OUT_BIT11;
470 if (!strcmp("out_bit22", tname))
471 return TEST_OUT_BIT22;
472 if (!strcmp("out_full_supp_trace", tname))
473 return TEST_OUT_FULL_SUPP_TRACE;
474 if (!strcmp("in_undef_ns", tname))
475 return TEST_IN_UNDEF_NS;
476 if (!strcmp("in_no_room", tname))
477 return TEST_IN_NO_ROOM;
478 if (!strcmp("in_oflag", tname))
479 return TEST_IN_OFLAG;
480 if (!strcmp("in_bit0", tname))
482 if (!strcmp("in_bit1", tname))
484 if (!strcmp("in_bit2", tname))
486 if (!strcmp("in_bit3", tname))
488 if (!strcmp("in_bit4", tname))
490 if (!strcmp("in_bit5", tname))
492 if (!strcmp("in_bit6", tname))
494 if (!strcmp("in_bit7", tname))
496 if (!strcmp("in_bit8", tname))
498 if (!strcmp("in_bit9", tname))
500 if (!strcmp("in_bit10", tname))
501 return TEST_IN_BIT10;
502 if (!strcmp("in_bit11", tname))
503 return TEST_IN_BIT11;
504 if (!strcmp("in_bit22", tname))
505 return TEST_IN_BIT22;
506 if (!strcmp("in_full_supp_trace", tname))
507 return TEST_IN_FULL_SUPP_TRACE;
508 if (!strcmp("fwd_full_supp_trace", tname))
509 return TEST_FWD_FULL_SUPP_TRACE;
514 static int get_u32(__u32 *val, const char *arg, int base)
521 res = strtoul(arg, &ptr, base);
523 if (!ptr || ptr == arg || *ptr)
526 if (res == ULONG_MAX && errno == ERANGE)
529 if (res > 0xFFFFFFFFUL)
536 static int get_u16(__u16 *val, const char *arg, int base)
543 res = strtoul(arg, &ptr, base);
545 if (!ptr || ptr == arg || *ptr)
548 if (res == ULONG_MAX && errno == ERANGE)
558 static int (*func[__TEST_MAX])(int, struct ioam6_trace_hdr *, __u32, __u16) = {
559 [TEST_OUT_UNDEF_NS] = check_ioam_header,
560 [TEST_OUT_NO_ROOM] = check_ioam_header,
561 [TEST_OUT_BIT0] = check_ioam_header_and_data,
562 [TEST_OUT_BIT1] = check_ioam_header_and_data,
563 [TEST_OUT_BIT2] = check_ioam_header_and_data,
564 [TEST_OUT_BIT3] = check_ioam_header_and_data,
565 [TEST_OUT_BIT4] = check_ioam_header_and_data,
566 [TEST_OUT_BIT5] = check_ioam_header_and_data,
567 [TEST_OUT_BIT6] = check_ioam_header_and_data,
568 [TEST_OUT_BIT7] = check_ioam_header_and_data,
569 [TEST_OUT_BIT8] = check_ioam_header_and_data,
570 [TEST_OUT_BIT9] = check_ioam_header_and_data,
571 [TEST_OUT_BIT10] = check_ioam_header_and_data,
572 [TEST_OUT_BIT11] = check_ioam_header_and_data,
573 [TEST_OUT_BIT22] = check_ioam_header_and_data,
574 [TEST_OUT_FULL_SUPP_TRACE] = check_ioam_header_and_data,
575 [TEST_IN_UNDEF_NS] = check_ioam_header,
576 [TEST_IN_NO_ROOM] = check_ioam_header,
577 [TEST_IN_OFLAG] = check_ioam_header,
578 [TEST_IN_BIT0] = check_ioam_header_and_data,
579 [TEST_IN_BIT1] = check_ioam_header_and_data,
580 [TEST_IN_BIT2] = check_ioam_header_and_data,
581 [TEST_IN_BIT3] = check_ioam_header_and_data,
582 [TEST_IN_BIT4] = check_ioam_header_and_data,
583 [TEST_IN_BIT5] = check_ioam_header_and_data,
584 [TEST_IN_BIT6] = check_ioam_header_and_data,
585 [TEST_IN_BIT7] = check_ioam_header_and_data,
586 [TEST_IN_BIT8] = check_ioam_header_and_data,
587 [TEST_IN_BIT9] = check_ioam_header_and_data,
588 [TEST_IN_BIT10] = check_ioam_header_and_data,
589 [TEST_IN_BIT11] = check_ioam_header_and_data,
590 [TEST_IN_BIT22] = check_ioam_header_and_data,
591 [TEST_IN_FULL_SUPP_TRACE] = check_ioam_header_and_data,
592 [TEST_FWD_FULL_SUPP_TRACE] = check_ioam_header_and_data,
595 int main(int argc, char **argv)
597 int fd, size, hoplen, tid, ret = 1, on = 1;
598 struct ioam6_hdr *opt;
599 struct cmsghdr *cmsg;
610 tid = str2id(argv[1]);
611 if (tid < 0 || !func[tid])
614 if (get_u32(&tr_type, argv[2], 16) ||
615 get_u16(&ioam_ns, argv[3], 0))
618 fd = socket(PF_INET6, SOCK_RAW,
619 !strcmp(argv[4], "encap") ? IPPROTO_IPV6 : IPPROTO_ICMPV6);
623 setsockopt(fd, IPPROTO_IPV6, IPV6_RECVHOPOPTS, &on, sizeof(on));
626 iov.iov_base = malloc(CMSG_SPACE(sizeof(buffer)));
630 memset(&msg, 0, sizeof(msg));
633 msg.msg_control = buffer;
634 msg.msg_controllen = CMSG_SPACE(sizeof(buffer));
636 size = recvmsg(fd, &msg, 0);
640 for (cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg)) {
641 if (cmsg->cmsg_level != IPPROTO_IPV6 ||
642 cmsg->cmsg_type != IPV6_HOPOPTS ||
643 cmsg->cmsg_len < sizeof(struct ipv6_hopopt_hdr))
646 ptr = (__u8 *)CMSG_DATA(cmsg);
648 hoplen = (ptr[1] + 1) << 3;
649 ptr += sizeof(struct ipv6_hopopt_hdr);
652 opt = (struct ioam6_hdr *)ptr;
654 if (opt->opt_type == IPV6_TLV_IOAM &&
655 opt->type == IOAM6_TYPE_PREALLOC) {
658 (struct ioam6_trace_hdr *)ptr,
663 ptr += opt->opt_len + 2;
664 hoplen -= opt->opt_len + 2;