Commit a540c81a authored by Stanislav Fomichev's avatar Stanislav Fomichev Committed by Alexei Starovoitov
Browse files

selftests/bpf: Extend bind{4,6} programs with a call to bpf_setsockopt 



To make sure it doesn't trigger sock_owned_by_me splat.

Signed-off-by: default avatarStanislav Fomichev <sdf@google.com>
Signed-off-by: default avatarAlexei Starovoitov <ast@kernel.org>
Acked-by: default avatarAndrii Nakryiko <andrii@kernel.org>
Link: https://lore.kernel.org/bpf/20201202172516.3483656-4-sdf@google.com
parent 427167c0

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

+31 −0
Original line number Diff line number Diff line
@@ -19,6 +19,33 @@ 
#define SERV4_REWRITE_IP	0x7f000001U /* 127.0.0.1 */

#define SERV4_REWRITE_PORT	4444



#ifndef IFNAMSIZ

#define IFNAMSIZ 16

#endif



static __inline int bind_to_device(struct bpf_sock_addr *ctx)

{

	char veth1[IFNAMSIZ] = "test_sock_addr1";

	char veth2[IFNAMSIZ] = "test_sock_addr2";

	char missing[IFNAMSIZ] = "nonexistent_dev";

	char del_bind[IFNAMSIZ] = "";



	if (bpf_setsockopt(ctx, SOL_SOCKET, SO_BINDTODEVICE,

				&veth1, sizeof(veth1)))

		return 1;

	if (bpf_setsockopt(ctx, SOL_SOCKET, SO_BINDTODEVICE,

				&veth2, sizeof(veth2)))

		return 1;

	if (bpf_setsockopt(ctx, SOL_SOCKET, SO_BINDTODEVICE,

				&missing, sizeof(missing)) != -ENODEV)

		return 1;

	if (bpf_setsockopt(ctx, SOL_SOCKET, SO_BINDTODEVICE,

				&del_bind, sizeof(del_bind)))

		return 1;



	return 0;

}



SEC("cgroup/bind4")

int bind_v4_prog(struct bpf_sock_addr *ctx)

{
@@ -62,6 +89,10 @@ int bind_v4_prog(struct bpf_sock_addr *ctx) 
	if (ctx->user_ip4 != user_ip4)

		return 0;



	/* Bind to device and unbind it. */

	if (bind_to_device(ctx))

		return 0;



	ctx->user_ip4 = bpf_htonl(SERV4_REWRITE_IP);

	ctx->user_port = bpf_htons(SERV4_REWRITE_PORT);

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

+31 −0
Original line number Diff line number Diff line
@@ -25,6 +25,33 @@ 
#define SERV6_REWRITE_IP_3	0x00000001

#define SERV6_REWRITE_PORT	6666



#ifndef IFNAMSIZ

#define IFNAMSIZ 16

#endif



static __inline int bind_to_device(struct bpf_sock_addr *ctx)

{

	char veth1[IFNAMSIZ] = "test_sock_addr1";

	char veth2[IFNAMSIZ] = "test_sock_addr2";

	char missing[IFNAMSIZ] = "nonexistent_dev";

	char del_bind[IFNAMSIZ] = "";



	if (bpf_setsockopt(ctx, SOL_SOCKET, SO_BINDTODEVICE,

				&veth1, sizeof(veth1)))

		return 1;

	if (bpf_setsockopt(ctx, SOL_SOCKET, SO_BINDTODEVICE,

				&veth2, sizeof(veth2)))

		return 1;

	if (bpf_setsockopt(ctx, SOL_SOCKET, SO_BINDTODEVICE,

				&missing, sizeof(missing)) != -ENODEV)

		return 1;

	if (bpf_setsockopt(ctx, SOL_SOCKET, SO_BINDTODEVICE,

				&del_bind, sizeof(del_bind)))

		return 1;



	return 0;

}



SEC("cgroup/bind6")

int bind_v6_prog(struct bpf_sock_addr *ctx)

{
@@ -76,6 +103,10 @@ int bind_v6_prog(struct bpf_sock_addr *ctx) 
			return 0;

	}



	/* Bind to device and unbind it. */

	if (bind_to_device(ctx))

		return 0;



	ctx->user_ip6[0] = bpf_htonl(SERV6_REWRITE_IP_0);

	ctx->user_ip6[1] = bpf_htonl(SERV6_REWRITE_IP_1);

	ctx->user_ip6[2] = bpf_htonl(SERV6_REWRITE_IP_2);