/
chunks.go
63 lines (53 loc) · 1.15 KB
/
chunks.go
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
package chunk
import (
"encoding"
"encoding/hex"
"fmt"
)
type Chunks []Chunk
var _ fmt.Stringer = (*Chunk)(nil)
var _ encoding.BinaryMarshaler = (*Chunk)(nil)
var _ encoding.BinaryUnmarshaler = (*Chunk)(nil)
func (c Chunks) MarshalBinary() ([]byte, error) {
buf := make([]byte, chunkSize*len(c))
for i, chunk := range c {
bytes, err := chunk.MarshalBinary()
if err != nil {
return nil, err
}
copy(buf[i*chunkSize:], bytes)
}
return buf, nil
}
func (c *Chunks) UnmarshalBinary(buf []byte) error {
if len(buf)%chunkSize != 0 {
return ErrInvalidChunkSize
}
chunks := make(Chunks, 0, len(buf)/chunkSize)
for i := 0; i < len(buf); i += chunkSize {
chunk := Chunk{}
err := chunk.UnmarshalBinary(buf[i : i+chunkSize])
if err != nil {
return err
}
chunks = append(chunks, chunk)
}
*c = chunks
return nil
}
func (c Chunks) String() string {
bytes, _ := c.MarshalBinary()
return hex.EncodeToString(bytes)
}
func NewChunksFromString(s string) (Chunks, error) {
c := Chunks{}
bytes, err := hex.DecodeString(s)
if err != nil {
return nil, err
}
err = c.UnmarshalBinary(bytes)
if err != nil {
return nil, err
}
return c, nil
}