From: Sergey Matveev <stargrave@stargrave.org>
Date: Mon, 15 Apr 2024 22:00:07 +0000 (+0300)
Subject: One possible round-trip less
X-Git-Tag: v2.0.0~1
X-Git-Url: http://www.git.stargrave.org/?a=commitdiff_plain;h=9f22296ea23e88cb49642d45f68dcd77240bc1fdc4b46f623428a968a7f123f3;p=vors.git

One possible round-trip less
---

diff --git a/cmd/client/main.go b/cmd/client/main.go
index 7ff841c..fadaf51 100644
--- a/cmd/client/main.go
+++ b/cmd/client/main.go
@@ -162,10 +162,6 @@ func main() {
 	if err = ctrl.SetNoDelay(true); err != nil {
 		log.Fatalln("nodelay:", err)
 	}
-	if _, err = io.Copy(ctrl, strings.NewReader(vors.NoisePrologue)); err != nil {
-		log.Fatalln("handshake: write prologue", err)
-		return
-	}
 
 	hs, err := noise.NewHandshakeState(noise.Config{
 		CipherSuite: vors.NoiseCipherSuite,
@@ -181,7 +177,16 @@ func main() {
 	if err != nil {
 		log.Fatalln("handshake encrypt:", err)
 	}
-	if err = vors.PktWrite(ctrl, buf); err != nil {
+	buf = append(
+		append(
+			[]byte(vors.NoisePrologue),
+			byte((len(buf) & 0xFF00) >> 8),
+			byte((len(buf) & 0x00FF) >> 0),
+		),
+		buf...
+	)
+	_, err = io.Copy(ctrl, bytes.NewReader(buf))
+	if err != nil {
 		log.Fatalln("write handshake:", err)
 		return
 	}