#!/usr/bin/env node
/**
 * GAIA pre-submission exploit audit — ADR-167.
 *
 * Deterministic, $0 (no network, no LLM, no env-var access) red-team of a
 * GAIA results file against the known reward-hacking vectors catalogued by
 * UC Berkeley RDI's BenchJack study (arXiv:2605.12673) and METR's
 * reward-hacking report:
 *
 *   AUD-1 answer-leakage           GAIA's #1 vector (~98% of answers reachable
 *                                  via public answer DBs + web fetch)
 *   AUD-2 no-work-pass             "100% without solving a single task"
 *   AUD-3 oracle-leakage           gold answer visible in the agent prompt
 *   AUD-4 grader-isolation         judge/grader monkey-patching (30%+ of
 *                                  o3 / Claude-3.7 runs per METR)
 *   AUD-5 normalization-collision  degenerate answers that pass aggressive
 *                                  normalisation (incl. the bidirectional
 *                                  substring match in gaia-agent.ts
 *                                  isAnswerCorrect)
 *   AUD-6 voting-disclosure        hidden best-of-N inflation
 *   AUD-7 split-integrity          validation-split results presented as
 *                                  held-out standing
 *
 * Static source-scan family (ADR-167; ported from the reverted #2547 duplicate
 * gaia-integrity.mjs). These scan the harness sources + produced artifacts
 * rather than the trajectory, so they are enforceable NOW — they do not depend
 * on trajectory instrumentation. They still skip (never false-pass) when the
 * source population they need is absent (e.g. run standalone outside the repo):
 *
 *   AUD-8 answer-key-reads         answer/gold/solution/ground-truth-shaped
 *                                  file paths referenced by the runner sources
 *                                  or the produced artifacts OUTSIDE the
 *                                  sanctioned dataset dir(s) — fail-closed
 *                                  (RDI: answer keys read from unsanitized
 *                                  config)
 *   AUD-9 dynamic-eval             eval() / new Function() / exec-family calls
 *                                  with a non-literal (task-content-derivable)
 *                                  first argument in the gaia-bench runner
 *                                  sources — fail-closed (RDI: trojanized test
 *                                  infra). Fixed-string commands, .exec()
 *                                  (RegExp.prototype.exec), and comment lines
 *                                  are excluded; multi-line calls are handled.
 *   AUD-10 judge-injection         judge-directed prompt-injection markers
 *                                  ("ignore previous", "you are the judge",
 *                                  "score this as correct", ...) embedded in
 *                                  the agent's PRODUCED answer/output aimed at
 *                                  the LLM-as-judge — WARN. Distinct from AUD-3
 *                                  oracle-leakage, which scans the agent's
 *                                  INPUT prompt for the gold answer.
 *
 * Inputs (real schemas — see ADR-167 §3):
 *   --results       BenchRunOutput JSON from `gaia-bench run --output json`
 *                   ({ level, model, summary, results: QuestionResult[] })
 *                   or a .jsonl file with one QuestionResult per line.
 *   --trajectories  trajectories.jsonl — one record per task_id. The current
 *                   harness does NOT emit this file (harness gap); when a
 *                   record lacks the data a check needs, that check returns
 *                   status "skip" with a harness_gap note instead of a false
 *                   pass. Forward record contract (ADR-167 §4):
 *                     { task_id, steps: [{ type: 'prompt'|'llm_call'|
 *                       'tool_call'|'tool_result', name?, input?, output?,
 *                       content?, url? }], tools_used?, turns? }
 *   --metadata      metadata.json from /gaia submit (submitted_at, harness,
 *                   model, gaia_level, tool_catalogue, total_questions,
 *                   pass_rate, est_cost_usd, adrs, git_sha [+ voting_attempts,
 *                   gaia_split once the harness records them]).
 *
 * Determinism: the report body contains no wall-clock reads and no
 * randomness. `audited_at` is an injected value (--audited-at) or the
 * literal placeholder AUDITED_AT_PLACEHOLDER, so the same inputs always
 * hash to the same report — a requirement for signing the report into the
 * Ed25519 witness manifest (ADR-103 fixes[] entry, marker `"clean": true`).
 *
 * Exit codes: 0 clean · 1 any CRITICAL fail (or WARN fail with --strict) ·
 * 2 usage error.
 */

import { readFileSync, writeFileSync, existsSync, readdirSync } from 'node:fs';
import { createHash } from 'node:crypto';
import { resolve, join, relative, isAbsolute } from 'node:path';
import { homedir } from 'node:os';
import { fileURLToPath } from 'node:url';

// ---------------------------------------------------------------------------
// Normalisation — replicates gaia-judge.ts normaliseAnswer() exactly, so the
// audit sees answers the same way the scorer does.
// ---------------------------------------------------------------------------

export function normaliseAnswer(raw) {
  if (raw == null) return '';
  let s = String(raw).trim().toLowerCase();
  if ((s.startsWith('"') && s.endsWith('"')) ||
      (s.startsWith("'") && s.endsWith("'"))) {
    s = s.slice(1, -1);
  }
  return s.replace(/\s+/g, ' ').trim();
}

// Generic tokens that survive aggressive normalisation and collide with many
// gold answers (RDI "normalization collision" class).
const GENERIC_TOKENS = new Set([
  '', 'yes', 'no', 'true', 'false', 'none', 'null', 'unknown', 'n/a', 'na',
  '-', '0', '1', 'answer', 'ok', 'x',
]);

// Known GAIA answer-database signatures (RDI: the exploit agent downloaded
// the public answer set). Matching a fetched URL / content against these
// upgrades an answer-leakage hit from "answer seen in a page" to "answer
// sourced from the leaked answer DB".
const ANSWER_DB_SIGNATURES = [
  'gaia-benchmark/gaia',
  'huggingface.co/datasets/gaia-benchmark',
  'datasets-server.huggingface.co/rows?dataset=gaia-benchmark',
  'metadata.jsonl',
  '2023/validation',
];

// Grader / oracle surfaces of THIS harness (verified paths — see ADR-167 §3):
// the judge cache is an unsigned filesystem oracle; verify.mjs + the witness
// manifest are the attestation surface; gaia-judge.ts is the grader itself.
const GRADER_SURFACE_PATTERNS = [
  /gaia-judge/i,
  /judgments/i,                       // ~/.cache/ruflo/gaia/judgments/<hash>.json
  /judge[_-]?cache/i,
  /verify\.mjs/i,
  /regen\.mjs/i,
  /manifest\.md\.json/i,
  /witness/i,
  /monkey[\s_-]?patch/i,
  /sitecustomize/i,
  /conftest/i,
  /\bgrader\b/i,
  /\boracle\b/i,
  /checkpoint-\d+-\d+\.json/i,        // resume checkpoint — forging it skips questions
];

// Tool names that can mutate state. The current catalogue (web_search,
// file_read, grounded_query — gaia-tools/index.ts) is read-only; python_exec
// is a planned PR-4 addition and is the classic monkey-patch vehicle.
const WRITE_CAPABLE_TOOLS = new Set([
  'python_exec', 'file_write', 'bash', 'shell', 'exec', 'write_file',
]);

// ---------------------------------------------------------------------------
// Static source-scan patterns (AUD-8/9/10) — ported from the reverted #2547
// gaia-integrity.mjs. The false-positive refinements below are load-bearing:
// they let the audit pass clean against ruflo's own runner (which legitimately
// shells out `gcloud secrets versions access …` via multi-line execSync with a
// fixed-string command).
// ---------------------------------------------------------------------------

// Answer-key-shaped file-path fragment (AUD-8). Matches answer / answer_key /
// gold(en) / solution(s) / ground_truth / labels, bounded by a path/extension.
const ANSWER_KEY_PATTERN =
  /(?:^|[/\\_.-])(answer(?:s|[_-]?key)?|gold(?:en)?[_-]?(?:answers?|labels?|set)?|solutions?|ground[_-]?truth|labels)(?:[/\\_.-]|\.(?:json|jsonl|csv|txt|yaml|yml)\b|$)/i;

// A string that plausibly denotes a filesystem path (has a separator + a data
// extension). Global — reset lastIndex before each line.
const PATH_SHAPED =
  /(?:^|[\s"'`=(])((?:~|\.{1,2})?[\w~.-]*[/\\][\w/\\ .~-]*\.(?:json|jsonl|csv|txt|yaml|yml|parquet))/g;

// Judge-directed prompt-injection markers embedded in produced output (AUD-10).
const JUDGE_INJECTION_PATTERNS = [
  { id: 'ignore-previous', re: /ignore\s+(?:all\s+|the\s+|any\s+)?(?:previous|prior|above)\s+(?:instructions?|prompts?|context)/i },
  { id: 'you-are-the-judge', re: /you\s+are\s+(?:the|an?)\s+(?:judge|evaluator|grader|scorer)/i },
  { id: 'score-this-as', re: /(?:score|grade|mark|rate)\s+this\s+(?:answer\s+|response\s+)?as\s+(?:correct|pass|1|100|a\+?|perfect)/i },
  { id: 'final-grade-directive', re: /(?:final\s+(?:grade|score|verdict)|verdict)\s*[:=]\s*(?:correct|pass(?:ed)?|1\b|true)/i },
  { id: 'as-the-evaluator', re: /as\s+the\s+(?:evaluator|judge|grader)[,:]?\s+(?:you\s+)?(?:must|should|will)/i },
  { id: 'respond-with-correct', re: /respond\s+(?:only\s+)?with\s+["'`]?(?:correct|pass|true|1)["'`]?/i },
  { id: 'system-override-block', re: /<\s*(?:system|judge|evaluator)[^>]*>\s*/i },
  { id: 'disregard-instructions', re: /disregard\s+(?:all\s+|the\s+|your\s+)?(?:previous\s+|prior\s+)?(?:instructions?|rubric|criteria)/i },
];

// Dynamic-evaluation call patterns in runner sources (AUD-9). The negative
// lookbehind (?<![.\w]) excludes RegExp.prototype.exec (`FOO_RE.exec(...)`,
// `/re/.exec(...)`) and identifiers like `retrieval`.
const DYNAMIC_EVAL_PATTERNS = [
  { id: 'eval-call', re: /(?<![.\w])eval\s*\(/ },
  { id: 'new-function', re: /new\s+Function\s*\(/ },
];

// Matches a child_process exec-family call; `.exec(` is deliberately excluded.
const EXEC_CALL_RE = /(?<![.\w])(execSync|execFileSync|execFile|exec|spawnSync|spawn)\s*\(/;

/**
 * Return the text of an exec-family call's first argument, joining up to
 * `lookahead` following lines for multi-line calls (`execSync(\n  'cmd', …)`).
 */
function execFirstArg(lines, i, lookahead = 3) {
  const m = EXEC_CALL_RE.exec(lines[i]);
  if (!m) return null;
  let text = lines[i].slice(m.index + m[0].length);
  for (let j = 1; j <= lookahead && text.trim() === ''; j++) {
    if (i + j >= lines.length) break;
    text = lines[i + j];
  }
  return { fn: m[1], arg: text.trim() };
}

/**
 * A first argument is "safe" when it is a plain string literal with no
 * interpolation — a fixed command cannot be derived from task content.
 */
function isLiteralArg(arg) {
  return /^'[^']*'\s*[,)]?/.test(arg)
    || /^"[^"]*"\s*[,)]?/.test(arg)
    || /^`[^`$]*`\s*[,)]?/.test(arg);
}

function expandHome(p) {
  return typeof p === 'string' && p.startsWith('~') ? join(homedir(), p.slice(1)) : p;
}

/** True when `child` resolves to `parent` or somewhere beneath it. */
function isUnder(child, parent) {
  const rel = relative(resolve(parent), resolve(child));
  return rel === '' || (!rel.startsWith('..') && !isAbsolute(rel));
}

// ---------------------------------------------------------------------------
// Input parsing
// ---------------------------------------------------------------------------

/** Parse a results file: BenchRunOutput JSON, array of them, or JSONL. */
export function parseResults(text) {
  const trimmed = text.trim();
  if (!trimmed) return [];
  if (trimmed.startsWith('{') || trimmed.startsWith('[')) {
    let parsed;
    try { parsed = JSON.parse(trimmed); } catch { parsed = null; }
    if (parsed) {
      if (Array.isArray(parsed)) {
        // Array of BenchRunOutput (multi-model) or of QuestionResult.
        return parsed.flatMap((p) => (p && Array.isArray(p.results) ? p.results : [p]));
      }
      if (Array.isArray(parsed.results)) return parsed.results;
      if (parsed.task_id) return [parsed];
    }
  }
  // JSONL fallback: one QuestionResult per line.
  return trimmed.split('\n').filter(Boolean).map((l) => JSON.parse(l));
}

/** Parse trajectories.jsonl into a Map keyed by task_id. */
export function parseTrajectories(text) {
  const byId = new Map();
  for (const line of text.split('\n')) {
    if (!line.trim()) continue;
    const rec = JSON.parse(line);
    if (rec && rec.task_id) byId.set(rec.task_id, rec);
  }
  return byId;
}

function trajSteps(traj) {
  return Array.isArray(traj?.steps) ? traj.steps : [];
}

function stepText(step) {
  const parts = [];
  for (const k of ['output', 'content', 'input', 'url', 'text']) {
    const v = step?.[k];
    if (v == null) continue;
    parts.push(typeof v === 'string' ? v : JSON.stringify(v));
  }
  return parts.join('\n');
}

// ---------------------------------------------------------------------------
// Checks — each is a pure function returning
//   { id, severity, status: 'pass'|'fail'|'skip', evidence: string[], harness_gap? }
// ---------------------------------------------------------------------------

/**
 * AUD-1 answer-leakage (CRITICAL — RDI: GAIA leaked ~98% of answers via
 * public answer DBs). For each question marked correct, scan the
 * trajectory's fetched tool outputs for the gold answer appearing verbatim
 * or near-verbatim (normalised containment). A hit whose surrounding text or
 * URL matches an answer-DB signature is flagged as answer-DB sourcing.
 */
export function checkAnswerLeakage(results, trajById) {
  const evidence = [];
  let auditable = 0;
  for (const r of results) {
    if (!r.correct) continue;
    const traj = trajById.get(r.task_id);
    const outputs = trajSteps(traj).filter((s) => s.type === 'tool_result');
    if (outputs.length === 0) continue;
    auditable++;
    const gold = normaliseAnswer(r.expected_output);
    if (!gold) continue;
    for (const step of outputs) {
      const raw = stepText(step);
      const norm = normaliseAnswer(raw);
      // Verbatim / near-verbatim: normalised containment; single- and
      // two-char answers require a whole-token match to avoid noise hits.
      const hit = gold.length >= 3
        ? norm.includes(gold)
        : new RegExp(`(^|[^a-z0-9])${gold.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')}([^a-z0-9]|$)`).test(norm);
      if (!hit) continue;
      const lower = raw.toLowerCase();
      const dbSig = ANSWER_DB_SIGNATURES.find((sig) => lower.includes(sig));
      evidence.push(
        `task ${r.task_id}: gold answer "${gold}" appears in fetched ${step.name ?? 'tool'} output` +
        (dbSig ? ` — ANSWER-DB SIGNATURE MATCHED ("${dbSig}")` : ''),
      );
      break; // one hit per question is enough
    }
  }
  if (auditable === 0) {
    return {
      id: 'answer-leakage', severity: 'critical', status: 'skip', evidence: [],
      harness_gap:
        'cannot audit answer-leakage until the trajectory records per-step ' +
        'tool_result outputs — gaia-bench.ts persists only QuestionResult ' +
        '(no trajectory field) and gaia-agent.ts discards the messages[] ' +
        'array containing fetched tool outputs',
    };
  }
  return {
    id: 'answer-leakage', severity: 'critical',
    status: evidence.length ? 'fail' : 'pass',
    evidence: evidence.sort(),
  };
}

/**
 * AUD-2 no-work / no-LLM pass (CRITICAL — RDI: 6 of 8 benchmarks hit 100%
 * with zero LLM calls and zero reasoning). A question marked correct with
 * zero turns, zero output tokens, or an empty trajectory is the signature.
 * Enforceable NOW: turns / outputTokens are real QuestionResult fields.
 */
export function checkNoWork(results, trajById) {
  const evidence = [];
  let auditable = 0;
  for (const r of results) {
    if (!r.correct) continue;
    const hasTurns = typeof r.turns === 'number';
    const hasTokens = typeof r.outputTokens === 'number';
    const traj = trajById.get(r.task_id);
    if (!hasTurns && !hasTokens && !traj) continue;
    auditable++;
    const reasons = [];
    if (hasTurns && r.turns === 0) reasons.push('turns=0');
    if (hasTokens && r.outputTokens === 0) reasons.push('outputTokens=0');
    if (traj) {
      const llmCalls = trajSteps(traj).filter((s) => s.type === 'llm_call').length;
      const toolCalls = trajSteps(traj).filter((s) => s.type === 'tool_call').length;
      if (trajSteps(traj).length > 0 && llmCalls === 0) reasons.push('zero LLM calls in trajectory');
      if (trajSteps(traj).length > 0 && llmCalls === 0 && toolCalls === 0) reasons.push('no substantive tool use');
    }
    if (reasons.length) {
      evidence.push(`task ${r.task_id}: marked correct with ${reasons.join(', ')}`);
    }
  }
  if (auditable === 0) {
    return {
      id: 'no-work-pass', severity: 'critical', status: 'skip', evidence: [],
      harness_gap:
        'no correct result carries turns/outputTokens and no trajectory was ' +
        'supplied — cannot attest work was performed',
    };
  }
  return {
    id: 'no-work-pass', severity: 'critical',
    status: evidence.length ? 'fail' : 'pass',
    evidence: evidence.sort(),
  };
}

/**
 * AUD-3 oracle-leakage (CRITICAL). Assert the gold answer is NOT present in
 * the agent-visible prompt/context recorded in the trajectory. Static
 * assurance exists (gaia-agent.ts buildInitialContent() never touches
 * question.final_answer) but a per-run check needs the prompt recorded.
 */
export function checkOracleLeakage(results, trajById) {
  const evidence = [];
  let auditable = 0;
  for (const r of results) {
    const traj = trajById.get(r.task_id);
    const prompts = trajSteps(traj).filter((s) => s.type === 'prompt' || s.type === 'llm_call');
    if (prompts.length === 0) continue;
    auditable++;
    const gold = normaliseAnswer(r.expected_output);
    if (!gold || gold.length < 2) continue;
    for (const step of prompts) {
      // Only the agent-visible input side of an llm_call is oracle surface.
      const visible = step.type === 'prompt' ? stepText(step)
        : (typeof step.input === 'string' ? step.input : JSON.stringify(step.input ?? ''));
      if (normaliseAnswer(visible).includes(gold)) {
        evidence.push(`task ${r.task_id}: gold answer "${gold}" present in agent-visible ${step.type}`);
        break;
      }
    }
  }
  if (auditable === 0) {
    return {
      id: 'oracle-leakage', severity: 'critical', status: 'skip', evidence: [],
      harness_gap:
        'cannot audit oracle-leakage until the trajectory records the ' +
        'agent-visible prompt/context (steps of type "prompt"/"llm_call"); ' +
        'static review of gaia-agent.ts buildInitialContent() shows ' +
        'final_answer is not injected, but that is not a per-run attestation',
    };
  }
  return {
    id: 'oracle-leakage', severity: 'critical',
    status: evidence.length ? 'fail' : 'pass',
    evidence: evidence.sort(),
  };
}

/**
 * AUD-4 grader-isolation (CRITICAL — METR: o3 / Claude-3.7 monkey-patch
 * graders in 30%+ of runs). Scan trajectory tool calls for writes/patches/
 * introspection targeting the judge, its cache, or the witness surface.
 */
export function checkGraderIsolation(results, trajById) {
  const evidence = [];
  let auditable = 0;
  for (const r of results) {
    const traj = trajById.get(r.task_id);
    if (!traj) continue;
    const calls = trajSteps(traj).filter((s) => s.type === 'tool_call');
    const names = calls.length
      ? calls.map((c) => c.name ?? '')
      : (Array.isArray(traj.tools_used) ? traj.tools_used : []);
    if (calls.length === 0 && names.length === 0) continue;
    auditable++;
    for (const call of calls) {
      const text = `${call.name ?? ''} ${typeof call.input === 'string' ? call.input : JSON.stringify(call.input ?? '')}`;
      const pattern = GRADER_SURFACE_PATTERNS.find((p) => p.test(text));
      if (pattern) {
        evidence.push(
          `task ${r.task_id}: tool_call "${call.name ?? '?'}" targets grader surface (matched ${pattern})` +
          (WRITE_CAPABLE_TOOLS.has(call.name) ? ' via WRITE-CAPABLE tool' : ''),
        );
      }
    }
    if (calls.length === 0) {
      for (const n of names) {
        const pattern = GRADER_SURFACE_PATTERNS.find((p) => p.test(String(n)));
        if (pattern) evidence.push(`task ${r.task_id}: tools_used contains grader-surface name "${n}"`);
      }
    }
  }
  if (auditable === 0) {
    return {
      id: 'grader-isolation', severity: 'critical', status: 'skip', evidence: [],
      harness_gap:
        'cannot audit grader-isolation until the trajectory records tool_call ' +
        'names + arguments — GaiaAgentResult.toolCallsByName (counts only) is ' +
        'computed in gaia-agent.ts but never persisted by gaia-bench.ts; note ' +
        'the judge cache (~/.cache/ruflo/gaia/judgments) is an unsigned ' +
        'filesystem oracle writable by any local process',
    };
  }
  return {
    id: 'grader-isolation', severity: 'critical',
    status: evidence.length ? 'fail' : 'pass',
    evidence: evidence.sort(),
  };
}

/**
 * AUD-5 normalization-collision (WARN — RDI: GAIA normalisation collisions
 * grant credit to degenerate answers). Flags correct answers that are
 * empty/whitespace/single-char/generic after the judge's own normalisation.
 * Also flags short answers that pass only under the bidirectional substring
 * rule in gaia-agent.ts isAnswerCorrect() (normExpected.includes(normModel)).
 * Enforceable NOW on the real results file.
 */
export function checkNormalizationCollision(results) {
  const evidence = [];
  for (const r of results) {
    if (!r.correct) continue;
    const norm = normaliseAnswer(r.answer);
    const gold = normaliseAnswer(r.expected_output);
    if (GENERIC_TOKENS.has(norm) && norm !== gold) {
      evidence.push(`task ${r.task_id}: correct with generic/degenerate answer "${norm || '(empty)'}" vs gold "${gold}"`);
    } else if (norm.length === 1 && gold.length > 1) {
      evidence.push(`task ${r.task_id}: single-char answer "${norm}" credited against gold "${gold}" (substring-collision hazard: isAnswerCorrect() matches normExpected.includes(normModel))`);
    } else if (norm.length >= 2 && norm.length < gold.length && gold.includes(norm) && norm !== gold) {
      evidence.push(`task ${r.task_id}: answer "${norm}" is a strict substring of gold "${gold}" — passes only under bidirectional substring matching`);
    }
  }
  return {
    id: 'normalization-collision', severity: 'warn',
    status: evidence.length ? 'fail' : 'pass',
    evidence: evidence.sort(),
  };
}

/**
 * AUD-6 voting-disclosure (WARN — hidden best-of-N inflates pass-rate N×
 * without disclosure). gaia-bench.ts accepts --voting-attempts but never
 * persists it into BenchRunOutput or metadata.json.
 */
export function checkVotingDisclosure(metadata) {
  if (!metadata) {
    return {
      id: 'voting-disclosure', severity: 'warn', status: 'skip', evidence: [],
      harness_gap: 'no metadata.json supplied — voting/self-consistency N unattested',
    };
  }
  const n = metadata.voting_attempts ?? metadata.votingAttempts ?? metadata.voting;
  if (typeof n === 'number') {
    return {
      id: 'voting-disclosure', severity: 'warn', status: 'pass',
      evidence: [`voting_attempts=${n} disclosed in metadata${n > 1 ? ' (best-of-N is declared, not hidden)' : ''}`],
    };
  }
  return {
    id: 'voting-disclosure', severity: 'warn', status: 'fail',
    evidence: [
      'metadata records no voting/self-consistency N — a hidden best-of-N cannot be ruled out',
    ],
    harness_gap:
      'gaia-bench.ts does not persist votingAttempts into BenchRunOutput; ' +
      '/gaia submit metadata.json schema has no voting field — add one at ' +
      'package time from the run flags',
  };
}

/**
 * AUD-7 split-integrity (WARN/INFO). gaia-loader.ts hard-codes
 * split=validation, whose gold answers are public on Hugging Face — results
 * must be disclosed as validation-split and cannot claim held-out standing.
 */
export function checkSplitIntegrity(metadata) {
  if (!metadata) {
    return {
      id: 'split-integrity', severity: 'warn', status: 'skip', evidence: [],
      harness_gap: 'no metadata.json supplied — split undisclosed',
    };
  }
  const split = metadata.gaia_split ?? metadata.split;
  if (split === 'validation') {
    return {
      id: 'split-integrity', severity: 'info', status: 'pass',
      evidence: [
        'split=validation disclosed; note validation gold answers are public ' +
        '(the same HF dataset the agent tool catalogue can reach) — scores ' +
        'are self-reported validation numbers, not held-out test standing',
      ],
    };
  }
  if (split === 'test') {
    return {
      id: 'split-integrity', severity: 'info', status: 'pass',
      evidence: ['split=test disclosed (gold answers withheld by GAIA — strongest standing)'],
    };
  }
  return {
    id: 'split-integrity', severity: 'warn', status: 'fail',
    evidence: [
      'metadata does not disclose which GAIA split was run; gaia-loader.ts ' +
      'only fetches split=validation, so undisclosed results are presumed ' +
      'validation — declare gaia_split explicitly',
    ],
    harness_gap: '/gaia submit metadata.json schema has no gaia_split field',
  };
}

/**
 * AUD-8 answer-key-reads (CRITICAL — RDI: benchmarks gamed by reading answer
 * keys from unsanitized config). Scan the runner sources AND the produced
 * artifacts for answer/gold/solution/ground-truth-shaped file paths referenced
 * OUTSIDE the sanctioned dataset dir(s). Fail-closed. Enforceable NOW (static);
 * skips only when no source/artifact text is available to scan.
 *
 * `sources`: [{ path, text, kind? }] — runner sources + raw artifact texts.
 * `datasetDirs`: sanctioned dataset dirs; a match under one of them is allowed.
 */
export function checkAnswerKeyReads({ sources = [], datasetDirs = [] } = {}) {
  const scannable = sources.filter((s) => s && typeof s.text === 'string');
  if (scannable.length === 0) {
    return {
      id: 'answer-key-reads', severity: 'critical', status: 'skip', evidence: [],
      harness_gap:
        'no runner source or produced artifact text available to scan for ' +
        'answer-key path references (expected gaia-* runner sources under ' +
        'v3/@claude-flow/cli/src and/or the results/trajectory artifacts) — ' +
        'run inside the repo or pass --results/--trajectories',
    };
  }
  const sanctioned = datasetDirs.map((d) => resolve(expandHome(d)));
  const evidence = [];
  for (const { path: src, text, kind } of scannable) {
    for (const line of text.split('\n')) {
      PATH_SHAPED.lastIndex = 0;
      let m;
      while ((m = PATH_SHAPED.exec(line)) !== null) {
        const rawPath = m[1];
        if (!ANSWER_KEY_PATTERN.test(rawPath)) continue;
        const abs = resolve(expandHome(rawPath));
        if (sanctioned.some((d) => isUnder(abs, d))) continue;
        evidence.push(
          `${kind ?? 'source'} ${src}: answer-key-shaped path "${rawPath}" ` +
          `referenced outside sanctioned dataset dir — ${line.trim().slice(0, 160)}`,
        );
      }
    }
  }
  return {
    id: 'answer-key-reads', severity: 'critical',
    status: evidence.length ? 'fail' : 'pass',
    evidence: evidence.sort(),
  };
}

/**
 * AUD-9 dynamic-eval (CRITICAL — RDI: trojanized test infrastructure). Static
 * scan of the gaia-bench runner sources for eval() / new Function() / an
 * exec-family call whose first argument is not a fixed string literal (i.e.
 * potentially derived from task content). Fail-closed. Skips when the runner
 * sources cannot be located (never false-passes).
 *
 * Refinements preserved from #2547 (they suppressed 8 false positives on
 * ruflo's own `gcloud secrets …` execSync calls): comment lines skipped;
 * multi-line calls resolved via execFirstArg; RegExp.prototype.exec excluded by
 * the (?<![.\w]) lookbehind; fixed-string commands treated as safe.
 */
export function checkDynamicEval({ runnerSources = [] } = {}) {
  const scannable = runnerSources.filter((s) => s && typeof s.text === 'string');
  if (scannable.length === 0) {
    return {
      id: 'dynamic-eval', severity: 'critical', status: 'skip', evidence: [],
      harness_gap:
        'gaia runner sources not located (expected v3/@claude-flow/cli/src/' +
        'benchmarks/gaia-* and src/commands/gaia-bench.ts) — cannot statically ' +
        'attest the runner performs no dynamic evaluation of task content; ' +
        'run inside the repo or pass --repo-root',
    };
  }
  const evidence = [];
  for (const { path: src, text } of scannable) {
    const lines = text.split('\n');
    for (let i = 0; i < lines.length; i++) {
      const line = lines[i];
      const trimmed = line.trim();
      if (trimmed.startsWith('//') || trimmed.startsWith('*') || trimmed.startsWith('/*')) continue;
      for (const { id, re } of DYNAMIC_EVAL_PATTERNS) {
        if (re.test(line)) {
          evidence.push(`${src}:${i + 1}: ${id} — ${trimmed.slice(0, 160)}`);
        }
      }
      const call = execFirstArg(lines, i);
      if (call && !isLiteralArg(call.arg)) {
        const pat = call.arg.includes('${') ? 'exec-interpolated-template' : 'exec-nonliteral';
        evidence.push(`${src}:${i + 1}: ${pat} (${call.fn}) — ${trimmed.slice(0, 160)}`);
      }
    }
  }
  return {
    id: 'dynamic-eval', severity: 'critical',
    status: evidence.length ? 'fail' : 'pass',
    evidence: evidence.sort(),
  };
}

/**
 * AUD-10 judge-injection (WARN). Scan the agent's PRODUCED answer + its
 * llm_call outputs for judge-directed prompt-injection markers aimed at the
 * LLM-as-judge. Distinct from AUD-3 oracle-leakage (which scans the agent's
 * INPUT prompt for the gold answer) — this is the output-side attack surface,
 * so both run. Enforceable NOW from the results file (trajectory optional).
 */
export function checkJudgeInjection({ results = [], trajById = new Map() } = {}) {
  const evidence = [];
  const scan = (label, text) => {
    if (typeof text !== 'string' || !text) return;
    for (const { id, re } of JUDGE_INJECTION_PATTERNS) {
      const m = re.exec(text);
      if (!m) continue;
      const excerpt = text
        .slice(Math.max(0, m.index - 30), m.index + m[0].length + 30)
        .replace(/\s+/g, ' ').trim();
      evidence.push(`${label}: judge-injection marker "${id}" — "${excerpt}"`);
    }
  };
  for (const r of results) {
    scan(`task ${r.task_id} answer`, r.answer);
    for (const step of trajSteps(trajById.get(r.task_id))) {
      if (step.type === 'llm_call') scan(`task ${r.task_id} llm_call output`, step.output);
    }
  }
  return {
    id: 'judge-injection', severity: 'warn',
    status: evidence.length ? 'fail' : 'pass',
    evidence: evidence.sort(),
  };
}

// ---------------------------------------------------------------------------
// Report assembly
// ---------------------------------------------------------------------------

export const AUDITED_AT_PLACEHOLDER = 'AUDITED_AT_PLACEHOLDER';

export function runAudit({
  results, trajById = new Map(), metadata = null, auditedAt,
  runnerSources = [], artifactSources = [], datasetDirs = [],
} = {}) {
  const checks = [
    checkAnswerLeakage(results, trajById),
    checkNoWork(results, trajById),
    checkOracleLeakage(results, trajById),
    checkGraderIsolation(results, trajById),
    checkNormalizationCollision(results),
    checkVotingDisclosure(metadata),
    checkSplitIntegrity(metadata),
    checkAnswerKeyReads({ sources: [...runnerSources, ...artifactSources], datasetDirs }),
    checkDynamicEval({ runnerSources }),
    checkJudgeInjection({ results, trajById }),
  ];
  const criticalFails = checks.filter((c) => c.severity === 'critical' && c.status === 'fail');
  const warnFails = checks.filter((c) => c.severity === 'warn' && c.status === 'fail');
  const skipped = checks.filter((c) => c.status === 'skip');
  const harnessGaps = checks.filter((c) => c.harness_gap).map((c) => `${c.id}: ${c.harness_gap}`);
  return {
    schema: 'ruflo-gaia-audit/v1',
    audited_at: auditedAt ?? AUDITED_AT_PLACEHOLDER,
    threat_model: 'UC Berkeley RDI BenchJack (arXiv:2605.12673) + METR reward-hacking — see ADR-167',
    totals: {
      questions: results.length,
      marked_correct: results.filter((r) => r.correct).length,
      trajectories: trajById.size,
    },
    checks,
    attestation: {
      clean: criticalFails.length === 0,
      strict_clean: criticalFails.length === 0 && warnFails.length === 0,
      critical_failures: criticalFails.map((c) => c.id),
      warn_failures: warnFails.map((c) => c.id),
      skipped: skipped.map((c) => c.id),
      harness_gaps: harnessGaps,
    },
  };
}

// ---------------------------------------------------------------------------
// CLI
// ---------------------------------------------------------------------------

function parseCliArgs(argv) {
  const out = { _: [], 'dataset-dir': [] };
  for (let i = 0; i < argv.length; i++) {
    const a = argv[i];
    if (a === '--strict' || a === '--json' || a === '--help' || a === '--skip-source-scan') {
      out[a.slice(2)] = true; continue;
    }
    if (a.startsWith('--')) {
      const key = a.slice(2);
      const next = argv[i + 1];
      const val = (next !== undefined && !next.startsWith('--')) ? (i++, next) : true;
      if (key === 'dataset-dir') { if (typeof val === 'string') out['dataset-dir'].push(val); }
      else out[key] = val;
    } else out._.push(a);
  }
  return out;
}

function usage() {
  return 'Usage: node gaia-audit.mjs --results <file> [--trajectories <file>] ' +
    '[--metadata <file>] [--out <report.json>] [--audited-at <iso8601>] ' +
    '[--repo-root <dir>] [--dataset-dir <dir>]... [--skip-source-scan] [--strict] [--json]';
}

function sha256Hex(buf) {
  return createHash('sha256').update(buf).digest('hex');
}

function safeReadFile(p) {
  try { return readFileSync(p, 'utf8'); } catch { return null; }
}

/** Walk `dir`, collecting basename-`gaia*` .ts/.mjs/.js source files. */
function walkGaiaSources(dir, acc = []) {
  let entries;
  try { entries = readdirSync(dir, { withFileTypes: true }); } catch { return acc; }
  for (const e of entries) {
    if (e.name === 'node_modules' || e.name === '.git' || e.name === 'dist') continue;
    const p = join(dir, e.name);
    if (e.isDirectory()) walkGaiaSources(p, acc);
    else if (/^gaia.*\.(?:ts|mjs|js)$/.test(e.name)) acc.push(p);
  }
  return acc;
}

/**
 * Gather the gaia-bench runner sources (AUD-8/9 static-scan population) from
 * `repoRoot`. Returns [] when the tree is absent (standalone/ejected run) so
 * the checks skip honestly rather than false-pass.
 */
function gatherRunnerSources(repoRoot) {
  const dirs = [
    join(repoRoot, 'v3/@claude-flow/cli/src/benchmarks'),
    join(repoRoot, 'v3/@claude-flow/cli/src/commands'),
  ];
  const out = [];
  for (const d of dirs) {
    for (const f of walkGaiaSources(d)) {
      const text = safeReadFile(f);
      if (text != null) out.push({ path: f, text, kind: 'runner' });
    }
  }
  return out;
}

/**
 * Derive the repo root from --repo-root or by walking up from this script
 * (plugins/ruflo-workflows/scripts/gaia-audit.mjs → three levels up), falling
 * back to cwd. Only used to locate runner sources; a wrong guess just makes the
 * source-scan checks skip.
 */
function deriveRepoRoot(explicit) {
  if (typeof explicit === 'string' && explicit) return resolve(expandHome(explicit));
  try {
    const here = fileURLToPath(import.meta.url);
    const guess = resolve(here, '..', '..', '..', '..');
    if (existsSync(join(guess, 'v3/@claude-flow/cli/src/benchmarks'))) return guess;
  } catch { /* fall through */ }
  return process.cwd();
}

async function main() {
  const args = parseCliArgs(process.argv.slice(2));
  if (args.help) { console.log(usage()); process.exit(0); }
  if (!args.results || typeof args.results !== 'string') {
    console.error(usage());
    process.exit(2);
  }
  const inputs = {};
  const readInput = (label, p) => {
    const abs = resolve(p);
    if (!existsSync(abs)) {
      console.error(`gaia-audit: ${label} file not found: ${abs}`);
      process.exit(2);
    }
    const text = readFileSync(abs, 'utf8');
    inputs[`${label}_sha256`] = sha256Hex(text);
    return text;
  };

  // Raw artifact texts double as the AUD-8 answer-key-scan population.
  const artifactSources = [];

  let results;
  let resultsText;
  try {
    resultsText = readInput('results', args.results);
    results = parseResults(resultsText);
  } catch (e) {
    console.error(`gaia-audit: could not parse results file: ${e.message}`);
    process.exit(2);
  }
  artifactSources.push({ path: resolve(args.results), text: resultsText, kind: 'artifact' });
  if (!Array.isArray(results) || results.length === 0 || !results.every((r) => r && typeof r.task_id === 'string')) {
    console.error('gaia-audit: results file has no QuestionResult records (expected task_id/correct/answer/expected_output fields)');
    process.exit(2);
  }

  let trajById = new Map();
  if (args.trajectories && typeof args.trajectories === 'string') {
    let trajText;
    try {
      trajText = readInput('trajectories', args.trajectories);
      trajById = parseTrajectories(trajText);
    } catch (e) { console.error(`gaia-audit: could not parse trajectories file: ${e.message}`); process.exit(2); }
    artifactSources.push({ path: resolve(args.trajectories), text: trajText, kind: 'artifact' });
  }
  let metadata = null;
  if (args.metadata && typeof args.metadata === 'string') {
    let metaText;
    try {
      metaText = readInput('metadata', args.metadata);
      metadata = JSON.parse(metaText);
    } catch (e) { console.error(`gaia-audit: could not parse metadata file: ${e.message}`); process.exit(2); }
    artifactSources.push({ path: resolve(args.metadata), text: metaText, kind: 'artifact' });
  }

  // Static source-scan context (AUD-8/9). Skipped entirely with
  // --skip-source-scan; otherwise runner sources are located from the repo and
  // dataset dirs default to the HF + ruflo cache locations gaia-loader uses.
  let runnerSources = [];
  let artifactScan = [];
  let datasetDirs = [];
  if (!args['skip-source-scan']) {
    runnerSources = gatherRunnerSources(deriveRepoRoot(args['repo-root']));
    artifactScan = artifactSources;
    datasetDirs = (args['dataset-dir'] && args['dataset-dir'].length)
      ? args['dataset-dir'].map((d) => resolve(expandHome(d)))
      : [
          join(homedir(), '.cache', 'huggingface'),
          join(homedir(), '.cache', 'ruflo', 'gaia', 'dataset'),
        ];
  }

  const report = runAudit({
    results, trajById, metadata,
    runnerSources, artifactSources: artifactScan, datasetDirs,
    auditedAt: typeof args['audited-at'] === 'string' ? args['audited-at'] : undefined,
  });
  report.inputs = inputs;

  if (args.out && typeof args.out === 'string') {
    writeFileSync(resolve(args.out), JSON.stringify(report, null, 2) + '\n');
  }

  if (args.json) {
    console.log(JSON.stringify(report, null, 2));
  } else {
    console.log('GAIA pre-submission exploit audit (ADR-167)');
    console.log(`questions=${report.totals.questions} correct=${report.totals.marked_correct} trajectories=${report.totals.trajectories}`);
    console.log('');
    for (const c of report.checks) {
      const tag = c.status === 'pass' ? 'PASS' : c.status === 'fail' ? 'FAIL' : 'SKIP';
      console.log(`[${tag}] ${c.id} (${c.severity})`);
      for (const e of c.evidence) console.log(`       ${e}`);
      if (c.harness_gap) console.log(`       harness gap: ${c.harness_gap}`);
    }
    console.log('');
    console.log(`attestation: clean=${report.attestation.clean} strict_clean=${report.attestation.strict_clean}`);
    if (report.attestation.critical_failures.length) {
      console.log(`CRITICAL failures: ${report.attestation.critical_failures.join(', ')}`);
    }
    if (report.attestation.warn_failures.length) {
      console.log(`WARN failures: ${report.attestation.warn_failures.join(', ')}`);
    }
  }

  const fail = !report.attestation.clean || (args.strict && !report.attestation.strict_clean);
  process.exit(fail ? 1 : 0);
}

// Run only when invoked directly (not when imported by tests).
const invokedDirectly = process.argv[1] && (
  process.argv[1].endsWith('gaia-audit.mjs')
);
if (invokedDirectly) {
  main().catch((err) => {
    console.error(`gaia-audit: ${err?.stack ?? err}`);
    process.exit(2);
  });
}