Commit 6a3a3dcc authored by Stanislav Fomichev's avatar Stanislav Fomichev Committed by Andrii Nakryiko
Browse files

selftests/bpf: Verify bpf_get_netns_cookie in BPF_PROG_TYPE_CGROUP_SOCKOPT 



Add extra calls to sockopt_sk.c.

Signed-off-by: default avatarStanislav Fomichev <sdf@google.com>
Signed-off-by: default avatarAndrii Nakryiko <andrii@kernel.org>
Acked-by: default avatarMartin KaFai Lau <kafai@fb.com>
Link: https://lore.kernel.org/bpf/20210813230530.333779-3-sdf@google.com
parent f1248dee

tools/testing/selftests/bpf/progs/sockopt_sk.c

+16 −0
Original line number Diff line number Diff line
@@ -33,6 +33,14 @@ int _getsockopt(struct bpf_sockopt *ctx) 
	__u8 *optval = ctx->optval;

	struct sockopt_sk *storage;



	/* Make sure bpf_get_netns_cookie is callable.

	 */

	if (bpf_get_netns_cookie(NULL) == 0)

		return 0;



	if (bpf_get_netns_cookie(ctx) == 0)

		return 0;



	if (ctx->level == SOL_IP && ctx->optname == IP_TOS) {

		/* Not interested in SOL_IP:IP_TOS;

		 * let next BPF program in the cgroup chain or kernel
@@ -123,6 +131,14 @@ int _setsockopt(struct bpf_sockopt *ctx) 
	__u8 *optval = ctx->optval;

	struct sockopt_sk *storage;



	/* Make sure bpf_get_netns_cookie is callable.

	 */

	if (bpf_get_netns_cookie(NULL) == 0)

		return 0;



	if (bpf_get_netns_cookie(ctx) == 0)

		return 0;



	if (ctx->level == SOL_IP && ctx->optname == IP_TOS) {

		/* Not interested in SOL_IP:IP_TOS;

		 * let next BPF program in the cgroup chain or kernel