aboutsummaryrefslogtreecommitdiffstats
path: root/vendor/github.com/lunixbochs/struc/binary.go
blob: 4899d08fd2bacd81c14f32dc247817a824206c30 (plain)
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
package struc

import (
	"encoding/binary"
	"io"
	"reflect"
)

type byteWriter struct {
	buf []byte
	pos int
}

func (b byteWriter) Write(p []byte) (int, error) {
	capacity := len(b.buf) - b.pos
	if capacity < len(p) {
		p = p[:capacity]
	}
	if len(p) > 0 {
		copy(b.buf[b.pos:], p)
		b.pos += len(p)
	}
	return len(p), nil
}

type binaryFallback reflect.Value

func (b binaryFallback) String() string {
	return b.String()
}

func (b binaryFallback) Sizeof(val reflect.Value, options *Options) int {
	return binary.Size(val.Interface())
}

func (b binaryFallback) Pack(buf []byte, val reflect.Value, options *Options) (int, error) {
	tmp := byteWriter{buf: buf}
	var order binary.ByteOrder = binary.BigEndian
	if options.Order != nil {
		order = options.Order
	}
	err := binary.Write(tmp, order, val.Interface())
	return tmp.pos, err
}

func (b binaryFallback) Unpack(r io.Reader, val reflect.Value, options *Options) error {
	var order binary.ByteOrder = binary.BigEndian
	if options.Order != nil {
		order = options.Order
	}
	return binary.Read(r, order, val.Interface())
}