fix: fix the bug that authentication always fails after token change (#…

…59)
Li4n0 · Sep 13, 2022 · 3fa2cae · 3fa2cae
1 parent fe31622
commit 3fa2cae
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/pkg/server/controller.go b/pkg/server/controller.go
@@ -18,7 +18,7 @@ import (
 
 func (revsuit *Revsuit) auth(c *gin.Context) {
 	c.SetSameSite(http.SameSiteLaxMode)
-	c.SetCookie("token", c.Request.Header["Token"][0], 0, revsuit.config.AdminPathPrefix, "", false, true)
+	c.SetCookie("token", c.Request.Header.Get("Token"), 0, revsuit.config.AdminPathPrefix, "", false, true)
 	c.String(200, "pong")
 }
 

diff --git a/pkg/server/router.go b/pkg/server/router.go
@@ -32,13 +32,12 @@ func (revsuit *Revsuit) registerPlatformRouter() {
 	// /api need Authorization
 	api := revsuit.http.Router.Group(revsuit.config.AdminPathPrefix + "/api")
 	api.Use(func(c *gin.Context) {
-		token, err := c.Cookie("token")
-		if err == http.ErrNoCookie {
-			token = c.Request.Header.Get("Token")
-		}
+		token, _ := c.Cookie("token")
 		if token != revsuit.http.Token {
-			c.Abort()
-			c.Status(403)
+			if c.Request.Header.Get("Token") != revsuit.http.Token {
+				c.Abort()
+				c.Status(403)
+			}
 		}
 	})
 	revsuit.http.ApiGroup = api