Lolia-FRP/lolia-frp
4
0
Fork 0
forked from Mxmilu666/frp
Code Pull Requests Activity

modify the getListener function && add TPS in dashboard

Browse Source
This commit is contained in:
XueBing
2016-12-18 23:35:14 +08:00
parent edf4bc431d
commit d93dd82ed9
2 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/utils/vhost/vhost.go
View File

@@ -76,10 +76,19 @@ func (v *VhostMuxer) Listen(p *config.ProxyServerConf) (ls []*Listener) {
return ls
}
func (v *VhostMuxer) getListener(name string) (l *Listener, exist bool) {
func (v *VhostMuxer) getListener(reqInfoMap map[string]string) (l *Listener, exist bool) {
v.mutex.RLock()
defer v.mutex.RUnlock()
//host
name := strings.ToLower(reqInfoMap["Host"])
// http
scheme := strings.ToLower(reqInfoMap["Scheme"])
if scheme == "http" || scheme == "" {
name = name + ":" + reqInfoMap["Path"]
}
// // first we check the full hostname
vr, found := v.registryRouter.get(name)
if found {
@@ -126,12 +135,7 @@ func (v *VhostMuxer) handle(c *conn.Conn) {
return
}
name := strings.ToLower(reqInfoMap["Host"])
// get listener by hostname
if reqInfoMap["Scheme"] == "http" {
name = name + ":" + reqInfoMap["Path"]
}
l, ok := v.getListener(name)
l, ok := v.getListener(reqInfoMap)
if !ok {
c.Close()
return