/
packet.go
105 lines (89 loc) · 1.88 KB
/
packet.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
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
96
97
98
99
100
101
102
103
104
105
package protocol
import (
"encoding/binary"
"errors"
"io"
"strconv"
"github.com/feel-easy/hole-server/utils"
)
var (
lenSize = 4
)
type Packet struct {
Body []byte `json:"data"`
}
func (p Packet) Int() (int, error) {
v, err := strconv.ParseInt(p.String(), 10, 64)
return int(v), err
}
func (p Packet) Int64() (int64, error) {
v, _ := strconv.ParseInt(p.String(), 10, 64)
return v, nil
}
func (p Packet) String() string {
return string(p.Body)
}
func (p Packet) Unmarshal(v interface{}) error {
return utils.UnmarshalJson(p.Body, v)
}
func StringPacket(msg string) Packet {
return Packet{
Body: []byte(msg),
}
}
func ErrorPacket(err error) Packet {
return Packet{
Body: []byte(err.Error()),
}
}
func ObjectPacket(obj interface{}) Packet {
return Packet{
Body: utils.MarshalJson(obj),
}
}
type ReadWriteCloser interface {
Read() (*Packet, error)
Write(msg Packet) error
Close() error
IP() string
}
func readUint32(reader io.Reader) (uint32, error) {
data := make([]byte, 4)
_, err := io.ReadFull(reader, data)
if err != nil {
return 0, err
}
return binary.BigEndian.Uint32(data), nil
}
func readUint64(reader io.Reader) (uint64, error) {
data := make([]byte, 8)
_, err := io.ReadFull(reader, data)
if err != nil {
return 0, err
}
return binary.BigEndian.Uint64(data), nil
}
func encode(msg Packet) []byte {
lenBytes := make([]byte, lenSize)
binary.BigEndian.PutUint32(lenBytes, uint32(len(msg.Body)))
data := make([]byte, 0)
data = append(data, lenBytes...)
return append(data, msg.Body...)
}
func decode(r io.Reader) (*Packet, error) {
l, err := readUint32(r)
if err != nil {
return nil, err
}
if l > MaxPacketSize {
return nil, errors.New("Overflow max packet size " + strconv.Itoa(MaxPacketSize))
}
dataBytes := make([]byte, l)
_, err = io.ReadFull(r, dataBytes)
if err != nil {
return nil, err
}
return &Packet{
Body: dataBytes,
}, nil
}