forked from multipath-tcp/mptcp_net-next
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
selftests/bpf: Add bpf_stale scheduler & test
This patch implements the setting a subflow as stale/unstale in BPF MPTCP scheduler, named bpf_stale. The staled subflow id will be added into a map in sk_storage. Two helper mptcp_subflow_set_stale() and mptcp_subflow_clear_stale() are added. In this test, subflow 1 is set as stale in bpf_stale_data_init(). Each subflow is checked whether it's a stale one in bpf_stale_get_subflow() to select a unstale subflow to send data. This patch adds the bpf_stale scheduler test: test_stale(). Use sysctl to set net.mptcp.scheduler to use this sched. Add two veth net devices to simulate the multiple addresses case. Use 'ip mptcp endpoint' command to add the new endpoint ADDR_2 to PM netlink. Send data and check bytes_sent of 'ss' output after it to make sure the data has been only sent on ADDR_1 since ADDR_2 is set as stale. Signed-off-by: Geliang Tang <[email protected]>
- Loading branch information
1 parent
666334b
commit 61a9b43
Showing
3 changed files
with
173 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,157 @@ | ||
// SPDX-License-Identifier: GPL-2.0 | ||
/* Copyright (c) 2023, SUSE. */ | ||
/* Copyright (c) 2024, Kylin Software */ | ||
|
||
#include <linux/bpf.h> | ||
#include "bpf_tcp_helpers.h" | ||
|
||
char _license[] SEC("license") = "GPL"; | ||
|
||
struct mptcp_stale_storage { | ||
__u8 nr; | ||
__u32 ids[MPTCP_SUBFLOWS_MAX]; | ||
}; | ||
|
||
struct { | ||
__uint(type, BPF_MAP_TYPE_SK_STORAGE); | ||
__uint(map_flags, BPF_F_NO_PREALLOC); | ||
__type(key, int); | ||
__type(value, struct mptcp_stale_storage); | ||
} mptcp_stale_map SEC(".maps"); | ||
|
||
static void mptcp_subflow_set_stale(struct mptcp_stale_storage *storage, | ||
__u32 subflow_id) | ||
{ | ||
if (!subflow_id) | ||
return; | ||
|
||
for (int i = 0; i < storage->nr && i < MPTCP_SUBFLOWS_MAX; i++) { | ||
if (storage->ids[i] == subflow_id) | ||
return; | ||
} | ||
|
||
if (storage->nr < MPTCP_SUBFLOWS_MAX - 1) | ||
storage->ids[storage->nr++] = subflow_id; | ||
} | ||
|
||
static void mptcp_subflow_clear_stale(struct mptcp_stale_storage *storage, | ||
__u32 subflow_id) | ||
{ | ||
if (!subflow_id) | ||
return; | ||
|
||
for (int i = 0; i < storage->nr && i < MPTCP_SUBFLOWS_MAX; i++) { | ||
if (storage->ids[i] == subflow_id) { | ||
for (int j = i; j < MPTCP_SUBFLOWS_MAX - 1; j++) { | ||
if (!storage->ids[j + 1]) | ||
break; | ||
storage->ids[j] = storage->ids[j + 1]; | ||
storage->ids[j + 1] = 0; | ||
} | ||
storage->nr--; | ||
return; | ||
} | ||
} | ||
} | ||
|
||
static bool mptcp_subflow_is_stale(struct mptcp_stale_storage *storage, | ||
__u32 subflow_id) | ||
{ | ||
for (int i = 0; i < storage->nr && i < MPTCP_SUBFLOWS_MAX; i++) { | ||
if (storage->ids[i] == subflow_id) | ||
return true; | ||
} | ||
|
||
return false; | ||
} | ||
|
||
static bool mptcp_subflow_is_active(struct mptcp_sched_data *data, | ||
__u32 subflow_id) | ||
{ | ||
for (int i = 0; i < data->subflows && i < MPTCP_SUBFLOWS_MAX; i++) { | ||
struct mptcp_subflow_context *subflow; | ||
|
||
subflow = bpf_mptcp_subflow_ctx_by_pos(data, i); | ||
if (!subflow) | ||
break; | ||
if (subflow->subflow_id == subflow_id) | ||
return true; | ||
} | ||
|
||
return false; | ||
} | ||
|
||
SEC("struct_ops/mptcp_sched_stale_init") | ||
void BPF_PROG(mptcp_sched_stale_init, struct mptcp_sock *msk) | ||
{ | ||
struct mptcp_stale_storage *storage; | ||
|
||
storage = bpf_sk_storage_get(&mptcp_stale_map, msk, 0, | ||
BPF_LOCAL_STORAGE_GET_F_CREATE); | ||
if (!storage) | ||
return; | ||
|
||
for (int i = 0; i < MPTCP_SUBFLOWS_MAX; i++) | ||
storage->ids[i] = 0; | ||
storage->nr = 0; | ||
|
||
mptcp_subflow_set_stale(storage, 2); | ||
mptcp_subflow_set_stale(storage, 3); | ||
} | ||
|
||
SEC("struct_ops/mptcp_sched_stale_release") | ||
void BPF_PROG(mptcp_sched_stale_release, struct mptcp_sock *msk) | ||
{ | ||
bpf_sk_storage_delete(&mptcp_stale_map, msk); | ||
} | ||
|
||
int BPF_STRUCT_OPS(bpf_stale_get_subflow, struct mptcp_sock *msk, | ||
struct mptcp_sched_data *data) | ||
{ | ||
struct mptcp_stale_storage *storage; | ||
int nr = -1, i; | ||
|
||
if (data->subflows == 1) { | ||
mptcp_subflow_set_scheduled(bpf_mptcp_subflow_ctx_by_pos(data, 0), true); | ||
return 0; | ||
} | ||
|
||
storage = bpf_sk_storage_get(&mptcp_stale_map, msk, 0, 0); | ||
if (!storage) | ||
return -1; | ||
|
||
/* Handle invalid subflow ids for subflows that have been closed */ | ||
if (data->subflows < storage->nr + 1) { | ||
for (i = 0; i < storage->nr && i < MPTCP_SUBFLOWS_MAX; i++) { | ||
if (!mptcp_subflow_is_active(data, storage->ids[i])) | ||
mptcp_subflow_clear_stale(storage, storage->ids[i]); | ||
} | ||
} | ||
|
||
for (i = 0; i < data->subflows && i < MPTCP_SUBFLOWS_MAX; i++) { | ||
struct mptcp_subflow_context *subflow; | ||
|
||
subflow = bpf_mptcp_subflow_ctx_by_pos(data, i); | ||
if (!subflow) | ||
break; | ||
|
||
if (mptcp_subflow_is_stale(storage, subflow->subflow_id)) | ||
continue; | ||
|
||
nr = i; | ||
} | ||
|
||
if (nr != -1) { | ||
mptcp_subflow_set_scheduled(bpf_mptcp_subflow_ctx_by_pos(data, nr), true); | ||
return -1; | ||
} | ||
return 0; | ||
} | ||
|
||
SEC(".struct_ops") | ||
struct mptcp_sched_ops stale = { | ||
.init = (void *)mptcp_sched_stale_init, | ||
.release = (void *)mptcp_sched_stale_release, | ||
.get_subflow = (void *)bpf_stale_get_subflow, | ||
.name = "bpf_stale", | ||
}; |