Commit f7c917ba authored by brakmo's avatar brakmo Committed by Alexei Starovoitov
Browse files

bpf: add bpf helper bpf_skb_ecn_set_ce 



This patch adds a new bpf helper BPF_FUNC_skb_ecn_set_ce
"int bpf_skb_ecn_set_ce(struct sk_buff *skb)". It is added to
BPF_PROG_TYPE_CGROUP_SKB typed bpf_prog which currently can
be attached to the ingress and egress path. The helper is needed
because his type of bpf_prog cannot modify the skb directly.

This helper is used to set the ECN field of ECN capable IP packets to ce
(congestion encountered) in the IPv6 or IPv4 header of the skb. It can be
used by a bpf_prog to manage egress or ingress network bandwdith limit
per cgroupv2 by inducing an ECN response in the TCP sender.
This works best when using DCTCP.

Signed-off-by: default avatarLawrence Brakmo <brakmo@fb.com>
Signed-off-by: default avatarMartin KaFai Lau <kafai@fb.com>
Acked-by: default avatarSong Liu <songliubraving@fb.com>
Signed-off-by: default avatarAlexei Starovoitov <ast@kernel.org>
parent b74e21ab

include/uapi/linux/bpf.h

+9 −1
Original line number Diff line number Diff line
@@ -2359,6 +2359,13 @@ union bpf_attr { 
 *	Return

 *		A **struct bpf_tcp_sock** pointer on success, or NULL in

 *		case of failure.

 *

 * int bpf_skb_ecn_set_ce(struct sk_buf *skb)

 *     Description

 *             Sets ECN of IP header to ce (congestion encountered) if

 *             current value is ect (ECN capable). Works with IPv6 and IPv4.

 *     Return

 *             1 if set, 0 if not set.

 */

#define __BPF_FUNC_MAPPER(FN)		\

	FN(unspec),			\
@@ -2457,7 +2464,8 @@ union bpf_attr { 
	FN(spin_lock),			\

	FN(spin_unlock),		\

	FN(sk_fullsock),		\

	FN(tcp_sock),

	FN(tcp_sock),			\

	FN(skb_ecn_set_ce),



/* integer value in 'imm' field of BPF_CALL instruction selects which helper

 * function eBPF program intends to call

net/core/filter.c

+28 −0
Original line number Diff line number Diff line
@@ -5426,6 +5426,32 @@ static const struct bpf_func_proto bpf_tcp_sock_proto = { 
	.arg1_type	= ARG_PTR_TO_SOCK_COMMON,

};



BPF_CALL_1(bpf_skb_ecn_set_ce, struct sk_buff *, skb)

{

	unsigned int iphdr_len;



	if (skb->protocol == cpu_to_be16(ETH_P_IP))

		iphdr_len = sizeof(struct iphdr);

	else if (skb->protocol == cpu_to_be16(ETH_P_IPV6))

		iphdr_len = sizeof(struct ipv6hdr);

	else

		return 0;



	if (skb_headlen(skb) < iphdr_len)

		return 0;



	if (skb_cloned(skb) && !skb_clone_writable(skb, iphdr_len))

		return 0;



	return INET_ECN_set_ce(skb);

}



static const struct bpf_func_proto bpf_skb_ecn_set_ce_proto = {

	.func           = bpf_skb_ecn_set_ce,

	.gpl_only       = false,

	.ret_type       = RET_INTEGER,

	.arg1_type      = ARG_PTR_TO_CTX,

};

#endif /* CONFIG_INET */



bool bpf_helper_changes_pkt_data(void *func)
@@ -5585,6 +5611,8 @@ cg_skb_func_proto(enum bpf_func_id func_id, const struct bpf_prog *prog) 
#ifdef CONFIG_INET

	case BPF_FUNC_tcp_sock:

		return &bpf_tcp_sock_proto;

	case BPF_FUNC_skb_ecn_set_ce:

		return &bpf_skb_ecn_set_ce_proto;

#endif

	default:

		return sk_filter_func_proto(func_id, prog);

CRA Git | Maintained and supported by SUSTech CRA and CCSE