wangjia
6e99e32285
实现「一个二进制三个监听」中的管理端：
- 独立监听 ADMIN_LISTEN（默认 127.0.0.1:9443，FromEnv 拒绝 0.0.0.0/:: 等公网绑定）
- 中间件链：mw_ipallow（CIDR 白名单，仅信任 RemoteAddr，默认内网段）→
mw_session（HttpOnly+Secure+SameSite=Strict cookie，Redis 30min 滑动 TTL）
- 登录双因素：argon2id 密码 + RFC6238 TOTP；失败限流与临时锁定；
TOTP 密钥 AES-GCM 加密入库；migration 000008 增补 admins 表；
首个管理员由 cmd/adminctl create 创建（终端输出 otpauth URI/Secret）
- 功能三块（html/template + embed 静态资源，原生 JS 二次确认）：
1. 码批次：表单生成→明文仅在本次 CSV 下载出现（不落盘/不入日志）；
批次列表 + 整批作废（复用 #3 codes，新增 Store.ListBatches/VoidBatch）
2. 节点操作：列表 + 近期 node_events；replace 经 #14 ProvisionService、
draining/up 经 #5 Lifecycle（二者未就绪→注入 stub，UI 置灰）；二次确认 + CSRF
3. audit_log 查看：actor/action/target/时间范围过滤分页；同页查 node_events
- 所有写操作写 audit_log（actor=用户名，meta 不含明文）；管理端仅记安全事件
（登录失败/锁定/白名单拦截），不记常规访问日志
- internal/totp 复用包（与 doc/05 用户中心 2FA 同算法，纯标准库）
测试：totp RFC6238 向量、argon2/AES 往返、白名单放行/拦截、会话滑动过期、
登录成功/密码错/TOTP 错/未知用户/锁定、批次 CSV 含明文且 audit 不泄露、
作废/节点操作的二次确认与 CSRF 缺失被拒、mock 断言 service 调用参数、审计过滤。
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
85 lines
1.9 KiB
Go
85 lines
1.9 KiB
Go
package admin
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/alicebob/miniredis/v2"
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
func newTestRedis(t *testing.T) (*redis.Client, *miniredis.Miniredis) {
|
|
t.Helper()
|
|
mr := miniredis.RunT(t)
|
|
rdb := redis.NewClient(&redis.Options{Addr: mr.Addr()})
|
|
t.Cleanup(func() { rdb.Close() })
|
|
return rdb, mr
|
|
}
|
|
|
|
func TestSessionStore_Lifecycle(t *testing.T) {
|
|
rdb, _ := newTestRedis(t)
|
|
store := NewSessionStore(rdb, 30*time.Minute)
|
|
ctx := context.Background()
|
|
|
|
sid, sess, err := store.Create(ctx, 42, "alice")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if sess.AdminID != 42 || sess.Username != "alice" || sess.CSRFToken == "" {
|
|
t.Fatalf("unexpected session %+v", sess)
|
|
}
|
|
|
|
got, err := store.Get(ctx, sid)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got == nil || got.AdminID != 42 {
|
|
t.Fatalf("Get returned %+v", got)
|
|
}
|
|
|
|
// CSRF check.
|
|
if !got.ValidCSRF(sess.CSRFToken) {
|
|
t.Error("valid CSRF token rejected")
|
|
}
|
|
if got.ValidCSRF("bogus") {
|
|
t.Error("bogus CSRF token accepted")
|
|
}
|
|
|
|
// Delete.
|
|
if err := store.Delete(ctx, sid); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
got, _ = store.Get(ctx, sid)
|
|
if got != nil {
|
|
t.Error("session still present after delete")
|
|
}
|
|
}
|
|
|
|
func TestSessionStore_SlidingExpiry(t *testing.T) {
|
|
rdb, mr := newTestRedis(t)
|
|
store := NewSessionStore(rdb, 30*time.Minute)
|
|
ctx := context.Background()
|
|
|
|
sid, _, err := store.Create(ctx, 1, "bob")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Advance 20 min, access (slides TTL forward).
|
|
mr.FastForward(20 * time.Minute)
|
|
if got, _ := store.Get(ctx, sid); got == nil {
|
|
t.Fatal("session expired prematurely")
|
|
}
|
|
// Another 20 min: still alive because the previous Get slid the TTL.
|
|
mr.FastForward(20 * time.Minute)
|
|
if got, _ := store.Get(ctx, sid); got == nil {
|
|
t.Fatal("sliding TTL did not extend session")
|
|
}
|
|
// Idle past the full TTL: gone.
|
|
mr.FastForward(31 * time.Minute)
|
|
if got, _ := store.Get(ctx, sid); got != nil {
|
|
t.Error("session survived past idle timeout")
|
|
}
|
|
}
|