netbird/gvisor
0
0
Fork 0
mirror of https://github.com/netbirdio/gvisor.git synced 2026-05-22 17:12:49 -07:00
Code Issues Packages Projects Releases Wiki Activity

Avoid accidental zero-checksum

Browse Source 
PiperOrigin-RevId: 320250773
This commit is contained in:
Tamir Duberstein
2020-07-08 13:13:44 -07:00
committed by gVisor bot
parent e3db9bda60
commit e1f11dea28
1 changed files with 16 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/tcpip/transport/udp/udp_test.go
+16 -5
View File
@@ -1721,9 +1721,11 @@ func TestIncrementMalformedPacketsReceived(t *testing.T) {
payload := newPayload()
h := unicastV6.header4Tuple(incoming)
buf := c.buildV6Packet(payload, &h)
// Invalidate the packet length field in the UDP header by adding one.
// Invalidate the UDP header length field.
u := header.UDP(buf[header.IPv6MinimumSize:])
u.SetLength(u.Length() + 1)
c.linkEP.InjectInbound(ipv6.ProtocolNumber, &stack.PacketBuffer{
Data: buf.ToVectorisedView(),
})
@@ -1803,9 +1805,16 @@ func TestIncrementChecksumErrorsV4(t *testing.T) {
payload := newPayload()
h := unicastV4.header4Tuple(incoming)
buf := c.buildV4Packet(payload, &h)
// Invalidate the checksum field in the UDP header by adding one.
u := header.UDP(buf[header.IPv4MinimumSize:])
u.SetChecksum(u.Checksum() + 1)
// Invalidate the UDP header checksum field, taking care to avoid
// overflow to zero, which would disable checksum validation.
for u := header.UDP(buf[header.IPv4MinimumSize:]); ; {
u.SetChecksum(u.Checksum() + 1)
if u.Checksum() != 0 {
break
}
}
c.linkEP.InjectInbound(ipv4.ProtocolNumber, &stack.PacketBuffer{
Data: buf.ToVectorisedView(),
})
@@ -1834,9 +1843,11 @@ func TestIncrementChecksumErrorsV6(t *testing.T) {
payload := newPayload()
h := unicastV6.header4Tuple(incoming)
buf := c.buildV6Packet(payload, &h)
// Invalidate the checksum field in the UDP header by adding one.
// Invalidate the UDP header checksum field.
u := header.UDP(buf[header.IPv6MinimumSize:])
u.SetChecksum(u.Checksum() + 1)
c.linkEP.InjectInbound(ipv6.ProtocolNumber, &stack.PacketBuffer{
Data: buf.ToVectorisedView(),
})