epic/play-with-docker
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add more info to router

Browse Source
This commit is contained in:
Jonathan Leibiusky @xetorthio
2017-09-19 20:39:45 -03:00
parent a268ad6988
commit 495219cdec
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
router/router.go
View File

@@ -329,12 +329,15 @@ func (r *proxyRouter) ListenSshAddress() string {
func (r *proxyRouter) handleConnection(c net.Conn) {
defer c.Close()
// first try tls
start := time.Now()
vhostConn, err := vhost.TLS(c)
discoverElapsed := time.Since(start)
if err == nil {
// It is a TLS connection
defer vhostConn.Close()
host := vhostConn.ClientHelloMsg.ServerName
log.Printf("Proxying TLS connection to %s. Discover took %s\n", host, discoverElapsed)
dstHost, err := r.director(ProtocolHTTPS, host)
if err != nil {
log.Printf("Error directing request: %v\n", err)
@@ -351,7 +354,9 @@ func (r *proxyRouter) handleConnection(c net.Conn) {
// it is not TLS
// treat it as an http connection
start := time.Now()
req, err := http.ReadRequest(bufio.NewReader(vhostConn))
httpReadElapsed := time.Since(start)
if err != nil {
// It is not http neither. So just close the connection.
return
@@ -360,6 +365,7 @@ func (r *proxyRouter) handleConnection(c net.Conn) {
if host == "" {
host = req.Host
}
log.Printf("Proxying http connection to %s. Discover took %s. Http read took %s\n", host, discoverElapsed, httpReadElapsed)
dstHost, err := r.director(ProtocolHTTP, host)
if err != nil {
log.Printf("Error directing request: %v\n", err)