Victor_Jay
daf87f895b
- 新增 internal/session 包：Session 结构体、Store 接口、MemoryStore（内存+后台清理）、RedisStore 预留 - Cookie 从 3 个精简为 1 个 mlb_sid（HttpOnly），移除 JWT access/refresh cookie - 会话过期采用滑动窗口：每次请求自动续期，记住我 30 天无操作过期 - AuthMiddleware/AuthAdmin/Maintenance 中间件改用 SessionManager.Validate() - AuthService Login/Register/ConfirmRestore 去除 token 生成，返回 *model.User - Controller 层在登录/注册后调用 SessionManager.Create 创建会话 - AdminService UpdateUserStatus/ResetUserToken 同步销毁 session 实现即时退登 - 改密/注销时通过 DestroyByUID 删除所有 session（替换 TokenVersion 检查） - config.yaml 新增 session 配置段（idle_timeout/remember_timeout/cleanup_interval） - TokenService/auth_parser/auth_token 标记废弃，移除 JWT 到期字段依赖
55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
package middleware
|
||
|
||
import (
|
||
"crypto/rand"
|
||
"encoding/hex"
|
||
"net/http"
|
||
|
||
"metazone.cc/metalab/internal/config"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
const csrfTokenLen = 32
|
||
|
||
// SetCSRFToken 在首次页面访问时下发 CSRF Cookie（由页面路由中间件调用）
|
||
func SetCSRFToken(c *gin.Context, cfg *config.Config) string {
|
||
secure := cfg.Server.Mode != "debug"
|
||
if existing, err := c.Cookie(csrfCookieName); err == nil && existing != "" {
|
||
c.Set(csrfMetaName, existing)
|
||
return existing
|
||
}
|
||
|
||
token, err := generateCSRFToken()
|
||
if err != nil {
|
||
token = "fallback-" + hex.EncodeToString([]byte("metazone"))
|
||
}
|
||
|
||
c.SetSameSite(http.SameSiteStrictMode)
|
||
maxAge := cfg.Session.RememberTimeout * 60
|
||
c.SetCookie(csrfCookieName, token, maxAge, "/", "", secure, false)
|
||
c.Set(csrfMetaName, token)
|
||
return token
|
||
}
|
||
|
||
// generateCSRFToken 生成 64 字符十六进制随机 CSRF token
|
||
func generateCSRFToken() (string, error) {
|
||
b := make([]byte, csrfTokenLen)
|
||
if _, err := rand.Read(b); err != nil {
|
||
return "", err
|
||
}
|
||
return hex.EncodeToString(b), nil
|
||
}
|
||
|
||
// constantTimeEq 恒定时间字符串比较（防时序攻击）
|
||
func constantTimeEq(a, b string) bool {
|
||
if len(a) != len(b) {
|
||
return false
|
||
}
|
||
var result byte
|
||
for i := 0; i < len(a); i++ {
|
||
result |= a[i] ^ b[i]
|
||
}
|
||
return result == 0
|
||
}
|