311 lines
7.6 KiB
Go
311 lines
7.6 KiB
Go
package struc
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/binary"
|
|
"fmt"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
type Nested struct {
|
|
Test2 int `struc:"int8"`
|
|
}
|
|
|
|
type Example struct {
|
|
Pad []byte `struc:"[5]pad"` // 00 00 00 00 00
|
|
I8f int `struc:"int8"` // 01
|
|
I16f int `struc:"int16"` // 00 02
|
|
I32f int `struc:"int32"` // 00 00 00 03
|
|
I64f int `struc:"int64"` // 00 00 00 00 00 00 00 04
|
|
U8f int `struc:"uint8,little"` // 05
|
|
U16f int `struc:"uint16,little"` // 06 00
|
|
U32f int `struc:"uint32,little"` // 07 00 00 00
|
|
U64f int `struc:"uint64,little"` // 08 00 00 00 00 00 00 00
|
|
Boolf int `struc:"bool"` // 01
|
|
Byte4f []byte `struc:"[4]byte"` // "abcd"
|
|
|
|
I8 int8 // 09
|
|
I16 int16 // 00 0a
|
|
I32 int32 // 00 00 00 0b
|
|
I64 int64 // 00 00 00 00 00 00 00 0c
|
|
U8 uint8 `struc:"little"` // 0d
|
|
U16 uint16 `struc:"little"` // 0e 00
|
|
U32 uint32 `struc:"little"` // 0f 00 00 00
|
|
U64 uint64 `struc:"little"` // 10 00 00 00 00 00 00 00
|
|
BoolT bool // 01
|
|
BoolF bool // 00
|
|
Byte4 [4]byte // "efgh"
|
|
Float1 float32 // 41 a0 00 00
|
|
Float2 float64 // 41 35 00 00 00 00 00 00
|
|
|
|
I32f2 int64 `struc:"int32"` // ff ff ff ff
|
|
U32f2 int64 `struc:"uint32"` // ff ff ff ff
|
|
|
|
I32f3 int32 `struc:"int64"` // ff ff ff ff ff ff ff ff
|
|
|
|
Size int `struc:"sizeof=Str,little"` // 0a 00 00 00
|
|
Str string `struc:"[]byte"` // "ijklmnopqr"
|
|
Strb string `struc:"[4]byte"` // "stuv"
|
|
|
|
Size2 int `struc:"uint8,sizeof=Str2"` // 04
|
|
Str2 string // "1234"
|
|
|
|
Size3 int `struc:"uint8,sizeof=Bstr"` // 04
|
|
Bstr []byte // "5678"
|
|
|
|
Size4 int `struc:"little"` // 07 00 00 00
|
|
Str4a string `struc:"[]byte,sizefrom=Size4"` // "ijklmno"
|
|
Str4b string `struc:"[]byte,sizefrom=Size4"` // "pqrstuv"
|
|
|
|
Size5 int `struc:"uint8"` // 04
|
|
Bstr2 []byte `struc:"sizefrom=Size5"` // "5678"
|
|
|
|
Nested Nested // 00 00 00 01
|
|
NestedP *Nested // 00 00 00 02
|
|
TestP64 *int `struc:"int64"` // 00 00 00 05
|
|
|
|
NestedSize int `struc:"sizeof=NestedA"` // 00 00 00 02
|
|
NestedA []Nested // [00 00 00 03, 00 00 00 04]
|
|
|
|
Skip int `struc:"skip"`
|
|
|
|
CustomTypeSize Int3 `struc:"sizeof=CustomTypeSizeArr"` // 00 00 00 04
|
|
CustomTypeSizeArr []byte // "ABCD"
|
|
}
|
|
|
|
var five = 5
|
|
|
|
type ExampleStructWithin struct {
|
|
a uint8
|
|
}
|
|
|
|
type ExampleSlice struct {
|
|
PropsLen uint8 `struc:"sizeof=Props"`
|
|
Props []ExampleStructWithin
|
|
}
|
|
|
|
type ExampleArray struct {
|
|
PropsLen uint8
|
|
Props [16]ExampleStructWithin `struc:"[16]ExampleStructWithin"`
|
|
}
|
|
|
|
var arraySliceReferenceBytes = []byte{
|
|
16,
|
|
0, 0, 0, 1,
|
|
0, 0, 0, 1,
|
|
0, 0, 0, 2,
|
|
0, 0, 0, 3,
|
|
0, 0, 0, 4,
|
|
0, 0, 0, 5,
|
|
0, 0, 0, 6,
|
|
0, 0, 0, 7,
|
|
0, 0, 0, 8,
|
|
0, 0, 0, 9,
|
|
0, 0, 0, 10,
|
|
0, 0, 0, 11,
|
|
0, 0, 0, 12,
|
|
0, 0, 0, 13,
|
|
0, 0, 0, 14,
|
|
0, 0, 0, 15,
|
|
0, 0, 0, 16,
|
|
}
|
|
|
|
var arrayReference = &ExampleArray{
|
|
16,
|
|
[16]ExampleStructWithin{
|
|
ExampleStructWithin{1},
|
|
ExampleStructWithin{2},
|
|
ExampleStructWithin{3},
|
|
ExampleStructWithin{4},
|
|
ExampleStructWithin{5},
|
|
ExampleStructWithin{6},
|
|
ExampleStructWithin{7},
|
|
ExampleStructWithin{8},
|
|
ExampleStructWithin{9},
|
|
ExampleStructWithin{10},
|
|
ExampleStructWithin{11},
|
|
ExampleStructWithin{12},
|
|
ExampleStructWithin{13},
|
|
ExampleStructWithin{14},
|
|
ExampleStructWithin{15},
|
|
ExampleStructWithin{16},
|
|
},
|
|
}
|
|
|
|
var sliceReference = &ExampleSlice{
|
|
16,
|
|
[]ExampleStructWithin{
|
|
ExampleStructWithin{1},
|
|
ExampleStructWithin{2},
|
|
ExampleStructWithin{3},
|
|
ExampleStructWithin{4},
|
|
ExampleStructWithin{5},
|
|
ExampleStructWithin{6},
|
|
ExampleStructWithin{7},
|
|
ExampleStructWithin{8},
|
|
ExampleStructWithin{9},
|
|
ExampleStructWithin{10},
|
|
ExampleStructWithin{11},
|
|
ExampleStructWithin{12},
|
|
ExampleStructWithin{13},
|
|
ExampleStructWithin{14},
|
|
ExampleStructWithin{15},
|
|
ExampleStructWithin{16},
|
|
},
|
|
}
|
|
|
|
var reference = &Example{
|
|
nil,
|
|
1, 2, 3, 4, 5, 6, 7, 8, 0, []byte{'a', 'b', 'c', 'd'},
|
|
9, 10, 11, 12, 13, 14, 15, 16, true, false, [4]byte{'e', 'f', 'g', 'h'},
|
|
20, 21,
|
|
-1,
|
|
4294967295,
|
|
-1,
|
|
10, "ijklmnopqr", "stuv",
|
|
4, "1234",
|
|
4, []byte("5678"),
|
|
7, "ijklmno", "pqrstuv",
|
|
4, []byte("5678"),
|
|
Nested{1}, &Nested{2}, &five,
|
|
6, []Nested{{3}, {4}, {5}, {6}, {7}, {8}},
|
|
0,
|
|
Int3(4), []byte("ABCD"),
|
|
}
|
|
|
|
var referenceBytes = []byte{
|
|
0, 0, 0, 0, 0, // pad(5)
|
|
1, 0, 2, 0, 0, 0, 3, 0, 0, 0, 0, 0, 0, 0, 4, // fake int8-int64(1-4)
|
|
5, 6, 0, 7, 0, 0, 0, 8, 0, 0, 0, 0, 0, 0, 0, // fake little-endian uint8-uint64(5-8)
|
|
0, // fake bool(0)
|
|
'a', 'b', 'c', 'd', // fake [4]byte
|
|
|
|
9, 0, 10, 0, 0, 0, 11, 0, 0, 0, 0, 0, 0, 0, 12, // real int8-int64(9-12)
|
|
13, 14, 0, 15, 0, 0, 0, 16, 0, 0, 0, 0, 0, 0, 0, // real little-endian uint8-uint64(13-16)
|
|
1, 0, // real bool(1), bool(0)
|
|
'e', 'f', 'g', 'h', // real [4]byte
|
|
65, 160, 0, 0, // real float32(20)
|
|
64, 53, 0, 0, 0, 0, 0, 0, // real float64(21)
|
|
|
|
255, 255, 255, 255, // fake int32(-1)
|
|
255, 255, 255, 255, // fake uint32(4294967295)
|
|
|
|
255, 255, 255, 255, 255, 255, 255, 255, // fake int64(-1)
|
|
|
|
10, 0, 0, 0, // little-endian int32(10) sizeof=Str
|
|
'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', // Str
|
|
's', 't', 'u', 'v', // fake string([4]byte)
|
|
04, '1', '2', '3', '4', // real string
|
|
04, '5', '6', '7', '8', // fake []byte(string)
|
|
|
|
7, 0, 0, 0, // little-endian int32(7)
|
|
'i', 'j', 'k', 'l', 'm', 'n', 'o', // Str4a sizefrom=Size4
|
|
'p', 'q', 'r', 's', 't', 'u', 'v', // Str4b sizefrom=Size4
|
|
04, '5', '6', '7', '8', // fake []byte(string)
|
|
|
|
1, 2, // Nested{1}, Nested{2}
|
|
0, 0, 0, 0, 0, 0, 0, 5, // &five
|
|
|
|
0, 0, 0, 6, // int32(6)
|
|
3, 4, 5, 6, 7, 8, // [Nested{3}, ...Nested{8}]
|
|
|
|
0, 0, 4, 'A', 'B', 'C', 'D', // Int3(4), []byte("ABCD")
|
|
}
|
|
|
|
func TestCodec(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
if err := Pack(&buf, reference); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
out := &Example{}
|
|
if err := Unpack(&buf, out); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !reflect.DeepEqual(reference, out) {
|
|
fmt.Printf("got: %#v\nwant: %#v\n", out, reference)
|
|
t.Fatal("encode/decode failed")
|
|
}
|
|
}
|
|
|
|
func TestEncode(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
if err := Pack(&buf, reference); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !bytes.Equal(buf.Bytes(), referenceBytes) {
|
|
fmt.Printf("got: %#v\nwant: %#v\n", buf.Bytes(), referenceBytes)
|
|
t.Fatal("encode failed")
|
|
}
|
|
}
|
|
|
|
func TestDecode(t *testing.T) {
|
|
buf := bytes.NewReader(referenceBytes)
|
|
out := &Example{}
|
|
if err := Unpack(buf, out); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !reflect.DeepEqual(reference, out) {
|
|
fmt.Printf("got: %#v\nwant: %#v\n", out, reference)
|
|
t.Fatal("decode failed")
|
|
}
|
|
}
|
|
|
|
func TestSizeof(t *testing.T) {
|
|
size, err := Sizeof(reference)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if size != len(referenceBytes) {
|
|
t.Fatalf("sizeof failed; expected %d, got %d", len(referenceBytes), size)
|
|
}
|
|
}
|
|
|
|
type ExampleEndian struct {
|
|
T int `struc:"int16,big"`
|
|
}
|
|
|
|
func TestEndianSwap(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
big := &ExampleEndian{1}
|
|
if err := PackWithOrder(&buf, big, binary.BigEndian); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
little := &ExampleEndian{}
|
|
if err := UnpackWithOrder(&buf, little, binary.LittleEndian); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if little.T != 256 {
|
|
t.Fatal("big -> little conversion failed")
|
|
}
|
|
}
|
|
|
|
func TestNilValue(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
if err := Pack(&buf, nil); err == nil {
|
|
t.Fatal("failed throw error for bad struct value")
|
|
}
|
|
if err := Unpack(&buf, nil); err == nil {
|
|
t.Fatal("failed throw error for bad struct value")
|
|
}
|
|
if _, err := Sizeof(nil); err == nil {
|
|
t.Fatal("failed to throw error for bad struct value")
|
|
}
|
|
}
|
|
|
|
type sliceUnderrun struct {
|
|
Str string `struc:"[10]byte"`
|
|
Arr []uint16 `struc:"[10]uint16"`
|
|
}
|
|
|
|
func TestSliceUnderrun(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
v := sliceUnderrun{
|
|
Str: "foo",
|
|
Arr: []uint16{1, 2, 3},
|
|
}
|
|
if err := Pack(&buf, &v); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|