fix: release COUNT expression values

2026-09-24 17:34:46 +00:00
parent 4f0219dd11
commit eca642d8d8
3 changed files with 54 additions and 2 deletions
@@ -0,0 +1,52 @@
+#!/usr/bin/env python3
+"""Regression check for discarded COUNT expression Values."""
+from pathlib import Path
+import os
+import subprocess
+import sys
+import tempfile
+
+ROOT = Path(__file__).resolve().parent.parent
+
+
+def main() -> int:
+    if os.name == "nt":
+        print("SKIP: LeakSanitizer check is POSIX-only")
+        return 0
+    with tempfile.TemporaryDirectory(prefix="claro-count-cleanup-") as tmp:
+        tmp_path = Path(tmp)
+        binary = tmp_path / "claro-lsan"
+        script = tmp_path / "count_expressions.claro"
+        script.write_text(
+            'SET items TO LIST\n'
+            + ''.join('COUNT items AS item_count\n' for _ in range(2000))
+            + 'SAY item_count\n', encoding="utf-8"
+        )
+        build = subprocess.run(
+            ["gcc", "-std=c99", "-O0", "-g", "-fsanitize=address,undefined",
+             "src/claro.c", "-o", str(binary), "-lm"],
+            cwd=ROOT, text=True, capture_output=True,
+        )
+        if build.returncode:
+            print(build.stdout, end="")
+            print(build.stderr, end="", file=sys.stderr)
+            return build.returncode
+        run = subprocess.run(
+            [str(binary), str(script)], cwd=ROOT, text=True, capture_output=True,
+            env={**os.environ, "ASAN_OPTIONS": "detect_leaks=1"},
+        )
+        if run.returncode or "LeakSanitizer" in run.stderr or "SUMMARY:" in run.stderr:
+            print("FAIL: discarded COUNT expression values still leak")
+            print(run.stdout, end="")
+            print(run.stderr, end="", file=sys.stderr)
+            return 1
+        if run.stdout != "0\n":
+            print("FAIL: COUNT cleanup probe produced the wrong output")
+            print(run.stdout, end="", file=sys.stderr)
+            return 1
+        print("PASS: discarded COUNT expression values are released")
+        return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())