Files
pangolin/server/internal/admin/config_test.go
T
wangjia 6e99e32285 feat(admin): 管理端最小后台(独立监听 + 白名单 + 2FA)tsk_SCMtcGF4F434
实现「一个二进制三个监听」中的管理端:

- 独立监听 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>
2026-06-13 14:49:32 +08:00

96 lines
2.3 KiB
Go

package admin
import (
"encoding/hex"
"net"
"strings"
"testing"
)
func TestValidateListen_RejectsPublic(t *testing.T) {
bad := []string{"0.0.0.0:9443", ":9443", "[::]:9443", "::"}
for _, addr := range bad {
if err := validateListen(addr); err == nil {
t.Errorf("validateListen(%q) = nil; want error (public bind)", addr)
}
}
good := []string{"127.0.0.1:9443", "10.0.0.5:9443", "192.168.1.2:9090", "internal.host:9443"}
for _, addr := range good {
if err := validateListen(addr); err != nil {
t.Errorf("validateListen(%q) = %v; want nil", addr, err)
}
}
}
func TestParseSecretKey(t *testing.T) {
key := make([]byte, 32)
for i := range key {
key[i] = byte(i)
}
if _, err := parseSecretKey(hex.EncodeToString(key)); err != nil {
t.Errorf("hex key rejected: %v", err)
}
if _, err := parseSecretKey("short"); err == nil {
t.Error("short key accepted")
}
if _, err := parseSecretKey(""); err == nil {
t.Error("empty key accepted")
}
}
func TestParseCIDRs_Defaults(t *testing.T) {
nets, err := ParseCIDRs(defaultInternalCIDRs)
if err != nil {
t.Fatal(err)
}
// Loopback and an RFC1918 address must be inside; a public one must not.
inside := []string{"127.0.0.1", "10.1.2.3", "192.168.0.9", "172.16.5.5"}
for _, ip := range inside {
if !anyContains(nets, ip) {
t.Errorf("%s not covered by default allowlist", ip)
}
}
if anyContains(nets, "8.8.8.8") {
t.Error("public IP 8.8.8.8 unexpectedly allowed")
}
}
func TestParseCIDRs_BareIP(t *testing.T) {
nets, err := ParseCIDRs([]string{"203.0.113.7"})
if err != nil {
t.Fatal(err)
}
if !anyContains(nets, "203.0.113.7") {
t.Error("bare IP host route not matched")
}
if anyContains(nets, "203.0.113.8") {
t.Error("bare IP matched neighbour")
}
}
func TestParseCIDRs_Invalid(t *testing.T) {
if _, err := ParseCIDRs([]string{"not-a-cidr"}); err == nil {
t.Error("invalid CIDR accepted")
}
if _, err := ParseCIDRs(nil); err == nil {
t.Error("empty allowlist accepted")
}
}
func anyContains(nets []*net.IPNet, ip string) bool {
parsed := net.ParseIP(ip)
for _, n := range nets {
if n.Contains(parsed) {
return true
}
}
return false
}
func TestSplitTrim(t *testing.T) {
got := splitTrim(" a , b ,, c ")
if strings.Join(got, ",") != "a,b,c" {
t.Errorf("splitTrim = %v", got)
}
}