Received-SPF: pass (google.com: domain of linux-kernel-owner@vger.kernel.org designates 2620:137:e000::1:20 as permitted sender) client-ip=2620:137:e000::1:20;
From:   Alan Maguire <alan.maguire@oracle.com>
To:     ast@kernel.org, daniel@iogearbox.net, andrii@kernel.org,
        jolsa@kernel.org
Cc:     kafai@fb.com, songliubraving@fb.com, yhs@fb.com,
        john.fastabend@gmail.com, kpsingh@kernel.org, mhiramat@kernel.org,
        akpm@linux-foundation.org, void@manifault.com, swboyd@chromium.org,
        ndesaulniers@google.com, linux-kernel@vger.kernel.org,
        netdev@vger.kernel.org, bpf@vger.kernel.org,
        Alan Maguire <alan.maguire@oracle.com>
Subject: [RFC bpf-next 2/2] selftests/bpf: add a kallsyms iter subtest
Date:   Fri, 24 Jun 2022 17:45:18 +0100
Message-Id: <1656089118-577-3-git-send-email-alan.maguire@oracle.com>
In-Reply-To: <1656089118-577-1-git-send-email-alan.maguire@oracle.com>
References: <1656089118-577-1-git-send-email-alan.maguire@oracle.com>
Content-Type: text/plain
MIME-Version: 1.0
X-MS-Exchange-AntiSpam-MessageData-ChunkCount: 1
X-MS-Exchange-AntiSpam-MessageData-0: =?us-ascii?Q?/71vQTuK5It4uNb8XvgQRIPQTQ0XJdYBKLcypboDP9V6VVxihUvytsS/DfOz?=
 =?us-ascii?Q?WA+Cm7a9ZP26FP1TTEZpVKLKgjNzUyxuZVjKN1adKk0BSraov+rFNWY/bP7a?=
 =?us-ascii?Q?icSsvAdNiXS/ZkhxR4TPrZfhvlldseHPrWTsFckjNPC5NQT0RKK7Lf2haIpa?=
 =?us-ascii?Q?iTMYS3XPZXko0mwobZQ+hZpq4D27claweJwJhQFrndwwlkfmHrVf9aCS2RIz?=
 =?us-ascii?Q?IQDpX9pshfzMJaOVSFWw+4NA5Xa7aaka0api5y7o1vOJckiWpqzG8cOKwQCJ?=
 =?us-ascii?Q?n3L6t/F26WSVIrv/P1xcNDL72vprWEECZUQH2zaLQ+XilkeiX1rFDcRnyq4+?=
 =?us-ascii?Q?8dQZ8oyTUfMqES9j6g9RE7f7theaZdSj+NOpGr7chSscbCcEjUtwIP/KoYmv?=
 =?us-ascii?Q?MVyvBUbQmvgn23Hvqt2NScGG4sFQqE5M4qWs0f/uW5XZZrpYrYhkovPXeOqX?=
 =?us-ascii?Q?2j5mIWQIAunVpqqYaS/ckDxHcXr+9YvnwNJlfCktY0c/DvCcF7jgH/zYxb0I?=
 =?us-ascii?Q?Asj2/c+ocwh9WjxCBmKK0o3fqovM+NVIqn1juDq4oCcv78zg2h3G3Cm+OL1R?=
 =?us-ascii?Q?wm76JCSMdt4OhRdQj2eEPLQf9iGzrHPjdfq35V7yFeq1e2+Hz9qIoey5SIia?=
 =?us-ascii?Q?uEWw7APRuxzl+kw0TevNuQuLCM/GiiwY3PKWH7b0qJecimZF6CKlKsKveNQX?=
 =?us-ascii?Q?p20eSYxbpmNNUIn6LAAGaMcSLLArzy2RybWMbzlsvU5TLfpHH/3Gl+V3uJ4P?=
 =?us-ascii?Q?dOK+4GusTfAHx56e0NM8QnQXM/JZ4Wey26ok1NEyxiqiEZRZur1IyRpoJC0I?=
 =?us-ascii?Q?sYjSwXmOeIL1hST/FsArreRkjGsLvEnakJVOB9Nor5flmcOefwzMXOc+QJqw?=
 =?us-ascii?Q?+bchGpllpjfhdwOsc1ZCgEZ2N3ko6zv0SvERak5k6CLkyaUEEDKu8s0TlSE2?=
 =?us-ascii?Q?r2Uk9hLuAN3vmOHHnHg2fC7XoyVT2CJgI56IaiKUssNW+MTUysl1O95LdTuF?=
 =?us-ascii?Q?10UoWf9TvrCOv3FTHw6AESTJ7IZAWEU4FLXncvi1y0x7gsdb80c5f6+7lwFD?=
 =?us-ascii?Q?CLgQbmCHagVTFG7X0B4O7t1ts8fes4ZFGWkYGdfbwtN1a3Rsst7fQr0lFodI?=
 =?us-ascii?Q?7rKJRmORYToC05WFdP71Cqkw1Na03vUgW4ktmECq6WwWzf9OitLSvaHgA6g6?=
 =?us-ascii?Q?EqxcPqXKfqFYt25SQ4DWudMh4O1e82qCPZ5DtoIuo8BuFY09sSCa13Yv4YlI?=
 =?us-ascii?Q?IQJurXwKZZd/Rk/17s67diu4ERplFGtPekAiHlk1ueSgry37p2W/wpO3eq/E?=
 =?us-ascii?Q?lHEK7oIIao+1P5A9a++lE+v2NvXXP/Me+AAJ3zszWgGjrk32JGUacJXv2Hhu?=
 =?us-ascii?Q?46Hpgjls7Ls17iobnSqxM1RnEZ93CzalYrxSbdOhK5NInbF/o+ceymFZVsND?=
 =?us-ascii?Q?kTSoBGmAHyIxB76Bw9d20RiwdwrRngJO6HPZe4Mi7SPMDnnxMmJhn4RmjRJe?=
 =?us-ascii?Q?f4AmJcUZfBzGT+muPsOgMsh7bSJ6TBu84+lNquQMvXCLM2ELb2F3QaibiJdB?=
 =?us-ascii?Q?FpsgdYOUR9gRydQBWOhfgEnLTVzZjXPJplEm/2O2?=
X-OriginatorOrg: oracle.com
X-MS-Exchange-CrossTenant-Network-Message-Id: 28f339d6-520d-4eae-effe-08da5600f292
X-MS-Exchange-CrossTenant-AuthSource: BLAPR10MB5267.namprd10.prod.outlook.com
X-MS-Exchange-CrossTenant-AuthAs: Internal
X-MS-Exchange-CrossTenant-OriginalArrivalTime: 24 Jun 2022 16:45:29.7769
 (UTC)
X-MS-Exchange-CrossTenant-FromEntityHeader: Hosted
X-MS-Exchange-CrossTenant-Id: 4e2c6054-71cb-48f1-bd6c-3a9705aca71b
X-MS-Exchange-CrossTenant-MailboxType: HOSTED
X-MS-Exchange-CrossTenant-UserPrincipalName: BFDlszoOi1r9BJx6EDKMrCLXr/f1l7p/OgTrSuK0q/kQmLSJdWUUW+OrffTECHdll6QA2cPw0jUie17bL+F2yA==
X-MS-Exchange-Transport-CrossTenantHeadersStamped: CH0PR10MB5115
X-Proofpoint-Virus-Version: vendor=fsecure engine=2.50.10434:6.0.517,18.0.883
 definitions=2022-06-24_08:2022-06-23,2022-06-24 signatures=0
X-Proofpoint-Spam-Details: rule=notspam policy=default score=0 adultscore=0 bulkscore=0 phishscore=0
 mlxlogscore=999 mlxscore=0 suspectscore=0 spamscore=0 malwarescore=0
 classifier=spam adjust=0 reason=mlx scancount=1 engine=8.12.0-2204290000
 definitions=main-2206240064
X-Proofpoint-GUID: BhlkWN6hnY_0ucS9EB3RMc0v-tJOBqr7
X-Proofpoint-ORIG-GUID: BhlkWN6hnY_0ucS9EB3RMc0v-tJOBqr7
X-Spam-Status: No, score=-2.8 required=5.0 tests=BAYES_00,DKIM_SIGNED,
        DKIM_VALID,DKIM_VALID_AU,DKIM_VALID_EF,RCVD_IN_DNSWL_LOW,
        RCVD_IN_MSPIKE_H2,SPF_HELO_NONE,SPF_NONE,T_SCC_BODY_TEXT_LINE
        autolearn=ham autolearn_force=no version=3.4.6
X-Spam-Checker-Version: SpamAssassin 3.4.6 (2021-04-09) on
        lindbergh.monkeyblade.net
Precedence: bulk
List-ID: <linux-kernel.vger.kernel.org>
X-Mailing-List: linux-kernel@vger.kernel.org

add subtest verifying BPF kallsyms iter behaviour.  The BPF kallsyms
iter program shows an example of dumping a format different to
kallsyms.  It adds KIND and MAX_SIZE fields which represent the
kind of symbol (core kernel, module, ftrace, bpf, or kprobe) and
the maximum size the symbol can be.  The latter is calculated from
the difference between current symbol value and the next symbol
value.

The key benefit for this iterator will likely be supporting in-kernel
data-gathering rather than dumping symbol details to userspace and
parsing the results.

Signed-off-by: Alan Maguire <alan.maguire@oracle.com>
---
 tools/testing/selftests/bpf/prog_tests/bpf_iter.c  | 16 +++++
 .../selftests/bpf/progs/bpf_iter_kallsyms.c        | 71 ++++++++++++++++++++++
 2 files changed, 87 insertions(+)
 create mode 100644 tools/testing/selftests/bpf/progs/bpf_iter_kallsyms.c

diff --git a/tools/testing/selftests/bpf/prog_tests/bpf_iter.c b/tools/testing/selftests/bpf/prog_tests/bpf_iter.c
index 7ff5fa9..1197a58 100644
--- a/tools/testing/selftests/bpf/prog_tests/bpf_iter.c
+++ b/tools/testing/selftests/bpf/prog_tests/bpf_iter.c
@@ -27,6 +27,7 @@
 #include "bpf_iter_test_kern5.skel.h"
 #include "bpf_iter_test_kern6.skel.h"
 #include "bpf_iter_bpf_link.skel.h"
+#include "bpf_iter_kallsyms.skel.h"
 
 static int duration;
 
@@ -1120,6 +1121,19 @@ static void test_link_iter(void)
 	bpf_iter_bpf_link__destroy(skel);
 }
 
+static void test_kallsyms_iter(void)
+{
+	struct bpf_iter_kallsyms *skel;
+
+	skel = bpf_iter_kallsyms__open_and_load();
+	if (!ASSERT_OK_PTR(skel, "bpf_iter_kallsyms__open_and_load"))
+		return;
+
+	do_dummy_read(skel->progs.dump_kallsyms);
+
+	bpf_iter_kallsyms__destroy(skel);
+}
+
 #define CMP_BUFFER_SIZE 1024
 static char task_vma_output[CMP_BUFFER_SIZE];
 static char proc_maps_output[CMP_BUFFER_SIZE];
@@ -1267,4 +1281,6 @@ void test_bpf_iter(void)
 		test_buf_neg_offset();
 	if (test__start_subtest("link-iter"))
 		test_link_iter();
+	if (test__start_subtest("kallsyms"))
+		test_kallsyms_iter();
 }
diff --git a/tools/testing/selftests/bpf/progs/bpf_iter_kallsyms.c b/tools/testing/selftests/bpf/progs/bpf_iter_kallsyms.c
new file mode 100644
index 0000000..20c8d85
--- /dev/null
+++ b/tools/testing/selftests/bpf/progs/bpf_iter_kallsyms.c
@@ -0,0 +1,71 @@
+// SPDX-License-Identifier: GPL-2.0
+/* Copyright (c) 2022, Oracle and/or its affiliates. */
+#include "bpf_iter.h"
+#include <bpf/bpf_helpers.h>
+
+char _license[] SEC("license") = "GPL";
+
+unsigned long last_sym_value = 0;
+
+static inline char tolower(char c)
+{
+	if (c >= 'A' && c <= 'Z')
+		c += ('a' - 'A');
+	return c;
+}
+
+static inline char toupper(char c)
+{
+	if (c >= 'a' && c <= 'z')
+		c -= ('a' - 'A');
+	return c;
+}
+
+/* Dump symbols with max size; the latter is calculated by caching symbol N value
+ * and when iterating on symbol N+1, we can print max size of symbol N via
+ * address of N+1 - address of N.
+ */
+SEC("iter/kallsyms")
+int dump_kallsyms(struct bpf_iter__kallsyms *ctx)
+{
+	struct seq_file *seq = ctx->meta->seq;
+	struct kallsym_iter *iter = ctx->kallsym_iter;
+	__u32 seq_num = ctx->meta->seq_num;
+	char type;
+	int ret;
+
+	if (!iter)
+		return 0;
+
+	if (seq_num == 0) {
+		BPF_SEQ_PRINTF(seq, "ADDR TYPE NAME MODULE_NAME KIND MAX_SIZE\n");
+		return 0;
+	}
+	if (last_sym_value)
+		BPF_SEQ_PRINTF(seq, "0x%x\n", iter->value - last_sym_value);
+	else
+		BPF_SEQ_PRINTF(seq, "\n");
+
+	last_sym_value = iter->value;
+
+	type = iter->type;
+
+	if (iter->module_name[0]) {
+		type = iter->exported ? toupper(type) : tolower(type);
+		BPF_SEQ_PRINTF(seq, "0x%llx %c %s [ %s ] ",
+			       iter->value, type, iter->name, iter->module_name);
+	} else {
+		BPF_SEQ_PRINTF(seq, "0x%llx %c %s ", iter->value, type, iter->name);
+	}
+	if (!iter->pos_arch_end || iter->pos_arch_end > iter->pos)
+		BPF_SEQ_PRINTF(seq, "CORE ");
+	else if (!iter->pos_mod_end || iter->pos_mod_end > iter->pos)
+		BPF_SEQ_PRINTF(seq, "MOD ");
+	else if (!iter->pos_ftrace_mod_end || iter->pos_ftrace_mod_end > iter->pos)
+		BPF_SEQ_PRINTF(seq, "FTRACE_MOD ");
+	else if (!iter->pos_bpf_end || iter->pos_bpf_end > iter->pos)
+		BPF_SEQ_PRINTF(seq, "BPF ");
+	else
+		BPF_SEQ_PRINTF(seq, "KPROBE ");
+	return 0;
+}
-- 
1.8.3.1