feat(agent): unified cross-platform shell execution contract (cmd/bash)

Windows runs 'cmd.exe /d /s /c <command>' as a string command line; Linux uses ['/bin/bash', '-lc', command]. Removes implicit shell=True behavior and aligns the agent system prompt per platform.
This commit is contained in:
2026-09-17 16:40:02 +08:00
parent ce56c77023
commit 60200260ab
5 changed files with 424 additions and 54 deletions
+141
View File
@@ -0,0 +1,141 @@
# -*- coding: utf-8 -*-
"""P1-02:窄平台进程适配 —— 显式 shell 契约 + 进程树终止 + 提示词平台段。
契约(不依赖 shell=True 的平台默认值):
- Windows:字符串命令行 ``cmd.exe /d /s /c "<command>"``/s 保留内部引号;
不能传 argv 列表——list2cmdline 的引号转义 cmd 不认)
- Linux ``/bin/bash -lc <command>``(登录 shell,可读 profile 别名/函数)
进程树终止:
- Windowstaskkill /F /T(既有等价机制,整树强杀)
- Linux start_new_session 独立 POSIX 进程组;SIGTERM 整组 → 宽限期 → SIGKILL 整组
(安全不变量:仅当 getpgid == 子进程 pid,即确认独立组后才 killpg,
否则退化为单进程 kill,绝不误杀调用方所在组)
系统提示词:通用正文 + 运行时插入的短平台 shell/path 段(占位符
``{{SHELL_PLATFORM_SECTION}}``)。不维护两份完整提示词。
"""
import os
import signal
import subprocess
import time
#: 系统提示词中的平台段占位符
SHELL_SECTION_PLACEHOLDER = "{{SHELL_PLATFORM_SECTION}}"
_KILL_GRACE_S = 3.0 # LinuxSIGTERM → SIGKILL 宽限
_TASKKILL_TIMEOUT_S = 10 # Windowstaskkill 自身超时
def is_windows() -> bool:
return os.name == "nt"
def shell_command(command: str):
"""显式 shell 命令(bash 工具的 shell 契约,替代 shell=True 平台默认值)。
- Windows:返回**字符串命令行** ``cmd.exe /d /s /c "<command>"``。
必须走字符串而非 argv 列表:列表形态会被 CPython ``list2cmdline``
给内部引号加反斜杠转义,cmd.exe 不认 ``\"``;字符串形态下命令行
原样交给 CreateProcessWcmd 自行按 /s 规则解析 /c 参数
(外层引号剥离、内部引号保留,实测引号路径/&&/管道/%VAR% 均正确)。
- Linux:返回 argv 列表 ``[\ "/bin/bash", "-lc", command]``POSIX 下
列表参数无引号转义问题,-l 登录 shell 可读 profile 别名/函数)。
"""
if is_windows():
return 'cmd.exe /d /s /c "' + command + '"'
return ["/bin/bash", "-lc", command]
def popen_flags() -> dict:
"""进程树管理所需的 Popen 附加参数:Linux 独立进程组。"""
if is_windows():
return {}
return {"start_new_session": True}
def kill_process_tree(proc, grace_s: float = _KILL_GRACE_S) -> None:
"""终止 proc 的整棵子进程树;最坏情况也保证返回(不抛异常)。
- Windowstaskkill /F /T 整树强杀(既有机制)。
- Linux:独立进程组 SIGTERM → 宽限 grace_s 秒 → SIGKILL 整组。
若无法确认独立组(进程已退 / 标志未生效)→ 退化单进程 kill。
"""
if proc is None:
return
if is_windows():
try:
subprocess.run(["taskkill", "/F", "/T", "/PID", str(proc.pid)],
capture_output=True, timeout=_TASKKILL_TIMEOUT_S)
except Exception:
try:
proc.kill()
except Exception:
pass
return
# ---- POSIX ----
pgid = None
try:
pgid = os.getpgid(proc.pid)
except Exception:
pass
if pgid is None or pgid != proc.pid:
# 进程已退出,或没有独立组 → 不能 killpg(可能误伤调用方组)
try:
proc.kill()
except Exception:
pass
return
def _pg(sig: int) -> None:
try:
os.killpg(pgid, sig)
except Exception:
pass
_pg(signal.SIGTERM)
deadline = time.time() + grace_s
while time.time() < deadline:
try:
if proc.poll() is not None:
return
except Exception:
return
time.sleep(0.05)
_pg(signal.SIGKILL)
# ----------------------------------------------------------------------
# 系统提示词平台段(短段,运行时插入;通用正文保持平台无关)
# ----------------------------------------------------------------------
_WIN_SECTION = """### 1.1 平台 shell 与路径(Windows
- 操作系统:Windows。路径形如 `C:\\Users\\14890\\Desktop\\haocode`;不确定真实路径时先 `dir` / `ls`。
- bash 命令经 **cmd.exe** 执行(不是 git-bash);`C:\\Program Files\\Git\\usr\\bin` 在 PATH 上,
`ls` `grep` `cat` `head` `tail` `wc` `rm` `sed` `awk` 可直接用,`|` `>` `2>&1` `&&` 可用。
- ⚠️ cmd 陷阱:`;` 不是分隔符(用 `&&` 或换行);环境变量用 `%VAR%``$VAR` 不展开);
内建命令不剥单引号(`echo 'x'` 原样带引号);每次调用都是新进程,`cd` 不跨调用保留
(写 `cd core && <命令>`);需要 `for` / `$(...)` / `[ ]` 测试等真实 bash 语义时
一律 `bash -c "..."`。"""
_LINUX_SECTION = """### 1.1 平台 shell 与路径(Linux
- 操作系统:Linux。路径为 POSIX 风格(`/home/user/...`、`/tmp/...`)。
- bash 命令经 **/bin/bash -lc** 执行(登录 shell):`;` `&&` `|` `$(...)`、单引号、
环境变量 `$VAR`、profile 里的别名与函数都可用。
- 每次调用都是新进程:`cd` 不跨调用保留(写 `cd core && <命令>` 或直接用绝对路径)。"""
def shell_prompt_section() -> str:
"""当前平台的短 shell/path 段。"""
return _WIN_SECTION if is_windows() else _LINUX_SECTION
def apply_platform_section(text: str) -> str:
"""把占位符替换为当前平台段;无占位符(如兜底提示词)时原样返回。"""
if SHELL_SECTION_PLACEHOLDER in text:
return text.replace(SHELL_SECTION_PLACEHOLDER, shell_prompt_section())
return text