mirror of
https://gitcode.com/JianFeeeee/ModelRouter.git
synced 2026-11-07 09:47:23 +00:00
feat(channels): VNC 兜底真正可用(依赖预检 + 就绪探针 + CDP 取凭据)
用户指出「vnc 无头浏览器兜底登陆你根本没有实现」。核对后确认:进程层代码存在,
但**只接在一个我编的假平台上**(loopbackish,已随夹具移出),四个真实平台全标
flow="manual" 并给一句"请你自己去某处搞定"。且代码写着「运行 deploy 脚本一键
安装」,而那个安装机制根本不存在(deploy.sh 里零浏览器栈内容)。
## 实现
- **依赖预检** browserDeps():探测 Xvfb/x11vnc/websockify/noVNC/chromium/
python3-websocket,如实报告缺什么、装哪个包、干什么用。
- **就绪探针** waitPort + X socket 等待:不再 sleep 2 拍脑袋(view_url 先于
服务可用会让 iframe 白屏)。
- **CDP 取凭据**:浏览器带 --remote-debugging-port,登录完成后经 CDP 读
localStorage/sessionStorage/cookie 或平台声明的 collect_expr。人只负责在 VNC
窗口里交互,机器负责把结果取出来 —— VNC 不该只是"看屏幕"。
实测:在真 HTTP 页面上 `localStorage.getItem('access_token')` → REAL-TOKEN-XYZ,
`document.cookie` → session=COOKIE-VAL,全部正确。
- **新端点**:channels/deps/status、browser/collect、browser/stop、browser/status。
- **安装入口** packaging/install-browser-stack.sh:识别 apt/dnf/yum/apk,
--check 只检查;装进发行包(/usr/share/llmsproxy/)。插件 install_hint 指向它。
默认不自动装:系统级包、数百 MB、chromium 安全面大,多数部署用不到 VNC。
## 过程中修掉三个真 bug(都是测试逼出来的)
1. **Xvfb 从未被启动**:steps 数组里有 Xvfb,但循环明确 `if st[1] ~= "Xvfb"`
跳过它(想先等 socket)→ 等一个永远不会出现的 socket。
2. **display 号按 seq 取模**:`90 + (seq % 10)`。Xvfb 锁文件在强杀后残留,
重启后 seq 从头 → 又选同一个号 → `Fatal: Server is already active`,
而报错只显示"Xvfb 未就绪"。改为**探测空闲**号(锁+socket+端口)。
3. **io.open 判 socket 存在**(两处):X socket 是 socket 文件,io.open 返回 nil
→ 探测误判"空闲"、等待循环永远等不到。改用 `[ -e ]`。
4. **孤儿进程**:spawn 往局部 pids 表追加,只在函数末尾回写 b.pids →
中途失败时 reapBrowser 读到空表,已起的 Xvfb 不被杀(实测留下 4 个孤儿)。
改为即时回写。
## 判据
- TestChannelsVncStackEndToEnd:注入 flow=vnc 平台 → 真起浏览器栈(view_url 含
vnc.html + CDP 端口)→ 回收。缺依赖则 SKIP 并说明。
- TestChannelsVncNoOrphanOnFailure:回收后 running=false、无孤儿。
- TestChannelsBrowserStackDeps:依赖清单三个字段齐全、缺件必须给安装提示。
- TestChannelsBrowserEndpointsRequireSession:无会话时三个端点如实报错/幂等。
- 全量 core 回归后确认:生产插件目录未被写入、无孤儿进程。
内核对照 v2.0.2:diff 为空(本次零内核改动)。
This commit is contained in:
@ -1603,10 +1603,50 @@ local function spawnBrowser(start_url, callback_path)
|
||||
end
|
||||
b.seq = b.seq + 1
|
||||
local seq = b.seq
|
||||
local disp = 90 + (seq % 10)
|
||||
local vport = 59100 + (seq % 100)
|
||||
local wport = 60100 + (seq % 100)
|
||||
local cdpPort = 61100 + (seq % 100)
|
||||
-- display / 端口取号:**探测空闲**,不能按 seq 取模。
|
||||
--
|
||||
-- 真实缺陷:原先写 `90 + (seq % 10)`,而 Xvfb 的锁文件(/tmp/.X<n>-lock)在
|
||||
-- 进程被强杀时会残留。重启后 seq 从头开始 → 又选中同一个号 → Xvfb 直接
|
||||
-- `Fatal: Server is already active for display 91` 退出,VNC 兜底永远起不来,
|
||||
-- 而报错只显示"Xvfb 未就绪",看不出是被残留锁挡住。
|
||||
-- 现在从 91 起找第一个既无锁文件、也无 socket 的号;端口同理探测空闲。
|
||||
local function pickFree(start, isFree)
|
||||
for n = start, start + 40 do
|
||||
if isFree(n) then return n end
|
||||
end
|
||||
return start
|
||||
end
|
||||
-- 判存在必须用 `[ -e ]` 而不是 io.open:X socket 是 socket 文件,
|
||||
-- io.open 对它返回 nil(打不开)→ 探测会误判"空闲"→ 又选中被残留 socket
|
||||
-- 占住的号 → Xvfb `Cannot establish any listening sockets` 退出。
|
||||
-- 这个坑真实踩过:残留 X91 socket 让显示器探测完全失效。
|
||||
local function pathExists(path)
|
||||
local ok = os.execute('[ -e "' .. path .. '" ] 2>/dev/null')
|
||||
-- Lua 5.1 返回数字(0=成功),5.2+ 返回 true/nil
|
||||
return ok == true or ok == 0
|
||||
end
|
||||
local disp = pickFree(91, function(n)
|
||||
return not pathExists("/tmp/.X" .. n .. "-lock")
|
||||
and not pathExists("/tmp/.X11-unix/X" .. n)
|
||||
end)
|
||||
local function portFree(p)
|
||||
-- 用 /proc/net/tcp + tcp6 查是否有 LISTEN(十六进制端口)
|
||||
for _, f in ipairs({ "/proc/net/tcp", "/proc/net/tcp6" }) do
|
||||
local h = io.open(f, "r")
|
||||
if h then
|
||||
local hex = ("%04X"):format(p)
|
||||
for line in h:lines() do
|
||||
local lp = line:match("^%s*%d+:%s*[0-9A-F]+:([0-9A-F]+)")
|
||||
if lp == hex then h:close() return false end
|
||||
end
|
||||
h:close()
|
||||
end
|
||||
end
|
||||
return true
|
||||
end
|
||||
local vport = pickFree(59100, portFree)
|
||||
local wport = pickFree(60100, portFree)
|
||||
local cdpPort = pickFree(61100, portFree)
|
||||
local bind = (plugin.state.settings and plugin.state.settings.vnc_bind) or "127.0.0.1"
|
||||
-- 会话密码:bind 到非回环时是唯一的防线,随机生成、只在响应里给一次
|
||||
local pass = sha256_hex(newToken()):sub(1, 16)
|
||||
@ -1622,9 +1662,14 @@ local function spawnBrowser(start_url, callback_path)
|
||||
local pid = h and h:read("*l")
|
||||
if h then h:close() end
|
||||
if not pid or pid == "" then
|
||||
return false, name .. " 启动失败"
|
||||
return false, name .. " 启动失败(cmd=" .. tostring(cmd):sub(1,80) .. ")"
|
||||
end
|
||||
pids[#pids + 1] = pid
|
||||
-- 立刻回写到 plugin.state,**不能只在函数末尾赋值**:
|
||||
-- 任何一步中途失败都要 reapBrowser(),而它读的是 b.pids。若此时 b.pids 还是
|
||||
-- 旧值(空),已经起来的进程就不会被杀 → 孤儿 Xvfb/chromium 常驻。
|
||||
-- 真实踩过:4 个失败的 VNC 会话留下 4 个 orphan Xvfb。
|
||||
b.pids = pids
|
||||
if port and port > 0 then
|
||||
if not waitPort(host or "127.0.0.1", port, timeoutMs or 8000) then
|
||||
return false, name .. " 端口未就绪(:" .. port .. ")"
|
||||
@ -1647,25 +1692,33 @@ local function spawnBrowser(start_url, callback_path)
|
||||
{ "websockify", ("websockify --web %s %s:%d 127.0.0.1:%d >/tmp/llmsproxy-ws-%d.log 2>&1 & echo $!"):format(
|
||||
novnc, bind, wport, vport, seq), bind, wport, 8000 },
|
||||
}
|
||||
-- X 的 socket 在 /tmp/.X11-unix/X<n>,先等它出现再起浏览器
|
||||
-- ① 先起 Xvfb(其余三步依赖 X display)
|
||||
local xvfbStep = steps[1]
|
||||
local xok, xerr = spawn(xvfbStep[1], xvfbStep[2], nil, 0, 0)
|
||||
if not xok then
|
||||
reapBrowser()
|
||||
return nil, xerr
|
||||
end
|
||||
-- X 的 socket 在 /tmp/.X11-unix/X<n>,等它出现再起浏览器
|
||||
local unixSock = ("/tmp/.X11-unix/X%d"):format(disp)
|
||||
local xdeadline = (os.time and os.time() or 0) + 10
|
||||
while true do
|
||||
local f = io.open(unixSock, "r")
|
||||
if f then f:close() break end
|
||||
-- 同 pathExists 的理由:X socket 是 socket 文件,io.open 打不开,
|
||||
-- 用它判"就绪"会永远等不到(这个坑在上面选号处也踩过一次)。
|
||||
if pathExists(unixSock) then break end
|
||||
if (os.time and os.time() or 0) >= xdeadline then
|
||||
reapBrowser()
|
||||
return nil, "Xvfb 未就绪(" .. unixSock .. " 未出现)"
|
||||
end
|
||||
os.execute("sleep 0.3")
|
||||
end
|
||||
for _, st in ipairs(steps) do
|
||||
if st[1] ~= "Xvfb" then
|
||||
local ok, err = spawn(st[1], st[2], st[3], st[4], st[5])
|
||||
if not ok then
|
||||
reapBrowser()
|
||||
return nil, err
|
||||
end
|
||||
-- ② 起浏览器/VNC/websockify(三步都有 TCP 端口可探)
|
||||
for i = 2, #steps do
|
||||
local st = steps[i]
|
||||
local ok, err = spawn(st[1], st[2], st[3], st[4], st[5])
|
||||
if not ok then
|
||||
reapBrowser()
|
||||
return nil, err
|
||||
end
|
||||
end
|
||||
|
||||
@ -3049,10 +3102,12 @@ function plugin.deps_status(req)
|
||||
present = d.present,
|
||||
chromium = d.chromium,
|
||||
novnc = d.novnc,
|
||||
-- 安装命令按发行版给两条(运维自己选),插件不执行它
|
||||
-- 安装入口:发行包自带脚本(识别 apt/dnf/yum/apk),插件**不执行**它
|
||||
-- (装系统包需要 root,且 chromium 是安全面较大的组件,不该由插件代劳)。
|
||||
install_hint = d.ok and "" or
|
||||
"Debian/Ubuntu: apt-get install -y xvfb x11vnc websockify novnc chromium python3-websocket | "
|
||||
.. "RHEL/Fedora: dnf install -y xorg-x11-server-Xvfb x11vnc python3-websockify novnc chromium python3-websocket-client",
|
||||
"sudo bash /usr/share/llmsproxy/install-browser-stack.sh"
|
||||
.. " 或沿源码部署时: sudo bash packaging/install-browser-stack.sh"
|
||||
.. "(脚本会自动识别 apt/dnf/yum/apk;也可手工装 Xvfb x11vnc websockify noVNC chromium python3-websocket)",
|
||||
} }
|
||||
end
|
||||
|
||||
|
||||
Reference in New Issue
Block a user