cognium-dev 3.124.0 → 3.125.0

package/dist/cli.js

@@ -22604,6 +22604,9 @@ class LanguageSourcesPass {
     if (language === "python") {
       additionalSanitizers.push(...findPythonNetlocAllowlistGuardSanitizers(code));
       additionalSanitizers.push(...findPythonRangeCheckGuardSanitizers(code));
+      additionalSanitizers.push(...findPythonRegexAllowlistWrapperSanitizers(code));
+      additionalSanitizers.push(...findPythonSetMembershipXssGuardSanitizers(code));
+      additionalSanitizers.push(...findPythonDefusedXmlSanitizers(code));
       const pyMisconfigFindings = findPythonPatternFindings(code, graph.ir.meta.file);
       for (const finding of pyMisconfigFindings) {
         ctx.addFinding(finding);
@@ -24008,6 +24011,252 @@ function findPythonRangeCheckGuardSanitizers(code) {
   }
   return sanitizers;
 }
+function findPythonRegexAllowlistWrapperSanitizers(code) {
+  const sanitizers = [];
+  const lines = code.split(`
+`);
+  const defOpen = /^(\s*)def\s+([A-Za-z_][A-Za-z0-9_]*)\s*\(\s*([A-Za-z_][A-Za-z0-9_]*)\s*\)\s*:\s*$/;
+  const tightRegex = /r?["'](?:\^)?(?:\\w[+*]|\\d[+*]|\[[A-Za-z0-9_\\\-\.\s]+\][+*]|[A-Za-z0-9_\-\.]+)(?:\$)?["']/;
+  const terminator = /\b(return\s+(?:None|"|'|\(|\[|\{|False|jsonify|make_response|redirect|abort)|raise\s+\w|abort\s*\(|sys\.exit\s*\()/;
+  const wrappers = [];
+  for (let i2 = 0;i2 < lines.length; i2++) {
+    const m = defOpen.exec(lines[i2]);
+    if (!m)
+      continue;
+    const defIndent = m[1].length;
+    const wrapperName = m[2];
+    const argName = m[3];
+    let foundGuard = false;
+    let returnsArg = false;
+    let blockEnd = -1;
+    const maxScan = Math.min(lines.length, i2 + 40);
+    for (let j = i2 + 1;j < maxScan; j++) {
+      const line = lines[j];
+      if (line.trim() === "")
+        continue;
+      const indent = line.length - line.trimStart().length;
+      if (indent <= defIndent) {
+        blockEnd = j - 1;
+        break;
+      }
+      if (!foundGuard) {
+        const guard = new RegExp(`if\\s+not\\s+re\\.(?:fullmatch|match)\\s*\\(\\s*(${tightRegex.source})\\s*,\\s*${argName}\\s*\\)\\s*:`);
+        if (guard.test(line)) {
+          for (let k = j + 1;k < maxScan; k++) {
+            const klin = lines[k];
+            if (klin.trim() === "")
+              continue;
+            const kindent = klin.length - klin.trimStart().length;
+            if (kindent <= indent)
+              break;
+            if (terminator.test(klin))
+              foundGuard = true;
+            break;
+          }
+        }
+      }
+      if (new RegExp(`^\\s*return\\s+${argName}\\s*$`).test(line)) {
+        returnsArg = true;
+      }
+    }
+    if (blockEnd === -1)
+      blockEnd = Math.min(lines.length - 1, i2 + 39);
+    if (!foundGuard || !returnsArg)
+      continue;
+    wrappers.push({ name: wrapperName, defLine: i2, defIndent });
+  }
+  if (wrappers.length === 0)
+    return sanitizers;
+  const wrapperKinds = [
+    "ldap_injection",
+    "xpath_injection",
+    "sql_injection",
+    "command_injection",
+    "path_traversal",
+    "xss",
+    "external_taint_escape"
+  ];
+  for (const w of wrappers) {
+    const assignedCallRe = new RegExp(`^(\\s*)([A-Za-z_][A-Za-z0-9_]*)\\s*=\\s*${w.name}\\s*\\(`);
+    const bareCallRe = new RegExp(`\\b${w.name}\\s*\\(`);
+    for (let i2 = 0;i2 < lines.length; i2++) {
+      if (i2 === w.defLine)
+        continue;
+      const line = lines[i2];
+      if (!bareCallRe.test(line))
+        continue;
+      if (i2 > w.defLine) {
+        const lineIndent = line.length - line.trimStart().length;
+        if (lineIndent > w.defIndent) {
+          let stillInside = true;
+          for (let k = w.defLine + 1;k < i2; k++) {
+            const kline = lines[k];
+            if (kline.trim() === "")
+              continue;
+            const kindent = kline.length - kline.trimStart().length;
+            if (kindent <= w.defIndent) {
+              stillInside = false;
+              break;
+            }
+          }
+          if (stillInside)
+            continue;
+        }
+      }
+      sanitizers.push({
+        type: "python_regex_allowlist_wrapper",
+        method: w.name,
+        line: i2 + 1,
+        sanitizes: [...wrapperKinds]
+      });
+      const am = assignedCallRe.exec(line);
+      if (!am)
+        continue;
+      const callIndent = am[1].length;
+      const validated = am[2];
+      const validatedRe = new RegExp(`\\b${validated}\\b`);
+      for (let j = i2 + 1;j < lines.length; j++) {
+        const jline = lines[j];
+        if (jline.trim() === "")
+          continue;
+        const jindent = jline.length - jline.trimStart().length;
+        if (jindent < callIndent)
+          break;
+        if (validatedRe.test(jline)) {
+          sanitizers.push({
+            type: "python_regex_allowlist_wrapper",
+            method: w.name,
+            line: j + 1,
+            sanitizes: [...wrapperKinds]
+          });
+        }
+      }
+    }
+  }
+  return sanitizers;
+}
+function findPythonSetMembershipXssGuardSanitizers(code) {
+  const sanitizers = [];
+  const lines = code.split(`
+`);
+  const guardOpen = /^(\s*)if\s+([A-Za-z_][A-Za-z0-9_]*)\s+not\s+in\s+([A-Za-z_][A-Za-z0-9_]*)\s*:\s*$/;
+  const allowlistName = /^(?:[A-Z][A-Z0-9_]+|.*?(allowed|accepted|whitelist|permitted|valid|approved).*)$/i;
+  const terminator = /\b(return|raise|abort\s*\(|sys\.exit\s*\()/;
+  for (let i2 = 0;i2 < lines.length; i2++) {
+    const m = guardOpen.exec(lines[i2]);
+    if (!m)
+      continue;
+    const guardIndent = m[1].length;
+    const guardedVar = m[2];
+    const allowName = m[3];
+    if (!allowlistName.test(allowName))
+      continue;
+    let bodyHasTerminator = false;
+    let blockEnd = -1;
+    const maxScan = Math.min(lines.length, i2 + 26);
+    for (let j = i2 + 1;j < maxScan; j++) {
+      const line = lines[j];
+      if (line.trim() === "")
+        continue;
+      const indent = line.length - line.trimStart().length;
+      if (indent <= guardIndent) {
+        blockEnd = j - 1;
+        break;
+      }
+      if (terminator.test(line))
+        bodyHasTerminator = true;
+    }
+    if (blockEnd === -1)
+      blockEnd = Math.min(lines.length - 1, i2 + 25);
+    if (!bodyHasTerminator)
+      continue;
+    const varRe = new RegExp(`\\b${guardedVar}\\b`);
+    for (let l = blockEnd + 2;l <= lines.length; l++) {
+      const lineText = lines[l - 1];
+      if (!varRe.test(lineText))
+        continue;
+      sanitizers.push({
+        type: "python_set_membership_xss_guard",
+        method: "if",
+        line: l,
+        sanitizes: ["xss", "external_taint_escape"]
+      });
+    }
+  }
+  return sanitizers;
+}
+function findPythonDefusedXmlSanitizers(code) {
+  const sanitizers = [];
+  const lines = code.split(`
+`);
+  const moduleAliases = new Set;
+  const fromNames = new Set;
+  const importAs = /^\s*import\s+defusedxml(?:\.[A-Za-z_][A-Za-z0-9_.]*)?\s+as\s+([A-Za-z_][A-Za-z0-9_]*)\s*$/;
+  const importBare = /^\s*import\s+defusedxml(?:\.([A-Za-z_][A-Za-z0-9_.]*))?\s*$/;
+  const fromImport = /^\s*from\s+defusedxml(?:\.[A-Za-z_][A-Za-z0-9_.]*)?\s+import\s+(.+)$/;
+  for (const raw of lines) {
+    const line = raw.replace(/#.*$/, "");
+    let m;
+    if (m = importAs.exec(line)) {
+      moduleAliases.add(m[1]);
+      continue;
+    }
+    if (m = importBare.exec(line)) {
+      if (m[1]) {
+        moduleAliases.add(m[1].split(".").pop());
+      }
+      moduleAliases.add("defusedxml");
+      continue;
+    }
+    if (m = fromImport.exec(line)) {
+      const items = m[1].split(",");
+      for (const item of items) {
+        const trimmed = item.trim().replace(/[()\\]/g, "");
+        if (!trimmed)
+          continue;
+        const asMatch = /^([A-Za-z_][A-Za-z0-9_]*)\s+as\s+([A-Za-z_][A-Za-z0-9_]*)$/.exec(trimmed);
+        if (asMatch) {
+          fromNames.add(asMatch[2]);
+        } else if (/^[A-Za-z_][A-Za-z0-9_]*$/.test(trimmed)) {
+          fromNames.add(trimmed);
+        }
+      }
+    }
+  }
+  if (moduleAliases.size === 0 && fromNames.size === 0)
+    return sanitizers;
+  for (let i2 = 0;i2 < lines.length; i2++) {
+    const line = lines[i2];
+    if (/^\s*(?:import|from)\s+/.test(line))
+      continue;
+    let matched = false;
+    for (const alias of moduleAliases) {
+      const re = new RegExp(`\\b${alias}\\s*\\.\\s*[A-Za-z_][A-Za-z0-9_]*\\s*\\(`);
+      if (re.test(line)) {
+        matched = true;
+        break;
+      }
+    }
+    if (!matched) {
+      for (const name2 of fromNames) {
+        const re = new RegExp(`\\b${name2}\\s*\\(`);
+        if (re.test(line)) {
+          matched = true;
+          break;
+        }
+      }
+    }
+    if (!matched)
+      continue;
+    sanitizers.push({
+      type: "python_defusedxml_import",
+      method: "defusedxml",
+      line: i2 + 1,
+      sanitizes: ["xxe", "external_taint_escape"]
+    });
+  }
+  return sanitizers;
+}
 function findRustSetAllowlistGuardSanitizers(code) {
   const sanitizers = [];
   const lines = code.split(`
@@ -37779,7 +38028,7 @@ var colors = {
 };
 
 // src/version.ts
-var version = "3.124.0";
+var version = "3.125.0";
 
 // src/formatters.ts
 var LIBRARY_API_SURFACE_TAG2 = "library-api-surface:caller-responsibility";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cognium-dev",
-  "version": "3.124.0",
+  "version": "3.125.0",
   "description": "Static Application Security Testing CLI for detecting security vulnerabilities via taint tracking",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -66,7 +66,7 @@
   },
   "dependencies": {
     "@cognium/project-profile-detect": "^1.1.0",
-    "circle-ir": "^3.124.0"
+    "circle-ir": "^3.125.0"
   },
   "devDependencies": {
     "@types/node": "^25.5.0",