Changeset 28024 in vbox for trunk/src/VBox/Runtime/common/checksum

Timestamp:

Apr 7, 2010 6:34:16 AM (15 years ago)

Author:

vboxsync

svn:sync-xref-src-repo-rev:

59731

Message:

iprt: ipv6 header and pseudo checksumming; tcp flags; tcp/udp checksumming additions.

Location:

trunk/src/VBox/Runtime/common/checksum

Files:

• ipv4.cpp (modified) (7 diffs)
• ipv6.cpp (added)

trunk/src/VBox/Runtime/common/checksum/ipv4.cpp

@@ -99 +99 @@
  * @param   cbPktMax    The max IP packet size, IP header and payload. This doesn't have
  *                      to be mapped following pIpHdr.
- */
-RTDECL(bool) RTNetIPv4IsHdrValid(PCRTNETIPV4 pIpHdr, size_t cbHdrMax, size_t cbPktMax)
+ * @param   fChecksum   Whether to validate the checksum (GSO).
+ */
+RTDECL(bool) RTNetIPv4IsHdrValid(PCRTNETIPV4 pIpHdr, size_t cbHdrMax, size_t cbPktMax, bool fChecksum)
 {
     /*
@@ -121 +122 @@
 
     /*
-     * The header checksum.
+     * The header checksum if requested.
      */
-    uint16_t u16Sum = RTNetIPv4HdrChecksum(pIpHdr);
-    if (RT_UNLIKELY(pIpHdr->ip_sum != u16Sum))
-        return false;
+    if (fChecksum)
+    {
+        uint16_t u16Sum = RTNetIPv4HdrChecksum(pIpHdr);
+        if (RT_UNLIKELY(pIpHdr->ip_sum != u16Sum))
+            return false;
+    }
     return true;
 }
@@ -386 +390 @@
 
 /**
+ * Calculates the checksum for the UDP header given the UDP header w/ payload
+ * and the checksum of the pseudo header.
+ *
+ * @returns The checksum (network endian).
+ * @param   u32Sum          The checksum of the pseudo header. See
+ *                          RTNetIPv4PseudoChecksum and RTNetIPv6PseudoChecksum.
+ * @param   pUdpHdr         Pointer to the UDP header and the payload, in
+ *                          network endian (big).  We use the uh_ulen field to
+ *                          figure out how much to checksum.
+ */
+RTDECL(uint16_t) RTNetUDPChecksum(uint32_t u32Sum, PCRTNETUDP pUdpHdr)
+{
+    u32Sum = rtNetIPv4AddUDPChecksum(pUdpHdr, u32Sum);
+    bool fOdd = false;
+    u32Sum = rtNetIPv4AddDataChecksum(pUdpHdr + 1, RT_BE2H_U16(pUdpHdr->uh_ulen) - sizeof(*pUdpHdr), u32Sum, &fOdd);
+    return rtNetIPv4FinalizeChecksum(u32Sum);
+}
+RT_EXPORT_SYMBOL(RTNetUDPChecksum);
+
+
+/**
  * Calculates the checksum for the UDP header given the IP header,
  * UDP header and payload.
@@ -458 +483 @@
  *                          and that cbPktMax - sizeof(RTNETUDP) is mapped here.
  * @param   cbPktMax        The max UDP packet size, UDP header and payload (data).
- */
-RTDECL(bool) RTNetIPv4IsUDPValid(PCRTNETIPV4 pIpHdr, PCRTNETUDP pUdpHdr, void const *pvData, size_t cbPktMax)
+ * @param   fChecksum       Whether to validate the checksum (GSO).
+ */
+RTDECL(bool) RTNetIPv4IsUDPValid(PCRTNETIPV4 pIpHdr, PCRTNETUDP pUdpHdr, void const *pvData, size_t cbPktMax, bool fChecksum)
 {
     if (RT_UNLIKELY(!rtNetIPv4IsUDPSizeValid(pIpHdr, pUdpHdr, cbPktMax)))
         return false;
-    if (pUdpHdr->uh_sum)
+    if (fChecksum && pUdpHdr->uh_sum)
     {
         uint16_t u16Sum = RTNetIPv4UDPChecksum(pIpHdr, pUdpHdr, pvData);
@@ -481 +507 @@
  * @param   pIpHdr          Pointer to the IPv4 header, in network endian (big).
  * @param   pTcpHdr         Pointer to the TCP header, in network endian (big).
- * @param   pvData          Pointer to the TCP payload. The size is taken from the
- *                          TCP header and the caller is supposed to have validated
- *                          this before calling.
- *                          If NULL then we assume the data follows immediately after
- *                          the TCP header.
+ * @param   pvData          Pointer to the TCP payload. The size is derived from
+ *                          the two headers and the caller is supposed to have
+ *                          validated this before calling.  If NULL, we assume
+ *                          the data follows immediately after the TCP header.
  */
 RTDECL(uint16_t) RTNetIPv4TCPChecksum(PCRTNETIPV4 pIpHdr, PCRTNETTCP pTcpHdr, void const *pvData)
@@ -498 +523 @@
 }
 RT_EXPORT_SYMBOL(RTNetIPv4TCPChecksum);
+
+
+/**
+ * Calculates the checksum for the TCP header given the TCP header, payload and
+ * the checksum of the pseudo header.
+ *
+ * This is not specific to IPv4.
+ *
+ * @returns The checksum (network endian).
+ * @param   u32Sum          The checksum of the pseudo header. See
+ *                          RTNetIPv4PseudoChecksum and RTNetIPv6PseudoChecksum.
+ * @param   pTcpHdr         Pointer to the TCP header, in network endian (big).
+ * @param   pvData          Pointer to the TCP payload.
+ * @param   cbData          The size of the TCP payload.
+ */
+RTDECL(uint16_t) RTNetTCPChecksum(uint32_t u32Sum, PCRTNETTCP pTcpHdr, void const *pvData, size_t cbData)
+{
+    u32Sum = rtNetIPv4AddTCPChecksum(pTcpHdr, u32Sum);
+    bool fOdd = false;
+    u32Sum = rtNetIPv4AddDataChecksum(pvData, cbData, u32Sum, &fOdd);
+    return rtNetIPv4FinalizeChecksum(u32Sum);
+}
+RT_EXPORT_SYMBOL(RTNetTCPChecksum);
 
 
@@ -559 +607 @@
  *                          the TCP header.
  * @param   cbPktMax        The max TCP packet size, TCP header and payload (data).
- */
-RTDECL(bool) RTNetIPv4IsTCPValid(PCRTNETIPV4 pIpHdr, PCRTNETTCP pTcpHdr, size_t cbHdrMax, void const *pvData, size_t cbPktMax)
+ * @param   fChecksum       Whether to validate the checksum (GSO).
+ */
+RTDECL(bool) RTNetIPv4IsTCPValid(PCRTNETIPV4 pIpHdr, PCRTNETTCP pTcpHdr, size_t cbHdrMax, void const *pvData, size_t cbPktMax,
+                                 bool fChecksum)
 {
     if (RT_UNLIKELY(!rtNetIPv4IsTCPSizeValid(pIpHdr, pTcpHdr, cbHdrMax, cbPktMax)))
         return false;
-    uint16_t u16Sum = RTNetIPv4TCPChecksum(pIpHdr, pTcpHdr, pvData);
-    if (RT_UNLIKELY(pTcpHdr->th_sum != u16Sum))
-        return false;
+    if (fChecksum)
+    {
+        uint16_t u16Sum = RTNetIPv4TCPChecksum(pIpHdr, pTcpHdr, pvData);
+        if (RT_UNLIKELY(pTcpHdr->th_sum != u16Sum))
+            return false;
+    }
     return true;
 }