Ma/origin
1
0
Fork 0
mirror of https://github.com/duanhf2012/origin.git synced 2026-02-04 06:54:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

框架修复

Browse Source
This commit is contained in:
lifeiyi
2020-03-28 19:53:03 +08:00
parent 9a43744551
commit bffc2b5f79
5 changed files with 21 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
network/tcp_msg.go
View File

@@ -89,6 +89,7 @@ func (p *MsgParser) Read(conn *TCPConn) ([]byte, error) {
msgLen = binary.BigEndian.Uint32(bufMsgLen)
}
}
msgLen -= 2
// check len
if msgLen > p.maxMsgLen {
@@ -121,6 +122,7 @@ func (p *MsgParser) Write(conn *TCPConn, args ...[]byte) error {
return errors.New("message too short")
}
//msgLen -= 2
msg := make([]byte, uint32(p.lenMsgLen)+msgLen)
// write len
@@ -131,7 +133,7 @@ func (p *MsgParser) Write(conn *TCPConn, args ...[]byte) error {
if p.littleEndian {
binary.LittleEndian.PutUint16(msg, uint16(msgLen))
} else {
binary.BigEndian.PutUint16(msg, uint16(msgLen))
binary.BigEndian.PutUint16(msg, uint16(msgLen)+uint16(p.lenMsgLen))
}
case 4:
if p.littleEndian {