Sakurasan
ad10bdc400
链路功能已经跑通，缺的是边界。这几处默认值都是仓库里公开已知的，忘配就等于没有认证。
- 认证端点限速：/auth/login、/auth/register 此前完全没接限速器（Limiter 只挂在代理
Auth 上），爆破与批量建号零成本。新增 AllowWindow 滑动窗口，登录按「单 IP」和
「单 IP×账号」两个桶（默认 30/min、10/min），注册按 IP（5/min）且在 argon2 之前拦，
避免光靠建号请求打满 CPU；超限返回 429 + Retry-After。
- 生产启动断言：env=production 时拒绝默认 OT_JWT_SECRET、缺失或短于 16 字节的
OT_MASTER_KEY（缺失时 NewEncryptor 会用 sha256("") 派生，渠道密文可被公开还原）、
默认管理员密码；一次列全所有问题，不再改一个试一个。development 不拦。
- 请求体上限：parseBody 原来全量 io.ReadAll 且全项目没有一处 MaxBytesReader，
合法密钥可把任意大 body 整份压进内存再转发。/v1 组挂 BodyLimit（默认 32MB，
OT_PROXY_MAX_BODY_MB），超限按客户端协议返回 413 request_too_large 并点名上限。
- CORS：不再无条件反射 Origin + Allow-Credentials: true——那唯一挡住跨站凭据读取的
是 refresh cookie 的 SameSite=Lax，改成 None 就立刻变成窃取面。改为显式白名单，
空白名单完全不发跨域头（同源部署与 vite proxy 本来就不需要），通配 * 不配凭据；
允许头补上 X-Api-Key / Anthropic-Version / OpenAI-*，否则浏览器里跑官方 SDK 预检会挂。
- 配套 OT_SERVER_TRUSTED_PROXIES：不配置就不信 X-Forwarded-For（防伪造头绕过 IP 限速）。
顺手删掉 .env.example 末尾重复的 OT_PROXY_LOG_RAW 段。
40 lines
934 B
Go
40 lines
934 B
Go
package ratelimit
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestAllowWindow(t *testing.T) {
|
|
l := New()
|
|
const win = 80 * time.Millisecond
|
|
|
|
// 窗口内到量就拒
|
|
for i := 0; i < 3; i++ {
|
|
if !l.AllowWindow("login:ip:1.2.3.4", 3, win) {
|
|
t.Fatalf("第 %d 次应放行", i+1)
|
|
}
|
|
}
|
|
if l.AllowWindow("login:ip:1.2.3.4", 3, win) {
|
|
t.Fatal("超过限额应拒绝")
|
|
}
|
|
// 不同键互不影响（换 IP 不会被同一个桶牵连）
|
|
if !l.AllowWindow("login:ip:5.6.7.8", 3, win) {
|
|
t.Fatal("另一个 IP 应独立计数")
|
|
}
|
|
if !l.AllowWindow("login:u:1.2.3.4|admin", 3, win) {
|
|
t.Fatal("同一 IP 的另一个账号桶应独立")
|
|
}
|
|
// 窗口滑过即恢复
|
|
time.Sleep(120 * time.Millisecond)
|
|
if !l.AllowWindow("login:ip:1.2.3.4", 3, win) {
|
|
t.Fatal("窗口过期后应重新放行")
|
|
}
|
|
// limit<=0 不限制
|
|
for i := 0; i < 5; i++ {
|
|
if !l.AllowWindow("free", 0, win) {
|
|
t.Fatal("limit<=0 应始终放行")
|
|
}
|
|
}
|
|
}
|