浏览代码

修复登录验证可能的越界问题

Signed-off-by: allan716 <[email protected]>
allan716 3 年之前
父节点
当前提交
81d9d13eb2
共有 1 个文件被更改,包括 3 次插入2 次删除
  1. 3 2
      internal/backend/middle/auth.go

+ 3 - 2
internal/backend/middle/auth.go

@@ -14,12 +14,13 @@ func CheckAuth() gin.HandlerFunc {
 
 	return func(context *gin.Context) {
 		authHeader := context.Request.Header.Get("Authorization")
-		if len(authHeader) <= 1 {
+		fields := strings.Fields(authHeader)
+		if len(fields) != 2 {
 			context.JSON(http.StatusUnauthorized, backend.ReplyCheckAuth{Message: "Request Header Authorization Error"})
 			context.Abort()
 			return
 		}
-		nowAccessToken := strings.Fields(authHeader)[1]
+		nowAccessToken := fields[1]
 		if nowAccessToken == "" || nowAccessToken != common.GetAccessToken() {
 			context.JSON(http.StatusUnauthorized, backend.ReplyCheckAuth{Message: "AccessToken Error"})
 			context.Abort()