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>
95 lines
2.2 KiB
Go
95 lines
2.2 KiB
Go
package totp
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// rfc6238Secret is the Base32 encoding of the ASCII seed "12345678901234567890"
|
|
// from RFC 6238 Appendix B (the SHA-1 test vector).
|
|
const rfc6238Secret = "GEZDGNBVGY3TQOJQGEZDGNBVGY3TQOJQ"
|
|
|
|
func TestCode_RFC6238Vectors(t *testing.T) {
|
|
// 8-digit reference values from RFC 6238 truncated to our 6 digits.
|
|
cases := []struct {
|
|
unix int64
|
|
want string
|
|
}{
|
|
{59, "287082"},
|
|
{1111111109, "081804"},
|
|
{1111111111, "050471"},
|
|
{1234567890, "005924"},
|
|
{2000000000, "279037"},
|
|
{20000000000, "353130"},
|
|
}
|
|
for _, c := range cases {
|
|
got, err := Code(rfc6238Secret, time.Unix(c.unix, 0).UTC())
|
|
if err != nil {
|
|
t.Fatalf("Code(%d): %v", c.unix, err)
|
|
}
|
|
if got != c.want {
|
|
t.Errorf("Code(%d) = %s; want %s", c.unix, got, c.want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestValidate_SkewWindow(t *testing.T) {
|
|
secret, err := GenerateSecret()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
now := time.Now().UTC()
|
|
code, err := Code(secret, now)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !Validate(secret, code, now, 1) {
|
|
t.Error("current code rejected")
|
|
}
|
|
// Previous window must be accepted with skew=1.
|
|
if !Validate(secret, code, now.Add(Period), 1) {
|
|
t.Error("code from previous step rejected with skew=1")
|
|
}
|
|
// Two steps away must be rejected.
|
|
if Validate(secret, code, now.Add(2*Period+time.Second), 1) {
|
|
t.Error("stale code accepted outside skew window")
|
|
}
|
|
// Wrong code rejected.
|
|
if Validate(secret, "000000", now, 1) && code != "000000" {
|
|
t.Error("validate accepted obviously wrong code")
|
|
}
|
|
}
|
|
|
|
func TestValidate_BadInput(t *testing.T) {
|
|
secret, _ := GenerateSecret()
|
|
now := time.Now().UTC()
|
|
if Validate(secret, "12345", now, 1) { // too short
|
|
t.Error("accepted 5-digit code")
|
|
}
|
|
if Validate("not-base32!!", "123456", now, 1) {
|
|
t.Error("accepted invalid secret")
|
|
}
|
|
}
|
|
|
|
func TestProvisioningURI(t *testing.T) {
|
|
uri := ProvisioningURI(rfc6238Secret, "admin", "Pangolin")
|
|
if uri == "" {
|
|
t.Fatal("empty URI")
|
|
}
|
|
for _, sub := range []string{"otpauth://totp/", "secret=" + rfc6238Secret, "issuer=Pangolin"} {
|
|
if !contains(uri, sub) {
|
|
t.Errorf("URI %q missing %q", uri, sub)
|
|
}
|
|
}
|
|
}
|
|
|
|
func contains(s, sub string) bool {
|
|
for i := 0; i+len(sub) <= len(s); i++ {
|
|
if s[i:i+len(sub)] == sub {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|