From: David Gibson <david@gibson.dropbear.id.au>
To: Stefano Brivio <sbrivio@redhat.com>,
Laurent Vivier <lvivier@redhat.com>,
passt-dev@passt.top
Cc: David Gibson <david@gibson.dropbear.id.au>
Subject: [PATCH 6/7] tcp: Merge tcp_update_check_tcp[46]()
Date: Mon, 28 Oct 2024 20:40:49 +1100 [thread overview]
Message-ID: <20241028094050.1609090-7-david@gibson.dropbear.id.au> (raw)
In-Reply-To: <20241028094050.1609090-1-david@gibson.dropbear.id.au>
The only reason we need separate functions for the IPv4 and IPv6 case is
to calculate the checksum of the IP pseudo-header, which is different for
the two cases. However, the caller already knows which path it's on and
can access the values needed for the pseudo-header partial sum more easily
than tcp_update_check_tcp[46]() can.
So, merge these functions into a single tcp_update_csum() function that
just takes the pseudo-header partial sum, calculated in the caller.
Signed-off-by: David Gibson <david@gibson.dropbear.id.au>
---
tcp.c | 65 ++++++++++++++++++++---------------------------------------
1 file changed, 22 insertions(+), 43 deletions(-)
diff --git a/tcp.c b/tcp.c
index 787bc19..e9f62a4 100644
--- a/tcp.c
+++ b/tcp.c
@@ -751,50 +751,20 @@ static void tcp_sock_set_bufsize(const struct ctx *c, int s)
}
/**
- * tcp_update_check_tcp4() - Calculate TCP checksum for IPv4
- * @iph: IPv4 header
+ * tcp_update_csum() - Calculate TCP checksum
+ * @psum: Unfolded partial checksum of the IPv4 or IPv6 pseudo-header
* @th: TCP header (updated)
* @iov: IO vector containing the TCP payload
* @iov_cnt: Length of @iov
* @doffset: TCP payload offset in @iov
*/
-static void tcp_update_check_tcp4(const struct iphdr *iph, struct tcphdr *th,
- const struct iovec *iov, int iov_cnt,
- size_t doffset)
+static void tcp_update_csum(uint32_t psum, struct tcphdr *th,
+ const struct iovec *iov, int iov_cnt,
+ size_t doffset)
{
- 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;
-
- sum = proto_ipv4_header_psum(l4len, IPPROTO_TCP, saddr, daddr);
-
- th->check = 0;
- sum = csum_unfolded(th, sizeof(*th), sum);
- th->check = csum_iov(iov, iov_cnt, doffset, sum);
-}
-
-/**
- * tcp_update_check_tcp6() - Calculate TCP checksum for IPv6
- * @ip6h: IPv6 header
- * @th: TCP header (updated)
- * @iov: IO vector containing the TCP payload
- * @iov_cnt: Length of @iov
- * @doffset: TCP payload offset in @iov
- */
-static void tcp_update_check_tcp6(const struct ipv6hdr *ip6h, struct tcphdr *th,
- const struct iovec *iov, int iov_cnt,
- size_t doffset)
-{
- uint16_t l4len = ntohs(ip6h->payload_len);
- uint32_t sum;
-
- sum = proto_ipv6_header_psum(l4len, IPPROTO_TCP, &ip6h->saddr,
- &ip6h->daddr);
-
th->check = 0;
- sum = csum_unfolded(th, sizeof(*th), sum);
- th->check = csum_iov(iov, iov_cnt, doffset, sum);
+ psum = csum_unfolded(th, sizeof(*th), psum);
+ th->check = csum_iov(iov, iov_cnt, doffset, psum);
}
/**
@@ -946,10 +916,14 @@ void tcp_fill_headers4(const struct tcp_tap_conn *conn,
tcp_fill_header(th, conn, seq);
- if (no_tcp_csum)
+ if (no_tcp_csum) {
th->check = 0;
- else
- tcp_update_check_tcp4(iph, th, iov, iov_cnt, doffset);
+ } else {
+ uint32_t psum = proto_ipv4_header_psum(l4len, IPPROTO_TCP,
+ *src4, *dst4);
+
+ tcp_update_csum(psum, th, iov, iov_cnt, doffset);
+ }
tap_hdr_update(taph, l3len + sizeof(struct ethhdr));
}
@@ -991,10 +965,15 @@ void tcp_fill_headers6(const struct tcp_tap_conn *conn,
tcp_fill_header(th, conn, seq);
- if (no_tcp_csum)
+ if (no_tcp_csum) {
th->check = 0;
- else
- tcp_update_check_tcp6(ip6h, th, iov, iov_cnt, doffset);
+ } else {
+ uint32_t psum = proto_ipv6_header_psum(l4len, IPPROTO_TCP,
+ &ip6h->saddr,
+ &ip6h->daddr);
+
+ tcp_update_csum(psum, th, iov, iov_cnt, doffset);
+ }
tap_hdr_update(taph, l4len + sizeof(*ip6h) + sizeof(struct ethhdr));
}
--
@@ -751,50 +751,20 @@ static void tcp_sock_set_bufsize(const struct ctx *c, int s)
}
/**
- * tcp_update_check_tcp4() - Calculate TCP checksum for IPv4
- * @iph: IPv4 header
+ * tcp_update_csum() - Calculate TCP checksum
+ * @psum: Unfolded partial checksum of the IPv4 or IPv6 pseudo-header
* @th: TCP header (updated)
* @iov: IO vector containing the TCP payload
* @iov_cnt: Length of @iov
* @doffset: TCP payload offset in @iov
*/
-static void tcp_update_check_tcp4(const struct iphdr *iph, struct tcphdr *th,
- const struct iovec *iov, int iov_cnt,
- size_t doffset)
+static void tcp_update_csum(uint32_t psum, struct tcphdr *th,
+ const struct iovec *iov, int iov_cnt,
+ size_t doffset)
{
- 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;
-
- sum = proto_ipv4_header_psum(l4len, IPPROTO_TCP, saddr, daddr);
-
- th->check = 0;
- sum = csum_unfolded(th, sizeof(*th), sum);
- th->check = csum_iov(iov, iov_cnt, doffset, sum);
-}
-
-/**
- * tcp_update_check_tcp6() - Calculate TCP checksum for IPv6
- * @ip6h: IPv6 header
- * @th: TCP header (updated)
- * @iov: IO vector containing the TCP payload
- * @iov_cnt: Length of @iov
- * @doffset: TCP payload offset in @iov
- */
-static void tcp_update_check_tcp6(const struct ipv6hdr *ip6h, struct tcphdr *th,
- const struct iovec *iov, int iov_cnt,
- size_t doffset)
-{
- uint16_t l4len = ntohs(ip6h->payload_len);
- uint32_t sum;
-
- sum = proto_ipv6_header_psum(l4len, IPPROTO_TCP, &ip6h->saddr,
- &ip6h->daddr);
-
th->check = 0;
- sum = csum_unfolded(th, sizeof(*th), sum);
- th->check = csum_iov(iov, iov_cnt, doffset, sum);
+ psum = csum_unfolded(th, sizeof(*th), psum);
+ th->check = csum_iov(iov, iov_cnt, doffset, psum);
}
/**
@@ -946,10 +916,14 @@ void tcp_fill_headers4(const struct tcp_tap_conn *conn,
tcp_fill_header(th, conn, seq);
- if (no_tcp_csum)
+ if (no_tcp_csum) {
th->check = 0;
- else
- tcp_update_check_tcp4(iph, th, iov, iov_cnt, doffset);
+ } else {
+ uint32_t psum = proto_ipv4_header_psum(l4len, IPPROTO_TCP,
+ *src4, *dst4);
+
+ tcp_update_csum(psum, th, iov, iov_cnt, doffset);
+ }
tap_hdr_update(taph, l3len + sizeof(struct ethhdr));
}
@@ -991,10 +965,15 @@ void tcp_fill_headers6(const struct tcp_tap_conn *conn,
tcp_fill_header(th, conn, seq);
- if (no_tcp_csum)
+ if (no_tcp_csum) {
th->check = 0;
- else
- tcp_update_check_tcp6(ip6h, th, iov, iov_cnt, doffset);
+ } else {
+ uint32_t psum = proto_ipv6_header_psum(l4len, IPPROTO_TCP,
+ &ip6h->saddr,
+ &ip6h->daddr);
+
+ tcp_update_csum(psum, th, iov, iov_cnt, doffset);
+ }
tap_hdr_update(taph, l4len + sizeof(*ip6h) + sizeof(struct ethhdr));
}
--
2.47.0
next prev parent reply other threads:[~2024-10-28 9:41 UTC|newest]
Thread overview: 14+ messages / expand[flat|nested] mbox.gz Atom feed top
2024-10-28 9:40 [PATCH 0/7] Rework some IOV handling in TCP code David Gibson
2024-10-28 9:40 ` [PATCH 1/7] tcp: Pass TCP header and payload separately to tcp_update_check_tcp[46]() David Gibson
2024-10-28 18:42 ` Stefano Brivio
2024-10-29 3:02 ` David Gibson
2024-10-29 4:07 ` David Gibson
2024-10-29 9:09 ` Stefano Brivio
2024-10-29 9:26 ` David Gibson
2024-10-29 10:32 ` Stefano Brivio
2024-10-28 9:40 ` [PATCH 2/7] tcp: Move tcp_l2_buf_fill_headers() to tcp_buf.c David Gibson
2024-10-28 9:40 ` [PATCH 3/7] tcp: Rework tcp_l2_buf_fill_headers() into tcp_buf_make_frame() David Gibson
2024-10-28 9:40 ` [PATCH 4/7] tcp: Don't use return value from tcp_fill_headers[46] to adjust iov_len David Gibson
2024-10-28 9:40 ` [PATCH 5/7] tcp: Pass TCP header and payload separately to tcp_fill_headers[46]() David Gibson
2024-10-28 9:40 ` David Gibson [this message]
2024-10-28 9:40 ` [PATCH 7/7] tcp: Fold tcp_update_csum() into tcp_fill_header() 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=20241028094050.1609090-7-david@gibson.dropbear.id.au \
--to=david@gibson.dropbear.id.au \
--cc=lvivier@redhat.com \
--cc=passt-dev@passt.top \
--cc=sbrivio@redhat.com \
/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).