Katalog CVE

CVE-2026-31253

Wysokie
Opublikowano: Przetłumaczono: NVD NIST

Streszczenie

Framework do treningu flash-attention zawiera podatność na niebezpieczną deserializację w mechanizmie ładowania punktów kontrolnych. Funkcja load_checkpoint() oraz kod ładowania punktów kontrolnych w eval.py używają torch.load() bez włączenia parametru weights_only=True, co umożliwia deserializację dowolnych obiektów Pythona.

Ocena ryzyka

Atakujący może wykorzystać tę podatność, dostarczając złośliwie skonstruowany plik punktu kontrolnego, co prowadzi do wykonania dowolnego kodu na systemie ofiary. To stwarza poważne zagrożenie dla bezpieczeństwa systemu.

Rekomendacja

Zaleca się włączenie parametru weights_only=True w funkcji torch.load() w celu zminimalizowania ryzyka. Należy również weryfikować źródło plików punktów kontrolnych przed ich załadowaniem.

Powiązane podatności

Oryginalny opis (angielski, źródło NVD)

The flash-attention training framework thru commit e724e2588cbe754beb97cf7c011b5e7e34119e62 (2025-13-04) contains an insecure deserialization vulnerability (CWE-502) in its checkpoint loading mechanism. The load_checkpoint() function in checkpoint.py and the checkpoint loading code in eval.py use torch.load() without enabling the security-restrictive weights_only=True parameter. This allows the deserialization of arbitrary Python objects via the pickle module. An attacker can exploit this by providing a maliciously crafted checkpoint file. When a victim loads this checkpoint during model warmstarting or evaluation, arbitrary code is executed on the victim's system.

Dane podatności pochodzą z NVD (NIST) · CISA KEV · EPSS