From 623c17293121287037ff7193e575fb2be9a346d3 Mon Sep 17 00:00:00 2001
From: octeep <github@bandersnatch.anonaddy.com>
Date: Mon, 28 Mar 2022 18:58:13 +0100
Subject: [PATCH] simplify connForward

---
 main.go | 16 +++++-----------
 1 file changed, 5 insertions(+), 11 deletions(-)

diff --git a/main.go b/main.go
index 479e355..163d3f3 100644
--- a/main.go
+++ b/main.go
@@ -7,6 +7,7 @@ import (
 	"encoding/hex"
 	"errors"
 	"fmt"
+	"io"
 	"log"
 	"math/rand"
 	"net"
@@ -266,17 +267,10 @@ func (c CredentialValidator) Valid(username, password string) bool {
 
 func connForward(bufSize int, from, to net.Conn) {
 	buf := make([]byte, bufSize)
-	for {
-		size, err := from.Read(buf)
-		if err != nil {
-			to.Close()
-			return
-		}
-		_, err = to.Write(buf[:size])
-		if err != nil {
-			to.Close()
-			return
-		}
+	_, err := io.CopyBuffer(to, from, buf)
+	if err != nil {
+		to.Close()
+		return
 	}
 }