UNPKG

@gguf/claw

Version:

Multi-channel AI gateway with extensible messaging integrations

91 lines (89 loc) 3.65 kB
import { constants } from "node:fs"; import path from "node:path"; import fs$1 from "node:fs/promises"; //#region src/infra/fs-safe.ts var SafeOpenError = class extends Error { constructor(code, message, options) { super(message, options); this.code = code; this.name = "SafeOpenError"; } }; const NOT_FOUND_CODES = new Set(["ENOENT", "ENOTDIR"]); const SUPPORTS_NOFOLLOW = process.platform !== "win32" && "O_NOFOLLOW" in constants; const OPEN_READ_FLAGS = constants.O_RDONLY | (SUPPORTS_NOFOLLOW ? constants.O_NOFOLLOW : 0); const ensureTrailingSep = (value) => value.endsWith(path.sep) ? value : value + path.sep; const isNodeError = (err) => Boolean(err && typeof err === "object" && "code" in err); const isNotFoundError = (err) => isNodeError(err) && typeof err.code === "string" && NOT_FOUND_CODES.has(err.code); const isSymlinkOpenError = (err) => isNodeError(err) && (err.code === "ELOOP" || err.code === "EINVAL" || err.code === "ENOTSUP"); async function openVerifiedLocalFile(filePath) { let handle; try { handle = await fs$1.open(filePath, OPEN_READ_FLAGS); } catch (err) { if (isNotFoundError(err)) throw new SafeOpenError("not-found", "file not found"); if (isSymlinkOpenError(err)) throw new SafeOpenError("symlink", "symlink open blocked", { cause: err }); throw err; } try { const [stat, lstat] = await Promise.all([handle.stat(), fs$1.lstat(filePath)]); if (lstat.isSymbolicLink()) throw new SafeOpenError("symlink", "symlink not allowed"); if (!stat.isFile()) throw new SafeOpenError("not-file", "not a file"); if (stat.ino !== lstat.ino || stat.dev !== lstat.dev) throw new SafeOpenError("path-mismatch", "path changed during read"); const realPath = await fs$1.realpath(filePath); const realStat = await fs$1.stat(realPath); if (stat.ino !== realStat.ino || stat.dev !== realStat.dev) throw new SafeOpenError("path-mismatch", "path mismatch"); return { handle, realPath, stat }; } catch (err) { await handle.close().catch(() => {}); if (err instanceof SafeOpenError) throw err; if (isNotFoundError(err)) throw new SafeOpenError("not-found", "file not found"); throw err; } } async function openFileWithinRoot(params) { let rootReal; try { rootReal = await fs$1.realpath(params.rootDir); } catch (err) { if (isNotFoundError(err)) throw new SafeOpenError("not-found", "root dir not found"); throw err; } const rootWithSep = ensureTrailingSep(rootReal); const resolved = path.resolve(rootWithSep, params.relativePath); if (!resolved.startsWith(rootWithSep)) throw new SafeOpenError("invalid-path", "path escapes root"); let opened; try { opened = await openVerifiedLocalFile(resolved); } catch (err) { if (err instanceof SafeOpenError) { if (err.code === "not-found") throw err; throw new SafeOpenError("invalid-path", "path is not a regular file under root", { cause: err }); } throw err; } if (!opened.realPath.startsWith(rootWithSep)) { await opened.handle.close().catch(() => {}); throw new SafeOpenError("invalid-path", "path escapes root"); } return opened; } async function readLocalFileSafely(params) { const opened = await openVerifiedLocalFile(params.filePath); try { if (params.maxBytes !== void 0 && opened.stat.size > params.maxBytes) throw new SafeOpenError("too-large", `file exceeds limit of ${params.maxBytes} bytes (got ${opened.stat.size})`); return { buffer: await opened.handle.readFile(), realPath: opened.realPath, stat: opened.stat }; } finally { await opened.handle.close().catch(() => {}); } } //#endregion export { openFileWithinRoot as n, readLocalFileSafely as r, SafeOpenError as t };