External/passt
1
0
Fork 0
mirror of https://passt.top/passt synced 2024-12-22 13:45:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

tcp: Pass TCP header and payload separately to tcp_update_check_tcp[46]()

Browse Source 
Currently these expects both the TCP header and payload in a single IOV,
and goes to some trouble to locate the checksum field within it.  In the
current caller we've already know where the TCP header is, so we might as
well just pass it in.  This will need to work a bit differently for
vhost-user, but that code already needs to locate the TCP header for other
reasons, so again we can just pass it in.

Signed-off-by: David Gibson <david@gibson.dropbear.id.au>
Signed-off-by: Stefano Brivio <sbrivio@redhat.com>
This commit is contained in:
David Gibson 2024-11-27 14:54:06 +11:00 committed by Stefano Brivio
parent 67151090bc
commit 2ee07697c4
3 changed files with 34 additions and 94 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

106
tcp.c
View File

@ -755,106 +755,42 @@ static void tcp_sock_set_bufsize(const struct ctx *c, int s)
/** /**
* tcp_update_check_tcp4() - Calculate TCP checksum for IPv4 * tcp_update_check_tcp4() - Calculate TCP checksum for IPv4
* @iph: IPv4 header * @iph: IPv4 header
* @iov: Pointer to the array of IO vectors * @th: TCP header (updated)
* @iov_cnt: Length of the array * @payload: TCP payload
* @l4offset: IPv4 payload offset in the iovec array
*/ */
void tcp_update_check_tcp4(const struct iphdr *iph, void tcp_update_check_tcp4(const struct iphdr *iph, struct tcphdr *th,
const struct iovec *iov, int iov_cnt, struct iov_tail *payload)
size_t l4offset)
{ {
uint16_t l4len = ntohs(iph->tot_len) - sizeof(struct iphdr); uint16_t l4len = ntohs(iph->tot_len) - sizeof(struct iphdr);
struct iov_tail l4 = IOV_TAIL(iov, iov_cnt, l4offset);
struct in_addr saddr = { .s_addr = iph->saddr }; struct in_addr saddr = { .s_addr = iph->saddr };
struct in_addr daddr = { .s_addr = iph->daddr }; struct in_addr daddr = { .s_addr = iph->daddr };
size_t check_ofs;
uint16_t *check;
int check_idx;
uint32_t sum; uint32_t sum;
char *ptr;
sum = proto_ipv4_header_psum(l4len, IPPROTO_TCP, saddr, daddr); sum = proto_ipv4_header_psum(l4len, IPPROTO_TCP, saddr, daddr);
check_idx = iov_skip_bytes(iov, iov_cnt, th->check = 0;
l4offset + offsetof(struct tcphdr, check), sum = csum_unfolded(th, sizeof(*th), sum);
&check_ofs); th->check = csum_iov_tail(payload, sum);
if (check_idx >= iov_cnt) {
err("TCP4 buffer is too small, iov size %zd, check offset %zd",
iov_size(iov, iov_cnt),
l4offset + offsetof(struct tcphdr, check));
return;
}
if (check_ofs + sizeof(*check) > iov[check_idx].iov_len) {
err("TCP4 checksum field memory is not contiguous "
"check_ofs %zd check_idx %d iov_len %zd",
check_ofs, check_idx, iov[check_idx].iov_len);
return;
}
ptr = (char *)iov[check_idx].iov_base + check_ofs;
if ((uintptr_t)ptr & (__alignof__(*check) - 1)) {
err("TCP4 checksum field is not correctly aligned in memory");
return;
}
check = (uint16_t *)ptr;
*check = 0;
*check = csum_iov_tail(&l4, sum);
} }
/** /**
* tcp_update_check_tcp6() - Calculate TCP checksum for IPv6 * tcp_update_check_tcp6() - Calculate TCP checksum for IPv6
* @ip6h: IPv6 header * @ip6h: IPv6 header
* @iov: Pointer to the array of IO vectors * @th: TCP header (updated)
* @iov_cnt: Length of the array * @payload: TCP payload
* @l4offset: IPv6 payload offset in the iovec array
*/ */
void tcp_update_check_tcp6(const struct ipv6hdr *ip6h, void tcp_update_check_tcp6(const struct ipv6hdr *ip6h, struct tcphdr *th,
const struct iovec *iov, int iov_cnt, struct iov_tail *payload)
size_t l4offset)
{ {
struct iov_tail l4 = IOV_TAIL(iov, iov_cnt, l4offset);
uint16_t l4len = ntohs(ip6h->payload_len); uint16_t l4len = ntohs(ip6h->payload_len);
size_t check_ofs;
uint16_t *check;
int check_idx;
uint32_t sum; uint32_t sum;
char *ptr;
sum = proto_ipv6_header_psum(l4len, IPPROTO_TCP, &ip6h->saddr, sum = proto_ipv6_header_psum(l4len, IPPROTO_TCP, &ip6h->saddr,
&ip6h->daddr); &ip6h->daddr);
check_idx = iov_skip_bytes(iov, iov_cnt, th->check = 0;
l4offset + offsetof(struct tcphdr, check), sum = csum_unfolded(th, sizeof(*th), sum);
&check_ofs); th->check = csum_iov_tail(payload, sum);
if (check_idx >= iov_cnt) {
err("TCP6 buffer is too small, iov size %zd, check offset %zd",
iov_size(iov, iov_cnt),
l4offset + offsetof(struct tcphdr, check));
return;
}
if (check_ofs + sizeof(*check) > iov[check_idx].iov_len) {
err("TCP6 checksum field memory is not contiguous "
"check_ofs %zd check_idx %d iov_len %zd",
check_ofs, check_idx, iov[check_idx].iov_len);
return;
}
ptr = (char *)iov[check_idx].iov_base + check_ofs;
if ((uintptr_t)ptr & (__alignof__(*check) - 1)) {
err("TCP6 checksum field is not correctly aligned in memory");
return;
}
check = (uint16_t *)ptr;
*check = 0;
*check = csum_iov_tail(&l4, sum);
} }
/** /**
@ -1005,11 +941,12 @@ void tcp_fill_headers4(const struct tcp_tap_conn *conn,
bp->th.check = 0; bp->th.check = 0;
} else { } else {
const struct iovec iov = { const struct iovec iov = {
.iov_base = bp, .iov_base = bp->data,
.iov_len = ntohs(iph->tot_len) - sizeof(struct iphdr), .iov_len = dlen,
}; };
struct iov_tail payload = IOV_TAIL(&iov, 1, 0);
tcp_update_check_tcp4(iph, &iov, 1, 0); tcp_update_check_tcp4(iph, &bp->th, &payload);
} }
tap_hdr_update(taph, l3len + sizeof(struct ethhdr)); tap_hdr_update(taph, l3len + sizeof(struct ethhdr));
@ -1052,11 +989,12 @@ void tcp_fill_headers6(const struct tcp_tap_conn *conn,
bp->th.check = 0; bp->th.check = 0;
} else { } else {
const struct iovec iov = { const struct iovec iov = {
.iov_base = bp, .iov_base = bp->data,
.iov_len = ntohs(ip6h->payload_len) .iov_len = dlen,
}; };
struct iov_tail payload = IOV_TAIL(&iov, 1, 0);
tcp_update_check_tcp6(ip6h, &iov, 1, 0); tcp_update_check_tcp6(ip6h, &bp->th, &payload);
} }
tap_hdr_update(taph, l4len + sizeof(*ip6h) + sizeof(struct ethhdr)); tap_hdr_update(taph, l4len + sizeof(*ip6h) + sizeof(struct ethhdr));

10
tcp_internal.h
View File

@ -162,12 +162,10 @@ void tcp_rst_do(const struct ctx *c, struct tcp_tap_conn *conn);
struct tcp_info_linux; struct tcp_info_linux;
void tcp_update_check_tcp4(const struct iphdr *iph, void tcp_update_check_tcp4(const struct iphdr *iph, struct tcphdr *th,
const struct iovec *iov, int iov_cnt, struct iov_tail *payload);
size_t l4offset); void tcp_update_check_tcp6(const struct ipv6hdr *ip6h, struct tcphdr *th,
void tcp_update_check_tcp6(const struct ipv6hdr *ip6h, struct iov_tail *payload);
const struct iovec *iov, int iov_cnt,
size_t l4offset);
void tcp_fill_headers4(const struct tcp_tap_conn *conn, void tcp_fill_headers4(const struct tcp_tap_conn *conn,
struct tap_hdr *taph, struct iphdr *iph, struct tap_hdr *taph, struct iphdr *iph,
struct tcp_payload_t *bp, size_t dlen, struct tcp_payload_t *bp, size_t dlen,

12
tcp_vu.c
View File

@ -73,15 +73,19 @@ static void tcp_vu_update_check(const struct flowside *tapside,
char *base = iov[0].iov_base; char *base = iov[0].iov_base;
if (inany_v4(&tapside->oaddr)) { if (inany_v4(&tapside->oaddr)) {
struct tcphdr *th = vu_payloadv4(base);
const struct iphdr *iph = vu_ip(base); const struct iphdr *iph = vu_ip(base);
struct iov_tail payload = IOV_TAIL(iov, iov_cnt,
(char *)(th + 1) - base);
tcp_update_check_tcp4(iph, iov, iov_cnt, tcp_update_check_tcp4(iph, th, &payload);
(char *)vu_payloadv4(base) - base);
} else { } else {
struct tcphdr *th = vu_payloadv6(base);
const struct ipv6hdr *ip6h = vu_ip(base); const struct ipv6hdr *ip6h = vu_ip(base);
struct iov_tail payload = IOV_TAIL(iov, iov_cnt,
(char *)(th + 1) - base);
tcp_update_check_tcp6(ip6h, iov, iov_cnt, tcp_update_check_tcp6(ip6h, th, &payload);
(char *)vu_payloadv6(base) - base);
} }
} }