From 1d233a5b5e655830bc8771681b73c412cb4a53f5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Zolt=C3=A1n=20Papp?= <zoltan.pmail@gmail.com>
Date: Tue, 10 Sep 2024 16:31:36 +0200
Subject: [PATCH] Decrease the max size of the handshake message

---
 relay/client/client.go    | 2 +-
 relay/messages/message.go | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/relay/client/client.go b/relay/client/client.go
index 7e5379ba5..6560c81e1 100644
--- a/relay/client/client.go
+++ b/relay/client/client.go
@@ -249,7 +249,7 @@ func (c *Client) handShake() error {
 		log.Errorf("failed to send auth message: %s", err)
 		return err
 	}
-	buf := make([]byte, messages.MaxHandshakeSize)
+	buf := make([]byte, messages.MaxHandshakeRespSize)
 	n, err := c.readWithTimeout(buf)
 	if err != nil {
 		log.Errorf("failed to read auth response: %s", err)
diff --git a/relay/messages/message.go b/relay/messages/message.go
index 73279be00..04ed1d09b 100644
--- a/relay/messages/message.go
+++ b/relay/messages/message.go
@@ -7,7 +7,8 @@ import (
 )
 
 const (
-	MaxHandshakeSize = 8192
+	MaxHandshakeSize     = 212
+	MaxHandshakeRespSize = 8192
 
 	CurrentProtocolVersion = 1