127 lines
3.7 KiB
Python
127 lines
3.7 KiB
Python
#!/usr/bin/env python3
|
|
import argparse
|
|
import base64
|
|
import hashlib
|
|
import json
|
|
import subprocess
|
|
import sys
|
|
import time
|
|
from pathlib import Path
|
|
|
|
|
|
SAMPLE_B64 = (
|
|
"REtJRgAAIABWUDkwQABAAAEAAAABAAAAAgAAAAAAAABeAAAAAAAAAAAAAACCSYNCAAPwA/YGOCQcGEoAACBAAGtD///lXb23/SskhXr7zdPyoCRyEjNuPymkNJQgETBR424BCv//rXCHLKdpldqOXFZdaWk1nVibjsmAd3pGejzlO0+dlygBOCSA/wAAAAEAAAAAAAAAgkmDQgAD8f/2BjgkHBhKAADQR9j9Ye4xQAev+/8OAGxOd+f8niRqQFa1U/7kzgammYg1AcYQFrhfX6tE38imv1MXtaAO/yiEiKaaDpaMxLBBYGTZ80JtDb8+6GWkt+fdDLSW/PuhlpLfn3Qy0lvz7oZaS3590MtJb8+6GWkt+fdDLSW/PuhlpLfn3Qy0lvz7oZaS3590MtJb8+6GWkt+fdDLSW/PuhlpLfn3Qy0lvz7oZaS3590MtJb8+6GWkt+fdDLSW/PuhlpLfn3Qy0lvz7oZaS3590MtJb8+6GWkt+fdDLSW/PuhlpLfn3Qy0lvz7oZaS3590MtJb8+6GUoA"
|
|
)
|
|
|
|
EXPECTED_SHA256 = "F26BDEFBDFD0B44359E314E0BFDE7AEA979D29F80F598749DCCA68AB34F54649"
|
|
|
|
CRASH_CODES = {
|
|
0xC0000005: "access_violation",
|
|
0xC0000374: "heap_corruption",
|
|
0xC0000409: "stack_buffer_overrun",
|
|
}
|
|
|
|
|
|
def code32(value):
|
|
if value is None:
|
|
return None
|
|
return value & 0xFFFFFFFF
|
|
|
|
|
|
def classify_returncode(value):
|
|
code = code32(value)
|
|
if code is None:
|
|
return "timeout"
|
|
if code == 0:
|
|
return "clean"
|
|
if code in CRASH_CODES:
|
|
return f"crash:{CRASH_CODES[code]}"
|
|
return "nonzero"
|
|
|
|
|
|
def write_sample(path):
|
|
data = base64.b64decode(SAMPLE_B64)
|
|
digest = hashlib.sha256(data).hexdigest().upper()
|
|
if digest != EXPECTED_SHA256:
|
|
raise RuntimeError(f"embedded sample hash mismatch: {digest}")
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
path.write_bytes(data)
|
|
return data
|
|
|
|
|
|
def run_vlc(vlc, sample, timeout):
|
|
cmd = [
|
|
str(vlc),
|
|
"-I",
|
|
"dummy",
|
|
"--dummy-quiet",
|
|
"--ignore-config",
|
|
"--no-media-library",
|
|
"--play-and-exit",
|
|
"--run-time",
|
|
"2",
|
|
"--no-one-instance",
|
|
"--no-qt-privacy-ask",
|
|
"--no-qt-error-dialogs",
|
|
"--no-crashdump",
|
|
"--no-audio",
|
|
"--vout",
|
|
"dummy",
|
|
str(sample),
|
|
"vlc://quit",
|
|
]
|
|
started = time.time()
|
|
try:
|
|
proc = subprocess.run(
|
|
cmd,
|
|
cwd=str(vlc.parent),
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
timeout=timeout,
|
|
)
|
|
returncode = proc.returncode
|
|
stdout = proc.stdout.decode("utf-8", "replace")
|
|
stderr = proc.stderr.decode("utf-8", "replace")
|
|
except subprocess.TimeoutExpired as exc:
|
|
returncode = None
|
|
stdout = (exc.stdout or b"").decode("utf-8", "replace")
|
|
stderr = (exc.stderr or b"").decode("utf-8", "replace")
|
|
return {
|
|
"status": classify_returncode(returncode),
|
|
"returncode": returncode,
|
|
"returncode_hex": f"0x{code32(returncode):08x}" if returncode is not None else None,
|
|
"elapsed": round(time.time() - started, 3),
|
|
"stdout_tail": stdout[-2000:],
|
|
"stderr_tail": stderr[-2000:],
|
|
}
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
"-o",
|
|
"--output",
|
|
type=Path,
|
|
default=Path("vp9_reschange_64x64_to_64x8192_tc0.ivf"),
|
|
)
|
|
parser.add_argument("--vlc", type=Path, help="optional path to vlc.exe for local replay")
|
|
parser.add_argument("--timeout", type=float, default=8)
|
|
args = parser.parse_args()
|
|
|
|
data = write_sample(args.output)
|
|
result = {
|
|
"sample": str(args.output.resolve()),
|
|
"sha256": hashlib.sha256(data).hexdigest().upper(),
|
|
"size": len(data),
|
|
}
|
|
if args.vlc:
|
|
result["vlc"] = run_vlc(args.vlc.resolve(), args.output.resolve(), args.timeout)
|
|
print(json.dumps(result, indent=2))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
main()
|
|
except KeyboardInterrupt:
|
|
sys.exit(130)
|