197 lines
7.4 KiB
Python
197 lines
7.4 KiB
Python
import t, c
|
||
from stdint import *
|
||
import platmacro
|
||
import memhub
|
||
import string
|
||
|
||
# Windows 后端依赖
|
||
import w32.win32base
|
||
import w32.win32file
|
||
import w32.win32process
|
||
import w32.win32sync
|
||
|
||
# POSIX 后端依赖
|
||
import posix
|
||
|
||
|
||
# 子进程输出缓冲区大小
|
||
OUTPUT_BUF_SIZE: t.CDefine = 65536
|
||
|
||
# 模块级全局 MBuddy 指针（由用户在 main 中赋值初始化）
|
||
_mbuddy: memhub.MemBuddy | t.CPtr
|
||
|
||
|
||
# =============================================================================
|
||
# CompletedProcess 结果类
|
||
# =============================================================================
|
||
|
||
class CompletedProcess:
|
||
"""subprocess.run 的返回结果"""
|
||
args: str # 原始命令行
|
||
returncode: int # 进程退出码（0 表示成功）
|
||
stdout: str # 标准输出（capture_output=True 时）
|
||
stderr: str # 标准错误（capture_output=True 时）
|
||
|
||
def __new__(self, args: str, returncode: int, stdout: str, stderr: str) -> t.CPtr:
|
||
# 堆分配（避免返回栈地址导致 use-after-return）
|
||
return t.CPtr(_mbuddy.alloc(CompletedProcess.__sizeof__()))
|
||
|
||
def __init__(self, args: str, returncode: int, stdout: str, stderr: str):
|
||
self.args = args
|
||
self.returncode = returncode
|
||
self.stdout = stdout
|
||
self.stderr = stderr
|
||
|
||
|
||
# =============================================================================
|
||
# 入口函数
|
||
# =============================================================================
|
||
|
||
def run(args: str, capture_output: bool = True, text: bool = True) -> CompletedProcess | t.CPtr:
|
||
"""同步执行子进程并等待完成。
|
||
args 为完整命令行字符串。
|
||
capture_output=True 时捕获 stdout/stderr。
|
||
text=True 时输出为字符串（当前实现总是返回字符串）。
|
||
返回 CompletedProcess 对象。"""
|
||
if platmacro.IS_WINDOWS:
|
||
return _run_win32(args, capture_output, text)
|
||
return _run_posix(args, capture_output, text)
|
||
|
||
|
||
# =============================================================================
|
||
# Windows 后端实现
|
||
# =============================================================================
|
||
|
||
def _run_win32(args: str, capture_output: bool, text: bool) -> CompletedProcess | t.CPtr:
|
||
"""Windows 平台子进程执行（CreateProcessA + 管道）"""
|
||
# 复制命令行（CreateProcessA 需要可修改的 lpCommandLine）
|
||
args_len: t.CSizeT = len(args)
|
||
cmd_line: bytes = _mbuddy.alloc(args_len + 1)
|
||
if cmd_line == None:
|
||
return CompletedProcess(args, -1, None, None)
|
||
i: t.CSizeT = 0
|
||
for ch in args:
|
||
cmd_line[i] = ch
|
||
i += 1
|
||
cmd_line[args_len] = 0
|
||
|
||
# 安全属性（允许句柄继承）
|
||
sa: w32.win32base.SECURITY_ATTRIBUTES = w32.win32base.SECURITY_ATTRIBUTES()
|
||
sa.nLength = w32.win32base.SECURITY_ATTRIBUTES.__sizeof__()
|
||
sa.lpSecurityDescriptor = t.CPtr(0)
|
||
sa.bInheritHandle = 1 # TRUE
|
||
|
||
# 创建 stdout 管道
|
||
stdout_read: w32.win32base.HANDLE = t.CPtr(0)
|
||
stdout_write: w32.win32base.HANDLE = t.CPtr(0)
|
||
if capture_output:
|
||
pipe_result: BOOL = w32.win32file.CreatePipe(
|
||
t.CPtr(c.Addr(stdout_read)),
|
||
t.CPtr(c.Addr(stdout_write)),
|
||
t.CPtr(c.Addr(sa)),
|
||
0
|
||
)
|
||
if pipe_result == 0:
|
||
return CompletedProcess(args, -1, None, None)
|
||
# 防止读端被子进程继承
|
||
w32.win32file.SetHandleInformation(stdout_read, w32.win32file.HANDLE_FLAG_INHERIT, 0)
|
||
|
||
# 设置启动信息
|
||
si: w32.win32process.STARTUPINFOA = w32.win32process.STARTUPINFOA()
|
||
si.cb = w32.win32process.STARTUPINFOA.__sizeof__()
|
||
if capture_output:
|
||
si.dwFlags = w32.win32process.STARTF_USESTDHANDLES
|
||
si.hStdOutput = stdout_write
|
||
si.hStdError = stdout_write # stderr 合并到 stdout
|
||
|
||
pi: w32.win32process.PROCESS_INFORMATION = w32.win32process.PROCESS_INFORMATION()
|
||
|
||
# 创建进程
|
||
# capture_output=True 时用 CREATE_NO_WINDOW（输出到管道，不需要控制台）
|
||
# capture_output=False 时用 0（继承父进程控制台，让 stdout 直接显示）
|
||
create_flags: ULONG = w32.win32process.CREATE_NO_WINDOW if capture_output else 0
|
||
create_result: BOOL = w32.win32process.CreateProcessA(
|
||
t.CPtr(0), # lpApplicationName (NULL，从命令行解析)
|
||
cmd_line, # lpCommandLine (可修改)
|
||
t.CPtr(0), # lpProcessAttributes
|
||
t.CPtr(0), # lpThreadAttributes
|
||
1, # bInheritHandles = TRUE
|
||
create_flags, # dwCreationFlags
|
||
t.CPtr(0), # lpEnvironment (继承父进程)
|
||
t.CPtr(0), # lpCurrentDirectory (继承父进程)
|
||
t.CPtr(c.Addr(si)),
|
||
t.CPtr(c.Addr(pi))
|
||
)
|
||
|
||
if create_result == 0:
|
||
# 创建失败
|
||
if capture_output:
|
||
w32.win32base.CloseHandle(stdout_read)
|
||
w32.win32base.CloseHandle(stdout_write)
|
||
return CompletedProcess(args, -1, None, None)
|
||
|
||
# 关闭父进程的写端（子进程已有写端的继承副本）
|
||
if capture_output:
|
||
w32.win32base.CloseHandle(stdout_write)
|
||
|
||
# 读取 stdout + 等待进程结束
|
||
# 注意：必须边读边等，否则子进程输出超过管道缓冲区时会死锁
|
||
# （子进程阻塞在写 stdout，父进程阻塞在 WaitForSingleObject）
|
||
stdout_str: str = None
|
||
if capture_output:
|
||
buf_size: ULONG = OUTPUT_BUF_SIZE
|
||
buf: bytes = _mbuddy.alloc(buf_size)
|
||
if buf != None:
|
||
total: ULONG = 0
|
||
while 1:
|
||
bytes_read: ULONG = 0
|
||
read_result: BOOL = w32.win32file.ReadFile(
|
||
stdout_read,
|
||
buf + total,
|
||
buf_size - 1 - total,
|
||
t.CPtr(c.Addr(bytes_read)),
|
||
t.CPtr(0)
|
||
)
|
||
if read_result == 0 or bytes_read == 0:
|
||
break
|
||
total += bytes_read
|
||
# 缓冲区将满，扩容（翻倍）
|
||
if total >= buf_size - 1 - 4096:
|
||
new_size: ULONG = buf_size * 2
|
||
new_buf: bytes = _mbuddy.alloc(new_size)
|
||
if new_buf != None:
|
||
string.memcpy(new_buf, buf, total)
|
||
buf = new_buf
|
||
buf_size = new_size
|
||
else:
|
||
# 扩容失败，停止读取
|
||
break
|
||
buf[total] = 0
|
||
stdout_str = str(buf)
|
||
|
||
# 等待进程结束（如果还没退出的话）
|
||
w32.win32sync.WaitForSingleObject(pi.hProcess, w32.win32base.INFINITE)
|
||
|
||
# 获取退出码
|
||
exit_code: ULONG = 0
|
||
w32.win32process.GetExitCodeProcess(pi.hProcess, t.CPtr(c.Addr(exit_code)))
|
||
|
||
# 关闭句柄
|
||
w32.win32base.CloseHandle(pi.hProcess)
|
||
w32.win32base.CloseHandle(pi.hThread)
|
||
if capture_output:
|
||
w32.win32base.CloseHandle(stdout_read)
|
||
|
||
return CompletedProcess(args, exit_code, stdout_str, None)
|
||
|
||
|
||
# =============================================================================
|
||
# POSIX 后端实现
|
||
# =============================================================================
|
||
|
||
def _run_posix(args: str, capture_output: bool, text: bool) -> CompletedProcess | t.CPtr:
|
||
"""POSIX 平台子进程执行（fork + execvp + pipe + waitpid）
|
||
注意：当前为桩函数。POSIX 实现需要 pipe/fork/waitpid，
|
||
这些函数在 Windows MinGW 上不可用，待 POSIX 平台测试时再完善。"""
|
||
return CompletedProcess(args, -1, None, None)
|