[PATCH v1] tests: add check for netlink protocol with -yy option
Fabien Siron
fabien.siron at epita.fr
Wed May 18 16:45:50 UTC 2016
* tests/netlink_netlink_diag.c: New file.
* tests/net-yy-netlink.c: New file.
* tests/net-yy-netlink.test: New test.
* tests/.gitignore: Add net-yy-netlink and netlink_netlink_diag
* tests/Makefile.am (check_PROGRAMS): Likewise.
(TESTS): Add net-yy-netlink.test.
---
tests/.gitignore | 2 +
tests/Makefile.am | 3 +
tests/net-yy-netlink.c | 81 +++++++++++++++++++++++++
tests/net-yy-netlink.test | 47 +++++++++++++++
tests/netlink_netlink_diag.c | 138 +++++++++++++++++++++++++++++++++++++++++++
5 files changed, 271 insertions(+)
create mode 100644 tests/net-yy-netlink.c
create mode 100755 tests/net-yy-netlink.test
create mode 100644 tests/netlink_netlink_diag.c
diff --git a/tests/.gitignore b/tests/.gitignore
index 05cbd43..4890f36 100644
--- a/tests/.gitignore
+++ b/tests/.gitignore
@@ -132,8 +132,10 @@ net-accept-connect
net-icmp_filter
net-y-unix
net-yy-inet
+net-yy-netlink
net-yy-unix
netlink_inet_diag
+netlink_netlink_diag
netlink_unix_diag
newfstatat
nsyscalls
diff --git a/tests/Makefile.am b/tests/Makefile.am
index a664275..ca4ed9d 100644
--- a/tests/Makefile.am
+++ b/tests/Makefile.am
@@ -185,8 +185,10 @@ check_PROGRAMS = \
net-icmp_filter \
net-y-unix \
net-yy-inet \
+ net-yy-netlink \
net-yy-unix \
netlink_inet_diag \
+ netlink_netlink_diag \
netlink_unix_diag \
newfstatat \
nsyscalls \
@@ -469,6 +471,7 @@ DECODER_TESTS = \
net-icmp_filter.test \
net-y-unix.test \
net-yy-inet.test \
+ net-yy-netlink.test \
net-yy-unix.test \
net.test \
newfstatat.test \
diff --git a/tests/net-yy-netlink.c b/tests/net-yy-netlink.c
new file mode 100644
index 0000000..a6cedc2
--- /dev/null
+++ b/tests/net-yy-netlink.c
@@ -0,0 +1,81 @@
+/*
+ * This file is part of net-yy-unix strace test.
+ *
+ * Copyright (c) 2016 Fabien Siron <fabien.siron at epita.fr>
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ * 3. The name of the author may not be used to endorse or promote products
+ * derived from this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "tests.h"
+#include <assert.h>
+#include <stddef.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <stdint.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/param.h>
+#include <sys/socket.h>
+#include <linux/netlink.h>
+#include <linux/sock_diag.h>
+#include <linux/netlink_diag.h>
+
+int
+main(void)
+{
+ char text[] = "text";
+ unsigned magic = 1234;
+ struct sockaddr_nl addr = {
+ .nl_family = AF_NETLINK,
+ .nl_pid = 1234
+ };
+ struct sockaddr *const sa = tail_memdup(&addr, sizeof(addr));
+ socklen_t * const len = tail_alloc(sizeof(socklen_t));
+ *len = sizeof(addr);
+
+ const int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_SOCK_DIAG);
+ if (fd < 0)
+ perror_msg_and_skip("socket");
+ const unsigned long inode = inode_of_sockfd(fd);
+ printf("socket(AF_NETLINK, SOCK_RAW, NETLINK_SOCK_DIAG) = "
+ "%d<NETLINK:[%lu]>\n", fd, inode);
+
+ if (bind(fd, sa, *len))
+ perror_msg_and_skip("bind");
+ printf("bind(%d<NETLINK:[%lu]>, {sa_family=AF_NETLINK, pid=%u, "
+ "groups=00000000}, %u) = 0\n", fd, inode, magic,
+ (unsigned) *len);
+
+ assert(sendto(fd, text, sizeof(text) - 1, MSG_DONTWAIT, NULL, 0)
+ == sizeof(text) - 1);
+ printf("sendto(%d<NETLINK:[SOCK_DIAG:%u]>, \"%s\", %u, MSG_DONTWAIT"
+ ", NULL, 0) = %u\n", fd, magic, text,
+ (unsigned) sizeof(text) - 1, (unsigned) sizeof(text) - 1);
+
+ assert(close(fd) == 0);
+ printf("close(%d<NETLINK:[SOCK_DIAG:%u]>) = 0\n", fd, magic);
+
+ puts("+++ exited with 0 +++");
+ return 0;
+}
diff --git a/tests/net-yy-netlink.test b/tests/net-yy-netlink.test
new file mode 100755
index 0000000..ffb59e9
--- /dev/null
+++ b/tests/net-yy-netlink.test
@@ -0,0 +1,47 @@
+#!/bin/sh
+#
+# Check decoding of ip:port pairs associated with socket descriptors
+#
+# Copyright (c) 2016 Fabien Siron <fabien.siron at epita.fr>
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions
+# are met:
+# 1. Redistributions of source code must retain the above copyright
+# notice, this list of conditions and the following disclaimer.
+# 2. Redistributions in binary form must reproduce the above copyright
+# notice, this list of conditions and the following disclaimer in the
+# documentation and/or other materials provided with the distribution.
+# 3. The name of the author may not be used to endorse or promote products
+# derived from this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
+# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
+# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+. "${srcdir=.}/init.sh"
+
+# strace -yy is implemented using /proc/self/fd
+[ -d /proc/self/fd/ ] ||
+ framework_skip_ '/proc/self/fd/ is not available'
+
+check_prog sed
+run_prog ./netlink_netlink_diag
+
+addr="$NAME-local-stream"
+run_prog "./$NAME" $addr > /dev/null
+
+run_strace -a22 -yy -eclose,network $args > "$EXP"
+# Filter out close() calls made by ld.so and libc.
+sed -n '/socket/,$p' < "$LOG" > "$OUT"
+
+match_diff "$OUT" "$EXP"
+rm -f "$EXP" "$OUT"
diff --git a/tests/netlink_netlink_diag.c b/tests/netlink_netlink_diag.c
new file mode 100644
index 0000000..9765edb
--- /dev/null
+++ b/tests/netlink_netlink_diag.c
@@ -0,0 +1,138 @@
+/*
+ * This file is part of net-yy-netlink strace test.
+ *
+ * Copyright (c) 2016 Fabien Siron <fabien.siron at epita.fr>
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ * 3. The name of the author may not be used to endorse or promote products
+ * derived from this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "tests.h"
+#include <errno.h>
+#include <string.h>
+#include <unistd.h>
+#include <netinet/in.h>
+#include <linux/netlink.h>
+#include <linux/sock_diag.h>
+#include <linux/netlink_diag.h>
+
+static void
+send_query(const int fd)
+{
+ struct sockaddr_nl nladdr = {
+ .nl_family = AF_NETLINK
+ };
+ struct {
+ struct nlmsghdr nlh;
+ struct netlink_diag_req ndr;
+ } req = {
+ .nlh = {
+ .nlmsg_len = sizeof(req),
+ .nlmsg_type = SOCK_DIAG_BY_FAMILY,
+ .nlmsg_flags = NLM_F_DUMP | NLM_F_REQUEST
+ },
+ .ndr = {
+ .sdiag_family = AF_NETLINK,
+ .sdiag_protocol = NDIAG_PROTO_ALL,
+ .ndiag_show = NDIAG_SHOW_MEMINFO
+ }
+ };
+ struct iovec iov = {
+ .iov_base = &req,
+ .iov_len = sizeof(req)
+ };
+ struct msghdr msg = {
+ .msg_name = (void *) &nladdr,
+ .msg_namelen = sizeof(nladdr),
+ .msg_iov = &iov,
+ .msg_iovlen = 1
+ };
+
+ if (sendmsg(fd, &msg, 0) <= 0)
+ perror_msg_and_skip("sendmsg");
+}
+
+static void
+check_responses(const int fd)
+{
+ static long buf[8192 / sizeof(long)];
+ struct sockaddr_nl nladdr = {
+ .nl_family = AF_NETLINK
+ };
+ struct iovec iov = {
+ .iov_base = buf,
+ .iov_len = sizeof(buf)
+ };
+ struct msghdr msg = {
+ .msg_name = (void *) &nladdr,
+ .msg_namelen = sizeof(nladdr),
+ .msg_iov = &iov,
+ .msg_iovlen = 1
+ };
+
+ ssize_t ret = recvmsg(fd, &msg, 0);
+ if (ret <= 0)
+ perror_msg_and_skip("recvmsg");
+
+ struct nlmsghdr *h = (struct nlmsghdr *) buf;
+ if (!NLMSG_OK(h, ret))
+ error_msg_and_skip("!NLMSG_OK");
+ if (h->nlmsg_type == NLMSG_ERROR) {
+ const struct nlmsgerr *err = NLMSG_DATA(h);
+ if (h->nlmsg_len < NLMSG_LENGTH(sizeof(*err)))
+ error_msg_and_skip("NLMSG_ERROR");
+ errno = -err->error;
+ perror_msg_and_skip("NLMSG_ERROR");
+ }
+ if (h->nlmsg_type != SOCK_DIAG_BY_FAMILY)
+ error_msg_and_skip("unexpected nlmsg_type %u",
+ (unsigned) h->nlmsg_type);
+
+ const struct netlink_diag_msg *diag = NLMSG_DATA(h);
+ if (h->nlmsg_len < NLMSG_LENGTH(sizeof(*diag)))
+ error_msg_and_skip("short response");
+}
+
+int main(void)
+{
+ struct sockaddr_nl addr;
+ socklen_t len = sizeof(addr);
+
+ memset(&addr, 0, sizeof(addr));
+ addr.nl_family = AF_NETLINK;
+
+ close(0);
+ close(1);
+
+ if (socket(AF_NETLINK, SOCK_RAW, NETLINK_SOCK_DIAG))
+ perror_msg_and_skip("socket AF_NETLINK");
+ if (bind(0, (struct sockaddr *) &addr, len))
+ perror_msg_and_skip("bind");
+
+ if (socket(AF_NETLINK, SOCK_RAW, NETLINK_SOCK_DIAG) != 1)
+ perror_msg_and_skip("socket AF_NETLINK");
+
+ send_query(1);
+ check_responses(1);
+ return 0;
+}
--
2.8.0
More information about the Strace-devel
mailing list