|
@@ -6,6 +6,7 @@ package ipv4
|
|
|
|
|
|
|
|
import (
|
|
import (
|
|
|
"bytes"
|
|
"bytes"
|
|
|
|
|
+ "encoding/binary"
|
|
|
"net"
|
|
"net"
|
|
|
"reflect"
|
|
"reflect"
|
|
|
"runtime"
|
|
"runtime"
|
|
@@ -13,46 +14,54 @@ import (
|
|
|
"testing"
|
|
"testing"
|
|
|
)
|
|
)
|
|
|
|
|
|
|
|
-var (
|
|
|
|
|
- wireHeaderFromKernel = [HeaderLen]byte{
|
|
|
|
|
|
|
+type headerTest struct {
|
|
|
|
|
+ wireHeaderFromKernel [HeaderLen]byte
|
|
|
|
|
+ wireHeaderToKernel [HeaderLen]byte
|
|
|
|
|
+ wireHeaderFromTradBSDKernel [HeaderLen]byte
|
|
|
|
|
+ wireHeaderFromFreeBSD10Kernel [HeaderLen]byte
|
|
|
|
|
+ wireHeaderToTradBSDKernel [HeaderLen]byte
|
|
|
|
|
+ *Header
|
|
|
|
|
+}
|
|
|
|
|
+
|
|
|
|
|
+var headerLittleEndianTest = headerTest{
|
|
|
|
|
+ // TODO(mikio): Add platform dependent wire header formats when
|
|
|
|
|
+ // we support new platforms.
|
|
|
|
|
+ wireHeaderFromKernel: [HeaderLen]byte{
|
|
|
0x45, 0x01, 0xbe, 0xef,
|
|
0x45, 0x01, 0xbe, 0xef,
|
|
|
0xca, 0xfe, 0x45, 0xdc,
|
|
0xca, 0xfe, 0x45, 0xdc,
|
|
|
0xff, 0x01, 0xde, 0xad,
|
|
0xff, 0x01, 0xde, 0xad,
|
|
|
172, 16, 254, 254,
|
|
172, 16, 254, 254,
|
|
|
192, 168, 0, 1,
|
|
192, 168, 0, 1,
|
|
|
- }
|
|
|
|
|
- wireHeaderToKernel = [HeaderLen]byte{
|
|
|
|
|
|
|
+ },
|
|
|
|
|
+ wireHeaderToKernel: [HeaderLen]byte{
|
|
|
0x45, 0x01, 0xbe, 0xef,
|
|
0x45, 0x01, 0xbe, 0xef,
|
|
|
0xca, 0xfe, 0x45, 0xdc,
|
|
0xca, 0xfe, 0x45, 0xdc,
|
|
|
0xff, 0x01, 0xde, 0xad,
|
|
0xff, 0x01, 0xde, 0xad,
|
|
|
172, 16, 254, 254,
|
|
172, 16, 254, 254,
|
|
|
192, 168, 0, 1,
|
|
192, 168, 0, 1,
|
|
|
- }
|
|
|
|
|
- wireHeaderFromTradBSDKernel = [HeaderLen]byte{
|
|
|
|
|
|
|
+ },
|
|
|
|
|
+ wireHeaderFromTradBSDKernel: [HeaderLen]byte{
|
|
|
0x45, 0x01, 0xdb, 0xbe,
|
|
0x45, 0x01, 0xdb, 0xbe,
|
|
|
0xca, 0xfe, 0xdc, 0x45,
|
|
0xca, 0xfe, 0xdc, 0x45,
|
|
|
0xff, 0x01, 0xde, 0xad,
|
|
0xff, 0x01, 0xde, 0xad,
|
|
|
172, 16, 254, 254,
|
|
172, 16, 254, 254,
|
|
|
192, 168, 0, 1,
|
|
192, 168, 0, 1,
|
|
|
- }
|
|
|
|
|
- wireHeaderFromFreeBSD10Kernel = [HeaderLen]byte{
|
|
|
|
|
|
|
+ },
|
|
|
|
|
+ wireHeaderFromFreeBSD10Kernel: [HeaderLen]byte{
|
|
|
0x45, 0x01, 0xef, 0xbe,
|
|
0x45, 0x01, 0xef, 0xbe,
|
|
|
0xca, 0xfe, 0xdc, 0x45,
|
|
0xca, 0xfe, 0xdc, 0x45,
|
|
|
0xff, 0x01, 0xde, 0xad,
|
|
0xff, 0x01, 0xde, 0xad,
|
|
|
172, 16, 254, 254,
|
|
172, 16, 254, 254,
|
|
|
192, 168, 0, 1,
|
|
192, 168, 0, 1,
|
|
|
- }
|
|
|
|
|
- wireHeaderToTradBSDKernel = [HeaderLen]byte{
|
|
|
|
|
|
|
+ },
|
|
|
|
|
+ wireHeaderToTradBSDKernel: [HeaderLen]byte{
|
|
|
0x45, 0x01, 0xef, 0xbe,
|
|
0x45, 0x01, 0xef, 0xbe,
|
|
|
0xca, 0xfe, 0xdc, 0x45,
|
|
0xca, 0xfe, 0xdc, 0x45,
|
|
|
0xff, 0x01, 0xde, 0xad,
|
|
0xff, 0x01, 0xde, 0xad,
|
|
|
172, 16, 254, 254,
|
|
172, 16, 254, 254,
|
|
|
192, 168, 0, 1,
|
|
192, 168, 0, 1,
|
|
|
- }
|
|
|
|
|
- // TODO(mikio): Add platform dependent wire header formats when
|
|
|
|
|
- // we support new platforms.
|
|
|
|
|
-
|
|
|
|
|
- testHeader = &Header{
|
|
|
|
|
|
|
+ },
|
|
|
|
|
+ Header: &Header{
|
|
|
Version: Version,
|
|
Version: Version,
|
|
|
Len: HeaderLen,
|
|
Len: HeaderLen,
|
|
|
TOS: 1,
|
|
TOS: 1,
|
|
@@ -65,26 +74,31 @@ var (
|
|
|
Checksum: 0xdead,
|
|
Checksum: 0xdead,
|
|
|
Src: net.IPv4(172, 16, 254, 254),
|
|
Src: net.IPv4(172, 16, 254, 254),
|
|
|
Dst: net.IPv4(192, 168, 0, 1),
|
|
Dst: net.IPv4(192, 168, 0, 1),
|
|
|
- }
|
|
|
|
|
-)
|
|
|
|
|
|
|
+ },
|
|
|
|
|
+}
|
|
|
|
|
|
|
|
func TestMarshalHeader(t *testing.T) {
|
|
func TestMarshalHeader(t *testing.T) {
|
|
|
- b, err := testHeader.Marshal()
|
|
|
|
|
|
|
+ tt := &headerLittleEndianTest
|
|
|
|
|
+ if nativeEndian != binary.LittleEndian {
|
|
|
|
|
+ t.Skip("no test for non-little endian machine yet")
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ b, err := tt.Header.Marshal()
|
|
|
if err != nil {
|
|
if err != nil {
|
|
|
t.Fatal(err)
|
|
t.Fatal(err)
|
|
|
}
|
|
}
|
|
|
var wh []byte
|
|
var wh []byte
|
|
|
switch runtime.GOOS {
|
|
switch runtime.GOOS {
|
|
|
case "darwin", "dragonfly", "netbsd":
|
|
case "darwin", "dragonfly", "netbsd":
|
|
|
- wh = wireHeaderToTradBSDKernel[:]
|
|
|
|
|
|
|
+ wh = tt.wireHeaderToTradBSDKernel[:]
|
|
|
case "freebsd":
|
|
case "freebsd":
|
|
|
if freebsdVersion < 1000000 {
|
|
if freebsdVersion < 1000000 {
|
|
|
- wh = wireHeaderToTradBSDKernel[:]
|
|
|
|
|
|
|
+ wh = tt.wireHeaderToTradBSDKernel[:]
|
|
|
} else {
|
|
} else {
|
|
|
- wh = wireHeaderFromFreeBSD10Kernel[:]
|
|
|
|
|
|
|
+ wh = tt.wireHeaderFromFreeBSD10Kernel[:]
|
|
|
}
|
|
}
|
|
|
default:
|
|
default:
|
|
|
- wh = wireHeaderToKernel[:]
|
|
|
|
|
|
|
+ wh = tt.wireHeaderToKernel[:]
|
|
|
}
|
|
}
|
|
|
if !bytes.Equal(b, wh) {
|
|
if !bytes.Equal(b, wh) {
|
|
|
t.Fatalf("got %#v; want %#v", b, wh)
|
|
t.Fatalf("got %#v; want %#v", b, wh)
|
|
@@ -92,25 +106,30 @@ func TestMarshalHeader(t *testing.T) {
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
func TestParseHeader(t *testing.T) {
|
|
func TestParseHeader(t *testing.T) {
|
|
|
|
|
+ tt := &headerLittleEndianTest
|
|
|
|
|
+ if nativeEndian != binary.LittleEndian {
|
|
|
|
|
+ t.Skip("no test for big endian machine yet")
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
var wh []byte
|
|
var wh []byte
|
|
|
switch runtime.GOOS {
|
|
switch runtime.GOOS {
|
|
|
case "darwin", "dragonfly", "netbsd":
|
|
case "darwin", "dragonfly", "netbsd":
|
|
|
- wh = wireHeaderFromTradBSDKernel[:]
|
|
|
|
|
|
|
+ wh = tt.wireHeaderFromTradBSDKernel[:]
|
|
|
case "freebsd":
|
|
case "freebsd":
|
|
|
if freebsdVersion < 1000000 {
|
|
if freebsdVersion < 1000000 {
|
|
|
- wh = wireHeaderFromTradBSDKernel[:]
|
|
|
|
|
|
|
+ wh = tt.wireHeaderFromTradBSDKernel[:]
|
|
|
} else {
|
|
} else {
|
|
|
- wh = wireHeaderFromFreeBSD10Kernel[:]
|
|
|
|
|
|
|
+ wh = tt.wireHeaderFromFreeBSD10Kernel[:]
|
|
|
}
|
|
}
|
|
|
default:
|
|
default:
|
|
|
- wh = wireHeaderFromKernel[:]
|
|
|
|
|
|
|
+ wh = tt.wireHeaderFromKernel[:]
|
|
|
}
|
|
}
|
|
|
h, err := ParseHeader(wh)
|
|
h, err := ParseHeader(wh)
|
|
|
if err != nil {
|
|
if err != nil {
|
|
|
t.Fatal(err)
|
|
t.Fatal(err)
|
|
|
}
|
|
}
|
|
|
- if !reflect.DeepEqual(h, testHeader) {
|
|
|
|
|
- t.Fatalf("got %#v; want %#v", h, testHeader)
|
|
|
|
|
|
|
+ if !reflect.DeepEqual(h, tt.Header) {
|
|
|
|
|
+ t.Fatalf("got %#v; want %#v", h, tt.Header)
|
|
|
}
|
|
}
|
|
|
s := h.String()
|
|
s := h.String()
|
|
|
if strings.Contains(s, ",") {
|
|
if strings.Contains(s, ",") {
|