diff --git a/YHProto/protocol.go b/YHProto/protocol.go index 0dbac00..8e168a0 100644 --- a/YHProto/protocol.go +++ b/YHProto/protocol.go @@ -139,9 +139,8 @@ func (codec *EchoPackage) DecodeBuffer(msg []byte) (err error) { return ErrNotEnoughStream } - bBuf := msg[echoPkgHeaderLen:codec.H.Len] codec.B = make([]byte, codec.H.Len) - copy(codec.B, bBuf) + copy(codec.B, msg[echoPkgHeaderLen:echoPkgHeaderLen+int(codec.H.Len)]) return nil } @@ -207,9 +206,7 @@ func (codec *EchoPackage) DecodeWsMessage(msg wsutil.Message) (err error) { if len(msg.Payload)-echoPkgHeaderLen < int(codec.H.Len) { return ErrNotEnoughStream } - - bBuf := msg.Payload[echoPkgHeaderLen:codec.H.Len] codec.B = make([]byte, codec.H.Len) - copy(codec.B, bBuf) + copy(codec.B, msg.Payload[echoPkgHeaderLen:echoPkgHeaderLen+int(codec.H.Len)]) return nil }