refactor(routing): 暴露 system_locked_domains + 服务端 follow-up 清理(FK/Validate/dedup/exclude单源/共用Store/Builtin保留)
This commit is contained in:
@@ -244,10 +244,18 @@ components:
|
|||||||
description: 国内域名/IP 直连
|
description: 国内域名/IP 直连
|
||||||
lan_direct:
|
lan_direct:
|
||||||
type: boolean
|
type: boolean
|
||||||
description: 局域网直连
|
readOnly: true
|
||||||
|
description: >-
|
||||||
|
reserved/always-on:局域网直连由系统层恒渲染(route_exclude_address +
|
||||||
|
固定 LAN 直连规则),渲染器不读此开关,当前始终生效;字段保留供未来扩展
|
||||||
|
per-user 关闭,目前无实际效果。
|
||||||
private_via_tunnel:
|
private_via_tunnel:
|
||||||
type: boolean
|
type: boolean
|
||||||
description: 私有服务域名强制走隧道(家庭内网穿透场景)
|
readOnly: true
|
||||||
|
description: >-
|
||||||
|
reserved/always-on:配置了 PANGOLIN_PRIVATE_SPLIT_DOMAINS 时,这些私有
|
||||||
|
服务域名恒强制走隧道,渲染器不读此开关,当前始终生效;字段保留供未来扩展
|
||||||
|
per-user 关闭,目前无实际效果。
|
||||||
|
|
||||||
RoutingProfile:
|
RoutingProfile:
|
||||||
type: object
|
type: object
|
||||||
@@ -267,6 +275,24 @@ components:
|
|||||||
type: string
|
type: string
|
||||||
enum: [proxy, direct]
|
enum: [proxy, direct]
|
||||||
|
|
||||||
|
RoutingProfileResponse:
|
||||||
|
description: GET /me/routing 响应体:档案字段 + 只读的系统强制域名清单
|
||||||
|
allOf:
|
||||||
|
- $ref: '#/components/schemas/RoutingProfile'
|
||||||
|
- type: object
|
||||||
|
required: [system_locked_domains]
|
||||||
|
properties:
|
||||||
|
system_locked_domains:
|
||||||
|
type: array
|
||||||
|
readOnly: true
|
||||||
|
items:
|
||||||
|
type: string
|
||||||
|
description: >-
|
||||||
|
系统强制走隧道的私有服务域名(PANGOLIN_PRIVATE_SPLIT_DOMAINS)。
|
||||||
|
用户对这些域名的自定义规则会静默失效;仅供客户端提示用户,不可写入
|
||||||
|
(POST 请求体中出现该字段会被忽略,不会持久化)。
|
||||||
|
example: [nas.example.com]
|
||||||
|
|
||||||
RoutingValidationError:
|
RoutingValidationError:
|
||||||
type: object
|
type: object
|
||||||
required: [code, message_zh, message_en, errors]
|
required: [code, message_zh, message_en, errors]
|
||||||
@@ -500,11 +526,11 @@ paths:
|
|||||||
summary: 获取当前用户的分流配置(可配置分流;未自定义时返回默认档案)
|
summary: 获取当前用户的分流配置(可配置分流;未自定义时返回默认档案)
|
||||||
responses:
|
responses:
|
||||||
'200':
|
'200':
|
||||||
description: 分流档案
|
description: 分流档案(含只读 system_locked_domains)
|
||||||
content:
|
content:
|
||||||
application/json:
|
application/json:
|
||||||
schema:
|
schema:
|
||||||
$ref: '#/components/schemas/RoutingProfile'
|
$ref: '#/components/schemas/RoutingProfileResponse'
|
||||||
'401':
|
'401':
|
||||||
description: 未认证
|
description: 未认证
|
||||||
content:
|
content:
|
||||||
|
|||||||
+30
-4
@@ -259,11 +259,11 @@ paths:
|
|||||||
tags: [Account]
|
tags: [Account]
|
||||||
responses:
|
responses:
|
||||||
"200":
|
"200":
|
||||||
description: 分流档案
|
description: 分流档案(含只读 system_locked_domains)
|
||||||
content:
|
content:
|
||||||
application/json:
|
application/json:
|
||||||
schema:
|
schema:
|
||||||
$ref: "#/components/schemas/RoutingProfile"
|
$ref: "#/components/schemas/RoutingProfileResponse"
|
||||||
"401":
|
"401":
|
||||||
$ref: "#/components/responses/Unauthorized"
|
$ref: "#/components/responses/Unauthorized"
|
||||||
"500":
|
"500":
|
||||||
@@ -1209,10 +1209,18 @@ components:
|
|||||||
description: 国内域名/IP 直连
|
description: 国内域名/IP 直连
|
||||||
lan_direct:
|
lan_direct:
|
||||||
type: boolean
|
type: boolean
|
||||||
description: 局域网直连
|
readOnly: true
|
||||||
|
description: >-
|
||||||
|
reserved/always-on:局域网直连由系统层恒渲染(route_exclude_address +
|
||||||
|
固定 LAN 直连规则),渲染器不读此开关,当前始终生效;字段保留供未来扩展
|
||||||
|
per-user 关闭,目前无实际效果。
|
||||||
private_via_tunnel:
|
private_via_tunnel:
|
||||||
type: boolean
|
type: boolean
|
||||||
description: 私有服务域名强制走隧道(家庭内网穿透场景)
|
readOnly: true
|
||||||
|
description: >-
|
||||||
|
reserved/always-on:配置了 PANGOLIN_PRIVATE_SPLIT_DOMAINS 时,这些私有
|
||||||
|
服务域名恒强制走隧道,渲染器不读此开关,当前始终生效;字段保留供未来扩展
|
||||||
|
per-user 关闭,目前无实际效果。
|
||||||
|
|
||||||
RoutingProfile:
|
RoutingProfile:
|
||||||
type: object
|
type: object
|
||||||
@@ -1232,6 +1240,24 @@ components:
|
|||||||
type: string
|
type: string
|
||||||
enum: [proxy, direct]
|
enum: [proxy, direct]
|
||||||
|
|
||||||
|
RoutingProfileResponse:
|
||||||
|
description: GET /me/routing 响应体:档案字段 + 只读的系统强制域名清单
|
||||||
|
allOf:
|
||||||
|
- $ref: "#/components/schemas/RoutingProfile"
|
||||||
|
- type: object
|
||||||
|
required: [system_locked_domains]
|
||||||
|
properties:
|
||||||
|
system_locked_domains:
|
||||||
|
type: array
|
||||||
|
readOnly: true
|
||||||
|
items:
|
||||||
|
type: string
|
||||||
|
description: >-
|
||||||
|
系统强制走隧道的私有服务域名(PANGOLIN_PRIVATE_SPLIT_DOMAINS)。
|
||||||
|
用户对这些域名的自定义规则会静默失效;仅供客户端提示用户,不可写入
|
||||||
|
(POST 请求体中出现该字段会被忽略,不会持久化)。
|
||||||
|
example: [nas.example.com]
|
||||||
|
|
||||||
RoutingValidationError:
|
RoutingValidationError:
|
||||||
type: object
|
type: object
|
||||||
required: [code, message_zh, message_en, errors]
|
required: [code, message_zh, message_en, errors]
|
||||||
|
|||||||
+13
-10
@@ -408,7 +408,18 @@ func mountV1(r chi.Router, sqlDB *sql.DB, rdb *redis.Client, nodeSvc *nodes.Serv
|
|||||||
accountAPI := httpapi.NewAccountAPI(sqlDB)
|
accountAPI := httpapi.NewAccountAPI(sqlDB)
|
||||||
|
|
||||||
// ── Routing profile (可配置分流) ───────────────────────────────────────────
|
// ── Routing profile (可配置分流) ───────────────────────────────────────────
|
||||||
routingAPI := httpapi.NewRoutingAPI(routing.NewStore(sqlDB))
|
// 私有服务域名分流(家庭内网穿透,如 nas/git/win.yanmeiai.com):逗号分隔;
|
||||||
|
// 这些域名用系统 DNS 解析、在外强制走隧道(排在国内分流前)。用户对这些域名的
|
||||||
|
// 自定义规则会静默失效——同一份 slice 注入 RoutingAPI,GET /v1/me/routing
|
||||||
|
// 附带 system_locked_domains 供客户端提示,以及 NodeAPI(渲染 sing-box 配置)。
|
||||||
|
var privateSplitDomains []string
|
||||||
|
for _, d := range strings.Split(os.Getenv("PANGOLIN_PRIVATE_SPLIT_DOMAINS"), ",") {
|
||||||
|
if d = strings.TrimSpace(d); d != "" {
|
||||||
|
privateSplitDomains = append(privateSplitDomains, d)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
routingStore := routing.NewStore(sqlDB)
|
||||||
|
routingAPI := httpapi.NewRoutingAPI(routingStore, privateSplitDomains)
|
||||||
|
|
||||||
// ── Nodes + Connect ───────────────────────────────────────────────────────
|
// ── Nodes + Connect ───────────────────────────────────────────────────────
|
||||||
var nodeAPI *httpapi.NodeAPI
|
var nodeAPI *httpapi.NodeAPI
|
||||||
@@ -423,15 +434,7 @@ func mountV1(r chi.Router, sqlDB *sql.DB, rdb *redis.Client, nodeSvc *nodes.Serv
|
|||||||
slog.Warn("PANGOLIN_PUBLIC_URL 未设置:国内分流(split_cn)将被静默跳过,客户端全量走隧道。" +
|
slog.Warn("PANGOLIN_PUBLIC_URL 未设置:国内分流(split_cn)将被静默跳过,客户端全量走隧道。" +
|
||||||
"如需国内直连,设为控制面对外公网基址(如 http://<公网IP>:8080)")
|
"如需国内直连,设为控制面对外公网基址(如 http://<公网IP>:8080)")
|
||||||
}
|
}
|
||||||
// 私有服务域名分流(家庭内网穿透,如 nas/git/win.yanmeiai.com):
|
nodeAPI = httpapi.NewNodeAPI(nodeStore, nodeSvc.Hub(), nodeSvc.Load(), os.Getenv("NODE_DERIVE_KEY"), publicURL, privateSplitDomains, routingStore)
|
||||||
// 逗号分隔;这些域名用系统 DNS 解析、在外强制走隧道(排在国内分流前)。
|
|
||||||
var privateSplitDomains []string
|
|
||||||
for _, d := range strings.Split(os.Getenv("PANGOLIN_PRIVATE_SPLIT_DOMAINS"), ",") {
|
|
||||||
if d = strings.TrimSpace(d); d != "" {
|
|
||||||
privateSplitDomains = append(privateSplitDomains, d)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
nodeAPI = httpapi.NewNodeAPI(nodeStore, nodeSvc.Hub(), nodeSvc.Load(), os.Getenv("NODE_DERIVE_KEY"), publicURL, privateSplitDomains, routing.NewStore(sqlDB))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// 国内分流(#5)的 rule-set 静态服务:GET /v1/rules/{name}.srs(自托管,
|
// 国内分流(#5)的 rule-set 静态服务:GET /v1/rules/{name}.srs(自托管,
|
||||||
|
|||||||
@@ -270,9 +270,15 @@ func BuildClientConfig(node *nodes.NodeRow, dpUUID, deriveKey string, opts Clien
|
|||||||
if len(extraExclude) > 0 {
|
if len(extraExclude) > 0 {
|
||||||
// IP 直连真生效:并入 TUN 入站的 route_exclude_address(auto_route 层排除,
|
// IP 直连真生效:并入 TUN 入站的 route_exclude_address(auto_route 层排除,
|
||||||
// 见 tunIn 构造处注释——单靠 route.rules 的 ip_cidr→direct 在 macOS
|
// 见 tunIn 构造处注释——单靠 route.rules 的 ip_cidr→direct 在 macOS
|
||||||
// strict_route 下不生效)。与既有 LAN 网段去重合并。
|
// strict_route 下不生效)。单源读 tunIn 现有种子(不重复硬编码 LAN 网段;
|
||||||
exclude := []string{"192.168.0.0/16", "10.0.0.0/8"}
|
// 其为 192.168+10,与上面 route.rules 的 LAN 直连 4 段有意不同——不能改成
|
||||||
seen := map[string]bool{"192.168.0.0/16": true, "10.0.0.0/8": true}
|
// LAN 全集,否则会漏排除/多排除),与 extraExclude 去重合并、写回。
|
||||||
|
base, _ := tunIn["route_exclude_address"].([]string)
|
||||||
|
exclude := append([]string{}, base...)
|
||||||
|
seen := make(map[string]bool, len(base))
|
||||||
|
for _, e := range base {
|
||||||
|
seen[e] = true
|
||||||
|
}
|
||||||
for _, e := range extraExclude {
|
for _, e := range extraExclude {
|
||||||
if !seen[e] {
|
if !seen[e] {
|
||||||
seen[e] = true
|
seen[e] = true
|
||||||
|
|||||||
@@ -64,6 +64,19 @@ func openRoutingDB(t *testing.T) *sql.DB {
|
|||||||
return db
|
return db
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// seedConnectUser inserts a minimal users row so routing_profiles inserts
|
||||||
|
// (which now declare FOREIGN KEY (user_id) REFERENCES users(id)) satisfy the
|
||||||
|
// constraint — this DB opens with _pragma=foreign_keys(1) (internal/db/db.go),
|
||||||
|
// so SQLite does enforce it, unlike a bare default SQLite connection.
|
||||||
|
func seedConnectUser(t *testing.T, db *sql.DB, id int64) {
|
||||||
|
t.Helper()
|
||||||
|
uuid := "u-connect"
|
||||||
|
if _, err := db.Exec(`INSERT INTO users (id,uuid,email,pw_hash,dp_uuid,status,created_at)
|
||||||
|
VALUES (?,?,?, 'x','dp-'||?, 'active', ?)`, id, uuid, uuid+"@x", uuid, time.Now().UTC()); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// newOnlineHub returns a real *nodes.Hub (backed by miniredis) with nodeUUID
|
// newOnlineHub returns a real *nodes.Hub (backed by miniredis) with nodeUUID
|
||||||
// registered online, so ConnectNode's a.hub.IsOnline(...) gate passes and
|
// registered online, so ConnectNode's a.hub.IsOnline(...) gate passes and
|
||||||
// Push(...) succeeds without a real Redis deployment.
|
// Push(...) succeeds without a real Redis deployment.
|
||||||
@@ -98,6 +111,7 @@ func TestConnectNode_ReadsRoutingProfile(t *testing.T) {
|
|||||||
const uid = int64(42)
|
const uid = int64(42)
|
||||||
|
|
||||||
db := openRoutingDB(t)
|
db := openRoutingDB(t)
|
||||||
|
seedConnectUser(t, db, uid)
|
||||||
rst := routing.NewStore(db)
|
rst := routing.NewStore(db)
|
||||||
p := routing.Default()
|
p := routing.Default()
|
||||||
p.Rules = []routing.Rule{
|
p.Rules = []routing.Rule{
|
||||||
@@ -167,6 +181,7 @@ func TestConnectNode_RoutingStoreErr_FailsSafe(t *testing.T) {
|
|||||||
const uid = int64(44)
|
const uid = int64(44)
|
||||||
|
|
||||||
db := openRoutingDB(t)
|
db := openRoutingDB(t)
|
||||||
|
seedConnectUser(t, db, uid)
|
||||||
// 写入一条无法反序列化的 profile_json,模拟 Get 出错。
|
// 写入一条无法反序列化的 profile_json,模拟 Get 出错。
|
||||||
if _, err := db.Exec(`INSERT INTO routing_profiles (user_id, profile_json, updated_at) VALUES (?,?,?)`,
|
if _, err := db.Exec(`INSERT INTO routing_profiles (user_id, profile_json, updated_at) VALUES (?,?,?)`,
|
||||||
uid, "{not-json", time.Now().UTC()); err != nil {
|
uid, "{not-json", time.Now().UTC()); err != nil {
|
||||||
|
|||||||
@@ -14,10 +14,28 @@ import (
|
|||||||
// user hasn't customized one yet; POST validates and upserts.
|
// user hasn't customized one yet; POST validates and upserts.
|
||||||
type RoutingAPI struct {
|
type RoutingAPI struct {
|
||||||
store *routing.Store
|
store *routing.Store
|
||||||
|
// lockedDomains are the system-forced-tunnel private-service domains
|
||||||
|
// (PANGOLIN_PRIVATE_SPLIT_DOMAINS, same slice injected into NodeAPI) —
|
||||||
|
// read-only, surfaced to GET so clients can warn users that rules against
|
||||||
|
// these domains silently have no effect. Never written to the persisted
|
||||||
|
// Profile.
|
||||||
|
lockedDomains []string
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewRoutingAPI creates a RoutingAPI backed by the given routing.Store.
|
// NewRoutingAPI creates a RoutingAPI backed by the given routing.Store.
|
||||||
func NewRoutingAPI(store *routing.Store) *RoutingAPI { return &RoutingAPI{store: store} }
|
// lockedDomains is the system-forced private-service domain list (may be nil).
|
||||||
|
func NewRoutingAPI(store *routing.Store, lockedDomains []string) *RoutingAPI {
|
||||||
|
return &RoutingAPI{store: store, lockedDomains: lockedDomains}
|
||||||
|
}
|
||||||
|
|
||||||
|
// profileResponse wraps routing.Profile for GET /v1/me/routing, adding the
|
||||||
|
// read-only system_locked_domains list. It deliberately lives here — not on
|
||||||
|
// routing.Profile itself — so the field can never leak into the writable
|
||||||
|
// Profile contract that SaveProfile decodes POST bodies into.
|
||||||
|
type profileResponse struct {
|
||||||
|
*routing.Profile
|
||||||
|
SystemLockedDomains []string `json:"system_locked_domains"`
|
||||||
|
}
|
||||||
|
|
||||||
// GetProfile handles GET /v1/me/routing.
|
// GetProfile handles GET /v1/me/routing.
|
||||||
func (a *RoutingAPI) GetProfile(w http.ResponseWriter, r *http.Request) {
|
func (a *RoutingAPI) GetProfile(w http.ResponseWriter, r *http.Request) {
|
||||||
@@ -34,7 +52,11 @@ func (a *RoutingAPI) GetProfile(w http.ResponseWriter, r *http.Request) {
|
|||||||
if p == nil {
|
if p == nil {
|
||||||
p = routing.Default()
|
p = routing.Default()
|
||||||
}
|
}
|
||||||
writeJSON(w, http.StatusOK, p)
|
locked := a.lockedDomains
|
||||||
|
if locked == nil {
|
||||||
|
locked = []string{}
|
||||||
|
}
|
||||||
|
writeJSON(w, http.StatusOK, profileResponse{Profile: p, SystemLockedDomains: locked})
|
||||||
}
|
}
|
||||||
|
|
||||||
// SaveProfile handles POST /v1/me/routing. On validation failure it returns
|
// SaveProfile handles POST /v1/me/routing. On validation failure it returns
|
||||||
|
|||||||
@@ -65,7 +65,7 @@ func doAuthReq(t *testing.T, method, target string, body *strings.Reader, uid in
|
|||||||
func TestRoutingGetDefaultThenSave(t *testing.T) {
|
func TestRoutingGetDefaultThenSave(t *testing.T) {
|
||||||
db := openRoutingTestDB(t)
|
db := openRoutingTestDB(t)
|
||||||
seedRoutingUser(t, db, 7)
|
seedRoutingUser(t, db, 7)
|
||||||
api := NewRoutingAPI(routing.NewStore(db))
|
api := NewRoutingAPI(routing.NewStore(db), nil)
|
||||||
|
|
||||||
// GET 无档案 → 200 + Default
|
// GET 无档案 → 200 + Default
|
||||||
rr := doAuthReq(t, http.MethodGet, "/v1/me/routing", nil, 7, api.GetProfile)
|
rr := doAuthReq(t, http.MethodGet, "/v1/me/routing", nil, 7, api.GetProfile)
|
||||||
@@ -120,7 +120,7 @@ func TestRoutingGetDefaultThenSave(t *testing.T) {
|
|||||||
|
|
||||||
func TestRoutingGetUnauthorized(t *testing.T) {
|
func TestRoutingGetUnauthorized(t *testing.T) {
|
||||||
db := openRoutingTestDB(t)
|
db := openRoutingTestDB(t)
|
||||||
api := NewRoutingAPI(routing.NewStore(db))
|
api := NewRoutingAPI(routing.NewStore(db), nil)
|
||||||
req := httptest.NewRequest(http.MethodGet, "/v1/me/routing", nil)
|
req := httptest.NewRequest(http.MethodGet, "/v1/me/routing", nil)
|
||||||
rr := httptest.NewRecorder()
|
rr := httptest.NewRecorder()
|
||||||
api.GetProfile(rr, req)
|
api.GetProfile(rr, req)
|
||||||
@@ -128,3 +128,82 @@ func TestRoutingGetUnauthorized(t *testing.T) {
|
|||||||
t.Fatalf("expected 401, got %d", rr.Code)
|
t.Fatalf("expected 401, got %d", rr.Code)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TestRoutingGetExposesSystemLockedDomains: GET must surface the injected
|
||||||
|
// lockedDomains list under system_locked_domains, without it ever ending up
|
||||||
|
// as part of routing.Profile's own field set.
|
||||||
|
func TestRoutingGetExposesSystemLockedDomains(t *testing.T) {
|
||||||
|
db := openRoutingTestDB(t)
|
||||||
|
seedRoutingUser(t, db, 8)
|
||||||
|
api := NewRoutingAPI(routing.NewStore(db), []string{"nas.x.com"})
|
||||||
|
|
||||||
|
rr := doAuthReq(t, http.MethodGet, "/v1/me/routing", nil, 8, api.GetProfile)
|
||||||
|
if rr.Code != 200 {
|
||||||
|
t.Fatalf("GET code %d", rr.Code)
|
||||||
|
}
|
||||||
|
var resp struct {
|
||||||
|
SystemLockedDomains []string `json:"system_locked_domains"`
|
||||||
|
}
|
||||||
|
if err := json.Unmarshal(rr.Body.Bytes(), &resp); err != nil {
|
||||||
|
t.Fatalf("unmarshal: %v", err)
|
||||||
|
}
|
||||||
|
if len(resp.SystemLockedDomains) != 1 || resp.SystemLockedDomains[0] != "nas.x.com" {
|
||||||
|
t.Fatalf("want system_locked_domains=[nas.x.com], got %v", resp.SystemLockedDomains)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestRoutingGetSystemLockedDomainsNilBecomesEmptyArray: a nil lockedDomains
|
||||||
|
// slice (no PANGOLIN_PRIVATE_SPLIT_DOMAINS configured) must serialize as
|
||||||
|
// `[]`, not `null` — clients shouldn't need a nil-check.
|
||||||
|
func TestRoutingGetSystemLockedDomainsNilBecomesEmptyArray(t *testing.T) {
|
||||||
|
db := openRoutingTestDB(t)
|
||||||
|
seedRoutingUser(t, db, 9)
|
||||||
|
api := NewRoutingAPI(routing.NewStore(db), nil)
|
||||||
|
|
||||||
|
rr := doAuthReq(t, http.MethodGet, "/v1/me/routing", nil, 9, api.GetProfile)
|
||||||
|
if rr.Code != 200 {
|
||||||
|
t.Fatalf("GET code %d", rr.Code)
|
||||||
|
}
|
||||||
|
var raw map[string]json.RawMessage
|
||||||
|
if err := json.Unmarshal(rr.Body.Bytes(), &raw); err != nil {
|
||||||
|
t.Fatalf("unmarshal: %v", err)
|
||||||
|
}
|
||||||
|
if got := string(raw["system_locked_domains"]); got != "[]" {
|
||||||
|
t.Fatalf("want system_locked_domains=`[]`, got %s", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestRoutingSaveIgnoresSystemLockedDomains: a client POSTing a body that
|
||||||
|
// includes system_locked_domains must not have it persisted — SaveProfile
|
||||||
|
// decodes straight into routing.Profile, which has no such field, so the
|
||||||
|
// key is silently dropped. Verify against the raw stored row (not the GET
|
||||||
|
// response, which always injects it from a.lockedDomains regardless of what
|
||||||
|
// was ever saved).
|
||||||
|
func TestRoutingSaveIgnoresSystemLockedDomains(t *testing.T) {
|
||||||
|
db := openRoutingTestDB(t)
|
||||||
|
seedRoutingUser(t, db, 10)
|
||||||
|
store := routing.NewStore(db)
|
||||||
|
api := NewRoutingAPI(store, []string{"nas.x.com"})
|
||||||
|
|
||||||
|
body := `{"mode":"rule","builtin":{"china_direct":true,"lan_direct":true,"private_via_tunnel":true},` +
|
||||||
|
`"rules":[],"final":"proxy","system_locked_domains":["evil.attacker.com"]}`
|
||||||
|
rr := doAuthReq(t, http.MethodPost, "/v1/me/routing", strings.NewReader(body), 10, api.SaveProfile)
|
||||||
|
if rr.Code != 200 {
|
||||||
|
t.Fatalf("POST code %d body %s", rr.Code, rr.Body)
|
||||||
|
}
|
||||||
|
|
||||||
|
stored, err := store.Get(context.Background(), 10)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if stored == nil {
|
||||||
|
t.Fatal("expected a persisted profile")
|
||||||
|
}
|
||||||
|
raw, err := json.Marshal(stored)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if strings.Contains(string(raw), "system_locked_domains") || strings.Contains(string(raw), "evil.attacker.com") {
|
||||||
|
t.Fatalf("system_locked_domains must never be persisted, got stored profile JSON: %s", raw)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -21,8 +21,17 @@ type Rule struct {
|
|||||||
// Builtin toggles the built-in routing behaviors that previously were
|
// Builtin toggles the built-in routing behaviors that previously were
|
||||||
// hardcoded into the server's sing-box config rendering.
|
// hardcoded into the server's sing-box config rendering.
|
||||||
type Builtin struct {
|
type Builtin struct {
|
||||||
ChinaDirect bool `json:"china_direct"`
|
ChinaDirect bool `json:"china_direct"`
|
||||||
LanDirect bool `json:"lan_direct"`
|
// LanDirect is reserved/always-on: the system layer (route_exclude_address
|
||||||
|
// + the hardcoded LAN direct rule in clientconfig.go) renders unconditionally
|
||||||
|
// regardless of this flag's value — the renderer does not read it. Kept in
|
||||||
|
// the wire contract for a possible future per-user opt-out; today it has no
|
||||||
|
// effect on rendering.
|
||||||
|
LanDirect bool `json:"lan_direct"`
|
||||||
|
// PrivateViaTunnel is reserved/always-on: when PANGOLIN_PRIVATE_SPLIT_DOMAINS
|
||||||
|
// is configured, those domains are always force-routed through the tunnel —
|
||||||
|
// the renderer does not read this flag. Kept in the wire contract for a
|
||||||
|
// possible future per-user opt-out; today it has no effect on rendering.
|
||||||
PrivateViaTunnel bool `json:"private_via_tunnel"`
|
PrivateViaTunnel bool `json:"private_via_tunnel"`
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -64,11 +73,11 @@ var geoWhitelist = map[string]bool{"cn": true} // geoip/geosite 仅自托管 cn
|
|||||||
|
|
||||||
// Validate checks the profile against the type/action whitelist, CIDR
|
// Validate checks the profile against the type/action whitelist, CIDR
|
||||||
// syntax, the geo set whitelist, and the rule count cap. It returns an empty
|
// syntax, the geo set whitelist, and the rule count cap. It returns an empty
|
||||||
// (nil) slice when the profile is valid; every violation is reported
|
// (non-nil) slice when the profile is valid — so JSON encoding produces `[]`
|
||||||
// independently (no short-circuiting) so callers can surface all errors at
|
// rather than `null` — and every violation is reported independently (no
|
||||||
// once.
|
// short-circuiting) so callers can surface all errors at once.
|
||||||
func (p *Profile) Validate() []FieldError {
|
func (p *Profile) Validate() []FieldError {
|
||||||
var errs []FieldError
|
errs := []FieldError{}
|
||||||
if p.Mode != "rule" && p.Mode != "global" && p.Mode != "direct" {
|
if p.Mode != "rule" && p.Mode != "global" && p.Mode != "direct" {
|
||||||
errs = append(errs, FieldError{-1, "mode", "must be rule|global|direct"})
|
errs = append(errs, FieldError{-1, "mode", "must be rule|global|direct"})
|
||||||
}
|
}
|
||||||
@@ -102,14 +111,20 @@ func (p *Profile) Validate() []FieldError {
|
|||||||
return errs
|
return errs
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// dedupKey identifies a rule for Normalize's de-duplication. Using a struct
|
||||||
|
// (rather than string-concatenating Type+Value+Action with a separator)
|
||||||
|
// avoids false-collision when a value itself contains the separator
|
||||||
|
// character.
|
||||||
|
type dedupKey struct{ Type, Value, Action string }
|
||||||
|
|
||||||
// Normalize trims rule values and de-duplicates rules by (type, value,
|
// Normalize trims rule values and de-duplicates rules by (type, value,
|
||||||
// action), keeping the first occurrence's position (and its Note/Enabled).
|
// action), keeping the first occurrence's position (and its Note/Enabled).
|
||||||
func (p *Profile) Normalize() {
|
func (p *Profile) Normalize() {
|
||||||
seen := map[string]bool{}
|
seen := map[dedupKey]bool{}
|
||||||
out := p.Rules[:0]
|
out := p.Rules[:0]
|
||||||
for _, r := range p.Rules {
|
for _, r := range p.Rules {
|
||||||
r.Value = strings.TrimSpace(r.Value)
|
r.Value = strings.TrimSpace(r.Value)
|
||||||
k := r.Type + "|" + r.Value + "|" + r.Action
|
k := dedupKey{r.Type, r.Value, r.Action}
|
||||||
if seen[k] {
|
if seen[k] {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,13 +1,32 @@
|
|||||||
package routing
|
package routing
|
||||||
|
|
||||||
import "testing"
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
// TestValidateEmptyErrorsSerializeAsArray guards the JSON shape API clients
|
||||||
|
// depend on: a valid profile's Validate() must marshal to `[]`, not `null`.
|
||||||
|
func TestValidateEmptyErrorsSerializeAsArray(t *testing.T) {
|
||||||
|
raw, err := json.Marshal(Default().Validate())
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if string(raw) != "[]" {
|
||||||
|
t.Fatalf("want `[]`, got %s", raw)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestValidate(t *testing.T) {
|
func TestValidate(t *testing.T) {
|
||||||
ok := Default()
|
ok := Default()
|
||||||
ok.Rules = []Rule{{Type: "domain_suffix", Value: "example.com", Action: "direct", Enabled: true}}
|
ok.Rules = []Rule{{Type: "domain_suffix", Value: "example.com", Action: "direct", Enabled: true}}
|
||||||
if e := ok.Validate(); len(e) != 0 {
|
e := ok.Validate()
|
||||||
|
if len(e) != 0 {
|
||||||
t.Fatalf("valid profile got errors %v", e)
|
t.Fatalf("valid profile got errors %v", e)
|
||||||
}
|
}
|
||||||
|
if e == nil {
|
||||||
|
t.Fatal("Validate must return a non-nil empty slice (serializes to [] not null)")
|
||||||
|
}
|
||||||
|
|
||||||
bad := Default()
|
bad := Default()
|
||||||
bad.Mode = "weird" // 非法 mode
|
bad.Mode = "weird" // 非法 mode
|
||||||
@@ -48,3 +67,19 @@ func TestNormalizeDedupAndTrim(t *testing.T) {
|
|||||||
t.Fatalf("want first-occurrence order preserved, got %v", p.Rules)
|
t.Fatalf("want first-occurrence order preserved, got %v", p.Rules)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TestNormalizeDedupNoSeparatorCollision guards against the historical
|
||||||
|
// string-concatenation dedup key ("type|value|action"): two distinct rules
|
||||||
|
// whose Value contains "|" could concatenate to the same string even though
|
||||||
|
// (type, value, action) differ. The struct-keyed dedup must tell them apart.
|
||||||
|
func TestNormalizeDedupNoSeparatorCollision(t *testing.T) {
|
||||||
|
p := Default()
|
||||||
|
p.Rules = []Rule{
|
||||||
|
{Type: "domain", Value: "a|b", Action: "proxy", Enabled: true},
|
||||||
|
{Type: "domain", Value: "a", Action: "b|proxy", Enabled: true},
|
||||||
|
}
|
||||||
|
p.Normalize()
|
||||||
|
if len(p.Rules) != 2 {
|
||||||
|
t.Fatalf("want 2 distinct rules preserved (no false collision), got %d: %v", len(p.Rules), p.Rules)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -32,6 +32,27 @@ func seedU(t *testing.T, db *sql.DB, id int64, uuid string) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TestSQLiteRoutingStoreGetNoProfileRow isolates the "queried the DB, found
|
||||||
|
// no row" path from TestSQLiteRoutingStoreUpsertGet's combined
|
||||||
|
// empty→upsert→overwrite flow: a store backed by a real (non-nil) *sql.DB,
|
||||||
|
// with the user seeded but no routing_profiles row for them, must return
|
||||||
|
// (nil, nil) — not an error — via the sql.ErrNoRows branch in Store.Get.
|
||||||
|
func TestSQLiteRoutingStoreGetNoProfileRow(t *testing.T) {
|
||||||
|
db := openDB(t)
|
||||||
|
seedU(t, db, 2, "u2")
|
||||||
|
st := NewStore(db)
|
||||||
|
if st == nil {
|
||||||
|
t.Fatal("NewStore returned nil")
|
||||||
|
}
|
||||||
|
got, err := st.Get(context.Background(), 2)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("want nil error for no-rows, got %v", err)
|
||||||
|
}
|
||||||
|
if got != nil {
|
||||||
|
t.Fatalf("want nil profile for seeded user with no profile row, got %+v", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestSQLiteRoutingStoreUpsertGet(t *testing.T) {
|
func TestSQLiteRoutingStoreUpsertGet(t *testing.T) {
|
||||||
db := openDB(t) // 内存库 + MigrateUp(sqlite)
|
db := openDB(t) // 内存库 + MigrateUp(sqlite)
|
||||||
seedU(t, db, 1, "u1")
|
seedU(t, db, 1, "u1")
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
CREATE TABLE routing_profiles (
|
CREATE TABLE routing_profiles (
|
||||||
user_id INTEGER NOT NULL PRIMARY KEY,
|
user_id INTEGER NOT NULL PRIMARY KEY,
|
||||||
profile_json TEXT NOT NULL,
|
profile_json TEXT NOT NULL,
|
||||||
updated_at DATETIME NOT NULL
|
updated_at DATETIME NOT NULL,
|
||||||
|
FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE CASCADE
|
||||||
);
|
);
|
||||||
|
|||||||
Reference in New Issue
Block a user