mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
3fe3a8524a
Add the `revng mass-testing` commands and auxiliary executables which simplify running revng on a huge quantity of input executables.
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
import argparse
|
|
import os
|
|
import time
|
|
from pathlib import Path
|
|
from subprocess import run
|
|
|
|
import yaml
|
|
|
|
from revng.internal.cli.support import get_root
|
|
|
|
from ...commands_registry import Command, Options
|
|
|
|
|
|
class MassTestingRunCommand(Command):
|
|
def __init__(self):
|
|
super().__init__(("mass-testing", "run"), "Run a mass-testing configuration")
|
|
|
|
def register_arguments(self, parser: argparse.ArgumentParser):
|
|
parser.description = "Run a mass-testing configuration"
|
|
parser.add_argument("build_dir", help="Build directory")
|
|
|
|
def run(self, options: Options):
|
|
args = options.parsed_args
|
|
|
|
meta_file = Path(args.build_dir) / "meta.yml"
|
|
with open(meta_file) as f:
|
|
meta_data = yaml.safe_load(f)
|
|
if "cpu_count" not in meta_data:
|
|
meta_data["cpu_count"] = len(os.sched_getaffinity(0))
|
|
if "start_time" not in meta_data:
|
|
meta_data["start_time"] = time.time()
|
|
with open(meta_file, "w") as f:
|
|
yaml.safe_dump(meta_data, f)
|
|
|
|
new_env = os.environ.copy()
|
|
bin_dir = get_root() / "libexec/revng/mass-testing"
|
|
new_env["PATH"] = f"{bin_dir.resolve()!s}:{new_env['PATH']}"
|
|
|
|
run(["ninja", "--quiet", "-k0", "-C", args.build_dir], env=new_env, check=True)
|