447 lines
20 KiB
Go
447 lines
20 KiB
Go
// Copyright 2022 E99p1ant. All rights reserved.
|
||
// Use of this source code is governed by a MIT-style
|
||
// license that can be found in the LICENSE file.
|
||
|
||
package route
|
||
|
||
import (
|
||
"encoding/gob"
|
||
"io"
|
||
"net/http"
|
||
"regexp"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/flamego/cache"
|
||
cacheRedis "github.com/flamego/cache/redis"
|
||
"github.com/flamego/csrf"
|
||
"github.com/flamego/flamego"
|
||
"github.com/flamego/recaptcha"
|
||
"github.com/flamego/session"
|
||
"github.com/flamego/session/mysql"
|
||
sessionRedis "github.com/flamego/session/redis"
|
||
"github.com/flamego/template"
|
||
"github.com/sirupsen/logrus"
|
||
|
||
"github.com/tamakyi/TamaBox/internal/captcha"
|
||
"github.com/tamakyi/TamaBox/internal/conf"
|
||
"github.com/tamakyi/TamaBox/internal/context"
|
||
"github.com/tamakyi/TamaBox/internal/form"
|
||
"github.com/tamakyi/TamaBox/internal/i18n"
|
||
"github.com/tamakyi/TamaBox/internal/mail"
|
||
"github.com/tamakyi/TamaBox/internal/route/chat" // 私聊路由
|
||
"github.com/tamakyi/TamaBox/internal/security/autoban"
|
||
"github.com/tamakyi/TamaBox/internal/security/ipinfo"
|
||
"github.com/tamakyi/TamaBox/internal/shared"
|
||
templatepkg "github.com/tamakyi/TamaBox/internal/template"
|
||
"github.com/tamakyi/TamaBox/route"
|
||
"github.com/tamakyi/TamaBox/route/auth"
|
||
"github.com/tamakyi/TamaBox/route/pixel"
|
||
"github.com/tamakyi/TamaBox/route/question"
|
||
"github.com/tamakyi/TamaBox/route/service"
|
||
"github.com/tamakyi/TamaBox/route/user"
|
||
"github.com/tamakyi/TamaBox/route/vote"
|
||
"github.com/tamakyi/TamaBox/static"
|
||
"github.com/tamakyi/TamaBox/templates"
|
||
)
|
||
|
||
// indexNowKeyPattern 限制 IndexNow 密钥文件名的合法形态：32 位小写十六进制 + ".txt"。
|
||
// 用于防止把路由参数直接拼进 embed FS 路径时发生路径穿越（如 "../robots.txt"）。
|
||
var indexNowKeyPattern = regexp.MustCompile(`^[0-9a-f]{32}\.txt$`)
|
||
|
||
// CSRFUserSessionKey 是 CSRF token 绑定的会话键，必须与登录时写入的键一致
|
||
// （见 route/auth/login.go 的 ctx.Session.Set("uid", ...)）。flamego/csrf 默认用
|
||
// "userID"，不对齐则 token id 恒为 "0"，匿名访客从公开页取到的 token 对所有已登录
|
||
// 用户均有效，CSRF 退化为全站公开常量。
|
||
// 本常量被 csrf_binding_test.go 引用作为单一事实来源，误改会让回归测试失败。
|
||
const CSRFUserSessionKey = "uid"
|
||
|
||
// securityHeaders adds security-related HTTP headers to every response.
|
||
func securityHeaders() flamego.Handler {
|
||
return func(ctx flamego.Context) {
|
||
w := ctx.ResponseWriter()
|
||
w.Header().Set("X-Content-Type-Options", "nosniff")
|
||
w.Header().Set("X-Frame-Options", "DENY")
|
||
w.Header().Set("X-XSS-Protection", "1; mode=block")
|
||
w.Header().Set("Referrer-Policy", "strict-origin-when-cross-origin")
|
||
// HSTS：仅当站点以 https 对外提供服务时下发，避免 http 部署下把用户锁死在 https。
|
||
if strings.HasPrefix(conf.App.ExternalURL, "https") {
|
||
w.Header().Set("Strict-Transport-Security", "max-age=31536000; includeSubDomains")
|
||
}
|
||
}
|
||
}
|
||
|
||
func New() *flamego.Flame {
|
||
f := flamego.Classic()
|
||
if conf.App.Production {
|
||
flamego.SetEnv(flamego.EnvTypeProd)
|
||
}
|
||
|
||
// 自动封禁实例：整个请求链路共享同一个，供中间件与 f.NotFound 复用，
|
||
// 避免对业务性 302→/（语言切换、登出等）误判为路径扫描。
|
||
// 404 是否计入由 ban_track_notfound 控制（默认 false，仅敏感扫描路径计数）。
|
||
ab := autoban.New(autoban.Options{
|
||
Enabled: conf.Security.EnableAutoBan,
|
||
TrackNotFound: conf.Security.BanTrackNotFound,
|
||
ThresholdSensitive: conf.Security.BanThresholdSensitive,
|
||
Threshold404: conf.Security.BanThreshold404,
|
||
Window: time.Duration(conf.Security.BanWindowSeconds) * time.Second,
|
||
Duration: time.Duration(conf.Security.BanDurationMinutes) * time.Minute,
|
||
AlertEmail: conf.Security.SecurityAlertEmail,
|
||
// 告警发送前先离线识别来源 IP 的网络归属（CDN/云厂商/WAF，零外部请求），
|
||
// 识别失败不影响告警发送（ipinfo.Lookup 内部已折叠错误）。
|
||
SendAlert: func(ip string, paths []string, durationMinutes int) error {
|
||
provider, proxied := ipinfo.Lookup(ip)
|
||
return mail.SendSecurityAlertMail(ip, provider, proxied, paths, durationMinutes)
|
||
},
|
||
})
|
||
|
||
templateFS, err := template.EmbedFS(templates.FS, ".", []string{".html"})
|
||
if err != nil {
|
||
logrus.WithError(err).Fatal("Failed to embed templates file system")
|
||
}
|
||
|
||
gob.Register(time.Time{})
|
||
gob.Register(context.Flash{})
|
||
gob.Register(captcha.StoredAnswer{})
|
||
var sessionStorage interface{}
|
||
initer := session.MemoryIniter()
|
||
if conf.Database.DSN != "" {
|
||
initer = mysql.Initer()
|
||
sessionStorage = mysql.Config{
|
||
DSN: conf.Database.DSN,
|
||
Lifetime: 7 * 24 * time.Hour,
|
||
}
|
||
}
|
||
if conf.Redis.Addr != "" {
|
||
initer = sessionRedis.Initer()
|
||
sessionStorage = sessionRedis.Config{
|
||
Options: &sessionRedis.Options{
|
||
Addr: conf.Redis.Addr,
|
||
Password: conf.Redis.Password,
|
||
DB: 1,
|
||
},
|
||
Lifetime: 7 * 24 * time.Hour,
|
||
}
|
||
}
|
||
sessioner := session.Sessioner(session.Options{
|
||
Initer: initer,
|
||
Config: sessionStorage,
|
||
// 显式声明 cookie 安全属性：HTTPOnly 防 XSS 窃取会话；SameSite=Lax 阻断跨站表单
|
||
// POST；Secure 仅在 https 对外服务时下发，避免 http 本地开发下 cookie 不生效
|
||
// （缺失 Secure 时，HSTS 尚未落地的首次访问可被 MITM 诱导明文请求窃取会话）。
|
||
Cookie: session.CookieOptions{
|
||
HTTPOnly: true,
|
||
SameSite: http.SameSiteLaxMode,
|
||
Secure: strings.HasPrefix(conf.App.ExternalURL, "https"),
|
||
},
|
||
})
|
||
|
||
f.Use(flamego.Static(flamego.StaticOptions{
|
||
FileSystem: http.FS(static.FS),
|
||
Prefix: "/static",
|
||
// 静态资源随二进制发布，内容变更需重新部署；1 天缓存是安全折中
|
||
CacheControl: func() string {
|
||
return "public, max-age=86400"
|
||
},
|
||
}))
|
||
|
||
f.Use(securityHeaders())
|
||
f.Use(ipBlacklistMiddleware())
|
||
f.Use(autobanMiddleware(ab))
|
||
// 全局请求体上限：必须在任何会读取 body 的中间件（尤其 Contexter 内的 CSRF 校验）
|
||
// 之前执行，否则超大 multipart 会在 Bind 的路由级限制生效前落盘临时文件耗尽磁盘。
|
||
f.Use(requestBodyLimitMiddleware())
|
||
|
||
reqUserSignOut := context.Toggle(&context.ToggleOptions{UserSignOutRequired: true})
|
||
reqUserSignIn := context.Toggle(&context.ToggleOptions{UserSignInRequired: true})
|
||
|
||
f.Group("", func() {
|
||
f.Get("/", route.Home)
|
||
f.Get("/pixel", reqUserSignIn, pixel.Index)
|
||
f.Get("/sponsor", route.Sponsor)
|
||
f.Get("/change-logs", route.ChangeLogs)
|
||
f.Get("/about-me", route.AboutMe)
|
||
// 社区规则：公开页，处置时的引用依据。
|
||
f.Get("/rules", route.Rules)
|
||
f.Get("/robots.txt", func(c context.Context) {
|
||
fs, err := static.FS.Open("robots.txt")
|
||
if err != nil {
|
||
c.ResponseWriter().WriteHeader(404)
|
||
return
|
||
}
|
||
defer func() { _ = fs.Close() }()
|
||
c.ResponseWriter().Header().Set("Content-Type", "text/plain; charset=utf-8")
|
||
c.ResponseWriter().Header().Set("Cache-Control", "public, max-age=86400")
|
||
_, _ = io.Copy(c.ResponseWriter(), fs)
|
||
})
|
||
f.Get("/favicon.ico", func(c context.Context) {
|
||
fs, _ := static.FS.Open("favicon.ico")
|
||
defer func() { _ = fs.Close() }()
|
||
c.ResponseWriter().Header().Set("Content-Type", "image/x-icon")
|
||
c.ResponseWriter().Header().Set("Cache-Control", "public, max-age=86400")
|
||
_, _ = io.Copy(c.ResponseWriter(), fs)
|
||
})
|
||
|
||
f.Get("/sitemap.xml", func(c context.Context) {
|
||
fs, err := static.FS.Open("sitemap.xml")
|
||
if err != nil {
|
||
c.ResponseWriter().WriteHeader(404)
|
||
return
|
||
}
|
||
defer func() { _ = fs.Close() }()
|
||
c.ResponseWriter().Header().Set("Content-Type", "application/xml")
|
||
c.ResponseWriter().Header().Set("Content-Type", "text/xml; charset=utf-8")
|
||
c.ResponseWriter().Header().Set("Cache-Control", "public, max-age=86400")
|
||
_, _ = io.Copy(c.ResponseWriter(), fs)
|
||
})
|
||
|
||
f.Get("/llms.txt", func(c context.Context) {
|
||
fs, err := static.FS.Open("llms.txt")
|
||
if err != nil {
|
||
c.ResponseWriter().WriteHeader(404)
|
||
return
|
||
}
|
||
defer func() { _ = fs.Close() }()
|
||
c.ResponseWriter().Header().Set("Content-Type", "text/plain; charset=utf-8")
|
||
c.ResponseWriter().Header().Set("Cache-Control", "public, max-age=86400")
|
||
_, _ = io.Copy(c.ResponseWriter(), fs)
|
||
})
|
||
|
||
f.Get("/llms-full.txt", func(c context.Context) {
|
||
fs, err := static.FS.Open("llms-full.txt")
|
||
if err != nil {
|
||
c.ResponseWriter().WriteHeader(404)
|
||
return
|
||
}
|
||
defer func() { _ = fs.Close() }()
|
||
c.ResponseWriter().Header().Set("Content-Type", "text/plain; charset=utf-8")
|
||
c.ResponseWriter().Header().Set("Cache-Control", "public, max-age=86400")
|
||
_, _ = io.Copy(c.ResponseWriter(), fs)
|
||
})
|
||
|
||
// IndexNow 密钥文件：static/index-now/<32 位十六进制>.txt，供 Bing / Yandex 等
|
||
// 搜索引擎校验站点所有权。
|
||
// ⚠️ flamego Static 的 Prefix 是 "/static"，不会接管本路径，必须显式注册，
|
||
// 否则会落到 f.NotFound 被 302 到首页，校验必然失败。
|
||
// 文件名经 indexNowKeyPattern 白名单校验，避免路径穿越读 embed FS 内其它文件。
|
||
f.Get("/index-now/{key}", func(c context.Context) {
|
||
key := c.Param("key")
|
||
if !indexNowKeyPattern.MatchString(key) {
|
||
c.ResponseWriter().WriteHeader(http.StatusNotFound)
|
||
return
|
||
}
|
||
fs, err := static.FS.Open("index-now/" + key)
|
||
if err != nil {
|
||
c.ResponseWriter().WriteHeader(http.StatusNotFound)
|
||
return
|
||
}
|
||
defer func() { _ = fs.Close() }()
|
||
c.ResponseWriter().Header().Set("Content-Type", "text/plain; charset=utf-8")
|
||
c.ResponseWriter().Header().Set("Cache-Control", "public, max-age=86400")
|
||
_, _ = io.Copy(c.ResponseWriter(), fs)
|
||
})
|
||
|
||
f.Group("", func() {
|
||
f.Combo("/register").Get(auth.Register).Post(form.Bind(form.Register{}), rateLimitMiddleware(10, 1*time.Minute), auth.RegisterAction)
|
||
f.Combo("/login").Get(auth.Login).Post(form.Bind(form.Login{}), rateLimitMiddleware(20, 1*time.Minute), auth.LoginAction)
|
||
f.Combo("/forgot-password").Get(auth.ForgotPassword).Post(form.Bind(form.ForgotPassword{}), rateLimitMiddleware(5, 5*time.Minute), auth.ForgotPasswordAction)
|
||
f.Combo("/recover-password").Get(auth.RecoverPassword).Post(form.Bind(form.RecoverPassword{}), auth.RecoverPasswordAction)
|
||
}, reqUserSignOut)
|
||
|
||
// 页面密码验证路由（独立于密码保护中间件）；限流防爆破
|
||
f.Combo("/_/{domain}/password").Get(question.PasswordPage).Post(rateLimitMiddleware(10, 1*time.Minute), question.VerifyPagePassword)
|
||
|
||
// 独立的分享图路由（不经过 Pager 中间件，直接返回图片）
|
||
// 注意：仍挂载 question.Pager，页面设置了密码时必须先通过密码验证，
|
||
// 否则分享图会绕过页面密码保护（与单题分享图行为一致）。
|
||
f.Group("/_/{domain}", func() {
|
||
f.Get("/share-card", question.ShareCard)
|
||
}, question.Pager)
|
||
|
||
f.Group("/_/{domain}", func() {
|
||
f.Combo("").Get(question.List).Post(form.Bind(form.NewQuestion{}), rateLimitMiddleware(30, 1*time.Minute), question.New)
|
||
f.Group("/{questionID}", func() {
|
||
f.Get("", question.Item)
|
||
f.Get("/share-card", question.QuestionShareCard)
|
||
f.Post("/delete", question.Delete)
|
||
f.Post("/set-private", question.SetPrivate)
|
||
f.Post("/set-public", question.SetPublic)
|
||
f.Post("/pin", question.TogglePin)
|
||
f.Post("/read", question.SetRead)
|
||
f.Post("/unread", question.SetUnread)
|
||
f.Post("/answer", reqUserSignIn, form.Bind(form.PublishAnswerQuestion{}), question.PublishAnswer)
|
||
// 举报入口：公开、不要求登录（主要目击者是匿名访客），限流防刷。
|
||
// 举报只写入 content_reports，不改变任何可见性；处置由箱主或 CLI 完成。
|
||
f.Post("/report", form.Bind(form.ReportQuestion{}), rateLimitMiddleware(10, 1*time.Minute), question.Report)
|
||
f.Post("/follow-up", form.Bind(form.PublishFollowUpQuestion{}), question.PublishFollowUp)
|
||
f.Post("/follow-up/{followUpID}/answer", reqUserSignIn, form.Bind(form.AnswerFollowUpQuestion{}), question.AnswerFollowUp)
|
||
}, question.Questioner)
|
||
|
||
// 私聊路由（需登录，Token 标识对话）
|
||
f.Group("/chat/{token}", func() {
|
||
f.Get("", chat.ConversationPage)
|
||
f.Post("/send", rateLimitMiddleware(20, 1*time.Minute), chat.SendMessage)
|
||
f.Post("/delete", chat.DeleteConversation)
|
||
f.Post("/block", chat.ToggleBlockUser)
|
||
}, reqUserSignIn)
|
||
}, question.Pager) // Pager 中间件已内置页面密码检查
|
||
|
||
f.Get("/explore", user.Explore)
|
||
|
||
// 账号停用成功页（公开）：停用后会话已清空，不能再要求登录，
|
||
// 故为独立公开页，让用户明确看到「已停用」而非被丢回登录页。
|
||
f.Get("/deactivated", user.DeactivatedProfile)
|
||
// 处置停用页（L4）：必须注册在公开区，被处置箱主自己也只能看到这一页。
|
||
f.Get("/restricted", user.RestrictedProfile)
|
||
|
||
// 匿名投票参与端（公开）：token 链接入口
|
||
f.Group("/vote/{token}", func() {
|
||
f.Get("", vote.Page)
|
||
// 提交限流优先按登录用户/签名 vote_fp 计数，IP 仅作高阈值兜底，避免共享出口误伤。
|
||
f.Post("/submit", voteSubmitRateLimitMiddleware(20, 1*time.Minute), vote.Submit)
|
||
})
|
||
|
||
// 公开页语言切换：语言偏好写入 session 后跳回原页面。此前写独立 lang cookie
|
||
// 会被 session 中间件的 Set-Cookie 整体覆盖而失效，故托管在 session 中。
|
||
f.Get("/i18n/{lang}", func(c context.Context) {
|
||
lang := i18n.Resolve(c.Param("lang"))
|
||
c.Session.Set("lang", lang)
|
||
to := c.Query("to")
|
||
// 仅允许站内相对路径，防开放重定向：拒绝空、非 "/" 开头、协议相对路径 "//"，
|
||
// 以及含反斜杠 / 控制字符的变体。
|
||
if to == "" || !strings.HasPrefix(to, "/") ||
|
||
strings.HasPrefix(to, "//") || strings.ContainsAny(to, "\\\x00\r\n\t") {
|
||
to = "/"
|
||
}
|
||
c.Redirect(to)
|
||
})
|
||
|
||
// 算数验证码 API（仅在 captcha_type = "math" 时生效，但路由本身不依赖中间件）
|
||
// 加限流：防止匿名脚本无限生成/轮询爆破（答案本身对客户端可见，限流是主要节流手段）。
|
||
f.Get("/api/captcha/generate", rateLimitMiddleware(20, 1*time.Minute), captcha.GenerateHandler)
|
||
f.Post("/api/captcha/verify", rateLimitMiddleware(20, 1*time.Minute), captcha.VerifyHandler)
|
||
|
||
f.Group("/user", func() {
|
||
f.Get("/questions", user.QuestionList)
|
||
|
||
// 匿名投票管理（箱主侧）
|
||
f.Group("/votes", func() {
|
||
f.Get("", user.VoteList)
|
||
f.Get("/new", user.VoteNew)
|
||
f.Post("", rateLimitMiddleware(10, 1*time.Minute), user.VoteCreate)
|
||
f.Get("/{voteID}/edit", user.VoteEdit)
|
||
f.Post("/{voteID}", rateLimitMiddleware(10, 1*time.Minute), user.VoteUpdate)
|
||
f.Post("/{voteID}/publish", user.VotePublish)
|
||
f.Post("/{voteID}/close", user.VoteClose)
|
||
f.Post("/{voteID}/delete", user.VoteDelete)
|
||
f.Get("/{voteID}/results", user.VoteResults)
|
||
f.Get("/{voteID}/preview", user.VotePreview)
|
||
// 导出为只读操作（幂等、无副作用），用 GET 链接下载，避免表单提交
|
||
f.Get("/{voteID}/export", user.VoteExportCSV)
|
||
})
|
||
|
||
f.Group("/profile", func() {
|
||
f.Get("", user.Profile)
|
||
f.Post("/update", form.Bind(form.UpdateProfile{}), rateLimitMiddleware(10, 1*time.Minute), user.UpdateProfile)
|
||
f.Post("/export", user.ExportProfile)
|
||
f.Post("/export-json", user.ExportProfileJSON)
|
||
f.Post("/theme", form.Bind(form.UpdateProfileTheme{}), rateLimitMiddleware(10, 1*time.Minute), user.UpdateTheme)
|
||
f.Combo("/deactivate").Get(user.DeactivateProfile).Post(user.DeactivateProfileAction)
|
||
|
||
f.Post("/update-page-password", form.Bind(form.UpdatePagePassword{}), rateLimitMiddleware(10, 1*time.Minute), user.UpdatePagePassword)
|
||
f.Post("/update-ask-password", form.Bind(form.UpdateAskPassword{}), rateLimitMiddleware(10, 1*time.Minute), user.UpdateAskPassword)
|
||
|
||
f.Post("/update-chat-settings", user.UpdateChatSettings)
|
||
|
||
// 功能管理（提问 / 私聊 / 投票 的开关、定时开关与定量收集）：
|
||
// 不单独开页面，作为账号设置页的三个面板，保存后回跳到对应面板。
|
||
f.Group("/features", func() {
|
||
f.Post("/update", rateLimitMiddleware(20, 1*time.Minute), user.FeaturesUpdate)
|
||
f.Post("/reset-quota", rateLimitMiddleware(20, 1*time.Minute), user.FeatureResetQuota)
|
||
})
|
||
})
|
||
// 防骚扰与全局屏蔽词拆为两个独立入口，避免同表单一次写两列时相互清空。
|
||
// 保存后经 ?tab= 参数回跳到来源面板。
|
||
f.Post("/harassment/update", form.Bind(form.UpdateHarassment{}), user.UpdateHarassment)
|
||
f.Post("/blockwords/update", form.Bind(form.UpdateBlockWords{}), user.UpdateBlockWords)
|
||
f.Post("/moderation-strictness/update", form.Bind(form.UpdateModerationStrictness{}), user.UpdateModerationStrictness)
|
||
f.Post("/tags/update", form.Bind(form.UpdateTags{}), user.UpdateTags)
|
||
// Logout 必须用 POST：GET 会留下 CSRF 风险（第三方图片/链接可触发退出登录）
|
||
f.Post("/logout", auth.Logout)
|
||
}, reqUserSignIn)
|
||
|
||
f.Group("/api/v1", func() {
|
||
f.Group("/user", func() {
|
||
f.Get("", reqUserSignIn, user.ProfileAPI)
|
||
f.Group("/{domain}", func() {
|
||
f.Group("/questions", func() {
|
||
f.Get("", rateLimitMiddleware(60, 1*time.Minute), question.ListAPI)
|
||
})
|
||
})
|
||
})
|
||
// 像素追踪代理：仅暴露只读 GET（f.Any 会放行写方法，曾有 CSRF 豁免 + 写方法放行的风险），
|
||
// 并加限流防止被用作高频打点通道。
|
||
f.Get("/pixel/{**}", reqUserSignIn, rateLimitMiddleware(30, 1*time.Minute), pixel.Proxy)
|
||
// 服务反代仅暴露只读 GET：本网关是把已登录用户的请求转发到配置好的内部服务，
|
||
// 不应允许任意写方法，避免被用作对内网后端的越权操作入口。
|
||
f.Get("/service/{**}", reqUserSignIn, service.Proxy)
|
||
f.Get("/explore/random", rateLimitMiddleware(60, 1*time.Minute), user.RandomRecommend)
|
||
// 首页"提问箱信息统计"公开接口（服务端 30 分钟缓存），供前端定时刷新
|
||
f.Get("/site-stats", rateLimitMiddleware(60, 1*time.Minute), route.SiteStatsAPI)
|
||
}, context.APIEndpoint)
|
||
},
|
||
cache.Cacher(cache.Options{
|
||
Initer: cacheRedis.Initer(),
|
||
Config: cacheRedis.Config{
|
||
Options: &cacheRedis.Options{
|
||
Addr: conf.Redis.Addr,
|
||
Password: conf.Redis.Password,
|
||
DB: 0,
|
||
},
|
||
},
|
||
}),
|
||
recaptcha.V3(
|
||
recaptcha.Options{
|
||
Secret: conf.Recaptcha.ServerKey,
|
||
VerifyURL: func() recaptcha.VerifyURL {
|
||
if conf.Recaptcha.TurnstileStyle {
|
||
return "https://challenges.cloudflare.com/turnstile/v0/siteverify"
|
||
}
|
||
return recaptcha.VerifyURLGlobal
|
||
}(),
|
||
},
|
||
),
|
||
sessioner,
|
||
csrf.Csrfer(csrf.Options{
|
||
Secret: conf.Server.XSRFKey,
|
||
Header: "X-CSRF-Token",
|
||
// ⚠️ 必须与 login.go 的 ctx.Session.Set("uid", ...) 一致。flamego/csrf 默认
|
||
// SessionKey 为 "userID"，不显式指定则 s.Get("userID") 恒为 nil、token 内 id
|
||
// 恒为 "0"；校验是无状态 HMAC 重算（不与 session 内 token 比对），匿名访客从
|
||
// 公开页取到的 token 会对所有已登录用户有效，CSRF 退化为全站公开常量。
|
||
// 回归测试见 route/csrf_binding_test.go。
|
||
SessionKey: CSRFUserSessionKey,
|
||
}),
|
||
template.Templater(template.Options{
|
||
FileSystem: templateFS,
|
||
FuncMaps: templatepkg.FuncMap(),
|
||
}),
|
||
context.Contexter(),
|
||
)
|
||
f.NotFound(func(ctx flamego.Context) {
|
||
// 真正的 404 探测在此记录（不把业务 302→/ 误判为扫描）。flamego 的 NotFound 会先
|
||
// 写 404 状态再 Redirect，故须在 Redirect 前记录。
|
||
// 键必须与 autobanMiddleware 的 IsBanned/Record 一致（同样经 LimitKey 聚合），
|
||
// 否则计数落在另一个键上，封禁永不生效。
|
||
// ⚠️ 计数受 ban_track_notfound 控制，默认关闭；Record 内部会忽略未开启的
|
||
// KindNotFound。
|
||
if ip := shared.LimitKey(shared.ResolveClientIP(ctx.Request().Request)); ip != "" {
|
||
ab.Record(ip, ctx.Request().URL.Path, autoban.KindNotFound)
|
||
}
|
||
ctx.Redirect("/")
|
||
})
|
||
|
||
return f
|
||
}
|