TRKJ/EdgeAdmin
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAdmin.git synced 2025-11-09 00:20:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

登录界面屏蔽个别敏感区域

Browse Source
This commit is contained in:
GoEdgeLab
2023-05-27 18:12:40 +08:00
parent 229e2c5144
commit c2804868dc
1 changed files with 24 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
internal/web/actions/default/index/index.go
View File

@@ -13,15 +13,19 @@ import (
adminserverutils "github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/settings/server/admin-server-utils" adminserverutils "github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/settings/server/admin-server-utils"
"github.com/TeaOSLab/EdgeAdmin/internal/web/helpers" "github.com/TeaOSLab/EdgeAdmin/internal/web/helpers"
"github.com/TeaOSLab/EdgeCommon/pkg/configutils" "github.com/TeaOSLab/EdgeCommon/pkg/configutils"
"github.com/TeaOSLab/EdgeCommon/pkg/iplibrary"
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/dao" "github.com/TeaOSLab/EdgeCommon/pkg/rpc/dao"
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb" "github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
"github.com/iwind/TeaGo/actions" "github.com/iwind/TeaGo/actions"
"github.com/iwind/TeaGo/lists"
"github.com/iwind/TeaGo/types" "github.com/iwind/TeaGo/types"
stringutil "github.com/iwind/TeaGo/utils/string" stringutil "github.com/iwind/TeaGo/utils/string"
"net" "net"
"time" "time"
) )
const regionDenyMessage = "当前软件系统暂时不为你所在的区域提供服务。"
type IndexAction struct { type IndexAction struct {
actionutils.ParentAction actionutils.ParentAction
} }
@@ -36,6 +40,11 @@ func (this *IndexAction) RunGet(params struct {
Auth *helpers.UserShouldAuth Auth *helpers.UserShouldAuth
}) { }) {
if !this.checkRegion() {
this.WriteString(regionDenyMessage)
return
}
// 是否自动从HTTP跳转到HTTPS // 是否自动从HTTP跳转到HTTPS
if this.Request.TLS == nil { if this.Request.TLS == nil {
httpsPort, _ := adminserverutils.ReadServerHTTPS() httpsPort, _ := adminserverutils.ReadServerHTTPS()
@@ -125,6 +134,11 @@ func (this *IndexAction) RunPost(params struct {
Auth *helpers.UserShouldAuth Auth *helpers.UserShouldAuth
CSRF *actionutils.CSRF CSRF *actionutils.CSRF
}) { }) {
if !this.checkRegion() {
this.Fail(regionDenyMessage)
return
}
params.Must. params.Must.
Field("username", params.Username). Field("username", params.Username).
Require("请输入用户名"). Require("请输入用户名").
@@ -216,3 +230,13 @@ func (this *IndexAction) RunPost(params struct {
this.Success() this.Success()
} }
// 检查登录区域
func (this *IndexAction) checkRegion() bool {
var ip = this.RequestRemoteIP()
var result = iplibrary.LookupIP(ip)
if result != nil && result.IsOk() && result.CountryId() > 0 && lists.ContainsInt64([]int64{10}, result.CountryId()) {
return false
}
return true
}