xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-09-17 17:16:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: improve session handling for non-base64 encoded session secrets

Browse Source
This commit is contained in:
Laisky.Cai 2024-12-20 05:56:35 +00:00
parent 09bfd61f2e
commit 524606c4ae
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
main.go
View File

@ -104,12 +104,13 @@ func main() {
sessionSecret, err := base64.StdEncoding.DecodeString(config.SessionSecret) sessionSecret, err := base64.StdEncoding.DecodeString(config.SessionSecret)
if err != nil { if err != nil {
logger.SysLog("session secret is not base64 encoded, using raw value instead") logger.SysLog("session secret is not base64 encoded, using raw value instead")
sessionSecret = []byte(config.SessionSecret) store := cookie.NewStore([]byte(config.SessionSecret))
server.Use(sessions.Sessions("session", store))
} else {
store := cookie.NewStore(sessionSecret, sessionSecret)
server.Use(sessions.Sessions("session", store))
} }
store := cookie.NewStore(sessionSecret, sessionSecret)
server.Use(sessions.Sessions("session", store))
router.SetRouter(server, buildFS) router.SetRouter(server, buildFS)
var port = os.Getenv("PORT") var port = os.Getenv("PORT")
if port == "" { if port == "" {