SKYMirror
7f060dd0e4
Go 1.27.1 (Gin+GORM) + Vue 3 文件快传服务： - 安全审计全部修复（docs/security-audit-2026-09-05.md）： bcrypt 密码哈希与自动升级、presign 直传服务端大小/内容校验、 全局请求体上限、依赖升级（govulncheck 0 命中）、janitor 后台清理、 管理端审计动作落库、/admin CORS 收紧、通知内容白名单净化、 会话默认 7 天、限流缓存故障降级、robots.txt 端点等 - 前端：取件链接复制修复（不再重复拼接提取码）、markdown 净化器加固 - Redis 支持库号（FCB_REDIS_DB / redis://…/db URL） - 文档：docs/api/* 与 openapi.yaml 同步最新行为（robots.txt、 提码 5 位起、chunk 32MiB 上限、admin 审计动作等） 验证：gofmt/go vet/go test 全绿；二进制端到端冒烟通过
67 lines
1.9 KiB
Go
67 lines
1.9 KiB
Go
package middleware
|
||
|
||
import (
|
||
"context"
|
||
"net/http/httptest"
|
||
"testing"
|
||
"time"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
|
||
"filecodebox/internal/cache"
|
||
)
|
||
|
||
// failingCache 模拟缓存故障（Get/Incr 均返回非 ErrNotFound 错误）。
|
||
type failingCache struct{ cache.Cache }
|
||
|
||
func (f *failingCache) Get(_ context.Context, _ string) (string, error) {
|
||
return "", context.DeadlineExceeded
|
||
}
|
||
func (f *failingCache) Set(_ context.Context, _, _ string, _ time.Duration) error {
|
||
return context.DeadlineExceeded
|
||
}
|
||
func (f *failingCache) Incr(_ context.Context, _ string, _ time.Duration) (int64, error) {
|
||
return 0, context.DeadlineExceeded
|
||
}
|
||
|
||
func newLimiterTest(c *gin.Context, cacheImpl cache.Cache, count int) *RateLimiter {
|
||
return NewRateLimiter(cacheImpl, map[string]LimitRule{
|
||
LimitLogin: {Count: count, Window: time.Minute},
|
||
})
|
||
}
|
||
|
||
func ginTestContext() *gin.Context {
|
||
gin.SetMode(gin.TestMode)
|
||
c, _ := gin.CreateTestContext(httptest.NewRecorder())
|
||
c.Request = httptest.NewRequest("POST", "/admin/login", nil)
|
||
return c
|
||
}
|
||
|
||
// TestRateLimiterFallbackOnCacheFailure L9：缓存故障时限流降级为进程内计数，
|
||
// 超过上限后 Check 拒绝（此前 fail-open 会一直放行）。
|
||
func TestRateLimiterFallbackOnCacheFailure(t *testing.T) {
|
||
c := ginTestContext()
|
||
rl := newLimiterTest(c, &failingCache{cache.NewMemory()}, 3)
|
||
for i := 0; i < 3; i++ {
|
||
if ok, _ := rl.Check(c, LimitLogin); !ok {
|
||
t.Fatalf("第 %d 次检查不应拒绝", i+1)
|
||
}
|
||
rl.Add(c, LimitLogin)
|
||
}
|
||
if ok, _ := rl.Check(c, LimitLogin); ok {
|
||
t.Fatal("缓存故障降级下，超过上限后 Check 应拒绝（fail-close）")
|
||
}
|
||
}
|
||
|
||
// TestRateLimiterNormalCacheCounting 正常缓存路径行为不变。
|
||
func TestRateLimiterNormalCacheCounting(t *testing.T) {
|
||
c := ginTestContext()
|
||
rl := newLimiterTest(c, cache.NewMemory(), 2)
|
||
for i := 0; i < 2; i++ {
|
||
rl.Add(c, LimitLogin)
|
||
}
|
||
if ok, _ := rl.Check(c, LimitLogin); ok {
|
||
t.Fatal("达到上限后 Check 应拒绝")
|
||
}
|
||
}
|