Files
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

58 lines
1.5 KiB
Go

package admin
import (
"context"
"net/http"
)
type ctxKey string
const ctxKeySession ctxKey = "admin_session"
// SessionMiddleware loads and validates the session cookie for protected
// routes. Requests without a valid session are redirected to the login page
// (GET) or rejected with 401 (other methods). On success the session is stored
// in the request context and its idle TTL is slid forward.
type SessionMiddleware struct {
sessions *SessionStore
next http.Handler
}
// NewSessionMiddleware wraps next so it only runs with a valid session.
func NewSessionMiddleware(sessions *SessionStore, next http.Handler) *SessionMiddleware {
return &SessionMiddleware{sessions: sessions, next: next}
}
func (m *SessionMiddleware) ServeHTTP(w http.ResponseWriter, r *http.Request) {
c, err := r.Cookie(SessionCookieName)
if err != nil {
m.deny(w, r)
return
}
sess, serr := m.sessions.Get(r.Context(), c.Value)
if serr != nil {
http.Error(w, "internal error", http.StatusInternalServerError)
return
}
if sess == nil {
m.deny(w, r)
return
}
ctx := context.WithValue(r.Context(), ctxKeySession, sess)
m.next.ServeHTTP(w, r.WithContext(ctx))
}
func (m *SessionMiddleware) deny(w http.ResponseWriter, r *http.Request) {
if r.Method == http.MethodGet {
http.Redirect(w, r, "/login", http.StatusFound)
return
}
http.Error(w, "unauthorized", http.StatusUnauthorized)
}
// SessionFromContext returns the authenticated session, or nil.
func SessionFromContext(ctx context.Context) *Session {
s, _ := ctx.Value(ctxKeySession).(*Session)
return s
}