This commit is contained in:
刘河
2019-01-12 01:22:53 +08:00
parent a67cdff02a
commit 0ce6ab1e15
4 changed files with 20 additions and 25 deletions

View File

@@ -10,7 +10,6 @@ import (
"log"
"net"
"net/http"
"net/http/httputil"
"net/url"
"strconv"
"strings"
@@ -245,16 +244,20 @@ func (s *Conn) SetAlive() {
//从tcp报文中解析出host连接类型等 TODO 多种情况
func (s *Conn) GetHost() (method, address string, rb []byte, err error, r *http.Request) {
r, err = http.ReadRequest(bufio.NewReader(s))
if err != nil {
var b [32 * 1024]byte
var n int
if n, err = s.Read(b[:]); err != nil {
return
}
rb, err = httputil.DumpRequest(r, true)
rb = b[:n]
r, err = http.ReadRequest(bufio.NewReader(bytes.NewReader(rb)))
if err != nil {
return
}
hostPortURL, err := url.Parse(r.Host)
if err != nil {
address = r.Host
err = nil
return
}
if hostPortURL.Opaque == "443" { //https访问