fix: Hide reset by peer

* fix: Hide reset by peer
qemus · Dec 28, 2023 · c81bf10 · c81bf10
1 parent f8568df
commit c81bf10
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/main.go b/src/main.go
@@ -10,8 +10,10 @@ import (
 	"flag"
 	"sync"
 	"bytes"
-	"strconv"
+	"errors"
 	"strings"
+	"syscall"
+	"strconv"
 	"os/exec"
 	"net/http"
 	"math/rand"
@@ -141,10 +143,10 @@ func incoming_conn(conn net.Conn) {
 		len, err := conn.Read(buf)
 
 		if err != nil {
-			if err != io.EOF {
-				log.Println("Read error:", err)
-			} else {
+			if err == io.EOF || errors.Is(err, syscall.ECONNRESET) {
 				fmt.Println("Disconnected:", err)
+			} else {
+				log.Println("Read error:", err)
 			}
 			if len != Packet { return }
 		}