🇰🇷 한글 설명
TensorFlow의 TFLite 연산자인 BatchToSpaceNd 구현에서 0으로 나누기 오류가 발생할 수 있는 취약점이 발견되었습니다. 공격자가 block 입력의 한 차원을 0으로 설정한 모델을 제작할 경우 프로그램이 비정상 종료될 수 있습니다. 이 문제는 TensorFlow 2.5.0 및 주요 이전 버전의 패치 버전을 통해 해결되었습니다.
💡 쉬운 설명
머신러닝 모델을 처리하는 과정에서 수학적으로 불가능한 '0으로 나누기' 계산이 발생하여 프로그램이 멈출 수 있는 문제입니다. 악의적으로 만들어진 모델 파일을 실행할 때 시스템이 갑자기 꺼지는 현상이 발생할 수 있습니다.
🇺🇸 원문 (English)
TensorFlow is an end-to-end open source platform for machine learning. The implementation of the `BatchToSpaceNd` TFLite operator is vulnerable to a division by zero error(https://github.com/tensorflow/tensorflow/blob/b5ed552fe55895aee8bd8b191f744a069957d18d/tensorflow/lite/kernels/batch_to_space_nd.cc#L81-L82). An attacker can craft a model such that one dimension of the `block` input is 0. Hence, the corresponding value in `block_shape` is 0. The fix will be included in TensorFlow 2.5.0. We will also cherrypick this commit on TensorFlow 2.4.2, TensorFlow 2.3.3, TensorFlow 2.2.3 and TensorFlow 2.1.4, as these are also affected and still in supported range.
📍 영향받는 시스템
TensorFlow 2.5.0 미만 버전 (2.4.2, 2.3.3, 2.2.3, 2.1.4 포함)
⚔️ 공격 벡터
0의 값을 포함하도록 조작된 특수 TensorFlow Lite 모델 파일을 시스템에 입력하여 실행
🛡️ 대응 방법
TensorFlow를 2.5.0 이상 버전으로 업데이트하거나, 사용 중인 버전의 보안 패치 버전으로 업데이트
🔧 검증 코드
출력된 버전 번호가 2.5.0 미만인지 확인하여 취약 여부를 판단합니다.
python3 -c 'import tensorflow as tf; print(tf.__version__)'