fixed tls recv bug

Author: gal

clients/http2client.go

@@ -109,8 +109,8 @@ func (client *Http2Client) Connect() error {
 	return nil
 }
 
-// getTCPConn tries to unwrap the net.Conn to get the underlying *net.TCPConn
-func getTCPConn(c net.Conn) (*net.TCPConn, error) {
+// GetTCPConn tries to unwrap the net.Conn to get the underlying *net.TCPConn
+func GetTCPConn(c net.Conn) (*net.TCPConn, error) {
 	switch conn := c.(type) {
 	case *net.TCPConn:
 		return conn, nil
@@ -132,7 +132,7 @@ func (client *Http2Client) CloseFIN() error {
 		return fmt.Errorf("connection is already closed")
 	}
 	//time.Sleep(time.Second * 1)
-	conn, err := getTCPConn(client.conn)
+	conn, err := GetTCPConn(client.conn)
 	if err != nil {
 		return fmt.Errorf("error getting TCP connection: %v", err)
 	}

clients/repeaterClientBase.go

@@ -1,6 +1,7 @@
 package clients
 
 import (
+	"crypto/tls"
 	"io"
 	"log"
 	"net"
@@ -39,7 +40,7 @@ func (c *RepeaterClientBase) Recv() (http2.Frame, error) {
 	return c.Client.ReadFrame()
 }
 
-func (c *RepeaterClientBase) StartRecvToDevNull2() {
+func (c *RepeaterClientBase) StartRecvToDevNull() {
 	go func() {
 		buf := make([]byte, 4000000) // Standard buffer size for socket reads
 		for {
@@ -59,14 +60,19 @@ func (c *RepeaterClientBase) StartRecvToDevNull2() {
 	}()
 }
 
-func (c *RepeaterClientBase) StartRecvToDevNull() {
+func (c *RepeaterClientBase) StartRecvToDevNull2() {
 	go func() {
-		if tc, ok := c.Client.conn.(*net.TCPConn); ok {
+		if tc, err := GetTCPConn(c.Client.conn); err == nil {
 			err := tc.SetReadBuffer(1 << 27) // Ask the OS for 128MB recv buffer so that the TCP window won't block
 			if err != nil {
 				log.Println("Error setting read buffer:", err)
 			}
-			_, _ = io.Copy(io.Discard, tc)
+		}
+		switch conn := c.Client.conn.(type) {
+		case *net.TCPConn:
+			_, _ = io.Copy(io.Discard, conn)
+		case *tls.Conn:
+			//c.StartRecvToDevNullTLS()
 		}
 	}()
 }