From cb81d4f3c27456983290393cc7a3cee883ec0e66 Mon Sep 17 00:00:00 2001
From: mic <562710164>
Date: Thu, 9 Feb 2023 17:52:21 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96proto=E7=BC=96=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 YHProto/protocol.go | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

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
 }