hysteria2: Fix missing message limit

SagerNet · May 12, 2024 · 4223bde · 4223bde
1 parent a7af781
commit 4223bde
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/hysteria2/internal/protocol/proxy.go b/hysteria2/internal/protocol/proxy.go
@@ -119,6 +119,9 @@ func ReadTCPResponse(r io.Reader) (bool, string, error) {
 }
 
 func WriteTCPResponse(ok bool, msg string, payload []byte) *buf.Buffer {
+	if len(msg) > MaxMessageLength {
+		msg = msg[:MaxMessageLength]
+	}
 	padding := tcpResponsePadding.String()
 	paddingLen := len(padding)
 	msgLen := len(msg)
@@ -198,7 +201,7 @@ func ParseUDPMessage(msg []byte) (*UDPMessage, error) {
 	if err != nil {
 		return nil, err
 	}
-	if lAddr == 0 || lAddr > MaxMessageLength {
+	if lAddr == 0 || lAddr > MaxAddressLength {
 		return nil, E.New("invalid address length")
 	}
 	bs := buf.Bytes()