Commit e15a2209 authored by Daniel Xu's avatar Daniel Xu Committed by Alexei Starovoitov
Browse files

bpf: selftests: Support custom type and proto for client sockets 



Extend connect_to_fd_opts() to take optional type and protocol
parameters for the client socket. These parameters are useful when
opening a raw socket to send IP fragments.

Signed-off-by: default avatarDaniel Xu <dxu@dxuuu.xyz>
Link: https://lore.kernel.org/r/9067db539efdfd608aa86a2b143c521337c111fc.1689970773.git.dxu@dxuuu.xyz


Signed-off-by: default avatarAlexei Starovoitov <ast@kernel.org>
parent 3495e89c

tools/testing/selftests/bpf/network_helpers.c

+15 −6
Original line number Diff line number Diff line
@@ -270,15 +270,24 @@ int connect_to_fd_opts(int server_fd, const struct network_helper_opts *opts) 
		opts = &default_opts;



	optlen = sizeof(type);



	if (opts->type) {

		type = opts->type;

	} else {

		if (getsockopt(server_fd, SOL_SOCKET, SO_TYPE, &type, &optlen)) {

			log_err("getsockopt(SOL_TYPE)");

			return -1;

		}

	}



	if (opts->proto) {

		protocol = opts->proto;

	} else {

		if (getsockopt(server_fd, SOL_SOCKET, SO_PROTOCOL, &protocol, &optlen)) {

			log_err("getsockopt(SOL_PROTOCOL)");

			return -1;

		}

	}



	addrlen = sizeof(addr);

	if (getsockname(server_fd, (struct sockaddr *)&addr, &addrlen)) {

tools/testing/selftests/bpf/network_helpers.h

+2 −0
Original line number Diff line number Diff line
@@ -22,6 +22,8 @@ struct network_helper_opts { 
	int timeout_ms;

	bool must_fail;

	bool noconnect;

	int type;

	int proto;

};



/* ipv4 test vector */