vendor/golang.org/x/net/icmp/ipv4_test.go

   1 // Copyright 2014 The Go Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style
   3 // license that can be found in the LICENSE file.
   4
   5 package icmp
   6
   7 import (
   8         "encoding/binary"
   9         "net"
  10         "reflect"
  11         "runtime"
  12         "testing"
  13
  14         "golang.org/x/net/internal/socket"
  15         "golang.org/x/net/ipv4"
  16 )
  17
  18 type ipv4HeaderTest struct {
  19         wireHeaderFromKernel        [ipv4.HeaderLen]byte
  20         wireHeaderFromTradBSDKernel [ipv4.HeaderLen]byte
  21         Header                      *ipv4.Header
  22 }
  23
  24 var ipv4HeaderLittleEndianTest = ipv4HeaderTest{
  25         // TODO(mikio): Add platform dependent wire header formats when
  26         // we support new platforms.
  27         wireHeaderFromKernel: [ipv4.HeaderLen]byte{
  28                 0x45, 0x01, 0xbe, 0xef,
  29                 0xca, 0xfe, 0x45, 0xdc,
  30                 0xff, 0x01, 0xde, 0xad,
  31                 172, 16, 254, 254,
  32                 192, 168, 0, 1,
  33         },
  34         wireHeaderFromTradBSDKernel: [ipv4.HeaderLen]byte{
  35                 0x45, 0x01, 0xef, 0xbe,
  36                 0xca, 0xfe, 0x45, 0xdc,
  37                 0xff, 0x01, 0xde, 0xad,
  38                 172, 16, 254, 254,
  39                 192, 168, 0, 1,
  40         },
  41         Header: &ipv4.Header{
  42                 Version:  ipv4.Version,
  43                 Len:      ipv4.HeaderLen,
  44                 TOS:      1,
  45                 TotalLen: 0xbeef,
  46                 ID:       0xcafe,
  47                 Flags:    ipv4.DontFragment,
  48                 FragOff:  1500,
  49                 TTL:      255,
  50                 Protocol: 1,
  51                 Checksum: 0xdead,
  52                 Src:      net.IPv4(172, 16, 254, 254),
  53                 Dst:      net.IPv4(192, 168, 0, 1),
  54         },
  55 }
  56
  57 func TestParseIPv4Header(t *testing.T) {
  58         tt := &ipv4HeaderLittleEndianTest
  59         if socket.NativeEndian != binary.LittleEndian {
  60                 t.Skip("no test for non-little endian machine yet")
  61         }
  62
  63         var wh []byte
  64         switch runtime.GOOS {
  65         case "darwin":
  66                 wh = tt.wireHeaderFromTradBSDKernel[:]
  67         case "freebsd":
  68                 if freebsdVersion >= 1000000 {
  69                         wh = tt.wireHeaderFromKernel[:]
  70                 } else {
  71                         wh = tt.wireHeaderFromTradBSDKernel[:]
  72                 }
  73         default:
  74                 wh = tt.wireHeaderFromKernel[:]
  75         }
  76         h, err := ParseIPv4Header(wh)
  77         if err != nil {
  78                 t.Fatal(err)
  79         }
  80         if !reflect.DeepEqual(h, tt.Header) {
  81                 t.Fatalf("got %#v; want %#v", h, tt.Header)
  82         }
  83 }