CVE-2021-37635: Heap out of bounds access in sparse reduction operations in TensorFlow
Impact The implementation of sparse reduction operations in TensorFlow can trigger accesses outside of bounds of heap allocated data:
python import tensorflow as tf
x = tf.SparseTensor( indices=[[773, 773, 773], [773, 773, 773]], values=[1, 1], denseshape=[337, 337, 337]) tf.sparse.reducesum(x, 1)
The implementation fails to validate that each reduction group does not overflow and that each corresponding index does not point to outside the bounds of the input tensor.
Patches We have patched the issue in GitHub commit 87158f43f05f2720a374f3e6d22a7aaa3a33f750.
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.
For more information Please consult our security guide for more information regarding the security model and how to contact us with issues and questions.
Attribution This vulnerability has been reported by members of the Aivul Team from Qihoo 360.
Other sources
TensorFlow is an end-to-end open source platform for machine learning. In affected versions the implementation of sparse reduction operations in TensorFlow can trigger accesses outside of bounds of heap allocated data. The implementation fails to validate that each reduction group does not overflow and that each corresponding index does not point to outside the bounds of the input tensor. We have patched the issue in GitHub commit 87158f43f05f2720a374f3e6d22a7aaa3a33f750. 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 Software
Remediation
Recommended actions to resolve this vulnerability, in priority order.
- Upgrade
Upgrade
pip/tensorflow-gputo a version that resolves this vulnerability.Fixed in 2.5.1 - Upgrade
Upgrade
pip/tensorflow-gputo a version that resolves this vulnerability.Fixed in 2.4.3 - Upgrade
Upgrade
pip/tensorflow-gputo a version that resolves this vulnerability.Fixed in 2.3.4 - Upgrade
Upgrade
pip/tensorflow-cputo a version that resolves this vulnerability.Fixed in 2.5.1 - Upgrade
Upgrade
pip/tensorflow-cputo a version that resolves this vulnerability.Fixed in 2.4.3 - Upgrade
Upgrade
pip/tensorflow-cputo a version that resolves this vulnerability.Fixed in 2.3.4 - Upgrade
Upgrade
pip/tensorflowto a version that resolves this vulnerability.Fixed in 2.5.1 - Upgrade
Upgrade
pip/tensorflowto a version that resolves this vulnerability.Fixed in 2.4.3 - Upgrade
Upgrade
pip/tensorflowto a version that resolves this vulnerability.Fixed in 2.3.4 - Upgrade
Upgrade
TensorFlowto a version that resolves this vulnerability.Fixed in 2.6.0Patch 87158f43f05f2720a374f3e6d22a7aaa3a33f750 - Upgrade
Upgrade
TensorFlowto a version that resolves this vulnerability.Fixed in 2.5.1Patch 87158f43f05f2720a374f3e6d22a7aaa3a33f750 - Upgrade
Upgrade
TensorFlowto a version that resolves this vulnerability.Fixed in 2.4.3Patch 87158f43f05f2720a374f3e6d22a7aaa3a33f750 - Upgrade
Upgrade
TensorFlowto a version that resolves this vulnerability.Fixed in 2.3.4Patch 87158f43f05f2720a374f3e6d22a7aaa3a33f750
Event History
Frequently Asked Questions
What is the severity of CVE-2021-37635?
CVE-2021-37635 has been classified with a high severity level due to potential out-of-bounds access leading to security vulnerabilities.
How do I fix CVE-2021-37635?
To fix CVE-2021-37635, upgrade to TensorFlow version 2.5.1, 2.4.3, or 2.3.4.
Which versions of TensorFlow are affected by CVE-2021-37635?
CVE-2021-37635 affects TensorFlow versions between 2.3.0 and 2.3.4, versions between 2.4.0 and 2.4.3, as well as versions 2.5.0 and 2.6.0-rc0, rc1, and rc2.
What type of vulnerability is CVE-2021-37635?
CVE-2021-37635 is a memory safety vulnerability that can lead to out-of-bounds access in TensorFlow.
Who is affected by CVE-2021-37635?
Developers and users of TensorFlow versions listed in the vulnerability details are at risk from CVE-2021-37635.