selftests/net: ipv6 flowlabel (3fb321fd) · Commits · 戴 / test

tools/testing/selftests/net/.gitignore

+2 −0

Original line number	Diff line number	Diff line
		@@ -18,3 +18,5 @@ tls
		txring_overwrite
		ip_defrag
		so_txtime
		flowlabel
		flowlabel_mgr

tools/testing/selftests/net/Makefile

+2 −2

Original line number	Diff line number	Diff line
		@@ -9,13 +9,13 @@ TEST_PROGS := run_netsocktests run_afpackettests test_bpf.sh netdevice.sh \
		TEST_PROGS += fib_tests.sh fib-onlink-tests.sh pmtu.sh udpgso.sh ip_defrag.sh
		TEST_PROGS += udpgso_bench.sh fib_rule_tests.sh msg_zerocopy.sh psock_snd.sh
		TEST_PROGS += udpgro_bench.sh udpgro.sh test_vxlan_under_vrf.sh reuseport_addr_any.sh
		TEST_PROGS += test_vxlan_fdb_changelink.sh so_txtime.sh
		TEST_PROGS += test_vxlan_fdb_changelink.sh so_txtime.sh ipv6_flowlabel.sh
		TEST_PROGS_EXTENDED := in_netns.sh
		TEST_GEN_FILES = socket
		TEST_GEN_FILES += psock_fanout psock_tpacket msg_zerocopy reuseport_addr_any
		TEST_GEN_FILES += tcp_mmap tcp_inq psock_snd txring_overwrite
		TEST_GEN_FILES += udpgso udpgso_bench_tx udpgso_bench_rx ip_defrag
		TEST_GEN_FILES += so_txtime
		TEST_GEN_FILES += so_txtime ipv6_flowlabel ipv6_flowlabel_mgr
		TEST_GEN_PROGS = reuseport_bpf reuseport_bpf_cpu reuseport_bpf_numa
		TEST_GEN_PROGS += reuseport_dualstack reuseaddr_conflict tls

tools/testing/selftests/net/ipv6_flowlabel.c

0 → 100644

+229 −0

Original line number	Diff line number	Diff line
		// SPDX-License-Identifier: GPL-2.0
		/* Test IPV6_FLOWINFO cmsg on send and recv */

		#define _GNU_SOURCE

		#include <arpa/inet.h>
		#include <asm/byteorder.h>
		#include <error.h>
		#include <errno.h>
		#include <fcntl.h>
		#include <limits.h>
		#include <linux/in6.h>
		#include <stdbool.h>
		#include <stdio.h>
		#include <stdint.h>
		#include <stdlib.h>
		#include <string.h>
		#include <sys/socket.h>
		#include <sys/stat.h>
		#include <sys/time.h>
		#include <sys/types.h>
		#include <unistd.h>

		/* uapi/glibc weirdness may leave this undefined */
		#ifndef IPV6_FLOWINFO
		#define IPV6_FLOWINFO 11
		#endif

		#ifndef IPV6_FLOWLABEL_MGR
		#define IPV6_FLOWLABEL_MGR 32
		#endif

		#define FLOWLABEL_WILDCARD ((uint32_t) -1)

		static const char cfg_data[] = "a";
		static uint32_t cfg_label = 1;

		static void do_send(int fd, bool with_flowlabel, uint32_t flowlabel)
		{
		char control[CMSG_SPACE(sizeof(flowlabel))] = {0};
		struct msghdr msg = {0};
		struct iovec iov = {0};
		int ret;

		iov.iov_base = (char *)cfg_data;
		iov.iov_len = sizeof(cfg_data);

		msg.msg_iov = &iov;
		msg.msg_iovlen = 1;

		if (with_flowlabel) {
		struct cmsghdr *cm;

		cm = (void *)control;
		cm->cmsg_len = CMSG_LEN(sizeof(flowlabel));
		cm->cmsg_level = SOL_IPV6;
		cm->cmsg_type = IPV6_FLOWINFO;
		(uint32_t )CMSG_DATA(cm) = htonl(flowlabel);

		msg.msg_control = control;
		msg.msg_controllen = sizeof(control);
		}

		ret = sendmsg(fd, &msg, 0);
		if (ret == -1)
		error(1, errno, "send");

		if (with_flowlabel)
		fprintf(stderr, "sent with label %u\n", flowlabel);
		else
		fprintf(stderr, "sent without label\n");
		}

		static void do_recv(int fd, bool with_flowlabel, uint32_t expect)
		{
		char control[CMSG_SPACE(sizeof(expect))];
		char data[sizeof(cfg_data)];
		struct msghdr msg = {0};
		struct iovec iov = {0};
		struct cmsghdr *cm;
		uint32_t flowlabel;
		int ret;

		iov.iov_base = data;
		iov.iov_len = sizeof(data);

		msg.msg_iov = &iov;
		msg.msg_iovlen = 1;

		memset(control, 0, sizeof(control));
		msg.msg_control = control;
		msg.msg_controllen = sizeof(control);

		ret = recvmsg(fd, &msg, 0);
		if (ret == -1)
		error(1, errno, "recv");
		if (msg.msg_flags & (MSG_TRUNC \| MSG_CTRUNC))
		error(1, 0, "recv: truncated");
		if (ret != sizeof(cfg_data))
		error(1, 0, "recv: length mismatch");
		if (memcmp(data, cfg_data, sizeof(data)))
		error(1, 0, "recv: data mismatch");

		cm = CMSG_FIRSTHDR(&msg);
		if (with_flowlabel) {
		if (!cm)
		error(1, 0, "recv: missing cmsg");
		if (CMSG_NXTHDR(&msg, cm))
		error(1, 0, "recv: too many cmsg");
		if (cm->cmsg_level != SOL_IPV6 \|\|
		cm->cmsg_type != IPV6_FLOWINFO)
		error(1, 0, "recv: unexpected cmsg level or type");

		flowlabel = ntohl((uint32_t )CMSG_DATA(cm));
		fprintf(stderr, "recv with label %u\n", flowlabel);

		if (expect != FLOWLABEL_WILDCARD && expect != flowlabel)
		fprintf(stderr, "recv: incorrect flowlabel %u != %u\n",
		flowlabel, expect);

		} else {
		fprintf(stderr, "recv without label\n");
		}
		}

		static bool get_autoflowlabel_enabled(void)
		{
		int fd, ret;
		char val;

		fd = open("/proc/sys/net/ipv6/auto_flowlabels", O_RDONLY);
		if (fd == -1)
		error(1, errno, "open sysctl");

		ret = read(fd, &val, 1);
		if (ret == -1)
		error(1, errno, "read sysctl");
		if (ret == 0)
		error(1, 0, "read sysctl: 0");

		if (close(fd))
		error(1, errno, "close sysctl");

		return val == '1';
		}

		static void flowlabel_get(int fd, uint32_t label, uint8_t share, uint16_t flags)
		{
		struct in6_flowlabel_req req = {
		.flr_action = IPV6_FL_A_GET,
		.flr_label = htonl(label),
		.flr_flags = flags,
		.flr_share = share,
		};

		/* do not pass IPV6_ADDR_ANY or IPV6_ADDR_MAPPED */
		req.flr_dst.s6_addr[0] = 0xfd;
		req.flr_dst.s6_addr[15] = 0x1;

		if (setsockopt(fd, SOL_IPV6, IPV6_FLOWLABEL_MGR, &req, sizeof(req)))
		error(1, errno, "setsockopt flowlabel get");
		}

		static void parse_opts(int argc, char **argv)
		{
		int c;

		while ((c = getopt(argc, argv, "l:")) != -1) {
		switch (c) {
		case 'l':
		cfg_label = strtoul(optarg, NULL, 0);
		break;
		default:
		error(1, 0, "%s: parse error", argv[0]);
		}
		}
		}

		int main(int argc, char **argv)
		{
		struct sockaddr_in6 addr = {
		.sin6_family = AF_INET6,
		.sin6_port = htons(8000),
		.sin6_addr = IN6ADDR_LOOPBACK_INIT,
		};
		const int one = 1;
		int fdt, fdr;

		parse_opts(argc, argv);

		fdt = socket(PF_INET6, SOCK_DGRAM, 0);
		if (fdt == -1)
		error(1, errno, "socket t");

		fdr = socket(PF_INET6, SOCK_DGRAM, 0);
		if (fdr == -1)
		error(1, errno, "socket r");

		if (connect(fdt, (void *)&addr, sizeof(addr)))
		error(1, errno, "connect");
		if (bind(fdr, (void *)&addr, sizeof(addr)))
		error(1, errno, "bind");

		flowlabel_get(fdt, cfg_label, IPV6_FL_S_EXCL, IPV6_FL_F_CREATE);

		if (setsockopt(fdr, SOL_IPV6, IPV6_FLOWINFO, &one, sizeof(one)))
		error(1, errno, "setsockopt flowinfo");

		if (get_autoflowlabel_enabled()) {
		fprintf(stderr, "send no label: recv auto flowlabel\n");
		do_send(fdt, false, 0);
		do_recv(fdr, true, FLOWLABEL_WILDCARD);
		} else {
		fprintf(stderr, "send no label: recv no label (auto off)\n");
		do_send(fdt, false, 0);
		do_recv(fdr, false, 0);
		}

		fprintf(stderr, "send label\n");
		do_send(fdt, true, cfg_label);
		do_recv(fdr, true, cfg_label);

		if (close(fdr))
		error(1, errno, "close r");
		if (close(fdt))
		error(1, errno, "close t");

		return 0;
		}

tools/testing/selftests/net/ipv6_flowlabel.sh

0 → 100755

+21 −0

Original line number	Diff line number	Diff line
		#!/bin/sh
		# SPDX-License-Identifier: GPL-2.0
		#
		# Regression tests for IPv6 flowlabels
		#
		# run in separate namespaces to avoid mgmt db conflicts betweent tests

		set -e

		echo "TEST management"
		./in_netns.sh ./ipv6_flowlabel_mgr

		echo "TEST datapath"
		./in_netns.sh \
		sh -c 'sysctl -q -w net.ipv6.auto_flowlabels=0 && ./ipv6_flowlabel -l 1'

		echo "TEST datapath (with auto-flowlabels)"
		./in_netns.sh \
		sh -c 'sysctl -q -w net.ipv6.auto_flowlabels=1 && ./ipv6_flowlabel -l 1'

		echo OK. All tests passed

tools/testing/selftests/net/ipv6_flowlabel_mgr.c

0 → 100644

+199 −0

Original line number	Diff line number	Diff line
		// SPDX-License-Identifier: GPL-2.0
		/* Test IPV6_FLOWINFO_MGR */

		#define _GNU_SOURCE

		#include <arpa/inet.h>
		#include <error.h>
		#include <errno.h>
		#include <limits.h>
		#include <linux/in6.h>
		#include <stdbool.h>
		#include <stdio.h>
		#include <stdint.h>
		#include <stdlib.h>
		#include <string.h>
		#include <sys/socket.h>
		#include <sys/stat.h>
		#include <sys/time.h>
		#include <sys/types.h>
		#include <sys/wait.h>
		#include <unistd.h>

		/* uapi/glibc weirdness may leave this undefined */
		#ifndef IPV6_FLOWLABEL_MGR
		#define IPV6_FLOWLABEL_MGR 32
		#endif

		/* from net/ipv6/ip6_flowlabel.c */
		#define FL_MIN_LINGER 6

		#define explain(x) \
		do { if (cfg_verbose) fprintf(stderr, " " x "\n"); } while (0)

		#define __expect(x) \
		do { \
		if (!(x)) \
		fprintf(stderr, "[OK] " #x "\n"); \
		else \
		error(1, 0, "[ERR] " #x " (line %d)", __LINE__); \
		} while (0)

		#define expect_pass(x) __expect(x)
		#define expect_fail(x) __expect(!(x))

		static bool cfg_long_running;
		static bool cfg_verbose;

		static int flowlabel_get(int fd, uint32_t label, uint8_t share, uint16_t flags)
		{
		struct in6_flowlabel_req req = {
		.flr_action = IPV6_FL_A_GET,
		.flr_label = htonl(label),
		.flr_flags = flags,
		.flr_share = share,
		};

		/* do not pass IPV6_ADDR_ANY or IPV6_ADDR_MAPPED */
		req.flr_dst.s6_addr[0] = 0xfd;
		req.flr_dst.s6_addr[15] = 0x1;

		return setsockopt(fd, SOL_IPV6, IPV6_FLOWLABEL_MGR, &req, sizeof(req));
		}

		static int flowlabel_put(int fd, uint32_t label)
		{
		struct in6_flowlabel_req req = {
		.flr_action = IPV6_FL_A_PUT,
		.flr_label = htonl(label),
		};

		return setsockopt(fd, SOL_IPV6, IPV6_FLOWLABEL_MGR, &req, sizeof(req));
		}

		static void run_tests(int fd)
		{
		int wstatus;
		pid_t pid;

		explain("cannot get non-existent label");
		expect_fail(flowlabel_get(fd, 1, IPV6_FL_S_ANY, 0));

		explain("cannot put non-existent label");
		expect_fail(flowlabel_put(fd, 1));

		explain("cannot create label greater than 20 bits");
		expect_fail(flowlabel_get(fd, 0x1FFFFF, IPV6_FL_S_ANY,
		IPV6_FL_F_CREATE));

		explain("create a new label (FL_F_CREATE)");
		expect_pass(flowlabel_get(fd, 1, IPV6_FL_S_ANY, IPV6_FL_F_CREATE));
		explain("can get the label (without FL_F_CREATE)");
		expect_pass(flowlabel_get(fd, 1, IPV6_FL_S_ANY, 0));
		explain("can get it again with create flag set, too");
		expect_pass(flowlabel_get(fd, 1, IPV6_FL_S_ANY, IPV6_FL_F_CREATE));
		explain("cannot get it again with the exclusive (FL_FL_EXCL) flag");
		expect_fail(flowlabel_get(fd, 1, IPV6_FL_S_ANY,
		IPV6_FL_F_CREATE \| IPV6_FL_F_EXCL));
		explain("can now put exactly three references");
		expect_pass(flowlabel_put(fd, 1));
		expect_pass(flowlabel_put(fd, 1));
		expect_pass(flowlabel_put(fd, 1));
		expect_fail(flowlabel_put(fd, 1));

		explain("create a new exclusive label (FL_S_EXCL)");
		expect_pass(flowlabel_get(fd, 2, IPV6_FL_S_EXCL, IPV6_FL_F_CREATE));
		explain("cannot get it again in non-exclusive mode");
		expect_fail(flowlabel_get(fd, 2, IPV6_FL_S_ANY, IPV6_FL_F_CREATE));
		explain("cannot get it again in exclusive mode either");
		expect_fail(flowlabel_get(fd, 2, IPV6_FL_S_EXCL, IPV6_FL_F_CREATE));
		expect_pass(flowlabel_put(fd, 2));

		if (cfg_long_running) {
		explain("cannot reuse the label, due to linger");
		expect_fail(flowlabel_get(fd, 2, IPV6_FL_S_ANY,
		IPV6_FL_F_CREATE));
		explain("after sleep, can reuse");
		sleep(FL_MIN_LINGER * 2 + 1);
		expect_pass(flowlabel_get(fd, 2, IPV6_FL_S_ANY,
		IPV6_FL_F_CREATE));
		}

		explain("create a new user-private label (FL_S_USER)");
		expect_pass(flowlabel_get(fd, 3, IPV6_FL_S_USER, IPV6_FL_F_CREATE));
		explain("cannot get it again in non-exclusive mode");
		expect_fail(flowlabel_get(fd, 3, IPV6_FL_S_ANY, 0));
		explain("cannot get it again in exclusive mode");
		expect_fail(flowlabel_get(fd, 3, IPV6_FL_S_EXCL, 0));
		explain("can get it again in user mode");
		expect_pass(flowlabel_get(fd, 3, IPV6_FL_S_USER, 0));
		explain("child process can get it too, but not after setuid(nobody)");
		pid = fork();
		if (pid == -1)
		error(1, errno, "fork");
		if (!pid) {
		expect_pass(flowlabel_get(fd, 3, IPV6_FL_S_USER, 0));
		if (setuid(USHRT_MAX))
		fprintf(stderr, "[INFO] skip setuid child test\n");
		else
		expect_fail(flowlabel_get(fd, 3, IPV6_FL_S_USER, 0));
		exit(0);
		}
		if (wait(&wstatus) == -1)
		error(1, errno, "wait");
		if (!WIFEXITED(wstatus) \|\| WEXITSTATUS(wstatus) != 0)
		error(1, errno, "wait: unexpected child result");

		explain("create a new process-private label (FL_S_PROCESS)");
		expect_pass(flowlabel_get(fd, 4, IPV6_FL_S_PROCESS, IPV6_FL_F_CREATE));
		explain("can get it again");
		expect_pass(flowlabel_get(fd, 4, IPV6_FL_S_PROCESS, 0));
		explain("child process cannot can get it");
		pid = fork();
		if (pid == -1)
		error(1, errno, "fork");
		if (!pid) {
		expect_fail(flowlabel_get(fd, 4, IPV6_FL_S_PROCESS, 0));
		exit(0);
		}
		if (wait(&wstatus) == -1)
		error(1, errno, "wait");
		if (!WIFEXITED(wstatus) \|\| WEXITSTATUS(wstatus) != 0)
		error(1, errno, "wait: unexpected child result");
		}

		static void parse_opts(int argc, char **argv)
		{
		int c;

		while ((c = getopt(argc, argv, "lv")) != -1) {
		switch (c) {
		case 'l':
		cfg_long_running = true;
		break;
		case 'v':
		cfg_verbose = true;
		break;
		default:
		error(1, 0, "%s: parse error", argv[0]);
		}
		}
		}

		int main(int argc, char **argv)
		{
		int fd;

		parse_opts(argc, argv);

		fd = socket(PF_INET6, SOCK_DGRAM, 0);
		if (fd == -1)
		error(1, errno, "socket");

		run_tests(fd);

		if (close(fd))
		error(1, errno, "close");

		return 0;
		}

Admin message