53 lines
3.2 KiB
Python
53 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
"""Test native telemetry storage/acknowledgments with ASan/UBSan, without the VM.
|
|
|
|
Uses deps/json/single_include/nlohmann/json.hpp if restored, or --json-header
|
|
pointing to the single header at the revision in the plugin's upstream.json.
|
|
"""
|
|
import argparse
|
|
import os
|
|
from pathlib import Path
|
|
import re
|
|
import shutil
|
|
import subprocess
|
|
import tempfile
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument("--json-header", help="Header path; {game} expands to t5/t6 for their respective pinned headers")
|
|
args = parser.parse_args()
|
|
|
|
with tempfile.TemporaryDirectory(prefix="tws-telemetry-") as temp:
|
|
directory = Path(temp)
|
|
for game in ("t5", "t6"):
|
|
plugin = ROOT / game / "plugins" / f"tws-{game}-gsc-utils"
|
|
header = Path(args.json_header.format(game=game)) if args.json_header else plugin / "deps/json/single_include/nlohmann/json.hpp"
|
|
if not header.is_file():
|
|
parser.error(f"Missing {header}; restore dependencies or pass --json-header")
|
|
shutil.copyfile(header, directory / "json.hpp")
|
|
(directory / "stdinc.hpp").write_text("#include <algorithm>\n#include <fstream>\n#include <sstream>\n#include <json.hpp>\n")
|
|
executable = directory / game
|
|
subprocess.run([os.environ.get("CXX", "c++"), "-std=c++20", "-Wall", "-Wextra", "-Werror",
|
|
"-fsanitize=address,undefined", "-I", str(directory), "-I", str(plugin / "src/utils"),
|
|
str(plugin / "src/utils/telemetry.cpp"), str(ROOT / "tools/tests/telemetry_cases.cpp"),
|
|
"-o", str(executable)], check=True)
|
|
work = directory / f"{game}-data"
|
|
subprocess.run([str(executable), str(work)], check=True)
|
|
# Separate OS processes prove snapshots/identities don't depend on VM or
|
|
# DLL memory. A newly booted producer recovers before its own start.
|
|
subprocess.run([str(executable), str(work), "produce"], check=True)
|
|
subprocess.run([str(executable), str(work), "recover"], check=True)
|
|
subprocess.run([str(executable), str(work), "empty"], check=True)
|
|
|
|
# Compile the actual registration adapter as well, substituting only
|
|
# engine interfaces. This does not validate the Windows ABI/VM linker.
|
|
stub = (ROOT / "tools/tests/http_component_stub.hpp").read_text()
|
|
stub = stub.replace("std::vector<script_value> get_keys() const", "void emplace_back(const script_value& value) { (*data)[std::to_string(data->size())] = value; }\n std::vector<script_value> get_keys() const")
|
|
(directory / "component_stub.hpp").write_text(stub)
|
|
component = re.sub(r'^#include[^\n]*\n', '', (plugin / "src/component/telemetry.cpp").read_text(), flags=re.MULTILINE)
|
|
source = directory / "component.cpp"
|
|
source.write_text('#include "component_stub.hpp"\n#include "telemetry.hpp"\n' + component)
|
|
subprocess.run([os.environ.get("CXX", "c++"), "-std=c++20", "-Wall", "-Wextra", "-Werror", "-fsyntax-only",
|
|
"-I", str(directory), "-I", str(plugin / "src/utils"), str(source)], check=True)
|
|
print(f"PASS {game}: native storage, acknowledgment, process restart and component checks", flush=True)
|