From 3c4ecd0180d170bd199d9ff0c7dc5664481f90f1 Mon Sep 17 00:00:00 2001
From: xvzc <pkhu2000@gmail.com>
Date: Sun, 5 Jun 2022 14:50:55 +0900
Subject: [PATCH] update readBytes EOF strategy

---
 net/conn.go | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)

diff --git a/net/conn.go b/net/conn.go
index 17b3ee3..24870e0 100644
--- a/net/conn.go
+++ b/net/conn.go
@@ -2,6 +2,7 @@ package net
 
 import (
 	"errors"
+	"io"
 	"net"
 	"time"
 
@@ -94,13 +95,15 @@ func (conn *Conn) ReadBytes() ([]byte, error) {
         }
     }
 
+    if len(ret) == 0 {
+        return nil, io.EOF
+    }
+
     return ret, nil
 }
 
 func (lConn *Conn) HandleHttp(p *packet.HttpPacket) {
     defer func() {
-        lConn.CloseRead()
-        lConn.CloseWrite()
         lConn.Close()
         log.Debug("[HTTP] Closing client Connection.. ", lConn.RemoteAddr())
     }()
@@ -129,8 +132,6 @@ func (lConn *Conn) HandleHttp(p *packet.HttpPacket) {
 	}
 
     defer func() {
-        rConn.CloseRead()
-        rConn.CloseWrite()
         rConn.Close()
         log.Debug("[HTTP] Closing server Connection.. ", p.Domain(), " ", rConn.LocalAddr())
     }()
@@ -152,8 +153,6 @@ func (lConn *Conn) HandleHttp(p *packet.HttpPacket) {
 
 func (lConn *Conn) HandleHttps(p *packet.HttpPacket) {
     defer func() {
-        lConn.CloseRead()
-        lConn.CloseWrite()
         lConn.Close()
         log.Debug("[HTTPS] Closing client Connection.. ", lConn.RemoteAddr())
     }()
@@ -180,8 +179,6 @@ func (lConn *Conn) HandleHttps(p *packet.HttpPacket) {
 	}
 
     defer func() {
-        rConn.CloseRead()
-        rConn.CloseWrite()
         rConn.Close()
         log.Debug("[HTTPS] Closing server Connection.. ", p.Domain(), " ", rConn.LocalAddr())
     }()
@@ -193,7 +190,8 @@ func (lConn *Conn) HandleHttps(p *packet.HttpPacket) {
 		log.Debug("[HTTPS] Error sending 200 Connection Established to the client", err)
         return
 	}
-	log.Debug("[HTTPS] Sent 200 Connection Estabalished to the client")
+
+	log.Debug("[HTTPS] Sent 200 Connection Estabalished to ", lConn.RemoteAddr())
 
 	// Read client hello
 	clientHello, err := lConn.ReadBytes()