From: Stefano Brivio <sbrivio@redhat.com>
To: Jon Maloy <jmaloy@redhat.com>
Cc: passt-dev@passt.top, lvivier@redhat.com, dgibson@redhat.com
Subject: Re: [PATCH v2 2/2] selftests: add selftest for tcp SO_PEEK_OFF support
Date: Fri, 23 Aug 2024 09:43:49 +0200 [thread overview]
Message-ID: <20240823094349.1f0df608@elisabeth> (raw)
In-Reply-To: <20240823013637.72719-3-jmaloy@redhat.com>
On Thu, 22 Aug 2024 21:36:37 -0400
Jon Maloy <jmaloy@redhat.com> wrote:
> We add a selftest to check that the new feature added in
> commit 05ea491641d3 ("tcp: add support for SO_PEEK_OFF socket option")
> works correctly.
>
> Signed-off-by: Jon Maloy <jmaloy@redhat.com>
Reviewed-by: Stefano Brivio <sbrivio@redhat.com>
Tested-by: Stefano Brivio <sbrivio@redhat.com>
With or without fixing two nits below:
> ---
> v2: Fixed Makefile and improved code after suggestion from Stefano Brivio
> ---
> tools/testing/selftests/net/Makefile | 1 +
> tools/testing/selftests/net/tcp_so_peek_off.c | 181 ++++++++++++++++++
> 2 files changed, 182 insertions(+)
> create mode 100644 tools/testing/selftests/net/tcp_so_peek_off.c
>
> diff --git a/tools/testing/selftests/net/Makefile b/tools/testing/selftests/net/Makefile
> index 8eaffd7a641c..1179e3261bef 100644
> --- a/tools/testing/selftests/net/Makefile
> +++ b/tools/testing/selftests/net/Makefile
> @@ -80,6 +80,7 @@ TEST_PROGS += io_uring_zerocopy_tx.sh
> TEST_GEN_FILES += bind_bhash
> TEST_GEN_PROGS += sk_bind_sendto_listen
> TEST_GEN_PROGS += sk_connect_zero_addr
> +TEST_GEN_PROGS += tcp_so_peek_off
> TEST_PROGS += test_ingress_egress_chaining.sh
> TEST_GEN_PROGS += so_incoming_cpu
> TEST_PROGS += sctp_vrf.sh
> diff --git a/tools/testing/selftests/net/tcp_so_peek_off.c b/tools/testing/selftests/net/tcp_so_peek_off.c
> new file mode 100644
> index 000000000000..8379ea02e3d7
> --- /dev/null
> +++ b/tools/testing/selftests/net/tcp_so_peek_off.c
> @@ -0,0 +1,181 @@
> +// SPDX-License-Identifier: GPL-2.0
> +
> +#include <stdio.h>
> +#include <stdlib.h>
> +#include <string.h>
> +#include <unistd.h>
> +#include <errno.h>
> +#include <sys/types.h>
> +#include <netinet/in.h>
> +#include <arpa/inet.h>
> +#include "../kselftest.h"
> +
> +static char *afstr(int af)
> +{
> + return af == AF_INET ? "TCP/IPv4" : "TCP/IPv6";
> +}
> +
> +int tcp_peek_offset_probe(sa_family_t af)
> +{
> + int optv = 0;
> + int ret = 0;
> + int s;
> +
> + s = socket(af, SOCK_STREAM | SOCK_CLOEXEC, IPPROTO_TCP);
> + if (s < 0) {
> + ksft_perror("Temporary TCP socket creation failed");
> + } else {
> + if (!setsockopt(s, SOL_SOCKET, SO_PEEK_OFF, &optv, sizeof(int)))
> + ret = 1;
> + else
> + printf("%s does not support SO_PEEK_OFF\n", afstr(af));
You could use ksft_print_msg() for this one.
But the naming of those functions is pretty redundant (what else do you
want to do with a 'msg' other than printing it, I wonder), so it might
make the code unnecessarily unreadable.
> + close(s);
> + }
> + return ret;
> +}
> +
> +static void tcp_peek_offset_set(int s, int offset)
> +{
> + if (setsockopt(s, SOL_SOCKET, SO_PEEK_OFF, &offset, sizeof(offset)))
> + ksft_perror("Failed to set SO_PEEK_OFF value\n");
> +}
> +
> +static int tcp_peek_offset_get(int s)
> +{
> + int offset;
> + socklen_t len = sizeof(offset);
> +
> + if (getsockopt(s, SOL_SOCKET, SO_PEEK_OFF, &offset, &len))
> + ksft_perror("Failed to get SO_PEEK_OFF value\n");
> + return offset;
> +}
> +
> +static int tcp_peek_offset_test(sa_family_t af)
> +{
> + union {
> + struct sockaddr sa;
> + struct sockaddr_in a4;
> + struct sockaddr_in6 a6;
> + } a;
> + int res = 0;
> + int s[2] = {0, 0};
> + int recv_sock = 0;
> + int offset = 0;
> + ssize_t len;
> + char buf;
> +
> + memset(&a, 0, sizeof(a));
> + a.sa.sa_family = af;
> +
> + s[0] = socket(af, SOCK_STREAM, IPPROTO_TCP);
> + s[1] = socket(af, SOCK_STREAM | SOCK_NONBLOCK, IPPROTO_TCP);
> +
> + if (s[0] < 0 || s[1] < 0) {
> + ksft_perror("Temporary probe socket creation failed\n");
> + goto out;
> + }
> + if (bind(s[0], &a.sa, sizeof(a)) < 0) {
> + ksft_perror("Temporary probe socket bind() failed\n");
> + goto out;
> + }
> + if (getsockname(s[0], &a.sa, &((socklen_t) { sizeof(a) })) < 0) {
> + ksft_perror("Temporary probe socket getsockname() failed\n");
> + goto out;
> + }
> + if (listen(s[0], 0) < 0) {
> + ksft_perror("Temporary probe socket listen() failed\n");
> + goto out;
> + }
> + if (connect(s[1], &a.sa, sizeof(a)) >= 0 || errno != EINPROGRESS) {
> + ksft_perror("Temporary probe socket connect() failed\n");
> + goto out;
> + }
> + recv_sock = accept(s[0], NULL, NULL);
> + if (recv_sock <= 0) {
> + ksft_perror("Temporary probe socket accept() failed\n");
> + goto out;
> + }
> +
> + /* Some basic tests of getting/setting offset */
> + offset = tcp_peek_offset_get(recv_sock);
> + if (offset != -1) {
> + ksft_perror("Initial value of socket offset not -1\n");
> + goto out;
> + }
> + tcp_peek_offset_set(recv_sock, 0);
> + offset = tcp_peek_offset_get(recv_sock);
> + if (offset != 0) {
> + ksft_perror("Failed to set socket offset to 0\n");
> + goto out;
> + }
> +
> + /* Transfer a message */
> + if (send(s[1], (char *)("ab"), 2, 0) <= 0 || errno != EINPROGRESS) {
> + ksft_perror("Temporary probe socket send() failed\n");
> + goto out;
> + }
> + /* Read first byte */
> + len = recv(recv_sock, &buf, 1, MSG_PEEK);
> + if (len != 1 || buf != 'a') {
> + ksft_perror("Failed to read first byte of message\n");
> + goto out;
> + }
> + offset = tcp_peek_offset_get(recv_sock);
> + if (offset != 1) {
> + ksft_perror("Offset not forwarded correctly at first byte\n");
> + goto out;
> + }
> + /* Try to read beyond last byte */
> + len = recv(recv_sock, &buf, 2, MSG_PEEK);
> + if (len != 1 || buf != 'b') {
> + ksft_perror("Failed to read last byte of message\n");
> + goto out;
> + }
> + offset = tcp_peek_offset_get(recv_sock);
> + if (offset != 2) {
> + ksft_perror("Offset not forwarded correctly at last byte\n");
> + goto out;
> + }
> + /* Flush message */
> + len = recv(recv_sock, NULL, 2, MSG_TRUNC);
> + if (len != 2) {
> + ksft_perror("Failed to flush message\n");
> + goto out;
> + }
> + offset = tcp_peek_offset_get(recv_sock);
> + if (offset != 0) {
> + ksft_perror("Offset not reverted correctly after flush\n");
> + goto out;
> + }
> +
> + printf("%s with MSG_PEEK_OFF works correctly\n", afstr(af));
Same here.
> + res = 1;
> +out:
> + close(recv_sock);
> + close(s[1]);
> + close(s[0]);
> + return res;
> +}
> +
> +int main(void)
> +{
> + int res4, res6;
> +
> + res4 = tcp_peek_offset_probe(AF_INET);
> + res6 = tcp_peek_offset_probe(AF_INET6);
> +
> + if (!res4 && !res6)
> + return KSFT_SKIP;
> +
> + if (res4)
> + res4 = tcp_peek_offset_test(AF_INET);
> +
> + if (res6)
> + res6 = tcp_peek_offset_test(AF_INET6);
> +
> + if (!res4 || !res6)
> + return KSFT_FAIL;
> +
> + return KSFT_PASS;
> +}
> +
--
Stefano
prev parent reply other threads:[~2024-08-23 7:47 UTC|newest]
Thread overview: 6+ messages / expand[flat|nested] mbox.gz Atom feed top
2024-08-23 1:36 [PATCH v2 0/2] Adding SO_PEEK_OFF for TCPv6 Jon Maloy
2024-08-23 1:36 ` [PATCH v2 1/2] tcp: add SO_PEEK_OFF socket option tor TCPv6 Jon Maloy
2024-08-23 7:30 ` David Gibson
2024-08-23 7:43 ` Stefano Brivio
2024-08-23 1:36 ` [PATCH v2 2/2] selftests: add selftest for tcp SO_PEEK_OFF support Jon Maloy
2024-08-23 7:43 ` Stefano Brivio [this message]
Reply instructions:
You may reply publicly to this message via plain-text email
using any one of the following methods:
* Save the following mbox file, import it into your mail client,
and reply-to-all from there: mbox
Avoid top-posting and favor interleaved quoting:
https://en.wikipedia.org/wiki/Posting_style#Interleaved_style
* Reply using the --to, --cc, and --in-reply-to
switches of git-send-email(1):
git send-email \
--in-reply-to=20240823094349.1f0df608@elisabeth \
--to=sbrivio@redhat.com \
--cc=dgibson@redhat.com \
--cc=jmaloy@redhat.com \
--cc=lvivier@redhat.com \
--cc=passt-dev@passt.top \
/path/to/YOUR_REPLY
https://kernel.org/pub/software/scm/git/docs/git-send-email.html
* If your mail client supports setting the In-Reply-To header
via mailto: links, try the mailto: link
Be sure your reply has a Subject: header at the top and a blank line
before the message body.
Code repositories for project(s) associated with this public inbox
https://passt.top/passt
This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox;
as well as URLs for IMAP folder(s).