Dispark/frp
1
0
Fork 0
mirror of https://github.com/fatedier/frp.git synced 2026-03-24 00:48:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

update reverseproxy from std libraries

Browse Source
This commit is contained in:
fatedier
2019-08-06 16:49:22 +08:00
parent ee0df07a3c
commit b53a2c1ed9
4 changed files with 289 additions and 147 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
utils/vhost/http.go
View File

@@ -89,14 +89,18 @@ func NewHttpReverseProxy(option HttpReverseProxyOptions, vhostRouter *VhostRoute
return rp.CreateConnection(host, url, remote)
},
},
WebSocketDialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
url := ctx.Value("url").(string)
host := getHostFromAddr(ctx.Value("host").(string))
remote := ctx.Value("remote").(string)
return rp.CreateConnection(host, url, remote)
},
//WebSocketDialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
//url := ctx.Value("url").(string)
//host := getHostFromAddr(ctx.Value("host").(string))
//remote := ctx.Value("remote").(string)
//return rp.CreateConnection(host, url, remote)
//},
BufferPool: newWrapPool(),
ErrorLog: log.New(newWrapLogger(), "", 0),
ErrorHandler: func(rw http.ResponseWriter, req *http.Request, err error) {
rw.WriteHeader(http.StatusNotFound)
rw.Write(getNotFoundPageContent())
},
}
rp.proxy = proxy
return rp