Skip to content
OpenAdvisories
Medium · 5.5 CVE-2021-41197 BIT-tensorflow-2021-41197EUVD-2021-0420GHSA-prcg-wp5q-rv7pPYSEC-2021-390PYSEC-2021-607PYSEC-2021-805

Crashes due to overflow and `CHECK`-fail in ops with large tensor shapes

Published Nov 5, 2021 · Updated Aug 4, 2024 · Assigned by GitHub_M

Description

TensorFlow is an open source platform for machine learning. In affected versions TensorFlow allows tensor to have a large number of dimensions and each dimension can be as large as desired. However, the total number of elements in a tensor must fit within an `int64_t`. If an overflow occurs, `MultiplyWithoutOverflow` would return a negative result. In the majority of TensorFlow codebase this then results in a `CHECK`-failure. Newer constructs exist which return a `Status` instead of crashing the binary. This is similar to CVE-2021-29584. The fix will be included in TensorFlow 2.7.0. We will also cherrypick this commit on TensorFlow 2.6.1, TensorFlow 2.5.2, and TensorFlow 2.4.4, as these are also affected and still in supported range.

Affected products

VendorProductAffectedFixed
tensorflow tensorflow >= 2.6.0, < 2.6.1, >= 2.5.0, < 2.5.2, < 2.4.4 —

Affected packages

EcosystemPackageAffectedFixedRecord
PyPI tensorflow >= 2.6.0, < 2.6.1 2.6.1 GHSA-prcg-wp5q-rv7p
PyPI tensorflow >= 2.5.0, < 2.5.2 2.5.2 GHSA-prcg-wp5q-rv7p
PyPI tensorflow < 2.4.4 2.4.4 GHSA-prcg-wp5q-rv7p
PyPI tensorflow < 2.4.4 · >= 2.5.0, < 2.5.2 · >= 2.6.0, < 2.6.1 · >= 2.7.0rc0, < 2.7.0 2.4.4, 2.5.2, 2.6.1, 2.7.0 PYSEC-2021-390
PyPI tensorflow-cpu >= 2.6.0, < 2.6.1 2.6.1 GHSA-prcg-wp5q-rv7p
PyPI tensorflow-cpu >= 2.5.0, < 2.5.2 2.5.2 GHSA-prcg-wp5q-rv7p
PyPI tensorflow-cpu < 2.4.4 2.4.4 GHSA-prcg-wp5q-rv7p
PyPI tensorflow-cpu < 2.4.4 · >= 2.5.0, < 2.5.2 · >= 2.6.0, < 2.6.1 · >= 2.7.0rc0, < 2.7.0 2.4.4, 2.5.2, 2.6.1, 2.7.0 PYSEC-2021-607
PyPI tensorflow-gpu >= 2.6.0, < 2.6.1 2.6.1 GHSA-prcg-wp5q-rv7p
PyPI tensorflow-gpu >= 2.5.0, < 2.5.2 2.5.2 GHSA-prcg-wp5q-rv7p
PyPI tensorflow-gpu < 2.4.4 2.4.4 GHSA-prcg-wp5q-rv7p
PyPI tensorflow-gpu < 2.4.4 · >= 2.5.0, < 2.5.2 · >= 2.6.0, < 2.6.1 · >= 2.7.0rc0, < 2.7.0 2.4.4, 2.5.2, 2.6.1, 2.7.0 PYSEC-2021-805

References