Add a test to demonstrate that qp-trie doesn't care about the
unused part of key.
Signed-off-by: Hou Tao <houtao1@xxxxxxxxxx>
---
.../selftests/bpf/prog_tests/str_key.c | 69 +++++++++++++++
tools/testing/selftests/bpf/progs/str_key.c | 85 +++++++++++++++++++
2 files changed, 154 insertions(+)
create mode 100644 tools/testing/selftests/bpf/prog_tests/str_key.c
create mode 100644 tools/testing/selftests/bpf/progs/str_key.c
diff --git a/tools/testing/selftests/bpf/prog_tests/str_key.c b/tools/testing/selftests/bpf/prog_tests/str_key.c
new file mode 100644
index 000000000000..8f134dd45902
--- /dev/null
+++ b/tools/testing/selftests/bpf/prog_tests/str_key.c
@@ -0,0 +1,69 @@
+// SPDX-License-Identifier: GPL-2.0
+/* Copyright (C) 2022. Huawei Technologies Co., Ltd */
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <test_progs.h>
+#include "str_key.skel.h"
+
+#define FILE_PATH_SIZE 64
+
+struct file_path_str {
+ unsigned int len;
+ char raw[FILE_PATH_SIZE];
+};
+
+static int setup_maps(struct str_key *skel, const char *name, unsigned int value)
+{
+ struct file_path_str key;
+ int fd, err;
+
+ memset(&key, 0, sizeof(key));
+ strncpy(key.raw, name, sizeof(key.raw) - 1);
+ key.len = strlen(name) + 1;
+
+ fd = bpf_map__fd(skel->maps.trie);
+ err = bpf_map_update_elem(fd, &key, &value, BPF_NOEXIST);
+ if (!ASSERT_OK(err, "trie add"))
+ return -EINVAL;
+
+ fd = bpf_map__fd(skel->maps.htab);
+ err = bpf_map_update_elem(fd, key.raw, &value, BPF_NOEXIST);
+ if (!ASSERT_OK(err, "htab add"))
+ return -EINVAL;
+
+ return 0;
+}
+
+void test_str_key(void)
+{
+ const char *name = "/tmp/str_key_test";
+ struct str_key *skel;
+ unsigned int value;
+ int err, fd;
+
+ skel = str_key__open_and_load();
+ if (!ASSERT_OK_PTR(skel, "open_load str key"))
+ return;
+
+ value = time(NULL);
+ if (setup_maps(skel, name, value))
+ goto out;
+
+ skel->bss->pid = getpid();
+ err = str_key__attach(skel);
+ if (!ASSERT_OK(err, "attach"))
+ goto out;
+
+ fd = open(name, O_RDONLY | O_CREAT, 0644);
+ if (!ASSERT_GE(fd, 0, "open tmp file"))
+ goto out;
+ close(fd);
+ unlink(name);
+
+ ASSERT_EQ(skel->bss->trie_value, value, "trie lookup str");
+ ASSERT_EQ(skel->bss->htab_value, -1, "htab lookup str");
+out:
+ str_key__destroy(skel);
+}
diff --git a/tools/testing/selftests/bpf/progs/str_key.c b/tools/testing/selftests/bpf/progs/str_key.c
new file mode 100644
index 000000000000..2e8becdd58c6
--- /dev/null
+++ b/tools/testing/selftests/bpf/progs/str_key.c
@@ -0,0 +1,85 @@
+// SPDX-License-Identifier: GPL-2.0
+/* Copyright (C) 2022. Huawei Technologies Co., Ltd */
+#include <linux/types.h>
+#include <linux/bpf.h>
+#include <bpf/bpf_helpers.h>
+#include <bpf/bpf_core_read.h>
+#include <bpf/bpf_tracing.h>
+
+char _license[] SEC("license") = "GPL";
+
+struct path {
+} __attribute__((preserve_access_index));
+
+struct file {
+ struct path f_path;
+} __attribute__((preserve_access_index));
+
+#define FILE_PATH_SIZE 64
+
+struct file_path_str {
+ unsigned int len;
+ char raw[FILE_PATH_SIZE];
+};
+
+struct {
+ __uint(type, BPF_MAP_TYPE_ARRAY);
+ __uint(max_entries, 1);
+ __type(key, int);
+ __type(value, struct file_path_str);
+} array SEC(".maps");
+
+struct {
+ __uint(type, BPF_MAP_TYPE_QP_TRIE);
+ __uint(max_entries, 1);
+ __type(key, struct file_path_str);
+ __type(value, __u32);
+ __uint(map_flags, BPF_F_NO_PREALLOC);
+} trie SEC(".maps");
+
+struct {
+ __uint(type, BPF_MAP_TYPE_HASH);
+ __uint(max_entries, 1);
+ __uint(key_size, FILE_PATH_SIZE);
+ __uint(value_size, sizeof(__u32));
+} htab SEC(".maps");
+
+int pid = 0;
+unsigned int trie_value = 0;
+unsigned int htab_value = 0;
+
+SEC("fentry/filp_close")
+int BPF_PROG(filp_close, struct file *filp)
+{
+ struct path *p = &filp->f_path;
+ struct file_path_str *str;
+ unsigned int *value;
+ int idx, len;
+
+ if (bpf_get_current_pid_tgid() >> 32 != pid)
+ return 0;
+
+ idx = 0;
+ str = bpf_map_lookup_elem(&array, &idx);
+ if (!str)
+ return 0;
+
+ len = bpf_d_path(p, str->raw, sizeof(str->raw));
+ if (len < 0 || len > sizeof(str->raw))
+ return 0;
+
+ str->len = len;
+ value = bpf_map_lookup_elem(&trie, str);
+ if (value)
+ trie_value = *value;
+ else
+ trie_value = -1;
+
+ value = bpf_map_lookup_elem(&htab, str->raw);
+ if (value)
+ htab_value = *value;
+ else
+ htab_value = -1;
+
+ return 0;
+}
--
2.29.2
