bitbank-lab-mcp 0.2.0 → 0.2.1

package/lib/validate.ts

@@ -1,3 +1,5 @@
+import { realpathSync } from 'node:fs';
+import { basename, delimiter, dirname, join, resolve, sep } from 'node:path';
 import type { Pair } from '../src/schemas.js';
 import { nowIso } from './datetime.js';
 
@@ -148,6 +150,74 @@ export function validateDate(
 	return { ok: true, value: date };
 }
 
+/** チャートファイルの既定出力ディレクトリ（Claude.ai 環境の出力先） */
+export const DEFAULT_CHART_OUTPUT_DIR = '/mnt/user-data/outputs';
+
+/** チャート出力の許可 root を運用側が追加する環境変数（path.delimiter 区切り） */
+export const OUTPUT_DIR_ALLOWLIST_ENV = 'BACKTEST_OUTPUT_DIR_ALLOWLIST';
+
+/**
+ * チャート出力先として許可される root ディレクトリ一覧。
+ * 既定: DEFAULT_CHART_OUTPUT_DIR とサーバー作業ディレクトリ配下
+ * （相対パス指定・Cursor 等でワークスペース直下に書き出すユースケース用）。
+ * それ以外は運用側が環境変数で明示的に許可する（LLM 入力からは追加できない）。
+ */
+export function allowedOutputRoots(): string[] {
+	const extra = (process.env[OUTPUT_DIR_ALLOWLIST_ENV] ?? '')
+		.split(delimiter)
+		.map((p) => p.trim())
+		.filter((p) => p.length > 0)
+		.map((p) => resolve(p));
+	return [resolve(DEFAULT_CHART_OUTPUT_DIR), process.cwd(), ...extra];
+}
+
+/**
+ * パスを実パスに正規化する。実在する最深の祖先を realpathSync で解決し、
+ * 未作成の末尾セグメントは字句のまま結合する（mkdir 前の検証で使うため）。
+ * 許可 root 配下に置かれた symlink を経由した外部への書き込みを防ぐ。
+ */
+function canonicalizePath(p: string): string {
+	let current = resolve(p);
+	const pending: string[] = [];
+	for (;;) {
+		try {
+			return join(realpathSync(current), ...pending);
+		} catch {
+			const parent = dirname(current);
+			if (parent === current) return join(current, ...pending);
+			pending.unshift(basename(current));
+			current = parent;
+		}
+	}
+}
+
+/**
+ * outputDir が許可 root 配下かを検証する。
+ * LLM 入力（プロンプトインジェクション含む）経由でプロセス権限内の任意パスへ
+ * ディレクトリ作成・ファイル書き込みされるのを防ぐ。`..` を含むパスも
+ * symlink も実パスに解決してから判定するため、トラバーサル・symlink では
+ * 迂回できない（検証と書き込みの間に symlink を差し替える TOCTOU は、
+ * それができる時点でローカルアクセスを持つため脅威モデル外）。
+ */
+export function ensureAllowedOutputDir(
+	dir: string,
+): { ok: true; dir: string } | { ok: false; error: { type: 'user' | 'internal'; message: string } } {
+	const canonical = canonicalizePath(dir);
+	const roots = allowedOutputRoots();
+	const canonicalRoots = roots.map(canonicalizePath);
+	const allowed = canonicalRoots.some((root) => canonical === root || canonical.startsWith(root + sep));
+	if (!allowed) {
+		return {
+			ok: false,
+			error: {
+				type: 'user',
+				message: `outputDir '${dir}' は許可されていません。許可 root: ${roots.join(', ')} 配下のみ（追加は環境変数 ${OUTPUT_DIR_ALLOWLIST_ENV} で指定）`,
+			},
+		};
+	}
+	return { ok: true, dir: canonical };
+}
+
 export function createMeta(
 	pair: Pair,
 	additional: Record<string, unknown> = {},

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bitbank-lab-mcp",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Experimental MCP server for bitbank — botters lab community edition. Market data, technical analysis, charting, and Private API trading tools.",
   "type": "module",
   "bin": {

package/src/schema/backtest.ts

@@ -60,7 +60,15 @@ export const RunBacktestInputSchema = z
 				'One-way fee in basis points. When omitted, resolved dynamically from the current /spot/pairs taker rate (falls back to nominal 12 bp if unavailable). Explicit values are always respected.',
 			),
 		execution: z.literal('t+1_open').optional().default('t+1_open').describe('Execution timing (fixed: t+1_open)'),
-		outputDir: z.string().optional().default('/mnt/user-data/outputs').describe('Output directory for chart files'),
+		outputDir: z
+			.string()
+			.optional()
+			.default('/mnt/user-data/outputs')
+			.describe(
+				'Output directory for chart files. Must be under an allowed root: ' +
+					'/mnt/user-data/outputs, the server working directory, or a directory listed in ' +
+					'the BACKTEST_OUTPUT_DIR_ALLOWLIST environment variable.',
+			),
 		savePng: z
 			.boolean()
 			.optional()

package/tools/trading_process/lib/svg_to_png.ts

@@ -49,6 +49,15 @@ export async function svgToPng(
 	return outputPath;
 }
 
+/**
+ * ファイル名に埋め込む値からパス区切り・ドット等を除去する。
+ * pair は上流（ensurePair）で検証済みだが、ファイル名の安全性を
+ * 上流バリデーションに依存させないための防御的サニタイズ。
+ */
+function sanitizeFilenamePart(part: string): string {
+	return part.replace(/[^a-zA-Z0-9_-]/g, '');
+}
+
 /**
  * Generate a unique filename for backtest chart
  */
@@ -59,6 +68,6 @@ export function generateBacktestChartFilename(
 	format: 'png' | 'svg' = 'png',
 ): string {
 	const timestamp = dayjs().format('YYYY-MM-DDTHH-mm-ss');
-	const safePair = pair.replace('_', '');
-	return `backtest_${safePair}_${timeframe}_${strategy}_${timestamp}.${format}`;
+	const safePair = sanitizeFilenamePart(pair.replace('_', ''));
+	return `backtest_${safePair}_${sanitizeFilenamePart(timeframe)}_${sanitizeFilenamePart(strategy)}_${timestamp}.${format}`;
 }

package/tools/trading_process/run_backtest.ts

@@ -7,6 +7,7 @@
 import { existsSync, mkdirSync } from 'node:fs';
 import { join } from 'node:path';
 import { dayjs, formatDateInTz } from '../../lib/datetime.js';
+import { DEFAULT_CHART_OUTPUT_DIR, ensureAllowedOutputDir } from '../../lib/validate.js';
 import { prependWarnings } from '../../lib/warning-propagation.js';
 import { type BacktestEngineInput, type BacktestEngineResult, runBacktestEngine } from './lib/backtest_engine.js';
 import { fetchCandlesForBacktest } from './lib/fetch_candles.js';
@@ -21,7 +22,7 @@ import {
 import type { BacktestRange, Period, Timeframe } from './types.js';
 
 // Claude.ai のデフォルト出力ディレクトリ
-const DEFAULT_OUTPUT_DIR = '/mnt/user-data/outputs';
+const DEFAULT_OUTPUT_DIR = DEFAULT_CHART_OUTPUT_DIR;
 
 /**
  * 書き込み可能なディレクトリを確保
@@ -48,7 +49,7 @@ export interface RunBacktestInput {
 	strategy: StrategyConfig;
 	fee_bp?: number;
 	execution?: 't+1_open';
-	/** 出力ディレクトリ（デフォルト: /mnt/user-data/outputs/） */
+	/** 出力ディレクトリ（デフォルト: /mnt/user-data/outputs/）。許可 root 配下のみ（lib/validate.ts の ensureAllowedOutputDir） */
 	outputDir?: string;
 	/** PNG ファイルを生成する（デフォルト: false、ファイルシステム非共有のため） */
 	savePng?: boolean;
@@ -118,6 +119,17 @@ export default async function runBacktest(input: RunBacktestInput): Promise<RunB
 		}
 		const params = validation.normalizedParams;
 
+		// 出力ディレクトリをバリデーション（savePng 時のみ・fetch 前に弾く）。
+		// LLM 指定の任意パスへの書き込みを許可 root 配下に制限する。
+		let resolvedOutputDir: string | undefined;
+		if (savePng) {
+			const dirCheck = ensureAllowedOutputDir(outputDir);
+			if (!dirCheck.ok) {
+				return { ok: false, error: dirCheck.error.message };
+			}
+			resolvedOutputDir = dirCheck.dir;
+		}
+
 		// 必要なバー数を params から動的に計算（ユーザーが long / slow / signal 等を
 		// デフォルトより大きく指定した場合でもウォームアップを十分確保するため）
 		const requiredBars = strategy.computeRequiredBars(params);
@@ -214,11 +226,11 @@ export default async function runBacktest(input: RunBacktestInput): Promise<RunB
 		};
 
 		// PNG ファイル保存（savePng: true の場合のみ）
-		if (savePng && svg) {
+		if (savePng && svg && resolvedOutputDir) {
 			try {
-				ensureOutputDir(outputDir);
+				ensureOutputDir(resolvedOutputDir);
 				const filename = generateBacktestChartFilename(pair, timeframe, strategyConfig.type, 'png');
-				const pngPath = join(outputDir, filename);
+				const pngPath = join(resolvedOutputDir, filename);
 				await svgToPng(svg, pngPath, { density: 150 });
 				output.chartPath = pngPath;
 			} catch (pngError) {