/
pack.go
95 lines (78 loc) · 1.89 KB
/
pack.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
package snet
import (
"bytes"
"encoding/binary"
"fmt"
"gcom/gcmd"
"hash/crc32"
)
//Package 消息包
type Package struct {
Msg []byte
}
//Pack 封包
func (p *Package) Pack() []byte {
buf := new(bytes.Buffer)
msgLength := uint32(len(p.Msg))
err := binary.Write(buf, binary.BigEndian, msgLength)
if err != nil {
gcmd.Println(gcmd.Err, "pack msg length err:", err)
return nil
}
checkLength := p.CheckCRC32(msgLength)
err = binary.Write(buf, binary.BigEndian, checkLength)
if err != nil {
gcmd.Println(gcmd.Err, "pack msg check length err:", err)
return nil
}
err = binary.Write(buf, binary.BigEndian, p.Msg)
if err != nil {
gcmd.Println(gcmd.Err, "pack msg err:", err)
return nil
}
return buf.Bytes()
}
// UnPack 解包
func (p *Package) UnPack(buff []byte) *Package {
buf := bytes.NewBuffer(buff)
var msgLength uint32
err := binary.Read(buf, binary.BigEndian, &msgLength)
if err != nil {
gcmd.Println(gcmd.Err, "unpack msg length err:", err)
return &Package{}
}
var checkLength uint32
err = binary.Read(buf, binary.BigEndian, &checkLength)
if err != nil {
gcmd.Println(gcmd.Err, "unpack msg check length err:", err)
return &Package{}
}
msg := make([]byte, msgLength)
err = binary.Read(buf, binary.BigEndian, &p.Msg)
if err != nil {
gcmd.Println(gcmd.Err, "unpack msg err:", err)
return &Package{}
}
p.Msg = append(p.Msg, msg...)
return p
}
// UnPackMsgLength 解包长度
func (p Package) UnPackMsgLength(buff []byte) (length uint32) {
buf := bytes.NewBuffer(buff)
err := binary.Read(buf, binary.BigEndian, &length)
if err != nil {
fmt.Println("unpack length err:", err)
return 0
}
return
}
// SetMsg 封包消息
func (p *Package) SetMsg(msg []byte) {
p.Msg = append(p.Msg, msg...)
}
// CheckCRC32 CRC校验
func (p *Package) CheckCRC32(l interface{}) uint32 {
ieee := crc32.NewIEEE()
binary.Write(ieee, binary.BigEndian, l)
return ieee.Sum32()
}