-
Notifications
You must be signed in to change notification settings - Fork 0
/
packet.go
67 lines (51 loc) · 1.34 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
//通讯协议处理,主要处理封包和解包的过程
package protocol
import (
"fmt"
"bufio"
"bytes"
"encoding/binary"
"gomore/global"
)
func Packet(message string) ([]byte, error) {
return []byte( message+"\n" ), nil
fmt.Println( global.PackSplitType )
if global.PackSplitType =="breakline" {
return []byte( message +"\n" ), nil
}
var length int32 = int32(len(message))
fmt.Println( "Set length :", length )
var pkg *bytes.Buffer = new(bytes.Buffer)
err := binary.Write(pkg, binary.LittleEndian, length)
if err != nil {
return nil, err
}
err = binary.Write(pkg, binary.LittleEndian, []byte(message))
if err != nil {
return nil, err
}
return pkg.Bytes(), nil
}
func Unpack(reader *bufio.Reader) ([]byte, error) {
return reader.ReadBytes('\n')
//if global.PackSplitType =="breakline" {
// return reader.ReadBytes('\n')
//}
lengthByte, _ := reader.Peek(4)
lengthBuff := bytes.NewBuffer(lengthByte)
var length int32
err := binary.Read(lengthBuff, binary.LittleEndian, &length)
if err != nil {
return nil, err
}
fmt.Println( "Get length :", length )
if int32(reader.Buffered()) < length+4 {
return nil, err
}
pack := make([]byte, int(4+length))
_, err = reader.Read(pack)
if err != nil {
return nil, err
}
return pack[4:] , nil
}