BIT-tensorflow-2021-41197

Dashboard / Vulnerabilities / BIT-tensorflow-2021-41197

BIT-tensorflow-2021-41197

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

Details: 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.

References: https://github.com/tensorflow/tensorflow/commit/7c1692bd417eb4f9b33ead749a41166d6080af85, https://github.com/tensorflow/tensorflow/commit/a871989d7b6c18cdebf2fb4f0e5c5b62fbc19edf, https://github.com/tensorflow/tensorflow/commit/d81b1351da3e8c884ff836b64458d94e4a157c15, https://github.com/tensorflow/tensorflow/issues/46890, https://github.com/tensorflow/tensorflow/issues/51908, https://github.com/tensorflow/tensorflow/security/advisories/GHSA-prcg-wp5q-rv7p, https://nvd.nist.gov/vuln/detail/CVE-2021-41197