Skip to content
OpenAdvisories
Medium · 5.5 CVE-2021-37685 BIT-tensorflow-2021-37685EUVD-2021-0411GHSA-c545-c4f9-rf6vPYSEC-2021-307PYSEC-2021-598PYSEC-2021-796

Heap OOB in TensorFlow Lite

Published Aug 12, 2021 · Updated Aug 4, 2024 · Assigned by GitHub_M

Description

TensorFlow is an end-to-end open source platform for machine learning. In affected versions TFLite's [`expand_dims.cc`](https://github.com/tensorflow/tensorflow/blob/149562d49faa709ea80df1d99fc41d005b81082a/tensorflow/lite/kernels/expand_dims.cc#L36-L50) contains a vulnerability which allows reading one element outside of bounds of heap allocated data. If `axis` is a large negative value (e.g., `-100000`), then after the first `if` it would still be negative. The check following the `if` statement will pass and the `for` loop would read one element before the start of `input_dims.data` (when `i = 0`). We have patched the issue in GitHub commit d94ffe08a65400f898241c0374e9edc6fa8ed257. The fix will be included in TensorFlow 2.6.0. We will also cherrypick this commit on TensorFlow 2.5.1, TensorFlow 2.4.3, and TensorFlow 2.3.4, as these are also affected and still in supported range.

Affected products

VendorProductAffectedFixed
tensorflow tensorflow >= 2.5.0, < 2.5.1, >= 2.4.0, < 2.4.3, < 2.3.4 —

Fixed in Linux distributions

Distributions backport fixes, so the version that fixes this vulnerability differs per release — upgrade to at least the version shown. From the distributions' security trackers and advisories.

ReleasePackageFixed inAdvisory
SUSE Package Hub 15 SP3 bazel-skylib1.0.3 1.0.3-bp153.2.1 openSUSE-SU-2022:10014-1
SUSE Package Hub 15 SP3 bazel3.7 3.7.2-bp153.4.1 openSUSE-SU-2022:10014-1
SUSE Package Hub 15 SP3 tensorflow2 +3 2.6.0-bp153.2.3.1 openSUSE-SU-2022:10014-1
ReleasePackageFixed inAdvisory
openSUSE Leap 15.3 bazel-skylib1.0.3 1.0.3-bp153.2.1 openSUSE-SU-2022:10014-1
openSUSE Leap 15.3 bazel3.7 3.7.2-bp153.4.1 openSUSE-SU-2022:10014-1
openSUSE Leap 15.3 tensorflow2 +3 2.6.0-bp153.2.3.1 openSUSE-SU-2022:10014-1

Affected packages

EcosystemPackageAffectedFixedRecord
PyPI tensorflow < 2.3.4 2.3.4 GHSA-c545-c4f9-rf6v
PyPI tensorflow >= 2.4.0, < 2.4.3 2.4.3 GHSA-c545-c4f9-rf6v
PyPI tensorflow >= 2.5.0, < 2.5.1 2.5.1 GHSA-c545-c4f9-rf6v
PyPI tensorflow >= 2.3.0, < 2.3.4 · >= 2.4.0, < 2.4.3 2.3.4, 2.4.3 PYSEC-2021-307
PyPI tensorflow-cpu < 2.3.4 2.3.4 GHSA-c545-c4f9-rf6v
PyPI tensorflow-cpu >= 2.4.0, < 2.4.3 2.4.3 GHSA-c545-c4f9-rf6v
PyPI tensorflow-cpu >= 2.5.0, < 2.5.1 2.5.1 GHSA-c545-c4f9-rf6v
PyPI tensorflow-cpu >= 2.3.0, < 2.3.4 · >= 2.4.0, < 2.4.3 2.3.4, 2.4.3 PYSEC-2021-598
PyPI tensorflow-gpu < 2.3.4 2.3.4 GHSA-c545-c4f9-rf6v
PyPI tensorflow-gpu >= 2.4.0, < 2.4.3 2.4.3 GHSA-c545-c4f9-rf6v
PyPI tensorflow-gpu >= 2.5.0, < 2.5.1 2.5.1 GHSA-c545-c4f9-rf6v
PyPI tensorflow-gpu >= 2.3.0, < 2.3.4 · >= 2.4.0, < 2.4.3 2.3.4, 2.4.3 PYSEC-2021-796

Advisories 1

References