Ma/origin
1
0
Fork 0
mirror of https://github.com/duanhf2012/origin.git synced 2026-02-03 22:45:13 +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 -= 2
// check len
if msgLen > p.maxMsgLen {
@@ -97,7 +97,8 @@ func (p *MsgParser) Read(conn *TCPConn) ([]byte, error) {
} else if msgLen < p.minMsgLen {
return nil, errors.New("message too short")
}
msgLen -= 2
// data
msgData := make([]byte, msgLen)
if _, err := io.ReadFull(conn, msgData); err != nil {