@jshookmcp/jshook 0.2.7 → 0.2.9

package/dist/handlers-Cgyg6c0U.mjs

@@ -0,0 +1,645 @@
+import { dr as UNIDBG_TIMEOUT_MS } from "./constants-B0OANIBL.mjs";
+import { a as invokePlugin, c as FridaSession, i as getAvailablePlugins, n as UnidbgRunner, o as HookCodeGenerator, r as HookGenerator, s as GhidraAnalyzer } from "./binary-instrument-CXfpx6fT.mjs";
+import { access } from "node:fs/promises";
+import { execFile } from "node:child_process";
+//#region src/server/domains/binary-instrument/handlers/shared.ts
+/**
+* Shared types and helpers for binary-instrument sub-handlers.
+*/
+const UNIDBG_MAX_BUFFER_BYTES = 8 * 1024 * 1024;
+function textResponse(text) {
+	return { content: [{
+		type: "text",
+		text
+	}] };
+}
+function jsonResponse(payload) {
+	return textResponse(JSON.stringify(payload));
+}
+function readRequiredString(args, key) {
+	const value = args[key];
+	if (typeof value !== "string" || value.trim().length === 0) throw new Error(`${key} is required`);
+	return value.trim();
+}
+function readOptionalString(args, key) {
+	const value = args[key];
+	return typeof value === "string" && value.trim().length > 0 ? value.trim() : void 0;
+}
+function readOptionalNumber(args, key) {
+	const value = args[key];
+	return typeof value === "number" && Number.isFinite(value) ? value : void 0;
+}
+function readStringArray(args, key) {
+	const value = args[key];
+	if (!Array.isArray(value)) return [];
+	return value.filter((entry) => typeof entry === "string" && entry.length > 0);
+}
+function isRecord(value) {
+	return typeof value === "object" && value !== null;
+}
+function isServerContext(value) {
+	return isRecord(value) && value["extensionPluginsById"] instanceof Map && value["extensionPluginRuntimeById"] instanceof Map;
+}
+function hasInstalledLegacyPlugin(context, pluginId) {
+	if (!context) return void 0;
+	const installed = context.extensionPluginsById;
+	if (!(installed instanceof Map)) return void 0;
+	return installed.has(pluginId);
+}
+async function invokeLegacyPlugin(context, pluginId, toolName, args) {
+	if (!context || hasInstalledLegacyPlugin(context, pluginId) === false) return textResponse(`Plugin ${pluginId.replaceAll("_", "-")} is not installed`);
+	const result = await invokePlugin(context, {
+		pluginId,
+		toolName,
+		args
+	});
+	if (result.success) return jsonResponse(result);
+	return textResponse(result.error ?? "Plugin invocation failed");
+}
+function readHookOptions(args, key) {
+	const raw = args[key];
+	if (!isRecord(raw)) return void 0;
+	const options = {};
+	const includeArgs = raw["includeArgs"];
+	const includeRetAddr = raw["includeRetAddr"];
+	if (typeof includeArgs === "boolean") options.includeArgs = includeArgs;
+	if (typeof includeRetAddr === "boolean") options.includeRetAddr = includeRetAddr;
+	return options;
+}
+function parsePid(target) {
+	if (!/^\d+$/.test(target)) return null;
+	const parsed = Number.parseInt(target, 10);
+	return Number.isNaN(parsed) ? null : parsed;
+}
+function makeMockId(value) {
+	return value.toLowerCase().replace(/[^a-z0-9]+/g, "-").replace(/^-+|-+$/g, "").slice(0, 24);
+}
+async function getUnidbgAvailability() {
+	const jarPath = process.env["UNIDBG_JAR"] ?? "";
+	if (jarPath.length === 0) return {
+		available: false,
+		reason: "UNIDBG_JAR is not configured",
+		command: "java",
+		jarPath: ""
+	};
+	try {
+		await access(jarPath);
+	} catch {
+		return {
+			available: false,
+			reason: `UNIDBG_JAR not found: ${jarPath}`,
+			command: "java",
+			jarPath
+		};
+	}
+	return {
+		available: true,
+		reason: "",
+		command: "java",
+		jarPath
+	};
+}
+function execFileUtf8(file, args, timeoutMs) {
+	return new Promise((resolve, reject) => {
+		execFile(file, args, {
+			timeout: timeoutMs,
+			windowsHide: true,
+			maxBuffer: UNIDBG_MAX_BUFFER_BYTES,
+			encoding: "utf8"
+		}, (error, stdout, stderr) => {
+			if (error) {
+				reject(error);
+				return;
+			}
+			resolve({
+				stdout: typeof stdout === "string" ? stdout : "",
+				stderr: typeof stderr === "string" ? stderr : ""
+			});
+		});
+	});
+}
+function isGhidraAnalysisOutput(value) {
+	return isRecord(value) && Array.isArray(value["functions"]) && Array.isArray(value["imports"]);
+}
+function toHookTemplates(value) {
+	const templates = [];
+	for (const entry of value) {
+		if (!isRecord(entry)) continue;
+		const functionName = readStringRecordField(entry, "functionName");
+		const hookCode = readStringRecordField(entry, "hookCode");
+		const description = readStringRecordField(entry, "description");
+		const parameters = parseHookParameters(entry["parameters"]);
+		if (!functionName || !hookCode || !description) continue;
+		templates.push({
+			functionName,
+			hookCode,
+			description,
+			parameters
+		});
+	}
+	return templates;
+}
+function readStringRecordField(record, key) {
+	const value = record[key];
+	return typeof value === "string" ? value : void 0;
+}
+function parseHookParameters(value) {
+	if (!Array.isArray(value)) return [];
+	const parameters = [];
+	for (const entry of value) {
+		if (!isRecord(entry)) continue;
+		const name = readStringRecordField(entry, "name");
+		const type = readStringRecordField(entry, "type");
+		const description = readStringRecordField(entry, "description");
+		if (name && type && description) parameters.push({
+			name,
+			type,
+			description
+		});
+	}
+	return parameters;
+}
+//#endregion
+//#region src/server/domains/binary-instrument/handlers/frida-handlers.ts
+/**
+* Frida sub-handler — attach, runScript, detach, listSessions, generateScript,
+* enumerateModules, enumerateFunctions, findSymbols.
+*/
+var FridaHandlers = class {
+	state;
+	constructor(state) {
+		this.state = state;
+	}
+	async handleFridaAttach(args) {
+		const legacyPid = readOptionalString(args, "pid");
+		if (!readOptionalString(args, "target") && legacyPid) return invokeLegacyPlugin(this.state.context, "plugin_frida_bridge", "frida_attach", args);
+		const target = readRequiredString(args, "target");
+		const frida = this.getFridaSession();
+		const availability = await frida.getAvailability();
+		if (!availability.available) {
+			const sessionId = `mock-frida-${makeMockId(target)}`;
+			return jsonResponse({
+				available: false,
+				target,
+				sessionId,
+				reason: availability.reason ?? "Frida CLI is not available",
+				sessions: [{
+					id: sessionId,
+					target,
+					pid: parsePid(target),
+					status: "unavailable"
+				}]
+			});
+		}
+		const sessionId = await frida.attach(target);
+		this.state.context?.eventBus.emit("frida:attached", {
+			target,
+			sessionId,
+			timestamp: (/* @__PURE__ */ new Date()).toISOString()
+		});
+		return jsonResponse({
+			available: true,
+			target,
+			sessionId,
+			sessions: frida.listSessions()
+		});
+	}
+	async handleFridaEnumerateModules(args) {
+		const sessionId = readRequiredString(args, "sessionId");
+		const frida = this.getFridaSession();
+		const availability = await frida.getAvailability();
+		if (!availability.available) return jsonResponse({
+			available: false,
+			sessionId,
+			reason: availability.reason ?? "Frida CLI is not available",
+			modules: [{
+				name: "mock-module",
+				base: "0x0",
+				size: 0,
+				path: "<unavailable>"
+			}]
+		});
+		if (!frida.useSession(sessionId)) return jsonResponse({
+			available: false,
+			sessionId,
+			reason: `Unknown Frida session: ${sessionId}`,
+			modules: []
+		});
+		return jsonResponse({
+			available: true,
+			sessionId,
+			modules: await frida.enumerateModules()
+		});
+	}
+	async handleFridaRunScript(args) {
+		const sessionId = readOptionalString(args, "sessionId");
+		if (!sessionId) return textResponse("Missing required string argument: sessionId");
+		const script = readRequiredString(args, "script");
+		const frida = this.getFridaSession();
+		const availability = await frida.getAvailability();
+		if (!availability.available) return {
+			available: false,
+			sessionId,
+			reason: availability.reason ?? "Frida CLI is not available",
+			execution: {
+				output: "",
+				error: "Frida unavailable"
+			}
+		};
+		if (!frida.useSession(sessionId)) return {
+			available: false,
+			sessionId,
+			reason: `Unknown Frida session: ${sessionId}`,
+			execution: {
+				output: "",
+				error: "Unknown session"
+			}
+		};
+		return {
+			available: true,
+			sessionId,
+			execution: await frida.executeScript(script)
+		};
+	}
+	async handleFridaDetach(args) {
+		const sessionId = readOptionalString(args, "sessionId");
+		if (!sessionId) return textResponse("Missing required string argument: sessionId");
+		const frida = this.getFridaSession();
+		if ((await frida.getAvailability()).available && frida.hasSession(sessionId)) {
+			frida.useSession(sessionId);
+			await frida.detach();
+			return jsonResponse({
+				success: true,
+				sessionId,
+				detached: true
+			});
+		}
+		return invokeLegacyPlugin(this.state.context, "plugin_frida_bridge", "frida_detach", args);
+	}
+	async handleFridaListSessions(_args) {
+		if (hasInstalledLegacyPlugin(this.state.context, "plugin_frida_bridge") === false) return textResponse("Plugin plugin-frida-bridge is not installed");
+		const frida = this.getFridaSession();
+		if ((await frida.getAvailability()).available) {
+			const sessions = frida.listSessions();
+			return jsonResponse({
+				success: true,
+				sessions,
+				count: sessions.length
+			});
+		}
+		return invokeLegacyPlugin(this.state.context, "plugin_frida_bridge", "frida_list_sessions", _args);
+	}
+	async handleFridaGenerateScript(args) {
+		if (hasInstalledLegacyPlugin(this.state.context, "plugin_frida_bridge") === false) return textResponse("Plugin plugin-frida-bridge is not installed");
+		const target = readOptionalString(args, "target") ?? "unknown";
+		const template = readOptionalString(args, "template") ?? "trace";
+		const functionName = readOptionalString(args, "functionName") ?? "target_function";
+		const templates = [{
+			functionName,
+			hookCode: `console.log('[${template}] ${functionName} called');`,
+			description: `${template} hook for ${functionName}`,
+			parameters: []
+		}];
+		return jsonResponse({
+			success: true,
+			target,
+			template,
+			functionName,
+			script: this.state.hookCodeGenerator.exportScript(templates, "frida")
+		});
+	}
+	async handleFridaEnumerateFunctions(args) {
+		const sessionId = readRequiredString(args, "sessionId");
+		const moduleName = readRequiredString(args, "moduleName");
+		const frida = this.getFridaSession();
+		const availability = await frida.getAvailability();
+		if (!availability.available) return {
+			available: false,
+			sessionId,
+			moduleName,
+			reason: availability.reason ?? "Frida CLI is not available",
+			functions: []
+		};
+		if (!frida.useSession(sessionId)) return {
+			available: false,
+			sessionId,
+			reason: `Unknown Frida session: ${sessionId}`,
+			functions: []
+		};
+		const functions = await frida.enumerateFunctions(moduleName);
+		return {
+			available: true,
+			sessionId,
+			moduleName,
+			functions,
+			count: functions.length
+		};
+	}
+	async handleFridaFindSymbols(args) {
+		const sessionId = readRequiredString(args, "sessionId");
+		const pattern = readRequiredString(args, "pattern");
+		const frida = this.getFridaSession();
+		const availability = await frida.getAvailability();
+		if (!availability.available) return {
+			available: false,
+			sessionId,
+			pattern,
+			reason: availability.reason ?? "Frida CLI is not available",
+			symbols: []
+		};
+		if (!frida.useSession(sessionId)) return {
+			available: false,
+			sessionId,
+			reason: `Unknown Frida session: ${sessionId}`,
+			symbols: []
+		};
+		const symbols = await frida.findSymbols(pattern);
+		return {
+			available: true,
+			sessionId,
+			pattern,
+			symbols,
+			count: symbols.length
+		};
+	}
+	getFridaSession() {
+		if (!this.state.fridaSession) this.state.fridaSession = new FridaSession();
+		return this.state.fridaSession;
+	}
+};
+//#endregion
+//#region src/server/domains/binary-instrument/handlers/analysis-handlers.ts
+/**
+* Static analysis sub-handler — Ghidra, IDA, JADX, Unidbg, hooks, plugins.
+*/
+var AnalysisHandlers = class {
+	state;
+	constructor(state) {
+		this.state = state;
+	}
+	async handleGhidraAnalyze(args) {
+		const legacyTargetPath = readOptionalString(args, "targetPath");
+		if (!readOptionalString(args, "binaryPath") && legacyTargetPath) return invokeLegacyPlugin(this.state.context, "plugin_ghidra_bridge", "ghidra_analyze", args);
+		const binaryPath = readRequiredString(args, "binaryPath");
+		const timeout = readOptionalNumber(args, "timeout");
+		const ghidra = this.getGhidraAnalyzer();
+		const availability = await ghidra.getAvailability();
+		const analysis = await ghidra.analyze(binaryPath, timeout !== void 0 ? { timeout } : void 0);
+		if (!availability.available) return {
+			available: false,
+			binaryPath,
+			reason: availability.reason ?? "Ghidra analyzeHeadless is not available",
+			analysis
+		};
+		return {
+			available: true,
+			binaryPath,
+			analysis
+		};
+	}
+	async handleGhidraDecompile(args) {
+		return invokeLegacyPlugin(this.state.context, "plugin_ghidra_bridge", "ghidra_decompile", args);
+	}
+	async handleIdaDecompile(args) {
+		return invokeLegacyPlugin(this.state.context, "plugin_ida_bridge", "ida_decompile", args);
+	}
+	async handleJadxDecompile(args) {
+		return invokeLegacyPlugin(this.state.context, "plugin_jadx_bridge", "jadx_decompile", args);
+	}
+	async handleGenerateHooks(args) {
+		const legacyGhidraOutput = readOptionalString(args, "ghidraOutput");
+		if (legacyGhidraOutput) return this.handleLegacyGenerateHooks(legacyGhidraOutput);
+		const legacyGhidraOutputObj = args["ghidraOutput"];
+		if (isRecord(legacyGhidraOutputObj)) return this.handleLegacyGenerateHooks(JSON.stringify(legacyGhidraOutputObj));
+		const symbols = readStringArray(args, "symbols");
+		if (symbols.length === 0) return textResponse("symbols or ghidraOutput is required");
+		const options = readHookOptions(args, "options");
+		const script = this.getHookGenerator().generateFridaHookScript(symbols, options);
+		return jsonResponse({
+			available: true,
+			symbolCount: symbols.length,
+			script
+		});
+	}
+	async handleExportHookScript(args) {
+		const rawTemplates = readOptionalString(args, "hookTemplates");
+		if (!rawTemplates) {
+			const generated = this.state.hookCodeGenerator.exportScript([], "frida");
+			return jsonResponse({
+				format: "frida",
+				hookCount: 0,
+				script: generated.includes("Java.perform") ? generated : `Java.perform(function() {\n${generated}\n});`
+			});
+		}
+		try {
+			const parsed = JSON.parse(rawTemplates);
+			if (!Array.isArray(parsed)) return textResponse("Invalid JSON");
+			const templates = toHookTemplates(parsed);
+			const script = this.state.hookCodeGenerator.exportScript(templates, "frida");
+			return jsonResponse({
+				format: "frida",
+				hookCount: templates.length,
+				script
+			});
+		} catch {
+			return textResponse("Invalid JSON");
+		}
+	}
+	async handleUnidbgEmulate(args) {
+		const binaryPath = readRequiredString(args, "binaryPath");
+		const functionName = readRequiredString(args, "functionName");
+		const invokeArgs = readStringArray(args, "args");
+		const availability = await getUnidbgAvailability();
+		if (!availability.available) return {
+			available: false,
+			binaryPath,
+			functionName,
+			args: invokeArgs,
+			reason: availability.reason,
+			result: {
+				returnValue: "0x0",
+				stdout: "",
+				stderr: "",
+				trace: ["mock-unidbg-unavailable"]
+			}
+		};
+		const result = await execFileUtf8(availability.command, [
+			"-jar",
+			availability.jarPath,
+			binaryPath,
+			functionName,
+			...invokeArgs
+		], UNIDBG_TIMEOUT_MS);
+		return {
+			available: true,
+			binaryPath,
+			functionName,
+			args: invokeArgs,
+			result: {
+				returnValue: "0x0",
+				stdout: result.stdout.trim(),
+				stderr: result.stderr.trim(),
+				trace: []
+			}
+		};
+	}
+	async handleUnidbgLaunch(args) {
+		const soPath = readOptionalString(args, "soPath");
+		if (!soPath) return textResponse("Missing required string argument: soPath");
+		const arch = readOptionalString(args, "arch") ?? "arm";
+		try {
+			const result = await this.state.unidbgRunner.launch(soPath, arch);
+			return {
+				available: true,
+				sessionId: result.sessionId,
+				soPath: result.soPath,
+				arch: result.arch,
+				sessions: this.state.unidbgRunner.listSessions()
+			};
+		} catch (error) {
+			return {
+				available: false,
+				soPath,
+				arch,
+				reason: error instanceof Error ? error.message : String(error),
+				sessions: this.state.unidbgRunner.listSessions()
+			};
+		}
+	}
+	async handleUnidbgCall(args) {
+		const sessionId = readOptionalString(args, "sessionId");
+		if (!sessionId) return textResponse("Missing required string argument: sessionId");
+		const functionName = readOptionalString(args, "functionName");
+		if (!functionName) return textResponse("Missing required string argument: functionName");
+		const callArgs = isRecord(args["args"]) ? args["args"] : {};
+		try {
+			return jsonResponse(await this.state.unidbgRunner.callFunction(sessionId, functionName, callArgs));
+		} catch (error) {
+			const message = error instanceof Error ? error.message : String(error);
+			return textResponse(message.startsWith("No unidbg session found") ? `${message} (not found)` : message);
+		}
+	}
+	async handleUnidbgTrace(args) {
+		const sessionId = readOptionalString(args, "sessionId");
+		if (!sessionId) return textResponse("Missing required string argument: sessionId");
+		try {
+			return jsonResponse(await this.state.unidbgRunner.trace(sessionId));
+		} catch (error) {
+			const message = error instanceof Error ? error.message : String(error);
+			return textResponse(message.startsWith("No unidbg session found") ? `${message} (not found)` : message);
+		}
+	}
+	async handleGetAvailablePlugins(_args) {
+		const plugins = this.state.context ? getAvailablePlugins(this.state.context) : [];
+		return jsonResponse({
+			plugins,
+			count: plugins.length
+		});
+	}
+	getGhidraAnalyzer() {
+		if (!this.state.ghidra) this.state.ghidra = new GhidraAnalyzer();
+		return this.state.ghidra;
+	}
+	getHookGenerator() {
+		if (!this.state.hookGen) this.state.hookGen = new HookGenerator();
+		return this.state.hookGen;
+	}
+	handleLegacyGenerateHooks(ghidraOutput) {
+		let parsed;
+		try {
+			parsed = JSON.parse(ghidraOutput);
+		} catch {
+			return textResponse("Invalid JSON");
+		}
+		if (!isGhidraAnalysisOutput(parsed)) return textResponse("ghidraOutput is required");
+		const hooks = this.state.hookCodeGenerator.generateHooks(parsed);
+		return jsonResponse({
+			count: hooks.length,
+			hooks
+		});
+	}
+};
+//#endregion
+//#region src/server/domains/binary-instrument/handlers.impl.core.ts
+/**
+* BinaryInstrument domain — composition facade.
+*
+* Frida operations in ./handlers/frida-handlers.ts.
+* Analysis/unidbg/hook operations in ./handlers/analysis-handlers.ts.
+*/
+var BinaryInstrumentHandlers = class {
+	state;
+	frida;
+	analysis;
+	constructor(first, second, third) {
+		this.state = {
+			hookCodeGenerator: new HookCodeGenerator(),
+			unidbgRunner: new UnidbgRunner()
+		};
+		if (first instanceof FridaSession) this.state.fridaSession = first;
+		else if (isServerContext(first)) this.state.context = first;
+		if (second) this.state.ghidra = second;
+		if (third) this.state.hookGen = third;
+		this.frida = new FridaHandlers(this.state);
+		this.analysis = new AnalysisHandlers(this.state);
+	}
+	handleFridaAttach(args) {
+		return this.frida.handleFridaAttach(args);
+	}
+	handleFridaEnumerateModules(args) {
+		return this.frida.handleFridaEnumerateModules(args);
+	}
+	handleFridaRunScript(args) {
+		return this.frida.handleFridaRunScript(args);
+	}
+	handleFridaDetach(args) {
+		return this.frida.handleFridaDetach(args);
+	}
+	handleFridaListSessions(args) {
+		return this.frida.handleFridaListSessions(args);
+	}
+	handleFridaGenerateScript(args) {
+		return this.frida.handleFridaGenerateScript(args);
+	}
+	handleFridaEnumerateFunctions(args) {
+		return this.frida.handleFridaEnumerateFunctions(args);
+	}
+	handleFridaFindSymbols(args) {
+		return this.frida.handleFridaFindSymbols(args);
+	}
+	handleGhidraAnalyze(args) {
+		return this.analysis.handleGhidraAnalyze(args);
+	}
+	handleGhidraDecompile(args) {
+		return this.analysis.handleGhidraDecompile(args);
+	}
+	handleIdaDecompile(args) {
+		return this.analysis.handleIdaDecompile(args);
+	}
+	handleJadxDecompile(args) {
+		return this.analysis.handleJadxDecompile(args);
+	}
+	handleGenerateHooks(args) {
+		return this.analysis.handleGenerateHooks(args);
+	}
+	handleExportHookScript(args) {
+		return this.analysis.handleExportHookScript(args);
+	}
+	handleUnidbgEmulate(args) {
+		return this.analysis.handleUnidbgEmulate(args);
+	}
+	handleUnidbgLaunch(args) {
+		return this.analysis.handleUnidbgLaunch(args);
+	}
+	handleUnidbgCall(args) {
+		return this.analysis.handleUnidbgCall(args);
+	}
+	handleUnidbgTrace(args) {
+		return this.analysis.handleUnidbgTrace(args);
+	}
+	handleGetAvailablePlugins(args) {
+		return this.analysis.handleGetAvailablePlugins(args);
+	}
+};
+//#endregion
+export { BinaryInstrumentHandlers };