| // 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 |
| } |
| } |
| }) |
| } |
