archive/revsuit
1
0
Fork 0
mirror of https://github.com/Li4n0/revsuit.git synced 2026-09-21 22:30:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Li4n0
2022-09-14 01:06:26 +08:00
committed by GitHub
parent fe316220a1
commit 3fa2cae862
2 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/server/controller.go
+1 -1
View File
@@ -18,7 +18,7 @@ import (
func (revsuit *Revsuit) auth(c *gin.Context) { func (revsuit *Revsuit) auth(c *gin.Context) {
c.SetSameSite(http.SameSiteLaxMode) 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") c.String(200, "pong")
} }
pkg/server/router.go
+5 -6
View File
@@ -32,13 +32,12 @@ func (revsuit *Revsuit) registerPlatformRouter() {
// /api need Authorization // /api need Authorization
api := revsuit.http.Router.Group(revsuit.config.AdminPathPrefix + "/api") api := revsuit.http.Router.Group(revsuit.config.AdminPathPrefix + "/api")
api.Use(func(c *gin.Context) { api.Use(func(c *gin.Context) {
token, err := c.Cookie("token") token, _ := c.Cookie("token")
if err == http.ErrNoCookie {
token = c.Request.Header.Get("Token")
}
if token != revsuit.http.Token { if token != revsuit.http.Token {
c.Abort() if c.Request.Header.Get("Token") != revsuit.http.Token {
c.Status(403) c.Abort()
c.Status(403)
}
} }
}) })
revsuit.http.ApiGroup = api revsuit.http.ApiGroup = api