From: JingPiao Chen <chenjingpiao@gmail.com>
Date: Tue, 27 Jun 2017 10:10:05 +0000 (+0800)
Subject: netlink: add a basic netlink attribute parser of AF_INET diag
X-Git-Tag: v4.18~36
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=ab17493bb2335a06bf3c604167cea8541b151ffa;p=strace

netlink: add a basic netlink attribute parser of AF_INET diag

* linux/inet_diag.h (INET_DIAG_REQ_*): New enum.
* netlink_sock_diag.c: Include "xlat/inet_diag_attrs.h"
and "xlat/inet_diag_req_attrs.h".
(decode_inet_diag_req_compat, decode_inet_diag_req_v2,
 decode_inet_diag_msg): Use decode_nlattr.
* xlat/inet_diag_attrs.in: New file.
* xlat/inet_diag_req_attrs.in: Likewise.

Co-authored-by: Fabien Siron <fabien.siron@epita.fr>
---

diff --git a/linux/inet_diag.h b/linux/inet_diag.h
index 245e4761..e1df3bb6 100644
--- a/linux/inet_diag.h
+++ b/linux/inet_diag.h
@@ -34,6 +34,11 @@ struct inet_diag_req_v2 {
 	struct inet_diag_sockid id;
 };
 
+enum {
+	INET_DIAG_REQ_NONE,
+	INET_DIAG_REQ_BYTECODE,
+};
+
 /* Info structure */
 struct inet_diag_msg {
 	uint8_t idiag_family;
diff --git a/netlink_sock_diag.c b/netlink_sock_diag.c
index d15df970..47ee8aba 100644
--- a/netlink_sock_diag.c
+++ b/netlink_sock_diag.c
@@ -40,7 +40,9 @@
 #endif
 #include <linux/unix_diag.h>
 
+#include "xlat/inet_diag_attrs.h"
 #include "xlat/inet_diag_extended_flags.h"
+#include "xlat/inet_diag_req_attrs.h"
 
 #include "xlat/tcp_states.h"
 #include "xlat/tcp_state_flags.h"
@@ -327,7 +329,8 @@ decode_inet_diag_req_compat(struct tcb *const tcp,
 			    const kernel_ulong_t len)
 {
 	struct inet_diag_req req = { .idiag_family = family };
-	const size_t offset = sizeof(req.idiag_family);
+	size_t offset = sizeof(req.idiag_family);
+	bool decode_nla = false;
 
 	PRINT_FIELD_XVAL("{", req, idiag_family, addrfams, "AF_???");
 	tprints(", ");
@@ -345,10 +348,18 @@ decode_inet_diag_req_compat(struct tcb *const tcp,
 			PRINT_FIELD_FLAGS(", ", req, idiag_states,
 					  tcp_state_flags, "1<<TCP_???");
 			PRINT_FIELD_U(", ", req, idiag_dbs);
+			decode_nla = true;
 		}
 	} else
 		tprints("...");
 	tprints("}");
+
+	offset = NLA_ALIGN(sizeof(req));
+	if (decode_nla && len > offset) {
+		tprints(", ");
+		decode_nlattr(tcp, addr + offset, len - offset,
+			      inet_diag_req_attrs, "INET_DIAG_REQ_???");
+	}
 }
 
 static void
@@ -359,7 +370,8 @@ decode_inet_diag_req_v2(struct tcb *const tcp,
 			const kernel_ulong_t len)
 {
 	struct inet_diag_req_v2 req = { .sdiag_family = family };
-	const size_t offset = sizeof(req.sdiag_family);
+	size_t offset = sizeof(req.sdiag_family);
+	bool decode_nla = false;
 
 	PRINT_FIELD_XVAL("{", req, sdiag_family, addrfams, "AF_???");
 	tprints(", ");
@@ -376,10 +388,18 @@ decode_inet_diag_req_v2(struct tcb *const tcp,
 					  tcp_state_flags, "1<<TCP_???");
 			tprints(", id=");
 			print_inet_diag_sockid(&req.id, req.sdiag_family);
+			decode_nla = true;
 		}
 	} else
 		tprints("...");
 	tprints("}");
+
+	offset = NLA_ALIGN(sizeof(req));
+	if (decode_nla && len > offset) {
+		tprints(", ");
+		decode_nlattr(tcp, addr + offset, len - offset,
+			      inet_diag_req_attrs, "INET_DIAG_REQ_???");
+	}
 }
 
 static void
@@ -406,7 +426,8 @@ decode_inet_diag_msg(struct tcb *const tcp,
 		     const kernel_ulong_t len)
 {
 	struct inet_diag_msg msg = { .idiag_family = family };
-	const size_t offset = sizeof(msg.idiag_family);
+	size_t offset = sizeof(msg.idiag_family);
+	bool decode_nla = false;
 
 	PRINT_FIELD_XVAL("{", msg, idiag_family, addrfams, "AF_???");
 	tprints(", ");
@@ -425,10 +446,18 @@ decode_inet_diag_msg(struct tcb *const tcp,
 			PRINT_FIELD_U(", ", msg, idiag_wqueue);
 			PRINT_FIELD_U(", ", msg, idiag_uid);
 			PRINT_FIELD_U(", ", msg, idiag_inode);
+			decode_nla = true;
 		}
 	} else
 		tprints("...");
 	tprints("}");
+
+	offset = NLA_ALIGN(sizeof(msg));
+	if (decode_nla && len > offset) {
+		tprints(", ");
+		decode_nlattr(tcp, addr + offset, len - offset,
+			      inet_diag_attrs, "INET_DIAG_???");
+	}
 }
 
 #ifdef AF_SMC
diff --git a/xlat/inet_diag_attrs.in b/xlat/inet_diag_attrs.in
new file mode 100644
index 00000000..17519c83
--- /dev/null
+++ b/xlat/inet_diag_attrs.in
@@ -0,0 +1,18 @@
+#unconditional
+INET_DIAG_NONE
+INET_DIAG_MEMINFO
+INET_DIAG_INFO
+INET_DIAG_VEGASINFO
+INET_DIAG_CONG
+INET_DIAG_TOS
+INET_DIAG_TCLASS
+INET_DIAG_SKMEMINFO
+INET_DIAG_SHUTDOWN
+INET_DIAG_DCTCPINFO
+INET_DIAG_PROTOCOL
+INET_DIAG_SKV6ONLY
+INET_DIAG_LOCALS
+INET_DIAG_PEERS
+INET_DIAG_PAD
+INET_DIAG_MARK
+INET_DIAG_BBRINFO
diff --git a/xlat/inet_diag_req_attrs.in b/xlat/inet_diag_req_attrs.in
new file mode 100644
index 00000000..c36369ee
--- /dev/null
+++ b/xlat/inet_diag_req_attrs.in
@@ -0,0 +1,3 @@
+#unconditional
+INET_DIAG_REQ_NONE
+INET_DIAG_REQ_BYTECODE