mirrors/frp
1
0
Fork 0
mirror of https://github.com/fatedier/frp.git synced 2025-01-03 13:31:04 +07:00
Code Issues Packages Projects Releases Wiki Activity

fix_cfg_bug

Browse Source
This commit is contained in:
jcy 2016-04-13 17:44:14 +08:00
parent 2c39719cc0
commit a9398ada8f
2 changed files with 103 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
src/frp/models/client/config.go
View File

@ -47,11 +47,23 @@ func LoadConf(confFile string) (err error) {
tmpStr, ok = conf.Get("common", "server_addr")
if ok {
ServerAddr = tmpStr
} else {
tmpStr, ok = conf.Get("common", "SERVER_ADDR")
if ok {
ServerAddr = tmpStr
}
}
tmpStr, ok = conf.Get("common", "server_port")
if ok {
ServerPort, _ = strconv.ParseInt(tmpStr, 10, 64)
} else {
tmpStr, ok = conf.Get("common", "SERVER_PORT")
if ok {
ServerPort, _ = strconv.ParseInt(tmpStr, 10, 64)
}
}
tmpStr, ok = conf.Get("common", "log_file")
@ -62,11 +74,28 @@ func LoadConf(confFile string) (err error) {
} else {
LogWay = "file"
}
} else {
tmpStr, ok = conf.Get("common", "LOG_FILE")
if ok {
LogFile = tmpStr
if LogFile == "console" {
LogWay = "console"
} else {
LogWay = "file"
}
}
}
tmpStr, ok = conf.Get("common", "log_level")
if ok {
LogLevel = tmpStr
} else {
tmpStr, ok = conf.Get("common", "LOG_LEVEL")
if ok {
LogLevel = tmpStr
}
}
var authToken string
@ -74,7 +103,13 @@ func LoadConf(confFile string) (err error) {
if ok {
authToken = tmpStr
} else {
return fmt.Errorf("auth_token not found")
tmpStr, ok = conf.Get("common", "AUTH_TOKEN")
if ok {
authToken = tmpStr
} else {
return fmt.Errorf("auth_token not found")
}
}
// proxies
@ -90,8 +125,12 @@ func LoadConf(confFile string) (err error) {
// local_ip
proxyClient.LocalIp, ok = section["local_ip"]
if !ok {
// use 127.0.0.1 as default
proxyClient.LocalIp = "127.0.0.1"
proxyClient.LocalIp, ok = section["LOCAL_IP"]
if !ok {
// use 127.0.0.1 as default
proxyClient.LocalIp = "127.0.0.1"
}
}
// local_port
@ -102,7 +141,16 @@ func LoadConf(confFile string) (err error) {
return fmt.Errorf("Parse ini file error: proxy [%s] local_port error", proxyClient.Name)
}
} else {
return fmt.Errorf("Parse ini file error: proxy [%s] local_port not found", proxyClient.Name)
portStr, ok := section["LOCAL_PORT"]
if ok {
proxyClient.LocalPort, err = strconv.ParseInt(portStr, 10, 64)
if err != nil {
return fmt.Errorf("Parse ini file error: proxy [%s] local_port error", proxyClient.Name)
}
} else {
return fmt.Errorf("Parse ini file error: proxy [%s] local_port not found", proxyClient.Name)
}
}
// use_encryption
@ -110,8 +158,12 @@ func LoadConf(confFile string) (err error) {
useEncryptionStr, ok := section["use_encryption"]
if ok && useEncryptionStr == "true" {
proxyClient.UseEncryption = true
} else {
useEncryptionStr, ok := section["USE_ENCRYPTION"]
if ok && useEncryptionStr == "true" {
proxyClient.UseEncryption = true
}
}
ProxyClients[proxyClient.Name] = proxyClient
}
}

49
src/frp/models/server/config.go
View File

@ -47,11 +47,23 @@ func LoadConf(confFile string) (err error) {
tmpStr, ok = conf.Get("common", "bind_addr")
if ok {
BindAddr = tmpStr
} else {
tmpStr, ok = conf.Get("common", "BIND_ADDR")
if ok {
BindAddr = tmpStr
}
}
tmpStr, ok = conf.Get("common", "bind_port")
if ok {
BindPort, _ = strconv.ParseInt(tmpStr, 10, 64)
} else {
tmpStr, ok = conf.Get("common", "BIND_PORT")
if ok {
BindPort, _ = strconv.ParseInt(tmpStr, 10, 64)
}
}
tmpStr, ok = conf.Get("common", "log_file")
@ -62,11 +74,28 @@ func LoadConf(confFile string) (err error) {
} else {
LogWay = "file"
}
} else {
tmpStr, ok = conf.Get("common", "LOG_FILE")
if ok {
LogFile = tmpStr
if LogFile == "console" {
LogWay = "console"
} else {
LogWay = "file"
}
}
}
tmpStr, ok = conf.Get("common", "log_level")
if ok {
LogLevel = tmpStr
} else {
tmpStr, ok = conf.Get("common", "LOG_LEVEL")
if ok {
LogLevel = tmpStr
}
}
// servers
@ -77,12 +106,18 @@ func LoadConf(confFile string) (err error) {
proxyServer.AuthToken, ok = section["auth_token"]
if !ok {
return fmt.Errorf("Parse ini file error: proxy [%s] no auth_token found", proxyServer.Name)
proxyServer.AuthToken, ok = section["AUTH_TOKEN"]
if !ok {
return fmt.Errorf("Parse ini file error: proxy [%s] no auth_token found", proxyServer.Name)
}
}
proxyServer.BindAddr, ok = section["bind_addr"]
if !ok {
proxyServer.BindAddr = "0.0.0.0"
proxyServer.BindAddr, ok = section["BIND_ADDR"]
if !ok {
proxyServer.BindAddr = "0.0.0.0"
}
}
portStr, ok := section["listen_port"]
@ -92,7 +127,15 @@ func LoadConf(confFile string) (err error) {
return fmt.Errorf("Parse ini file error: proxy [%s] listen_port error", proxyServer.Name)
}
} else {
return fmt.Errorf("Parse ini file error: proxy [%s] listen_port not found", proxyServer.Name)
portStr, ok := section["LISTEN_PORT"]
if ok {
proxyServer.ListenPort, err = strconv.ParseInt(portStr, 10, 64)
if err != nil {
return fmt.Errorf("Parse ini file error: proxy [%s] listen_port error", proxyServer.Name)
}
} else {
return fmt.Errorf("Parse ini file error: proxy [%s] listen_port not found", proxyServer.Name)
}
}
proxyServer.Init()