The branch main has been updated by kp:

URL: 
https://cgit.FreeBSD.org/src/commit/?id=8aa098b469144e56517b1e59302d918151cd04a1

commit 8aa098b469144e56517b1e59302d918151cd04a1
Author:     Kristof Provost <k...@freebsd.org>
AuthorDate: 2025-02-11 15:09:53 +0000
Commit:     Kristof Provost <k...@freebsd.org>
CommitDate: 2025-02-19 10:41:09 +0000

    pf: pf_send_tcp: ask the stack to do the cksum instead of doing it manually
    
    ok benno lteo naddy (back in january)
    
    Obtained from:  OpenBSD, henning <henn...@openbsd.org>, 674c6722fd
    Sponsored by:   Rubicon Communications, LLC ("Netgate")
---
 sys/netpfil/pf/pf.c | 55 ++++++++++++++++++++++++-----------------------------
 1 file changed, 25 insertions(+), 30 deletions(-)

diff --git a/sys/netpfil/pf/pf.c b/sys/netpfil/pf/pf.c
index da741364d372..2a1ee73b3aef 100644
--- a/sys/netpfil/pf/pf.c
+++ b/sys/netpfil/pf/pf.c
@@ -2354,7 +2354,9 @@ pf_intr(void *v)
 
                                pfse->pfse_m->m_flags |= M_SKIP_FIREWALL;
                                pfse->pfse_m->m_pkthdr.csum_flags |=
-                                   CSUM_IP_VALID | CSUM_IP_CHECKED;
+                                   CSUM_IP_VALID | CSUM_IP_CHECKED |
+                                   CSUM_DATA_VALID | CSUM_PSEUDO_HDR;
+                               pfse->pfse_m->m_pkthdr.csum_data = 0xffff;
                                ip_input(pfse->pfse_m);
                        } else {
                                ip_output(pfse->pfse_m, NULL, NULL, 0, NULL,
@@ -2375,6 +2377,9 @@ pf_intr(void *v)
 
                                pfse->pfse_m->m_flags |= M_SKIP_FIREWALL |
                                    M_LOOP;
+                               pfse->pfse_m->m_pkthdr.csum_flags |=
+                                   CSUM_DATA_VALID | CSUM_PSEUDO_HDR;
+                               pfse->pfse_m->m_pkthdr.csum_data = 0xffff;
                                ip6_input(pfse->pfse_m);
                        } else {
                                ip6_output(pfse->pfse_m, NULL, NULL, 0, NULL,
@@ -4003,28 +4008,46 @@ pf_build_tcp(const struct pf_krule *r, sa_family_t af,
        switch (af) {
 #ifdef INET
        case AF_INET:
+               m->m_pkthdr.csum_flags |= CSUM_TCP;
+               m->m_pkthdr.csum_data = offsetof(struct tcphdr, th_sum);
+
                h = mtod(m, struct ip *);
 
-               /* IP header fields included in the TCP checksum */
                h->ip_p = IPPROTO_TCP;
                h->ip_len = htons(tlen);
+               h->ip_v = 4;
+               h->ip_hl = sizeof(*h) >> 2;
+               h->ip_tos = IPTOS_LOWDELAY;
+               h->ip_len = htons(len);
+               h->ip_off = htons(V_path_mtu_discovery ? IP_DF : 0);
+               h->ip_ttl = ttl ? ttl : V_ip_defttl;
+               h->ip_sum = 0;
                h->ip_src.s_addr = saddr->v4.s_addr;
                h->ip_dst.s_addr = daddr->v4.s_addr;
 
                th = (struct tcphdr *)((caddr_t)h + sizeof(struct ip));
+               th->th_sum = in_pseudo(h->ip_src.s_addr, h->ip_dst.s_addr,
+                   htons(len - sizeof(struct ip) + IPPROTO_TCP));
                break;
 #endif /* INET */
 #ifdef INET6
        case AF_INET6:
+               m->m_pkthdr.csum_flags |= CSUM_TCP_IPV6;
+               m->m_pkthdr.csum_data = offsetof(struct tcphdr, th_sum);
+
                h6 = mtod(m, struct ip6_hdr *);
 
                /* IP header fields included in the TCP checksum */
                h6->ip6_nxt = IPPROTO_TCP;
                h6->ip6_plen = htons(tlen);
+               h6->ip6_vfc |= IPV6_VERSION;
+               h6->ip6_hlim = V_ip6_defhlim;
                memcpy(&h6->ip6_src, &saddr->v6, sizeof(struct in6_addr));
                memcpy(&h6->ip6_dst, &daddr->v6, sizeof(struct in6_addr));
 
                th = (struct tcphdr *)((caddr_t)h6 + sizeof(struct ip6_hdr));
+               th->th_sum = in6_cksum_pseudo(h6, len - sizeof(struct ip6_hdr),
+                   IPPROTO_TCP, 0);
                break;
 #endif /* INET6 */
        }
@@ -4046,34 +4069,6 @@ pf_build_tcp(const struct pf_krule *r, sa_family_t af,
                memcpy((opt + 2), &mss, 2);
        }
 
-       switch (af) {
-#ifdef INET
-       case AF_INET:
-               /* TCP checksum */
-               th->th_sum = in_cksum(m, len);
-
-               /* Finish the IP header */
-               h->ip_v = 4;
-               h->ip_hl = sizeof(*h) >> 2;
-               h->ip_tos = IPTOS_LOWDELAY;
-               h->ip_off = htons(V_path_mtu_discovery ? IP_DF : 0);
-               h->ip_len = htons(len);
-               h->ip_ttl = ttl ? ttl : V_ip_defttl;
-               h->ip_sum = 0;
-               break;
-#endif /* INET */
-#ifdef INET6
-       case AF_INET6:
-               /* TCP checksum */
-               th->th_sum = in6_cksum(m, IPPROTO_TCP,
-                   sizeof(struct ip6_hdr), tlen);
-
-               h6->ip6_vfc |= IPV6_VERSION;
-               h6->ip6_hlim = IPV6_DEFHLIM;
-               break;
-#endif /* INET6 */
-       }
-
        return (m);
 }
 

Reply via email to