Function assertICMPChecksum

packet/packet_test.go:112–160

(t *testing.T, icmpPacket *ICMP)

}

func assertICMPChecksum(t *testing.T, icmpPacket *ICMP) {
	buf := gopacket.NewSerializeBuffer()
	if icmpPacket.Protocol == layers.IPProtocolICMPv4 {
		icmpv4 := layers.ICMPv4{
			TypeCode: layers.CreateICMPv4TypeCode(uint8(icmpPacket.Type.(ipv4.ICMPType)), uint8(icmpPacket.Code)),
		}
		switch body := icmpPacket.Body.(type) {
		case *icmp.Echo:
			icmpv4.Id = uint16(body.ID)
			icmpv4.Seq = uint16(body.Seq)
			payload := gopacket.Payload(body.Data)
			require.NoError(t, payload.SerializeTo(buf, serializeOpts))
		default:
			require.NoError(t, serializeICMPAsPayload(icmpPacket.Message, buf))
		}
		// SerializeTo sets the checksum in icmpv4
		require.NoError(t, icmpv4.SerializeTo(buf, serializeOpts))
		require.Equal(t, icmpv4.Checksum, uint16(icmpPacket.Checksum))
	} else {
		switch body := icmpPacket.Body.(type) {
		case *icmp.Echo:
			payload := gopacket.Payload(body.Data)
			require.NoError(t, payload.SerializeTo(buf, serializeOpts))
			echo := layers.ICMPv6Echo{
				Identifier: uint16(body.ID),
				SeqNumber:  uint16(body.Seq),
			}
			require.NoError(t, echo.SerializeTo(buf, serializeOpts))
		default:
			require.NoError(t, serializeICMPAsPayload(icmpPacket.Message, buf))
		}

		icmpv6 := layers.ICMPv6{
			TypeCode: layers.CreateICMPv6TypeCode(uint8(icmpPacket.Type.(ipv6.ICMPType)), uint8(icmpPacket.Code)),
		}
		ipLayer := layers.IPv6{
			Version:    6,
			SrcIP:      icmpPacket.Src.AsSlice(),
			DstIP:      icmpPacket.Dst.AsSlice(),
			NextHeader: icmpPacket.Protocol,
			HopLimit:   icmpPacket.TTL,
		}
		require.NoError(t, icmpv6.SetNetworkLayerForChecksum(&ipLayer))

		// SerializeTo sets the checksum in icmpv4
		require.NoError(t, icmpv6.SerializeTo(buf, serializeOpts))
		require.Equal(t, icmpv6.Checksum, uint16(icmpPacket.Checksum))
	}
}

func serializeICMPAsPayload(message *icmp.Message, buf gopacket.SerializeBuffer) error {
	serializedBody, err := message.Body.Marshal(message.Type.Protocol())