#!/usr/bin/env bash
# CWE-407 scan — Apache Ant (Java build tool)
# Old Java code (~2000s); high probability of ArrayList/Vector for visited sets
# Strategy: generic Java patterns — List.contains, ArrayList membership in loops
set -euo pipefail
echo "# scan=ant host=$(hostname) date=$(date -u +%Y-%m-%dT%H:%M:%SZ)"

cd /tmp && git clone --depth 1 https://github.com/apache/ant ant 2>&1 | tail -1

# O(n) membership patterns in Java
JAVA_MEM='\.contains\s*\(|\.indexOf\s*\(|Collections\.frequency\s*\('

# Loop constructs
JAVA_LOOP='for\s*\(|while\s*\(|\.forEach\s*\(|\.iterator\s*\('

for dir in \
  src/main/org/apache/tools/ant \
  src/main/org/apache/tools/ant/taskdefs \
  src/main/org/apache/tools/ant/types; do
  [ -d /tmp/ant/$dir ] || continue
  echo "# roots: /tmp/ant/$dir"
  find /tmp/ant/$dir -name "*.java" 2>/dev/null | sort | while IFS= read -r f; do
    ml=$(grep -nE "$JAVA_MEM" "$f" 2>/dev/null | cut -d: -f1 || true)
    [ -z "$ml" ] && continue
    while IFS= read -r ln; do
      s=$(( ln > 20 ? ln - 20 : 1 )); e=$(( ln + 20 ))
      ctx=$(sed -n "${s},${e}p" "$f" 2>/dev/null || true)
      # Only flag if context mentions graph/dag/dep/order/sort/visit/cycle/topo
      graph_hit=$(echo "$ctx" | grep -iE 'graph|depend|topo|visit|cycle|dag|sort|stack|order|reachab|ancestor|transit' | head -1 | sed 's/^\s*//' || true)
      [ -z "$graph_hit" ] && continue
      loop_hit=$(echo "$ctx" | grep -iE "$JAVA_LOOP" | head -1 | sed 's/^\s*//' || true)
      [ -n "$loop_hit" ] && {
        mem_line=$(sed -n "${ln}p" "$f" | sed 's/^\s*//')
        echo "CANDIDATE\t${f}:${ln}\t${mem_line}  [loop: ${loop_hit}]"
      }
    done <<< "$ml"
  done
done
echo "# scan complete"