fix: tls in HTTP1 after protocol detect

Author: mrhaoxx

http/http.go

@@ -6,6 +6,7 @@ import (
 	"compress/gzip"
 	"context"
 	"crypto/rand"
+	"crypto/tls"
 	"encoding/hex"
 	"fmt"
 	"io"
@@ -153,6 +154,21 @@ func newReqID() string {
 	return hex.EncodeToString(b)
 }
 
+func (h *Midware) preparetls(rw http.ResponseWriter, r *http.Request, conn *tcp.Conn) {
+	if r.TLS == nil {
+		head := conn.Head()
+		if head > 1 {
+			if conn.Protocol()[head-1] == "TLS" {
+				if conn, ok := conn.TopConn().(*utils.RwConn); ok {
+					r.TLS = new(tls.ConnectionState)
+					*r.TLS = conn.Rawconn.(*tls.Conn).ConnectionState()
+				}
+			}
+		}
+	}
+
+}
+
 func (h *Midware) head(rw http.ResponseWriter, r *http.Request, conn *tcp.Conn) {
 	ngrw := &NgResponseWriter{
 		writer:         nil,

http/midware.go

@@ -91,6 +91,7 @@ func (h *Midware) Handle(c *tcp.Conn) tcp.SerRet {
 	switch top {
 	case "HTTP1":
 		http.Serve(utils.ConnGetSocket(c.TopConn()), http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
+			h.preparetls(rw, r, c)
 			h.head(rw, r, c)
 		}))
 		return tcp.Close

tcp/connection.go

@@ -54,10 +54,17 @@ func (c *Conn) TopConn() net.Conn {
 	return c.conn[c.head]
 }
 func (c *Conn) TopProtocol() string {
-	c.mu.RLock()
-	defer c.mu.RUnlock()
 	return c.proto[c.head]
 }
+
+func (c *Conn) Protocol() []string {
+	return c.proto
+}
+
+func (c *Conn) Head() int {
+	return int(c.head)
+}
+
 func (c *Conn) unsync_genProtocols() {
 	c.protos = ""
 	for _, v := range c.proto {