AnyPS5/tools/relinker_fuzz_seeds.py

import argparse
import importlib
import inspect
import sys
from pathlib import Path

ROOT = Path(__file__).resolve().parent.parent
TESTS = ROOT / "core" / "relinker" / "relinker" / "tests"
BUILDERS = ("fixture", "image", "make_image", "tls_image", "padded_image", "executable_image",
            "empty_tls_image", "make_eof_segment_prefix_image")
SELECTORS = (0, 1, 2, 4)


def without_arguments(function):
    for parameter in inspect.signature(function).parameters.values():
        positional = parameter.kind in (parameter.POSITIONAL_ONLY, parameter.POSITIONAL_OR_KEYWORD)
        if positional and parameter.default is inspect.Parameter.empty:
            return False
    return True


def images():
    sys.path.insert(0, str(TESTS))
    for path in sorted(TESTS.glob("test_*.py")):
        try:
            module = importlib.import_module(path.stem)
        except Exception as error:
            print(f"warning: no seeds from {path.name}: it cannot be imported: {error}", file=sys.stderr)
            continue
        for name in BUILDERS:
            function = getattr(module, name, None)
            if not callable(function) or not without_arguments(function):
                continue
            try:
                data = function()
            except Exception as error:
                print(f"warning: no seed from {path.stem}.{name}: {error}", file=sys.stderr)
                continue
            if isinstance(data, (bytes, bytearray)) and len(data) >= 64:
                yield f"{path.stem}-{name}", bytes(data)


def main():
    parser = argparse.ArgumentParser(description="Write relinker test fixtures as relinker_conversion_fuzzer seeds")
    parser.add_argument("output", type=Path)
    arguments = parser.parse_args()
    arguments.output.mkdir(parents=True, exist_ok=True)
    count = 0
    for name, data in images():
        for selector in SELECTORS:
            (arguments.output / f"{name}-{selector}").write_bytes(bytes([selector]) + data)
            count += 1
    print(f"{count} seeds written to {arguments.output}")


if __name__ == "__main__":
    main()