@astrasyncai/verification-gateway 3.9.0 → 3.10.1

package/dist/gateway/gateway.mjs

@@ -18,7 +18,9 @@ var HIGH_RISK_COMMANDS = [
   "git push",
   "npm publish",
   "docker",
-  "kubectl"
+  "kubectl",
+  "gh pr create",
+  "gh pr merge"
 ];
 var SENSITIVE_PATHS = [
   ".ssh",
@@ -37,6 +39,164 @@ var SENSITIVE_PATHS = [
   ".pypirc"
 ];
 
+// src/local-evaluator/command-matcher.ts
+var WRAPPERS = /* @__PURE__ */ new Set([
+  "sudo",
+  "doas",
+  "env",
+  "nohup",
+  "time",
+  "timeout",
+  "nice",
+  "ionice",
+  "stdbuf",
+  "xargs",
+  "command",
+  "exec",
+  "builtin"
+]);
+var ASSIGN_RE = /^[a-z_][a-z0-9_]*=/;
+var MAX_WRAPPER_HOPS = 4;
+var MAX_RECURSION_DEPTH = 3;
+var MAX_COMMAND_LENGTH = 1e5;
+var LOCAL_HOST_RE = /^(localhost|127\.0\.0\.1|::1|[a-z0-9.-]*\.localhost)$/;
+function buildRiskList(entries) {
+  const singleWord = /* @__PURE__ */ new Set();
+  const multiWord = [];
+  for (const entry of entries) {
+    const words = entry.toLowerCase().trim().split(/\s+/);
+    if (words.length === 1) singleWord.add(words[0]);
+    else multiWord.push(words);
+  }
+  multiWord.sort((a, b) => b.length - a.length);
+  return { singleWord, multiWord };
+}
+function stripHeredocBodies(text) {
+  const lines = text.split("\n");
+  const kept = [];
+  let terminator = null;
+  for (const line of lines) {
+    if (terminator !== null) {
+      if (line.trim() === terminator) terminator = null;
+      continue;
+    }
+    kept.push(line);
+    const heredoc = line.match(/<<-?\s*['"]?(\w+)['"]?/);
+    if (heredoc) terminator = heredoc[1];
+  }
+  return kept.join("\n");
+}
+function extractDollarSubstitutions(text, collect) {
+  let result = text;
+  let start = result.indexOf("$(");
+  let iterations = 0;
+  while (start !== -1 && iterations < 50) {
+    iterations++;
+    let depth = 0;
+    let end = result.length;
+    for (let i = start + 1; i < result.length; i++) {
+      if (result[i] === "(") depth++;
+      else if (result[i] === ")") {
+        depth--;
+        if (depth === 0) {
+          end = i;
+          break;
+        }
+      }
+    }
+    collect(result.slice(start + 2, end));
+    result = `${result.slice(0, start)} ${result.slice(Math.min(end + 1, result.length))}`;
+    start = result.indexOf("$(");
+  }
+  return result;
+}
+function extractBacktickSubstitutions(text, collect) {
+  if (!text.includes("`")) return text;
+  const chunks = text.split("`");
+  const kept = [];
+  for (let i = 0; i < chunks.length; i++) {
+    if (i % 2 === 1) collect(chunks[i]);
+    else kept.push(chunks[i]);
+  }
+  return kept.join(" ");
+}
+function basename(token) {
+  const slash = token.lastIndexOf("/");
+  return slash === -1 ? token : token.slice(slash + 1);
+}
+function matchSegment(segment, list) {
+  const tokens = segment.split(/\s+/).filter((t) => /[a-z0-9]/.test(t));
+  let i = 0;
+  let hops = 0;
+  while (i < tokens.length) {
+    while (i < tokens.length && (tokens[i] === "export" || ASSIGN_RE.test(tokens[i]))) i++;
+    if (i >= tokens.length) return null;
+    const cmd = basename(tokens[i]);
+    const significant = [cmd, ...tokens.slice(i + 1).filter((t) => !t.startsWith("-"))];
+    for (const entry of list.multiWord) {
+      if (entry.length <= significant.length && entry.every((w, j) => significant[j] === w)) {
+        return entry.join(" ");
+      }
+    }
+    if (list.singleWord.has(cmd)) return cmd;
+    if (WRAPPERS.has(cmd) && hops < MAX_WRAPPER_HOPS) {
+      hops++;
+      i++;
+      while (i < tokens.length && (tokens[i].startsWith("-") || ASSIGN_RE.test(tokens[i]))) i++;
+      continue;
+    }
+    return null;
+  }
+  return null;
+}
+function collectMatches(command, list, matches, depth) {
+  if (depth > MAX_RECURSION_DEPTH || command.length > MAX_COMMAND_LENGTH) return;
+  let text = command.toLowerCase();
+  text = stripHeredocBodies(text);
+  const recurse = (inner) => collectMatches(inner, list, matches, depth + 1);
+  text = extractDollarSubstitutions(text, recurse);
+  text = extractBacktickSubstitutions(text, recurse);
+  text = text.replace(/'[^']*'/g, " ").replace(/"[^"]*"/g, " ");
+  text = text.replace(/(^|\s)#[^\n]*/g, " ");
+  const segments = text.split(/\|\||&&|[;|&\n]|[({]/);
+  for (const segment of segments) {
+    const matched = matchSegment(segment, list);
+    if (matched) matches.add(matched);
+  }
+}
+function matchHighRiskCommands(command, highRiskCommands = HIGH_RISK_COMMANDS) {
+  if (!command) return [];
+  const matches = /* @__PURE__ */ new Set();
+  collectMatches(command, buildRiskList(highRiskCommands), matches, 0);
+  return [...matches];
+}
+function isLocalhostOnlyNetworkCommand(command) {
+  const text = command.toLowerCase();
+  const tokens = text.split(/\s+/).map((t) => t.replace(/^["'`]+|["'`,;)]+$/g, ""));
+  let localCount = 0;
+  for (const token of tokens) {
+    let host = null;
+    if (token.includes("://")) {
+      let rest = token.slice(token.indexOf("://") + 3);
+      const at = rest.indexOf("@");
+      if (at !== -1) rest = rest.slice(at + 1);
+      host = rest.split(/[/?#]/)[0];
+    } else if (/^[a-z0-9.[\]:-]+:\d+(\/.*)?$/.test(token)) {
+      host = token.split("/")[0];
+    } else if (/^(localhost|127\.0\.0\.1|\[?::1\]?)(\/.*)?$/.test(token)) {
+      host = token.split("/")[0];
+    }
+    if (host === null) continue;
+    const bare = host.replace(/:\d+$/, "").replace(/^\[|\]$/g, "");
+    if (LOCAL_HOST_RE.test(bare)) {
+      localCount++;
+    } else {
+      return false;
+    }
+  }
+  return localCount > 0;
+}
+
 // src/local-evaluator/evaluator.ts
 var LocalEvaluator = class {
   constructor(policy) {
@@ -187,12 +347,10 @@ var LocalEvaluator = class {
       return score;
     }
     if (context.purpose === "shell.exec" && context.target) {
-      const targetLower = context.target.toLowerCase();
-      for (const cmd of HIGH_RISK_COMMANDS) {
-        if (targetLower.startsWith(cmd) || targetLower.includes(` ${cmd} `) || targetLower.includes(` ${cmd}`)) {
-          score = Math.max(score, 80);
-          break;
-        }
+      const matched = matchHighRiskCommands(context.target);
+      const localProbeOnly = matched.length > 0 && matched.every((cmd) => cmd === "curl" || cmd === "wget") && isLocalhostOnlyNetworkCommand(context.target);
+      if (matched.length > 0 && !localProbeOnly) {
+        score = Math.max(score, 80);
       }
     }
     if ((context.purpose === "file.read" || context.purpose === "file.write" || context.purpose === "file.delete") && context.target) {
@@ -3503,7 +3661,7 @@ function getTrustLevel(score) {
 }
 
 // src/version.ts
-var SDK_VERSION = "3.9.0";
+var SDK_VERSION = "3.10.1";
 
 // src/well-known.ts
 var CACHE_TTL_MS = 60 * 60 * 1e3;
@@ -3919,6 +4077,9 @@ async function verify(config, request, options) {
 // src/adapters/stepup-poll.ts
 var FALLBACK_MAX_WAIT_MS = 5 * 60 * 1e3;
 
+// src/metadata-capture.ts
+var MAX_TOTAL_BYTES = 16 * 1024;
+
 // src/adapters/express.ts
 var DEFAULT_ROUTES_REFRESH_MS = 5 * 60 * 1e3;
 
@@ -4061,8 +4222,75 @@ async function verifyOnline(config, context, astraId) {
 // src/gateway/modes/hybrid.ts
 import { existsSync, readFileSync as readFileSync2, writeFileSync, mkdirSync } from "fs";
 import { dirname } from "path";
+
+// src/gateway/redact.ts
+var marker = (type2) => `[REDACTED:${type2}]`;
+var REDACTION_PATTERNS = [
+  // Multi-line blocks first so later single-line patterns don't shred them.
+  {
+    type: "private-key",
+    re: /-----BEGIN [A-Z ]*PRIVATE KEY-----[\s\S]*?(?:-----END [A-Z ]*PRIVATE KEY-----|$)/g
+  },
+  // URL userinfo credentials (https://user:pass@, x-access-token:ghp_…@).
+  {
+    type: "url-credentials",
+    re: /([a-z][a-z0-9+.-]*:\/\/)[^\s/@]+:[^\s/@]+@/gi,
+    replacement: `$1${marker("url-credentials")}@`
+  },
+  // Known token formats.
+  { type: "github-token", re: /\bgh[pousr]_[A-Za-z0-9]{20,255}\b/g },
+  { type: "github-token", re: /\bgithub_pat_[A-Za-z0-9_]{20,255}\b/g },
+  { type: "astrasync-key", re: /\bkya_[A-Fa-f0-9]{32,128}\b/g },
+  { type: "openai-key", re: /\bsk-[A-Za-z0-9_-]{20,}\b/g },
+  { type: "aws-access-key-id", re: /\bAKIA[0-9A-Z]{16}\b/g },
+  { type: "jwt", re: /\beyJ[A-Za-z0-9_-]{10,}\.[A-Za-z0-9_-]{10,}\.[A-Za-z0-9_-]{10,}\b/g },
+  // Bearer/token headers.
+  {
+    type: "bearer-token",
+    re: /\b(bearer\s+)[a-z0-9._~+/=-]{16,}/gi,
+    replacement: `$1${marker("bearer-token")}`
+  },
+  // Generic KEY=value / key: value assignments with secret-ish names.
+  {
+    type: "assigned-secret",
+    re: /\b((?:password|passwd|secret|token|api[_-]?key|apikey|access[_-]?key|client[_-]?secret|private[_-]?key)\s*[=:]\s*)(['"]?)[^\s'"&;]{8,}\2/gi,
+    replacement: `$1$2${marker("assigned-secret")}$2`
+  }
+];
+function redactSecrets(text) {
+  let result = text;
+  for (const { type: type2, re, replacement } of REDACTION_PATTERNS) {
+    result = result.replace(re, replacement ?? marker(type2));
+  }
+  return result;
+}
+function redactDeep(value) {
+  if (typeof value === "string") {
+    return redactSecrets(value);
+  }
+  if (Array.isArray(value)) {
+    return value.map((item) => redactDeep(item));
+  }
+  if (value !== null && typeof value === "object" && !(value instanceof Date)) {
+    const out = {};
+    for (const [k, v] of Object.entries(value)) {
+      out[k] = redactDeep(v);
+    }
+    return out;
+  }
+  return value;
+}
+
+// src/gateway/modes/hybrid.ts
 var MAX_QUEUE_SIZE = 1e3;
 var MAX_RETRIES = 3;
+var FLUSH_TIMEOUT = /* @__PURE__ */ Symbol("flush-timeout");
+function delay(ms) {
+  return new Promise((resolve) => {
+    const timer = setTimeout(() => resolve(FLUSH_TIMEOUT), ms);
+    if (typeof timer.unref === "function") timer.unref();
+  });
+}
 var HybridSyncQueue = class {
   constructor(config, persistPath) {
     this.queue = [];
@@ -4083,12 +4311,12 @@ var HybridSyncQueue = class {
   /**
    * Stop periodic sync and flush remaining entries.
    */
-  async stop() {
+  async stop(flushOptions = {}) {
     if (this.timer) {
       clearInterval(this.timer);
       this.timer = null;
     }
-    await this.flush();
+    await this.flush(flushOptions);
     this.saveToDisk();
   }
   /**
@@ -4100,8 +4328,8 @@ var HybridSyncQueue = class {
     }
     this.queue.push({
       id: `${Date.now()}-${Math.random().toString(36).slice(2, 8)}`,
-      context,
-      decision,
+      context: redactDeep(context),
+      decision: redactDeep(decision),
       timestamp: /* @__PURE__ */ new Date(),
       retryCount: 0,
       status: "pending"
@@ -4111,16 +4339,21 @@ var HybridSyncQueue = class {
   /**
    * Flush pending entries to the API.
    */
-  async flush() {
+  async flush(options = {}) {
     if (this.processing) return;
     this.processing = true;
     try {
       const pending = this.queue.filter((e) => e.status === "pending");
-      for (const entry of pending) {
-        try {
-          await verifyOnline(this.config, entry.context);
+      const batch = typeof options.limit === "number" ? pending.slice(0, options.limit) : pending;
+      for (const entry of batch) {
+        const attempt = verifyOnline(this.config, entry.context, this.config.agentId).then(() => "synced").catch(() => "error");
+        const result = options.perEntryTimeoutMs ? await Promise.race([attempt, delay(options.perEntryTimeoutMs)]) : await attempt;
+        if (result === FLUSH_TIMEOUT) {
+          break;
+        }
+        if (result === "synced") {
           entry.status = "synced";
-        } catch {
+        } else {
           entry.retryCount++;
           if (entry.retryCount >= MAX_RETRIES) {
             entry.status = "failed";
@@ -4197,8 +4430,8 @@ var TraceLogger = class {
       id: this.generateId(),
       timestamp: /* @__PURE__ */ new Date(),
       type: "evaluation",
-      context,
-      decision
+      context: redactDeep(context),
+      decision: redactDeep(decision)
     });
   }
   /**
@@ -4315,10 +4548,12 @@ var AstraSyncGateway = class {
   }
   /**
    * Graceful shutdown. Flushes sync queue and cleans up resources.
+   * Stateless callers (hooks) pass bounded flush options so sync never
+   * sits on a tool call's critical path.
    */
-  async shutdown() {
+  async shutdown(options) {
     if (this.syncQueue) {
-      await this.syncQueue.stop();
+      await this.syncQueue.stop(options?.syncFlush ?? {});
       this.syncQueue = null;
     }
     this.evaluator = null;
@@ -4341,7 +4576,7 @@ var AstraSyncGateway = class {
     let decision;
     switch (this.config.mode) {
       case "online":
-        decision = await verifyOnline(this.config, context, astraId);
+        decision = await verifyOnline(this.config, context, astraId ?? this.config.agentId);
         break;
       case "local":
         if (!this.evaluator) throw new Error("Local evaluator not initialized");