feat: 配对管理端点与中继侧设备注册上报
- agent --pair:临时监听 :3001 仅开放 /provision 与 /destroy,以节点密码为共享秘密; 窗口由三个独立条件关闭(首次成功配对 / 失败尝试 10 次 / 10 分钟超时),配对成功后 管理监听即关(设计 §7 第 4 步) - agent 为每台配对手机生成中继 bearer(config.Save 0600 原子写;启动时为手写条目补发) - transport.PushDevices:握手成功后全量上报本节点受信设备集;空集是吊销声明而非无操作 - relay /v1/agent/devices:agent 会话后鉴权的设备注册;全量替换本节点条目; 操作员 MSA_DEVICES 为 static 不可被覆盖;跨节点 id 不可抢占 - 中继设备注册替代了 §11 第 6 项的一部分:节点 destroy + 重握手即可在路由层吊销 - 配置:provisionSecret 字段(无密码保护节点需显式设置才可配对);Validate(pairMode) - 测试:pair 流程 8 项、设备注册 6 项、transport 1 项;文档 §7 实现修订注记与 README 状态
This commit is contained in:
@@ -0,0 +1,282 @@
|
||||
package pair
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"crypto/ed25519"
|
||||
"crypto/rand"
|
||||
"encoding/base64"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"fun.chick/msa-agent/internal/config"
|
||||
)
|
||||
|
||||
func agentConfig(t *testing.T) (string, config.File, ed25519.PrivateKey) {
|
||||
t.Helper()
|
||||
_, sign, err := ed25519.GenerateKey(rand.Reader)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
kx := make([]byte, 32)
|
||||
rand.Read(kx)
|
||||
cfg := config.File{
|
||||
NodeID: "nd_7f3a9c2e",
|
||||
RelayURL: "https://relay.example",
|
||||
AgentToken: "agent-token",
|
||||
LocalPort: 8788,
|
||||
SignKey: base64.StdEncoding.EncodeToString(sign),
|
||||
KxKey: base64.StdEncoding.EncodeToString(kx),
|
||||
AuditPath: filepath.Join(t.TempDir(), "audit.log"),
|
||||
NodePassword: "hunter2",
|
||||
}
|
||||
path := filepath.Join(t.TempDir(), "config.json")
|
||||
if err := config.Save(path, cfg); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
return path, cfg, sign
|
||||
}
|
||||
|
||||
func phoneKeys(t *testing.T) (string, string) {
|
||||
t.Helper()
|
||||
pub, _, err := ed25519.GenerateKey(rand.Reader)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
kx := make([]byte, 32)
|
||||
rand.Read(kx)
|
||||
return base64.StdEncoding.EncodeToString(pub), base64.StdEncoding.EncodeToString(kx)
|
||||
}
|
||||
|
||||
func provisionBody(t *testing.T, password, signPub, kxPub string) *bytes.Buffer {
|
||||
t.Helper()
|
||||
body, err := json.Marshal(map[string]string{
|
||||
"password": password, "appSignPub": signPub, "appKxPub": kxPub, "deviceName": "chick",
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
return bytes.NewBuffer(body)
|
||||
}
|
||||
|
||||
func openHandler(t *testing.T, path string, cfg config.File) *Handler {
|
||||
t.Helper()
|
||||
h, err := New(path, cfg, nil)
|
||||
if err != nil {
|
||||
t.Fatalf("New: %v", err)
|
||||
}
|
||||
return h
|
||||
}
|
||||
|
||||
func TestProvisionPairsThePhoneAndClosesTheWindow(t *testing.T) {
|
||||
path, cfg, sign := agentConfig(t)
|
||||
h := openHandler(t, path, cfg)
|
||||
signPub, kxPub := phoneKeys(t)
|
||||
|
||||
rec := httptest.NewRecorder()
|
||||
h.Routes().ServeHTTP(rec, httptest.NewRequest("POST", "/provision",
|
||||
provisionBody(t, "hunter2", signPub, kxPub)))
|
||||
if rec.Code != 200 {
|
||||
t.Fatalf("provision status %d: %s", rec.Code, rec.Body)
|
||||
}
|
||||
var out provisionResponse
|
||||
if err := json.Unmarshal(rec.Body.Bytes(), &out); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if out.NodeID != "nd_7f3a9c2e" || !strings.HasPrefix(out.AppID, "ap_") || out.DeviceToken == "" {
|
||||
t.Fatalf("response: %+v", out)
|
||||
}
|
||||
if len(out.AgentSignPub) != base64.StdEncoding.EncodedLen(ed25519.PublicKeySize) {
|
||||
t.Fatalf("agent signing key not published: %q", out.AgentSignPub)
|
||||
}
|
||||
// The published key must verify what the agent's private key signs: the pairing response is
|
||||
// the phone's only source for the trust anchor.
|
||||
proof := ed25519.Sign(sign, []byte("x"))
|
||||
gotPub, err := base64.StdEncoding.DecodeString(out.AgentSignPub)
|
||||
if err != nil || !ed25519.Verify(gotPub, []byte("x"), proof) {
|
||||
t.Fatal("published signing key does not match the private key")
|
||||
}
|
||||
// The config on disk carries the device, with a relay token generated by the agent.
|
||||
reloaded, err := config.Load(path)
|
||||
if err != nil {
|
||||
t.Fatalf("config after pairing must be loadable: %v", err)
|
||||
}
|
||||
if len(reloaded.Devices) != 1 || reloaded.Devices[0].ID != out.AppID ||
|
||||
reloaded.Devices[0].SignKey != signPub || reloaded.Devices[0].RelayToken != out.DeviceToken {
|
||||
t.Fatalf("persisted devices: %+v", reloaded.Devices)
|
||||
}
|
||||
// The window closed with the pairing: a second phone is refused.
|
||||
rec = httptest.NewRecorder()
|
||||
h.Routes().ServeHTTP(rec, httptest.NewRequest("POST", "/provision",
|
||||
provisionBody(t, "hunter2", signPub, kxPub)))
|
||||
if rec.Code != http.StatusGone {
|
||||
t.Fatalf("second pairing got %d, want 410", rec.Code)
|
||||
}
|
||||
select {
|
||||
case <-h.Done:
|
||||
default:
|
||||
t.Fatal("Done was not closed after a completed pairing")
|
||||
}
|
||||
}
|
||||
|
||||
func TestAWrongSecretIsRefusedAndNothingIsPersisted(t *testing.T) {
|
||||
path, cfg, _ := agentConfig(t)
|
||||
h := openHandler(t, path, cfg)
|
||||
signPub, kxPub := phoneKeys(t)
|
||||
rec := httptest.NewRecorder()
|
||||
h.Routes().ServeHTTP(rec, httptest.NewRequest("POST", "/provision",
|
||||
provisionBody(t, "wrong", signPub, kxPub)))
|
||||
if rec.Code != http.StatusUnauthorized {
|
||||
t.Fatalf("status %d", rec.Code)
|
||||
}
|
||||
// LoadForPairing, not Load: this config legitimately has no devices yet, which is exactly the
|
||||
// state that must survive a refused attempt.
|
||||
reloaded, err := config.LoadForPairing(path)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(reloaded.Devices) != 0 {
|
||||
t.Fatal("a refused pairing was still persisted")
|
||||
}
|
||||
}
|
||||
|
||||
// The pairing secret is a human password and the window is LAN-reachable, so the attempt ceiling is
|
||||
// the difference between a management port and a password oracle.
|
||||
func TestTheAttemptCeilingShutsTheWindow(t *testing.T) {
|
||||
path, cfg, _ := agentConfig(t)
|
||||
h := openHandler(t, path, cfg)
|
||||
signPub, kxPub := phoneKeys(t)
|
||||
for i := 0; i < MaxAttempts; i++ {
|
||||
rec := httptest.NewRecorder()
|
||||
h.Routes().ServeHTTP(rec, httptest.NewRequest("POST", "/provision",
|
||||
provisionBody(t, "guess", signPub, kxPub)))
|
||||
if rec.Code != http.StatusUnauthorized {
|
||||
t.Fatalf("attempt %d: status %d", i, rec.Code)
|
||||
}
|
||||
}
|
||||
rec := httptest.NewRecorder()
|
||||
h.Routes().ServeHTTP(rec, httptest.NewRequest("POST", "/provision",
|
||||
provisionBody(t, "hunter2", signPub, kxPub)))
|
||||
if rec.Code != http.StatusGone {
|
||||
t.Fatalf("after the ceiling, a correct password got %d, want 410", rec.Code)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBadKeyMaterialIsRefused(t *testing.T) {
|
||||
path, cfg, _ := agentConfig(t)
|
||||
h := openHandler(t, path, cfg)
|
||||
_, kxPub := phoneKeys(t)
|
||||
for _, body := range []string{
|
||||
`{"password":"hunter2","appSignPub":"not base64!!","appKxPub":"` + kxPub + `"}`,
|
||||
// 31 bytes decodes fine but is the wrong size; the strict length check is the point.
|
||||
`{"password":"hunter2","appSignPub":"` + base64.StdEncoding.EncodeToString(make([]byte, 31)) + `","appKxPub":"` + kxPub + `"}`,
|
||||
`{"password":"hunter2","appSignPub":"` + strings.Repeat("A", 44) + `","appKxPub":"` + base64.StdEncoding.EncodeToString(make([]byte, 1)) + `"}`,
|
||||
} {
|
||||
rec := httptest.NewRecorder()
|
||||
h.Routes().ServeHTTP(rec, httptest.NewRequest("POST", "/provision", strings.NewReader(body)))
|
||||
if rec.Code != http.StatusBadRequest {
|
||||
t.Fatalf("accepted key material %q with %d", body, rec.Code)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestDestroyClearsDevicesAndCloses(t *testing.T) {
|
||||
path, cfg, _ := agentConfig(t)
|
||||
h := openHandler(t, path, cfg)
|
||||
signPub, kxPub := phoneKeys(t)
|
||||
rec := httptest.NewRecorder()
|
||||
h.Routes().ServeHTTP(rec, httptest.NewRequest("POST", "/provision",
|
||||
provisionBody(t, "hunter2", signPub, kxPub)))
|
||||
if rec.Code != 200 {
|
||||
t.Fatal(rec.Body.String())
|
||||
}
|
||||
// Destroy after close (pairing closed it) is Gone: the one-window-one-action rule.
|
||||
rec = httptest.NewRecorder()
|
||||
h.Routes().ServeHTTP(rec, httptest.NewRequest("POST", "/destroy",
|
||||
bytes.NewBufferString(`{"password":"hunter2"}`)))
|
||||
if rec.Code != http.StatusGone {
|
||||
t.Fatalf("destroy after pairing got %d, want 410", rec.Code)
|
||||
}
|
||||
|
||||
// A fresh window can destroy an existing device list.
|
||||
h2 := openHandler(t, path, cfg)
|
||||
cfg2 := config.File{}
|
||||
raw, _ := os.ReadFile(path)
|
||||
json.Unmarshal(raw, &cfg2)
|
||||
h2 = openHandler(t, path, cfg2)
|
||||
rec = httptest.NewRecorder()
|
||||
h2.Routes().ServeHTTP(rec, httptest.NewRequest("POST", "/destroy",
|
||||
bytes.NewBufferString(`{"password":"hunter2"}`)))
|
||||
if rec.Code != 200 {
|
||||
t.Fatalf("destroy got %d: %s", rec.Code, rec.Body)
|
||||
}
|
||||
reloaded, err := config.LoadForPairing(path)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(reloaded.Devices) != 0 {
|
||||
t.Fatal("destroy left devices behind")
|
||||
}
|
||||
}
|
||||
|
||||
func TestAnUnprotectedNodeNeedsAnExplicitSecret(t *testing.T) {
|
||||
path, cfg, _ := agentConfig(t)
|
||||
cfg.NodePassword = ""
|
||||
if _, err := New(path, cfg, nil); err != ErrNoSecret {
|
||||
t.Fatalf("got %v, want ErrNoSecret", err)
|
||||
}
|
||||
cfg.ProvisionSecret = "pair-only"
|
||||
h, err := New(path, cfg, nil)
|
||||
if err != nil {
|
||||
t.Fatalf("explicit secret was refused: %v", err)
|
||||
}
|
||||
signPub, kxPub := phoneKeys(t)
|
||||
rec := httptest.NewRecorder()
|
||||
h.Routes().ServeHTTP(rec, httptest.NewRequest("POST", "/provision",
|
||||
provisionBody(t, "pair-only", signPub, kxPub)))
|
||||
if rec.Code != 200 {
|
||||
t.Fatalf("status %d: %s", rec.Code, rec.Body)
|
||||
}
|
||||
}
|
||||
|
||||
func TestUnknownRoutesDoNotDescribeTheSurface(t *testing.T) {
|
||||
path, cfg, _ := agentConfig(t)
|
||||
h := openHandler(t, path, cfg)
|
||||
rec := httptest.NewRecorder()
|
||||
h.Routes().ServeHTTP(rec, httptest.NewRequest("GET", "/provision", nil))
|
||||
if rec.Code != http.StatusMethodNotAllowed && rec.Code != http.StatusNotFound {
|
||||
t.Fatalf("GET /provision got %d", rec.Code)
|
||||
}
|
||||
rec = httptest.NewRecorder()
|
||||
h.Routes().ServeHTTP(rec, httptest.NewRequest("POST", "/admin/list-devices", nil))
|
||||
if rec.Code != http.StatusNotFound || len(rec.Body.Bytes()) != 0 {
|
||||
t.Fatalf("unknown route leaked: %d %q", rec.Code, rec.Body.String())
|
||||
}
|
||||
}
|
||||
|
||||
// Serve enforces the wall-clock deadline even if nobody ever pairs: a management socket that
|
||||
// outlives the operator's session is the second hole the design forbids.
|
||||
func TestServeDeadlineClosesTheWindow(t *testing.T) {
|
||||
path, cfg, _ := agentConfig(t)
|
||||
h := openHandler(t, path, cfg)
|
||||
server := httptest.NewServer(h.Routes())
|
||||
defer server.Close()
|
||||
deadline := time.NewTimer(50 * time.Millisecond)
|
||||
<-deadline.C
|
||||
h.Close()
|
||||
signPub, kxPub := phoneKeys(t)
|
||||
response, err := http.Post(server.URL+"/provision", "application/json",
|
||||
provisionBody(t, "hunter2", signPub, kxPub))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer response.Body.Close()
|
||||
if response.StatusCode != http.StatusGone {
|
||||
t.Fatalf("status after the deadline: %d", response.StatusCode)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user