@crediolabs/policy-synth 0.1.3 → 0.1.4

package/src/synth/deny-cases.ts

@@ -0,0 +1,447 @@
+import type { PredicateLeaf, PredicateNode, ScVal } from '../types.ts'
+import type { EvalContext } from './evaluate.ts'
+import { literalNumericBigInt } from './predicate-literals.ts'
+
+export interface DenyCase {
+  dimension: string
+  ctx: EvalContext
+}
+
+export interface GeneratedCases {
+  permit: EvalContext
+  denies: DenyCase[]
+}
+
+type ComparisonOperator = 'eq' | 'lt' | 'lte' | 'gt' | 'gte'
+
+type ComparisonNode = {
+  op: ComparisonOperator
+  left: PredicateLeaf
+  right: PredicateLeaf
+}
+
+type MembershipNode = {
+  op: 'in'
+  needle: PredicateLeaf
+  haystack: PredicateLeaf[]
+}
+
+interface PredicateFacts {
+  comparisons: ComparisonNode[]
+  memberships: MembershipNode[]
+}
+
+const ORACLE_CASES = [
+  ['oracle_stale', 'stale'],
+  ['oracle_missing', 'missing'],
+  ['oracle_deviation_exceeded', 'deviation'],
+  ['oracle_paused', 'paused'],
+] as const
+
+// Deterministic XLM/USDC adjacency fixture; the shared registry can replace this boundary later.
+const ADJACENT_ASSETS = [
+  'CAS3J7GYLGXMF6TDJ5WQ2PEN4GRVNXJUIQ2TZU3ZB3OQ2V4DRCWI7WPF',
+  'CCWCLTASNDT57N3BCHOSVB5QWMV5URK4BXLDDF6ZZQYMBQ4OKZA3ZB2N',
+] as const
+
+/** Build deterministic model-evaluated alternatives without mutating the intended call. */
+export function generateCases(predicate: PredicateNode, permitCtx: EvalContext): GeneratedCases {
+  const facts = inspectPredicate(predicate)
+  const denies: DenyCase[] = []
+
+  for (const comparison of facts.comparisons) {
+    if (comparison.left.kind !== 'amount') continue
+    const mutated = mutateBigIntRecord(
+      permitCtx,
+      'amountByToken',
+      comparison.left.token,
+      comparison
+    )
+    if (mutated) denies.push({ dimension: 'amount', ctx: mutated })
+  }
+
+  const movedTokens = new Set<string>()
+  for (const comparison of facts.comparisons) {
+    if (comparison.left.kind === 'amount' || comparison.left.kind === 'window_spent') {
+      movedTokens.add(comparison.left.token)
+    }
+  }
+  for (const token of movedTokens) {
+    denies.push({ dimension: 'asset', ctx: mutateAsset(predicate, permitCtx, token) })
+  }
+
+  const contractConstraints = [
+    ...facts.comparisons.filter(
+      (node) =>
+        node.op === 'eq' &&
+        node.left.kind === 'call_contract' &&
+        node.right.kind === 'literal_address'
+    ),
+    ...facts.memberships.filter((node) => node.needle.kind === 'call_contract'),
+  ]
+  for (const _constraint of contractConstraints) {
+    const ctx = cloneContext(permitCtx)
+    ctx.contract = distinctText(permitCtx.contract, 'contract')
+    denies.push({ dimension: 'contract', ctx })
+  }
+
+  const functionConstraints = [
+    ...facts.comparisons.filter(
+      (node) =>
+        node.op === 'eq' && node.left.kind === 'call_fn' && node.right.kind === 'literal_symbol'
+    ),
+    ...facts.memberships.filter((node) => node.needle.kind === 'call_fn'),
+  ]
+  for (const _constraint of functionConstraints) {
+    const ctx = cloneContext(permitCtx)
+    ctx.fn = distinctText(permitCtx.fn, 'function')
+    denies.push({ dimension: 'function', ctx })
+  }
+
+  if (permitCtx.validUntilLedger !== undefined) {
+    const ctx = cloneContext(permitCtx)
+    ctx.atLedger = permitCtx.validUntilLedger + 1
+    denies.push({ dimension: 'timing', ctx })
+  }
+
+  for (const comparison of facts.comparisons) {
+    if (comparison.left.kind !== 'window_spent') continue
+    const mutated = mutateBigIntRecord(
+      permitCtx,
+      'windowSpentByToken',
+      comparison.left.token,
+      comparison,
+      false
+    )
+    if (mutated) denies.push({ dimension: 'time_window', ctx: mutated })
+  }
+
+  for (const comparison of facts.comparisons) {
+    if (comparison.left.kind !== 'invocation_count_in_window') continue
+    if (comparison.right.kind !== 'literal_u32') continue
+    const ctx = cloneContext(permitCtx)
+    ctx.invocationCountByWindow[comparison.left.windowSecs] = violatingNumber(
+      comparison.op,
+      comparison.right.value
+    )
+    denies.push({ dimension: 'invocation_count', ctx })
+  }
+
+  // Ordered numeric bound on a call_arg (e.g. a SoroSwap input-amount cap
+  // `call_arg[0] <= limit`). A violating deny case pushes the arg past the
+  // bound so the leaf is exercised - and, critically, so `minimize` keeps it:
+  // a conjunct that no deny case needs is pruned as redundant, which would
+  // silently drop a caller-requested restriction.
+  for (const comparison of facts.comparisons) {
+    if (comparison.op === 'eq' || comparison.left.kind !== 'call_arg') continue
+    const violating = violatingArgScVal(comparison.op, comparison.right)
+    if (!violating) continue
+    const ctx = cloneContext(permitCtx)
+    ctx.args[comparison.left.index] = violating
+    denies.push({ dimension: 'arg_amount_bound', ctx })
+  }
+
+  const argumentConstraints: Array<{ index: number }> = []
+  for (const comparison of facts.comparisons) {
+    if (
+      comparison.op === 'eq' &&
+      comparison.left.kind === 'call_arg' &&
+      comparison.right.kind !== 'literal_vec'
+    ) {
+      argumentConstraints.push({ index: comparison.left.index })
+    }
+  }
+  for (const membership of facts.memberships) {
+    if (membership.needle.kind === 'call_arg') {
+      argumentConstraints.push({ index: membership.needle.index })
+    }
+  }
+  for (const constraint of argumentConstraints) {
+    const ctx = cloneContext(permitCtx)
+    ctx.args[constraint.index] = { type: 'other', value: 'deny-case-opaque-argument' }
+    denies.push({ dimension: 'arg_bound', ctx })
+  }
+
+  const scopedArgumentIndices = new Set<number>(argumentConstraints.map(({ index }) => index))
+  for (const comparison of facts.comparisons) {
+    if (
+      comparison.op === 'eq' &&
+      comparison.left.kind === 'call_arg' &&
+      comparison.right.kind === 'literal_vec'
+    ) {
+      scopedArgumentIndices.add(comparison.left.index)
+    }
+  }
+  if (
+    contractConstraints.length > 0 &&
+    functionConstraints.length > 0 &&
+    scopedArgumentIndices.size > 0
+  ) {
+    const ctx = cloneContext(permitCtx)
+    ctx.contract = distinctText(permitCtx.contract, 'authorized-call-contract')
+    ctx.fn = distinctText(permitCtx.fn, 'authorized-call-function')
+    for (const index of scopedArgumentIndices) {
+      ctx.args[index] = { type: 'other', value: 'deny-case-authorized-call-argument' }
+    }
+    denies.push({ dimension: 'scope_contract_fn_arg', ctx })
+  }
+
+  const oracleComparisons = facts.comparisons.filter((node) => node.left.kind === 'oracle_price')
+  for (const [dimension, error] of ORACLE_CASES) {
+    for (const comparison of oracleComparisons) {
+      if (comparison.left.kind !== 'oracle_price') continue
+      const ctx = cloneContext(permitCtx)
+      ctx.oraclePriceByAsset[comparison.left.asset] = { error }
+      denies.push({ dimension, ctx })
+    }
+  }
+
+  for (const comparison of facts.comparisons) {
+    if (
+      comparison.op !== 'eq' ||
+      comparison.left.kind !== 'call_arg' ||
+      comparison.right.kind !== 'literal_vec'
+    ) {
+      continue
+    }
+    const ctx = cloneContext(permitCtx)
+    ctx.args[comparison.left.index] = differentVector(ctx.args[comparison.left.index])
+    denies.push({ dimension: 'soroswap_allowed_path', ctx })
+  }
+
+  // Version mismatch, malformed predicates, master authorization, and nonce replay are install-time checks and are intentionally omitted from model-evaluated cases.
+  return { permit: cloneContext(permitCtx), denies }
+}
+
+function inspectPredicate(predicate: PredicateNode): PredicateFacts {
+  const facts: PredicateFacts = { comparisons: [], memberships: [] }
+  visit(predicate, facts)
+  return facts
+}
+
+function visit(node: PredicateNode, facts: PredicateFacts): void {
+  switch (node.op) {
+    case 'and':
+    case 'or':
+      for (const child of node.children) visit(child, facts)
+      return
+    case 'not':
+      visit(node.child, facts)
+      return
+    case 'in':
+      facts.memberships.push(node)
+      return
+    case 'eq':
+    case 'lt':
+    case 'lte':
+    case 'gt':
+    case 'gte':
+      facts.comparisons.push(node)
+  }
+}
+
+function mutateBigIntRecord(
+  permitCtx: EvalContext,
+  recordKey: 'amountByToken' | 'windowSpentByToken',
+  token: string,
+  comparison: ComparisonNode,
+  preferScaledAmount = true
+): EvalContext | null {
+  if (comparison.right.kind !== 'literal_i128') return null
+
+  try {
+    const current = BigInt(permitCtx[recordKey][token] ?? '0')
+    const bound = BigInt(comparison.right.value)
+    const value = violatingBigInt(comparison.op, current, bound, preferScaledAmount)
+    const ctx = cloneContext(permitCtx)
+    ctx[recordKey][token] = value.toString()
+    return ctx
+  } catch {
+    return null
+  }
+}
+
+function violatingBigInt(
+  op: ComparisonOperator,
+  current: bigint,
+  bound: bigint,
+  preferScaledAmount: boolean
+): bigint {
+  if (preferScaledAmount) {
+    const scaledCandidates = [(current * 101n) / 100n, current * 10n]
+    for (const candidate of scaledCandidates) {
+      if (!bigIntComparison(op, candidate, bound)) return candidate
+    }
+  }
+
+  return boundaryViolatingBigInt(op, bound)
+}
+
+/** The single value at the boundary that violates `<selector> op bound`:
+ *  bound+1 for `lte`/`eq`, bound-1 for `gte`, and bound itself for the strict
+ *  `lt`/`gt`. */
+function boundaryViolatingBigInt(op: ComparisonOperator, bound: bigint): bigint {
+  switch (op) {
+    case 'lt':
+    case 'gt':
+      return bound
+    case 'lte':
+    case 'eq':
+      return bound + 1n
+    case 'gte':
+      return bound - 1n
+  }
+}
+
+function violatingNumber(op: ComparisonOperator, bound: number): number {
+  switch (op) {
+    case 'lt':
+    case 'gt':
+      return bound
+    case 'lte':
+    case 'eq':
+      return bound + 1
+    case 'gte':
+      return bound - 1
+  }
+}
+
+/** Build an ScVal that VIOLATES an ordered numeric bound on a call_arg, given
+ *  the comparison op and its numeric-literal right-hand side. Returns null when
+ *  the literal is not an integer (the bound is not a numeric compare). The arg
+ *  is emitted as an i128 - the recorder's numeric args are read via BigInt, so
+ *  the wire type only needs to be a numeric ScVal to exercise the bound. */
+function violatingArgScVal(op: ComparisonOperator, right: PredicateLeaf): ScVal | null {
+  const bound = literalNumericBigInt(right)
+  if (bound === null) return null
+  return { type: 'i128', value: boundaryViolatingBigInt(op, bound).toString() }
+}
+
+function bigIntComparison(op: ComparisonOperator, value: bigint, bound: bigint): boolean {
+  switch (op) {
+    case 'eq':
+      return value === bound
+    case 'lt':
+      return value < bound
+    case 'lte':
+      return value <= bound
+    case 'gt':
+      return value > bound
+    case 'gte':
+      return value >= bound
+  }
+}
+
+function mutateAsset(predicate: PredicateNode, permitCtx: EvalContext, token: string): EvalContext {
+  const binding = findAddressBinding(predicate, token)
+  const adjacent = adjacentAsset(token)
+  const ctx = cloneContext(permitCtx)
+  moveRecordEntry(ctx.amountByToken, token, adjacent)
+  moveRecordEntry(ctx.windowSpentByToken, token, adjacent)
+  if (binding.contract && ctx.contract === token) ctx.contract = adjacent
+  for (const index of binding.argumentIndices) {
+    const value = ctx.args[index]
+    if (value) ctx.args[index] = replaceAddress(value, token, adjacent)
+  }
+  return ctx
+}
+
+function findAddressBinding(
+  predicate: PredicateNode,
+  address: string
+): { contract: boolean; argumentIndices: Set<number> } {
+  const facts = inspectPredicate(predicate)
+  let contract = false
+  const argumentIndices = new Set<number>()
+
+  for (const comparison of facts.comparisons) {
+    if (comparison.op !== 'eq' || !leafContainsAddress(comparison.right, address)) continue
+    if (comparison.left.kind === 'call_contract') contract = true
+    if (comparison.left.kind === 'call_arg') argumentIndices.add(comparison.left.index)
+  }
+  for (const membership of facts.memberships) {
+    if (!membership.haystack.some((leaf) => leafContainsAddress(leaf, address))) continue
+    if (membership.needle.kind === 'call_contract') contract = true
+    if (membership.needle.kind === 'call_arg') argumentIndices.add(membership.needle.index)
+  }
+
+  return { contract, argumentIndices }
+}
+
+function leafContainsAddress(leaf: PredicateLeaf, address: string): boolean {
+  if (leaf.kind === 'literal_address') return leaf.value === address
+  if (leaf.kind === 'literal_vec') {
+    return leaf.elements.some((element) => leafContainsAddress(element, address))
+  }
+  return false
+}
+
+function moveRecordEntry(record: Record<string, string>, from: string, to: string): void {
+  const value = record[from]
+  if (value === undefined) return
+  delete record[from]
+  record[to] = value
+}
+
+function replaceAddress(value: ScVal, from: string, to: string): ScVal {
+  if (value.type === 'address') {
+    return value.value === from ? { type: 'address', value: to } : { ...value }
+  }
+  if (value.type === 'vec') {
+    return { type: 'vec', value: value.value.map((item) => replaceAddress(item, from, to)) }
+  }
+  return { ...value }
+}
+
+function differentVector(actual: ScVal | undefined): ScVal {
+  if (actual?.type !== 'vec') return { type: 'vec', value: [] }
+  if (actual.value.length === 0) {
+    return { type: 'vec', value: [{ type: 'other', value: 'deny-case-extra-hop' }] }
+  }
+  if (actual.value.length > 1) {
+    const reversed = actual.value.map(cloneScVal).reverse()
+    if (JSON.stringify(reversed) !== JSON.stringify(actual.value)) {
+      return { type: 'vec', value: reversed }
+    }
+  }
+  const value = actual.value.map(cloneScVal)
+  value[0] = { type: 'other', value: 'deny-case-different-hop' }
+  return { type: 'vec', value }
+}
+
+function adjacentAsset(asset: string): string {
+  return asset === ADJACENT_ASSETS[0] ? ADJACENT_ASSETS[1] : ADJACENT_ASSETS[0]
+}
+
+function distinctText(value: string, label: string): string {
+  return `${value}#${label}`
+}
+
+function cloneContext(ctx: EvalContext): EvalContext {
+  const cloned: EvalContext = {
+    contract: ctx.contract,
+    fn: ctx.fn,
+    args: ctx.args.map(cloneScVal),
+    atLedger: ctx.atLedger,
+    nowSeconds: ctx.nowSeconds,
+    amountByToken: { ...ctx.amountByToken },
+    windowSpentByToken: { ...ctx.windowSpentByToken },
+    invocationCountByWindow: { ...ctx.invocationCountByWindow },
+    oraclePriceByAsset: Object.fromEntries(
+      Object.entries(ctx.oraclePriceByAsset).map(([asset, entry]) => [
+        asset,
+        'error' in entry ? { error: entry.error } : { ...entry },
+      ])
+    ),
+  }
+  if (ctx.validUntilLedger !== undefined) cloned.validUntilLedger = ctx.validUntilLedger
+  if (ctx.signerWeights !== undefined) cloned.signerWeights = { ...ctx.signerWeights }
+  return cloned
+}
+
+function cloneScVal(value: ScVal): ScVal {
+  if (value.type === 'vec') {
+    return { type: 'vec', value: value.value.map(cloneScVal) }
+  }
+  return { ...value }
+}