public inbox for passt-dev@passt.top
 help / color / mirror / code / Atom feed
From: Laurent Vivier <lvivier@redhat.com>
To: passt-dev@passt.top
Cc: Laurent Vivier <lvivier@redhat.com>
Subject: [PATCH 4/4] tcp: Update TCP checksum using an iovec array
Date: Tue, 24 Sep 2024 17:46:42 +0200	[thread overview]
Message-ID: <20240924154642.182857-5-lvivier@redhat.com> (raw)
In-Reply-To: <20240924154642.182857-1-lvivier@redhat.com>

TCP header and payload are supposed to be in the same buffer,
and tcp_update_check_tcp4()/tcp_update_check_tcp6() compute
the checksum from the base address of the header using the
length of the IP payload.

In the future (for vhost-user) we need to dispatch the TCP header and
the TCP payload through several buffers. To be able to manage that, we
provide an iovec array that points to the data of the TCP frame.
We provide also an offset to be able to provide an array that contains
the TCP frame embedded in an lower level frame, and this offset points
to the TCP header inside the iovec array.

Signed-off-by: Laurent Vivier <lvivier@redhat.com>
---
 checksum.c |   1 -
 tcp.c      | 100 ++++++++++++++++++++++++++++++++++++++---------------
 2 files changed, 72 insertions(+), 29 deletions(-)

diff --git a/checksum.c b/checksum.c
index f80db4d309a2..96ccfe2af50b 100644
--- a/checksum.c
+++ b/checksum.c
@@ -503,7 +503,6 @@ uint16_t csum(const void *buf, size_t len, uint32_t init)
  *
  * Return: 16-bit folded, complemented checksum
  */
-/* cppcheck-suppress unusedFunction */
 uint16_t csum_iov(const struct iovec *iov, size_t n, size_t offset,
 		  uint32_t init)
 {
diff --git a/tcp.c b/tcp.c
index c9472d905520..efd4037ed008 100644
--- a/tcp.c
+++ b/tcp.c
@@ -755,36 +755,65 @@ static void tcp_sock_set_bufsize(const struct ctx *c, int s)
 }
 
 /**
- * tcp_update_check_tcp4() - Update TCP checksum from stored one
- * @iph:	IPv4 header
- * @bp:		TCP header followed by TCP payload
- */
-static void tcp_update_check_tcp4(const struct iphdr *iph,
-				  struct tcp_payload_t *bp)
+ * tcp_update_check_tcp4() - Calculate TCP checksum for IPv6
+ * @src:		IPv4 source address
+ * @dst:		IPv4 destination address
+ * @iov:		Pointer to the array of IO vectors
+ * @iov_cnt:		Length of the array
+ * @payload_offset:	IPv4 payload offset in the iovec array
+ */
+void tcp_update_check_tcp4(struct in_addr src,
+			   struct in_addr dst,
+			   const struct iovec *iov, int iov_cnt,
+			   size_t payload_offset)
 {
-	uint16_t l4len = ntohs(iph->tot_len) - sizeof(struct iphdr);
-	struct in_addr saddr = { .s_addr = iph->saddr };
-	struct in_addr daddr = { .s_addr = iph->daddr };
-	uint32_t sum = proto_ipv4_header_psum(l4len, IPPROTO_TCP, saddr, daddr);
+	size_t check_ofs;
+	__sum16 *check;
+	int check_idx;
+	uint32_t sum;
+
+	sum = proto_ipv4_header_psum(iov_size(iov, iov_cnt) - payload_offset,
+				     IPPROTO_TCP, src, dst);
+
+	check_idx = iov_skip_bytes(iov, iov_cnt,
+				payload_offset + offsetof(struct tcphdr, check),
+				&check_ofs);
+
+	check = (__sum16 *)((char *)iov[check_idx].iov_base + check_ofs);
 
-	bp->th.check = 0;
-	bp->th.check = csum(bp, l4len, sum);
+	*check = 0;
+	*check = csum_iov(iov, iov_cnt, payload_offset, sum);
 }
 
 /**
  * tcp_update_check_tcp6() - Calculate TCP checksum for IPv6
- * @ip6h:	IPv6 header
- * @bp:		TCP header followed by TCP payload
- */
-static void tcp_update_check_tcp6(const struct ipv6hdr *ip6h,
-				  struct tcp_payload_t *bp)
+ * @src:		IPv6 source address
+ * @dst:		IPv6 destination address
+ * @iov:		Pointer to the array of IO vectors
+ * @iov_cnt:		Length of the array
+ * @payload_offset:	IPv6 payload offset in the iovec array
+ */
+void tcp_update_check_tcp6(const struct in6_addr *src,
+			   const struct in6_addr *dst,
+			   const struct iovec *iov, int iov_cnt,
+			   size_t payload_offset)
 {
-	uint16_t l4len = ntohs(ip6h->payload_len);
-	uint32_t sum = proto_ipv6_header_psum(l4len, IPPROTO_TCP,
-					      &ip6h->saddr, &ip6h->daddr);
+	size_t check_ofs;
+	__sum16 *check;
+	int check_idx;
+	uint32_t sum;
+
+	sum = proto_ipv6_header_psum(iov_size(iov, iov_cnt) - payload_offset,
+				     IPPROTO_TCP, src, dst);
+
+	check_idx = iov_skip_bytes(iov, iov_cnt,
+				payload_offset + offsetof(struct tcphdr, check),
+				&check_ofs);
+
+	check = (__sum16 *)((char *)iov[check_idx].iov_base + check_ofs);
 
-	bp->th.check = 0;
-	bp->th.check = csum(bp, l4len, sum);
+	*check = 0;
+	*check = csum_iov(iov, iov_cnt, payload_offset, sum);
 }
 
 /**
@@ -935,10 +964,18 @@ static size_t tcp_fill_headers4(const struct tcp_tap_conn *conn,
 
 	tcp_fill_header(&bp->th, conn, seq);
 
-	if (no_tcp_csum)
+	if (no_tcp_csum) {
 		bp->th.check = 0;
-	else
-		tcp_update_check_tcp4(iph, bp);
+	} else {
+		const struct iovec iov = {
+			.iov_base = bp,
+			.iov_len = ntohs(iph->tot_len) - sizeof(struct iphdr),
+		};
+		struct in_addr saddr = { .s_addr = iph->saddr };
+		struct in_addr daddr = { .s_addr = iph->daddr };
+
+		tcp_update_check_tcp4(saddr, daddr, &iov, 1, 0);
+	}
 
 	tap_hdr_update(taph, l3len + sizeof(struct ethhdr));
 
@@ -980,10 +1017,17 @@ static size_t tcp_fill_headers6(const struct tcp_tap_conn *conn,
 
 	tcp_fill_header(&bp->th, conn, seq);
 
-	if (no_tcp_csum)
+	if (no_tcp_csum) {
 		bp->th.check = 0;
-	else
-		tcp_update_check_tcp6(ip6h, bp);
+	} else {
+		const struct iovec iov = {
+			.iov_base = bp,
+			.iov_len = ntohs(ip6h->payload_len)
+		};
+
+		tcp_update_check_tcp6(&ip6h->saddr, &ip6h->daddr,
+				      &iov, 1, 0);
+	}
 
 	tap_hdr_update(taph, l4len + sizeof(*ip6h) + sizeof(struct ethhdr));
 
-- 
@@ -755,36 +755,65 @@ static void tcp_sock_set_bufsize(const struct ctx *c, int s)
 }
 
 /**
- * tcp_update_check_tcp4() - Update TCP checksum from stored one
- * @iph:	IPv4 header
- * @bp:		TCP header followed by TCP payload
- */
-static void tcp_update_check_tcp4(const struct iphdr *iph,
-				  struct tcp_payload_t *bp)
+ * tcp_update_check_tcp4() - Calculate TCP checksum for IPv6
+ * @src:		IPv4 source address
+ * @dst:		IPv4 destination address
+ * @iov:		Pointer to the array of IO vectors
+ * @iov_cnt:		Length of the array
+ * @payload_offset:	IPv4 payload offset in the iovec array
+ */
+void tcp_update_check_tcp4(struct in_addr src,
+			   struct in_addr dst,
+			   const struct iovec *iov, int iov_cnt,
+			   size_t payload_offset)
 {
-	uint16_t l4len = ntohs(iph->tot_len) - sizeof(struct iphdr);
-	struct in_addr saddr = { .s_addr = iph->saddr };
-	struct in_addr daddr = { .s_addr = iph->daddr };
-	uint32_t sum = proto_ipv4_header_psum(l4len, IPPROTO_TCP, saddr, daddr);
+	size_t check_ofs;
+	__sum16 *check;
+	int check_idx;
+	uint32_t sum;
+
+	sum = proto_ipv4_header_psum(iov_size(iov, iov_cnt) - payload_offset,
+				     IPPROTO_TCP, src, dst);
+
+	check_idx = iov_skip_bytes(iov, iov_cnt,
+				payload_offset + offsetof(struct tcphdr, check),
+				&check_ofs);
+
+	check = (__sum16 *)((char *)iov[check_idx].iov_base + check_ofs);
 
-	bp->th.check = 0;
-	bp->th.check = csum(bp, l4len, sum);
+	*check = 0;
+	*check = csum_iov(iov, iov_cnt, payload_offset, sum);
 }
 
 /**
  * tcp_update_check_tcp6() - Calculate TCP checksum for IPv6
- * @ip6h:	IPv6 header
- * @bp:		TCP header followed by TCP payload
- */
-static void tcp_update_check_tcp6(const struct ipv6hdr *ip6h,
-				  struct tcp_payload_t *bp)
+ * @src:		IPv6 source address
+ * @dst:		IPv6 destination address
+ * @iov:		Pointer to the array of IO vectors
+ * @iov_cnt:		Length of the array
+ * @payload_offset:	IPv6 payload offset in the iovec array
+ */
+void tcp_update_check_tcp6(const struct in6_addr *src,
+			   const struct in6_addr *dst,
+			   const struct iovec *iov, int iov_cnt,
+			   size_t payload_offset)
 {
-	uint16_t l4len = ntohs(ip6h->payload_len);
-	uint32_t sum = proto_ipv6_header_psum(l4len, IPPROTO_TCP,
-					      &ip6h->saddr, &ip6h->daddr);
+	size_t check_ofs;
+	__sum16 *check;
+	int check_idx;
+	uint32_t sum;
+
+	sum = proto_ipv6_header_psum(iov_size(iov, iov_cnt) - payload_offset,
+				     IPPROTO_TCP, src, dst);
+
+	check_idx = iov_skip_bytes(iov, iov_cnt,
+				payload_offset + offsetof(struct tcphdr, check),
+				&check_ofs);
+
+	check = (__sum16 *)((char *)iov[check_idx].iov_base + check_ofs);
 
-	bp->th.check = 0;
-	bp->th.check = csum(bp, l4len, sum);
+	*check = 0;
+	*check = csum_iov(iov, iov_cnt, payload_offset, sum);
 }
 
 /**
@@ -935,10 +964,18 @@ static size_t tcp_fill_headers4(const struct tcp_tap_conn *conn,
 
 	tcp_fill_header(&bp->th, conn, seq);
 
-	if (no_tcp_csum)
+	if (no_tcp_csum) {
 		bp->th.check = 0;
-	else
-		tcp_update_check_tcp4(iph, bp);
+	} else {
+		const struct iovec iov = {
+			.iov_base = bp,
+			.iov_len = ntohs(iph->tot_len) - sizeof(struct iphdr),
+		};
+		struct in_addr saddr = { .s_addr = iph->saddr };
+		struct in_addr daddr = { .s_addr = iph->daddr };
+
+		tcp_update_check_tcp4(saddr, daddr, &iov, 1, 0);
+	}
 
 	tap_hdr_update(taph, l3len + sizeof(struct ethhdr));
 
@@ -980,10 +1017,17 @@ static size_t tcp_fill_headers6(const struct tcp_tap_conn *conn,
 
 	tcp_fill_header(&bp->th, conn, seq);
 
-	if (no_tcp_csum)
+	if (no_tcp_csum) {
 		bp->th.check = 0;
-	else
-		tcp_update_check_tcp6(ip6h, bp);
+	} else {
+		const struct iovec iov = {
+			.iov_base = bp,
+			.iov_len = ntohs(ip6h->payload_len)
+		};
+
+		tcp_update_check_tcp6(&ip6h->saddr, &ip6h->daddr,
+				      &iov, 1, 0);
+	}
 
 	tap_hdr_update(taph, l4len + sizeof(*ip6h) + sizeof(struct ethhdr));
 
-- 
2.46.0


  parent reply	other threads:[~2024-09-24 15:46 UTC|newest]

Thread overview: 12+ messages / expand[flat|nested]  mbox.gz  Atom feed  top
2024-09-24 15:46 [PATCH 0/4] tcp: use csum_iov() in tcp_update_check_tcp[4|6]() Laurent Vivier
2024-09-24 15:46 ` [PATCH 1/4] tcp: Use tcp_payload_t rather than tcphdr Laurent Vivier
2024-09-24 15:46 ` [PATCH 2/4] pcap: Add an offset argument in pcap_iov() Laurent Vivier
2024-09-25  1:21   ` David Gibson
2024-09-24 15:46 ` [PATCH 3/4] checksum: Add an offset argument in csum_iov() Laurent Vivier
2024-09-25  0:51   ` David Gibson
2024-09-24 15:46 ` Laurent Vivier [this message]
2024-09-25  1:12   ` [PATCH 4/4] tcp: Update TCP checksum using an iovec array David Gibson
2024-09-25  6:40     ` Laurent Vivier
2024-09-25  7:01       ` David Gibson
2024-09-25  7:27         ` Laurent Vivier
2024-09-25  7:31           ` David Gibson

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=20240924154642.182857-5-lvivier@redhat.com \
    --to=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).