File size: 2,735 Bytes
d04f74a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 | /** Lazy libc execve and descriptor bindings used by the one-shot Linux bootstrap. */
import { getSystemErrorMessage, getSystemErrorName } from 'node:util'
import { SUBPROCESS_CONTROL_FD } from '@deepseek-ai/dsh-subprocess/control'
import { createLazyRequire } from '@deepseek-ai/dsh-lazy-require'
const requireKoffi = createLazyRequire<typeof import('koffi')['default']>('koffi', import.meta.url)
/** Replace the current process image while preserving the supplied argv and environment. */
export type LinuxExecve = (
file: string,
argv: string[],
env: Record<string, string>,
control?: 'pipe',
) => never
type NativeExecve = (
file: string,
argv: Array<string | null>,
envp: Array<string | null>,
) => number
type NativeFcntl = (fd: number, command: number, argument: number) => number
const STANDARD_FILE_DESCRIPTORS = [0, 1, 2] as const
const F_GETFD = 1
const F_SETFD = 2
const FD_CLOEXEC = 1
let cachedExecve: LinuxExecve | undefined
function systemError(errno: number, syscall: string, path?: string): Error {
const uvError = -errno
const code = getSystemErrorName(uvError)
const detail = getSystemErrorMessage(uvError)
const subject = path === undefined ? syscall : `${syscall} '${path}'`
const error = Object.assign(new Error(`${code}: ${detail}, ${subject}`), {
code,
errno: uvError,
syscall,
})
return path === undefined ? error : Object.assign(error, { path })
}
/**
* Load libc's execve and fcntl symbols on first use and retain the native bindings.
* @returns a process-replacing execve operation that throws Node-style errors on failure.
*/
export function loadLinuxExecve(): LinuxExecve {
if (cachedExecve !== undefined) return cachedExecve
const koffi = requireKoffi()
const libc = koffi.load(null)
const nativeExecve = libc.func(
'int execve(const char *pathname, const char **argv, const char **envp)',
) as NativeExecve
const nativeFcntl = libc.func(
'int fcntl(int fd, int cmd, int arg)',
) as NativeFcntl
cachedExecve = (file, argv, env, control) => {
const descriptors = control === 'pipe'
? [...STANDARD_FILE_DESCRIPTORS, SUBPROCESS_CONTROL_FD]
: STANDARD_FILE_DESCRIPTORS
for (const fd of descriptors) {
const flags = nativeFcntl(fd, F_GETFD, 0)
if (flags === -1) throw systemError(koffi.errno(), 'fcntl')
if ((flags & FD_CLOEXEC) === 0) continue
if (nativeFcntl(fd, F_SETFD, flags & ~FD_CLOEXEC) === -1) {
throw systemError(koffi.errno(), 'fcntl')
}
}
nativeExecve(
file,
[...argv, null],
[...Object.entries(env).map(([key, value]) => `${key}=${value}`), null],
)
throw systemError(koffi.errno(), 'execve', file)
}
return cachedExecve
}
|