Ma/origin
1
0
Fork 0
mirror of https://github.com/duanhf2012/origin.git synced 2026-03-03 12:28:31 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

优化包头

Browse Source
This commit is contained in:
duanhf2012
2020-04-02 20:38:27 +08:00
parent 6498851e78
commit c98db7e042
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
network/tcp_msg.go
View File

@@ -89,7 +89,7 @@ func (p *MsgParser) Read(conn *TCPConn) ([]byte, error) {
msgLen = binary.BigEndian.Uint32(bufMsgLen) msgLen = binary.BigEndian.Uint32(bufMsgLen)
} }
} }
msgLen -= 2
// check len // check len
if msgLen > p.maxMsgLen { if msgLen > p.maxMsgLen {
@@ -97,7 +97,8 @@ func (p *MsgParser) Read(conn *TCPConn) ([]byte, error) {
} else if msgLen < p.minMsgLen { } else if msgLen < p.minMsgLen {
return nil, errors.New("message too short") return nil, errors.New("message too short")
} }
msgLen -= 2
// data // data
msgData := make([]byte, msgLen) msgData := make([]byte, msgLen)
if _, err := io.ReadFull(conn, msgData); err != nil { if _, err := io.ReadFull(conn, msgData); err != nil {