|
- // Copyright 2014 The Go Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style
- // license that can be found in the LICENSE file.
-
- package icmp_test
-
- import (
- "bytes"
- "net"
- "reflect"
- "testing"
-
- "golang.org/x/net/icmp"
- "golang.org/x/net/internal/iana"
- "golang.org/x/net/ipv4"
- "golang.org/x/net/ipv6"
- )
-
- func TestMarshalAndParseMessage(t *testing.T) {
- fn := func(t *testing.T, proto int, tms []icmp.Message) {
- var pshs [][]byte
- switch proto {
- case iana.ProtocolICMP:
- pshs = [][]byte{nil}
- case iana.ProtocolIPv6ICMP:
- pshs = [][]byte{
- icmp.IPv6PseudoHeader(net.ParseIP("fe80::1"), net.ParseIP("ff02::1")),
- nil,
- }
- }
- for i, tm := range tms {
- for _, psh := range pshs {
- b, err := tm.Marshal(psh)
- if err != nil {
- t.Fatalf("#%d: %v", i, err)
- }
- m, err := icmp.ParseMessage(proto, b)
- if err != nil {
- t.Fatalf("#%d: %v", i, err)
- }
- if m.Type != tm.Type || m.Code != tm.Code {
- t.Errorf("#%d: got %#v; want %#v", i, m, &tm)
- continue
- }
- if !reflect.DeepEqual(m.Body, tm.Body) {
- t.Errorf("#%d: got %#v; want %#v", i, m.Body, tm.Body)
- continue
- }
- }
- }
- }
-
- t.Run("IPv4", func(t *testing.T) {
- fn(t, iana.ProtocolICMP,
- []icmp.Message{
- {
- Type: ipv4.ICMPTypeDestinationUnreachable, Code: 15,
- Body: &icmp.DstUnreach{
- Data: []byte("ERROR-INVOKING-PACKET"),
- },
- },
- {
- Type: ipv4.ICMPTypeTimeExceeded, Code: 1,
- Body: &icmp.TimeExceeded{
- Data: []byte("ERROR-INVOKING-PACKET"),
- },
- },
- {
- Type: ipv4.ICMPTypeParameterProblem, Code: 2,
- Body: &icmp.ParamProb{
- Pointer: 8,
- Data: []byte("ERROR-INVOKING-PACKET"),
- },
- },
- {
- Type: ipv4.ICMPTypeEcho, Code: 0,
- Body: &icmp.Echo{
- ID: 1, Seq: 2,
- Data: []byte("HELLO-R-U-THERE"),
- },
- },
- {
- Type: ipv4.ICMPTypeExtendedEchoRequest, Code: 0,
- Body: &icmp.ExtendedEchoRequest{
- ID: 1, Seq: 2,
- Extensions: []icmp.Extension{
- &icmp.InterfaceIdent{
- Class: 3,
- Type: 1,
- Name: "en101",
- },
- },
- },
- },
- {
- Type: ipv4.ICMPTypeExtendedEchoReply, Code: 0,
- Body: &icmp.ExtendedEchoReply{
- State: 4 /* Delay */, Active: true, IPv4: true,
- },
- },
- })
- })
- t.Run("IPv6", func(t *testing.T) {
- fn(t, iana.ProtocolIPv6ICMP,
- []icmp.Message{
- {
- Type: ipv6.ICMPTypeDestinationUnreachable, Code: 6,
- Body: &icmp.DstUnreach{
- Data: []byte("ERROR-INVOKING-PACKET"),
- },
- },
- {
- Type: ipv6.ICMPTypePacketTooBig, Code: 0,
- Body: &icmp.PacketTooBig{
- MTU: 1<<16 - 1,
- Data: []byte("ERROR-INVOKING-PACKET"),
- },
- },
- {
- Type: ipv6.ICMPTypeTimeExceeded, Code: 1,
- Body: &icmp.TimeExceeded{
- Data: []byte("ERROR-INVOKING-PACKET"),
- },
- },
- {
- Type: ipv6.ICMPTypeParameterProblem, Code: 2,
- Body: &icmp.ParamProb{
- Pointer: 8,
- Data: []byte("ERROR-INVOKING-PACKET"),
- },
- },
- {
- Type: ipv6.ICMPTypeEchoRequest, Code: 0,
- Body: &icmp.Echo{
- ID: 1, Seq: 2,
- Data: []byte("HELLO-R-U-THERE"),
- },
- },
- {
- Type: ipv6.ICMPTypeExtendedEchoRequest, Code: 0,
- Body: &icmp.ExtendedEchoRequest{
- ID: 1, Seq: 2,
- Extensions: []icmp.Extension{
- &icmp.InterfaceIdent{
- Class: 3,
- Type: 2,
- Index: 911,
- },
- },
- },
- },
- {
- Type: ipv6.ICMPTypeExtendedEchoReply, Code: 0,
- Body: &icmp.ExtendedEchoReply{
- State: 5 /* Probe */, Active: true, IPv6: true,
- },
- },
- })
- })
- }
-
- func TestMarshalAndParseRawMessage(t *testing.T) {
- t.Run("RawBody", func(t *testing.T) {
- for i, tt := range []struct {
- m icmp.Message
- wire []byte
- parseShouldFail bool
- }{
- { // Nil body
- m: icmp.Message{
- Type: ipv4.ICMPTypeDestinationUnreachable, Code: 127,
- },
- wire: []byte{
- 0x03, 0x7f, 0xfc, 0x80,
- },
- parseShouldFail: true,
- },
- { // Empty body
- m: icmp.Message{
- Type: ipv6.ICMPTypeDestinationUnreachable, Code: 128,
- Body: &icmp.RawBody{},
- },
- wire: []byte{
- 0x01, 0x80, 0x00, 0x00,
- },
- parseShouldFail: true,
- },
- { // Crafted body
- m: icmp.Message{
- Type: ipv6.ICMPTypeDuplicateAddressConfirmation, Code: 129,
- Body: &icmp.RawBody{
- Data: []byte{0xca, 0xfe},
- },
- },
- wire: []byte{
- 0x9e, 0x81, 0x00, 0x00,
- 0xca, 0xfe,
- },
- parseShouldFail: false,
- },
- } {
- b, err := tt.m.Marshal(nil)
- if err != nil {
- t.Errorf("#%d: %v", i, err)
- continue
- }
- if !bytes.Equal(b, tt.wire) {
- t.Errorf("#%d: got %#v; want %#v", i, b, tt.wire)
- continue
- }
- m, err := icmp.ParseMessage(tt.m.Type.Protocol(), b)
- if err != nil != tt.parseShouldFail {
- t.Errorf("#%d: got %v, %v", i, m, err)
- continue
- }
- if tt.parseShouldFail {
- continue
- }
- if m.Type != tt.m.Type || m.Code != tt.m.Code {
- t.Errorf("#%d: got %v; want %v", i, m, tt.m)
- continue
- }
- if !bytes.Equal(m.Body.(*icmp.RawBody).Data, tt.m.Body.(*icmp.RawBody).Data) {
- t.Errorf("#%d: got %#v; want %#v", i, m.Body, tt.m.Body)
- continue
- }
- }
- })
- t.Run("RawExtension", func(t *testing.T) {
- for i, tt := range []struct {
- m icmp.Message
- wire []byte
- }{
- { // Unaligned data and nil extension
- m: icmp.Message{
- Type: ipv6.ICMPTypeDestinationUnreachable, Code: 130,
- Body: &icmp.DstUnreach{
- Data: []byte("ERROR-INVOKING-PACKET"),
- },
- },
- wire: []byte{
- 0x01, 0x82, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00,
- 'E', 'R', 'R', 'O',
- 'R', '-', 'I', 'N',
- 'V', 'O', 'K', 'I',
- 'N', 'G', '-', 'P',
- 'A', 'C', 'K', 'E',
- 'T',
- },
- },
- { // Unaligned data and empty extension
- m: icmp.Message{
- Type: ipv6.ICMPTypeDestinationUnreachable, Code: 131,
- Body: &icmp.DstUnreach{
- Data: []byte("ERROR-INVOKING-PACKET"),
- Extensions: []icmp.Extension{
- &icmp.RawExtension{},
- },
- },
- },
- wire: []byte{
- 0x01, 0x83, 0x00, 0x00,
- 0x02, 0x00, 0x00, 0x00,
- 'E', 'R', 'R', 'O',
- 'R', '-', 'I', 'N',
- 'V', 'O', 'K', 'I',
- 'N', 'G', '-', 'P',
- 'A', 'C', 'K', 'E',
- 'T',
- 0x20, 0x00, 0xdf, 0xff,
- },
- },
- { // Nil extension
- m: icmp.Message{
- Type: ipv6.ICMPTypeExtendedEchoRequest, Code: 132,
- Body: &icmp.ExtendedEchoRequest{
- ID: 1, Seq: 2, Local: true,
- },
- },
- wire: []byte{
- 0xa0, 0x84, 0x00, 0x00,
- 0x00, 0x01, 0x02, 0x01,
- },
- },
- { // Empty extension
- m: icmp.Message{
- Type: ipv6.ICMPTypeExtendedEchoRequest, Code: 133,
- Body: &icmp.ExtendedEchoRequest{
- ID: 1, Seq: 2, Local: true,
- Extensions: []icmp.Extension{
- &icmp.RawExtension{},
- },
- },
- },
- wire: []byte{
- 0xa0, 0x85, 0x00, 0x00,
- 0x00, 0x01, 0x02, 0x01,
- 0x20, 0x00, 0xdf, 0xff,
- },
- },
- { // Crafted extension
- m: icmp.Message{
- Type: ipv6.ICMPTypeExtendedEchoRequest, Code: 134,
- Body: &icmp.ExtendedEchoRequest{
- ID: 1, Seq: 2, Local: true,
- Extensions: []icmp.Extension{
- &icmp.RawExtension{
- Data: []byte("CRAFTED"),
- },
- },
- },
- },
- wire: []byte{
- 0xa0, 0x86, 0x00, 0x00,
- 0x00, 0x01, 0x02, 0x01,
- 0x20, 0x00, 0xc3, 0x21,
- 'C', 'R', 'A', 'F',
- 'T', 'E', 'D',
- },
- },
- } {
- b, err := tt.m.Marshal(nil)
- if err != nil {
- t.Errorf("#%d: %v", i, err)
- continue
- }
- if !bytes.Equal(b, tt.wire) {
- t.Errorf("#%d: got %#v; want %#v", i, b, tt.wire)
- continue
- }
- m, err := icmp.ParseMessage(tt.m.Type.Protocol(), b)
- if err != nil {
- t.Errorf("#%d: %v", i, err)
- continue
- }
- if m.Type != tt.m.Type || m.Code != tt.m.Code {
- t.Errorf("#%d: got %v; want %v", i, m, tt.m)
- continue
- }
- }
- })
- }
|