HI-VPN/hi-server
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update: device middleware

Browse Source
This commit is contained in:
EUForest 2025-10-13 19:16:05 +08:00
parent 39306f3043
commit e99058b969
1 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
internal/middleware/deviceMiddleware.go
View File

@ -28,9 +28,23 @@ const (
func DeviceMiddleware(srvCtx *svc.ServiceContext) func(c *gin.Context) { func DeviceMiddleware(srvCtx *svc.ServiceContext) func(c *gin.Context) {
return func(c *gin.Context) { return func(c *gin.Context) {
loginType := c.GetString(string(constant.LoginType))
if loginType == "" { if !srvCtx.Config.Device.Enable {
c.Next()
return
}
if srvCtx.Config.Device.SecuritySecret == "" {
result.HttpResult(c, nil, errors.Wrapf(xerr.NewErrCode(xerr.SecretIsEmpty), "Secret is empty"))
c.Abort()
return
}
var loginType string
if c.Value(constant.CtxKeyUser) == nil {
loginType = c.GetHeader("Login-Type") loginType = c.GetHeader("Login-Type")
} else {
loginType = c.GetString(string(constant.LoginType))
} }
if loginType != "device" { if loginType != "device" {
@ -40,10 +54,6 @@ func DeviceMiddleware(srvCtx *svc.ServiceContext) func(c *gin.Context) {
c.Request = c.Request.WithContext(context.WithValue(c.Request.Context(), constant.LoginType, loginType)) c.Request = c.Request.WithContext(context.WithValue(c.Request.Context(), constant.LoginType, loginType))
if !srvCtx.Config.Device.Enable || srvCtx.Config.Device.SecuritySecret == "" {
c.Next()
return
}
rw := NewResponseWriter(c, srvCtx) rw := NewResponseWriter(c, srvCtx)
if !rw.Decrypt() { if !rw.Decrypt() {
result.HttpResult(c, nil, errors.Wrapf(xerr.NewErrCode(xerr.InvalidCiphertext), "Invalid ciphertext")) result.HttpResult(c, nil, errors.Wrapf(xerr.NewErrCode(xerr.InvalidCiphertext), "Invalid ciphertext"))