[PATCH 3/5] tests: check decoding of TCA_STATS2 netlink attribute of tcmsg

JingPiao Chen chenjingpiao at gmail.com
Tue Sep 5 13:16:30 UTC 2017


* tests/nlattr_tc_stats.c: New file.
* tests/gen_tests.in (nlattr_tc_stats): New entry.
* tests/pure_executables.list: Add nlattr_tc_stats.
* tests/.gitignore: Likewise.
---
 tests/.gitignore            |   1 +
 tests/gen_tests.in          |   1 +
 tests/nlattr_tc_stats.c     | 151 ++++++++++++++++++++++++++++++++++++++++++++
 tests/pure_executables.list |   1 +
 4 files changed, 154 insertions(+)
 create mode 100644 tests/nlattr_tc_stats.c

diff --git a/tests/.gitignore b/tests/.gitignore
index 3d1631e..c6bfc02 100644
--- a/tests/.gitignore
+++ b/tests/.gitignore
@@ -240,6 +240,7 @@ nlattr_packet_diag_msg
 nlattr_rtgenmsg
 nlattr_rtmsg
 nlattr_smc_diag_msg
+nlattr_tc_stats
 nlattr_tcamsg
 nlattr_tcmsg
 nlattr_unix_diag_msg
diff --git a/tests/gen_tests.in b/tests/gen_tests.in
index 8592e28..117f0a1 100644
--- a/tests/gen_tests.in
+++ b/tests/gen_tests.in
@@ -219,6 +219,7 @@ nlattr_packet_diag_msg		+netlink_sock_diag-v.sh
 nlattr_rtgenmsg			+netlink_sock_diag.test
 nlattr_rtmsg			+netlink_sock_diag.test
 nlattr_smc_diag_msg		+netlink_sock_diag.test
+nlattr_tc_stats			+netlink_sock_diag.test
 nlattr_tcamsg			+netlink_sock_diag.test
 nlattr_tcmsg			+netlink_sock_diag.test
 nlattr_unix_diag_msg		+netlink_sock_diag.test
diff --git a/tests/nlattr_tc_stats.c b/tests/nlattr_tc_stats.c
new file mode 100644
index 0000000..dd76cc7
--- /dev/null
+++ b/tests/nlattr_tc_stats.c
@@ -0,0 +1,151 @@
+/*
+ * Copyright (c) 2017 JingPiao Chen <chenjingpiao at gmail.com>
+ * Copyright (c) 2017 The strace developers.
+ * 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"
+
+#ifdef HAVE_STRUCT_GNET_STATS_BASIC
+
+# include <stdio.h>
+# include <stddef.h>
+# include "test_nlattr.h"
+# include <linux/gen_stats.h>
+# include <linux/rtnetlink.h>
+
+const unsigned int hdrlen = sizeof(struct tcmsg);
+
+static void
+init_tcmsg(struct nlmsghdr *const nlh, const unsigned int msg_len)
+{
+	SET_STRUCT(struct nlmsghdr, nlh,
+		.nlmsg_len = msg_len,
+		.nlmsg_type = RTM_GETQDISC,
+		.nlmsg_flags = NLM_F_DUMP
+	);
+
+	struct tcmsg *const msg = NLMSG_DATA(nlh);
+	SET_STRUCT(struct tcmsg, msg,
+		.tcm_family = AF_UNIX,
+		.tcm_ifindex = ifindex_lo()
+	);
+
+	struct nlattr *const nla = NLMSG_ATTR(nlh, sizeof(*msg));
+	SET_STRUCT(struct nlattr, nla,
+		.nla_len = msg_len - NLMSG_SPACE(hdrlen),
+		.nla_type = TCA_STATS2
+	);
+}
+
+static void
+print_tcmsg(const unsigned int msg_len)
+{
+	printf("{len=%u, type=RTM_GETQDISC, flags=NLM_F_DUMP"
+	       ", seq=0, pid=0}, {tcm_family=AF_UNIX"
+	       ", tcm_ifindex=" IFINDEX_LO_STR
+	       ", tcm_handle=0, tcm_parent=0, tcm_info=0}"
+	       ", {{nla_len=%u, nla_type=TCA_STATS2}",
+	       msg_len, msg_len - NLMSG_SPACE(hdrlen));
+}
+
+int
+main(void)
+{
+	skip_if_unavailable("/proc/self/fd/");
+
+	const int fd = create_nl_socket(NETLINK_ROUTE);
+	void *nlh0 = tail_alloc(NLMSG_SPACE(hdrlen));
+
+	static char pattern[4096];
+	fill_memory_ex(pattern, sizeof(pattern), 'a', 'z' - 'a' + 1);
+
+	static const struct gnet_stats_basic sb = {
+		.bytes = 0xabcdebdafefeadeb,
+		.packets = 0xbdcdeabf
+	};
+	char buf[offsetofend(struct gnet_stats_basic, packets)];
+	memcpy(buf, &sb, sizeof(buf));
+	TEST_NESTED_NLATTR_OBJECT(fd, nlh0, hdrlen,
+				  init_tcmsg, print_tcmsg,
+				  TCA_STATS_BASIC, pattern, buf,
+				  PRINT_FIELD_U("{", sb, bytes);
+				  PRINT_FIELD_U(", ", sb, packets);
+				  printf("}"));
+
+#ifdef HAVE_STRUCT_GNET_STATS_RATE_EST
+	static const struct gnet_stats_rate_est est = {
+		.bps = 0xebcdaebd,
+		.pps = 0xabdceade,
+	};
+	TEST_NESTED_NLATTR_OBJECT(fd, nlh0, hdrlen,
+				  init_tcmsg, print_tcmsg,
+				  TCA_STATS_RATE_EST, pattern, est,
+				  PRINT_FIELD_U("{", est, bps);
+				  PRINT_FIELD_U(", ", est, pps);
+				  printf("}"));
+#endif
+
+#ifdef HAVE_STRUCT_GNET_STATS_QUEUE
+	static const struct gnet_stats_queue qstats = {
+		.qlen = 0xabcdeabd,
+		.backlog = 0xbcdaebad,
+		.drops = 0xcdbeaedb,
+		.requeues = 0xdebaefab,
+		.overlimits = 0xefaebade
+	};
+	TEST_NESTED_NLATTR_OBJECT(fd, nlh0, hdrlen,
+				  init_tcmsg, print_tcmsg,
+				  TCA_STATS_QUEUE, pattern, qstats,
+				  PRINT_FIELD_U("{", qstats, qlen);
+				  PRINT_FIELD_U(", ", qstats, backlog);
+				  PRINT_FIELD_U(", ", qstats, drops);
+				  PRINT_FIELD_U(", ", qstats, requeues);
+				  PRINT_FIELD_U(", ", qstats, overlimits);
+				  printf("}"));
+#endif
+
+#ifdef HAVE_STRUCT_GNET_STATS_RATE_EST64
+	static const struct gnet_stats_rate_est64 est64 = {
+		.bps = 0xacbdcdefafecaebf,
+		.pps = 0xcdabeabdfeabceaf
+	};
+	TEST_NESTED_NLATTR_OBJECT(fd, nlh0, hdrlen,
+				  init_tcmsg, print_tcmsg,
+				  TCA_STATS_RATE_EST64, pattern, est64,
+				  PRINT_FIELD_U("{", est64, bps);
+				  PRINT_FIELD_U(", ", est64, pps);
+				  printf("}"));
+#endif
+
+	puts("+++ exited with 0 +++");
+	return 0;
+}
+
+#else
+
+SKIP_MAIN_UNDEFINED("HAVE_STRUCT_GNET_STATS_BASIC")
+
+#endif
diff --git a/tests/pure_executables.list b/tests/pure_executables.list
index dade57b..2878d3a 100755
--- a/tests/pure_executables.list
+++ b/tests/pure_executables.list
@@ -202,6 +202,7 @@ nlattr_packet_diag_msg
 nlattr_rtgenmsg
 nlattr_rtmsg
 nlattr_smc_diag_msg
+nlattr_tc_stats
 nlattr_tcamsg
 nlattr_tcmsg
 nlattr_unix_diag_msg
-- 
2.7.4





More information about the Strace-devel mailing list