From: Masatake YAMATO <yamato@redhat.com>
Date: Sun, 26 Feb 2023 21:23:13 +0000 (+0900)
Subject: tests: (mkfds) add netlink factory
X-Git-Tag: v2.39-rc1~41^2~2
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=4763f06709c162ea9f171b8b968c2ea5a1633a4e;p=thirdparty%2Futil-linux.git

tests: (mkfds) add netlink factory

Signed-off-by: Masatake YAMATO <yamato@redhat.com>
---

diff --git a/tests/helpers/test_mkfds.c b/tests/helpers/test_mkfds.c
index bfbc275fbb..601cd3dcc8 100644
--- a/tests/helpers/test_mkfds.c
+++ b/tests/helpers/test_mkfds.c
@@ -25,6 +25,7 @@
 #include <getopt.h>
 #include <linux/if_ether.h>
 #include <linux/if_packet.h>
+#include <linux/netlink.h>
 #include <linux/sockios.h>  /* SIOCGSKNS */
 #include <net/if.h>
 #include <netinet/in.h>
@@ -241,6 +242,12 @@ struct ptype_class ptype_classes [] = {
 		.read   = integer_read,
 		.free   = integer_free,
 	},
+	[PTYPE_UINTEGER] = {
+		.name = "uinteger",
+		.sprint = uinteger_sprint,
+		.read   = uinteger_read,
+		.free   = uinteger_free,
+	},
 	[PTYPE_BOOLEAN] = {
 		.name = "boolean",
 		.sprint = boolean_sprint,
@@ -1838,6 +1845,52 @@ static void *make_netns(const struct factory *factory _U_, struct fdesc fdescs[]
 	return NULL;
 }
 
+static void *make_netlink(const struct factory *factory, struct fdesc fdescs[],
+			  int argc, char ** argv)
+{
+	struct arg protocol = decode_arg("protocol", factory->params, argc, argv);
+	int iprotocol = ARG_INTEGER(protocol);
+	struct arg groups = decode_arg("groups", factory->params, argc, argv);
+	unsigned int ugroups = ARG_UINTEGER(groups);
+	int sd;
+
+	free_arg(&protocol);
+
+	sd = socket(AF_NETLINK, SOCK_RAW, iprotocol);
+	if (sd < 0)
+		err((errno == EPROTONOSUPPORT)? EXIT_EPROTONOSUPPORT: EXIT_FAILURE,
+		    "failed in socket()");
+
+	if (sd != fdescs[0].fd) {
+		if (dup2(sd, fdescs[0].fd) < 0) {
+			int e = errno;
+			close(sd);
+			errno = e;
+			err(EXIT_FAILURE, "failed to dup %d -> %d", sd, fdescs[0].fd);
+		}
+		close(sd);
+	}
+
+	struct sockaddr_nl nl;
+	memset(&nl, 0, sizeof(nl));
+	nl.nl_family = AF_NETLINK;
+	nl.nl_groups = ugroups;
+	if (bind(sd, (struct sockaddr*)&nl, sizeof(nl)) < 0) {
+		int e = errno;
+		close(sd);
+		errno = e;
+		err(EXIT_FAILURE, "failed in bind(2)");
+	}
+
+	fdescs[0] = (struct fdesc){
+		.fd    = fdescs[0].fd,
+		.close = close_fdesc,
+		.data  = NULL
+	};
+
+	return NULL;
+}
+
 #define PARAM_END { .name = NULL, }
 static const struct factory factories[] = {
 	{
@@ -2378,6 +2431,29 @@ static const struct factory factories[] = {
 			PARAM_END
 		}
 	},
+	{
+		.name = "netlink",
+		.desc = "AF_NETLINK sockets",
+		.priv = false,
+		.N    = 1,
+		.EX_N = 0,
+		.make = make_netlink,
+		.params = (struct parameter []) {
+			{
+				.name = "protocol",
+				.type = PTYPE_INTEGER,
+				.desc = "protocol passed to socket(AF_NETLINK, SOCK_RAW, protocol)",
+				.defv.integer = NETLINK_USERSOCK,
+			},
+			{
+				.name = "groups",
+				.type = PTYPE_UINTEGER,
+				.desc = "multicast groups of netlink communication (requires CAP_NET_ADMIN)",
+				.defv.uinteger = 0,
+			},
+			PARAM_END
+		}
+	},
 };
 
 static int count_parameters(const struct factory *factory)