1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
// Copyright 2021 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 slicereader
import (
"encoding/binary"
"io"
"testing"
)
func TestSliceReader(t *testing.T) {
b := []byte{}
bt := make([]byte, 4)
e32 := uint32(1030507)
binary.LittleEndian.PutUint32(bt, e32)
b = append(b, bt...)
bt = make([]byte, 8)
e64 := uint64(907050301)
binary.LittleEndian.PutUint64(bt, e64)
b = append(b, bt...)
b = appendUleb128(b, uint(e32))
b = appendUleb128(b, uint(e64))
b = appendUleb128(b, 6)
s1 := "foobar"
s1b := []byte(s1)
b = append(b, s1b...)
b = appendUleb128(b, 9)
s2 := "bazbasher"
s2b := []byte(s2)
b = append(b, s2b...)
readStr := func(slr *Reader) string {
len := slr.ReadULEB128()
return slr.ReadString(int64(len))
}
for i := 0; i < 2; i++ {
slr := NewReader(b, i == 0)
g32 := slr.ReadUint32()
if g32 != e32 {
t.Fatalf("slr.ReadUint32() got %d want %d", g32, e32)
}
g64 := slr.ReadUint64()
if g64 != e64 {
t.Fatalf("slr.ReadUint64() got %d want %d", g64, e64)
}
g32 = uint32(slr.ReadULEB128())
if g32 != e32 {
t.Fatalf("slr.ReadULEB128() got %d want %d", g32, e32)
}
g64 = slr.ReadULEB128()
if g64 != e64 {
t.Fatalf("slr.ReadULEB128() got %d want %d", g64, e64)
}
gs1 := readStr(slr)
if gs1 != s1 {
t.Fatalf("readStr got %s want %s", gs1, s1)
}
gs2 := readStr(slr)
if gs2 != s2 {
t.Fatalf("readStr got %s want %s", gs2, s2)
}
if _, err := slr.Seek(4, io.SeekStart); err != nil {
t.Fatal(err)
}
off := slr.Offset()
if off != 4 {
t.Fatalf("Offset() returned %d wanted 4", off)
}
g64 = slr.ReadUint64()
if g64 != e64 {
t.Fatalf("post-seek slr.ReadUint64() got %d want %d", g64, e64)
}
}
}
func appendUleb128(b []byte, v uint) []byte {
for {
c := uint8(v & 0x7f)
v >>= 7
if v != 0 {
c |= 0x80
}
b = append(b, c)
if c&0x80 == 0 {
break
}
}
return b
}
|
