mirror of
git://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
synced 2025-05-24 10:39:52 +00:00
c07b4bcd51
In order to prevent mptcpify prog from affecting the running results of other BPF tests, a pid limit was added to restrict it from only modifying its own program. Suggested-by: Martin KaFai Lau <martin.lau@kernel.org> Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn> Signed-off-by: Andrii Nakryiko <andrii@kernel.org> Link: https://lore.kernel.org/bpf/8987e2938e15e8ec390b85b5dcbee704751359dc.1712054986.git.tanggeliang@kylinos.cn
24 lines
538 B
C
24 lines
538 B
C
// SPDX-License-Identifier: GPL-2.0
|
|
/* Copyright (c) 2023, SUSE. */
|
|
|
|
#include "vmlinux.h"
|
|
#include <bpf/bpf_tracing.h>
|
|
#include "bpf_tracing_net.h"
|
|
|
|
char _license[] SEC("license") = "GPL";
|
|
int pid;
|
|
|
|
SEC("fmod_ret/update_socket_protocol")
|
|
int BPF_PROG(mptcpify, int family, int type, int protocol)
|
|
{
|
|
if (bpf_get_current_pid_tgid() >> 32 != pid)
|
|
return protocol;
|
|
|
|
if ((family == AF_INET || family == AF_INET6) &&
|
|
type == SOCK_STREAM &&
|
|
(!protocol || protocol == IPPROTO_TCP)) {
|
|
return IPPROTO_MPTCP;
|
|
}
|
|
|
|
return protocol;
|
|
}
|