Sanitize outputs on apu-memory-tuner

Author: danielholanda

skills/apu-memory-tuner/scripts/apply_profile.py

@@ -61,6 +61,8 @@
 from dataclasses import dataclass
 from pathlib import Path
 
+from _sanitize import sanitize_process_output
+
 PAGE_SIZE_BYTES = 4096
 MIN_VRAM_GB = 0.5    # Floor most BIOSes allow for the UMA frame buffer.
 MIN_GTT_GB = 1.0     # Below this, even routine GPU work fails.
@@ -83,9 +85,13 @@ def _run(cmd: list[str], timeout: float = 60.0) -> tuple[int, str, str]:
         r = subprocess.run(
             cmd, capture_output=True, text=True, timeout=timeout, check=False,
         )
-        return r.returncode, r.stdout or "", r.stderr or ""
+        return (
+            r.returncode,
+            sanitize_process_output(r.stdout),
+            sanitize_process_output(r.stderr),
+        )
     except (FileNotFoundError, subprocess.SubprocessError, OSError) as e:
-        return 127, "", str(e)
+        return 127, "", sanitize_process_output(str(e))
 
 
 def _read_text(path: str) -> str:

skills/apu-memory-tuner/scripts/detect_platform.py

@@ -36,6 +36,8 @@
 from dataclasses import asdict, dataclass, field
 from pathlib import Path
 
+from _sanitize import sanitize_process_output
+
 # Mapping from LLVM gfx target to the marketing-friendly generation bucket
 # the rest of the skill keys off. RDNA3.5 (gfx115x) is the only generation
 # the source ROCm doc describes shared-memory tuning for, so it is the only
@@ -91,7 +93,11 @@ def _run(cmd: list[str], timeout: float = 5.0) -> tuple[int, str, str]:
         r = subprocess.run(
             cmd, capture_output=True, text=True, timeout=timeout, check=False,
         )
-        return r.returncode, r.stdout or "", r.stderr or ""
+        return (
+            r.returncode,
+            sanitize_process_output(r.stdout),
+            sanitize_process_output(r.stderr),
+        )
     except (FileNotFoundError, subprocess.SubprocessError, OSError):
         return 127, "", ""
 

skills/apu-memory-tuner/scripts/show_config.py

@@ -36,6 +36,8 @@
 from dataclasses import asdict, dataclass
 from pathlib import Path
 
+from _sanitize import sanitize_process_output
+
 # 4 KiB is the page size assumed by the amdgpu/TTM accounting on every
 # platform we target. The kernel param is in pages, the user thinks in GB.
 PAGE_SIZE_BYTES = 4096
@@ -64,7 +66,11 @@ def _run(cmd: list[str], timeout: float = 10.0) -> tuple[int, str, str]:
         r = subprocess.run(
             cmd, capture_output=True, text=True, timeout=timeout, check=False,
         )
-        return r.returncode, r.stdout or "", r.stderr or ""
+        return (
+            r.returncode,
+            sanitize_process_output(r.stdout),
+            sanitize_process_output(r.stderr),
+        )
     except (FileNotFoundError, subprocess.SubprocessError, OSError):
         return 127, "", ""