From 363485794cbd299c7968c9284fe39a5d0280aa59 Mon Sep 17 00:00:00 2001
From: Geliang Tang <tanggeliang@kylinos.cn>
Date: Fri, 11 Oct 2024 15:27:32 +0800
Subject: [PATCH] bpf local list

Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn>
---
 .../bpf/progs/mptcp_bpf_userspace_pm.c        | 48 +++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/tools/testing/selftests/bpf/progs/mptcp_bpf_userspace_pm.c b/tools/testing/selftests/bpf/progs/mptcp_bpf_userspace_pm.c
index 8880ee085a59..5d1de3eed9be 100644
--- a/tools/testing/selftests/bpf/progs/mptcp_bpf_userspace_pm.c
+++ b/tools/testing/selftests/bpf/progs/mptcp_bpf_userspace_pm.c
@@ -5,17 +5,41 @@
 #include "mptcp_bpf.h"
 
 char _license[] SEC("license") = "GPL";
+int use_bpf_list = 0;
+
+struct {
+	__uint(type, BPF_MAP_TYPE_ARRAY);
+	__uint(max_entries, 256);
+	__type(key, int);
+	__type(value, struct mptcp_pm_addr_entry);
+} mptcp_pm_list SEC(".maps");
 
 SEC("struct_ops")
 void BPF_PROG(mptcp_pm_init, struct mptcp_sock *msk)
 {
+	struct mptcp_pm_addr_entry *entry;
+	__u32 key = 0;
+
 	bpf_printk("BPF userspace PM (%s)",
 		   CONFIG_MPTCP_IPV6 ? "IPv6" : "IPv4");
+
+	if (!use_bpf_list)
+		return;
+
+	//for (__u32 i = 0; i < MPTCP_SUBFLOWS_MAX; i++) {
+	entry = bpf_map_lookup_elem(&mptcp_pm_list, &key);
+	if (entry)
+		bpf_printk("id=%u", entry->addr.id);
+	//}
 }
 
 SEC("struct_ops")
 void BPF_PROG(mptcp_pm_release, struct mptcp_sock *msk)
 {
+	if (!use_bpf_list)
+		return;
+
+	//bpf_sk_storage_delete(&mptcp_pm_list_map, msk);
 }
 
 static int mptcp_userspace_pm_append_new_local_addr(struct mptcp_sock *msk,
@@ -134,11 +158,35 @@ static int mptcp_pm_remove_id_zero_address(struct mptcp_sock *msk)
 	return err;
 }
 
+#if 0
+static struct mptcp_pm_addr_entry *
+bpf_mptcp_userspace_pm_lookup_addr_by_id(struct mptcp_sock *msk, unsigned int id)
+{
+	struct mptcp_pm_list_storage *storage;
+	struct mptcp_pm_addr_entry *entry;
+
+	storage = bpf_sk_storage_get(&mptcp_pm_list_map, msk, 0, 0);
+	if (!storage)
+		return NULL;
+
+	for (int i = 0; i < storage->nr && i < MPTCP_SUBFLOWS_MAX; i++) {
+		entry = storage->pm_list[i];
+		if (entry->addr.id == id)
+			return entry;
+	}
+
+	return NULL;
+}
+#endif
+
 static struct mptcp_pm_addr_entry *
 mptcp_userspace_pm_lookup_addr_by_id(struct mptcp_sock *msk, unsigned int id)
 {
 	struct mptcp_pm_addr_entry *entry;
 
+	//if (use_bpf_list)
+	//	return bpf_mptcp_userspace_pm_lookup_addr_by_id(msk, id);
+
 	bpf_for_each(mptcp_userspace_pm_addr, entry, msk) {
 		if (entry->addr.id == id)
 			return entry;