🇰🇷 한글 설명
PyTorch의 weights_only unpickler에 취약점이 있어, 악의적인 체크포인트 파일(.pth)을 생성하여 torch.load() 함수를 통해 로드할 때 메모리 손상 및 임의 코드 실행이 가능하다.
💡 쉬운 설명
PyTorch 라이브러리에 취약점이 있어, 악의적인 파일을 통해 시스템을 공격할 수 있습니다. PyTorch 버전 2.10.0 이전의 버전에서 발생합니다. 체크포인트 파일을 로드할 때 메모리 손상 및 임의 코드 실행이 가능합니다.
🇺🇸 원문 (English)
PyTorch is a Python package that provides tensor computation. Prior to version 2.10.0, a vulnerability in PyTorch's `weights_only` unpickler allows an attacker to craft a malicious checkpoint file (`.pth`) that, when loaded with `torch.load(..., weights_only=True)`, can corrupt memory and potentially lead to arbitrary code execution. Version 2.10.0 fixes the issue.
📍 영향받는 시스템
PyTorch 버전 2.10.0 이전
⚔️ 공격 벡터
악의적인 체크포인트 파일(.pth) 생성 및 torch.load() 함수를 통해 로드
🛡️ 대응 방법
PyTorch를 버전 2.10.0 이상으로 업데이트하여 취약점을 해결
🔧 검증 코드
Python에서 torch 라이브러리를 임포트하고, 버전을 확인하여 취약점 여부를 판단합니다. torch.__version__ 변수를 통해 PyTorch 버전을 확인할 수 있습니다. 버전이 2.10.0 미만인 경우 취약하다고 판단합니다.
import torch
print(torch.__version__)
if torch.__version__ < '2.10.0':
print({"result": "취약", "reason": "PyTorch 버전이 2.10.0 미만"})
else:
print({"result": "양호", "reason": "PyTorch 버전이 2.10.0 이상"})