Alex-Devera/3x-ui
1
0
Fork 0
mirror of https://github.com/MHSanaei/3x-ui.git synced 2025-09-10 20:26:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Update subController.go

Browse source
This commit is contained in:
mhsanaei 2024-08-11 18:26:19 +02:00
parent bb70e61fcd
commit e4b9114463
1 changed files with 8 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
sub/subController.go
View file

@ -56,7 +56,7 @@ func (a *SUBController) initRouter(g *gin.RouterGroup) {
func (a *SUBController) subs(c *gin.Context) {
subId := c.Param("subid")
var host string
if h, err := getHostFromXFH(c.GetHeader("X-Forwarded-Host")); err != nil {
if h, err := getHostFromXFH(c.GetHeader("X-Forwarded-Host")); err == nil {
host = h
}
if host == "" {
@ -94,7 +94,7 @@ func (a *SUBController) subs(c *gin.Context) {
func (a *SUBController) subJsons(c *gin.Context) {
subId := c.Param("subid")
var host string
if h, err := getHostFromXFH(c.GetHeader("X-Forwarded-Host")); err != nil {
if h, err := getHostFromXFH(c.GetHeader("X-Forwarded-Host")); err == nil {
host = h
}
if host == "" {
@ -121,16 +121,13 @@ func (a *SUBController) subJsons(c *gin.Context) {
}
}
func getHostFromXFH(s string) (host string, err error) {
// X-Forwarded-Host can actually be a host:port pair, so we need to
// split it
if strings.Contains(host, ":") {
if realHost, _, err := net.SplitHostPort(host); err == nil {
return realHost, nil
} else {
func getHostFromXFH(s string) (string, error) {
if strings.Contains(s, ":") {
realHost, _, err := net.SplitHostPort(s)
if err != nil {
return "", err
}
} else {
return s, nil
return realHost, nil
}
return s, nil
}