monster-orchestrator 3.4.0 → 3.7.6

package/scripts/tmux-guard.mjs

@@ -0,0 +1,396 @@
+#!/usr/bin/env node
+// 席の PreToolUse(Bash) hook（箱 n=2320・n=2311 の直し）。席が艦隊の tmux サーバを消す命令を撃つのを止める。
+//
+// 🔬 なぜ要るか（2026-10-02 17:54 実害）: 席が試験で `TMUX_TMPDIR=… tmux kill-server` を撃ち、
+// **$TMUX が TMUX_TMPDIR より優先されて**艦隊サーバに当たり 13 席が同時に落ちた（2026-09-25 に続き 2 回目）。
+// tmux は -S／-L が無いと $TMUX の socket を、$TMUX も無ければ `${TMUX_TMPDIR:-/tmp}/tmux-<uid>/default` を使う。
+//
+// 止める物: ① 行き先（-S／-L）の無い tmux kill-server（`tmux-gui.sh kill-server` を含む）
+//           ② pkill／killall の tmux
+//           ③ 行き先を明示していても、それが席自身のサーバ（$TMUX の 1 欄目）か既定サーバを指す kill-server
+//           （行き先が変数などで読めないときも止める＝fail-safe。消すのは使い捨てなので字で書ける）
+// 通す物: それ以外の全部（他のサーバを指す kill-server・kill-session 等の他の tmux 命令・tmux を含まない命令）
+//
+// 入出力（Claude Code の PreToolUse 規約）: stdin の JSON（tool_name／tool_input.command／cwd）を読み、
+// 止めるときは exit 2＋stderr に理由。それ以外は exit 0。読めない・壊れた入力も exit 0（fail-open・席を止めない）。
+// ⛔ python にしない（会員の機体に python3 がある保証は無い・node は orchestrator が必ず持つ）。
+
+import * as fs from "node:fs";
+import * as path from "node:path";
+import { fileURLToPath } from "node:url";
+
+const GUIDE =
+  "使い捨ての tmux は `env -u TMUX tmux -S /tmp/<短い名>.sock …` で立て、同じ -S で消せ。手順: {CTOリポ}/05手順/使い捨てのtmuxの立て方.md";
+
+const SHELLS = new Set(["bash", "sh", "zsh", "dash", "ksh"]);
+// tmux の大域オプションのうち引数を取るもの（tmux 3.x: -c shell-command -f file -L socket-name -S socket-path -T features）
+const TMUX_OPTS_WITH_ARG = new Set(["c", "f", "L", "S", "T"]);
+
+/**
+ * シェルの 1 行を「文（単語の列）」に割る。引用（' " \）を解き、; & | && || 改行 ( ) で文を切る。
+ * `$(…)` と `` `…` `` の中身は別の命令として nested に返す（引用 " の中でも実行されるので拾う。' の中は拾わない）。
+ * 閉じていない引用は残りを字として扱う（bash はその行を実行しないので、ここで止める理由は無い）。
+ * @param {string} src
+ * @returns {{ statements: string[][]; nested: string[] }}
+ */
+export function splitShell(src) {
+  /** @type {string[][]} */
+  const statements = [];
+  /** @type {string[]} */
+  const nested = [];
+  /** @type {string[]} */
+  let words = [];
+  let word = "";
+  let inWord = false;
+  const endWord = () => {
+    if (inWord) words.push(word);
+    word = "";
+    inWord = false;
+  };
+  const endStatement = () => {
+    endWord();
+    if (words.length > 0) statements.push(words);
+    words = [];
+  };
+  // `$(` の直後から対応する `)` までを返す（引用と入れ子を数える）
+  const readSubst = (start) => {
+    let depth = 1;
+    let j = start;
+    let q = "";
+    for (; j < src.length; j++) {
+      const c = src[j];
+      if (q === "'") {
+        if (c === "'") q = "";
+      } else if (q === '"') {
+        if (c === "\\") j++;
+        else if (c === '"') q = "";
+      } else if (c === "\\") j++;
+      else if (c === "'" || c === '"') q = c;
+      else if (c === "(") depth++;
+      else if (c === ")" && --depth === 0) break;
+    }
+    return { body: src.slice(start, j), end: j };
+  };
+  const readBacktick = (start) => {
+    let j = start;
+    while (j < src.length && src[j] !== "`") {
+      if (src[j] === "\\") j++;
+      j++;
+    }
+    return { body: src.slice(start, j), end: j };
+  };
+
+  for (let i = 0; i < src.length; i++) {
+    const c = src[i];
+    if (c === "\\") {
+      if (src[i + 1] === "\n") {
+        i++; // 行の継続
+        continue;
+      }
+      word += src[i + 1] ?? "";
+      inWord = true;
+      i++;
+    } else if (c === "'") {
+      const j = src.indexOf("'", i + 1);
+      const end = j === -1 ? src.length : j;
+      word += src.slice(i + 1, end);
+      inWord = true;
+      i = end;
+    } else if (c === '"') {
+      inWord = true;
+      let j = i + 1;
+      for (; j < src.length && src[j] !== '"'; j++) {
+        if (src[j] === "\\" && j + 1 < src.length && '"\\$`\n'.includes(src[j + 1])) {
+          if (src[j + 1] !== "\n") word += src[j + 1];
+          j++;
+        } else if (src[j] === "$" && src[j + 1] === "(") {
+          const s = readSubst(j + 2);
+          nested.push(s.body);
+          word += "$(" + s.body + ")";
+          j = s.end;
+        } else if (src[j] === "`") {
+          const s = readBacktick(j + 1);
+          nested.push(s.body);
+          word += "`" + s.body + "`";
+          j = s.end;
+        } else word += src[j];
+      }
+      i = j;
+    } else if (c === "$" && src[i + 1] === "(") {
+      const s = readSubst(i + 2);
+      nested.push(s.body);
+      word += "$(" + s.body + ")";
+      inWord = true;
+      i = s.end;
+    } else if (c === "`") {
+      const s = readBacktick(i + 1);
+      nested.push(s.body);
+      word += "`" + s.body + "`";
+      inWord = true;
+      i = s.end;
+    } else if (c === ";" || c === "&" || c === "|" || c === "\n" || c === "(" || c === ")") {
+      endStatement();
+    } else if (c === " " || c === "\t") {
+      endWord();
+    } else if (c === "#" && !inWord) {
+      // 単語の頭の # から行末まではコメント
+      const j = src.indexOf("\n", i);
+      i = j === -1 ? src.length : j - 1;
+    } else {
+      word += c;
+      inWord = true;
+    }
+  }
+  endStatement();
+  return { statements, nested };
+}
+
+const ASSIGN = /^([A-Za-z_][A-Za-z0-9_]*)=(.*)$/s;
+
+/** `$VAR`・`${VAR}`・頭の `~` を展開する。展開できない物（未定義の変数・$(…)・`…`）が残れば null。 */
+function expand(value, vars, env) {
+  if (/`|\$\(/.test(value)) return null;
+  let unresolved = false;
+  let out = value.replace(/\$\{([A-Za-z_][A-Za-z0-9_]*)\}|\$([A-Za-z_][A-Za-z0-9_]*)/g, (_m, a, b) => {
+    const name = a ?? b;
+    const v = name in vars ? vars[name] : env[name];
+    if (v === undefined) {
+      unresolved = true;
+      return "";
+    }
+    return v;
+  });
+  if (unresolved || out.includes("$")) return null;
+  if (out === "~" || out.startsWith("~/")) {
+    if (!env.HOME) return null;
+    out = env.HOME + out.slice(1);
+  }
+  return out;
+}
+
+/** socket の道を比べられる形にする（macOS の /tmp → /private/tmp を realpath で揃える）。 */
+function canonical(p, cwd, realpath) {
+  const abs = path.resolve(cwd, p);
+  const dir = path.dirname(abs);
+  let real = dir;
+  try {
+    real = realpath(dir);
+  } catch {
+    // 無い道は字のまま比べる
+  }
+  return path.join(real, path.basename(abs));
+}
+
+/**
+ * tmux 呼び出しの引数から、kill-server を撃つか・行き先は何かを読む。
+ * tmux は `\;` で命令を続けられるので、全ての命令を見る。命令名は一意な前方一致も効く（kill-ser 以上）。
+ * @returns {{ kills: boolean; socket?: string; label?: string }}
+ */
+export function readTmuxArgs(args) {
+  /** @type {{ kills: boolean; socket?: string; label?: string }} */
+  const r = { kills: false };
+  let i = 0;
+  for (; i < args.length; i++) {
+    const a = args[i];
+    if (a === "--") {
+      i++;
+      break;
+    }
+    if (!a.startsWith("-") || a === "-") break;
+    // 束ねたフラグ（-2u）と、引数つきの直付け（-L名・-S/道）を読む
+    for (let k = 1; k < a.length; k++) {
+      const f = a[k];
+      if (TMUX_OPTS_WITH_ARG.has(f)) {
+        const val = k + 1 < a.length ? a.slice(k + 1) : args[++i];
+        if (f === "S") r.socket = val ?? "";
+        if (f === "L") r.label = val ?? "";
+        break;
+      }
+    }
+  }
+  let atCommand = true;
+  for (; i < args.length; i++) {
+    let a = args[i];
+    const ends = a.length > 1 && a.endsWith(";");
+    if (a === ";") {
+      atCommand = true;
+      continue;
+    }
+    if (ends) a = a.slice(0, -1);
+    if (atCommand && a.length >= "kill-ser".length && "kill-server".startsWith(a)) r.kills = true;
+    atCommand = ends;
+  }
+  return r;
+}
+
+/**
+ * Bash の命令 1 本を判定する。止めるなら理由（日本語 1 文）、通すなら null。
+ * @param {string} command
+ * @param {{ env?: Record<string, string | undefined>; cwd?: string; uid?: number; realpath?: (p: string) => string }} [ctx]
+ * @returns {string | null}
+ */
+export function verdict(command, ctx = {}) {
+  const env = ctx.env ?? process.env;
+  const cwd = ctx.cwd ?? process.cwd();
+  const uid = ctx.uid ?? (typeof process.getuid === "function" ? process.getuid() : 0);
+  const realpath = ctx.realpath ?? ((p) => fs.realpathSync.native(p));
+  return judge(command, { env, cwd, uid, realpath, vars: {} }, 0);
+}
+
+function judge(command, c, depth) {
+  if (typeof command !== "string" || command.length === 0 || depth > 4) return null;
+  const { statements, nested } = splitShell(command);
+  for (const sub of nested) {
+    const why = judge(sub, c, depth + 1);
+    if (why) return why;
+  }
+  for (const words of statements) {
+    const why = judgeStatement(words, c, depth);
+    if (why) return why;
+  }
+  return null;
+}
+
+function judgeStatement(words, c, depth) {
+  // 頭の代入（`X=1 cmd` の一時代入と、代入だけの文＝以後の行で効くシェル変数）
+  /** @type {Record<string, string>} */
+  const prefix = {};
+  let k = 0;
+  while (k < words.length && ASSIGN.test(words[k])) {
+    const [, name, value] = /** @type {RegExpMatchArray} */ (words[k].match(ASSIGN));
+    prefix[name] = expand(value, c.vars, c.env) ?? "";
+    k++;
+  }
+  if (k === words.length) {
+    Object.assign(c.vars, prefix);
+    return null;
+  }
+  if (words[k] === "export") {
+    for (const w of words.slice(k + 1)) {
+      const m = w.match(ASSIGN);
+      if (m) c.vars[m[1]] = expand(m[2], c.vars, c.env) ?? "";
+    }
+    return null;
+  }
+
+  const base = words.map((w) => path.basename(w));
+
+  // ② pkill／killall の tmux
+  const killer = base.findIndex((b) => b === "pkill" || b === "killall");
+  if (killer !== -1 && words.slice(killer + 1).some((w) => w.includes("tmux"))) {
+    return "pkill／killall の tmux は艦隊の tmux サーバを巻き込む";
+  }
+
+  // bash -c '…'・eval … の中身も命令として見る
+  const sh = base.findIndex((b) => SHELLS.has(b));
+  if (sh !== -1) {
+    const ci = words.indexOf("-c", sh + 1);
+    if (ci !== -1 && words[ci + 1] !== undefined) {
+      const why = judge(words[ci + 1], c, depth + 1);
+      if (why) return why;
+    }
+  }
+  if (base[k] === "eval") {
+    const why = judge(words.slice(k + 1).join(" "), c, depth + 1);
+    if (why) return why;
+  }
+
+  const t = base.findIndex((b) => b === "tmux" || b === "tmux-gui.sh");
+  if (t === -1) return null;
+  const r = readTmuxArgs(words.slice(t + 1));
+  if (!r.kills) return null;
+  if (base[t] === "tmux-gui.sh") {
+    return "tmux-gui.sh kill-server は艦隊の tmux サーバそのものを消す";
+  }
+
+  // この呼び出しで効く環境（`env -u TMUX`・一時代入・export を重ねる）
+  /** @type {Record<string, string | undefined>} */
+  const eff = { ...c.env, ...c.vars, ...prefix };
+  for (let j = k; j < t; j++) {
+    if (base[j] === "env") {
+      for (let m = j + 1; m < t; m++) {
+        if (words[m] === "-u" && words[m + 1] !== undefined) eff[words[++m]] = undefined;
+        else if (words[m].startsWith("--unset=")) eff[words[m].slice(8)] = undefined;
+        else if (ASSIGN.test(words[m])) {
+          const [, n, v] = /** @type {RegExpMatchArray} */ (words[m].match(ASSIGN));
+          eff[n] = expand(v, c.vars, c.env) ?? "";
+        }
+      }
+    }
+  }
+
+  if (r.socket === undefined && r.label === undefined) {
+    return "行き先（-S <socket> か -L <名>）の無い tmux kill-server（$TMUX が在ると TMUX_TMPDIR は効かず席自身のサーバに当たる・無ければ既定サーバに当たる）";
+  }
+
+  // ③ 行き先が席自身のサーバ／既定サーバか
+  const tmpdirOf = (e) => (e.TMUX_TMPDIR && e.TMUX_TMPDIR.length > 0 ? e.TMUX_TMPDIR : "/tmp");
+  let target;
+  if (r.socket !== undefined) {
+    const p = expand(r.socket, c.vars, c.env);
+    if (p === null || p === "") return "kill-server の -S の行き先が読めない（変数など）。使い捨ての socket は字で書け";
+    target = canonical(p, c.cwd, c.realpath);
+  } else {
+    const name = expand(/** @type {string} */ (r.label), c.vars, c.env);
+    if (name === null || name === "") return "kill-server の -L の名が読めない（変数など）。使い捨ての socket は -S と字で書け";
+    target = canonical(path.join(tmpdirOf(eff), `tmux-${c.uid}`, name), c.cwd, c.realpath);
+  }
+  /** @type {string[]} */
+  const protectedSockets = [];
+  const seat = (c.env.TMUX ?? "").split(",")[0];
+  if (seat) protectedSockets.push(canonical(seat, c.cwd, c.realpath));
+  protectedSockets.push(canonical(path.join(tmpdirOf(c.env), `tmux-${c.uid}`, "default"), c.cwd, c.realpath));
+  if (protectedSockets.includes(target)) {
+    return `kill-server の行き先（${target}）が席自身の tmux サーバ（艦隊）を指している`;
+  }
+  return null;
+}
+
+/** 拒否文。 */
+export function denyMessage(reason) {
+  return `⛔ tmux-guard（n=2320）: ${reason}。${GUIDE}`;
+}
+
+/**
+ * hook の本体。stdin の文字列を受けて終了コードを返す（0＝通す／2＝止める）。
+ * @param {string} raw
+ * @param {{ env?: Record<string, string | undefined>; uid?: number; realpath?: (p: string) => string; stderr?: (s: string) => void }} [ctx]
+ */
+export function runHook(raw, ctx = {}) {
+  let reason = null;
+  try {
+    const data = JSON.parse(raw);
+    if (!data || data.tool_name !== "Bash") return 0;
+    const command = data.tool_input?.command;
+    if (typeof command !== "string") return 0;
+    reason = verdict(command, {
+      env: ctx.env,
+      cwd: typeof data.cwd === "string" && data.cwd ? data.cwd : undefined,
+      uid: ctx.uid,
+      realpath: ctx.realpath,
+    });
+  } catch {
+    return 0; // 読めない入力で席を止めない
+  }
+  if (!reason) return 0;
+  (ctx.stderr ?? ((s) => process.stderr.write(s)))(denyMessage(reason) + "\n");
+  return 2;
+}
+
+function isMain() {
+  try {
+    return !!process.argv[1] && fs.realpathSync(process.argv[1]) === fs.realpathSync(fileURLToPath(import.meta.url));
+  } catch {
+    return false;
+  }
+}
+
+if (isMain()) {
+  let raw = "";
+  try {
+    raw = fs.readFileSync(0, "utf8");
+  } catch {
+    process.exit(0);
+  }
+  process.exit(runHook(raw));
+}