run_command module

Author: joyguoguo

fuzz/command_util.py

@@ -0,0 +1,42 @@
+# command_util.py
+
+import subprocess
+import logging
+from returns.maybe import Maybe
+from errors import CommandError
+from pathlib import Path
+
+def run_command(
+    cmd: str,
+    oss_fuzz_dir: Path,
+    project: str = "",
+    allowed_exit_codes: Maybe[list[int]] = Maybe.empty,
+    skip_yes: bool = False
+) -> int:
+    """Execute a command and return the exit code (no stdout/stderr capture)"""
+    allowed_codes = allowed_exit_codes.value_or([0])
+    cmd_str = f"yes | {cmd}" if not skip_yes else cmd
+    logging.debug(f"Executing command [{project}]: {cmd_str}")
+
+    try:
+        process = subprocess.Popen(
+            cmd_str,
+            shell=True,
+            cwd=str(oss_fuzz_dir)
+        )
+        exit_code = process.wait()
+
+        if exit_code in allowed_codes:
+            return exit_code
+
+        error_msg = f"Command failed (exit code: {exit_code})"
+        if project:
+            error_msg += f" for project: {project}"
+        raise CommandError(error_msg, project=project, exit_code=exit_code)
+
+    except FileNotFoundError as e:
+        raise CommandError(f"Command not found: {cmd.split()[0]}", project=project) from e
+    except OSError as e:
+        raise CommandError(f"System error: {e}", project=project) from e
+    except subprocess.SubprocessError as e:
+        raise CommandError(f"Subprocess error: {e}", project=project) from e