On 05/03, Aditi Ghag wrote:
> The helper will be used to programmatically retrieve,
> and pass ports in userspace and kernel selftest programs.
>
> Suggested-by: Stanislav Fomichev <sdf@xxxxxxxxxx>
> Signed-off-by: Aditi Ghag <aditi.ghag@xxxxxxxxxxxxx>
Acked-by: Stanislav Fomichev <sdf@xxxxxxxxxx>
Looks great, thank you!
> ---
> tools/testing/selftests/bpf/network_helpers.c | 23 +++++++++++++++++++
> tools/testing/selftests/bpf/network_helpers.h | 1 +
> 2 files changed, 24 insertions(+)
>
> diff --git a/tools/testing/selftests/bpf/network_helpers.c b/tools/testing/selftests/bpf/network_helpers.c
> index 596caa176582..a105c0cd008a 100644
> --- a/tools/testing/selftests/bpf/network_helpers.c
> +++ b/tools/testing/selftests/bpf/network_helpers.c
> @@ -427,3 +427,26 @@ void close_netns(struct nstoken *token)
> close(token->orig_netns_fd);
> free(token);
> }
> +
> +int get_socket_local_port(int sock_fd)
> +{
> + struct sockaddr_storage addr;
> + socklen_t addrlen = sizeof(addr);
> + int err;
> +
> + err = getsockname(sock_fd, (struct sockaddr *)&addr, &addrlen);
> + if (err < 0)
> + return err;
> +
> + if (addr.ss_family == AF_INET) {
> + struct sockaddr_in *sin = (struct sockaddr_in *)&addr;
> +
> + return sin->sin_port;
> + } else if (addr.ss_family == AF_INET6) {
> + struct sockaddr_in6 *sin = (struct sockaddr_in6 *)&addr;
> +
> + return sin->sin6_port;
> + }
> +
> + return -1;
> +}
> diff --git a/tools/testing/selftests/bpf/network_helpers.h b/tools/testing/selftests/bpf/network_helpers.h
> index f882c691b790..694185644da6 100644
> --- a/tools/testing/selftests/bpf/network_helpers.h
> +++ b/tools/testing/selftests/bpf/network_helpers.h
> @@ -56,6 +56,7 @@ int fastopen_connect(int server_fd, const char *data, unsigned int data_len,
> int make_sockaddr(int family, const char *addr_str, __u16 port,
> struct sockaddr_storage *addr, socklen_t *len);
> char *ping_command(int family);
> +int get_socket_local_port(int sock_fd);
>
> struct nstoken;
> /**
> --
> 2.34.1
>
