feat: 远程目录 remote_dir 改用相对路径(从远程用户 home 目录开始)

- 新增 _resolve_remote_dir(相对→~/xxx)+ _validate_remote_dir(防绝对路径/..穿越)
- 移除 _FORBIDDEN_DIRS 系统目录黑名单(相对路径天然限制在 home 内)
- rsync 迁移 + bwrap --bind 统一走 _resolve_remote_dir
This commit is contained in:
ymq 2026-08-14 14:20:47 +08:00
parent 07383a9dd5
commit dc4379b235

View File

@ -132,12 +132,28 @@ def get_org_pubkey(org_id: str) -> dict:
_HOST_RE = re.compile(r"^[a-zA-Z0-9]([a-zA-Z0-9\-\.]*[a-zA-Z0-9])?$")
_USER_RE = re.compile(r"^[a-zA-Z0-9._-]+$")
# 禁止作为 remote_dir 的系统目录(防 rsync --delete 误删系统文件)
_FORBIDDEN_DIRS = {
"/", "/bin", "/sbin", "/usr", "/etc", "/lib", "/lib64", "/var",
"/root", "/boot", "/dev", "/proc", "/sys", "/tmp", "/opt", "/run",
"/srv", "/mnt", "/media", "/home",
}
# remote_dir 现在是相对路径(从远程登录用户的 home 目录开始)。
# 相对路径天然限制在 home 内,无需系统目录黑名单,只需防路径穿越(..)。
def _resolve_remote_dir(remote_dir: str) -> str:
"""把相对路径规范化为远程 home 下的路径(~/xxx"""
rel = (remote_dir or "").strip().strip("/")
return "~/" + rel if rel else "~"
def _validate_remote_dir(remote_dir: str) -> str:
"""校验 remote_dir相对路径返回错误消息None 表示通过)。"""
d = (remote_dir or "").strip()
if not d:
return "缺少 remote_dir"
if d.startswith("/"):
return "remote_dir 必须是相对路径(从远程用户 home 目录开始)"
parts = [p for p in d.split("/") if p]
if ".." in parts:
return "remote_dir 不能包含 .."
if not parts:
return "remote_dir 不能为空"
return None
def _validate_remote(remote_config: dict) -> str:
@ -167,19 +183,9 @@ def _validate_remote(remote_config: dict) -> str:
except (ValueError, TypeError):
return "remote_port 必须是数字"
remote_dir = (remote_config.get("remote_dir") or "").strip()
if not remote_dir:
return "缺少 remote_dir"
if not remote_dir.startswith("/"):
return "remote_dir 必须是绝对路径"
_dir = remote_dir.rstrip("/")
if not _dir:
return "remote_dir 不能是根目录 /"
if _dir in _FORBIDDEN_DIRS:
return "remote_dir 不能是系统目录 " + _dir
for d in _FORBIDDEN_DIRS:
if d != "/" and _dir.startswith(d + "/"):
return "remote_dir 不能位于系统目录 " + d + ""
err = _validate_remote_dir(remote_config.get("remote_dir", ""))
if err:
return err
key = (remote_config.get("remote_key_path") or "").strip()
if key:
@ -214,17 +220,9 @@ async def migrate_work_dir(account_name: str, from_mode: str, to_mode: str, env:
# remote_dir 安全校验纵深防御DB 脏数据/历史记录也拦一道)
if from_mode == "remote" or to_mode == "remote":
remote_dir = (env.get("remote_dir") or "").strip()
if not remote_dir:
return {"ok": False, "error": "缺少远程目录 remote_dir"}
_d = remote_dir.rstrip("/")
if not _d:
return {"ok": False, "error": "远程目录是根目录,拒绝迁移"}
if _d in _FORBIDDEN_DIRS:
return {"ok": False, "error": "远程目录是系统目录,拒绝迁移: " + _d}
for d in _FORBIDDEN_DIRS:
if d != "/" and _d.startswith(d + "/"):
return {"ok": False, "error": "远程目录位于系统目录下,拒绝迁移: " + d}
err = _validate_remote_dir(env.get("remote_dir", ""))
if err:
return {"ok": False, "error": err}
# 构造 rsync 命令
try:
@ -233,7 +231,7 @@ async def migrate_work_dir(account_name: str, from_mode: str, to_mode: str, env:
remote_dir = env.get("remote_dir", "")
if not remote_dir:
return {"ok": False, "error": "缺少远程目录 remote_dir"}
target = _ssh_target(env) + ":" + remote_dir
target = _ssh_target(env) + ":" + _resolve_remote_dir(remote_dir)
cmd = ["rsync", "-az", "--delete", "-e",
"ssh " + " ".join(_ssh_common_args(env)),
local_dir.rstrip("/") + "/", target.rstrip("/") + "/"]
@ -242,7 +240,7 @@ async def migrate_work_dir(account_name: str, from_mode: str, to_mode: str, env:
remote_dir = env.get("remote_dir", "")
if not remote_dir:
return {"ok": False, "error": "缺少远程目录 remote_dir"}
src = _ssh_target(env) + ":" + remote_dir
src = _ssh_target(env) + ":" + _resolve_remote_dir(remote_dir)
cmd = ["rsync", "-az", "--delete", "-e",
"ssh " + " ".join(_ssh_common_args(env)),
src.rstrip("/") + "/", local_dir.rstrip("/") + "/"]
@ -366,7 +364,7 @@ def _remote_bwrap_cmd(env: dict, deploy_dir: str, command: str, workdir: str = "
"--proc", "/proc",
"--dev", "/dev",
"--tmpfs", "/tmp",
"--bind", deploy_dir, "/home",
"--bind", _resolve_remote_dir(deploy_dir), "/home",
"--chdir", "/home",
"--setenv", "HOME", "/home",
"--", "bash", "-c", command,