zwxi 11 miesięcy temu
rodzic
commit
822c65d626
1 zmienionych plików z 19 dodań i 18 usunięć
  1. 19 18
      middleware/token.go

+ 19 - 18
middleware/token.go

@@ -2,6 +2,7 @@ package middleware
 
 import (
 	"eta/eta_bridge/controller/resp"
+	"eta/eta_bridge/global"
 	"eta/eta_bridge/utils"
 	"github.com/gin-gonic/gin"
 )
@@ -31,24 +32,24 @@ func Token() gin.HandlerFunc {
 
 func InternalToken() gin.HandlerFunc {
 	return func(c *gin.Context) {
-		//token := c.Request.Header.Get("Authorization")
-		//if token == "" {
-		//	token = c.DefaultQuery("authorization", "")
-		//	if token == "" {
-		//		token = c.DefaultQuery("Authorization", "")
-		//	}
-		//}
-		//if token == "" {
-		//	resp.TokenError(nil, "非法访问", "非法访问", c)
-		//	c.Abort()
-		//	return
-		//}
-		//checkToken := utils.MD5(global.CONFIG.Serve.AppNameEn + global.CONFIG.Serve.Md5Key)
-		//if token != checkToken {
-		//	resp.TokenError(nil, "非法访问", "非法访问", c)
-		//	c.Abort()
-		//	return
-		//}
+		token := c.Request.Header.Get("Authorization")
+		if token == "" {
+			token = c.DefaultQuery("authorization", "")
+			if token == "" {
+				token = c.DefaultQuery("Authorization", "")
+			}
+		}
+		if token == "" {
+			resp.TokenError(nil, "非法访问", "非法访问", c)
+			c.Abort()
+			return
+		}
+		checkToken := utils.MD5(global.CONFIG.Serve.AppNameEn + global.CONFIG.Serve.Md5Key)
+		if token != checkToken {
+			resp.TokenError(nil, "非法访问", "非法访问", c)
+			c.Abort()
+			return
+		}
 		c.Next()
 	}
 }